如何解决如何在 Django 请求处理程序线程中的 SIGTERM 上执行代码
我目前正在尝试了解接收 SIGTERM 时 Django 中的信号处理。
背景信息
我有一个在 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
中注册我的应用程序,因此 TesterConfig.ready
中的代码实际上并未执行。
接下来,事实证明 Django 还为 SIGTERM
信号注册了一个处理程序,参见 the Django source code。因此,如果您向流程发送 SIGTERM
,则会触发该流程。我暂时在我的虚拟环境中评论了该行以进行更多调查,但这当然永远不会导致真正的解决方案。
sys.exit()
函数确实引发了 SystemExit
异常,但仅在线程本身中处理。如果您想在线程之间进行通信,您可能需要使用 Event
并在要执行的线程中定期检查它。
如果您正在寻找通过 gunicorn 运行 Django 时如何执行此类操作的建议,我发现如果您使用 sync
工作器,您可以在您的 views.py
中注册信号,因为请求将在主线程中处理。
最后,我最终在这里注册了信号,并在信号处理程序中编写了一个日志记录行并引发了一个异常。然后由已经存在的异常处理来处理。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。