Close the work queue on Executor shutdown

golemfactory · May 27, 2021 · 4b8891f · 4b8891f
1 parent ebe9282
commit 4b8891f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/yapapi/executor/__init__.py b/yapapi/executor/__init__.py
@@ -1113,6 +1113,9 @@ async def worker_starter() -> None:
             cancelled = True
 
         finally:
+
+            await work_queue.close()
+
             # Importing this at the beginning would cause circular dependencies
             from ..log import pluralize
 

diff --git a/yapapi/executor/_smartq.py b/yapapi/executor/_smartq.py
@@ -93,6 +93,7 @@ async def close(self):
         if self._buffer_task:
             self._buffer_task.cancel()
             await self._buffer_task
+            self._buffer_task = None
 
     def finished(self):
         return (