Simplify REALIP method (labstack#1193)

* Simplify REALIP method * Remove alloc
23doors · Sep 19, 2018 · a2d4cb9 · a2d4cb9
1 parent c185ebd
commit a2d4cb9
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/context.go b/context.go
@@ -256,14 +256,13 @@ func (c *context) Scheme() string {
 }
 
 func (c *context) RealIP() string {
-	ra := c.request.RemoteAddr
 	if ip := c.request.Header.Get(HeaderXForwardedFor); ip != "" {
-		ra = strings.Split(ip, ", ")[0]
-	} else if ip := c.request.Header.Get(HeaderXRealIP); ip != "" {
-		ra = ip
-	} else {
-		ra, _, _ = net.SplitHostPort(ra)
+		return strings.Split(ip, ", ")[0]
+	}
+	if ip := c.request.Header.Get(HeaderXRealIP); ip != "" {
+		return ip
 	}
+	ra, _, _ := net.SplitHostPort(c.request.RemoteAddr)
 	return ra
 }