1. 场景描述
举个例子,有时候执行一段程序需要1 ~ 10分钟才能运行完,但你最多只能等5分钟。如果5分钟以内程序能够运行完,那么直接返回正确结果;如果过了5分钟程序还没运行完,那么就中止程序的运行,直接返回一个错误结果。
2. 一个工具类(用来执行有时间限制的任务)
package org.demo.task;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
public class TimeoutTaskUtils {
/**
* 执行一个有时间限制的任务
* @param task 待执行的任务
* @param seconds 超时时间(单位: 秒)
* @return
*/
public static Boolean execute(Callable<Boolean> task, int seconds)
{
Boolean result = Boolean.FALSE;
ExecutorService threadPool = Executors.newCachedThreadPool();
try
{
Future<Boolean> future = threadPool.submit(task);
result = future.get(seconds, TimeUnit.SECONDS);
}
catch (Exception e)
{
result = Boolean.FALSE;
e.printStackTrace();
}
finally
{
threadPool.shutdownNow();
}
return result;
}
}
3. 测试
package org.demo.task;
import java.util.concurrent.Callable;
public class Main {
/**
* @param args
*/
public static void main(String[] args) {
boolean result = TimeoutTaskUtils.execute(new MyTimeoutTask(), 5);
System.out.println("result = " + result);
System.out.println("-- finished. --");
}
}
class MyTimeoutTask implements Callable<Boolean> {
@Override
public Boolean call() throws Exception {
for (int i=0; i<10; i++) {
System.out.println("i = " + i);
Thread.sleep(1000);
}
return true;
}
}