整合默认swagger文档

router/define.go

@@ -63,3 +63,8 @@ type UriParam struct {
 const (
 	FieldNameMeta = "Meta" // 元信息字段
 )
+
+const (
+	SwaggerUIThemeDefault  = "swaggerUI" // 文档默认主题
+	SwaggerUIThemeKnife4go = "knife4go"  // knife4go 主题
+)

router/server.go

@@ -82,4 +82,22 @@ 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, SwaggerUIThemeDefault)
+	// 默认swagger
+	s.router.GET("/swagger/*any", func(ctx *gin.Context) {
+		if ctx.Request.RequestURI == "/swagger/doc.json" {
+			ctx.JSON(http.StatusOK, swaggerInstance.docInstance.Data())
+			ctx.Abort()
+		}
+	}, swaggerInstance.HandleSwaggerUI())
 }

router/swagger_ui.go

@@ -0,0 +1,71 @@
+// Package router ...
+//
+// Description : router ...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 2025-02-15 21:20
+package router
+
+import (
+	"github.com/gin-gonic/gin"
+	knife4goFiles "github.com/go-webtools/knife4go"
+	knife4goGin "github.com/go-webtools/knife4go/gin"
+	swaggerFiles "github.com/swaggo/files"
+	ginSwagger "github.com/swaggo/gin-swagger"
+)
+
+// NewSwaggerUI ...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 21:21 2025/2/15
+func NewSwaggerUI(docInstance *Doc, uiTheme string) *SwaggerUI {
+	return &SwaggerUI{
+		docInstance: docInstance,
+		uiTheme:     uiTheme,
+	}
+}
+
+type SwaggerUI struct {
+	docInstance *Doc   // 文档实例
+	uiTheme     string // 文档主题, swaggerUI / knife4go, 默认 knife4go
+}
+
+// Handler 访问文档的接口处理
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 21:34 2025/2/15
+func (su *SwaggerUI) Handler() func(ctx *gin.Context) {
+	switch su.uiTheme {
+	case SwaggerUIThemeKnife4go:
+		return su.HandleKnife4goUI()
+	default:
+		return su.HandleSwaggerUI()
+	}
+}
+
+// HandleKnife4goUI ...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 21:38 2025/2/15
+func (su *SwaggerUI) HandleKnife4goUI() func(ctx *gin.Context) {
+	resetOption := func(cfg *knife4goGin.Config) {
+		if nil == cfg {
+			return
+		}
+		cfg.Title = "服务接口文档"
+	}
+	return knife4goGin.WrapHandler(knife4goFiles.Handler, resetOption)
+}
+
+// HandleSwaggerUI ...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 21:41 2025/2/15
+func (su *SwaggerUI) HandleSwaggerUI() func(ctx *gin.Context) {
+	return ginSwagger.WrapHandler(swaggerFiles.Handler)
+}