Django中如何通过QuerySet实现拖拽排序的批量更新操作

不需要查询全量数据再通过bulk_update提交，用Django ORM自带的F()表达式做数据库层面的原子偏移更新，性能更高，也不会出现更新过程中position值冲突的问题。
拖拽排序的更新只需要分两种场景处理区间偏移：

• 向后拖拽（旧位置 < 新位置）：所有位置在(旧位置, 新位置]区间的记录，position统一减1，最后把被拖拽记录的position设为新位置
• 向前拖拽（旧位置 > 新位置）：所有位置在[新位置, 旧位置)区间的记录，position统一加1，最后把被拖拽记录的position设为新位置

完整实现代码 #

所有更新逻辑要包裹在数据库事务中，避免中途报错导致排序错乱：

from django.db import transaction
from django.db.models import F
from rest_framework.response import Response


def put(self, request):
    total_count = Device.objects.count()
    drag_id = int(request.query_params.get('id'))
    old_pos = int(request.query_params.get('old'))
    new_pos = int(request.query_params.get('new'))

    # 参数合法性校验
    if new_pos == old_pos:
        return Response("新旧位置一致，无需更新", status=400)
    if new_pos > total_count or new_pos < 1 or old_pos < 1:
        return Response("传入的位置参数非法", status=400)
    
    with transaction.atomic():
        if old_pos < new_pos:
            # 向后拖拽场景：中间区间的位置统一减1
            Device.objects.filter(
                position__gt=old_pos,
                position__lte=new_pos
            ).update(position=F('position') - 1)
        else:
            # 向前拖拽场景：中间区间的位置统一加1
            Device.objects.filter(
                position__gte=new_pos,
                position__lt=old_pos
            ).update(position=F('position') + 1)
        # 更新被拖拽记录的最终位置
        Device.objects.filter(id=drag_id).update(position=new_pos)
    
    return Response("排序更新完成")

注意事项 #

• 新增Device记录时，给position字段默认赋值为当前总记录数+1，保证所有记录的position是从1开始连续不重复的正整数，否则区间更新会出现错位问题
• 不要用「查询全量数据→逐个修改position属性→调用bulk_update提交」的方案，该方案需要额外把受影响数据加载到Python内存，数据量较大时性能远低于F表达式直接更新，且并发场景下容易出现值覆盖问题
• 事务包裹是必要的，如果区间偏移更新成功但被拖拽记录更新失败，事务会自动回滚，不会出现排序混乱

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