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

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

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

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

查了大量资料后发现，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 ，打印出了"处理超时啦...." 的文字，在此处可进行超时后的处理。

评论

2 楼 380071587 2013-08-05  

学习了，太好了

1 楼 宋建勇 2012-07-23  

nice