1. 程式人生 > >java設定一段程式碼執行的超時時間的簡單方法 間接實現獲取Connection超時問題

java設定一段程式碼執行的超時時間的簡單方法 間接實現獲取Connection超時問題

最近有個需求, 當DB壓力過大時獲取Connction的時間過慢長時間不返回的話, 就不連線DB了, 研究了好久,DataSource裡面的setLoginTimeOut 根本沒法用, 剛開始一直糾結在大google搜尋"java get connection 超時"答案上, 但始終找不到答案, 偶然嘗試了下"java 設定超時" 問題就迎刃而解了. 

java早已經給我們提供瞭解決方案。jdk1.5自帶的併發庫中Future類就能滿足這個需求。Future類中重要方法包括get()和cancel()。get()獲取資料物件,如果資料沒有載入,就會阻塞直到取到資料,而 cancel()是取消資料載入。另外一個get(timeout)操作,表示如果在timeout時間內沒有取到就失敗返回,而不再阻塞。 
看來不能一直糾結在一條道上, 偶爾換個思路還是很有幫助的,  不多說了, 解決方案如下 

我的code: 
Java程式碼  
  1. public boolean checkDBStatus() {  
  2.     boolean bdStatus = false
    ;  
  3.     final ExecutorService exec = Executors.newFixedThreadPool(1);  
  4.     Callable<String> call = new Callable<String>() {  
  5.         public String call() throws Exception {  
  6.             DataSource dataSource = getJdbcTemplate().getDataSource();  
  7.             Connection connection = dataSource.getConnection();  
  8.             Statement statement = connection.createStatement();  
  9.             statement.executeQuery("select * from citirisk_menu_node");  
  10.             return "true";  
  11.         }  
  12.     };  
  13.     try {  
  14.         Future<String> future = exec.submit(call);  
  15.         // set db connection timeout to 10 seconds
      
  16.         String obj = future.get(1000 * 10, TimeUnit.MILLISECONDS);   
  17.         bdStatus = Boolean.parseBoolean(obj);  
  18.         System.out.println("the return value from call is :" + obj);  
  19.     } catch (TimeoutException ex) {  
  20.         System.out.println("====================task time out===============");  
  21.         ex.printStackTrace();  
  22.         bdStatus = false;  
  23.     } catch (Exception e) {  
  24.         System.out.println("failed to handle.");  
  25.         e.printStackTrace();  
  26.         bdStatus = false;  
  27.     }  
  28.     // close thread pool  
  29.     exec.shutdown();  
  30.     return bdStatus;  
  31. }  


在Java中,如果需要設定程式碼執行的最長時間,即超時,可以用Java執行緒池ExecutorService類配合Future介面來實現。 Future介面是Java標準API的一部分,在java.util.concurrent包中。Future介面是Java執行緒Future模式的實現,可以來進行非同步計算。 

Future模式可以這樣來描述:我有一個任務,提交給了Future,Future替我完成這個任務。期間我自己可以去做任何想做的事情。一段時間之後,我就便可以從Future那兒取出結果。就相當於下了一張訂貨單,一段時間後可以拿著提訂單來提貨,這期間可以幹別的任何事情。其中Future 介面就是訂貨單,真正處理訂單的是Executor類,它根據Future介面的要求來生產產品。 

Future介面提供方法來檢測任務是否被執行完,等待任務執行完獲得結果,也可以設定任務執行的超時時間。這個設定超時的方法就是實現Java程式執行超時的關鍵。 

Future介面是一個泛型介面,嚴格的格式應該是Future<V>,其中V代表了Future執行的任務返回值的型別。 Future介面的方法介紹如下: 

    boolean cancel (boolean mayInterruptIfRunning) 取消任務的執行。引數指定是否立即中斷任務執行,或者等等任務結束 
    boolean isCancelled () 任務是否已經取消,任務正常完成前將其取消,則返回 true 
    boolean isDone () 任務是否已經完成。需要注意的是如果任務正常終止、異常或取消,都將返回true 
    V get () throws InterruptedException, ExecutionException  等待任務執行結束,然後獲得V型別的結果。InterruptedException 執行緒被中斷異常, ExecutionException任務執行異常,如果任務被取消,還會丟擲CancellationException 
    V get (long timeout, TimeUnit unit) throws InterruptedException, ExecutionException, TimeoutException 同上面的get功能一樣,多了設定超時時間。引數timeout指定超時時間,uint指定時間的單位,在列舉類TimeUnit中有相關的定義。如果計算超時,將丟擲TimeoutException 

Future的實現類有java.util.concurrent.FutureTask<V>即 javax.swing.SwingWorker<T,V>。通常使用FutureTask來處理我們的任務。FutureTask類同時又實現了Runnable介面,所以可以直接提交給Executor執行。使用FutureTask實現超時執行的程式碼如下: 

Java程式碼  
  1. ExecutorService executor = Executors.newSingleThreadExecutor();    
  2. FutureTask<String> future =    
  3.        new FutureTask<String>(new Callable<String>() {//使用Callable介面作為構造引數    
  4.          public String call() {    
  5.            //真正的任務在這裡執行,這裡的返回值型別為String,可以為任意型別    
  6.        }});    
  7. executor.execute(future);    
  8. //在這裡可以做別的任何事情    
  9. try {    
  10.     result = future.get(5000, TimeUnit.MILLISECONDS); //取得結果,同時設定超時執行時間為5秒。同樣可以用future.get(),不設定執行超時時間取得結果    
  11. catch (InterruptedException e) {    
  12.     futureTask.cancel(true);    
  13. catch (ExecutionException e) {    
  14.     futureTask.cancel(true);    
  15. catch (TimeoutException e) {    
  16.     futureTask.cancel(true);    
  17. finally {    
  18.     executor.shutdown();    
  19. }    



不直接構造Future物件,也可以使用ExecutorService.submit方法來獲得Future物件,submit方法即支援以 Callable介面型別,也支援Runnable介面作為引數,具有很大的靈活性。使用示例如下: 
Java程式碼  
  1. ExecutorService executor = Executors.newSingleThreadExecutor();    
  2. FutureTask<String> future = executor.submit(    
  3.    new Callable<String>() {//使用Callable介面作為構造引數    
  4.        public String call() {    
  5.       //真正的任務在這裡執行,這裡的返回值型別為String,可以為任意型別    
  6.    }});    
  7. //在這裡可以做別的任何事情    
  8. //同上面取得結果的程式碼    

相關推薦

no