是否可以在Stream.parallel()中设置线程的优先级?

问题描述 投票:11回答:2

如果我想在后台任务中并行运行Stream,是否可以以较低优先级运行它?如果是这样怎么样?

java java-8 java-stream java-threads
2个回答
9
投票

对的,这是可能的。

程序如下:

  1. 创建一个ForkJoinWorkerThreadFactory,创建具有适当优先级的线程。
  2. 使用上面的线程工厂创建一个ForkJoinPool
  3. 实例化并行流。
  4. 通过将流提交到ForkJoinPool来运行流

像这样的东西:

public class MyThread extends ForkJoinWorkerThread {
    public MyThread(ForkJoinPool pool, int priority) {
        super(pool);
        setPriority(priority);
    }
}

final int poolSize = ...
final int priority = ...

List<Long> aList = LongStream.rangeClosed(firstNum, lastNum).boxed()
  .collect(Collectors.toList());

ForkJoinWorkerThreadFactory factory = new ForkJoinWorkerThreadFactory() {
    public ForkJoinWorkerThread newThread(ForkJoinPool pool) {
         return new MyThread(pool, priority);
    }
};
/*
ForkJoinWorkerThreadFactory factory = pool -> new MyThread(
  pool,
  priority
);
*/

ForkJoinPool customThreadPool = new ForkJoinPool(
    poolSize, factory, null, false);
long actualTotal = customThreadPool.submit(
    () -> aList.parallelStream().reduce(0L, Long::sum)).get();

(示例代码改编自http://www.baeldung.com/java-8-parallel-streams-custom-threadpool


0
投票

我认为更好的方法就像描述here

public class CustomForkJoinWorkerThreadFactory implements ForkJoinWorkerThreadFactory {

    private final int threadPriority;

    public CustomForkJoinWorkerThreadFactory(int threadPriority) {
        this.threadPriority = threadPriority;
    }

    @Override           
    public ForkJoinWorkerThread newThread(ForkJoinPool pool)
    {
        final ForkJoinWorkerThread worker = ForkJoinPool.defaultForkJoinWorkerThreadFactory.newThread(pool);
        worker.setPriority(threadPriority);
        return worker;
    }
}

它允许你仍然使用“默认”ForkJoinWorkerThread,但你可以设置优先级/名称/等。使用如下:

new ForkJoinPool(poolSize, new CustomForkJoinWorkerThreadFactory(Thread.MIN_PRIORITY), null, false);
© www.soinside.com 2019 - 2024. All rights reserved.