我的概念很简单,我想添加请求ID(由chi生成到响应标头)。middleware.RequestID我下面的代码添加了但未添加ID本身。我不确定为什么,因为它肯定在上下文中可用,因为在中间件链中处于较高位置?X-Request-Idmiddleware.RequestID我的路由器创建: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 回答
- 0 关注
- 96 浏览
添加回答
举报
0/150
提交
取消
