Skip to content

Commit

Permalink
reverseproxy: Connection termination cleanup (#5663)
Browse files Browse the repository at this point in the history
  • Loading branch information
mmm444 authored Aug 1, 2023
1 parent 94749e1 commit da23501
Showing 1 changed file with 19 additions and 6 deletions.
25 changes: 19 additions & 6 deletions modules/caddyhttp/reverseproxy/reverseproxy.go
Original file line number Diff line number Diff line change
Expand Up @@ -769,7 +769,7 @@ func (h *Handler) reverseProxy(rw http.ResponseWriter, req *http.Request, origRe
// regardless, and we should expect client disconnection in low-latency streaming
// scenarios (see issue #4922)
if h.FlushInterval == -1 {
req = req.WithContext(ignoreClientGoneContext{req.Context(), h.ctx.Done()})
req = req.WithContext(ignoreClientGoneContext{req.Context()})
}

// do the round-trip; emit debug log with values we know are
Expand Down Expand Up @@ -1398,15 +1398,28 @@ type handleResponseContext struct {
// ignoreClientGoneContext is a special context.Context type
// intended for use when doing a RoundTrip where you don't
// want a client disconnection to cancel the request during
// the roundtrip. Set its done field to a Done() channel
// of a context that doesn't get canceled when the client
// disconnects, such as caddy.Context.Done() instead.
// the roundtrip.
// This context clears cancellation, error, and deadline methods,
// but still allows values to pass through from its embedded
// context.
//
// TODO: This can be replaced with context.WithoutCancel once
// the minimum required version of Go is 1.21.
type ignoreClientGoneContext struct {
context.Context
done <-chan struct{}
}

func (c ignoreClientGoneContext) Done() <-chan struct{} { return c.done }
func (c ignoreClientGoneContext) Deadline() (deadline time.Time, ok bool) {
return
}

func (c ignoreClientGoneContext) Done() <-chan struct{} {
return nil
}

func (c ignoreClientGoneContext) Err() error {
return nil
}

// proxyHandleResponseContextCtxKey is the context key for the active proxy handler
// so that handle_response routes can inherit some config options
Expand Down

0 comments on commit da23501

Please sign in to comment.