如何在Django中使用aggregation聚合优化购物车价格计算逻辑

直接在模型的save方法内修改即可，无需改动视图层代码，优化后的逻辑依然属于模型层的计算逻辑，保持封装性更合理。

具体修改步骤 #

1. 先在模型文件头部引入Django内置的聚合函数：

from django.db.models import Sum

2. 替换原final_price的计算逻辑，修改后的save方法如下：

def save(self, *args, **kwargs):
    if self.id:
        self.total_products = self.meals.count()
        # 用aggregate单查询计算总价，空购物车时返回0避免None赋值报错
        price_agg = self.meals.aggregate(total=Sum('final_price'))
        self.final_price = price_agg['total'] or 0
    super().save(*args, **kwargs)

优化说明 #

• 原写法会先查询所有关联的CartMeal对象加载到内存，再用Python遍历求和，购物车商品较多时内存占用高、查询效率低。
• aggregate会直接生成带SUM()的SQL语句，计算完全在数据库端完成，仅返回求和结果，单条查询即可完成计算，性能提升明显。

进阶优化（可选） #

如果你想把total_products的计数查询也合并，进一步减少查询次数，可以同时引入Count聚合，将两次查询合并为一次：

# 头部导入新增Count
from django.db.models import Sum, Count

# save方法修改为
def save(self, *args, **kwargs):
    if self.id:
        agg_result = self.meals.aggregate(
            total_count=Count('id'),
            total_price=Sum('final_price')
        )
        self.total_products = agg_result['total_count'] or 0
        self.final_price = agg_result['total_price'] or 0
    super().save(*args, **kwargs)

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