diff --git a/cmd/server/main.go b/cmd/server/main.go index 436c213..e8476b1 100644 --- a/cmd/server/main.go +++ b/cmd/server/main.go @@ -105,7 +105,7 @@ func main() { zap.String("region", cfg.S3.Region), ) - engine := router.New(log, db, searchClient, rdb, cfg.Equalize, s3Storage, cfg.ShareCodeMaxPerMac, time.Duration(cfg.ShareCodeTTLMin)*time.Minute) + engine := router.New(log, db, searchClient, rdb, cfg.Equalize, s3Storage, cfg.ShareCodeMaxPerMac, time.Duration(cfg.ShareCodeTTLMin)*time.Minute, cfg.Env) srv := &http.Server{ Addr: cfg.Addr(), diff --git a/internal/router/router.go b/internal/router/router.go index 9699fa6..33daa99 100644 --- a/internal/router/router.go +++ b/internal/router/router.go @@ -19,7 +19,7 @@ import ( ginSwagger "github.com/swaggo/gin-swagger" ) -func New(log *zap.Logger, db *sql.DB, searchClient *search.Client, rdb *redis.Client, eqCfg config.EqualizeConfig, s3 *storage.S3Storage, shareCodeMaxPerMac int, shareCodeTTL time.Duration) *gin.Engine { +func New(log *zap.Logger, db *sql.DB, searchClient *search.Client, rdb *redis.Client, eqCfg config.EqualizeConfig, s3 *storage.S3Storage, shareCodeMaxPerMac int, shareCodeTTL time.Duration, env string) *gin.Engine { r := gin.New() r.Use(gin.Recovery()) r.Use(middleware.RequestID()) @@ -55,8 +55,10 @@ func New(log *zap.Logger, db *sql.DB, searchClient *search.Client, rdb *redis.Cl v1.GET("/health", health.Check) } - // Swagger - r.GET("/docs/*any", ginSwagger.WrapHandler(swaggerFiles.Handler)) + // Swagger — 仅非生产环境启用 + if env != "production" { + r.GET("/docs/*any", ginSwagger.WrapHandler(swaggerFiles.Handler)) + } audio := r.Group("/audio") {