Update to get real IP from X-Real-IP as well.

drkaka · Jul 3, 2017 · f326494 · f326494
1 parent 2da913e
commit f326494
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/glide.lock b/glide.lock
diff --git a/lg.go b/lg.go
@@ -74,9 +74,14 @@ func LogRequest(next http.Handler) http.Handler {
 		lw := &basicWriter{ResponseWriter: w}
 
 		// Get the real IP even behind a proxy
-		realIP := r.Header.Get("X-Forwarded-For")
-		if len(realIP) == 0 {
-			realIP = r.RemoteAddr
+		realIP := r.Header.Get(http.CanonicalHeaderKey("X-Forwarded-For"))
+		if realIP == "" {
+			// if no content in header "X-Forwarded-For", get "X-Real-IP"
+			if xrip := r.Header.Get(http.CanonicalHeaderKey("X-Real-IP")); xrip != "" {
+				realIP = xrip
+			} else {
+				realIP = r.RemoteAddr
+			}
 		}
 
 		// Process request