Celery Once 也是利用 Redis 加鎖來實現(xiàn), Celery Once 在 Task 類基礎(chǔ)上實現(xiàn)了 QueueOnce 類，該類提供了任務(wù)去重的功能，今天通過本文給大家介紹通過celery_one避免Celery定時任務(wù)重復(fù)執(zhí)行的問題，感興趣的朋友一起看看吧

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

參考：http://www.dbjr.com.cn/article/226849.htm

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

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

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

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

另外如果要手動設(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ù)排隊，防止多次執(zhí)行

安裝

pip install -U celery_once

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

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

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

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

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

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

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

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

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

在運行任務(wù)時，celery_once檢查是否沒有鎖定（針對Redis鍵）。否則，任務(wù)將正常運行。一旦任務(wù)完成（或由于異常而結(jié)束），鎖將被清除。如果在任務(wù)完成之前嘗試再次運行該任務(wù)，將會引發(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ù)的選項中設(shè)置了once={'graceful': True}，或者在運行時設(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!"

其他功能請訪問：https://pypi.org/project/celery_once/

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

您可能感興趣的文章: