Django Rest Framework按过滤后的预取关联对象计数排序问题

你的问题本质是Django默认的prefetch_related会预取所有关联对象，普通Count注解统计的是全局关联数据而非过滤后的子集，我们通过带过滤条件的Count注解+绑定同规则的Prefetch预取，即可同时满足计数准确、支持排序过滤、预取数据匹配的需求。

第一步：在get_queryset中实现过滤、计数、预取逻辑 #

你可以直接在查询集构造阶段完成所有逻辑，完全兼容原生的排序、分页规则：

from django.db.models import Count, Q, Prefetch
from rest_framework import generics

from .models import Provider, Provision, ProvisionLinkRegion
from .serializers import ProviderSerializer

class ProviderSearchView(generics.ListAPIView):
    serializer_class = ProviderSerializer
    # 直接把统计字段加入排序字段列表，即可支持前端传参排序
    ordering_fields = ['number_services', 'name', 'create_time']
    ordering = ['-number_services']

    def get_queryset(self):
        # 1. 提取并校验查询参数
        target_region_id = self.request.query_params.get('region_id')
        target_procedure_id = self.request.query_params.get('procedure_id')
        if not target_region_id or not target_procedure_id:
            return Provider.objects.none()

        # 2. 定义统一的匹配过滤规则，计数和预取共用避免逻辑不一致
        match_rule = Q(
            procedure_id=target_procedure_id,
            provisionlinkregion__region_id=target_region_id
        )

        # 3. 构造查询集
        return Provider.objects.annotate(
            # 仅统计匹配规则的Provision数量，distinct避免多表关联导致的计数重复
            number_services=Count(
                'provision',
                filter=match_rule,
                distinct=True
            )
        ).filter(
            # 直接过滤掉无匹配服务的Provider
            number_services__gt=0
        ).prefetch_related(
            # 预取同规则下的关联数据，和计数结果完全匹配，无N+1问题
            Prefetch(
                'provision',
                queryset=Provision.objects.filter(match_rule).prefetch_related(
                    Prefetch(
                        'provisionlinkregion_set',
                        queryset=ProvisionLinkRegion.objects.filter(region_id=target_region_id).select_related('region')
                    )
                ),
                # 自定义属性名存储匹配结果，不覆盖原关联字段
                to_attr='matched_provisions'
            )
        )

注：Count的filter参数需要Django 1.11及以上版本支持，是原生ORM提供的能力，不需要额外安装依赖。

第二步：序列化器适配 #

不需要使用SerializerMethodField，直接读取注解的统计字段和预取的关联数据即可：

from rest_framework import serializers
from .models import Provider, Provision, ProvisionLinkRegion, Region

class RegionSerializer(serializers.ModelSerializer):
    class Meta:
        model = Region
        fields = ['id', 'name']

class ProvisionLinkRegionSerializer(serializers.ModelSerializer):
    region = RegionSerializer(read_only=True)
    class Meta:
        model = ProvisionLinkRegion
        fields = ['location', 'region']

class MatchedProvisionSerializer(serializers.ModelSerializer):
    service_locations = ProvisionLinkRegionSerializer(
        source='provisionlinkregion_set',
        many=True,
        read_only=True
    )
    class Meta:
        model = Provision
        fields = ['id', 'name', 'service_locations']

class ProviderSerializer(serializers.ModelSerializer):
    # 直接读取ORM注解的统计字段，支持排序过滤
    number_services = serializers.IntegerField(read_only=True)
    # 读取预取的匹配服务列表
    matched_provisions = MatchedProvisionSerializer(many=True, read_only=True)
    class Meta:
        model = Provider
        fields = ['id', 'name', 'number_services', 'matched_provisions']

方案特性 #

• 所有逻辑在数据库层完成，性能远高于SerializerMethodField实现，和原生分页、排序逻辑完全兼容
• 计数规则和预取规则完全一致，不会出现统计数量和返回服务列表不匹配的问题
• 自动过滤无匹配服务的Provider，不需要在序列化层做二次处理
• 仅触发一次带关联Join的数据库查询，无N+1性能问题

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