lejianwen 1 год назад
Родитель
Сommit
c298ddea0c
4 измененных файлов с 11 добавлено и 4 удалено
  1. 1 0
      conf/config.yaml
  2. 3 2
      config/config.go
  3. 4 1
      http/router/admin.go
  4. 3 1
      http/router/api.go

+ 1 - 0
conf/config.yaml

@@ -2,6 +2,7 @@ lang: "zh-CN"
2 2
 app:
3 3
   web-client: 1  # 1:启用 0:禁用
4 4
   register: false #是否开启注册
5
+  show-swagger: 0 # 1:启用 0:禁用
5 6
 admin:
6 7
   title: "RustDesk Api Admin"
7 8
   hello-file: "./conf/admin/hello.html"  #优先使用file

+ 3 - 2
config/config.go

@@ -14,8 +14,9 @@ const (
14 14
 )
15 15
 
16 16
 type App struct {
17
-	WebClient int  `mapstructure:"web-client"`
18
-	Register  bool `mapstructure:"register"`
17
+	WebClient   int  `mapstructure:"web-client"`
18
+	Register    bool `mapstructure:"register"`
19
+	ShowSwagger int  `mapstructure:"show-swagger"`
19 20
 }
20 21
 type Admin struct {
21 22
 	Title     string `mapstructure:"title"`

+ 4 - 1
http/router/admin.go

@@ -2,6 +2,7 @@ package router
2 2
 
3 3
 import (
4 4
 	_ "Gwen/docs/admin"
5
+	"Gwen/global"
5 6
 	"Gwen/http/controller/admin"
6 7
 	"Gwen/http/controller/admin/my"
7 8
 	"Gwen/http/middleware"
@@ -14,7 +15,9 @@ func Init(g *gin.Engine) {
14 15
 
15 16
 	//swagger
16 17
 	//g.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
17
-	g.GET("/admin/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler, ginSwagger.InstanceName("admin")))
18
+	if global.Config.App.ShowSwagger == 1 {
19
+		g.GET("/admin/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler, ginSwagger.InstanceName("admin")))
20
+	}
18 21
 
19 22
 	adg := g.Group("/api/admin")
20 23
 	LoginBind(adg)

+ 3 - 1
http/router/api.go

@@ -15,7 +15,9 @@ func ApiInit(g *gin.Engine) {
15 15
 
16 16
 	//g.Use(middleware.Cors())
17 17
 	//swagger
18
-	g.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler, ginSwagger.InstanceName("api")))
18
+	if global.Config.App.ShowSwagger == 1 {
19
+		g.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler, ginSwagger.InstanceName("api")))
20
+	}
19 21
 
20 22
 	frg := g.Group("/api")
21 23