本文旨在介绍如何通过提取方法或使用累加器模式来优化Java Stream操作，以避免重复代码，并提高代码的可读性和效率。我们将通过示例代码演示如何将多个类似的Stream操作合并为一个更简洁的实现，从而减少代码冗余。
在Java编程中，我们经常会遇到需要对集合中的元素进行多次类似操作的情况。例如，计算一个商品列表中不同属性的总价。如果直接使用多个Stream分别计算，会导致代码冗余且难以维护。本文将介绍两种优化方案：提取方法和使用累加器模式，以更紧凑地实现多个变量的赋值。
方法提取
最直接的方法是将重复的Stream操作提取到一个单独的方法中。这样可以避免代码重复，并提高代码的可读性。
import java.math.BigDecimal;
import java.util.List;
import java.util.function.Function;
public class StreamOptimization {
private static final BigDecimal ZERO = BigDecimal.ZERO;
public static void main(String[] args) {
List items = List.of(
new CommodityItem(BigDecimal.valueOf(10), BigDecimal.valueOf(20), BigDecimal.valueOf(30)),
new CommodityItem(BigDecimal.valueOf(5), BigDecimal.valueOf(15), BigDecimal.valueOf(25))
);
BigDecimal batchPrice1 = sum(items, CommodityItem::getTotalPrice1);
BigDecimal batchPrice2 = sum(items, CommodityItem::getTotalPrice2);
BigDecimal batchPrice3 = sum(items, CommodityItem::getTotalPrice3);
System.out.println("batchPrice1: " + batchPrice1);
System.out.println("batchPrice2: " + batchPrice2);
System.out.println("batchPrice3: " + batchPrice3);
}
private static BigDecimal sum(List items, Function mapper) {
return items.stream()
.map(mapper)
.reduce(ZERO, BigDecimal::add);
}
static class CommodityItem {
private BigDecimal totalPrice1;
private BigDecimal totalPrice2;
private BigDecimal totalPrice3;
public CommodityItem(BigDecimal totalPrice1, BigDecimal totalPrice2, BigDecimal totalPrice3) {
this.totalPrice1 = totalPrice1;
this.totalPrice2 = totalPrice2;
this.totalPrice3 = totalPrice3;
}
public BigDecimal getTotalPrice1() {
return totalPrice1;
}
public BigDecimal getTotalPrice2() {
return totalPrice2;
}
public BigDecimal getTotalPrice3() {
return totalPrice3;
}
}
} 在这个例子中，sum 方法接收一个 List
立即学习“Java免费学习笔记（深入）”；
累加器模式
如果需要只迭代一次集合，可以使用累加器模式。这种模式通过创建一个累加器类，并在迭代过程中更新累加器的状态来实现。
import java.math.BigDecimal;
import java.util.List;
public class StreamOptimization {
public static void main(String[] args) {
List items = List.of(
new CommodityItem(BigDecimal.valueOf(10), BigDecimal.valueOf(20), BigDecimal.valueOf(30)),
new CommodityItem(BigDecimal.valueOf(5), BigDecimal.valueOf(15), BigDecimal.valueOf(25))
);
YouNameIt accumulator = new YouNameIt();
items.forEach(item -> accumulator.accumulate(item));
BigDecimal batchPrice1 = accumulator.getBatchPrice1();
BigDecimal batchPrice2 = accumulator.getBatchPrice2();
BigDecimal batchPrice3 = accumulator.getBatchPrice3();
System.out.println("batchPrice1: " + batchPrice1);
System.out.println("batchPrice2: " + batchPrice2);
System.out.println("batchPrice3: " + batchPrice3);
}
static class YouNameIt {
private BigDecimal batchPrice1 = BigDecimal.ZERO;
private BigDecimal batchPrice2 = BigDecimal.ZERO;
private BigDecimal batchPrice3 = BigDecimal.ZERO;
public BigDecimal getBatchPrice1() {
return batchPrice1;
}
public BigDecimal getBatchPrice2() {
return batchPrice2;
}
public BigDecimal getBatchPrice3() {
return batchPrice3;
}
void accumulate(CommodityItem item) {
batchPrice1 = batchPrice1.add(item.getTotalPrice1());
batchPrice2 = batchPrice2.add(item.getTotalPrice2());
batchPrice3 = batchPrice3.add(item.getTotalPrice3());
}
}
static class CommodityItem {
private BigDecimal totalPrice1;
private BigDecimal totalPrice2;
private BigDecimal totalPrice3;
public CommodityItem(BigDecimal totalPrice1, BigDecimal totalPrice2, BigDecimal totalPrice3) {
this.totalPrice1 = totalPrice1;
this.totalPrice2 = totalPrice2;
this.totalPrice3 = totalPrice3;
}
public BigDecimal getTotalPrice1() {
return totalPrice1;
}
public BigDecimal getTotalPrice2() {
return totalPrice2;
}
public BigDecimal getTotalPrice3() {
return totalPrice3;
}
}
} 在这个例子中，YouNameIt 类充当累加器。accumulate 方法用于更新累加器的状态。通过使用 forEach 方法迭代集合，并调用 accumulate 方法，我们可以只迭代一次集合就完成多个变量的赋值。
总结
本文介绍了两种优化Java Stream操作的方法，以更紧凑地为多个变量赋值。提取方法可以避免代码重复，并提高代码的可读性。累加器模式可以在只迭代一次集合的情况下完成多个变量的赋值。选择哪种方法取决于具体的需求。如果性能是关键，累加器模式可能更适合。如果代码可读性更重要，提取方法可能更合适。
