nyancat/ip-checker
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore(go): simplify sourceIP logic

Browse source
This commit is contained in:
Guanran Wang 2024-10-19 16:18:59 +08:00
parent 796b0b3839
commit 265c998df0
Signed by: nyancat
GPG key ID: 91F97D9ED12639CF
1 changed files with 1 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
main.go
View file

@ -80,11 +80,8 @@ func getIPCountry(ip string) string {
func handleRequest(w http.ResponseWriter, r *http.Request) { func handleRequest(w http.ResponseWriter, r *http.Request) {
sourceIP := func() string { sourceIP := func() string {
if ip := r.Header.Get("CF-Connecting-IP"); ip != "" {
return ip
}
if ip := r.Header.Get("X-Forwarded-For"); ip != "" { if ip := r.Header.Get("X-Forwarded-For"); ip != "" {
return ip return strings.Split(ip, ",")[0]
} }
return strings.Split(r.RemoteAddr, ":")[0] return strings.Split(r.RemoteAddr, ":")[0]
}() }()