Go chi 中间件在响应头中附加项目

在 Go chi 中，可以通过中间件向响应头添加项目。然而，在添加请求 ID 时遇到了问题，尽管中间件生成了 ID，但它并未添加到标头中。自定义中间件负责添加 ID，但使用了不正确的上下文，导致无法获取当前请求的 ID。通过使用正确的上下文，可以成功将请求 ID 添加到响应头。

我的概念很简单，我想添加请求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，则返回空字符串。

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~