更改 ip 获取位置

config/config.go

@@ -10,12 +10,11 @@ type Config struct {
 	Sql struct {
 		MysqlDsn string
 	}
-	Node   int64
-	Epoch  int64
 	Debug  bool
 	JwtKey string
 	Cache  struct {
 		Type string
 		Ram  int
 	}
+	RaelIP bool
 }

server/server.go

@@ -21,7 +21,7 @@ func NewServer(c config.Config, sl *slog.Logger, route *httprouter.Router) (*htt
 		Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 			ctx := r.Context()
 			if sl.Enabled(ctx, slog.LevelInfo) {
-				ip, _ := utils.GetIP(r)
+				ip, _ := utils.GetIP(r, c.RaelIP)
 				trackid.Add(1)
 				ctx = setCtx(ctx, &reqInfo{
 					URL:     r.URL.String(),

utils/ip.go

@@ -7,7 +7,8 @@ import (
 	"strings"
 )
 
-func GetIP(r *http.Request) (string, error) {
+func GetIP(r *http.Request, fromHeader bool) (string, error) {
+	if fromHeader {
 		//Get IP from the X-REAL-IP header
 		ip := r.Header.Get("X-REAL-IP")
 		netIP := net.ParseIP(ip)
@@ -24,13 +25,14 @@ func GetIP(r *http.Request) (string, error) {
 				return ip, nil
 			}
 		}
+	}
 
 	//Get IP from RemoteAddr
 	ip, _, err := net.SplitHostPort(r.RemoteAddr)
 	if err != nil {
 		return "", err
 	}
-	netIP = net.ParseIP(ip)
+	netIP := net.ParseIP(ip)
 	if netIP != nil {
 		return ip, nil
 	}