upgrade: 支持初始化上下文数据中间件 + 支持自动启用请求初始化中间件

router/option.go

@@ -31,7 +31,28 @@ type serverOption struct {
 	serverList           []*apiDocDefine.ServerItem // 服务器环境列表
 	enablePprof          bool                       // 启用pprof
 	enableCors           bool                       // 启动跨域支持
+	disableInitRequest   bool                       // 禁用初始化请求
 	loggerCfg            *middleware.AccessConfig   // 日志配置
+	initContextData      gin.HandlerFunc            // 初始化一些请求数据
+}
+
+// WithEnableInitRequest 启用pprof
+func WithEnableInitRequest(disable bool) SetServerOptionFunc {
+	return func(so *serverOption) {
+		so.disableInitRequest = disable
+	}
+}
+
+// WithInitContextData 初始化一些请求数据
+func WithInitContextData(data map[string]any) SetServerOptionFunc {
+	return func(so *serverOption) {
+		f := func(ctx *gin.Context) {
+			for k, v := range data {
+				ctx.Set(k, v)
+			}
+		}
+		so.initContextData = f
+	}
 }
 
 // WithSwaggerUITheme 设置swaggerUI主题