java设置一段代码执行超时时间

叁歲伎倆 2022-08-06 10:25 305阅读 0赞

前段时间在搞一个批量处理程序，涉及到多线程操作。但是后台服务很不给力，并发一大常常就挂了，长时间不给返回，导致我的程序也挂死在那里……   
  
那么能不能设置一段代码执行的超时时间呢？如果处理超时就忽略该错误继续向下执行。   
  
可是在网上搜了大半天，找到的都是无用的代码，根本不能用。   
  
查了大量资料后发现，java早已经给我们提供了解决方案。jdk1.5自带的并发库中Future类就能满足这个需求。Future类中重要方法包括get()和cancel()。get()获取数据对象，如果数据没有加载，就会阻塞直到取到数据，而 cancel()是取消数据加载。另外一个get(timeout)操作，表示如果在timeout时间内没有取到就失败返回，而不再阻塞。

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

1.  **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;**  
    
2.  **public** **class** ThreadTest \{  
3.    
4.      **public** **static** **void** main(String\[\] args) **throws** InterruptedException,  
5.              ExecutionException \{  
6.            
7.          **final** ExecutorService exec = Executors.newFixedThreadPool(1);  
8.            
9.          Callable<String> call = **new** Callable<String>() \{  
10.             **public** String call() **throws** Exception \{  
11.                 //开始执行耗时操作  
12.                 Thread.sleep(1000 \* 5);  
13.                 **return** "线程执行完成.";  
14.             \}  
15.         \};  
16.           
17.         **try** \{  
18.             Future<String> future = exec.submit(call);  
19.             String obj = future.get(1000 \* 1, TimeUnit.MILLISECONDS); //任务处理超时时间设为 1 秒  
20.             System.out.println("任务成功返回:" + obj);  
21.         \} **catch** (TimeoutException ex) \{  
22.             System.out.println("处理超时啦....");  
23.             ex.printStackTrace();  
24.         \} **catch** (Exception e) \{  
25.             System.out.println("处理失败.");  
26.             e.printStackTrace();  
27.         \}  
28.         // 关闭线程池  
29.         exec.shutdown();  
30.     \}  
31. \}

注意，以上代码需要 1.5 以上的 jdk 才可编译。   
执行后可以发现抛出了 TimeoutException ，打印出了"处理超时啦...." 的文字，在此处可进行超时后的处理。