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
 app:
2
 app:
3
   web-client: 1  # 1:启用 0:禁用
3
   web-client: 1  # 1:启用 0:禁用
4
   register: false #是否开启注册
4
   register: false #是否开启注册
5
+  show-swagger: 0 # 1:启用 0:禁用
5
 admin:
6
 admin:
6
   title: "RustDesk Api Admin"
7
   title: "RustDesk Api Admin"
7
   hello-file: "./conf/admin/hello.html"  #优先使用file
8
   hello-file: "./conf/admin/hello.html"  #优先使用file

+ 3 - 2
config/config.go

@@ -14,8 +14,9 @@ const (
14
 )
14
 )
15
 
15
 
16
 type App struct {
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
 type Admin struct {
21
 type Admin struct {
21
 	Title     string `mapstructure:"title"`
22
 	Title     string `mapstructure:"title"`

+ 4 - 1
http/router/admin.go

@@ -2,6 +2,7 @@ package router
2
 
2
 
3
 import (
3
 import (
4
 	_ "Gwen/docs/admin"
4
 	_ "Gwen/docs/admin"
5
+	"Gwen/global"
5
 	"Gwen/http/controller/admin"
6
 	"Gwen/http/controller/admin"
6
 	"Gwen/http/controller/admin/my"
7
 	"Gwen/http/controller/admin/my"
7
 	"Gwen/http/middleware"
8
 	"Gwen/http/middleware"
@@ -14,7 +15,9 @@ func Init(g *gin.Engine) {
14
 
15
 
15
 	//swagger
16
 	//swagger
16
 	//g.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
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
 	adg := g.Group("/api/admin")
22
 	adg := g.Group("/api/admin")
20
 	LoginBind(adg)
23
 	LoginBind(adg)

+ 3 - 1
http/router/api.go

@@ -15,7 +15,9 @@ func ApiInit(g *gin.Engine) {
15
 
15
 
16
 	//g.Use(middleware.Cors())
16
 	//g.Use(middleware.Cors())
17
 	//swagger
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
 	frg := g.Group("/api")
22
 	frg := g.Group("/api")
21
 
23