Go Chi中间件向响应头添加项

bzzcjhmw  于 2023-05-04  发布在  Go
关注(0)|答案(1)|浏览(111)

我的概念很简单,我想添加请求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
}
lvjbypge

lvjbypge1#

我相信你的问题在这里:

middleware.GetReqID(context.Background())

这会尝试获取一个新的空上下文的请求ID,但是你需要当前请求的请求ID,这意味着它应该是:

middleware.GetReqID(r.Context())

(Or,因为您已经在本地ctx中获取了请求上下文,所以也可以使用它。)
这在GetReqId中有记录(强调我的):
GetReqID返回给定上下文中的请求ID(如果存在)。如果找不到请求ID,则返回空字符串。

相关问题