在 Celery 工作人员中捕获 Heroku SIGTERM 以优雅地关闭工作人员

首先,我们需要在 Celery 工作人员中捕获 Heroku SIGTERM 信号。这可以通过定义一个信号处理函数来实现,并在应用程序启动时注册这个函数。以下是一个简单的步骤说明:

1. 在你的 `tasks.py` 文件中,添加一个信号处理函数。这个函数应该能够优雅地关闭 Celery 工作人员。例如:
```python
from celery import Celery
import signal

app = Celery('tasks', broker='amqp://guest@localhost//')

# 定义信号处理函数
def sigterm_handler(signal, frame):
    print("Caught SIGTERM, shutting down gracefully...")
    app.control.shutdown()  # 关闭 Celery 工作人员
    sys.exit(0)

# 注册信号处理函数
signal.signal(signal.SIGTERM, sigterm_handler)
```

2. 在你的 `main.py` 文件中,启动 Celery 工作人员。例如:
```python
from celery.bin import worker
from tasks import app  # 导入你的 Celery 应用

def start():
    worker = worker.Worker(app=app)
    worker.run()  # 启动 Celery 工作人员

if __name__ == '__main__':
    start()
```

3. 在你的 Heroku 应用程序中,设置启动脚本为 `python main.py`。例如,在 `Procfile` 文件中添加以下内容:
```
web: python main.py
worker: celery -A tasks worker --loglevel=info
```

以上步骤确保了在 Heroku 上运行 Celery 应用时,当接收到 SIGTERM 信号时,Celery 工作人员会优雅地关闭。

测试用例:

1. 在你的本地环境中,运行以下命令启动 Celery 工作人员:
```bash
celery -A tasks worker --loglevel=info
```

2. 在另一个终端中,发送 SIGTERM 信号给 Celery 工作人员进程(使用 `kill` 命令):
```bash
kill <worker_process_id>
```

3. 检查 Celery 工作人员是否已经关闭。你应该看到以下输出:
```
Caught SIGTERM, shutting down gracefully...
[2019-10-01 12:34:56,019: INFO/MainProcess] Shutting down...
[2019-10-01 12:34:56,020: INFO/MainProcess] Stopping consumer:
```

这个测试用例验证了在 Celery 工作人员中捕获 Heroku SIGTERM 信号的工作情况。

  • 3
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

潮易

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值