Update backend/pkg/commons/ratelimit/ratelimit.go

Co-authored-by: Lucca <[email protected]>
gobitfly · Jul 31, 2024 · faaa65f · faaa65f
1 parent f599336
commit faaa65f
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/backend/pkg/commons/ratelimit/ratelimit.go b/backend/pkg/commons/ratelimit/ratelimit.go
@@ -944,10 +944,7 @@ func getRoute(r *http.Request) string {
 
 // getIP returns the ip address from the http request
 func getIP(r *http.Request) string {
-	ips := r.Header.Get("CF-Connecting-IP")
-	if ips == "" {
-		ips = r.Header.Get("X-Forwarded-For")
-	}
+	ips := cmp.Or(r.Header.Get("CF-Connecting-IP"), r.Header.Get("X-Forwarded-For"))
 	splitIps := strings.Split(ips, ",")
 
 	if len(splitIps) > 0 {