如何用Java循环实现Spring Boot中用户订单总金额汇总

核心思路 #

不需要两层循环，用Map来分组存储每个用户的累计金额即可，时间复杂度为O(n)，比两层循环更高效。以下提供两种实现方式，以及后续保存到数据库的步骤。

方式一：普通循环实现分组汇总 #

// 初始化Map，key为user_id，value为该用户累计的总金额
Map<Long, Double> userTotalAmountMap = new HashMap<>();

// 遍历所有销售数据，完成金额累加
for (Sales sales : allSales) {
    long userId = sales.getUser_id();
    Double currentAmount = sales.getOrders().getTotal_amount();

    // 若用户已存在则累加，不存在则初始化金额
    userTotalAmountMap.put(
        userId,
        userTotalAmountMap.getOrDefault(userId, 0.0) + currentAmount
    );
}

方式二：Stream API实现（更简洁） #

利用Java Stream的分组和汇总Collector，一行代码完成分组计算：

Map<Long, Double> userTotalAmountMap = allSales.stream()
    .collect(Collectors.groupingBy(
        Sales::getUser_id, // 按user_id分组
        Collectors.summingDouble(s -> s.getOrders().getTotal_amount()) // 汇总每组的total_amount
    ));

保存到数据库的步骤 #

1. 创建对应数据库表的实体类

假设你的表名为user_total_sales，包含user_id和amount字段：

// UserTotalSales.java
@Entity
@Table(name = "user_total_sales")
@Getter
@Setter
@NoArgsConstructor
@AllArgsConstructor
public class UserTotalSales {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    @Column(name = "user_id", nullable = false)
    private Long userId;

    @Column(name = "amount", nullable = false)
    private Double amount;

    // 快速创建对象的构造方法
    public UserTotalSales(Long userId, Double amount) {
        this.userId = userId;
        this.amount = amount;
    }
}

2. 创建JPA Repository接口

// UserTotalSalesRepository.java
public interface UserTotalSalesRepository extends JpaRepository<UserTotalSales, Long> {
}

3. 转换数据并批量保存

// 将Map中的数据转换为实体列表
List<UserTotalSales> userTotalList = userTotalAmountMap.entrySet().stream()
    .map(entry -> new UserTotalSales(entry.getKey(), entry.getValue()))
    .collect(Collectors.toList());

// 批量保存到数据库
userTotalSalesRepository.saveAll(userTotalList);

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