如何通过Django ORM按指定主键顺序高效批量查询数据？

一、问题根源 #

使用Product.objects.filter(pk__in=target_pks)查询时，数据库会按照自身主键索引或默认排序规则返回结果，不会保留传入的target_pks列表顺序。

二、PostgreSQL下的数据库层面排序方案 #

1. 用extra()实现（适配旧版Django） #

直接借助PostgreSQL的array_position函数，在数据库层面计算并按主键在目标列表中的位置排序：

target_pks = [3, 1, 4, 2]
products = Product.objects.filter(pk__in=target_pks).extra(
    select={'position': "array_position('{}'::integer[], id)".format(target_pks)},
    order_by=['position']
)

原理是将主键列表转为PostgreSQL整数数组，通过array_position获取每个主键在数组中的位置，再按该位置字段排序，结果顺序将与target_pks完全一致。

2. 用Func表达式实现（推荐，Django 2.0+） #

由于extra()在Django 3.2+被标记为过时，更推荐用ORM原生的Func封装array_position：

from django.db.models import Func, Value, IntegerField

class ArrayPosition(Func):
    function = 'array_position'
    output_field = IntegerField()

target_pks = [3, 1, 4, 2]
products = Product.objects.filter(pk__in=target_pks).annotate(
    position=ArrayPosition(Value(target_pks, output_field=IntegerField()), 'id')
).order_by('position')

这种方式完全遵循Django ORM规范，避免直接拼接SQL字符串，更安全易维护。

三、与字典映射方案的性能对比 #

常见的字典映射排序逻辑为：

products = Product.objects.filter(pk__in=target_pks)
product_dict = {p.pk: p for p in products}
sorted_products = [product_dict[pk] for pk in target_pks if pk in product_dict]

数据库层面排序的优势：

• 降低内存占用：无需在Python端存储全量数据的字典，尤其适合大数据量场景。
• 更高执行效率：数据库对排序操作的优化能力强于Python端的字典遍历，主键列表越长，性能差距越明显。

四、注意事项 #

• 该方案仅适用于PostgreSQL，array_position是其特有函数；若使用MySQL，可替换为FIELD()函数，例如extra(order_by=['FIELD(id, {})'.format(','.join(map(str, target_pks)))])。
• 确保target_pks不为空，否则查询会返回空结果。

内容的提问来源于stack exchange，提问作者Tonis F. Piip