celery 定时任务实现

Celery 是什么?

  • 异步任务队列工具,主要解决 realtime 事件的异步操作,但也支持定时任务。
  • 什么是异步?那要先理解什么是同步,比如我去麦当劳吃饭,如果麦当劳前一个顾客点完单,拿到餐,吃完走人之后才能接待下一个顾客,就是同步。反过来我点完单,它马上就接下一个客人的单,我的流程虽然还没有走完(time-consuming),但也不影响下一个顾客点单(blocked),这就是异步。

Celery 安装

  • pip install celery,由于 celery 自己并不带队列存储,所以根据官方推荐,还需要安装 RabbitMQ 或者 Redis 来存储队列。方便起见,本文用 Redis。

Celery 机制

  1. celery 是一个装饰器类,本质上是把一个函数变成一个可以异步调用的函数。
  2. 开启 celery 进程。
  3. 在另一段程序中导入这个函数,当这个函数被以 delay 模式调用时:xxx.delay(),celery 会把这个任务写到任务队列然后返回,原程序不会被阻塞可以往下跑。注意这里需要用 delay 模式去跑这个函数,同时注意不要把 xxx.delay() 的结果赋值给一个变量,否则依然会被阻塞。
  4. celery 的另一个进程会去这个任务队列里取任务,完成之后写到 result 队列里面。或者多数情况下,这个被异步调用的函数不需要返回结果,比如发送一个邮件,提醒之类的,连 result 队列都不用。

Celery first blood

  1. 开一个 python 脚本,比如tasks.py

  2. 生成一个 Celery 对象实例,是一个装饰器。

    from celery import Celery
    app = Celery('__name__', broker='redis://localhost:6379') 
    
  3. 定义一个你想异步操作的函数,并加上 celery 装饰器 @app.task

    @app.task
    def add(x, y):
        return x + y
    
  4. 保存,退出,然后在 terminal 启动 celery 服务。

    celery -A tasks worker --loglevel=info
    

做个定时任务:每天发问候

  1. 接下来,我要搭配钉钉机器人了,我希望小仙女每天早上7点给我发个问候,然后在7点半的时候确认我有没有开始干活了。

  2. 先写一个简单的钉钉提醒程序,命名celery_worker.py

    #! /usr/bin/env python
    # coding: utf-8
    
    import requests
    import json
    import time
    
    from config import HOST_IP, NOTIFY_URL, MOBILE_NUMBER
    
    def notify_dingding(msg):
        headers = {"Content-Type": "application/json; charset=utf-8"}
        post_data = {
            "msgtype": "text",
            "text": {
                "content": msg
            },
            "at": {
                "atMobiles": [MOBILE_NUMBER]
            }
        }
    
        r = requests.post(NOTIFY_URL, headers=headers, data=json.dumps(post_data))
        print(r.content)
    

    注意这里从 config 里导入了一些参数,所以要在这个程序的同一层写一个 config.py 的配置文件。

    # config.py
    NOTIFY_URL = ("https://oapi.dingtalk.com/robot/send?access_token="
        "c6d5a2936381dfc29394f3c336bea5fad962d90ffd31809e92d95a1xxxxxxxx")
    MOBILE_NUMBER = "176xxxxx619"
    HOST_IP = "127.0.0.1"
    
  3. 导入 celery 包,给函数加上装饰器:

    from celery import Celery
    
    BROKER_URI = 'redis://%s:6379/6' % HOST_IP
    BACKEND_URI = 'redis://%s:6379/5' % HOST_IP
    
    worker = Celery('celery_worker', broker=BROKER_URI, backend=BACKEND_URI)
    
    @worker.task
    def notify_dingding(msg):
        ...
    
  4. 简单学习一下celery 的 crontab 定时任务

  5. 给 worker 加上定时任务

    from celery.schedules import crontab
    
    worker.conf.update(
        timezone='Asia/Shanghai',
        enable_utc=True,
        beat_schedule={
            "morning_msg_1": {
                "task": "celery_worker.notify_dingding",
                "schedule": crontab(minute=0, hour=7),
                "args": ("早,起床了哟,先去做个早饭吧",)
            },
            "morning_msg_2": {
                "task": "celery_worker.notify_dingding",
                "schedule": crontab(minute=30, hour=7),
                "args": ("我就问问你在干活咩?",)
            }
        }
    )
    
  6. 最后程序末尾加一个小测试,看看服务是不是起来了:

    notify_dingding("小仙女上线啦")
    
  7. 开启 redis-server

    nohup redis-server &
    
  8. 开启我们的 celery worker,这里的-B 是 celery 的 beat 服务,可以理解为一个周期任务。

    celery -A celery_worker worker -B
    
  9. 服务起来喽:

QQ20170818-163110.jpg
QQ20170818-162958.jpg
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,214评论 6 481
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,307评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 152,543评论 0 341
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,221评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,224评论 5 371
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,007评论 1 284
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,313评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,956评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,441评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,925评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,018评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,685评论 4 322
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,234评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,240评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,464评论 1 261
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,467评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,762评论 2 345

推荐阅读更多精彩内容