upgrade: 优化 WithInitContextData 配置

2025-08-17 21:50:59 +08:00
parent b8803409b0
commit d70c3a0cee
1 changed files with 5 additions and 1 deletions
--- a/router/option.go
+++ b/router/option.go
@ -44,9 +44,13 @@ func WithDisableInitRequest(disable bool) SetServerOptionFunc {
 }

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