Django中带GenericForeignKey的模型如何预取多类型关联数据

实现方法 #

你可以用Django原生的Prefetch对象组合实现多ContentType的GenericForeignKey预取，不需要额外引入第三方依赖，具体实现步骤如下：

步骤1：构造预取逻辑 #

先获取目标模型对应的ContentType实例，再通过Prefetch分别指定不同关联模型的预取规则，示例代码如下：

from django.contrib.contenttypes.models import ContentType
from django.db.models import Prefetch

# 获取两种内容块对应的ContentType
header_ct = ContentType.objects.get_for_model(Header)
paragraph_ct = ContentType.objects.get_for_model(Paragraph)

# 从Document侧发起查询，一次性预取所有关联的Block、以及Block对应的Header/Paragraph
document = Document.objects.prefetch_related(
    Prefetch(
        'block_set',
        queryset=Block.objects.prefetch_related(
            Prefetch('content_block', queryset=Header.objects.all(), to_attr='_header_content'),
            Prefetch('content_block', queryset=Paragraph.objects.all(), to_attr='_paragraph_content')
        )
    )
).get(id=your_document_id)

步骤2：给Block添加快捷访问属性 #

为了避免模板中重复判断内容类型，给Block模型加一个只读属性，直接返回预加载好的内容对象：

class Block(models.Model):
    # 原有字段定义保持不变
    @property
    def loaded_content(self):
        # 优先返回预加载的内容
        if hasattr(self, '_header_content') and self._header_content:
            return self._header_content[0]
        if hasattr(self, '_paragraph_content') and self._paragraph_content:
            return self._paragraph_content[0]
        # 兜底逻辑，未预取时走原有查询逻辑
        return self.content_block

步骤3：修改模板使用预加载属性 #

把模板中原来的block.content_block替换为block.loaded_content即可：

\documentclass[a4paper,10pt]{report}
\begin{document}
  {% for block in chapter.block_set.all %}
    {% if block.content_block_type.name == 'header' %}
      \section{ {{- block.loaded_content.latex_text -}} }
    {% elif block.content_block_type.name == 'paragraph' %}
      {{ block.loaded_content.latex_text }}
    {% endif %}
  {% endfor %}
\end{document}

性能说明 #

该方案总共只会触发4次数据库查询：

1. 查询Document实例
2. 查询关联的所有Block实例
3. 查询所有Block关联的Header实例
4. 查询所有Block关联的Paragraph实例
   完全解决了遍历Block时的N+1查询问题，渲染性能会有大幅提升。

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