github-mirror/caddy
github-mirror/caddy
2
0
Fork 0
mirror of https://github.com/caddyserver/caddy.git synced 2025-04-23 05:07:50 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

feat: change way to call delayCtx after roundtrip

Browse Source
This commit is contained in:
Lucas Botelho 2024-10-31 15:31:23 -03:00
parent 1beff2fcd6
commit 9dbc182ab0
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/caddyhttp/reverseproxy/reverseproxy.go
View File

@ -840,8 +840,10 @@ func (h *Handler) reverseProxy(rw http.ResponseWriter, req *http.Request, origRe
// user probably wants us to finish sending the data to the upstream regardless,
// and we should expect client disconnection in low-latency streaming scenarios
// (see issue #4922)
var delayCtx *delayClientDoneContext
if h.CloseAfterReceivedBody {
req = req.WithContext(NewDelayClientDoneContext(req.Context(), h.ctx.Done()))
delayCtx = NewDelayClientDoneContext(req.Context(), h.ctx.Done())
req = req.WithContext(delayCtx)
}
// do the round-trip
@ -850,7 +852,7 @@ func (h *Handler) reverseProxy(rw http.ResponseWriter, req *http.Request, origRe
duration := time.Since(start)
if h.CloseAfterReceivedBody {
req.Context().(*delayClientDoneContext).RoundTripDone()
delayCtx.RoundTripDone()
}
// record that the round trip is done for the 1xx response handler