hoteas/iedc-go
1
0
Fork 0
forked from golang/hotime
Code Issues Pull Requests Projects Releases Wiki Activity

增加记录IP

Browse Source
This commit is contained in:
hoteas 2022-08-20 22:38:15 +08:00
parent 78420a692f
commit 6505594fd0
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
context.go
View File

@ -5,6 +5,7 @@ import (
. "code.hoteas.com/golang/hotime/db" . "code.hoteas.com/golang/hotime/db"
"encoding/json" "encoding/json"
"net/http" "net/http"
"strings"
"time" "time"
) )
@ -74,6 +75,16 @@ 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" {
if that.Req.Header.Get("X-Forwarded-For") != "" {
ipStr = that.Req.Header.Get("X-Forwarded-For")
} else if that.Req.Header.Get("X-Real-IP") != "" {
ipStr = that.Req.Header.Get("X-Real-IP")
}
}
that.Log["ip"] = ipStr
that.Db.Insert("logs", that.Log) that.Db.Insert("logs", that.Log)
} }