如何通过Django ORM对JSONField中的列表元素求和聚合

问题背景 #

需要计算Django模型中JSONField（存储结构如[{"score": 10}, {"score": 0}, {"score": 40}, ...]）内所有元素的score字段总和，已通过原生SQL实现，但想通过ORM完成。之前尝试时遇到两个核心问题：

• 直接用jsonb_array_elements将JSON列表转集合后，在聚合函数中调用会触发PostgreSQL报错：aggregate function calls cannot contain set-returning function calls
• 尝试用.extra()实现LATERAL JOIN时，Django会自动给LATERAL子查询加引号，导致报错：ERROR: no relation "LATERAL (SELECT ..."

ORM实现方案 #

1. 计算单条记录的score总和

通过子查询结合Django原生的JSON函数，自动生成LATERAL JOIN逻辑，避免手动写原生SQL：

from django.db.models import Sum, Subquery, OuterRef
from django.contrib.postgres.fields.jsonb import KeyTextTransform
from django.contrib.postgres.functions import JsonbArrayElements
from myapp.models import MyModel

# 子查询：展开当前记录的data数组，提取每个元素的score并求和
score_sum_subquery = Subquery(
    MyModel.objects.filter(pk=OuterRef('pk'))
    .annotate(item=JsonbArrayElements('data'))  # 展开JSON数组为多行
    .annotate(score=KeyTextTransform('score', 'item'))  # 提取每个元素的score
    .values('pk')
    .annotate(total=Sum('score'))  # 对当前记录的所有score求和
    .values('total')
)

# 主查询：为每条记录标注score总和
results = MyModel.objects.annotate(total_score=score_sum_subquery).values('pk', 'total_score')

2. 计算所有记录的总score

如果需要统计所有记录中JSON列表的score总和，可直接先展开所有数组元素，再聚合求和：

total_all_scores = MyModel.objects.annotate(item=JsonbArrayElements('data'))\
    .annotate(score=KeyTextTransform('score', 'item'))\
    .aggregate(total=Sum('score'))['total']

方案说明 #

• 核心逻辑是先展开JSON数组为多行记录，再对展开后的score字段执行聚合，避开了PostgreSQL对“聚合函数嵌套集合返回函数”的限制。
• 第一种方案中的Subquery+OuterRef会让Django自动生成LATERAL JOIN SQL，无需手动处理表名引号问题，完全符合ORM规范。
• 用到的JsonbArrayElements和KeyTextTransform都是Django为PostgreSQL JSONB字段提供的原生函数，兼容性和可维护性优于原生SQL拼接。

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