wood/random-api-go
1
0
Fork 0
mirror of https://github.com/woodchen-ink/random-api-go.git synced 2025-07-18 13:52:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加来源域名日志及错误日志格式化输出

Browse Source
This commit is contained in:
wood chen 2024-10-14 12:19:03 +08:00
parent c4bb234c2c
commit 485594497c
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
main.go
View File

@ -8,6 +8,7 @@ import (
"math/rand" "math/rand"
"net" "net"
"net/http" "net/http"
"net/url"
"os" "os"
"path/filepath" "path/filepath"
"strings" "strings"
@ -196,11 +197,23 @@ func getCSVContent(path string) (*URLSelector, error) {
func handleAPIRequest(w http.ResponseWriter, r *http.Request) { func handleAPIRequest(w http.ResponseWriter, r *http.Request) {
start := time.Now() start := time.Now()
realIP := getRealIP(r) realIP := getRealIP(r)
referer := r.Referer()
// 获取来源域名
var sourceDomain string
if referer != "" {
if parsedURL, err := url.Parse(referer); err == nil {
sourceDomain = parsedURL.Hostname()
}
}
if sourceDomain == "" {
sourceDomain = "direct"
}
if time.Since(lastFetchTime) > cacheDuration { if time.Since(lastFetchTime) > cacheDuration {
if err := loadCSVPaths(); err != nil { if err := loadCSVPaths(); err != nil {
http.Error(w, "Failed to load CSV paths", http.StatusInternalServerError) http.Error(w, "Failed to load CSV paths", http.StatusInternalServerError)
log.Println("Error loading CSV paths:", err) log.Printf("Error loading CSV paths: %v", err)
return return
} }
} }
@ -228,7 +241,7 @@ func handleAPIRequest(w http.ResponseWriter, r *http.Request) {
selector, err := getCSVContent(csvPath) selector, err := getCSVContent(csvPath)
if err != nil { if err != nil {
http.Error(w, "Failed to fetch CSV content", http.StatusInternalServerError) http.Error(w, "Failed to fetch CSV content", http.StatusInternalServerError)
log.Println("Error fetching CSV content:", err) log.Printf("Error fetching CSV content: %v", err)
return return
} }
@ -240,8 +253,8 @@ func handleAPIRequest(w http.ResponseWriter, r *http.Request) {
randomURL := selector.GetRandomURL() randomURL := selector.GetRandomURL()
duration := time.Since(start) duration := time.Since(start)
log.Printf("Request: %s %s from %s - Duration: %v - Redirecting to: %s\n", log.Printf("Request: %s %s from %s - Source: %s - Duration: %v - Redirecting to: %s",
r.Method, r.URL.Path, realIP, duration, randomURL) r.Method, r.URL.Path, realIP, sourceDomain, duration, randomURL)
http.Redirect(w, r, randomURL, http.StatusFound) http.Redirect(w, r, randomURL, http.StatusFound)
} }