Use the gin framework

middleware/logging.go

@@ -0,0 +1,97 @@
+package middleware
+
+import (
+	"strings"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"networkDev/utils/logger"
+)
+
+// LoggingMiddleware 日志记录中间件结构体
+// 用于记录HTTP请求的详细信息，包括方法、路径、状态码和响应时间
+type LoggingMiddleware struct {
+	logger *logger.Logger
+}
+
+// NewLoggingMiddleware 创建新的日志记录中间件实例
+func NewLoggingMiddleware(logger *logger.Logger) *LoggingMiddleware {
+	return &LoggingMiddleware{
+		logger: logger,
+	}
+}
+
+// Handler 返回Gin中间件函数，用于记录HTTP请求日志
+// 记录格式遵循Apache Common Log Format
+func (lm *LoggingMiddleware) Handler() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		// 记录开始时间
+		start := time.Now()
+
+		// 处理请求
+		c.Next()
+
+		// 计算处理时间
+		duration := time.Since(start)
+
+		// 获取客户端IP
+		clientIP := getClientIP(c)
+
+		// 记录日志 - Apache Common Log Format
+		// 使用专门的HTTP日志方法避免User-Agent中的反斜杠被转义
+		lm.logger.LogRequestWithHeaders(
+			c.Request.Method,
+			c.Request.RequestURI,
+			clientIP,
+			c.Writer.Status(),
+			duration,
+			"-", // referer (已废弃)
+			c.Request.UserAgent(),
+		)
+	}
+}
+
+// getClientIP 获取客户端真实IP地址
+// 优先从X-Forwarded-For、X-Real-IP等头部获取，最后使用RemoteAddr
+func getClientIP(c *gin.Context) string {
+	// 检查X-Forwarded-For头部
+	xForwardedFor := c.GetHeader("X-Forwarded-For")
+	if xForwardedFor != "" {
+		// X-Forwarded-For可能包含多个IP，取第一个
+		ips := strings.Split(xForwardedFor, ",")
+		if len(ips) > 0 {
+			return strings.TrimSpace(ips[0])
+		}
+	}
+
+	// 检查X-Real-IP头部
+	xRealIP := c.GetHeader("X-Real-IP")
+	if xRealIP != "" {
+		return xRealIP
+	}
+
+	// 检查X-Forwarded头部
+	xForwarded := c.GetHeader("X-Forwarded")
+	if xForwarded != "" {
+		return xForwarded
+	}
+
+	// 使用RemoteAddr
+	remoteAddr := c.Request.RemoteAddr
+	if strings.Contains(remoteAddr, ":") {
+		// 移除端口号
+		if idx := strings.LastIndex(remoteAddr, ":"); idx != -1 {
+			return remoteAddr[:idx]
+		}
+	}
+
+	return remoteAddr
+}
+
+// WrapHandler 创建Gin日志中间件
+// 使用全局日志记录器创建日志中间件
+func WrapHandler() gin.HandlerFunc {
+	logger := logger.GetLogger()
+	middleware := NewLoggingMiddleware(logger)
+	return middleware.Handler()
+}