天天看點

SpringBoot開發案例之CountDownLatch多任務并行處理

SpringBoot開發案例之CountDownLatch多任務并行處理

前言

最近在做一個原始成績統計的功能,使用者通過前台設定相關參數,背景實時統計并傳回資料。相對來說統計功能點還是比較多的,這裡大體羅列一下。

  • 個人排名
  • 本次測試的優良線、及格線、低分線
  • 各個班級的排名人數(1-25、26-50 類比等等)
  • 各個班級的前X名人數統計(前10、前20 類比等等)
  • 各個班級的分數段學生人數統計(150-140、139-130 類比等等)

最好的使用者體驗,就是每一個操作都可以實時的展示資料,3秒之内應該是使用者的忍受範圍之内的了,是以做一款産品不僅要考慮使用者互動設計,後端的優化也是比不可少的。

大家可以簡單的看下以上這5項統計資料,總體來說,統計量還是不少的。最主要的還是要實時、實時、實時(重要的事情說三遍),顯然定時任務是不現實的。

改造前

程式邏輯

SpringBoot開發案例之CountDownLatch多任務并行處理

改造後

SpringBoot開發案例之CountDownLatch多任務并行處理

代碼實作

StatsDemo僞代碼:

/**      * 多任務并行統計      * 建立者 科幫網      * 建立時間    2018年4月16日      */     public class StatsDemo {         final static SimpleDateFormat sdf = new SimpleDateFormat(                 "yyyy-MM-dd HH:mm:ss");         final static String startTime = sdf.format(new Date());         public static void main(String[] args) throws InterruptedException {             CountDownLatch latch = new CountDownLatch(5);// 兩個賽跑者             Stats stats1 = new Stats("任務A", 1000, latch);             Stats stats2 = new Stats("任務B", 2000, latch);             Stats stats3 = new Stats("任務C", 2000, latch);             Stats stats4 = new Stats("任務D", 2000, latch);             Stats stats5 = new Stats("任務E", 2000, latch);             stats1.start();//任務A開始執行             stats2.start();//任務B開始執行             stats3.start();//任務C開始執行             stats4.start();//任務D開始執行             stats5.start();//任務E開始執行             latch.await();// 等待所有人任務結束             System.out.println("所有的統計任務執行完成:" + sdf.format(new Date()));         }         static class Stats extends Thread {             String statsName;             int runTime;             CountDownLatch latch;             public Stats(String statsName, int runTime, CountDownLatch latch) {                 this.statsName = statsName;                 this.runTime = runTime;                 this.latch = latch;             }             public void run() {                 try {                     System.out.println(statsName+ " do stats begin at "+ startTime);                     //模拟任務執行時間                     Thread.sleep(runTime);                     System.out.println(statsName + " do stats complete at "+ sdf.format(new Date()));                     latch.countDown();//單次任務結束,計數器減一                 } catch (InterruptedException e) {                     e.printStackTrace();                 }             }         }     }           

由于要同步傳回統計資料,這裡我們使用到了CountDownLatch類,它是Java5中新增的一個并發工具類,其使用非常簡單,參考上面的僞代碼給出了詳細的使用步驟。

CountDownLatch用于同步一個或多個任務,強制他們等待由其他任務執行的一組操作完成。CountDownLatch典型的用法是将一個程式分為N個互相獨立的可解決任務,并建立值為N的CountDownLatch。當每一個任務完成時,都會在這個鎖存器上調用countDown,等待問題被解決的任務調用這個鎖存器的await,将他們自己攔住,直至鎖存器計數結束。

具體的源碼解讀,大家可以參考:

【JUC】JDK1.8源碼分析之CountDownLatch

項目源碼:

https://gitee.com/52itstyle/spring-data-jpa

作者: 小柒2012

歡迎關注:

https://blog.52itstyle.com