don't keep client tasks

blacklanternsecurity · Aug 23, 2024 · bb6accc · bb6accc
1 parent 9eb9152
commit bb6accc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/bbot/core/engine.py b/bbot/core/engine.py
@@ -563,6 +563,9 @@ async def _shutdown(self):
 
     def new_child_task(self, client_id, coro):
         task = asyncio.create_task(coro)
+        def remove_task():
+            self.child_tasks.get(client_id, set()).discard(task)
+        task.add_done_callback(remove_task)
         try:
             self.child_tasks[client_id].add(task)
         except KeyError: