feat: added GetAddr func

pkgz · Jul 21, 2024 · ee52b7d · ee52b7d
1 parent 0f1c948
commit ee52b7d
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/request.go b/request.go
@@ -32,3 +32,17 @@ func ReadBody(r *http.Request, str interface{}) error {
 
 	return nil
 }
+
+func GetAddr(r *http.Request) string {
+	addr := r.RemoteAddr
+	if CFAddr := r.Header.Get("CF-Connecting-IP"); CFAddr != "" {
+		addr = CFAddr
+	}
+	if addr == "" {
+		addr = r.Header.Get("X-Forwarded-For")
+	}
+	if addr == "" {
+		addr = r.Header.Get("X-Real-Ip")
+	}
+	return addr
+}