理解 Go 中的上下文可能会令人困惑。让我们探讨如何有效地将上下文传递给中间件和处理程序。
要将上下文传递给中间件,请按照以下步骤操作:
例如,为请求添加超时:
ctx, cancel := context.WithTimeout(r.Context(), time.Duration(60*time.Second)) defer cancel() r = r.WithContext(ctx)
至将上下文传递给处理程序:
例如,要将用户 ID 添加到上下文中:
ctx := context.WithValue(r.Context(), ContextUserKey, "theuser") h.ServeHTTP(w, r.WithContext(ctx))
这里是中间件和处理程序的示例实现使用上下文:
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 } // Add authentication-specific context here h.ServeHTTP(w, r) }) } } type Handler struct { ... ... } func (h *HandlerW) ServeHTTP(w http.ResponseWriter, r *http.Request) { // Get context values here decoder := json.NewDecoder(r.Body) // ... } func main() { router := mux.NewRouter() authToken, ok := getAuthToken() if !ok { panic("...") } authCheck := checkAuth(authToken) h := Handler{ ... } router.Handle("/hello", util.UseMiddleware(authCheck, Handler, ...)) }
以上是如何在 Go 中正确传递中间件和处理程序中的上下文?的详细内容。更多信息请关注PHP中文网其他相关文章!