alemi/treepuncher
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix: handle SIGTERM too

Browse source
This commit is contained in:
əlemi 2023-11-20 17:51:40 +01:00
parent 5586d1fc9c
commit bd97b891f4
Signed by: alemi
GPG key ID: A4895B84D311642C
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/treepuncher/traits/runnable.py
View file

@ -2,7 +2,7 @@ import asyncio
import logging import logging
from typing import Optional from typing import Optional
from signal import signal, SIGINT, SIGTERM, SIGABRT from signal import signal, SIGINT, SIGTERM
class Runnable: class Runnable:
_is_running : bool _is_running : bool
@ -32,6 +32,10 @@ class Runnable:
else: else:
logging.info("Received SIGINT, stopping gracefully...") logging.info("Received SIGINT, stopping gracefully...")
self._stop_task = asyncio.get_event_loop().create_task(self.stop(force=self._stop_task is not None)) self._stop_task = asyncio.get_event_loop().create_task(self.stop(force=self._stop_task is not None))
if signum == SIGTERM:
logging.info("Received SIGTERM, terminating")
self._stop_task = asyncio.get_event_loop().create_task(self.stop(force=True))
signal(SIGINT, signal_handler) signal(SIGINT, signal_handler)