mirror of
https://github.com/ctwj/urldb.git
synced 2025-11-25 03:15:04 +08:00
115 lines
2.6 KiB
Go
115 lines
2.6 KiB
Go
package middleware
|
||
|
||
import (
|
||
"bytes"
|
||
"io"
|
||
"net/http"
|
||
"time"
|
||
|
||
"github.com/ctwj/urldb/utils"
|
||
)
|
||
|
||
// responseWriter 包装http.ResponseWriter以捕获响应状态码和内容
|
||
type responseWriter struct {
|
||
http.ResponseWriter
|
||
statusCode int
|
||
body bytes.Buffer
|
||
}
|
||
|
||
func (rw *responseWriter) WriteHeader(code int) {
|
||
rw.statusCode = code
|
||
rw.ResponseWriter.WriteHeader(code)
|
||
}
|
||
|
||
func (rw *responseWriter) Write(b []byte) (int, error) {
|
||
rw.body.Write(b)
|
||
return rw.ResponseWriter.Write(b)
|
||
}
|
||
|
||
// LoggingMiddleware HTTP请求日志中间件
|
||
func LoggingMiddleware(next http.Handler) http.Handler {
|
||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
start := time.Now()
|
||
|
||
// 包装ResponseWriter
|
||
rw := &responseWriter{
|
||
ResponseWriter: w,
|
||
statusCode: 200, // 默认状态码
|
||
}
|
||
|
||
// 读取请求体
|
||
var requestBody []byte
|
||
if r.Body != nil {
|
||
requestBody, _ = io.ReadAll(r.Body)
|
||
r.Body = io.NopCloser(bytes.NewBuffer(requestBody))
|
||
}
|
||
|
||
// 处理请求
|
||
next.ServeHTTP(rw, r)
|
||
|
||
// 计算处理时间
|
||
duration := time.Since(start)
|
||
|
||
// 记录请求日志
|
||
logRequest(r, rw, duration, requestBody)
|
||
})
|
||
}
|
||
|
||
// logRequest 记录请求日志
|
||
func logRequest(r *http.Request, rw *responseWriter, duration time.Duration, requestBody []byte) {
|
||
// 获取客户端IP
|
||
clientIP := getClientIP(r)
|
||
|
||
// 获取用户代理
|
||
userAgent := r.UserAgent()
|
||
if userAgent == "" {
|
||
userAgent = "Unknown"
|
||
}
|
||
|
||
// 记录请求信息
|
||
utils.Info("HTTP请求 - %s %s - IP: %s - User-Agent: %s - 状态码: %d - 耗时: %v",
|
||
r.Method, r.URL.Path, clientIP, userAgent, rw.statusCode, duration)
|
||
|
||
// 如果是错误状态码，记录详细信息
|
||
if rw.statusCode >= 400 {
|
||
utils.Error("HTTP错误 - %s %s - 状态码: %d - 响应体: %s",
|
||
r.Method, r.URL.Path, rw.statusCode, rw.body.String())
|
||
}
|
||
|
||
// 记录请求参数（仅对POST/PUT请求）
|
||
if (r.Method == "POST" || r.Method == "PUT") && len(requestBody) > 0 {
|
||
// 限制日志长度，避免日志文件过大
|
||
if len(requestBody) > 1000 {
|
||
utils.Debug("请求体(截断): %s...", string(requestBody[:1000]))
|
||
} else {
|
||
utils.Debug("请求体: %s", string(requestBody))
|
||
}
|
||
}
|
||
|
||
// 记录查询参数
|
||
if len(r.URL.RawQuery) > 0 {
|
||
utils.Debug("查询参数: %s", r.URL.RawQuery)
|
||
}
|
||
}
|
||
|
||
// getClientIP 获取客户端真实IP地址
|
||
func getClientIP(r *http.Request) string {
|
||
// 检查X-Forwarded-For头
|
||
if ip := r.Header.Get("X-Forwarded-For"); ip != "" {
|
||
return ip
|
||
}
|
||
|
||
// 检查X-Real-IP头
|
||
if ip := r.Header.Get("X-Real-IP"); ip != "" {
|
||
return ip
|
||
}
|
||
|
||
// 检查X-Client-IP头
|
||
if ip := r.Header.Get("X-Client-IP"); ip != "" {
|
||
return ip
|
||
}
|
||
|
||
// 返回远程地址
|
||
return r.RemoteAddr
|
||
}
|