使用Java Stream更紧凑地进行多值赋值_技术教程

本文旨在介绍如何使用Java Stream API更高效地对多个变量进行赋值，避免重复代码。通过提取通用方法或使用自定义累加器，可以显著提高代码的可读性和简洁性，并减少不必要的迭代。我们将探讨两种实现方式，并分析它们的优缺点。

在处理集合数据时，经常需要对集合中的元素进行计算，并将结果赋值给多个变量。如果直接使用多个独立的Stream操作，会导致代码冗余，不易维护。本文将介绍两种优化方法，帮助你使用Java Stream API更紧凑地完成多值赋值。

方法一：提取通用方法

这种方法的核心思想是将重复的Stream操作提取到一个通用方法中，然后通过传入不同的mapper函数来实现不同的计算。

示例代码：

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 即可。

缺点：

需要多次迭代集合，效率相对较低。

方法二：使用自定义累加器

这种方法通过自定义一个累加器类，在一次迭代中完成所有计算。

示例代码：

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;
        }
    }
}

代码解释：