Java调用任务集合(与invokeAll一样)

时间:2017-10-09 11:35:36

标签: java javafx concurrency task executorservice

我正在寻找与javafx.concurrent.Task相关的方法,java.util.concurrent.CallableExecutorService.invokeAll(Collection<? extends Callable<T>>)可以做什么这个方法无法与任务一起使用,因为它们是可运行的不是可赎回的。

使用Executors.callable(Runnable task)将我的任务类型转换为Callable无效,因为返回的Callable将始终返回null,我需要从Task.call()返回的值。

我了解Task并非设计用于Collections。但是我需要在我的JavaFX应用程序中使用Task并通过编写测试来解决这个问题。

除了简单地制作Task工具Callable之外,有什么想法可以绕过这个吗?我不想这样做,因为我会改变我的代码以适应我的测试。

1 个答案:

答案 0 :(得分:1)

您可以将每个任务包装在可调用的中。基本上对于每个Task<T> task,您需要Callable<T>,如下所示:

Callable<T> callable = () -> {
    task.run();
    return task.getValue();
};

因此,例如,您可以定义执行此映射的Function<Task<T>, Callable<T>>

Function<Task<T>, Callable<T>> taskWrapper = task -> () -> {
    task.run();
    return task.getValue();
};

然后给出ExecutorService execCollection<Task<T>> tasks即可

List<Future<T>> results = exec.invokeAll(tasks.stream()
    .map(taskWrapper)
    .collect(Collectors.toList()));

这是一个SSCCE:

import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.function.Function;
import java.util.stream.Collectors;

import javafx.application.Application;
import javafx.concurrent.Task;
import javafx.geometry.Insets;
import javafx.geometry.Pos;
import javafx.scene.Scene;
import javafx.scene.control.Button;
import javafx.scene.control.Label;
import javafx.scene.layout.VBox;
import javafx.stage.Stage;

public class InvokeAllTasks extends Application {

    private Random rng = new Random();
    private ExecutorService exec = Executors.newFixedThreadPool(5);
    private Function<Task<Integer>, Callable<Integer>> taskWrapper = task -> () -> {
        task.run();
        return task.getValue();
    };

    @Override
    public void start(Stage primaryStage) {
        Button runAll = new Button("Run all tasks");
        Label status = new Label();

        runAll.setOnAction(e -> {
            List<Task<Integer>> tasks = createTasks();


            Task<List<Future<Integer>>> runAllTask = new Task<List<Future<Integer>>>() {
                @Override
                protected List<Future<Integer>> call() throws Exception {
                    return exec.invokeAll(tasks.stream().map(taskWrapper).collect(Collectors.toList()));
                }
            };
            status.setText("Running...");
            runAllTask.setOnSucceeded(evt -> status.setText("All Done"));
            new Thread(runAllTask).start();


        });

        VBox root = new VBox(5, runAll, status);
        root.setMinHeight(120);
        root.setAlignment(Pos.CENTER);
        root.setPadding(new Insets(10));

        Scene scene = new Scene(root);
        primaryStage.setScene(scene);
        primaryStage.show();
    }

    @Override
    public void stop() {
        exec.shutdown();
    }

    private List<Task<Integer>> createTasks() {
        List<Task<Integer>> tasks = new ArrayList<>();
        for (int i = 1 ; i <= 8 ; i++) {
            String name = "Task "+i;
            Task<Integer> t = new Task<Integer>() {
                @Override
                protected Integer call() throws Exception {
                    System.out.println(name+" running");
                    Thread.sleep(rng.nextInt(1000)+500);
                    int result = rng.nextInt(500);
                    System.out.println(name+" computed "+result);
                    return result;
                }
            };
            tasks.add(t);
        }
        return tasks;
    }

    public static void main(String[] args) {
        launch(args);
    }
}