github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

fix(asset_user): 修改创建AuthBook对象锁机制，使用select_for_update替换redis_lock3

pull/4318/head
Bai 2020-07-14 19:02:23 +08:00 committed by 老广
parent 9550ea62fb
commit ddcd4ebbfc
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apps/assets/models/authbook.py
View File

@ -59,10 +59,9 @@ class AuthBook(BaseUser):
username = kwargs['username'] username = kwargs['username']
asset = kwargs['asset'] asset = kwargs['asset']
with transaction.atomic(): with transaction.atomic():
# 使用select_for_update限制并发创建相同的username、asset数据 # 使用select_for_update限制并发创建相同的username、asset条目
cls.objects.select_for_update().filter( instances = cls.objects.select_for_update().filter(username=username, asset=asset)
username=username, asset=asset, is_latest=True instances.filter(is_latest=True).update(is_latest=False)
).update(is_latest=False)
max_version = cls.get_max_version(username, asset) max_version = cls.get_max_version(username, asset)
kwargs.update({ kwargs.update({
'version': max_version + 1, 'version': max_version + 1,