server初始化支持设置各种option

2025-02-18 18:12:44 +08:00
parent 74dde8b41e
commit 1c2c5aa4ce
2 changed files with 165 additions and 11 deletions
--- a/router/option.go
+++ b/router/option.go
@ -6,3 +6,106 @@
 //
 // Date : 2025-02-18 17:26
 package router
+
+import (
+	apiDocDefine "git.zhangdeman.cn/gateway/api-doc/define"
+	"github.com/gin-gonic/gin"
+	"strings"
+)
+
+type SetServerOptionFunc func(so *serverOption)
+
+// serverOption 获取server实例的选项
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 17:28 2025/2/18
+type serverOption struct {
+	swaggerUiTheme       string                     // swagger 主题
+	swaggerBaseUri       string                     // swagger基础path
+	globalMiddlewareList []gin.HandlerFunc          // 全局中间件列表
+	disableSwaggerDoc    bool                       // 禁用swagger文档, 特定环境不想展示文档, 可通过次方式禁用
+	serverInfo           *apiDocDefine.Info         // 服务器信息
+	serverList           []*apiDocDefine.ServerItem // 服务器环境列表
+}
+
+// WithSwaggerUITheme 设置swaggerUI主题
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 17:29 2025/2/18
+func WithSwaggerUITheme(uiTheme string) SetServerOptionFunc {
+	return func(so *serverOption) {
+		uiTheme = strings.TrimSpace(uiTheme)
+		if len(uiTheme) == 0 {
+			return
+		}
+		so.swaggerUiTheme = uiTheme
+	}
+}
+
+// WithGlobalMiddlewareList 设置全局中间件
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 17:33 2025/2/18
+func WithGlobalMiddlewareList(middlewareList ...gin.HandlerFunc) SetServerOptionFunc {
+	return func(so *serverOption) {
+		so.globalMiddlewareList = middlewareList
+	}
+}
+
+// WithSwaggerBaseUri ...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 18:05 2025/2/18
+func WithSwaggerBaseUri(baseUri string) SetServerOptionFunc {
+	return func(so *serverOption) {
+		baseUri = strings.TrimSpace(baseUri)
+		if len(baseUri) == 0 {
+			return
+		}
+		baseUri = "/" + strings.TrimLeft(baseUri, "/")
+		so.swaggerBaseUri = baseUri
+	}
+}
+
+// WithDisableSwaggerDoc 禁用swagger文档
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 17:35 2025/2/18
+func WithDisableSwaggerDoc() SetServerOptionFunc {
+	return func(so *serverOption) {
+		so.disableSwaggerDoc = true
+	}
+}
+
+// WithServerInfo 设置serverInfo
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 17:51 2025/2/18
+func WithServerInfo(serverInfo *apiDocDefine.Info) SetServerOptionFunc {
+	return func(so *serverOption) {
+		if nil == serverInfo {
+			return
+		}
+		so.serverInfo = serverInfo
+	}
+}
+
+// WithServerList 设置服务器列表
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 17:52 2025/2/18
+func WithServerList(serverList []*apiDocDefine.ServerItem) SetServerOptionFunc {
+	return func(so *serverOption) {
+		if len(serverList) == 0 {
+			return
+		}
+		so.serverList = serverList
+	}
+}