如何在Django DRF中实现嵌套关联表预取并优化查询性能

方案可行性说明 #

你设想的链式调用序列化器预加载逻辑是完全可行的，且是非常推荐的DRF嵌套关联性能优化方案，符合DRY原则，也便于后续维护。
首先纠正一个基础使用误区：prefetch_related/select_related 只需指定到关联对象层级即可，不需要写字段后缀，例如写contactrole__contact就足够拉取关联Contact对象的所有字段，不需要写contactrole__contact__name这类到字段的路径。

具体实现方案 #

你可以封装一个通用的预加载Mixin，让所有序列化器继承，自动递归处理嵌套关联的预加载逻辑，无需手动拼接双下划线路径：

from rest_framework import serializers
from django.db.models import Prefetch

class EagerLoadingMixin:
    @classmethod
    def setup_eager_loading(cls, queryset, prefix=""):
        # 处理当前序列化器的直连预加载字段
        if hasattr(cls, "select_related_fields"):
            prefetch_paths = [f"{prefix}{field}" for field in cls.select_related_fields]
            queryset = queryset.select_related(*prefetch_paths)
        if hasattr(cls, "prefetch_related_fields"):
            prefetch_items = []
            for field in cls.prefetch_related_fields:
                # 支持直接传字符串或者Prefetch对象
                if isinstance(field, str):
                    prefetch_items.append(f"{prefix}{field}")
                else:
                    # 处理自定义Prefetch对象，拼接路径前缀
                    prefixed_prefetch = Prefetch(
                        f"{prefix}{field.prefetch_through}",
                        queryset=field.queryset,
                        to_attr=field.to_attr
                    )
                    prefetch_items.append(prefixed_prefetch)
            queryset = queryset.prefetch_related(*prefetch_items)
        
        # 递归处理嵌套序列化器的预加载逻辑
        for field in cls._declared_fields.values():
            # 处理多值关联（many=True）
            child_serializer = field.child if hasattr(field, "child") else field
            if hasattr(child_serializer, "setup_eager_loading"):
                field_name = field.source or field.field_name
                new_prefix = f"{prefix}{field_name}__"
                queryset = child_serializer.setup_eager_loading(queryset, new_prefix)
        return queryset

序列化器改造

# 最内层Contact序列化器
class ContactReadSerializer(EagerLoadingMixin, serializers.ModelSerializer):
    # 无下层关联，无需配置预加载字段
    class Meta:
        model = Contact
        # 仅指定需要返回的字段，不会多余返回数据
        fields = ("name", "surname_1", "surname_2")

class ContactSubTypeReadSerializer(EagerLoadingMixin, serializers.ModelSerializer):
    class Meta:
        model = ContactSubType
        fields = "__all__"

# 中间层关联序列化器
class LegalFileContactRoleSerializer(EagerLoadingMixin, serializers.ModelSerializer):
    contact = ContactReadSerializer()
    subtype = ContactSubTypeReadSerializer()
    # ForeignKey用select_related，性能优于prefetch_related
    select_related_fields = ["contact", "subtype"]
    class Meta:
        model = LegalFileContactRole
        fields = "__all__"

# 上层主序列化器
class LegalFileReadSerializer(EagerLoadingMixin, serializers.ModelSerializer):
    contactrole = LegalFileContactRoleSerializer(many=True)
    # ManyToMany用prefetch_related
    prefetch_related_fields = ["contactrole"]
    class Meta:
        model = LegalFile
        fields = '__all__'
        read_only_fields = ('__all__',)

ViewSet逻辑无需修改

你原有ViewSet的get_queryset逻辑可以直接复用，不需要调整。

常见问题解答 #

1. 会不会查询多余信息？
   不会，序列化器Meta.fields会自动过滤返回给前端的字段。如果需要进一步减少数据库查询的字段量，可以配合only()自定义Prefetch对象，例如：

prefetch_related_fields = [
    Prefetch("contact", queryset=Contact.objects.only("name", "surname_1", "surname_2"))
]

这样连数据库层面都只会查询需要的字段，不会有多余IO开销。
2. 性能如何？
这个方案是DRF生态的通用优化方案，生产环境被大量项目验证过，完全可以避免N+1查询问题，性能远优于手动拼接双下划线路径的实现，也更便于维护。

内容的提问来源于stack exchange，提问作者Ira Ramírez