中断并行流执行

考虑这段代码：

Thread thread = new Thread(() -> tasks.parallelStream().forEach(Runnable::run));

任务是应并行执行的 Runnable 列表。 当我们启动这个线程并开始执行时，根据一些计算，我们需要中断（取消）所有这些任务。

中断线程只会停止其中一个异常。我们如何对待别人？或者也许 Streams 不应该这样使用？或者你知道更好的解决方案？

您可以使用ForkJoinPool中断线程：

@Test
public void testInterruptParallelStream() throws Exception {
    final AtomicReference<InterruptedException> exc = new AtomicReference<>();

    final ForkJoinPool forkJoinPool = new ForkJoinPool(4);
    // use the pool with a parallel stream to execute some tasks
    forkJoinPool.submit(() -> {
        Stream.generate(Object::new).parallel().forEach(obj -> {
            synchronized (obj) {
                try {
                    // task that is blocking
                    obj.wait();
                } catch (final InterruptedException e) {
                    exc.set(e);
                }
            }
        });
    });

    // wait until the stream got started
    Threads.sleep(500);
    // now we want to interrupt the task execution
    forkJoinPool.shutdownNow();
    // wait for the interrupt to occur
    Threads.sleep(500);
    // check that we really got an interruption in the parallel stream threads
    assertTrue(exc.get() instanceof InterruptedException);
}

工作线程确实会被中断，从而终止阻塞操作。您也可以致电shutdown()内Consumer.

请注意，这些睡眠可能不会针对适当的单元测试进行调整，您可能有更好的想法，只需在必要时等待即可。但这足以表明它正在发挥作用。