Question

假设django模型有两个字段： - 创建 - 修改每个字段都是整数，唯一且价值不断增加。第一次保存时，对象应具有相同的创建和修改值，并且修改后的字段应在每次保存时增加到大于其自身的下一个自由值。

我为此目的创建了以下字段：

class CreatedVerisonField(models.BigIntegerField):
    update_on_each_save = False

    def __init__(self, *args, **kwargs):
        kwargs.setdefault('editable', False)
        kwargs.setdefault('blank', True)
        kwargs.setdefault('unique', True)
        models.BigIntegerField.__init__(self, *args, **kwargs)

    def pre_save(self, model, add):
        if add or self.update_on_each_save:
            value = self.get_next_value(model)
            setattr(model, self.attname, value)
            return value
        else:
            return super(CreatedVerisonField, self).pre_save(model, add)

    def get_next_value(self, model):
        objs = model.__class__.objects.all()
        fields = self._get_fields(model)
        if objs:
            # new version is max of all version fields + 1
            value = max(objs.aggregate(*map(lambda x: Max(x), fields)).values()) + 1
        else:
            value = 1
        return value

    def _get_fields(self, model):
        fields = []
        for f in model._meta.fields:
            if isinstance(f, CreatedVerisonField):
                fields.append(f.db_column if f.db_column else f.name)
        return fields

class ModifiedVersionField(CreatedVerisonField):
    def __init__(self, *args, **kwargs):
        self.update_on_each_save = True
        CreatedVerisonField.__init__(self, args, kwargs)

使用这些字段的简单测试类：

class TestModel(models.Model):
    created = CreatedVerisonField()
    modified = ModifiedVersionField()

当从多个线程创建并保存“TestModel”对象时，由于违反了唯一约束，我得到了IntegretyError。在添加pre_save代码时，很明显这个种族存在。我该如何解决这个问题？

Answer 1

进行线程安全数据库访问的唯一方法是在访问之前锁定表。在网上粗略搜索了这个片段：http://djangosnippets.org/snippets/2039/。也许你可以做点什么。关于交易的Django文档部分对您也很有用：https://docs.djangoproject.com/en/dev/topics/db/transactions/

在自定义Django字段中避免IntegrityError

1 个答案: