upgrade: 优化 WithInitContextData 配置
This commit is contained in:
@ -44,9 +44,13 @@ func WithDisableInitRequest(disable bool) SetServerOptionFunc {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// WithInitContextData 初始化一些请求数据
|
// WithInitContextData 初始化一些请求数据
|
||||||
func WithInitContextData(data map[string]any) SetServerOptionFunc {
|
func WithInitContextData(formatFunc func(ctx *gin.Context) map[string]any) SetServerOptionFunc {
|
||||||
return func(so *serverOption) {
|
return func(so *serverOption) {
|
||||||
|
if nil == formatFunc {
|
||||||
|
return
|
||||||
|
}
|
||||||
f := func(ctx *gin.Context) {
|
f := func(ctx *gin.Context) {
|
||||||
|
data := formatFunc(ctx)
|
||||||
for k, v := range data {
|
for k, v := range data {
|
||||||
ctx.Set(k, v)
|
ctx.Set(k, v)
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user