context.go

@@ -105,17 +105,14 @@ func (c *Context) Get(key string) interface{} {
 
 // RealIP uses proxy headers for the real ip, if none exist the IP of the current connection is returned
 func (c *Context) RealIP() string {
-	reqIP := c.Request.RemoteAddr
-
 	if ip := c.Request.Header.Get(`X-Forwarded-For`); ip != `` {
-		reqIP = strings.Split(ip, `, `)[0]
-	} else if ip := c.Request.Header.Get(`X-Real-IP`); ip != `` {
-		reqIP = ip
+		return strings.Split(ip, `, `)[0]
 	}
 
-	ra, _, _ := net.SplitHostPort(reqIP)
-	if ra != `` {
-		reqIP = ra
+	if ip := c.Request.Header.Get(`X-Real-IP`); ip != `` {
+		return ip
 	}
-	return reqIP
+
+	ra, _, _ := net.SplitHostPort(c.Request.RemoteAddr)
+	return ra
 }