golang框架在高并发场景中的中间件选择与使用

import (
    "github.com/gin-gonic/gin"
    "go.uber.org/zap"
    "time"
)

func main() {
    // 创建 Zap logger
    logger, _ := zap.NewProduction()
    defer logger.Sync()

    // 创建 Gin 路由器
    router := gin.Default()
    
    // 注册 Zap 日志记录中间件
    loggerMiddleware := zapLoggerMiddleware(logger)
    router.Use(loggerMiddleware)

    // 注册用户注册处理程序
    router.POST("/register", registerUser)
    
    // 启动服务
    router.Run(":8080")
}

// registerUser 处理用户注册请求
func registerUser(c *gin.Context) {
    user := User{}
    err := c.BindJSON(&user)
    if err != nil {
        c.JSON(400, gin.H{"error": err.Error()})
        return
    }
    
    // 将用户保存到数据库

    c.JSON(200, gin.H{"status": "success"})
}

// zapLoggerMiddleware 是一个 Gin 中间件，使用 Zap 记录请求和响应信息
func zapLoggerMiddleware(logger *zap.Logger) gin.HandlerFunc {
    return func(c *gin.Context) {
        start := time.Now()
        
        c.Next()
        
        latency := time.Since(start)
        logger.Info("handled request",
            zap.String("method", c.Request.Method),
            zap.String("path", c.Request.URL.Path),
            zap.Int("status", c.Writer.Status()),
            zap.Duration("latency", latency))
    }
}