feat: refactor application to use Gin framework, update routing and middleware handling

This commit is contained in:
王锦强
2025-03-15 12:39:17 +08:00
parent cab289dabf
commit 6fa5c1f467
13 changed files with 314 additions and 279 deletions

View File

@@ -2,45 +2,27 @@ package middleware
import (
"log"
"net/http"
"time"
"github.com/gin-gonic/gin"
)
// Logger 是一个HTTP中间件记录请求的详细信息
func Logger(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
func Logger() gin.HandlerFunc {
return func(c *gin.Context) {
start := time.Now()
// 包装ResponseWriter以捕获状态码
wrapper := &responseWriterWrapper{
ResponseWriter: w,
statusCode: http.StatusOK,
}
// 调用下一个处理器
next.ServeHTTP(wrapper, r)
// 处理请求
c.Next()
// 记录请求信息
duration := time.Since(start)
log.Printf(
"[%s] %s %s %d %s",
r.Method,
r.RequestURI,
r.RemoteAddr,
wrapper.statusCode,
log.Printf("[%s] %s %s %d %s",
c.Request.Method,
c.Request.URL.Path,
c.ClientIP(),
c.Writer.Status(),
duration,
)
})
}
// responseWriterWrapper 包装http.ResponseWriter以捕获状态码
type responseWriterWrapper struct {
http.ResponseWriter
statusCode int
}
// WriteHeader 捕获状态码
func (w *responseWriterWrapper) WriteHeader(statusCode int) {
w.statusCode = statusCode
w.ResponseWriter.WriteHeader(statusCode)
}
}