mirror of
https://github.com/skyle1995/NetworkAuth.git
synced 2026-05-25 02:24:05 +08:00
98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
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()
|
||
}
|