在 Java 8 引入 Stream API 后,集合数据处理的方式发生了革命性的变化。相比传统的 for
循环和 Iterator
,Stream API 提供了更 声明式(Declarative) 的编程风格,让代码更简洁、可读性更强,同时还能利用多核 CPU 进行并行计算,提升性能。
在企业开发中,我们经常面临:
Stream API 能很好地解决这些问题。本文将结合真实企业开发场景,分享 Stream API 的 最佳实践、性能优化技巧及常见坑点。
需求:从数据库查询 List
,筛选出 金额大于 1000 且状态为 “PAID” 的订单,并提取订单号。
List<String> paidOrderIds = new ArrayList<>();
for (Order order : orderList) {
if (order.getAmount() > 1000 && "PAID".equals(order.getStatus())) {
paidOrderIds.add(order.getOrderId());
}
}
NullPointerException
)。List<String> paidOrderIds = orderList.stream()
.filter(order -> order.getAmount() > 1000)
.filter(order -> "PAID".equals(order.getStatus()))
.map(Order::getOrderId)
.collect(Collectors.toList());
优点:
"PAID".equals(...)
比 order.getStatus().equals("PAID")
更安全)。需求:统计每个用户的订单总金额。
Map<String, BigDecimal> userTotalAmountMap = new HashMap<>();
for (Order order : orderList) {
String userId = order.getUserId();
BigDecimal amount = order.getAmount();
userTotalAmountMap.merge(userId, amount, BigDecimal::add);
}
null
。Collectors.groupingBy
)Map<String, BigDecimal> userTotalAmountMap = orderList.stream()
.collect(Collectors.groupingBy(
Order::getUserId,
Collectors.reducing(
BigDecimal.ZERO,
Order::getAmount,
BigDecimal::add
)
));
优点:
Map
。.parallelStream()
)。需求:按 订单金额降序,创建时间升序 排序。
orderList.sort((o1, o2) -> {
int amountCompare = o2.getAmount().compareTo(o1.getAmount());
if (amountCompare != 0) {
return amountCompare;
}
return o1.getCreateTime().compareTo(o2.getCreateTime());
});
Comparator.comparing
)List<Order> sortedOrders = orderList.stream()
.sorted(Comparator
.comparing(Order::getAmount).reversed()
.thenComparing(Order::getCreateTime)
)
.collect(Collectors.toList());
优点:
thenComparing
)。❌ 错误写法(多次调用 stream()
导致重复计算):
long count = orderList.stream().filter(...).count();
List<Order> filtered = orderList.stream().filter(...).collect(Collectors.toList());
正确写法(缓存 Stream 结果):
Stream<Order> filteredStream = orderList.stream().filter(...);
long count = filteredStream.count(); // 终端操作,流关闭
List<Order> filtered = orderList.stream().filter(...).collect(Collectors.toList()); // 重新创建流
parallelStream
)谨慎使用List<Order> bigDataList = ...; // 10W+ 数据
List<String> orderIds = bigDataList.parallelStream()
.map(Order::getOrderId)
.collect(Collectors.toList());
IntStream
、LongStream
)避免自动拆箱(Integer → int
)带来的性能损耗。
// 传统写法(涉及自动拆箱)
int totalAmount = orderList.stream()
.mapToInt(Order::getAmount) // 使用 IntStream 替代 Stream
.sum();
Stream<Order> stream = orderList.stream();
List<Order> paidOrders = stream.filter(...).collect(Collectors.toList());
List<Order> bigOrders = stream.filter(...).collect(Collectors.toList()); // ❌ IllegalStateException
解决方案:每次操作都重新创建流。
forEach
不能替代 for
循环forEach
是终端操作,不能 break
或 return
。for
循环)。Optional<Order> highestOrder = orderList.stream()
.max(Comparator.comparing(Order::getAmount));
highestOrder.ifPresent(order -> {
System.out.println("最高金额订单:" + order.getOrderId());
});
场景 | 推荐方式 | 理由 |
---|---|---|
简单遍历 | for 循环 |
代码更直观,性能无差别 |
复杂数据处理 | Stream API | 代码更简洁,可读性高 |
大数据量计算 | parallelStream |
利用多核 CPU 加速计算 |
需要提前终止循环 | for 循环 |
Stream 无法 break /return |
从函数式的角度上看,过程式的代码实现将收集元素、循环迭代、各种逻辑判断耦合在一起,暴露了太多细节。当未来需求变动和变得更加复杂的情况下,过程式的代码将变得难以理解和维护
函数式的解决方案解开了代码细节和业务逻辑的耦合,类似于sql语句,表达的是**“要做什么"而不是"如何去做”**,使程序员可以更加专注于业务逻辑,写出易于理解和维护的代码。