Java ThreadPoolExexecutor使用流和可調用對象

Question

我有一個實現Callable的類，並且它具有重寫call並返回Long 。

我創建了一個List的Callable<Long>作為

List<Callable<Long>> callables = new ArrayList<>();
for (File fileEntry : folder.listFiles()) {
    callables.add(new DataProcessor(fileEntry));

我有

ExecutorService threadPoolExecutor = Executors.newFixedThreadPool(10);

我打電話

threadPoolExecutor.invokeAll(callables)
    .stream()
    .map(future -> {
        try {
            return future.get();
        } catch (Exception e) {
            throw new IllegalStateException(e);
        }
        })
        .collect(Collectors.toLong(/* what goes here? */));

我要做的是將future.get()所有返回值future.get() 。

另外，由於我正在調用invokeAll，所以我仍然需要關閉執行器嗎？

Answer 1

您需要的是Collectors.summingLong ：

.collect(Collectors.summingLong(r -> r));

其中r -> r只是一個ToLongFunction ，它使Callable s返回的每個Long占一個long 。

另外，由於我正在調用invokeAll，所以我仍然需要關閉執行器嗎？

ExecutorService.invokeAll沒有記錄自動關機。 所以你需要自己關閉它

Answer 2

您可以使用Stream.mapToLong映射future.get為LongStream然后找到sum流為：

long sum = threadPoolExecutor.invokeAll(callables)
            .stream()
            .mapToLong(future -> {
                try {
                    return future.get();
                } catch (Exception e) {
                    throw new IllegalStateException(e);
                }
            }) // LongStream
            .sum(); // sum of the stream

注意 ：這簡化了使用Collectors.summingLong流API調用鏈。 它允許避免遍歷集合時創建冗余的臨時對象。

除了 ：您還可以按以下方式collect Callable ：

List<Callable<Long>> callables = fileList.stream()
                                         .map(fileEntry -> new DataProcessor(fileEntry))
                                         .collect(Collectors.toList());

由於我正在調用invokeAll，因此我仍然需要關閉執行器嗎？

是的，您必須關閉ExecutorService 。 您還可以使用isShutDown() API確認相同狀態：

System.out.println(threadPoolExecutor.isShutdown()); // would return false