我的概念很簡(jiǎn)單，我想添加請(qǐng)求ID（由chi生成到響應(yīng)標(biāo)頭）。middleware.RequestID我下面的代碼添加了但未添加ID本身。我不確定為什么，因?yàn)樗隙ㄔ谏舷挛闹锌捎?，因?yàn)樵谥虚g件鏈中處于較高位置？X-Request-Idmiddleware.RequestID我的路由器創(chuàng)建：package restimport ( "time" "github.com/go-chi/chi/v5" "github.com/go-chi/chi/v5/middleware" "goWebAPI/pkg/database" "goWebAPI/pkg/rest/mw" "goWebAPI/pkg/rest/routes")func GetRouter(db *database.Database) *chi.Mux { r := chi.NewRouter() r.Use(middleware.RequestID) r.Use(mw.SendRequestID) r.Use(middleware.RealIP) r.Use(middleware.Logger) r.Use(middleware.Recoverer) r.Use(middleware.Timeout(60 * time.Second)) r.Get("/", routes.IndexRoute()) return r}我的自定義中間件：package mwimport ( "context" "net/http" "github.com/go-chi/chi/v5/middleware")const requestIDHeader = "X-Request-Id"func SendRequestID(next http.Handler) http.Handler { fn := func(w http.ResponseWriter, r *http.Request) { ctx := r.Context() if w.Header().Get(requestIDHeader) == "" { w.Header().Add( requestIDHeader, middleware.GetReqID(context.Background()), ) } next.ServeHTTP(w, r.WithContext(ctx)) } return http.HandlerFunc(fn)}我的索引路由：package routesimport "net/http"func IndexRoute() http.HandlerFunc { fn:= func(w http.ResponseWriter, r *http.Request) { w.Write([]byte("welcome")) } return fn}
1 回答
胡子哥哥
TA貢獻(xiàn)1825條經(jīng)驗(yàn) 獲得超6個(gè)贊
我相信你的問題在這里：
middleware.GetReqID(context.Background())
這將嘗試獲取新的空上下文的請(qǐng)求 ID，而您需要的是當(dāng)前請(qǐng)求的請(qǐng)求 ID，這意味著它應(yīng)該是：
middleware.GetReqID(r.Context)
（或者，由于您已經(jīng)在本地獲取了請(qǐng)求上下文，因此也可以使用它。ctx
這記錄在獲取請(qǐng)求Id中（強(qiáng)調(diào)我的）：
如果存在請(qǐng)求 ID，則從給定的上下文中返回一個(gè)請(qǐng)求 ID。如果找不到請(qǐng)求 ID，則返回空字符串。
- 1 回答
- 0 關(guān)注
- 62 瀏覽
添加回答
舉報(bào)
0/150
提交
取消