使用自定义信号处理程序与gunicorn

时间:2016-12-11 21:45:03

标签: python flask signals gunicorn

我有一个带有自定义信号处理程序的Flask应用程序,可以在退出之前完成清理任务。当使用gunicorn运行应用程序时,gunicorn会杀死应用程序,然后才能完成所有清理任务。

1 个答案:

答案 0 :(得分:1)

您没有解释自定义信号处理程序的含义,但我不确定您是否应该使用Flask's signals来捕获进程级事件,例如shutdown。相反,您可以使用标准库中的signal module挂钩到SIGTERM信号,如下所示:

# app.py - CREATE THIS FILE
from flask import Flask
from time import sleep, time
import signal
import sys

def create_app():
  signal.signal(signal.SIGTERM, my_teardown_handler)
  app = Flask(__name__)

  @app.route('/')
  def home():
    return 'hi'

  return app


def my_teardown_handler(signal, frame):
  """Sleeps for 3 seconds, then creates/updates a file named app-log.txt with the timestamp."""
  sleep(3)
  with open('app-log.txt', 'w') as f:
    msg = ''.join(['The time is: ', str(time())])
    f.write(msg)
  sys.exit(0)


if __name__ == '__main__':
  app = create_app()
  app.run(port=8888)


# wsgi.py - CREATE THIS FILE, in same folder as app.py
import os
import sys
from werkzeug.wsgi import DispatcherMiddleware
from werkzeug.exceptions import NotFound

from app import create_app
app = DispatcherMiddleware(create_app())

假设您安装了Flask和Gunicorn的虚拟环境,那么您应该可以使用Gunicorn启动应用程序:

$ gunicorn --bind 127.0.0.1:8888 --log-level debug wsgi:app

接下来,在单独的终端中,您可以将TERM信号发送到您的应用,如下所示:

$ kill -s TERM [PROCESS ID OF GUNICORN PROCESS / $(ps ax | grep gunicorn | head -n 1 | awk '{print $1}')]

要观察结果,您应该注意到在{3}延迟后运行app-log.txt命令时,kill文件的内容会更新。您甚至可以在此目录中生成第三个终端窗口并运行watch -n 1 "cat app-log.txt"以观察此文件是否实时更新,同时在启动应用程序和发送TERM信号之间循环。

至于将其投入生产,我知道Supervisor has a configuration option指定stopsignal,如下所示:

[program:my-app]
command = /path/to/gunicorn [RUNTIME FLAGS]
stopsignal = TERM
...

但这与确保应用程序的清理任务完全执行的原始问题不同。