Forceably terminate greenlets with TERM signal.

coleifer · Jul 19, 2023 · 6284422 · 6284422
1 parent 4180cbe
commit 6284422
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/huey/consumer.py b/huey/consumer.py
@@ -522,8 +522,8 @@ def _handle_stop_signal(self, sig_num, frame):
         self._restart = False
         self._graceful = False
         if self.worker_type == WORKER_GREENLET:
-            # Ensure that the running greenlet (if not main) is interrupted.
-            raise KeyboardInterrupt
+            gevent.killall([t for _, t in self.worker_threads],
+                           KeyboardInterrupt)
 
     def _handle_restart_signal(self, sig_num, frame):
         self._logger.info('Received SIGHUP, will restart')