在使用Celery統(tǒng)計(jì)每日訪問(wèn)數(shù)量的時(shí)候，發(fā)現(xiàn)一個(gè)任務(wù)會(huì)同時(shí)執(zhí)行兩次，發(fā)現(xiàn)同一時(shí)間內(nèi)（1s內(nèi)）竟然同時(shí)發(fā)送了兩次任務(wù)，也就是同時(shí)產(chǎn)生了兩個(gè)worker，造成統(tǒng)計(jì)兩次，一直找不到原因。

參考：http://m.fzitv.net/article/226849.htm

有人使用 Redis 實(shí)現(xiàn)了分布式鎖，然后也有人使用了 Celery Once。

Celery Once 也是利用 Redis 加鎖來(lái)實(shí)現(xiàn), Celery Once 在 Task 類基礎(chǔ)上實(shí)現(xiàn)了 QueueOnce 類，該類提供了任務(wù)去重的功能，所以在使用時(shí)，我們自己實(shí)現(xiàn)的方法需要將 QueueOnce 設(shè)置為 base

@task(base=QueueOnce, once={'graceful': True})

后面的 once 參數(shù)表示，在遇到重復(fù)方法時(shí)的處理方式，默認(rèn) graceful 為 False，那樣 Celery 會(huì)拋出 AlreadyQueued 異常，手動(dòng)設(shè)置為 True，則靜默處理。

另外如果要手動(dòng)設(shè)置任務(wù)的 key，可以指定 keys 參數(shù)

@celery.task(base=QueueOnce, once={'keys': ['a']})
def slow_add(a, b):
    sleep(30)
    return a + b

解決步驟

Celery One允許你將Celery任務(wù)排隊(duì)，防止多次執(zhí)行

安裝

pip install -U celery_once

要求，需要Celery4.0，老版本可能運(yùn)行，但不是官方支持的。

使用celery_once，tasks需要繼承一個(gè)名為QueueOnce的抽象base tasks

Once安裝完成后，需要配置一些關(guān)于ONCE的選項(xiàng)在Celery配置中

from celery import Celery
from celery_once import QueueOnce
from time import sleep

celery = Celery('tasks', broker='amqp://guest@localhost//')

# 一般之前的配置沒(méi)有這個(gè)，需要添加上
celery.conf.ONCE = {
  'backend': 'celery_once.backends.Redis',
  'settings': {
    'url': 'redis://localhost:6379/0',
    'default_timeout': 60 * 60
  }
}

# 在原本沒(méi)有參數(shù)的里面加上base
@celery.task(base=QueueOnce)
def slow_task():
    sleep(30)
    return "Done!"

要確定配置，需要取決于使用哪個(gè)backend進(jìn)行鎖定，查看Backends

在后端，這將覆蓋apply_async和delay。它不影響直接調(diào)用任務(wù)。

在運(yùn)行任務(wù)時(shí)，celery_once檢查是否沒(méi)有鎖定（針對(duì)Redis鍵）。否則，任務(wù)將正常運(yùn)行。一旦任務(wù)完成（或由于異常而結(jié)束），鎖將被清除。如果在任務(wù)完成之前嘗試再次運(yùn)行該任務(wù)，將會(huì)引發(fā)AlreadyQueued異常。

example.delay(10)
example.delay(10)
Traceback (most recent call last):
    ..
AlreadyQueued()
result = example.apply_async(args=(10))
result = example.apply_async(args=(10))
Traceback (most recent call last):
    ..
AlreadyQueued()

graceful：如果在任務(wù)的選項(xiàng)中設(shè)置了once={'graceful': True}，或者在運(yùn)行時(shí)設(shè)置了apply_async，則任務(wù)可以返回None，而不是引發(fā)AlreadyQueued異常。

from celery_once import AlreadyQueued
# Either catch the exception,
try:
    example.delay(10)
except AlreadyQueued:
    pass
# Or, handle it gracefully at run time.
result = example.apply(args=(10), once={'graceful': True})
# or by default.
@celery.task(base=QueueOnce, once={'graceful': True})
def slow_task():
    sleep(30)
    return "Done!"

其他功能請(qǐng)?jiān)L問(wèn)：https://pypi.org/project/celery_once/

到此這篇關(guān)于通過(guò)celery_one避免Celery定時(shí)任務(wù)重復(fù)執(zhí)行的文章就介紹到這了,更多相關(guān)Celery定時(shí)任務(wù)重復(fù)執(zhí)行內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論