如何在 Go HTTP 处理程序和中间件中传递和访问上下文信息？

在中间件和 HandlerFunc 中传递上下文

在 Go 中， context 包提供了在请求处理程序之间传递信息的功能。这对于中间件特别有用，中间件可以在执行处理程序函数之前和之后执行操作。

创建和传递上下文

在中间件的上下文中，我们可以通过在请求的上下文上调用 context.WithValue 来创建新的上下文，并提供键和值对。然后，在调用处理程序的 ServeHTTP 方法时应使用这个新上下文。

例如，在提供的代码片段中，checkAuth 中间件函数接收请求上下文和身份验证令牌作为输入。它返回一个包装的处理函数，用于检查身份验证令牌。

func checkAuth(authToken string) util.Middleware {
    return func(h http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            if r.Header.Get("Auth") != authToken {
                util.SendError(w, "...", http.StatusForbidden, false)
                return
            }
            h.ServeHTTP(w, r)
        })
    }
}

要使用身份验证令牌初始化上下文，我们可以在请求的上下文上调用 context.WithValue，如下所示：

func main() {
    authToken, ok := getAuthToken()
    if !ok {
        panic("...")
    }
    ctx := context.WithValue(r.Context(), "auth_token", authToken)
    router.Handle("/hello", util.UseMiddleware(authCheck, Handler, ...))
}

当中间件处理程序调用原始处理程序的 ServeHTTP 时，将使用这个新上下文

在处理程序中访问上下文

处理程序可以通过调用 r.Context() 来访问上下文信息，该方法返回当前上下文。可以使用 Value 方法检索该值，如下面的代码片段所示：

func (h *HandlerW) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    authToken := r.Context().Value("auth_token").(string)
    // ...
}

以上是如何在 Go HTTP 处理程序和中间件中传递和访问上下文信息？的详细内容。更多信息请关注PHP中文网其他相关文章！