celery笔记六之worker介绍

谁践踏了优雅 2024-03-17 13:36 37阅读 0赞

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

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

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

#### 1、worker 启动 ####

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

celery -A hunter worker -l INFO

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

celery -A hunter worker --loglevel=INFO

**指定worker的hostname**

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

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

以下是示例：

<table> 
 <thead> 
  <tr> 
   <th>变量</th> 
   <th>示例</th> 
   <th>结果</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>%h</td> 
   <td>worker1@%h</td> 
   <td>worker1@george.example.com</td> 
  </tr> 
  <tr> 
   <td>%n</td> 
   <td>worker1@%n</td> 
   <td>worker1@george</td> 
  </tr> 
  <tr> 
   <td>%d</td> 
   <td>worker1@%d</td> 
   <td>worker1@example.com</td> 
  </tr> 
 </tbody> 
</table>

**指定日志文件地址**

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

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

**杀死 worker 进程**

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

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

**并发处理**

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

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

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

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

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

\--concurrency 也可以简写成 -c:

celery -A hunter worker -c 3 -l INFO

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

ps aux |grep 'celery -A hunter'

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

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

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

#### 2、worker与队列 ####

**消费指定队列的task**

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

指定的方式如下：

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

**列出所有活跃的queues**

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

celery -A hunter inspect active_queues

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

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',
        },
    }

我们这样启动worker：

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

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

celery -A hunter inspect active_queues

可以看到如下输出：

->  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.

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

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

celery -A hunter inspect active_queues -d worker1@localhost

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

# 获取所有的队列信息
    from hunter.celery import app
    app.control.inspect().active_queues()
    
    # 获取指定 worker 的队列信息
    app.control.inspect(['worker1@localhost']).active_queues()

#### 3、worker 的检测 ####

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

from hunter.celery import app
    
    i = app.control.inspect()

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

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

**获取已经注册的task列表**

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

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

输出的格式是一个 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}

输出的结果也是一个 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)

返回的结果每个 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': {...}}]}

**queue队列中等待的 task**

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

i.reserved()

来获取队列中等待的 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'}}]

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

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

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