server初始化支持设置各种option

router/server.go

@@ -10,6 +10,7 @@ package router
 import (
 	"fmt"
 	apiDoc "git.zhangdeman.cn/gateway/api-doc"
+	"git.zhangdeman.cn/zhangdeman/consts"
 	"git.zhangdeman.cn/zhangdeman/gin/middleware"
 	"net/http"
 	"strings"
@@ -18,32 +19,72 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
+func newServerOption(port int, optionList ...SetServerOptionFunc) *serverOption {
+	option := &serverOption{
+		swaggerUiTheme:       apiDocDefine.SwaggerUIThemeRedocFree,
+		swaggerBaseUri:       "/doc/swagger",
+		globalMiddlewareList: nil,
+		disableSwaggerDoc:    false,
+		serverInfo: &apiDocDefine.Info{
+			Description:    "这是一个微服务，提供一些必要的的数据接口功能",
+			Title:          "微服务接口文档",
+			TermsOfService: "",
+			Contact: &apiDocDefine.Contact{
+				Name:  "开发人员",
+				Url:   "",
+				Email: "developer@example.com",
+			},
+			License: &apiDocDefine.License{
+				Name: consts.LicenseApache20,
+				Url:  consts.LicenseUrlTable[consts.LicenseApache20],
+			},
+			Version: "0.0.1",
+		},
+		serverList: []*apiDocDefine.ServerItem{
+			{
+				Url:         fmt.Sprintf("http://127.0.0.1:%d", port),
+				Description: "测试服务器",
+				Variables:   nil,
+			},
+		},
+	}
+	for _, opt := range optionList {
+		if nil == opt {
+			continue
+		}
+		opt(option)
+	}
+	return option
+}
+
 // NewServer server实例
 //
 // Author : go_developer@163.com<白茶清欢>
 //
 // Date : 18:20 2025/2/7
-func NewServer(port int, option any) *server {
+func NewServer(port int, optionList ...SetServerOptionFunc) *server {
 	if port < 80 {
 		panic("port should be greater than 80")
 	}
+	option := newServerOption(port, optionList...)
 	r := gin.Default()
+
 	r.Use(
 		// 初始化请求
-		middleware.InitRequest(),
+		middleware.InitRequest(), // 初始化请求
 		/*request_cors.New(request_cors.Config{
 			AllowAllOrigins: true,
 		}), */// middleware.Ac
 	)
+	if nil != option.globalMiddlewareList {
+		// 启用全局中间件
+		r.Use(option.globalMiddlewareList...)
+	}
 	return &server{
-		router: r,
-		uiInstance: apiDoc.NewSwaggerUI(nil, []*apiDocDefine.ServerItem{
-			{
-				Url:         fmt.Sprintf("http://127.0.0.1:%d", port),
-				Description: "测试服务器",
-			},
-		}, apiDocDefine.SwaggerUIThemeRedocFree), // TODO : 配置相关信息
-		port: port,
+		router:     r,
+		uiInstance: apiDoc.NewSwaggerUI(option.serverInfo, option.serverList, option.swaggerUiTheme),
+		port:       port,
+		option:     option,
 	}
 }
 
@@ -51,6 +92,7 @@ type server struct {
 	router     *gin.Engine
 	port       int
 	uiInstance *apiDoc.SwaggerUI
+	option     *serverOption
 }
 
 // Start 启动服务
@@ -60,12 +102,21 @@ type server struct {
 // Date : 18:31 2025/2/7
 func (s *server) Start() {
 	// 注册文档
-	s.uiInstance.RegisterHandler(s.router, "/doc/swagger")
+	s.uiInstance.RegisterHandler(s.router, s.option.swaggerBaseUri)
 	if err := s.router.Run(fmt.Sprintf(":%d", s.port)); err != nil {
 		panic("服务启动监听失败" + err.Error())
 	}
 }
 
+// Router 对外访问路由实例, 不建议直接用
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 17:35 2025/2/18
+func (s *server) Router() *gin.Engine {
+	return s.router
+}
+
 // Group 注册接口路由
 //
 // Author : go_developer@163.com<白茶清欢>