Celery笔记六之worker介绍

前面我们介绍过 celery 的理想的设计方式是几个 worker 处理特定的任务队列的数据，这样可以避免任务在队列中的积压。

这一篇笔记我们介绍一下如何使用 worker 提高系统中任务的处理效率。

1. worker启动
2. worker与队列
3. worker检测
4. 其他worker命令

1、worker 启动

前面介绍过 worker 的启动方式，在 celery 配置文件的上一级目录运行下面的命令：

celery -A hunter worker -l INFO
1

其中，-l 表示日志等级，相当于是 --loglevel=INFO

celery -A hunter worker --loglevel=INFO
1

指定worker的hostname

celery -A hunter worker -l INFO -n worker1@%h
1

其中，%h 表示主机名，包含域名在内，%n 表示仅包含主机名，%d 表示仅包含域名。

以下是示例：

指定日志文件地址

logfile 参数可以指定日志文件地址：

celery -A hunter worker --loglevel=INFO --logfile=/Users/hunter/python/celery_log/celery.log
1

杀死 worker 进程

我们可以通过获取 worker 的进程 id 来杀死这些进程：

ps aux | grep 'celery -A hunter' | awk '{print $2}' |xargs sudo kill -9
1

并发处理

一般来说，当我们直接启动 worker 的时候，会默认同时起好几个 worker 进程。

如果不指定 worker 的数量，worker 的进程会默认是所在机器的 CPU 的数量。

我们也可以通过 concurrency 参数来指定启动 worker 的进程数。

比如说，我们想启动三个 worker 的进程，可以如下指定：

celery -A hunter worker --concurrency=3 -l INFO
1

–concurrency 也可以简写成 -c:

celery -A hunter worker -c 3 -l INFO
1

这样，我们在启动的命令行里输入下面的参数就可以看到启动了三个 worker 的进程：

ps aux |grep 'celery -A hunter'
1

这里有一个关于 worker 进程数启动多少的问题，是不是我们的 worker 启动的越多，我们的定时任务和延时任务就会执行得越快呢？

并不是，有实验证明 worker 的数量启动得越多，对于 task 处理的性能有可能还会起到一个反向作用，这里不作展开讨论，我们可以设置 CPU 的数量即可。

当然，你也可以根据 worker 处理任务的情况，基于 application，基于工作负载，任务运行时间等试验出一个最佳的数量。

2、worker与队列

消费指定队列的task

我们可以在运行 worker 的时候指定 worker 只消费特定队列的 task，这个特定队列，可以是一个，也可以是多个，用逗号分隔开。

指定的方式如下：

celery -A hunter worker -l INFO -Q queue_1,queue_2
1

列出所有活跃的queues

下面的命令可以列出所有系统活跃的队列信息：

celery -A hunter inspect active_queues
1

假设目前我们相关配置如下：

app.conf.task_queues = (
    Queue('default_queue',),
    Queue('queue_1'),
    Queue('queue_2'),
)

app.conf.task_routes = {
    'blog.tasks.add': {
        'queue': 'queue_1',
    },
    'blog.tasks.minus': {
        'queue': 'queue_2',
    },
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14

我们这样启动worker：

celery -A hunter worker -l INFO -c 3 -n worker1@%h
1

然后运行上面的查看队列命令：

celery -A hunter inspect active_queues
1

可以看到如下输出：

->  worker1@localhost: OK
    * {'name': 'default_queue', 'exchange': {...}, 'routing_key': 'default_queue', ...}
    * {'name': 'queue_1', 'exchange': {...}, 'routing_key': 'default_queue', ...}
    * {'name': 'queue_2', 'exchange': {...}, 'routing_key': 'default_queue', ...}

1 node online.

1
2
3
4
5
6
7

其中，输出结果最上面的 worker1@localhost 就是我们启动 worker 通过 -n 指定的 hostnam，可以通过这个来指定 worker。

我们可以指定 worker 输出对应的队列数据：

celery -A hunter inspect active_queues -d worker1@localhost
1

除了命令行，我们也可以在交互界面来获取这些数据：

# 获取所有的队列信息
from hunter.celery import app
app.control.inspect().active_queues()

# 获取指定 worker 的队列信息
app.control.inspect(['worker1@localhost']).active_queues()
1
2
3
4
5
6

3、worker 的检测

app.control.inspect() 函数可以检测正在运行的 worker 信息，我们可以用下面的命令来操作：

from hunter.celery import app

i = app.control.inspect()
1
2
3

这个操作是获取所有节点，我们也可以指定单个或者多个节点检测：

# 输入数组参数，表示获取多个节点worker信息
i = app.control.inspect(['worker1@localhost', 'worker2@localhost'])

# 输入单个worker名，指定获取worker信息
i = app.control.inspect('worker1@localhost')
1
2
3
4
5

获取已经注册的task列表

用到前面的 app.control.inspect() 函数和其下的 registered() 函数

i.registered()

# 输出结果为 worker 及其下的 task name 
# 输出示例为 {'worker1@localhost': ['blog.tasks.add', 'blog.tasks.minus', 'polls.tasks.multi']}
1
2
3
4

输出的格式是一个 dict，worker 的名称为 key，task 列表为 value

正在执行的 task

active() 用于获取正在执行的 task 函数

i.active()

# 输出 worker 正在执行的 task
# 输出示例为 {'worker1@localhost': [{'id': 'xxx', 'name': 'blog.tasks.add', 'args': [3, 4], 'hostname': 'worker1@localhost', 'time_start': 1659450162.58197, ..., 'worker_pid': 41167}
1
2
3
4

输出的结果也是一个 dict，每个 worker 下有 n 个正在 worker 中执行的 task 信息，这个 n 的最大数量取决于前面我们启动 worker 时的 --concurrency 参数。

在其中的 task 信息里包含 task_id，task_name，和输入的参数，开始时间，worker name 等。

即将运行的 task

比如我们运行 add 延时任务，定时在 20s 之后运行：

add.apply_async((1, 1), countdown=20)
1

返回的结果每个 worker 下有一个任务列表，每个列表存有任务的信息：

i.scheduled()

# 输出信息如下
# {'worker1@localhost': [{'eta': '2022-08-02T22:56:49.503517+08:00', 'priority': 6, 'request': {'id': '23080c03-a906-4cc1-9ab1-f27890c58adb', 'name': 'blog.tasks.add', 'args': [1, 1], 'kwargs': {}, 'type': 'blog.tasks.add', 'hostname': 'worker1@localhost', 'time_start': None, 'acknowledged': False, 'delivery_info': {...}}]}
1
2
3
4

queue队列中等待的 task

如果我们有任务在 queue 中积压，我们可以使用：

i.reserved()
1

来获取队列中等待的 task 列表

4、其他 worker 命令

ping-pong

检测 worker 还活着的 worker

使用 ping() 函数，可以得到 pong 字符串的回复表明该 worker 是存活的。

from hunter.celery import app

app.control.ping(timeout=0.5)

# [{'worker1@localhost': {'ok': 'pong'}}]
1
2
3
4
5

我们也可以指定 worker 来操作：

app.control.ping(['worker1@localhost'])
1

如果你了解 redis 的存活检测操作的话，应该知道在 redis-cli 里也可以执行这个 ping-pong 的一来一回的检测操作。

本文首发于微信公众号，可搜索关注：Hunter后端

原文链接：celery笔记六之worker介绍