前段时间在搞一个批量处理程序,涉及到多线程操作。但是后台服务很不给力,并发一大常常就挂了,长时间不给返回,导致我的程序也挂死在那里……
那么能不能设置一段代码执行的超时时间呢?如果处理超时就忽略该错误继续向下执行。
可是在网上搜了大半天,找到的都是无用的代码,根本不能用。
查了大量资料后发现,java早已经给我们提供了解决方案。jdk1.5自带的并发库中Future类就能满足这个需求。Future类中重要方法包括get()和cancel()。get()获取数据对象,如果数据没有加载,就会阻塞直到取到数据,而 cancel()是取消数据加载。另外一个get(timeout)操作,表示如果在timeout时间内没有取到就失败返回,而不再阻塞。
到这里,此问题就迎刃而解了。废话不多说,直接上代码:
-
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import com.sun.corba.se.impl.orbutil.closure.Future;
import com.sun.corba.se.impl.orbutil.threadpool.TimeoutException;
-
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 ,打印出了"处理超时啦...." 的文字,在此处可进行超时后的处理。