feat: swagger switch.

2026-05-04 18:36:09 +08:00
parent 062ed63bb5
commit 30c8094f0f
3 changed files with 24 additions and 18 deletions
@@ -4,6 +4,7 @@ import (
 	"github.com/gin-gonic/gin"
 	swaggerFiles "github.com/swaggo/files"
 	ginSwagger "github.com/swaggo/gin-swagger"
+	"hr_receiver/config"
 	"hr_receiver/controllers"
 	_ "hr_receiver/docs"
 	"hr_receiver/middleware"
@@ -49,15 +50,15 @@ func SetupRouter() *gin.Engine {

 	r.GET("/auth/token", deviceTokenHandler)

-	// Swagger API文档 (swaggo/gin-swagger，持久化鉴权信息)
-	r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler,
-		ginSwagger.PersistAuthorization(true),
-	))
+	// 开发模式下注册 API 文档路由
+	if config.App.Swagger.Enabled {
+		r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler,
+			ginSwagger.PersistAuthorization(true),
+		))

-	// Scalar 精美API文档
-	r.GET("/scalar", func(c *gin.Context) {
-		c.Header("Content-Type", "text/html; charset=utf-8")
-		c.String(http.StatusOK, `<!doctype html>
+		r.GET("/scalar", func(c *gin.Context) {
+			c.Header("Content-Type", "text/html; charset=utf-8")
+			c.String(http.StatusOK, `<!doctype html>
 <html>
 <head>
    <title>API Reference</title>
@@ -71,16 +72,14 @@ func SetupRouter() *gin.Engine {
    <div id="app"></div>
    <script src="https://cdn.jsdelivr.net/npm/@scalar/api-reference"></script>
    <script>
-		Scalar.createApiReference('#app', {
-    spec: {
-        url: '/swagger/doc.json',
-    },
- persistAuth: true,
-})
+        Scalar.createApiReference('#app', {
+            url: '/swagger/doc.json',
+        })
    </script>
 </body>
 </html>`)
-	})
+		})
+	}

 	v1 := r.Group("/api/v1")
 	{