69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package routes
|
|
|
|
import (
|
|
"skeleton/config"
|
|
"skeleton/middlewares"
|
|
"skeleton/routes/rest"
|
|
"skeleton/routes/ws"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"go.opentelemetry.io/contrib/instrumentation/github.com/gin-gonic/gin/otelgin"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// SetupRoutes 设置主路由
|
|
func SetupRoutes(cfg *config.Config) *gin.Engine {
|
|
// 根据配置设置 Gin 模式
|
|
if !cfg.App.Debug {
|
|
gin.SetMode(gin.ReleaseMode)
|
|
}
|
|
|
|
// 创建 Gin 路由器（不使用默认中间件）
|
|
r := gin.New()
|
|
|
|
// 设置 multipart form 内存限制为 100MB
|
|
r.MaxMultipartMemory = 100 << 20 // 100 MB
|
|
|
|
// 添加自定义中间件
|
|
r.Use(middlewares.RequestID())
|
|
if cfg.Observability.TracingEnabled {
|
|
r.Use(otelgin.Middleware(cfg.Observability.TracingService))
|
|
}
|
|
if cfg.Observability.MetricsEnabled {
|
|
r.Use(middlewares.Metrics())
|
|
}
|
|
r.Use(middlewares.CORS()) // CORS跨域处理（需要在其他中间件之前）
|
|
r.Use(middlewares.GinLogger()) // 结构化日志
|
|
r.Use(middlewares.GinRecovery()) // 异常恢复
|
|
r.Use(middlewares.ErrorLogging()) // 错误响应日志（用于记录逻辑异常）
|
|
|
|
if cfg.Observability.MetricsEnabled {
|
|
r.GET(cfg.Observability.MetricsPath, gin.WrapH(promhttp.Handler()))
|
|
}
|
|
|
|
// API 路由组
|
|
api := r.Group("/api")
|
|
|
|
// 公开路由组（无需认证）
|
|
public := api.Group("")
|
|
rest.ApplyPublic(public)
|
|
|
|
// 私有路由组（需要 JWT 认证）
|
|
private := api.Group("/private")
|
|
private.Use(middlewares.JWTAuth()) // 启用JWT认证中间件
|
|
rest.ApplyPrivate(private)
|
|
|
|
// WebSocket 路由组（无需认证）
|
|
ws.ApplyWebSocketRoutes(api)
|
|
|
|
// 打印路由统计信息
|
|
rest.PrintStats()
|
|
|
|
zap.L().Info("主路由设置完成",
|
|
zap.Any("rest_stats", rest.GetRegistrarStats()),
|
|
)
|
|
|
|
return r
|
|
}
|