Skip to content

Commit 60e99d0

Browse files
committed
add old names to tasks
1 parent a80f4e1 commit 60e99d0

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

packages/service-library/src/servicelib/fastapi/requests_decorators.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,10 +60,17 @@ def cancel_on_disconnect(handler: _HandlerWithRequestArg):
6060

6161
@wraps(handler)
6262
async def wrapper(request: Request, *args, **kwargs):
63+
sentinel = object()
6364
try:
6465
async with asyncio.TaskGroup() as tg:
65-
handler_task = tg.create_task(handler(request, *args, **kwargs))
66-
tg.create_task(_disconnect_poller_for_task_group(request))
66+
handler_task = tg.create_task(
67+
handler(request, *args, **kwargs),
68+
name=f"cancel_on_disconnect/handler/{handler.__name__}/{id(sentinel)}",
69+
)
70+
tg.create_task(
71+
_disconnect_poller_for_task_group(request),
72+
name=f"cancel_on_disconnect/poller/{handler.__name__}/{id(sentinel)}",
73+
)
6774

6875
return handler_task.result()
6976

0 commit comments

Comments
 (0)