如何对GenericForeignKey关系使用prefetch_related实现预取优化

首先明确问题核心：list_items__content_object__parent 预取规则无法直接生效，是因为content_object是多态关联，只有ChildItem模型存在parent字段，ParentItem没有对应字段，Django无法自动适配多态关联下的差异化预取逻辑。

最优方案：视图层完整预取（推荐） #

将所有预取逻辑放在视图层，避免序列化器耦合查询逻辑，Django 3.2及以上版本支持为Prefetch对象指定content_type参数，专门处理多态关联的差异化预取：

# views.py
from django.contrib.contenttypes.models import ContentType
from django.db.models import Prefetch
from rest_framework import viewsets
from .models import List, ListItem, ChildItem
from .serializers import ListSerializer

class ListViewSet(viewsets.ModelViewSet):
    serializer_class = ListSerializer
    
    def get_queryset(self):
        # 预取ListItem关联的ContentType，以及对应的多态内容对象
        list_item_qs = ListItem.objects.select_related("content_type").prefetch_related("content_object")
        # 单独为ChildItem类型的内容对象预取parent关联
        child_content_type = ContentType.objects.get_for_model(ChildItem)
        child_with_parent_qs = ChildItem.objects.select_related("parent")
        
        return List.objects.prefetch_related(
            Prefetch("list_items", queryset=list_item_qs),
            Prefetch(
                "list_items__content_object",
                queryset=child_with_parent_qs,
                content_type=child_content_type
            )
        )

这个方案总共只会产生4次查询：查询所有List、查询所有关联的ListItem和对应的ContentType、查询关联的所有ParentItem、查询关联的所有ChildItem和对应的ParentItem，完全避免N+1问题。

低版本Django兼容方案 #

如果你的Django版本低于3.2，不支持content_type参数，可以在序列化器层做批量预取优化，比你当前的写法性能更好：

# serializers.py
from .models import ChildItem

class ListSerializer(serializers.ModelSerializer):
    items = serializers.SerializerMethodField()
    
    class Meta:
        model = List
        fields = ["title", "items"]

    def get_items(self, obj):
        list_items = list(obj.list_items.all().select_related("content_type"))
        # 批量收集所有ChildItem的ID，一次性预取关联的parent
        child_item_ids = [
            item.object_id for item in list_items 
            if item.content_type.model == "childitem"
        ]
        # 一次性查询所有需要的ChildItem和parent，存入字典做映射
        child_item_map = {
            child.id: child for child in 
            ChildItem.objects.filter(id__in=child_item_ids).select_related("parent")
        }
        # 替换content_object，避免后续重复查询
        for item in list_items:
            if item.content_type.model == "childitem":
                item.content_object = child_item_map[item.object_id]
        return ListItemSerializer(list_items, many=True).data

对应视图层只需要保留基础预取即可：

class ListViewSet(viewsets.ModelViewSet):
    queryset = List.objects.prefetch_related("list_items")
    serializer_class = ListSerializer

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