在中间件和 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中文网其他相关文章！
