pipeline(runner): add support for SIGTERM

This commit is contained in:
Aleix Conchillo Flaqué
2024-06-04 15:08:11 -07:00
parent 489060881d
commit c8d37a7227
2 changed files with 10 additions and 3 deletions

View File

@@ -43,11 +43,15 @@ class PipelineRunner:
loop = asyncio.get_running_loop()
loop.add_signal_handler(
signal.SIGINT,
lambda *args: asyncio.create_task(self._sigint_handler())
lambda *args: asyncio.create_task(self._sig_handler())
)
loop.add_signal_handler(
signal.SIGTERM,
lambda *args: asyncio.create_task(self._sig_handler())
)
async def _sigint_handler(self):
logger.warning(f"Ctrl-C detected. Canceling runner {self}")
async def _sig_handler(self):
logger.warning(f"Interruption detected. Canceling runner {self}")
await self.cancel()
def __str__(self):