decrement server_load on listen for disconnect (#18784)

Signed-off-by: Daniel Salib <danielsalib@meta.com>
This commit is contained in:
daniel-salib 2025-05-28 07:15:12 -07:00 committed by GitHub
parent 435fa95444
commit 6e4cea1cc5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -26,6 +26,11 @@ async def listen_for_disconnect(request: Request) -> None:
while True:
message = await request.receive()
if message["type"] == "http.disconnect":
if request.app.state.enable_server_load_tracking:
# on timeout/cancellation the BackgroundTask in load_aware_call
# cannot decrement the server load metrics.
# Must be decremented by with_cancellation instead.
request.app.state.server_load_metrics -= 1
break