我有一个Django应用程序将对象保存到数据库中,还有一个芹菜任务定期对其中一些对象进行一些处理。问题是,用户可以在芹菜任务选择对象进行处理后删除对象,但在芹菜任务实际完成处理和保存之前删除对象。因此,当芹菜任务调用.save()
时,即使用户删除了对象,该对象也会重新出现在数据库中。当然,这对用户来说真的很可怕。
下面是一些显示问题的代码:
def my_delete_view(request, pk):
thing = Thing.objects.get(pk=pk)
thing.delete()
return HttpResponseRedirect('yay')
@app.task
def my_periodic_task():
things = get_things_for_processing()
# if the delete happens anywhere between here and the .save(), we're hosed
for thing in things:
process_thing(thing) # could take a LONG time
thing.save()
我想通过添加一个原子块和一个事务来修复它,在保存之前测试对象是否真的存在:
@app.task
def my_periodic_task():
things = Thing.objects.filter(...some criteria...)
for thing in things:
process_thing(thing) # could take a LONG time
try:
with transaction.atomic():
# just see if it still exists:
unused = Thing.objects.select_for_update().get(pk=thing.pk)
# no exception means it exists. go ahead and save the
# processed version that has all of our updates.
thing.save()
except Thing.DoesNotExist:
logger.warning("Processed thing vanished")
这是做这种事情的正确模式吗?我的意思是,我会在生产中运行它的几天内发现它是否有效,但如果知道是否有其他公认的模式来完成这类事情,那就太好了。
我真正想要的是能够更新对象,如果它仍然存在于数据库中。我可以处理用户编辑和来自process_thing
的编辑之间的竞争,并且我总是可以在process_thing
之前放入refresh_from_db
,以最大限度地减少用户编辑丢失的时间。但我绝对不能让对象在用户删除后重新出现。
如果您在处理芹菜任务的时候打开了一个事务,您应该避免这样的问题:
@app.task
@transaction.atomic
def my_periodic_task():
things = get_things_for_processing()
# if the delete happens anywhere between here and the .save(), we're hosed
for thing in things:
process_thing(thing) # could take a LONG time
thing.save()
有时,您想向前端报告您正在处理数据,这样您就可以将select_for_update()
添加到查询集中(很可能是在get_things_for_procrocessing中),然后在负责删除的代码中,当db报告特定记录被锁定时,您需要处理错误。
目前,"再次原子选择,然后保存"的模式似乎已经足够了:
@app.task
def my_periodic_task():
things = Thing.objects.filter(...some criteria...)
for thing in things:
process_thing(thing) # could take a LONG time
try:
with transaction.atomic():
# just see if it still exists:
unused = Thing.objects.select_for_update().get(pk=thing.pk)
# no exception means it exists. go ahead and save the
# processed version that has all of our updates.
thing.save()
except Thing.DoesNotExist:
logger.warning("Processed thing vanished")
(这与我原来的问题中的代码相同)。