How to Safely Manipulate Context Values in Go HTTP Handlers?

Context Value Manipulation in HTTP Handlers

Setting context values within HTTP handlers is crucial for sharing data across request handlers. In Go's HTTP package, the http.HandleFunc function is commonly used to register handlers. However, the approach of setting context values using *r = *r.WithContext(ctx) raises concerns.

Recommended Approach

To avoid overwriting the request object, which can lead to unexpected behavior, it's best to return a new shallow copy of the request with the modified context value.

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")
})

This approach ensures that the original request object remains unchanged and that the updated context is propagated to any后续处理程序调用。

Passing the Updated Request

If the handler needs to invoke another handler, it's crucial to pass the updated request object to the new handler. Otherwise, the modified context value will not be available to the subsequent handler.

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

By following these best practices, you can effectively manage context values within HTTP handlers, enabling efficient data sharing and seamless transitions between request handlers.

The above is the detailed content of How to Safely Manipulate Context Values in Go HTTP Handlers?. For more information, please follow other related articles on the PHP Chinese website!