up

backend/models/log.go

@@ -1,17 +1,50 @@
 package models
 
-import "time"
+import (
+	"net"
+	"strings"
 
-type APIRequestLog struct {
-	ID         int64     `gorm:"primaryKey;column:id" json:"id"`
-	IPAddress  string    `gorm:"column:ip_address;size:45;not null" json:"ip_address"`
-	Path       string    `gorm:"column:path;size:500;not null" json:"path"`
-	Method     string    `gorm:"column:method;size:10;not null" json:"method"`
-	StatusCode int       `gorm:"column:status_code;index" json:"status_code"`
-	Message    string    `gorm:"column:error_message;type:text" json:"error_message"`
-	CreatedAt  time.Time `gorm:"column:created_at;default:CURRENT_TIMESTAMP" json:"created_at"`
+	"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() {
+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)
 
 }