Vasanko/sanic
2
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Add signal handling

Browse Source
This commit is contained in:
messense 2017-03-15 18:01:52 +08:00
parent a90d70feae
commit 11a3cf9b99
No known key found for this signature in database
GPG Key ID: BB41A8A2C716CCA9
1 changed files with 40 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
sanic/worker.py
View File

@ -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)