From 11a3cf9b99063a58c5ff06c81db8888fef32cdc9 Mon Sep 17 00:00:00 2001
From: messense <messense@icloud.com>
Date: Wed, 15 Mar 2017 18:01:52 +0800
Subject: [PATCH] Add signal handling

---
 sanic/worker.py | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/sanic/worker.py b/sanic/worker.py
index 3312b485..10b7e19b 100644
--- a/sanic/worker.py
+++ b/sanic/worker.py
@@ -1,4 +1,6 @@
 import os
+import sys
+import signal
 import asyncio
 import logging
 try:
@@ -24,6 +26,7 @@ class GunicornWorker(base.Worker):
             self.ssl_context = None
         self.servers = []
         self.connections = set()
+        self.exit_code = 0
 
     def init_process(self):
         # create new event_loop after fork
@@ -49,6 +52,8 @@ class GunicornWorker(base.Worker):
                            self.loop)
             self.loop.close()
 
+        sys.exit(self.exit_code)
+
     async def close(self):
         trigger_events(self._server_settings.get('before_stop', []),
                        self.loop)
@@ -119,3 +124,38 @@ class GunicornWorker(base.Worker):
         if cfg.ciphers:
             ctx.set_ciphers(cfg.ciphers)
         return ctx
+
+    def init_signals(self):
+        # Set up signals through the event loop API.
+
+        self.loop.add_signal_handler(signal.SIGQUIT, self.handle_quit,
+                                     signal.SIGQUIT, None)
+
+        self.loop.add_signal_handler(signal.SIGTERM, self.handle_exit,
+                                     signal.SIGTERM, None)
+
+        self.loop.add_signal_handler(signal.SIGINT, self.handle_quit,
+                                     signal.SIGINT, None)
+
+        self.loop.add_signal_handler(signal.SIGWINCH, self.handle_winch,
+                                     signal.SIGWINCH, None)
+
+        self.loop.add_signal_handler(signal.SIGUSR1, self.handle_usr1,
+                                     signal.SIGUSR1, None)
+
+        self.loop.add_signal_handler(signal.SIGABRT, self.handle_abort,
+                                     signal.SIGABRT, None)
+
+        # Don't let SIGTERM and SIGUSR1 disturb active requests
+        # by interrupting system calls
+        signal.siginterrupt(signal.SIGTERM, False)
+        signal.siginterrupt(signal.SIGUSR1, False)
+
+    def handle_quit(self, sig, frame):
+        self.alive = False
+        self.cfg.worker_int(self)
+
+    def handle_abort(self, sig, frame):
+        self.alive = False
+        self.exit_code = 1
+        self.cfg.worker_abort(self)