51 lines
931 B
Go
51 lines
931 B
Go
package models
|
||
|
||
import (
|
||
"net"
|
||
"strings"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// GetRealIP 获取真实IP（处理代理）
|
||
func GetRealIP(c *gin.Context) string {
|
||
// 优先级顺序
|
||
headers := []string{
|
||
"CF-Connecting-IP", // Cloudflare
|
||
"True-Client-IP",
|
||
"X-Forwarded-For",
|
||
"X-Real-IP",
|
||
}
|
||
|
||
for _, header := range headers {
|
||
if ip := c.GetHeader(header); ip != "" {
|
||
// 处理多个IP的情况（如 X-Forwarded-For: client, proxy1, proxy2）
|
||
if strings.Contains(ip, ",") {
|
||
ips := strings.Split(ip, ",")
|
||
ip = strings.TrimSpace(ips[0])
|
||
}
|
||
|
||
if net.ParseIP(ip) != nil {
|
||
return ip
|
||
}
|
||
}
|
||
}
|
||
|
||
// 最后使用Gin的ClientIP方法
|
||
return c.ClientIP()
|
||
}
|
||
|
||
func LogAdd(c *gin.Context, msg string) {
|
||
|
||
var logtemp APIRequestLog_
|
||
|
||
logtemp.IPAddress = GetRealIP(c)
|
||
logtemp.Path = c.Request.URL.Path
|
||
logtemp.Method = c.Request.Method
|
||
logtemp.Message = msg
|
||
|
||
//fmt.Println(logtemp)
|
||
DB.Create(&logtemp)
|
||
|
||
}
|