executor.invokeAll和Future.get指定限制时间

设定时间的任务

如果某一个子线程任务不能在规定的时间内完成,那么应该抛弃该任务,不再需要其结果,同时中断该任务,减少性能开销。

Future.get方法有一个重载,传入限制的时间

V get(long timeout, TimeUnit unit)
    throws InterruptedException, ExecutionException, TimeoutException;

如果阻塞时长超过指定的timeout,将抛出TimeoutException异常。
简单使用

public class TimeLimitTask {
    static class A implements Callable<String> {
        @Override
        public String call() throws Exception {
            return "Hello World";
        }
    }
    public static void main(String[] args) {
        ExecutorService executor = Executors.newCachedThreadPool();
        String ans = null;
        long endTime = System.nanoTime() + 10000000;
        Future<String> f = executor.submit(new A());
        long timeLeft = endTime - System.nanoTime();
        try {
            //获取结果超出指定时间,抛出Timeout异常
            ans = f.get(timeLeft, TimeUnit.NANOSECONDS);
        } catch (ExecutionException e) {
            ans = "default1";
        } catch (TimeoutException e) {
            e.printStackTrace();
            ans = "default2";
            //超时后,应该取消该任务,减小不必要的开销
            f.cancel(true);
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
            ans = "default3";
        }
        System.out.println(ans);
    }
}

使用ExecutorServiceinvokeAll方法

该方法可以传入一组任务,并给定完成的时间,返回一组FutureinvokeAll按照任务完成的顺序将Future添加到返回的集合中; 超过指定时间后,所有未完成的任务将会取消,客户可以通过调用FuturegetisCancelled判断情况。

简单使用

public class ExecutorInvokeAllTest {
    static class Task implements Callable<String> {
        static int val = 0;
        @Override
        public String call() throws Exception {
            if (val == 3) Thread.sleep(3000);
            return "Hello Wolrd" + (val ++);
        }
    }
    public static void main(String[] args) {
        ExecutorService executor = Executors.newCachedThreadPool();
        List<Task> tasks = new ArrayList<>();
        for (int i = 0; i < 5; i++) {
            tasks.add(new Task());
        }
        List<String> ans = new ArrayList<>(tasks.size());
        try {
            List<Future<String>> fts = executor.invokeAll(tasks, 2, TimeUnit.SECONDS);
            for (Future<String> f : fts) {
                try {
                    f.cancel(true);
                    if (!f.isCancelled()) {
                        ans.add(f.get());
                    } else {
                        System.out.println("task was cancelled!");
                    }
                } catch (ExecutionException e) {
                    ans.add("error");
                    System.out.println(e.getCause());
                }
            }
        } catch (InterruptedException e) {
            e.printStackTrace();
            Thread.currentThread().interrupt();
        }
        for (String s : ans) {
            System.out.println(s);
        }
    }
}
  • invokeAll添加任务组,线程池处理
  • f.cancel取消任务,f.isCancelled()判断任务是否已经取消,如果任务已经取消,再调用get方法,会抛出CancellationException异常。
  • 如果某个任务执行的时间超过指定,抛出CancellationException异常,所有剩下的未完成的任务将取消。

猜你喜欢

转载自blog.csdn.net/sweatott/article/details/71773411