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

refactor(api_handler): improve source info extraction and logging

Browse Source
This commit is contained in:
wood chen 2024-10-31 01:08:30 +08:00
parent f68fa981ce
commit 887e8a7ca8
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
handlers/api_handler.go
View File

@ -27,15 +27,18 @@ func HandleAPIRequest(w http.ResponseWriter, r *http.Request) {
realIP := utils.GetRealIP(r)
referer := r.Referer()
var sourceDomain string
// 修改这部分,获取完整的referer信息
sourceInfo := "direct"
if referer != "" {
if parsedURL, err := url.Parse(referer); err == nil {
sourceDomain = parsedURL.Hostname()
// 包含主机名和路径
sourceInfo = parsedURL.Host + parsedURL.Path
// 如果有查询参数,也可以加上
if parsedURL.RawQuery != "" {
sourceInfo += "?" + parsedURL.RawQuery
}
}
}
if sourceDomain == "" {
sourceDomain = "direct"
}
path := strings.TrimPrefix(r.URL.Path, "/")
pathSegments := strings.Split(path, "/")
@ -77,7 +80,7 @@ func HandleAPIRequest(w http.ResponseWriter, r *http.Request) {
duration := time.Since(start)
log.Printf("请求：%s %s，来自 %s -来源：%s -持续时间: %v - 重定向至: %s",
r.Method, r.URL.Path, realIP, sourceDomain, duration, randomURL)
r.Method, r.URL.Path, realIP, sourceInfo, duration, randomURL)
http.Redirect(w, r, randomURL, http.StatusFound)
}