如何高效递归查询Django Role模型及其父级关联的所有Skill对象？

针对你给出的Django模型，要高效获取某一Role对象及其所有递归父级关联的全部Skill，最优方案是利用数据库递归CTE（公共表表达式），在数据库层面完成递归查询与关联，避免多次请求数据库带来的性能损耗。

方法一：使用Django递归CTE（推荐） #

该方法适用于Django 1.11+，且数据库支持递归CTE（如PostgreSQL、MySQL 8.0+、SQLite 3.38.0+），一次查询即可获取所有目标数据：

from django.db.models import OuterRef
from django.db.models.expressions import RawSQL
from .models import Role, Skill

def get_all_skills(role: Role):
    # 构建递归CTE，获取当前角色及其所有父级的ID
    recursive_cte = Role.objects.filter(pk=role.pk).values("pk").annotate(
        depth=RawSQL("0", [])
    ).union(
        # 递归查询父级，层级深度+1
        Role.objects.filter(parents__pk=OuterRef("pk")).values("pk").annotate(
            depth=RawSQL("depth + 1", [])
        ),
        all=True
    )
    # 通过CTE结果筛选所有关联的Skill，去重返回
    return Skill.objects.filter(role__pk__in=recursive_cte.values("pk")).distinct()

优势 #

• 数据库层面一次性完成递归遍历与关联查询，性能最优，尤其适合层级较深的场景
• 返回的是QuerySet对象，支持后续的链式操作（如排序、分页）

方法二：Python递归实现（不推荐） #

如果受限于数据库版本或Django版本无法使用CTE，可以用Python递归实现，但会产生N+1查询问题，层级越多性能越差，仅适合小规模数据场景：

from .models import Role, Skill

def get_all_skills(role: Role):
    # 用集合去重，避免重复技能
    all_skills = set(role.skills.all())
    # 递归遍历每个父级，合并技能
    for parent in role.parents.all():
        all_skills.update(get_all_skills(parent))
    return list(all_skills)

注意事项 #

• 该方法会为每个父级角色单独发起数据库查询，数据量大或层级深时性能显著下降
• 返回的是Python列表，无法再使用Django QuerySet的特性

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