并行流,收集器和线程安全

时间:2014-03-12 11:37:16

标签: java concurrency parallel-processing java-8 java-stream

请参阅下面的简单示例,该示例计算列表中每个单词的出现次数:

Stream<String> words = Stream.of("a", "b", "a", "c");
Map<String, Integer> wordsCount = words.collect(toMap(s -> s, s -> 1,
                                                      (i, j) -> i + j));

最后,wordsCount{a=2, b=1, c=1}

但是我的信息流非常庞大,我希望将这项工作并行化,所以我写道:

Map<String, Integer> wordsCount = words.parallel()
                                       .collect(toMap(s -> s, s -> 1,
                                                      (i, j) -> i + j));

但是我注意到wordsCount是一个简单的HashMap所以我想知道是否需要明确要求并发映射以确保线程安全:

Map<String, Integer> wordsCount = words.parallel()
                                       .collect(toConcurrentMap(s -> s, s -> 1,
                                                                (i, j) -> i + j));

非并发收集器可以安全地与并行流一起使用,还是在从并行流中收集时我应该只使用并发版本?

3 个答案:

答案 0 :(得分:40)

  

非并发收集器可以安全地与并行流一起使用,还是在从并行流中收集时我应该只使用并发版本?

在并行流的collect操作中使用非并发收集器是安全的。

Collector界面的specification中,在包含六个要点的部分中,是:

  

对于非并发收集器,从结果提供者,累加器或组合器函数返回的任何结果必须是串行线程限制的。这使得集合可以并行发生,而收集器无需实现任何其他同步。减少实现必须管理输入被正确分区,分区是单独处理的,并且仅在累积完成后才进行组合。

这意味着Collectors类提供的各种实现可以与并行流一起使用,即使其中一些实现可能不是并发收集器。这也适用于您可能实现的任何非并发收集器。它们可以安全地与并行流一起使用,前提是您的收集器不会干扰流源,无副作用,顺序无关等。

我还建议阅读java.util.stream包文档的Mutable Reduction部分。在本节的中间是一个声明可并行化的示例,但它将结果收集到ArrayList,这不是线程安全的。

这种方式的工作方式是以非并发收集器结尾的并行流确保不同的线程始终在中间结果集合的不同实例上运行。这就是为什么收集器具有Supplier函数的原因,用于创建与线程一样多的中间集合,因此每个线程可以累积到它自己的集合中。当要合并中间结果时,它们会在线程之间安全地切换,并且在任何给定时间只有一个线程合并任何一对中间结果。

答案 1 :(得分:19)

如果所有收集者遵守规范中的规则,则可以安全地并行或顺序运行。并行准备是这里设计的关键部分。

并发和非并发收集器之间的区别与并行化方法有关。

普通(非并发)收集器通过合并子结果来操作。因此,源被分成一堆块,每个块被收集到一个结果容器(如列表或映射)中,然后子结果被合并到一个更大的结果容器中。这是安全且保持订单的,但对于某些类型的容器 - 尤其是地图 - 可能很昂贵,因为按键合并两个地图通常很昂贵。

并发收集器会创建一个结果容器,其插入操作保证是线程安全的,并从多个线程中将元素放入其中。使用像ConcurrentHashMap这样的高度并发的结果容器,这种方法可能比合并普通的HashMaps更好。

因此,并发收集器严格优于普通收集器。而且他们没有成本;因为元素是从许多线程中被抨击的,所以并发收集器通常不能保留遭遇顺序。 (但是,通常你不关心 - 在创建单词计数直方图时,你不关心你首先计算的“foo”实例。)

答案 2 :(得分:10)

使用非并发集合和非原子计数器与并行流是安全的。

如果您查看Stream::collect的文档,可以找到以下段落:

  

reduce(Object, BinaryOperator)类似,可以并行化收集操作,而无需额外的同步。

对于方法Stream::reduce

  

虽然与简单地在循环中改变运行总计相比,这似乎是一种更迂回的执行聚合的方式,但是还原操作可以更优雅地并行化,而无需额外的同步并且大大降低了数据竞争的风险。

这可能有点令人惊讶。但请注意,并行流基于 fork-join模型。这意味着并发执行的工作方式如下:

  • 将序列分成两个大小相同的部分
  • 单独处理每个部分
  • 收集两个部分的结果并将它们合并为一个结果

在第二步中,三个步骤以递归方式应用于子序列。

一个例子应该清楚说明。

IntStream.range(0, 4)
    .parallel()
    .collect(Trace::new, Trace::accumulate, Trace::combine);

Trace 的唯一目的是记录构造函数和方法调用。如果执行此语句,则会打印以下行:

thread:  9  /  operation: new
thread: 10  /  operation: new
thread: 10  /  operation: accumulate
thread:  1  /  operation: new
thread:  1  /  operation: accumulate
thread:  1  /  operation: combine
thread: 11  /  operation: new
thread: 11  /  operation: accumulate
thread:  9  /  operation: accumulate
thread:  9  /  operation: combine
thread:  9  /  operation: combine

您可以看到,已创建了四个 Trace 对象,每个对象上调用了一次 accumulate ,并且已使用 combine 三次将四个物体合二为一。每个对象一次只能由一个线程访问。这使代码线程安全,同样适用于方法 Collectors :: toMap