Django中Flight模型字符串表示引发重复查询的N+1问题求助

问题背景 #

开发机场API服务时，Flight模型出现N+1查询问题。通过debug_toolbar检测到重复查询，使用select_related后查询数从25降至17，但仍存在重复查询。经排查，问题根源在于Route模型的__str__方法调用了Airport的closest_big_city字段，且需要保留该字符串表示形式。

解决方案 #

方案1：通过数据库注解直接生成Route显示字符串 #

在查询Flight时，直接在数据库层面拼接出Route的显示字符串，避免在Python代码中访问关联的Airport对象，彻底消除N+1问题。

修改views.py中的get_queryset方法：

def get_queryset(self):
    queryset = self.queryset
    if self.action == "list":
        queryset = (
            queryset
            .select_related("airplane", "route__source", "route__destination")
            .annotate(
                # 直接注解出Route的显示字符串
                route_display=F("route__source__closest_big_city") + " - " + F("route__destination__closest_big_city"),
                tickets_available=(
                    F("airplane__rows") * F("airplane__seats_in_row")) - Count("tickets")
            )
        )
        return queryset

    return queryset

修改serializers.py中的FlightListSerializer，使用注解的字段替代StringRelatedField：

class FlightListSerializer(FlightSerializer):
    # 替换StringRelatedField为注解的字段
    route = serializers.CharField(source="route_display", read_only=True)
    airplane_name = serializers.CharField(source="airplane.name", read_only=True)
    tickets_available = serializers.IntegerField(read_only=True)

    class Meta:
        model = Flight
        fields = (
            "id",
            "route",
            "airplane_name",
            "departure_time",
            "arrival_time",
            "tickets_available",
        )

方案2：缓存Route的字符串表示 #

利用Django的cached_property缓存Route的__str__结果，避免重复计算和重复查询关联对象。

修改Route模型：

from django.utils.functional import cached_property

class Route(models.Model):
    source = models.ForeignKey(
        Airport,
        on_delete=models.CASCADE,
        related_name="sources"
    )
    destination = models.ForeignKey(
        Airport,
        on_delete=models.CASCADE,
        related_name="destinations"
    )
    distance = models.IntegerField()

    class Meta:
        unique_together = ["source", "destination"]

    @cached_property
    def _display_str(self):
        return f"{self.source.closest_big_city} - {self.destination.closest_big_city}"

    def __str__(self):
        return self._display_str

此方法结合已有的select_related配置，能确保每个Route对象的字符串只计算一次，避免重复触发查询。

方案3：确认select_related覆盖所有关联 #

当前代码中已配置预取关联对象，但可明确验证配置完整性：

queryset = queryset.select_related("airplane", "route__source", "route__destination")

该配置会一次性取出Flight关联的Airplane、Route，以及Route关联的source和destination Airport对象，确保调用Route的__str__时无需额外查询。若仍有问题，需检查是否存在其他未预取的关联场景（如其他视图或admin页面）。

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