diff --git a/router/option.go b/router/option.go index c80cbcb..88b862d 100644 --- a/router/option.go +++ b/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 } }