Django递归模型场景下如何避免多次数据库查询？

模型定义 #

class Topic(models.Model):
    ...

class Article(models.Model):
    ...

class ArticleInTopic(models.Model):
    topic = models.ForeignKey(Topic, on_delete=models.PROTECT)
    article = models.ForeignKey(Article, on_delete=models.PROTECT)
    depends_on = models.ForeignKey(ArticleInTopic, on_delete=models.PROTECT)

    class Meta:
        unique_together = ('topic', 'article', 'depends_on')

业务场景 #

该模型用于学习场景：存在多个学习主题，每个主题包含多篇文章；用户需按文章间的依赖关系阅读主题下的文章，且依赖的文章必属于同一主题。此结构为无环图，需对其进行拓扑排序以告知用户阅读顺序。

问题解答 #

1. 访问depends_on时是否会重复查询数据库？ #

当你已经通过一次查询获取了同一主题下的所有ArticleInTopic对象后，访问某个对象的depends_on字段时，Django会复用内存中已加载的对象，不会发起新的数据库查询。

原因是Django的ORM会维护实例缓存，对于已经加载到内存中的模型实例，只要其主键存在于缓存中，再次通过外键访问时就会直接从缓存中读取，不会触发新的SQL查询。

2. 如何彻底避免潜在的额外查询？ #

虽然Django会自动缓存实例，但可以通过以下方式进一步优化，确保完全避免额外查询：

• 预取所有关联对象：使用prefetch_related批量预取所有depends_on关联的自引用对象，一次查询加载全部所需数据。示例代码：

  topic = Topic.objects.get(pk=topic_id)
  # 获取当前主题下的所有ArticleInTopic，并预取关联的depends_on对象
  article_in_topics = ArticleInTopic.objects.filter(topic=topic).prefetch_related('depends_on')
  

• 手动构建内存映射：将查询到的ArticleInTopic对象按主键存入字典，后续直接通过字典查找关联对象，完全绕过ORM的外键访问逻辑。示例代码：

  topic = Topic.objects.get(pk=topic_id)
  article_in_topics = ArticleInTopic.objects.filter(topic=topic)
  # 构建主键到对象的映射字典
  ait_map = {ait.pk: ait for ait in article_in_topics}
  
  # 拓扑排序时直接通过映射获取依赖对象
  for ait in article_in_topics:
      # 用depends_on_id直接取主键，避免触发属性访问的ORM逻辑
      depends_on_ait = ait_map.get(ait.depends_on_id)
      # 执行拓扑排序相关逻辑
  

  这种方式效率最高，完全在内存中操作，不会产生任何额外数据库查询。

注意：如果depends_on允许为null（当前模型未设置null=True），处理时需判断空值情况，避免报错。

内容的提问来源于stack exchange，提问作者Daniel Richter