Django在批量插入/更新/删除时“模拟”数据库触发行为

Question

这是一个自我解释的问题，但我们开始吧。 我正在 Django 中创建一个业务应用程序，我不想在应用程序和数据库中“传播”所有逻辑，但另一方面，我不想让数据库处理这个任务（它可能通过使用Triggers)。

所以我想“重现”数据库触发器的行为，但在 Django 的模型类中（目前使用 Django 1.4）。

经过一些研究，我发现对于单个对象，我可以覆盖“models.Model”类的“save”和“delete”方法，插入“before”和“after”钩子以便它们可以被执行在父母的保存/删除之前和之后。像这样：

     class MyModel(models.Model):

         def __before(self):
             pass

         def __after(self):
            pass

         @commit_on_success #the decorator is only to ensure that everything occurs inside the same transaction
         def save(self, *args, *kwargs):
             self.__before()
             super(MyModel,self).save(args, kwargs)
             self.__after()

大问题在于批量操作。从 QuerySet 运行“update()”/“delete()”时，Django 不会触发模型的保存/删除。 Insted，它使用 QuerySet 自己的方法。更糟糕的是，它也不会触发任何信号。

编辑： 更具体一点：视图内的模型加载是动态的，因此不可能定义“特定于模型”的方式。在这种情况下，我应该创建一个抽象类并在那里处理它。

我最后一次尝试是创建一个自定义管理器，在这个自定义管理器中，覆盖更新方法，循环查询集中的模型，并触发每个模型的“save()”（考虑上面的实现，或“信号”系统）。它可以工作，但会导致数据库“过载”（想象一个 10k 行的查询集正在更新）。

Answer 1

有一些注意事项，您可以覆盖查询集的 update 方法来触发信号，同时仍然使用 SQL UPDATE 语句：

from django.db.models.signals import pre_save, post_save

def CustomQuerySet(QuerySet):
    @commit_on_success
    def update(self, **kwargs):
        for instance in self:
            pre_save.send(sender=instance.__class__, instance=instance, raw=False, 
                          using=self.db, update_fields=kwargs.keys())
        # use self instead of self.all() if you want to reload all data 
        # from the db for the post_save signal
        result = super(CustomQuerySet, self.all()).update(**kwargs)
        for instance in self:
            post_save.send(sender=instance.__class__, instance=instance, created=False,
                           raw=False, using=self.db, update_fields=kwargs.keys())
        return result

    update.alters_data = True

我克隆当前查询集（使用self.all()），因为update方法会清除查询集对象的缓存。

有一些问题可能会或可能不会破坏您的代码。首先，它将引入竞争条件。您根据更新数据库时可能不再准确的数据在pre_save 信号的接收器中执行某些操作。

大型查询集也可能存在一些严重的性能问题。与update 方法不同，所有模型都必须加载到内存中，然后仍然需要执行信号。特别是如果信号本身必须与数据库交互，性能可能会慢得令人无法接受。与常规的 pre_save 信号不同，更改模型实例不会自动导致数据库更新，因为模型实例不用于保存新数据。

可能还有一些问题会在少数极端情况下导致问题。

无论如何，如果您可以在不出现严重问题的情况下处理这些问题，我认为这是最好的方法。它产生尽可能少的开销，同时仍将模型加载到内存中，这是正确执行各种信号所必需的。

Answer 2

首先，您可以使用内置的 pre_save、post_save,、pre_delete, 和 post_delete 信号，而不是覆盖 save 以添加 __before 和 __after 方法。 https://docs.djangoproject.com/en/1.4/topics/signals/

from django.db.models.signals import post_save

class YourModel(models.Model):
    pass

def after_save_your_model(sender, instance, **kwargs):
     pass

# register the signal
post_save.connect(after_save_your_model, sender=YourModel, dispatch_uid=__file__)

pre_delete 和 post_delete 在查询集上调用 delete() 时会被触发。

但是，对于批量更新，您必须手动调用要自己触发的函数。您也可以将其全部放入事务中。

如果您使用的是动态模型，要调用正确的触发函数，您可以检查模型的 ContentType。例如：

from django.contrib.contenttypes.models import ContentType

def view(request, app, model_name, method):
    ...
    model = get_model(app, model_name)
    content_type = ContentType.objects.get_for_model(model)
    if content_type == ContenType.objects.get_for_model(YourModel):
        after_save_your_model(model)
    elif content_type == Contentype.objects.get_for_model(AnotherModel):
        another_trigger_function(model)