加速Django Rest Framework Model Serializer N+1 Query问题

Question

我有一个序列化Order 模型的 DRF ModelSerializer 类。这个序列化器有一个字段：

num_modelA = serializers.SerializerMethodField() `

def get_num_modelA(self, o):
    r = ModelA.objects.filter(modelB__modelC__order=o).count()

    return r

ModelA 有一个ForeignKey 字段modelB，ModelB 有一个ForeignKey 字段modelC，ModelC 有一个ForeignKey 字段order。

问题显然在于，对于每个序列化的订单，它都会对数据库进行额外的查询，这会降低性能。

我已经实现了一个静态方法setup_eager_loading，如here 所述，它修复了我拥有的其他字段的 N+1 查询问题。

@staticmethod
def setup_eager_loading(queryset):
    # select_related for "to-one" relationships
    queryset = queryset.select_related('modelD','modelE')
    
    return queryset

我的想法是我也可以使用prefetch_related 来减少查询次数。但我不确定如何执行此操作，因为 Order 和 ModelA 由多个外键分隔。让我知道是否有任何其他信息有用

Answer 1

您可以使用注释：

from django.db.models import Count

# …

@staticmethod
def setup_eager_loading(queryset):
    # select_related for "to-one" relationships
    return queryset.select_related('modelD','modelE').annotate(
        num_modelA=Count('modelC__modelB__modelA')
    )

在您的Order 的序列化程序中，您可以使用num_modelA 作为IntegerField：

from rest_framework import serializers

class OrderSerializer(serializers.ModelSerializer):
    num_modelA = serializers.IntegerField()

    class Meta:
        model = Order
        fields = ['num_modelA', 'and', 'other', 'fields']