FutureTask可用於異步獲取執行結果或取消執行任務的場景。通過傳入Runnable或者Callable的任務給FutureTask,直接調用其run方法或者放入線程池執行,之后可以在外部通過FutureTask的get方法異步獲取執行結果,因此,FutureTask非常適合用於耗時的計算,主線程可以在完成自己的任務后,再去獲取結果。另外,FutureTask還可以確保即使調用了多次run方法,它都只會執行一次Runnable或者Callable任務,或者通過cancel取消FutureTask的執行等。
1. FutureTask執行多任務計算的使用場景
利用FutureTask和ExecutorService,可以用多線程的方式提交計算任務,主線程繼續執行其他任務,當主線程需要子線程的計算結果時,在異步獲取子線程的執行結果。
1 package futuretask; 2 3 import java.util.ArrayList; 4 import java.util.List; 5 import java.util.concurrent.Callable; 6 import java.util.concurrent.ExecutionException; 7 import java.util.concurrent.ExecutorService; 8 import java.util.concurrent.Executors; 9 import java.util.concurrent.FutureTask; 10 11 public class FutureTaskForMultiCompute { 12 13 public static void main(String[] args) { 14 15 FutureTaskForMultiCompute inst=new FutureTaskForMultiCompute(); 16 // 創建任務集合 17 List<FutureTask<Integer>> taskList = new ArrayList<FutureTask<Integer>>(); 18 // 創建線程池 19 ExecutorService exec = Executors.newFixedThreadPool(5); 20 for (int i = 0; i < 10; i++) { 21 // 傳入Callable對象創建FutureTask對象 22 FutureTask<Integer> ft = new FutureTask<Integer>(inst.new ComputeTask(i, ""+i)); 23 taskList.add(ft); 24 // 提交給線程池執行任務,也可以通過exec.invokeAll(taskList)一次性提交所有任務; 25 exec.submit(ft); 26 } 27 28 System.out.println("所有計算任務提交完畢, 主線程接着干其他事情!"); 29 30 // 開始統計各計算線程計算結果 31 Integer totalResult = 0; 32 for (FutureTask<Integer> ft : taskList) { 33 try { 34 //FutureTask的get方法會自動阻塞,直到獲取計算結果為止 35 totalResult = totalResult + ft.get(); 36 } catch (InterruptedException e) { 37 e.printStackTrace(); 38 } catch (ExecutionException e) { 39 e.printStackTrace(); 40 } 41 } 42 43 // 關閉線程池 44 exec.shutdown(); 45 System.out.println("多任務計算后的總結果是:" + totalResult); 46 47 } 48 49 private class ComputeTask implements Callable<Integer> { 50 51 private Integer result = 0; 52 private String taskName = ""; 53 54 public ComputeTask(Integer iniResult, String taskName){ 55 result = iniResult; 56 this.taskName = taskName; 57 System.out.println("生成子線程計算任務: "+taskName); 58 } 59 60 public String getTaskName(){ 61 return this.taskName; 62 } 63 64 @Override 65 public Integer call() throws Exception { 66 // TODO Auto-generated method stub 67 68 for (int i = 0; i < 100; i++) { 69 result =+ i; 70 } 71 // 休眠5秒鍾,觀察主線程行為,預期的結果是主線程會繼續執行,到要取得FutureTask的結果是等待直至完成。 72 Thread.sleep(5000); 73 System.out.println("子線程計算任務: "+taskName+" 執行完成!"); 74 return result; 75 } 76 } 77 }
2. FutureTask在高並發環境下確保任務只執行一次
在很多高並發的環境下,往往我們只需要某些任務只執行一次。這種使用情景FutureTask的特性恰能勝任。舉一個例子,假設有一個帶key的連接池,當key存在時,即直接返回key對應的對象;當key不存在時,則創建連接。對於這樣的應用場景,通常采用的方法為使用一個Map對象來存儲key和連接池對應的對應關系,典型的代碼如下面所示:
1 private Map<String, Connection> connectionPool = new HashMap<String, Connection>(); 2 private ReentrantLock lock = new ReentrantLock(); 3 4 public Connection getConnection(String key){ 5 try{ 6 lock.lock(); 7 if(connectionPool.containsKey(key)){ 8 return connectionPool.get(key); 9 } 10 else{ 11 //創建 Connection 12 Connection conn = createConnection(); 13 connectionPool.put(key, conn); 14 return conn; 15 } 16 } 17 finally{ 18 lock.unlock(); 19 } 20 } 21 22 //創建Connection 23 private Connection createConnection(){ 24 return null; 25 }
在上面的例子中,我們通過加鎖確保高並發環境下的線程安全,也確保了connection只創建一次,然而確犧牲了性能。改用ConcurrentHash的情況下,幾乎可以避免加鎖的操作,性能大大提高,但是在高並發的情況下有可能出現Connection被創建多次的現象。這時最需要解決的問題就是當key不存在時,創建Connection的動作能放在connectionPool之后執行,這正是FutureTask發揮作用的時機,基於ConcurrentHashMap和FutureTask的改造代碼如下:
1 private ConcurrentHashMap<String,FutureTask<Connection>>connectionPool = new ConcurrentHashMap<String, FutureTask<Connection>>(); 2 3 public Connection getConnection(String key) throws Exception{ 4 FutureTask<Connection>connectionTask=connectionPool.get(key); 5 if(connectionTask!=null){ 6 return connectionTask.get(); 7 } 8 else{ 9 Callable<Connection> callable = new Callable<Connection>(){ 10 @Override 11 public Connection call() throws Exception { 12 // TODO Auto-generated method stub 13 return createConnection(); 14 } 15 }; 16 FutureTask<Connection>newTask = new FutureTask<Connection>(callable); 17 connectionTask = connectionPool.putIfAbsent(key, newTask); 18 if(connectionTask==null){ 19 connectionTask = newTask; 20 connectionTask.run(); 21 } 22 return connectionTask.get(); 23 } 24 } 25 26 //創建Connection 27 private Connection createConnection(){ 28 return null; 29 }