golang/hotime
4
0
Fork 1
Code Issues Pull Requests Releases 66 Wiki Activity

IP获取真实条件修改

Browse Source
This commit is contained in:
hoteas 2022-09-02 00:04:23 +08:00
parent f59e909d30
commit b6ba3486d6
2 changed files with 21 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
application.go
View File

@ -357,14 +357,18 @@ func (that *Application) handler(w http.ResponseWriter, req *http.Request) {
defer func() { defer func() {
//是否展示日志 //是否展示日志
if that.WebConnectLog != nil { if that.WebConnectLog != nil {
ipStr := Substr(context.Req.RemoteAddr, 0, strings.Index(context.Req.RemoteAddr, ":"))
//负载均衡优化 //负载均衡优化
if ipStr == "127.0.0.1" { ipStr := ""
if req.Header.Get("X-Forwarded-For") != "" { if req.Header.Get("X-Forwarded-For") != "" {
ipStr = req.Header.Get("X-Forwarded-For") ipStr = req.Header.Get("X-Forwarded-For")
} else if req.Header.Get("X-Real-IP") != "" { } else if req.Header.Get("X-Real-IP") != "" {
ipStr = req.Header.Get("X-Real-IP") ipStr = req.Header.Get("X-Real-IP")
} }
//负载均衡优化
if ipStr == "" {
//RemoteAddr := that.Req.RemoteAddr
ipStr = Substr(context.Req.RemoteAddr, 0, strings.Index(context.Req.RemoteAddr, ":"))
} }
that.WebConnectLog.Infoln(ipStr, context.Req.Method, that.WebConnectLog.Infoln(ipStr, context.Req.Method,

17
context.go
View File

@ -75,14 +75,17 @@ func (that *Context) View() {
if that.Session("user_id").Data != nil { if that.Session("user_id").Data != nil {
that.Log["user_id"] = that.Session("user_id").ToCeilInt() that.Log["user_id"] = that.Session("user_id").ToCeilInt()
} }
ipStr := Substr(that.Req.RemoteAddr, 0, strings.Index(that.Req.RemoteAddr, ":"))
//负载均衡优化 //负载均衡优化
if ipStr == "127.0.0.1" { ipStr := ""
if that.Req.Header.Get("X-Forwarded-For") != "" { if that.Req.Header.Get("X-Forwarded-For") != "" {
ipStr = that.Req.Header.Get("X-Forwarded-For") ipStr = that.Req.Header.Get("X-Forwarded-For")
} else if that.Req.Header.Get("X-Real-IP") != "" { } else if that.Req.Header.Get("X-Real-IP") != "" {
ipStr = that.Req.Header.Get("X-Real-IP") ipStr = that.Req.Header.Get("X-Real-IP")
} }
//负载均衡优化
if ipStr == "" {
//RemoteAddr := that.Req.RemoteAddr
ipStr = Substr(that.Req.RemoteAddr, 0, strings.Index(that.Req.RemoteAddr, ":"))
} }
that.Log["ip"] = ipStr that.Log["ip"] = ipStr
that.Db.Insert("logs", that.Log) that.Db.Insert("logs", that.Log)