From 193ed9f42f15e82ae7b96980997bd1abe8ca5f1b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=99=BD=E8=8C=B6=E6=B8=85=E6=AC=A2?=
 <go_developer@163.com>
Date: Sat, 3 Jan 2026 13:26:52 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E5=8C=96initContextFunc?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 router/option.go | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

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
 	}
 }