Django。线程安全更新或创建。

Question

我们知道，更新 - 是线程安全操作。 这意味着，当你这样做：

SomeModel.objects.filter(id=1).update(some_field=100) 

相反的：

sm = SomeModel.objects.get(id=1) 
sm.some_field=100 
sm.save() 

您的应用程序是relativly线程运行安全SomeModel.objects.filter(id=1).update(some_field=100)不会改写在其他模型字段中的数据。

我的问题是..如果有什么办法可以做到

SomeModel.objects.filter(id=1).update(some_field=100) 

但如果创作不存在，它的对象吗？

Answer 1

from django.db import IntegrityError 

def update_or_create(model, filter_kwargs, update_kwargs) 
    if not model.objects.filter(**filter_kwargs).update(**update_kwargs): 
     kwargs = filter_kwargs.copy() 
     kwargs.update(update_kwargs) 
     try: 
      model.objects.create(**kwargs) 
     except IntegrityError: 
      if not model.objects.filter(**filter_kwargs).update(**update_kwargs): 
       raise # re-raise IntegrityError 

我想，在问题中提供的代码不是很有说服力：谁想为模型设置id？ 让我们假设我们需要这个，我们必须同时操作：

def thread1(): 
    update_or_create(SomeModel, {'some_unique_field':1}, {'some_field': 1}) 

def thread2(): 
    update_or_create(SomeModel, {'some_unique_field':1}, {'some_field': 2}) 

随着update_or_create功能，取决于哪个线程至上，对象将被创建并没有例外更新。这将是线程安全的，但显然有什么用处：取决于SomeModek.objects.get(some__unique_field=1).some_field竞争条件值可以是1或2

Django提供˚F对象，所以我们可以提升我们的代码：

from django.db.models import F 

def thread1(): 
    update_or_create(SomeModel, 
        {'some_unique_field':1}, 
        {'some_field': F('some_field') + 1}) 

def thread2(): 
    update_or_create(SomeModel, 
        {'some_unique_field':1}, 
        {'some_field': F('some_field') + 2}) 

Answer 2

您可以使用Django的内置get_or_create，但它可以在模型本身上运行，而不是在查询集上运行。

您可以使用这样的：

me = SomeModel.objects.get_or_create(id=1) 
me.some_field = 100 
me.save() 

如果你有多个线程，您的应用程序将需要确定模型的实例是正确的。通常，我所做的是从数据库中刷新模型，进行更改，然后保存它，以便在断开连接的状态下没有很长时间。

Answer 3

这是不可能在Django做这样的upsert操作，更新。但是过滤领域的queryset的更新方法的返回号码，以便你可以这样做：

from django.db import router, connections, transaction 

class MySuperManager(models.Manager): 
    def _lock_table(self, lock='ACCESS EXCLUSIVE'): 
     cursor = connections[router.db_for_write(self.model)] 
     cursor.execute(
      'LOCK TABLE %s IN %s MODE' % (self.model._meta.db_table, lock) 
     ) 

    def create_or_update(self, id, **update_fields): 
     with transaction.commit_on_success():    
      self.lock_table() 
      if not self.get_query_set().filter(id=id).update(**update_fields): 
       self.model(id=id, **update_fields).save() 

这个例子中，如果Postgres的，你可以使用它没有SQL代码，但更新或插入操作不会是原子。如果你在表上创建一个锁，你将确保两个对象不会在另外两个线程中创建。

Answer 4

你希望django的select_for_update()方法（以及支持行级锁定的后端，如PostgreSQL）结合手动事务管理。

try: 
    with transaction.commit_on_success(): 
     SomeModel.objects.create(pk=1, some_field=100) 
except IntegrityError: #unique id already exists, so update instead 
    with transaction.commit_on_success(): 
     object = SomeModel.objects.select_for_update().get(pk=1) 
     object.some_field=100 
     object.save() 

请注意，如果其他进程删除两个查询之间的对象，您将得到一个SomeModel.DoesNotExist异常。

Django 1.7及以上版本还具有原子操作支持和内置update_or_create()方法。

Answer 5

我认为如果你对原子操作有严格的要求。你最好在数据库级别而不是Django ORM级别上设计它。

Django ORM系统专注于便利性而不是性能和安全性。有时您必须优化自动生成的SQL。

最具生产力的数据库中的“事务”可以很好地提供数据库锁定和回滚。

在mashup（混合）系统中，或者说您的系统添加了第三方零部件，如日志记录，统计信息。应用在不同的框架甚至语言中可能同时访问数据库，在这种情况下在Django中添加线程安全性是不够的。

Answer 6

SomeModel.objects.filter(id=1).update(set__some_field=100)