优化Java Stream：更紧凑地为多个变量赋值_技术教程

本文旨在介绍如何通过提取方法或使用累加器模式来优化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 和一个 Function 作为参数。Function 用于指定要提取的属性。通过调用 sum 方法，我们可以避免重复编写Stream操作的代码。

累加器模式

如果需要只迭代一次集合，可以使用累加器模式。这种模式通过创建一个累加器类，并在迭代过程中更新累加器的状态来实现。

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