fix: avoid data race when watching request cancellation

This commit is contained in:
beck-8
2026-03-19 00:15:52 +08:00
parent db63f9b5d6
commit b2921518ac

View File

@@ -339,12 +339,13 @@ func (h *BaseAPIHandler) GetContextWithCancel(handler interfaces.APIHandler, c *
} }
} }
newCtx, cancel := context.WithCancel(parentCtx) newCtx, cancel := context.WithCancel(parentCtx)
cancelCtx := newCtx
if requestCtx != nil && requestCtx != parentCtx { if requestCtx != nil && requestCtx != parentCtx {
go func() { go func() {
select { select {
case <-requestCtx.Done(): case <-requestCtx.Done():
cancel() cancel()
case <-newCtx.Done(): case <-cancelCtx.Done():
} }
}() }()
} }