我目前正在尝试了解 Django 在收到 SIGTERM 时的信号处理。
背景资料
我有一个可能需要长时间运行请求的应用程序,它在 Docker 容器中运行。当 Docker 想要停止一个容器时,它首先发送一个 SIGTERM 信号,等待一段时间,然后发送一个 SIGKILL。通常,在 SIGTERM 上,您会停止接收新请求,并希望当前正在运行的请求在 Docker 决定发送 SIGKILL 之前完成。 但是,在我的应用程序中,我想保存已尝试过哪些请求,并且发现这比立即完成请求更重要。所以我更希望当前请求在 SIGTERM 上关闭,这样我就可以优雅地结束它们(并保存它们的状态),而不是等待 SIGKILL。
我的尝试
我的理论是,您可以为 SIGTERM 注册一个信号监听器,它执行 sys.exit()
, 所以 SystemExit
引发异常。然后我想在我的请求处理程序中捕获该异常,并保存我的状态。作为第一个实验,我为 Django 开发服务器创建了一个模拟项目。
我在 Appconfig.ready()
中注册了信号功能:
import signal
import sys
from django.apps import AppConfig
import logging
logger = logging.getLogger(__name__)
def signal_handler(signal_num, frame):
sys.exit()
class TesterConfig(AppConfig):
default_auto_field = 'django.db.models.BigAutoField'
name = 'tester'
def ready(self):
logger.info('starting ready')
signal.signal(signal.SIGTERM, signal_handler)
并创建了一个捕获异常和基本异常的请求处理程序:
import logging
import sys
import time
from django.http import HttpResponse
def handler(request):
try:
logger.info('start')
while True:
time.sleep(1)
except Exception:
logger.info('exception')
except BaseException:
logger.info('baseexception')
return HttpResponse('hallo')
但是当我启动开发服务器用户时python manage.py runserver
然后使用 kill -n 15 <pid>
发送终止信号,没有“baseexception”消息被记录(“start”确实被记录)。
完整代码可见here .
我的问题
我的假设是 SIGTERM 信号在主线程中处理,所以 sys.exit()
调用在主线程中处理。因此在运行请求处理程序的线程中不会引发异常,也不会捕获任何内容。
如何更改我的代码以获得 SystemError
在请求处理程序线程中引发?我需要那个线程的一些信息来记录,所以我不能直接在信号处理程序中“记录”一些东西。
最佳答案
好的,我做了一些调查,并找到了我自己问题的答案。正如我在提出问题时有些怀疑的那样,在这种问题中,您可能想要与所要求的解决方案不同的解决方案。但是,我会在这里发布我的发现,以防将来有人发现问题并发现他和/或她自己处于类似情况。
上面的方法不起作用有几个原因。首先是我忘记在INSTALLED_APPS
中注册我的app,所以TesterConfig.ready
中的代码并没有真正执行。
接下来,事实证明 Django 还为 SIGTERM
信号注册了一个处理程序,参见 the Django source code .因此,如果您向进程发送一个 SIGTERM
,这就是被触发的那个。我暂时在我的虚拟环境中评论了该行以进行更多调查,但当然这永远不会导致真正的解决方案。
sys.exit()
函数确实引发了一个SystemExit
异常,但它只在线程本身中处理。如果您想在线程之间进行通信,您可能需要使用 Event
并定期在您要执行的线程中检查它。
如果您正在寻找如何在通过 gunicorn 运行 Django 时执行类似操作的建议,我发现如果您使用 sync
worker,您可以在 View 中注册信号。 py
,因为请求将在主线程中处理。
最后,我最终在这里注册了信号,并编写了一条日志记录行并在信号处理程序中引发了一个异常。然后由已经存在的异常处理来处理。
关于python - 如何在 Django 请求处理程序线程中对 SIGTERM 执行代码,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/67769562/