本文旨在介绍如何使用Java Stream API更高效地对多个变量进行赋值，避免重复代码。通过提取通用方法或使用自定义累加器，可以显著提高代码的可读性和简洁性，并减少不必要的迭代。我们将探讨两种实现方式，并分析它们的优缺点。
在处理集合数据时，经常需要对集合中的元素进行计算，并将结果赋值给多个变量。如果直接使用多个独立的Stream操作，会导致代码冗余，不易维护。本文将介绍两种优化方法，帮助你使用Java Stream API更紧凑地完成多值赋值。
方法一：提取通用方法
这种方法的核心思想是将重复的Stream操作提取到一个通用方法中，然后通过传入不同的mapper函数来实现不同的计算。
示例代码：
立即学习“Java免费学习笔记（深入）”；
import java.math.BigDecimal;
import java.util.List;
import java.util.function.Function;
public class StreamExample {
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
和一个 Function 作为参数。 - Function 接口用于指定如何从 CommodityItem 中提取需要计算的 BigDecimal 值。
- 在 main 方法中，我们分别调用 sum 方法，并传入不同的 CommodityItem::getTotalPriceX 方法引用，从而计算出不同的总价。
优点：
- 代码简洁，易于理解。
- 易于扩展，如果需要计算更多的值，只需要添加对应的 batchPriceX 和 CommodityItem::getTotalPriceX 即可。
缺点：
- 需要多次迭代集合，效率相对较低。
方法二：使用自定义累加器
这种方法通过自定义一个累加器类，在一次迭代中完成所有计算。
示例代码：
立即学习“Java免费学习笔记（深入）”；
import java.math.BigDecimal;
import java.util.List;
public class StreamExample2 {
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 类充当累加器，包含 batchPrice1、batchPrice2 和 batchPrice3 三个成员变量，分别用于存储计算结果。
- accumulate 方法接收一个 CommodityItem 对象，并将对应的价格累加到相应的成员变量中。
- 在 main 方法中，我们首先创建一个 YouNameIt 对象，然后使用 forEach 方法遍历集合，并调用 accumulate 方法进行累加。
- 最后，通过 getBatchPriceX 方法获取计算结果。
优点：
- 只需要一次迭代集合，效率较高。
缺点：
- 代码相对复杂，不易理解。
- 如果需要计算更多的值，需要修改 YouNameIt 类的代码。
总结
本文介绍了两种使用Java Stream API更紧凑地进行多值赋值的方法。第一种方法通过提取通用方法，代码简洁易懂，但效率较低；第二种方法通过自定义累加器，效率较高，但代码相对复杂。选择哪种方法取决于具体的应用场景和性能要求。
在实际开发中，可以根据以下原则进行选择：
- 如果集合大小较小，或者对性能要求不高，可以选择第一种方法。
- 如果集合大小较大，并且对性能要求较高，可以选择第二种方法。
- 如果需要计算的值的数量较少，可以选择第一种方法。
- 如果需要计算的值的数量较多，可以选择第二种方法。
希望本文能够帮助你更好地使用Java Stream API，提高代码效率和可读性。
