编辑2:
出于某种原因,我的头最初被困在重复任务的领域。这是一个更简单的解决方案。
您真正需要的是为每个用户操作定义一个任务。您可以跳过在数据库中执行的存储任务——这就是 celery 的用途!
再次重用您的 facebook 帖子示例,并再次假设您在post_to_facebook
某处有一个函数,它需要一个用户和一些文本,执行一些魔法,并将文本发布到该用户的 facebook,您可以将其定义为这样的任务:
# Task to send one update.
@celery.task(ignore_result=True)
def post_to_facebook(user, text):
# perform magic
return whatever_you_want
当用户准备好将此类帖子排入队列时,您只需告诉 celery 何时运行该任务:
post_to_facebook.apply_async(
(user, text), # args
eta=datetime.datetime(2012, 9, 15, 11, 45, 4, 126440) # pass execution options as kwargs
)
这一切都在这里详细说明,在一大堆可用的调用选项中: http: //docs.celeryproject.org/en/latest/userguide/calling.html#eta-and-countdown
如果需要调用结果,可以跳过任务定义中的 ignore_result 参数,取回一个 AsyncResult 对象,然后检查调用结果。更多信息:http: //docs.celeryproject.org/en/latest/getting-started/first-steps-with-celery.html#keeping-results
下面的一些答案仍然是相关的。您仍然希望每个用户操作都有一个任务,您仍然想考虑任务设计等,但这是完成您所要求的事情的更简单的方法。
使用重复任务的原始答案如下:
Dannyroa 的想法是正确的。我将在此基础上进行一点介绍。
编辑/ TLDR:
答案是肯定的,芹菜适合您的需求。您可能只需要重新考虑您的任务定义。
我假设您不允许您的用户编写任意 Python 代码来定义他们的任务。除此之外,您必须预定义用户可以安排的一些操作,然后允许他们根据需要安排这些操作。然后,您可以为每个用户操作运行一个计划任务,检查条目并为每个条目执行操作。
一项用户操作:
使用您的 Facebook 示例,您可以将用户的更新存储在一个表中:
class ScheduledPost(Model):
user = ForeignKey('auth.User')
text = TextField()
time = DateTimeField()
sent = BooleanField(default=False)
然后,您将每分钟运行一次任务,检查该表中计划在最后一分钟发布的条目(基于您提到的误差范围)。如果达到一分钟窗口非常重要,则可以更频繁地安排任务,例如每 30 秒一次。任务可能如下所示(在 myapp/tasks.py 中):
@celery.task
def post_scheduled_updates():
from celery import current_task
scheduled_posts = ScheduledPost.objects.filter(
sent=False,
time__gt=current_task.last_run_at, #with the 'sent' flag, you may or may not want this
time__lte=timezone.now()
)
for post in scheduled_posts:
if post_to_facebook(post.text):
post.sent = True
post.save()
配置可能如下所示:
CELERYBEAT_SCHEDULE = {
'fb-every-30-seconds': {
'task': 'tasks.post_scheduled_updates',
'schedule': timedelta(seconds=30),
},
}
其他用户操作:
除了发布到 Facebook 之外,对于每个用户操作,您都可以定义一个新表和一个新任务:
class EmailToMom(Model):
user = ForeignKey('auth.User')
text = TextField()
subject = CharField(max_length=255)
sent = BooleanField(default=False)
time = DateTimeField()
@celery.task
def send_emails_to_mom():
scheduled_emails = EmailToMom.objects.filter(
sent=False,
time__lt=timezone.now()
)
for email in scheduled_emails:
sent = send_mail(
email.subject,
email.text,
email.user.email,
[email.user.mom.email],
)
if sent:
email.sent = True
email.save()
CELERYBEAT_SCHEDULE = {
'fb-every-30-seconds': {
'task': 'tasks.post_scheduled_updates',
'schedule': timedelta(seconds=30),
},
'mom-every-30-seconds': {
'task': 'tasks.send_emails_to_mom',
'schedule': timedelta(seconds=30),
},
}
速度和优化:
post_scheduled_updates
为了获得更高的吞吐量,您可以生成一堆子任务并并行执行(给定足够的工人),而不是迭代更新以发布并在调用期间连续发送它们。然后,对的调用post_scheduled_updates
运行得非常快,并安排了一大堆任务——每个 fb 更新一个任务——尽快运行。看起来像这样:
# Task to send one update. This will be called by post_scheduled_updates.
@celery.task
def post_one_update(update_id):
try:
update = ScheduledPost.objects.get(id=update_id)
except ScheduledPost.DoesNotExist:
raise
else:
sent = post_to_facebook(update.text)
if sent:
update.sent = True
update.save()
return sent
@celery.task
def post_scheduled_updates():
from celery import current_task
scheduled_posts = ScheduledPost.objects.filter(
sent=False,
time__gt=current_task.last_run_at, #with the 'sent' flag, you may or may not want this
time__lte=timezone.now()
)
for post in scheduled_posts:
post_one_update.delay(post.id)
我发布的代码没有经过测试,当然也没有优化,但它应该让你走上正轨。在您的问题中,您暗示了对吞吐量的一些担忧,因此您需要仔细查看要优化的地方。一个明显的问题是批量更新而不是迭代调用post.sent=True;post.save()
.
更多信息:
有关定期任务的更多信息:http: //docs.celeryproject.org/en/latest/userguide/periodic-tasks.html。
关于任务设计策略的部分:http: //docs.celeryproject.org/en/latest/userguide/tasks.html#performance-and-strategies
这里有一整页关于优化 celery:http: //docs.celeryproject.org/en/latest/userguide/optimizing.html。
这个关于子任务的页面也可能很有趣:http ://docs.celeryproject.org/en/latest/userguide/canvas.html 。
事实上,我建议阅读所有 celery 文档。