Java8 Stream 并行流

简介: Java8 Stream 并行流

并行流就是把一系列数据自动拆分成多个数据块,并使用多个线程来处理这些数据块,这样就可以利用现代CPU多核的优势,把计算任务分配给多个CPU核心,最后再汇总结果。让它们都忙起来~


# 并行流使用的线程池


  • 先来看看并行流所使用的线程

public static void main(String[] args) {
    Random random = new Random();
    //非并行流
    Stream.generate(() -> random.nextInt(1000))
            .limit(10)
            .forEach(x -> System.out.println(x + ": " + Thread.currentThread().getName()));
    //并行流
    Stream.generate(() -> random.nextInt(1000))
            .limit(1000)
            //转换成并行流
            .parallel()
            .forEach(x -> System.out.println(x + ": " + Thread.currentThread().getName()));
}


  • 结果: 可以看到并行流除了使用main线程,还使用了ForkJoinPool线程。


image.png

image.png


  • ForkJoinPool使用演示

@Getter
@Setter
@Slf4j
public class ForkJoinCalculator extends RecursiveTask<Long> {
    /**
     * 获取通用的ForkJoinPool
     */
    private static final ForkJoinPool FORK_JOIN_POOL = ForkJoinPool.commonPool();
    /**
     * 最小批次元素数量
     */
    private static final int MIN_BATCH_SIZE = 1000;
    private long[] dataArray;
    private int startIndex;
    private int endIndex;
    public ForkJoinCalculator(long[] dataArray) {
        this.dataArray = dataArray;
        this.startIndex = 0;
        this.endIndex = dataArray.length;
    }
    private ForkJoinCalculator(long[] dataArray, int startIndex, int endIndex) {
        this.dataArray = dataArray;
        this.startIndex = startIndex;
        this.endIndex = endIndex;
    }
    @Override
    protected Long compute() {
        ForkJoinCalculator.printWithThread("cur startIndex=%s,endIndex=%s", startIndex, endIndex);
        long curTotal = 0L;
        if (endIndex - startIndex <= MIN_BATCH_SIZE) {
            //如果需要计算的元素个数小于最小阈值则直接计算
            for (int i = startIndex; i < endIndex; i++) {
                curTotal += dataArray[i];
            }
            ForkJoinCalculator.printWithThread("直接计算curTotal=%s", curTotal);
        } else {
            // 如要要计算的元素个数大于设定的最小阈值,则进行任务拆分
            // 将元素startIndex~endIndex个任务拆分成两份
            // 计算中间索引
            int middleIndex = (startIndex + endIndex) / 2;
            ForkJoinCalculator leftForkJoinCalculator = new ForkJoinCalculator(dataArray, startIndex, middleIndex);
            ForkJoinCalculator rightForkJoinCalculator = new ForkJoinCalculator(dataArray, middleIndex, endIndex);
            // fork():将任务push到线程的工作队列
            // join(): 计算结果
            long leftTotal = leftForkJoinCalculator.fork().join();
            // 第二个子任务,有可能继续划分
            long rightTotal = rightForkJoinCalculator.compute();
            ForkJoinCalculator.printWithThread("leftTotal=%s,rightTotal=%s", leftTotal, rightTotal);
            curTotal = leftTotal + rightTotal;
        }
        return curTotal;
    }
    public static void printWithThread(String format, Object... args) {
        String formatStr = String.format(format, args);
        String threadName = Thread.currentThread().getName();
        System.out.println(threadName + ": " + formatStr);
    }
}


  • 工作窃取: work stealing
  • 为了保证每个线程完成的任务量相对平均,每个线程都会将分配给自己的任务保存在一个双向队列,每执行完一个任务,就会从队列头取出下一个任务执行,如果当前线程比较优秀,早早地完成了自己队列内的所有任务,则会从其他线程的队列的尾巴上"窃取"一个任务来执行,直到所有线程的队列都清空,以保证以最快的速度执行完所有的任务。


# 性能与注意点


使用并行流并不保证性能一定比非并行流和for循环好,有时候可能更差,这取决于要处理的数据集的数据结构。并且,在使用并行流之前,必须确保用的对,否则可能出现计算结果错误的严重后果。请记住,并行化并不是没有代价的。


  • 演示一个错误使用并行流的例子

public static class Add {
    private long total;
    public void add(long curVal) {
        total += curVal;
    }
}
public void streamDemo() {
    Add add = new Add();
    LongStream.rangeClosed(1, 1_000_000)
            .forEach(add::add);
    System.out.println(add.total);
}
public void parallelStreamDemo() {
    Add add = new Add();
    LongStream.rangeClosed(1, 1_000_000)
            .parallel()
            .forEach(add::add);
    System.out.println(add.total);
}


  • 结果: 使用并行流计算出来的结果与正确结果500000500000出现了非常大的差异,这是因为在使用并行流时,多个线程同时访问total+=curVal;,会出现线程安全问题。当然,你可以将add()设置为synchronized同步方法,但是很显然性能会很差。

