如何在 Go 中正確傳遞中間件和處理程序中的上下文？

在 Go 中的中間件和處理程序中傳遞上下文

簡介

理解 Go 中的上下文可能會令人困惑。讓我們探討如何有效地將上下文傳遞給中間件和處理程序。

在中間件中傳遞上下文

要將上下文傳遞給中間件，請按照以下步驟操作：

1. 派生一個新的使用context.WithTimeout( ) 或context.WithValue() 從請求上下文取得上下文。
2. 使用更新後的請求呼叫 ServeHTTP，該請求現在包含新的上下文。
3. 在授權檢查器中，加入在呼叫 ServeHTTP 之前將使用者的資訊傳遞到上下文。

例如，為請求添加超時：

ctx, cancel := context.WithTimeout(r.Context(), time.Duration(60*time.Second))
defer cancel()
r = r.WithContext(ctx)

在處理程序中傳遞上下文

至將上下文傳遞給處理程序：

1. 至將上下文傳遞給處理程序：
2. 使用context.WithValue() 將值加到請求的上下文。

在處理程序中，使用request.Context().Value() 存取

ctx := context.WithValue(r.Context(), ContextUserKey, "theuser")
h.ServeHTTP(w, r.WithContext(ctx))

例如，要將使用者ID 新增至上下文：

範例程式碼

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中文網其他相關文章！