Django 3.2中如何基于多列元组列表利用索引查询Member对象

需求背景 #

现有Member模型，通过UniqueConstraint建立了user和group的组合唯一索引：

from django.db import models

class Member(models.Model):
    user = models.ForeignKey("User", on_delete=models.CASCADE)
    group = models.ForeignKey("Group", on_delete=models.CASCADE)

    class Meta:
        constraints = [
            models.UniqueConstraint(
                fields=["user", "group"], name="unique_user_group_tuple"
            )
        ]

需要基于给定的元组列表member_tuples = [(1, 1), (2, 1), (2, 4)]，实现两个目标：

1. 用PostgreSQL风格的元组IN语句过滤数据（WHERE (user_id, group_id) IN ((1,1), (2,1), (2,4))），同时命中组合唯一索引
2. 为每条记录注解member_tuple字段，对应SQL中的(user_id, group_id) as member_tuple

解决方案 #

1. ORM原生实现（推荐） #

使用Django的Func和F表达式，构造PostgreSQL的ROW函数来组合字段成元组，同时完成注解与过滤：

from django.db.models import Func, F

member_tuples = [(1, 1), (2, 1), (2, 4)]

# 注解member_tuple并执行元组过滤
existing_members = Member.objects.annotate(
    # 用ROW函数将user_id和group_id组合为元组
    member_tuple=Func(F('user_id'), F('group_id'), function='ROW')
).filter(
    # 直接传入元组列表作为过滤条件
    member_tuple__in=member_tuples
)

生成的SQL与预期完全一致：

SELECT 
    "members_member"."id",
    "members_member"."user_id",
    "members_member"."group_id",
    ROW("members_member"."user_id", "members_member"."group_id") AS "member_tuple"
FROM "members_member"
WHERE ROW("members_member"."user_id", "members_member"."group_id") IN ((1, 1), (2, 1), (2, 4))

该查询会正常命中unique_user_group_tuple组合索引，避免全表扫描。

2. 原生SQL实现 #

如果需要更直接的原生语法，可使用RawSQL，注意通过参数化避免SQL注入：

from django.db.models import RawSQL

member_tuples = [(1, 1), (2, 1), (2, 4)]

existing_members = Member.objects.raw(
    "SELECT *, (user_id, group_id) AS member_tuple FROM members_member WHERE (user_id, group_id) IN %s",
    [tuple(member_tuples)]
)

说明 #

• 原Q对象拼接OR条件的方式虽能命中索引，但SQL结构与预期不同；上述两种方法更贴近PostgreSQL的元组查询语法。
• ORM原生写法更利于后续维护和兼容Django的其他查询扩展。

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