设置一段代码执行的超时时间的简单方法

  • Post author:
  • Post category:其他


前段时间在搞一个批量处理程序,涉及到多线程操作。但是后台服务很不给力,并发一大常常就挂了,长时间不给返回,导致我的程序也挂死在那里……

那么能不能设置一段代码执行的超时时间呢?如果处理超时就忽略该错误继续向下执行。

可是在网上搜了大半天,找到的都是无用的代码,根本不能用。

查了大量资料后发现,java早已经给我们提供了解决方案。jdk1.5自带的并发库中Future类就能满足这个需求。Future类中重要方法包括get()和cancel()。get()获取数据对象,如果数据没有加载,就会阻塞直到取到数据,而 cancel()是取消数据加载。另外一个get(timeout)操作,表示如果在timeout时间内没有取到就失败返回,而不再阻塞。

到这里,此问题就迎刃而解了。废话不多说,直接上代码:

public class ThreadTest {

public static void main(String[] args) throws InterruptedException,
ExecutionException {

final ExecutorService exec = Executors.newFixedThreadPool(1);

Callable<String> call = new Callable<String>() {
public String call() throws Exception {
//开始执行耗时操作
Thread.sleep(1000 * 5);
return "线程执行完成.";
}
};

try {
Future<String> future = exec.submit(call);
String obj = future.get(1000 * 1, TimeUnit.MILLISECONDS); //任务处理超时时间设为 1 秒
System.out.println("任务成功返回:" + obj);
} catch (TimeoutException ex) {
System.out.println("处理超时啦....");
ex.printStackTrace();
} catch (Exception e) {
System.out.println("处理失败.");
e.printStackTrace();
}
// 关闭线程池
exec.shutdown();
}
}

注意,以上代码需要 1.5 以上的 jdk 才可编译。

执行后可以发现抛出了 TimeoutException ,打印出了”处理超时啦….” 的文字,在此处可进行超时后的处理。