feat: 优化initContextFunc

router/option.go

@@ -40,18 +40,13 @@ func WithDisableInitRequest(disable bool) SetServerOptionFunc {
 }
 
 // WithInitContextData 初始化一些请求数据
-func WithInitContextData(formatFunc func(ctx *gin.Context) map[string]any) SetServerOptionFunc {
+func WithInitContextData(formatFunc func(ctx *gin.Context)) SetServerOptionFunc {
 	return func(so *serverOption) {
 		if nil == formatFunc {
-			return
+			formatFunc = func(ctx *gin.Context) {}
 		}
-		f := func(ctx *gin.Context) {
-			data := formatFunc(ctx)
-			for k, v := range data {
-				ctx.Set(k, v)
-			}
-		}
-		so.initContextData = f
+
+		so.initContextData = formatFunc
 	}
 }