在序列化器字段中使用属性时如何优化 DRF 编号查询？

Question

我正在开发一个 DRF API，但我并不完全熟悉 django 属性。

数据库关系是经典的。公司有不同的职位可供候选人申请。每个职位都有多个匹配项，匹配项是职位和候选人之间的连接表。匹配具有不同的状态，代表申请流程的不同阶段。

所以这里是交易： 我正在使用 drf 视图集从 api 获取数据。此视图集使用序列化程序来获取特定字段，特别是作业的每个状态的匹配数。序列化程序的简化版本如下所示。

class Team2AmBackofficeSerializer(Normal2JobSerializer):

    class Meta:
        model = Job
        fields = (    
            'pk',
            'name',
            'company',
            'company_name',
            'job__nb_matches_proposition',
            'job__nb_matches_preselection',
            'job__nb_matches_valides',
            'job__nb_matches_pitches',
            'job__nb_matches_entretiens',
            'job__nb_matches_offre',
        )

job__xxx 字段使用装饰器@property，例如：

@property
def job__nb_matches_offre(self):
    return self.matches.filter(current_status__step_name='Offre').count()

问题是每次我将这些属性之一添加到我的序列化程序的字段时，数据库查询的数量都会显着增加。这当然是因为每个属性都会多次调用数据库。所以这是我的问题：

有没有办法优化对数据库的查询次数，或者通过更改序列化程序中的某些内容，或者通过以不同方式获取特定状态的匹配次数？ p>

我看过select_related 和prefetch_related。这让我在获取有关公司的信息时减少了查询次数，但实际上并没有减少匹配次数。

非常感谢任何帮助:)

Answer 1

您想要的是annotate 您的查询集使用这些值，这将导致数据库在一个查询中完成所有计数。结果明显快于您当前的解决方案。

例子：

from django.db.models import Count, Q

Job.objects.annotate(
    'nb_matches_offre'=Count(
        'pk',
        filter=Q(current_status__step_name='Offre')
    ),
    'nb_matches_entretiens'=Count(...)
).all()

生成的查询集将包含具有属性 job_obj.nb_matches_offre 和 job_obj.nb_matches_entretiens 的 Job 对象以及计数。

另见https://docs.djangoproject.com/en/3.0/topics/db/aggregation/