Please enable Javascript to view the contents

如何在 Django 中保证并发的数据一致性

 ·  ☕ 3 分钟

前面一部分主要讲乐观锁和悲观锁。锁从数据库层面,保证了并发时的数据一致性。了解锁,有助于对并发解决方案的理解。后面一部分主要讲的是 Django 中,并发场景下,保证数据一致性的解决办法。

1. 关于锁

1.1 乐观锁

乐观锁的出发点是,同一条数据很少会因为并发修改而产生冲突,适用于读多写少的场景,用以提高吞吐量。

实现方式,读取一个字段,执行处理逻辑,当需要更新数据时,再次检查该字段是否和第一次读取一致。如果一致,则更新数据,否则拒绝更新,重新读取后再提交。

1.2 悲观锁

悲观锁的出发点是,当一条数据正在被修改时,不允许其他任何关于这条数据的操作。

实现方式,读取一个字段之后,加锁,不允许其他任何读、写操作。执行处理逻辑,更新数据完毕后,释放锁。

1.3 比较

乐观锁的开销远低于悲观锁。

悲观锁可能会导致死锁。当 A 锁定了 a 资源,需要 b 资源。而 b 被 B 锁定,正在等待 a 资源。此时,导致出现死锁。但是,可以通过设置超时来处理这个问题。

悲观锁可以有效降低冲突后,重试的次数。

乐观锁可以提高响应速度。

2. Django 中的事务

Django 默认每条数据库操作都会被立即提交到数据库。

这样会导致一个问题,如果有一系列的数据库操作构成,要么全部执行,要么就全部都不执行,怎么办?

这时,就需要事务。将一系列数据库操作设置为一个事务,提交给数据库执行。

Django 提供 atomic 装饰器以开启事务。
atomic 使用一个参数来指定数据库的名字。如果不设置值,Django 就会使用系统默认的数据库。

2.1 整个 View 函数开启事务

1
2
3
4
5
6
from django.db import transaction

@transaction.atomic
def viewfunc(request):
    # This code executes inside a transaction.
    do_stuff()

2.2 部分函数 do_more_stuff() 开启事务。

1
2
3
4
5
6
7
8
9
from django.db import transaction

def viewfunc(request):
    # This code executes in autocommit mode (Django's default).
    do_stuff()

    with transaction.atomic():
        # This code executes inside a transaction.
        do_more_stuff()

2.3 不要在事务中处理异常

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
from django.db import transaction

def viewfunc(request):
    do_stuff()
    try:
        with transaction.commit_on_success():
            do_more_stuff_1() # in transaction
            try:
                do_more_stuff_2() # not in transaction
            except:
                pass
            do_more_stuff_3() # in transaction
    except:
        pass

当退出原子块时,Django 会查看它是否正常退出或者是否有异常来确定是否提交或者回滚。

如果你捕获并处理原子块中的异常,可以能会隐藏 Django 中发生问题的事实。这样可能会造成非预期的行为。

3. 利用 F 函数更新运算

通常更新数据库的操作,需要将对象读取到内存。在内存中进行修改之后,再写回数据库。

在内存中的操作,如果存在同时操作的情况,会导致运算逻辑错误。

F() 函数的作用就是直接生成 SQL 语句,不必将需要更新的对象读取到内存。避免了并发导致的数据不一致问题。

1
2
3
4
5
from django.db.models import F

reporter = Reporters.objects.get(name='OICQ')
reporter.stories_filed = F('stories_filed') + 1
reporter.save()

4. 利用 select_for_update 函数

select_for_update 使用的是悲观锁。

select for update 函数使用数据库查询语句, select ... for update 对数据库进行操作。

这是数据库层面的,解决并发取数据后再修改的问题方法。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
def mark_as_readed(self, notification_id):
    # 让s elect for update 和 update 语句发生在一个完整的事务里面
    with transaction.commit_on_success():
        # 使用select_for_update 来保证并发请求同时只有一个请求在处理,其他的请求
        # 等待锁释放
        notification = Notification.objects.select_for_update().get(pk=notification_id)
        # 没有必要重复标记一个已经读过的通知            
        if notication.has_readed:                
            return
        notification.has_readed = True
        notification.save()
        # 在这里更新我们的计数器,嗯,我感觉好极了
        self.update_unread_count(-1)

微信公众号
作者
微信公众号