Go chi 中间件将项目添加到响应标头

如何解决Go chi 中间件将项目添加到响应标头

我的概念很简单，我想在响应头中添加请求ID（由chi middleware.RequestID生成）。

我下面的代码添加了 X-Request-Id 但没有添加 ID 本身。我不知道为什么，因为它肯定在上下文中可用，因为 middleware.RequestID 在中间件链中处于更高的位置？

我的路由器创建：

package rest

import (
    "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 mw

import (
    "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 routes

import "net/http"

func IndexRoute() http.HandlerFunc {
    fn:= func(w http.ResponseWriter,r *http.Request) {
        w.Write([]byte("welcome"))
    }
    return fn
}

解决方法

我认为您的问题在这里：

middleware.GetReqID(context.Background())

这会尝试获取新的空上下文的请求 ID，当您需要的是当前请求的请求 ID 时，这意味着它应该是：

middleware.GetReqID(r.Context)

（或者，由于您已经在本地 ctx 中获取请求上下文，您也可以使用它。）

这是documented in GetReqId（强调我的）：

GetReqID 从给定的上下文返回一个请求 ID（如果存在）。如果找不到请求 ID，则返回空字符串。