Java - 超时长时间运行的可调用线程


public class FutureGetTimeoutTest {
    private static final ExecutorService THREAD_POOL = Executors.newFixedThreadPool(5);
    public static void main(String[] args) throws InterruptedException, ExecutionException {
        List<String> respList = new ArrayList<String>();
        List<Future<String>> futures = new ArrayList<Future<String>>();
        futures.add(THREAD_POOL.submit(new CallableTask(1L)));
        futures.add(THREAD_POOL.submit(new CallableTask(2L)));
        futures.add(THREAD_POOL.submit(new CallableTask(3L)));
        long start = System.currentTimeMillis();
        System.out.println(start);
        for (Future<String> future : futures) {
            try {
                respList.add(future.get(10000, TimeUnit.MILLISECONDS));
                /*
                 * Timeout time for 2nd Task starts only at the end of 1st Task Timeout
                 * and so 2nd task is able to run for 20s and 3rd task for 30s!
                 */
            } catch (TimeoutException e) {
                e.printStackTrace();
            }
        }
        long end = System.currentTimeMillis();
        System.out.println(end);
        System.out.println(end - start);
        System.out.println(respList);
    }
}
class CallableTask implements Callable<String> {
    private long ipAddressL;
    public CallableTask(long ipAddressL) {
        this.ipAddressL = ipAddressL;
    }
    @Override
    public String call() throws Exception {
        if (ipAddressL == 1) {
            Thread.sleep(10000);
            /* Imagine a DB operation taking more time. */
            return "1";
        } else if (ipAddressL == 2) {
            Thread.sleep(20000);
            return "2";
        } else {
            Thread.sleep(30000);
            return "3";
        }
    }
}

我想返回一个空字符串,或者只是终止线程或在每个任务中抛出一个 TimeoutException,如果每个任务单独花费超过 10 秒。

假设第一个线程需要 10 秒,Future.get()将等待 10 秒然后超时。我将捕获异常并继续迭代第二个未来对象。假设此时第二个线程尚未完成(这意味着它运行了 10 秒,而第一个线程运行并且仍在运行),现在第 2 个线程上的Future.get()将再等待 10 秒,以此类推,后续线程总共需要 20 秒,依此类推。

future.get(1000, TimeUnit.MILLISECONDS) (1 sec), 

将确保整个操作的限制为 10 秒,但我需要通过对每个单独的并发任务施加 10 秒的限制来对整个操作进行 10 秒的限制。

使用THREAD_POOL.invokeAll而不是submit等待 10 秒等待任务完成。

如果某些任务在 10 秒之前完成,您可以使用 future.isDone() 进行检查并使用 future.get 检索结果而不会阻止。

相关内容

  • 没有找到相关文章

最新更新