diff --git a/context.go b/context.go index e9c2fd6..eec329c 100644 --- a/context.go +++ b/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 }