Celery 是一个广泛应用于网络应用程序的任务处理系统。
它可以在以下情况下使用:
在请求响应周期中做网络调用。服务器应当立即响应任何网络请求。如果在请求响应周期内需要进行网络调用,则应在周期外完成调用。例如当用户在网站上注册时,需要发送激活邮件。发送邮件是一种网络调用,耗时 2 到 3 秒。用户应该无需等待这 2 到 3 秒。因此,发送激活邮件应当在请求响应周期外完成,celery 就能实现这一点。
将一个由几个独立部分组成的大任务分成多个小任务。假设你想知道脸书用户的时间流。脸书提供不同的端点来获取不同的数据。譬如,一个端点用以获取用户时间流中的图片,一个端点获取用户时间流中的博文,一个端点得到用户的点赞信息等。如果你的函数需要和脸书的 5 个端点依此通信,每个网络调用平均耗时 2 秒,你将需要 10 秒完成一次函数执行。但是,你可以把这项工作分为 5 个独立的任务(你很快就会发现这很容易做到),并让 celery 来处理这些任务。Celery 可以并行地与这 5 个端点通信,在 2 秒之内就能得到所有端点的响应。
假设我们有一个函数,并传给它一个网址列表。该函数需要获取这些网址的响应。
创建文件celery_blog.py
:
import requests
import time
def func(urls):
start = time.time()
for url in urls:
resp = requests.get(url)
print resp.status_code
print "It took", time.time() - start, "seconds"
if __name__ == "__main__":
func(["http://oneapm.com", "http://jd.com", "https://taobao.com", "http://baidu.com", "http://news.oneapm.com"])
运行:
python celery_blog.py
输出:
##### 使用 celery
调用 celery 的程序中最重要的组成部分为 celery worker。
在 web 应用程序注册的例子中,celery worker 用于发送邮件。
在脸书的例子中, celery worker 用于获取不同的网址。
在我们的 celery_blog.py
例子中, celery worker 用于获取 URL。
celery worker 和你的应用程序/脚本是不同的进程,彼此独立运行。所以你的应用程序/脚本和 celery 需要一些方法来相互沟通。
应用程序代码需要把任务放在 celery worker 可以取出并执行的位置。譬如,应用程序代码将任务放在消息队列中,celery worker 从消息队列领取任务并执行任务。我们将使用 Redis 作为消息队列。
请确认你已安装 Redis,并可以运行redis-server
。
请确认你已安装 celery。
修改文件 celery_blog.py
,如下:
from celery import Celery
app = Celery('celery_blog',bloker='redis://localhost:6379/1')
@app.task
def fetch_url(url):
resp = requests.get(url)
print resp.status_code
def func(urls):
for url in urls:
fetch_url.delay(url)
if __name__ == "__main__":
func(["http://oneapm.com", "http://jd.com", "https://taobao.com", "http://baidu.com", "http://news.oneapm.com"])
代码解释:我们需要一个 celery 实例来启动程序,因此创建了一个名为 app 的 celery 实例。
在 3 个终端中启动:
第一个终端,运行 redis-server
第二个终端,运行 celery worker -A celery_blog -l info -c 5
,通过输出可以看到 celery 成功运行。
第三个终端,运行脚本 python celery_blog.py
可以看到第二个终端输出如下:
上面的例子中,我们只写了一个 celery 任务。但您的项目可能涉及多个模块,您可能希望在不同的模块中有不同的任务。所以让我们将 celery 配置移到单独的文件中。
创建 celery_config.py
from celery import Celery
app = Celery('celery_config', broker='redis://localhost:6379/0', include=['celery_blog'])
修改 celery_blog.py
代码如下:
import requests
from celery_config import app
@app.task
def fetch_url(url):
resp = requests.get(url)
print resp.status_code
def func(urls):
for url in urls:
fetch_url.delay(url)
if __name__ == "__main__":
func(["http://oneapm.com", "http://jd.com", "https://taobao.com", "http://baidu.com", "http://news.oneapm.com"])
停掉之前的 celery worker
,运行:
celery worker -A celery_config -l info -c 5
打开 ipython ,运行如下命令:
In [1]: from celery_blog import func
In [2]: func(["http://oneapm.com", "http://jd.com", "https://taobao.com", "http://baidu.com", "http://news.oneapm.com"])
输出如下:
##### 在不同文件中添加新的任务
您可以添加新的模块,并在该模块中定义一个任务。用以下内容创建一个模块 celery_add.py
:
from celery_config import app
@app.task
def add(a, b):
return a + b
改变 celery_config.py
包含新的模块 celery_add.py
,如下:
from celery import Celery
app = Celery('celery_config', broker='redis://localhost:6379/0', include=['celery_blog', 'celery_add'])
在 ipython 输入:
In [1]: from celery_add import add
In [2]: add.delay(4, 5)
输出如下:
到目前为止,我们的脚本、celery worker 和 Redis 都运行在同一机器中。其实并无这种必要,这三者可以运行在不同机器上。
celery 任务涉及到网络请求,因此,在网络优化的机器上使用 celery worker 能提高任务运行速度。Redis 是一种内存数据库,在内存优化的机器上运行效率更高。
在这个例子中,我将在本地系统运行脚本和 celery worker,在分开的服务器上运行 Redis。
修改 celery_config.py
为:
app = Celery('celery_config', broker='redis://192.168.118.148:6379/0', include=['celery_blog'])
现在我运行任何任务,脚本都将把他放在 Redis 运行的服务器(192.168.118.148)上面。
celery worker 也与 192.168.118.148 沟通,在这个 Redis 服务器上得到任务并执行它。
注意:您必须使用正在运行 redis-server 的服务器地址。我的服务器已停止 Redis,所以你将无法连接到 Redis。
参考文章:Getting started with Celery and Redis
本文系 OneAPM 工程师整理。OneAPM 是应用性能管理领域的新兴领军企业,能帮助企业用户和开发者轻松实现:缓慢的程序代码和 SQL 语句的实时抓取。想阅读更多技术文章,请访问 OneAPM 官方博客。