访问没有 http.Request 的上下文

积累知识，胜过积蓄金银！毕竟在Golang开发的过程中，会遇到各种各样的问题，往往都是一些细节知识点还没有掌握好而导致的，因此基础知识点的积累是很重要的。下面本文《访问没有 http.Request 的上下文》，就带大家讲解一下知识点，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我在中间件内的 context 中设置 x-request-id （如下所示），以便我可以在有 *http.request 结构的地方使用它 - 例如req.context().value(中间件.reqidkey).但是，我的代码库中的某些地方无法访问 *http.request 结构，因此我无法使用 context 来获取 x-request-id。 go 有办法还是我试图做一些根本错误的事情？

内部/中间件/requestid.go

这是我在 context 中设置 x-request-id 的中间件。目前在我的“服务器”包中称为 http.listenandserve(":8080", middleware.requestid(somehandler)) 。

package middleware

import (
    "context"
    "github.com/google/uuid"
    "net/http"
)

type val string
const reqidkey val = "x-request-id"

func requestid(handler http.handler) http.handler {
    return http.handlerfunc(func(res http.responsewriter, req *http.request) {
        val := req.header.get("x-request-id")
        if val == "" {
            val = uuid.new().string()
        }

        ctx1 := context.withvalue(req.context(), reqidkey, val)
        ctx2 := req.withcontext(ctx1)

        handler.servehttp(res, ctx2)
    })
}

内部/logger/logger.go

这是另一个包，我需要访问 context 或只是 x-request-id 值。顺便说一句，调用 logger.config 是在启动服务器之前进行的。

package logger

import (
    "github.com/sirupsen/logrus"
    "os"
)

var Log *logrus.Entry

func Config() {
    logrus.SetLevel(logrus.InfoLevel)
    logrus.SetOutput(os.Stdout)
    logrus.SetFormatter(&logrus.JSONFormatter{})

    Log = logrus.WithFields(logrus.Fields{
        "request_id": ..., // I need X-Request-Id value to go here so that all logs have it
    })
}

解决方案

首先，您应该将上下文传递到需要的地方。如果您在 config() 函数中需要它，请将其传递到那里：

func config(ctx context.context) {
  /* ... * /
}

但是您可能会在启动时调用 config() 一次，而不是根据请求调用一次，这引出了我的第二点：

您不应该将上下文或一般请求范围的数据传递给配置函数。这完全是本末倒置。

相反，您应该将记录器传递到处理程序/中间件中，并让它记录请求数据：

func handleSomePath(logger *logrus.Entry) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        /* do something */
        logger.WithFields(logrus.Fields{
            "request_id": /* ... */
        })
    }
}

如果您有 http.request，您可以访问其中的上下文和值。如果您没有请求但需要上下文：获取上下文并将其作为显式参数传递到您的调用树（按照惯例，它是第一个参数）。

（go 中没有魔法，任何没有直接或间接传递到函数中的东西都不存在。）

到这里，我们也就讲完了《访问没有 http.Request 的上下文》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！