Django中如何从User模型批量查询关联的Reaction数据

针对你的场景，有两种高效的批量查询方式，完全避免逐个用户查询的N+1问题：

1. 从用户集合出发，预取关联的Reaction数据 #

先获取目标用户集合，再通过prefetch_related一次性预取所有关联的Reaction记录，后续遍历用户时直接访问关联属性即可：

示例代码 #

# 第一步：获取目标用户集合（替换成你的筛选条件，比如id范围、状态等）
target_users = User.objects.filter(id__range=(1, 1000))

# 预取用户作为发起者的Reaction（对应related_name='reaction_from_user'）
target_users = target_users.prefetch_related('reaction_from_user')

# 如需同时预取用户作为接收者的Reaction，追加即可
target_users = target_users.prefetch_related('reaction_to_user')

# 遍历用户，直接访问关联数据（无额外数据库查询）
for user in target_users:
    # 获取该用户发起的所有Reaction
    initiated_reactions = user.reaction_from_user.all()
    # 获取该用户收到的所有Reaction
    received_reactions = user.reaction_to_user.all()
    # 处理业务逻辑...

进阶：过滤预取的Reaction #

如果只需要特定条件的Reaction（比如最近30天的），可以用Prefetch对象定制预取查询：

from django.db.models import Prefetch
import datetime

# 定义筛选条件的Reaction查询集
recent_reactions = Reaction.objects.filter(created_at__gte=datetime.date.today()-datetime.timedelta(days=30))

# 预取并将结果存入自定义属性
target_users = target_users.prefetch_related(
    Prefetch('reaction_from_user', queryset=recent_reactions, to_attr='recent_initiated_reactions')
)

# 访问过滤后的结果
for user in target_users:
    print(user.recent_initiated_reactions)

2. 从Reaction模型出发，直接筛选关联目标用户的记录 #

如果需要一次性获取所有和目标用户相关的Reaction（无论作为发起者还是接收者），可以直接从Reaction模型过滤：

示例代码 #

from django.db.models import Q
from collections import defaultdict

# 获取目标用户ID列表
target_user_ids = User.objects.filter(id__range=(1, 1000)).values_list('id', flat=True)

# 筛选所有关联目标用户的Reaction
related_reactions = Reaction.objects.filter(
    Q(from_user_id__in=target_user_ids) | Q(to_user_id__in=target_user_ids)
).select_related('from_user', 'to_user')  # 如需关联用户信息，用select_related

# （可选）按用户ID分组整理数据
user_reaction_map = defaultdict(lambda: {'initiated': [], 'received': []})
for reaction in related_reactions:
    user_reaction_map[reaction.from_user.id]['initiated'].append(reaction)
    user_reaction_map[reaction.to_user.id]['received'].append(reaction)

# 通过用户ID快速获取对应Reaction
user_reactions = user_reaction_map[123]  # 获取ID为123的用户的发起/收到的Reaction

这两种方式都只需要1-2次数据库查询，完全适配超1000条用户的批量场景，性能远优于逐个查询。

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