phzhik/chain-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

* Fixed tasks cancellation on app shutdown

Browse Source
This commit is contained in:
Phil Zhitnikov 2024-08-05 00:01:55 +04:00
parent 095896d847
commit b387604b25
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
chain_service/app.py
View File

@ -25,12 +25,13 @@ async def startup_event():
@application.on_event("shutdown") @application.on_event("shutdown")
async def shutdown_event(): async def shutdown_event():
async with application.state.tasks_lock: async with application.state.tasks_lock:
for task in application.state.tasks.values(): for task_list in application.state.tasks.values():
try: for task in task_list:
task.cancel() try:
except asyncio.CancelledError: task.cancel()
pass except asyncio.CancelledError:
await asyncio.gather(*application.state.tasks.values(), return_exceptions=True) pass
application.state.tasks.clear() application.state.tasks.clear()