python - 如何在 Django 请求处理程序线程中对 SIGTERM 执行代码

标签 python django sigterm

我目前正在尝试了解 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/

相关文章:

java - Java 静态方法可以可视化为 python 中的函数 + 具有相同名称(作为函数)的实例方法吗?

python - 如何在没有 count() 的情况下计算查询集中的项目数

django-celery - Celery 任务在本地运行时在正常关闭时重新排队进入代理,但尽管配置相同但在 kubernetes 中丢失

java - 如何使用 SIGKILL Process.destroy() 执行 SIGTERM 在 java 中终止 Linux 进程

python - 当我使用 pygame 加载 Pillow 时,出现错误 : "ValueError: Unrecognized type of format"

python - Luhn 的算法伪代码代码

Python退出所有函数并停止执行

python - 如何在 Django 管理中预填充 UserProfile 字段?

django - 阻止 django-allauth 在确认电子邮件上登录

使用更多进程在 C 中捕获 SIGTERM