Django ORM在异步Telegram Bot中如何规避竞态条件？

我几乎在所有项目中都使用Django ORM，当前正基于Pyrogram开发支持async/await的异步Telegram Bot，希望继续在其中使用Django ORM。我的需求场景为：点击绑定数据库对象的按钮时，需在满足条件的情况下，减少Object1对象的field1字段值，同时增加Object2对象的field2字段值，简化代码如下：

async def on_button_click():
   obj1 = Object1.objects.get(id=X)
   if obj1.filed1 > 0:
      obj2 = Object2.objects.get(id=X2)
      obj1.field1 -= N
      obj2.field2 += N
      obj1.save()
      obj2.save()

但我担心并发点击会引发竞态条件，导致执行结果不符合预期。想请教此类场景的最佳实践：select_for_update方法能否满足需求？若Django ORM不适用，应选择哪些ORM或数据库库？若适用，具体该如何处理？

1. Django ORM的select_for_update完全适配你的需求 #

你的场景核心是要保证读取-修改-保存流程的原子性，select_for_update正是解决这类竞态问题的标准方案——它会在查询时给目标数据库行加排他锁，直到事务结束才释放，让并发请求必须排队执行，彻底避免数据不一致。

异步场景下的具体实现

由于你使用的是异步Bot，必须配合Django的异步ORM接口（带async_前缀的方法），同时手动包裹原子事务：

from django.db import transaction

async def on_button_click():
    async with transaction.atomic():
        # 加排他锁获取obj1，阻塞其他并发修改请求
        obj1 = await Object1.objects.select_for_update().aget(id=X)
        if obj1.field1 > 0:
            # 给obj2也加锁，防止其他操作同时修改
            obj2 = await Object2.objects.select_for_update().aget(id=X2)
            obj1.field1 -= N
            obj2.field2 += N
            await obj1.asave()
            await obj2.asave()

• transaction.atomic()：确保整个操作要么全部成功，要么全部回滚，避免部分更新的情况
• select_for_update()：对查询到的行加排他锁，阻止其他事务修改这些行
• 所有ORM操作使用异步版本（aget、asave）：匹配你的async/await开发环境

2. 更高效的优化方案：数据库原子更新 #

如果不想依赖锁机制，还可以用Django的F()表达式直接在数据库层面完成更新操作，这是性能更高的原子操作，从根源消除客户端的读取-修改间隙：

from django.db.models import F
from django.db import transaction

async def on_button_click():
    async with transaction.atomic():
        # 仅当obj1的field1>0时才执行更新，返回受影响的行数
        update_count = await Object1.objects.filter(id=X, field1__gt=0).aupdate(field1=F('field1') - N)
        if update_count > 0:
            # obj1更新成功后，再原子更新obj2
            await Object2.objects.filter(id=X2).aupdate(field2=F('field2') + N)

这种方式的优势：

• 完全在数据库端执行更新，没有客户端的读取-修改窗口期，彻底避免竞态
• 不需要加锁，并发性能更好，适合高频率点击的场景
• 通过update_count判断更新是否生效，替代原有的内存中条件判断

3. 极端场景下的ORM替代方案 #

如果因为特殊需求必须更换ORM，推荐以下异步友好的选项：

• SQLAlchemy 1.4+：支持异步操作，ORM功能全面，事务和锁机制成熟
• Tortoise-ORM：专为Python异步设计，语法接近Django，学习成本低
• Peewee（异步扩展）：轻量级ORM，支持异步操作，适合小型项目

但对你的场景来说，Django ORM完全能满足需求，没有必要更换。

内容的提问来源于stack exchange，提问作者ltad