500000500000
53692171876

# 高效使用并行流的建议



  • 如果不确定使用并行流是否能提高程序执行的效率:请测量、测试。使用并行流的结果并不一定会产生与我们预期相符的结果,最好的方法就是在使用之前做足够的基准测试来检测性能。
  • 注意装箱和拆箱成本,尽量使用原始类型特化流IntStream, LongStreamDoubleStream
  • 对于较少的数据量,使用并行流从来都不是一个好的决定。使用并行流带来的好处还抵不过线程的额外开销。
  • 因为并行流计算之前需要将数据集拆分,所以在使用并行流之前需要考虑数据集是否易于拆分。例如,ArrayList就比LinkedList易于拆分,因为ArrayList不需要遍历整个数据集就可拆分,而后者必须完整遍历。
  • 按照可拆分性:


数据源 可拆分性
ArrayList 极佳
LinkedList
Stream.range() 极佳
Stream.iterate()
HashSet
TreeSet


相关文章
|
1月前
|
安全 Java 测试技术
Java并行流陷阱:为什么指定线程池可能是个坏主意
本文探讨了Java并行流的使用陷阱,尤其是指定线程池的问题。文章分析了并行流的设计思想,指出了指定线程池的弊端,并提供了使用CompletableFuture等替代方案。同时,介绍了Parallel Collector库在处理阻塞任务时的优势和特点。
|
19天前
|
存储 Java 数据挖掘
Java 8 新特性之 Stream API:函数式编程风格的数据处理范式
Java 8 引入的 Stream API 提供了一种新的数据处理方式,支持函数式编程风格,能够高效、简洁地处理集合数据,实现过滤、映射、聚合等操作。
35 6
|
19天前
|
Java API 开发者
Java中的Lambda表达式与Stream API的协同作用
在本文中,我们将探讨Java 8引入的Lambda表达式和Stream API如何改变我们处理集合和数组的方式。Lambda表达式提供了一种简洁的方法来表达代码块,而Stream API则允许我们对数据流进行高级操作,如过滤、映射和归约。通过结合使用这两种技术,我们可以以声明式的方式编写更简洁、更易于理解和维护的代码。本文将介绍Lambda表达式和Stream API的基本概念,并通过示例展示它们在实际项目中的应用。
|
10天前
|
Rust 安全 Java
Java Stream 使用指南
本文介绍了Java中Stream流的使用方法,包括如何创建Stream流、中间操作(如map、filter、sorted等)和终结操作(如collect、forEach等)。此外,还讲解了并行流的概念及其可能带来的线程安全问题,并给出了示例代码。
|
21天前
|
安全 Java API
Java中的Lambda表达式与Stream API的高效结合####
探索Java编程中Lambda表达式与Stream API如何携手并进,提升数据处理效率,实现代码简洁性与功能性的双重飞跃。 ####
24 0
|
1月前
|
Java API 数据处理
探索Java中的Lambda表达式与Stream API
【10月更文挑战第22天】 在Java编程中,Lambda表达式和Stream API是两个强大的功能,它们极大地简化了代码的编写和提高了开发效率。本文将深入探讨这两个概念的基本用法、优势以及在实际项目中的应用案例,帮助读者更好地理解和运用这些现代Java特性。
|
2月前
|
Java 流计算
Flink-03 Flink Java 3分钟上手 Stream 给 Flink-02 DataStreamSource Socket写一个测试的工具!
Flink-03 Flink Java 3分钟上手 Stream 给 Flink-02 DataStreamSource Socket写一个测试的工具!
45 1
Flink-03 Flink Java 3分钟上手 Stream 给 Flink-02 DataStreamSource Socket写一个测试的工具!
|
2月前
|
Java Shell 流计算
Flink-02 Flink Java 3分钟上手 Stream SingleOutputStreamOpe ExecutionEnvironment DataSet FlatMapFunction
Flink-02 Flink Java 3分钟上手 Stream SingleOutputStreamOpe ExecutionEnvironment DataSet FlatMapFunction
25 1
Flink-02 Flink Java 3分钟上手 Stream SingleOutputStreamOpe ExecutionEnvironment DataSet FlatMapFunction
|
2月前
|
存储 Java 数据处理
Flink-01 介绍Flink Java 3分钟上手 HelloWorld 和 Stream ExecutionEnvironment DataSet FlatMapFunction
Flink-01 介绍Flink Java 3分钟上手 HelloWorld 和 Stream ExecutionEnvironment DataSet FlatMapFunction
37 1
|
3月前
|
SQL Java Linux
Java 8 API添加了一个新的抽象称为流Stream
Java 8 API添加了一个新的抽象称为流Stream