From e6b5abe909f3e8308d339f2e9dad24d85b1ea05c Mon Sep 17 00:00:00 2001
From: Fabio Manganiello <fabio@manganiello.tech>
Date: Thu, 31 Aug 2023 02:09:23 +0200
Subject: [PATCH] Added SIGTERM handler for clean termination in Docker
 contexts.

---
 platypush/runner/_runner.py | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/platypush/runner/_runner.py b/platypush/runner/_runner.py
index ce2a443a..cf89b6a4 100644
--- a/platypush/runner/_runner.py
+++ b/platypush/runner/_runner.py
@@ -1,4 +1,6 @@
 import logging
+import os
+import signal
 import sys
 from threading import Thread
 from typing import Optional
@@ -23,6 +25,7 @@ class ApplicationRunner:
         logging.basicConfig(level=logging.INFO, stream=sys.stdout)
         self.logger = logging.getLogger('platypush:runner')
         self._proc: Optional[ApplicationProcess] = None
+        self._stream: Optional[CommandStream] = None
 
     def _listen(self, stream: CommandStream):
         """
@@ -48,12 +51,16 @@ class ApplicationRunner:
         if parsed_args.version:
             self._print_version()
 
+        signal.signal(signal.SIGTERM, lambda *_: self.stop())
+
         while True:
-            with CommandStream(parsed_args.ctrl_sock) as stream, ApplicationProcess(
+            with CommandStream(
+                parsed_args.ctrl_sock
+            ) as self._stream, ApplicationProcess(
                 *args, pidfile=parsed_args.pidfile, timeout=self._default_timeout
             ) as self._proc:
                 try:
-                    self._listen(stream)
+                    self._listen(self._stream)
                 except KeyboardInterrupt:
                     pass
 
@@ -63,6 +70,8 @@ class ApplicationRunner:
 
                 break
 
+        self._stream = None
+
     def run(self, *args: str) -> None:
         try:
             self._run(*args)
@@ -73,6 +82,10 @@ class ApplicationRunner:
         if self._proc is not None:
             self._proc.stop()
 
+        if self._stream and self._stream.pid:
+            os.kill(self._stream.pid, signal.SIGKILL)
+            self._stream = None
+
     def restart(self):
         if self._proc is not None:
             self._proc.mark_for_restart()