如何在 Go 中的 HTTP 處理程序中安全地設定上下文值？

在HTTP 處理程序中設定上下文值

在Go 的HTTP 套件中，HandleFunc 函數允許註冊處理程序函數來處理傳入入>在Go 的HTTP 套件中，HandleFunc 函數允許註冊處理程序函數來處理傳入入的HTTP 請求。雖然在處理程序中設定上下文值是一種常見做法，但確定最佳方法可能會引起注意。

一種方法是使用 context 套件中的 WithValue 函數在處理程序中設定上下文值。但是，請務必注意，以下用法可能會引入潛在問題：

func setValue(r *http.Request, val string) {
  ctx := context.WithValue(r.Context(), myContext, val)
  *r = *r.WithContext(ctx)
}

將 *r 指派給新的請求物件可能會導致堆疊中的程式碼使用不正確的上下文值。為了解決這個問題，建議的方法是使用WithContext 建立請求的淺表副本並傳回指向它的指標：

func setValue(r *http.Request, val string) *http.Request {
  return r.WithContext(context.WithValue(r.Context(), myContext, val))
}

透過傳回指向淺表副本的指針，可以確保正確的上下文值在整個請求處理管道中使用。如果處理程序呼叫其他處理程序，請將修改後的請求傳遞給後續處理程序：

http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
  r = setValue(r, "foobar")
  someOtherHandler.ServeHTTP(w, r)
})

此方法有效地在HTTP 處理程序中設定上下文值，同時避免與修改原始請求物件相關的潛在陷阱。

以上是如何在 Go 中的 HTTP 處理程序中安全地設定上下文值？的詳細內容。更多資訊請關注PHP中文網其他相關文章！