Quick update to support being behind cloudflare
This commit is contained in:
parent
1f5eeed126
commit
2373062a39
10
main.go
10
main.go
|
@ -19,7 +19,7 @@ func Logger() gin.HandlerFunc {
|
||||||
t := time.Now()
|
t := time.Now()
|
||||||
ip, err := net.ResolveTCPAddr("tcp", c.Request.RemoteAddr)
|
ip, err := net.ResolveTCPAddr("tcp", c.Request.RemoteAddr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.Abort(500)
|
c.Abort()
|
||||||
}
|
}
|
||||||
|
|
||||||
// before request
|
// before request
|
||||||
|
@ -53,8 +53,14 @@ func mainHandler(c *gin.Context) {
|
||||||
fields := strings.Split(c.Params.ByName("field"), ".")
|
fields := strings.Split(c.Params.ByName("field"), ".")
|
||||||
ip, err := net.ResolveTCPAddr("tcp", c.Request.RemoteAddr)
|
ip, err := net.ResolveTCPAddr("tcp", c.Request.RemoteAddr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.Abort(500)
|
c.Abort()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
cfIP := net.ParseIP(c.Request.Header.Get("CF-Connecting-IP"))
|
||||||
|
if cfIP != nil {
|
||||||
|
ip.IP = cfIP
|
||||||
|
}
|
||||||
|
|
||||||
c.Set("ip", ip.IP.String())
|
c.Set("ip", ip.IP.String())
|
||||||
c.Set("port", ip.Port)
|
c.Set("port", ip.Port)
|
||||||
c.Set("ua", c.Request.UserAgent())
|
c.Set("ua", c.Request.UserAgent())
|
||||||
|
|
Loading…
Reference in New Issue