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
检索结果而不会阻止。