乐观锁与悲观锁的简单使用

乐观锁与悲观锁

悲观锁

Views.py(视图，悲观锁，select_fo_update)

from django.shortcuts import render
from django.http import HttpResponse
from django.views.generic import View
from django.db import transcation
from 应用.models import 模型类


# 类视图（并发，悲观锁）
class MyViews(View):
    @transaction.atomic
    def post(self,request):
        # 等同于SQL(select * from 表名 where id = 1 for update)
        # for update代表锁，只有获取到锁才会执行查询，否则阻塞等待。
        obj = 模型.objects.select_for_update().get(pk=1)
        # 等待事务提交后，会自动释放锁。
        return HttpResponse('ok')

乐观锁

乐观锁其实并不是锁。通过SQL的where子句中的条件是否满足来判断是否满足更新条件来更新数据库，
通过受影响行数判断是否更新成功，如果更新失败可以再次进行尝试，如果多次尝试失败就返回更新失败的结果。
使用乐观锁时，必须设置数据库的隔离级别是Read Committed(可以读到其他线程已提交的数据)。
如果隔离级别是Repeatable Read(可重复读，读到的数据都是开启事务时刻的数据，即使其他线程提交更新数据，该线程读取的数据也是之前读到的数据)，
乐观锁如果第一次尝试失败,那么不管尝试多少次都会失败。
(Mysql数据库的默认隔离级别是Repeatable Read，需要修改成Read Committed)。

Views.py(视图，乐观锁)：

from django.shortcuts import render
from django.http import JsonResponse
from django.views.generic import View
from django.db import transaction
from 应用.models import Goods(模型类名)


class MyView(view):
    @transaction.atomic
    def post(self,request):
        '''创建订单'''
        count = 3 #订购三件商品
        
        s1 = transaction.savepoint()  #设置事务保存点
        
        #乐观锁，设置尝试的次数
        for i in range(5):
            try:
                sku = Goods.objects.get(id=1)
            except:
                #商品不存在
                transaction.savepoint_rollback(s1)
                return JsonResponse({"result":1,"error_msg":"商品不存在"})
                
            if count > sku.stock:
                transaction.savepoint_rollback(s1)
                return JsonResponse({"result":1,"error_msg":"商品库存不足"})
                
            orgin_stock = sku.stock  #原库存（数据库隔离级别必须是Read Committed,如果是Repeatable Read,则多次读取的原库存都是一样的,读不到其他线程提交更新后的数据）
            new_stock = orgin_stock - count   # 更新后的库存
            new_sales = sku.sales + count   # 更新后的销量
            
            row = Goods.objects.filter(id=1,stock=orgin_stock).update(stock=new_stock,sales=new_sales)
            if row:
                if i==4:
                    # 如果五次全部失败
                    transaction.savepoint_rollback(s1)
                    return JsonResponse({"result":3,"error_msg":"下单失败"})
                continue
            #当规定次数内更新成功，跳出循环    
            break
            
        transaction.savepoint_commit(s1)
        return JsonResponse({"result":4,"message":"创建成功"})
                   

在并发比较少的时候建议使用乐观锁，减少加锁、释放锁的开销。在并发比较高的时候，建议使用悲观锁。
如果乐观锁尝试代价比较大，也建议使用悲观锁。