优化接口文档注册

router/server.go

@@ -9,11 +9,10 @@ package router
 
 import (
 	"fmt"
+	apiDoc "git.zhangdeman.cn/gateway/api-doc"
 	"net/http"
 	"strings"
 
-	"git.zhangdeman.cn/zhangdeman/consts"
-
 	apiDocDefine "git.zhangdeman.cn/gateway/api-doc/define"
 	"github.com/gin-gonic/gin"
 )
@@ -29,20 +28,20 @@ func NewServer(port int, option any) *server {
 	}
 	return &server{
 		router: gin.Default(),
-		docInstance: NewDoc(nil, []*apiDocDefine.ServerItem{
+		uiInstance: apiDoc.NewSwaggerUI(nil, []*apiDocDefine.ServerItem{
 			{
 				Url:         fmt.Sprintf("http://127.0.0.1:%d", port),
 				Description: "测试服务器",
 			},
-		}), // TODO : 配置相关信息
+		}, apiDocDefine.SwaggerUIThemeYDocLucky), // TODO : 配置相关信息
 		port: port,
 	}
 }
 
 type server struct {
-	router      *gin.Engine
-	port        int
-	docInstance *Doc
+	router     *gin.Engine
+	port       int
+	uiInstance *apiDoc.SwaggerUI
 }
 
 // Start 启动服务
@@ -68,7 +67,7 @@ func (s *server) Group(routerPrefix string, middlewareList []gin.HandlerFunc, cL
 	for _, c := range cList {
 		urlTable := cParser.Parse(c)
 		for _, itemUriCfg := range urlTable {
-			s.docInstance.Add(routerPrefix, itemUriCfg.FormDataType, itemUriCfg.ResultDataType)
+			_ = s.uiInstance.DocInstance().AddApiFromInAndOut(routerPrefix, itemUriCfg.FormDataType, itemUriCfg.ResultDataType)
 			method := strings.ToUpper(itemUriCfg.RequestMethod)
 			switch method {
 			case http.MethodGet:
@@ -93,47 +92,5 @@ func (s *server) Group(routerPrefix string, middlewareList []gin.HandlerFunc, cL
 		}
 	}
 	// 注册文档
-	s.SwaggerUI()
-}
-
-// SwaggerUI ...
-//
-// Author : go_developer@163.com<白茶清欢>
-//
-// Date : 21:45 2025/2/15
-func (s *server) SwaggerUI() {
-	swaggerInstance := NewSwaggerUI(s.docInstance, SwaggerUIThemeKnife4go)
-	s.router.GET("/swagger-resources", func(ctx *gin.Context) { // lucky UI获取分组信息
-		ctx.Writer.Header().Set("Access-Control-Allow-Origin", "*") // 允许访问所有域
-		ctx.JSON(http.StatusOK, []map[string]any{
-			{
-				"name":           "服务文档",
-				"url":            "/swagger-ui/doc/openapi.json",
-				"swaggerVersion": consts.SwaggerDocVersion3,
-			},
-		})
-		// ctx.JSON(http.StatusOK, swaggerInstance.docInstance.Data())
-	})
-	s.router.GET("/swagger-ui/doc/openapi.json", func(ctx *gin.Context) {
-		ctx.Writer.Header().Set("Access-Control-Allow-Origin", "*") // 允许访问所有域
-		ctx.JSON(http.StatusOK, swaggerInstance.docInstance.Data())
-	})
-	s.router.GET("/doc/swagger/*any", func(ctx *gin.Context) {
-		if ctx.Request.RequestURI == "/doc/swagger/doc.json" {
-			// 默认swagger, 通过此接口读取文档数据
-			ctx.JSON(http.StatusOK, swaggerInstance.docInstance.Data())
-			ctx.Abort()
-		}
-		if ctx.Request.RequestURI == "/doc/swagger/openapi.json" {
-			// knife4go 文档通过此接口读取文档列表
-			ctx.JSON(http.StatusOK, []map[string]any{
-				{
-					"name":           "服务文档",
-					"url":            "doc.json",
-					"swaggerVersion": consts.SwaggerDocVersion3,
-				},
-			})
-			ctx.Abort()
-		}
-	}, swaggerInstance.Handler())
+	s.uiInstance.RegisterHandler(s.router, "/doc/swagger")
 }