SpringBoot CountDownLatch多任務並行處理的實現方法

前言

最近在做一個原始成績統計的功能，用戶通過前台設置相關參數，後台實時統計並返回數據。相對來說統計功能點還是比較多的，這里大體羅列一下。

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

最好的用戶體驗，就是每一個操作都可以實時的展示數據，3秒之內應該是用戶的忍受範圍之內的了，所以做一款產品不僅要考慮用戶交互設計，後端的優化也是比不可少的。

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

改造前

程序邏輯

順序執行任務.png

改造後

程序邏輯

多任務並行處理.png

多任務並行處理，適用於多核CPU，單核CPU多線程執行任務可能會適得其反(上下文切換以及線程的創建和銷毀都會消耗資源)，特別是CPU密集型的任務。

代碼實現

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，將他們自己攔住，直至鎖存器計數結束。

具體的源碼解讀，大家可以參考： 源碼分析之CountDownLatch

項目源碼：https://gitee.com/52itstyle/spring-data-jpa

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持武林網。