首先,我们需要在 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 信号的工作情况。