Java Stream reduce：全元素为null返回null，含非null元素返回求和结果的实现问题

以BigDecimal为例，这类问题适用于所有可求和的类型。先看你的测试代码：

@Test
void streamWithNullsTest() {
  final var list = new ArrayList<BigDecimal>();
  // 正常：空列表返回初始值null
  assertNull(list.stream().reduce(null, BigDecimal::add));

  // 正常：空列表返回初始值BigDecimal.ONE
  assertEquals(BigDecimal.ONE, list.stream().filter(Objects::nonNull).reduce(BigDecimal.ONE, BigDecimal::add));
  
  // 正常：过滤空值后，返回初始值null
  list.add(null);
  assertNull(list.stream().filter(Objects::nonNull).reduce(null, BigDecimal::add));

  // 正常：过滤空值后，返回初始值BigDecimal.ONE
  assertEquals(BigDecimal.ONE, list.stream().filter(Objects::nonNull).reduce(BigDecimal.ONE, BigDecimal::add));

  // 异常：似乎将BigDecimal.ONE与初始值null相加？
  // 需求：所有元素为null时返回null，存在非null元素时返回求和结果
  list.add(BigDecimal.ONE);
  assertNull(list.stream().filter(Objects::nonNull).reduce(null, BigDecimal::add));
}

问题原因 #

你用的是reduce(U identity, BiFunction<U, ? super T, U>)重载方法，这个方法的初始值（identity）必须是累加器的单位元（比如加法的0、BigDecimal的ZERO）。当传入null作为初始值时：

• 若过滤后的流为空，直接返回null，符合预期；
• 若流中有非null元素，会执行null.add(元素)——本质是调用BigDecimal.add()实例方法时用null作为调用者，会触发NullPointerException，且该重载方法无法区分“流为空”和“累加结果为初始值”的场景，完全不符合你的需求。

解决方法 #

使用不带初始值的reduce(BinaryOperator<T>)重载，它返回Optional<T>，正好能区分流为空和有元素的情况：

// 实现需求的代码
BigDecimal result = list.stream()
    .filter(Objects::nonNull)
    .reduce(BigDecimal::add) // 流为空返回empty，否则返回封装求和结果的Optional
    .orElse(null); // 流为空返回null，否则返回求和结果

对应最后一个测试用例，断言应修改为：

assertEquals(BigDecimal.ONE, result);

泛型复用方案 #

如果需要在多个类型中复用这个逻辑，可以封装成泛型方法：

public static <T> T reduceWithNullWhenEmpty(Stream<T> stream, BinaryOperator<T> accumulator) {
    return stream.reduce(accumulator).orElse(null);
}

调用示例：

BigDecimal result = reduceWithNullWhenEmpty(list.stream().filter(Objects::nonNull), BigDecimal::add);

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