Django中如何对反向通用关联使用prefetch_related或select_related？

首先明确结论：

• select_related 无法用于这个泛型关联场景，它基于SQL JOIN实现，仅支持关联目标表固定的外键、一对一关系，GenericForeignKey属于多态关联，关联的目标表不固定，无法通过单次JOIN拉取所有关联数据。
• prefetch_related 可以完美解决这个场景的N+1问题，它通过分表查询+Python层内存匹配的方式实现关联数据预取，原生支持Django的GenericRelation、GenericForeignKey字段。

第一步：修正模型定义的语法问题 #

当前模型定义有几处语法错误，先调整为Django认可的写法：

from django.contrib.contenttypes.fields import GenericForeignKey, GenericRelation
from django.contrib.contenttypes.models import ContentType
from django.db import models

# 泛型关联表A，存储所有挂载到其他模型的关联数据
class A(models.Model):
  content_type = models.ForeignKey(ContentType, on_delete=models.PROTECT, null=True, blank=True)
  object_id = models.PositiveIntegerField(null=True, blank=True)
  attached_object = GenericForeignKey('content_type', 'object_id')
  # 你需要读取的A表业务字段
  some_field_of_class_A = models.CharField(max_length=128)

# 业务表B，通过GenericRelation建立反向泛型关联
class B(models.Model):
  # 注意GenericRelation直接传入关联模型类，不需要加class关键字，这里自定义反向关联字段名
  related_a_entries = GenericRelation(A)

第二步：用prefetch_related实现预取，消除N+1 #

直接在查询B的QuerySet上调用prefetch_related指定泛型反向关联字段即可，固定只产生2-3次数据库查询（1次查所有B实例，1次查所有关联的A实例，最多加1次查询ContentType表数据，和B的实例总数无关）：

# 基础预取写法
instances_of_b = B.objects.prefetch_related("related_a_entries")

some_list = []
for instance in instances_of_b:
  # 注意GenericRelation反向返回的是关联对象列表，若业务上每个B仅对应1个A，取第一个元素即可
  related_a = instance.related_a_entries.first()
  if related_a:
    some_list.append(related_a.some_field_of_class_A)

进阶优化：仅查询需要的字段，减少内存占用 #

如果你只需要A表的特定字段，不需要完整的A模型实例，可以结合Prefetch对象自定义关联查询的字段集，进一步提升性能：

from django.db.models import Prefetch

instances_of_b = B.objects.prefetch_related(
  Prefetch(
    lookup="related_a_entries",
    # 仅查询需要的字段，注意必须保留content_type_id、object_id两个字段用于预取匹配
    queryset=A.objects.only("some_field_of_class_A", "content_type_id", "object_id"),
    # 自定义预取结果挂载到实例上的属性名，避免和默认反向关联的查询集冲突
    to_attr="prefetched_a"
  )
)

some_list = []
for instance in instances_of_b:
  if instance.prefetched_a:
    some_list.append(instance.prefetched_a[0].some_field_of_class_A)

注意事项 #

• 之前手动调用ContentType.objects.get_for_model的逻辑完全可以去掉，Django的泛型关联预取逻辑会自动处理content_type的匹配，不需要手动查询。
• 预取完成后，不要对关联字段再次调用filter()、exclude()等链式查询方法，否则会绕过预取缓存触发新的SQL查询，自定义过滤逻辑请写到Prefetch的queryset参数中。

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