ThreadPoolExecutor + CountDownLatch 實際應用補充

注:(本例用submit實現會更簡單按傅,可以參見ThreadPoolExecutor execute 和 submit

昨日上線了我的并發(fā)編程改造后的代碼,速率提升十分明顯服协,原本需要運行將近30小時的任務縮短到十分之一(我開啟了最大十個并發(fā))绍昂,不過,雖然達到了提升速率的目的偿荷,但是結合日志我還是發(fā)現了兩個潛在Bug窘游。上一篇傳送門:ThreadPoolExecutor + CountDownLatch 實際應用

首先,我在線程池中捕獲了異常跳纳,并且記錄日志忍饰。

try {
    // 模擬耗時
    Thread.sleep(Long.valueOf(String.valueOf(new Double(Math.random() * 1000).intValue())));
} catch (InterruptedException e) {
    // 實際應用在此記錄日志
    log.error("ThreadPoolExecutor error", e);
}

接著我在日志中發(fā)現的確有報錯,這種情況下寺庄,我應該判斷任務運行結果為失敗艾蓝,然而數據庫中最終任務運行狀態(tài)是成功力崇。再一看代碼就明白了,因為我只判斷了線程池中子線程的運行狀態(tài)是否完成赢织,但是沒有判斷邏輯結果是不是對的亮靴,再加上線程內異常被吞了,最后任務的狀態(tài)肯定是成功的于置,哪怕所有線程都報錯茧吊,結果也能輸出,只不過是錯的八毯。解決方法如下:

1搓侄、創(chuàng)建內部類,作為狀態(tài)標識變量(還可以用其他引用類型變量代替)
static class Flag {
    private boolean f;

    public Flag() {
        f = true;// 初始設為true
    }

    public boolean isF() {
        return f;
    }

    public void setF(boolean f) {
        this.f = f;
    }
}
2话速、初始化Flag
final Flag flag = new Flag();
3休讳、在異常捕獲中設置為false
...
catch (Exception e) {
    // 如果子線程報錯,狀態(tài)標識為false
    flag.setF(false);
}
4尿孔、最后判斷狀態(tài)是否為true
if (flag.isF()) {
    // 打印計數
    System.out.println("結束:" + totalRows.get());
} else {
    System.out.println("有子線程報錯,結果不準確");
}

另外筹麸,我還發(fā)現了一個潛在Bug活合。我的任務核心部分是在while 循環(huán)內請求第三方api下載數據,如果第三方返回結果為空物赶,手動break退出循環(huán)白指,但是未在catch中退出。這樣就有問題了酵紫,一旦循環(huán)體內拋出運行時異常告嘲,代碼未執(zhí)行到判斷是否break時就被異常了,但是catch未執(zhí)行退出奖地,將可能導致死循環(huán)橄唬。好比示例代碼中:

while (true) {
    // 此處發(fā)送請求拉取數據,一旦報錯参歹,將跳過下面的break 判斷仰楚,而catch中未退出,將導致死循環(huán)
    String result = download();
    // 判斷是否break
    if (result  == null) {
      break;
    }
} catch (Exception e) {
      e.printStackTrace();
      // 未執(zhí)行退出
    }
}
修改后
boolean loop = true;
while (loop) {
    String result = download();
    // 判斷是否break
    if (result  == null) {
      loop = false;
    }
} catch (Exception e) {
      // 退出循環(huán)
      loop = false;
    }
}
完整示例(while循環(huán)內用次數模擬和真實業(yè)務邏輯還是有區(qū)別的)
package com.yzy.test;

import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

public class Main1 {
    static class Flag {
        private boolean f;

        public Flag() {
            f = true;
        }

        public boolean isF() {
            return f;
        }

        public void setF(boolean f) {
            this.f = f;
        }
    }

    public static void main(String[] args) throws InterruptedException {

        // 線程安全的計數器
        AtomicInteger totalRows = new AtomicInteger(0);

        // 創(chuàng)建線程池犬庇,其中核心線程10僧界,也是我期望的最大并發(fā)數,最大線程數和隊列大小都為30臭挽,即我的總任務數
        ThreadPoolExecutor executor = new ThreadPoolExecutor(10, 30, 60L, TimeUnit.SECONDS, new ArrayBlockingQueue<>(30));

        // 初始化CountDownLatch捂襟,大小為30
        CountDownLatch countDownLatch = new CountDownLatch(30);

        // 記錄狀態(tài)
        final Flag flag = new Flag();

        // 模擬遍歷參數集合
        for (int i = 0; i < 30; i++) {
            // 往線程池提交任務
            executor.execute(new Runnable() {
                @Override
                public void run() {
                    int times = 0;
                    boolean loop = true;
                    // 模擬數據拉取過程可能需要分頁
                    while (loop) {
                        // 模擬每個任務需要分頁5次
                        if (times >= 5) {
                            break;
                        }
                        times++;

                        // 模擬計數
                        totalRows.incrementAndGet();
                        try {
                            // 模擬耗時
                            Thread.sleep(Long.valueOf(String.valueOf(new Double(Math.random() * 1000).intValue())));
                        } catch (Exception e) {
                            // 如果子線程報錯,退出循環(huán)
                            loop = false;
                            // 如果子線程報錯欢峰,狀態(tài)標識為false
                            flag.setF(false);
                        }
                    }
                    // 子線程完成葬荷,countDownLatch執(zhí)行countDown
                    countDownLatch.countDown();
                }
            });
            // 打印線程池運行狀態(tài)
            System.out.println("線程池中線程數目:" + executor.getPoolSize() + "涨共,隊列中等待執(zhí)行的任務數目:" +
                    executor.getQueue().size() + ",已執(zhí)行結束的任務數目:" + executor.getCompletedTaskCount());
        }
        // 標記多線程關閉闯狱,但不會立馬關閉
        executor.shutdown();

        // 阻塞當前線程煞赢,知道所有子線程都執(zhí)行countDown方法才會繼續(xù)執(zhí)行
        countDownLatch.await();

        // 打印線程池運行狀態(tài)
        System.out.println("線程池中線程數目:" + executor.getPoolSize() + ",隊列中等待執(zhí)行的任務數目:" +
                executor.getQueue().size() + "哄孤,已執(zhí)行結束的任務數目:" + executor.getCompletedTaskCount());

        if (flag.isF()) {
            // 打印計數
            System.out.println("結束:" + totalRows.get());
        } else {
            System.out.println("有子線程報錯照筑,結果不準確");
        }
    }
}
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市瘦陈,隨后出現的幾起案子凝危,更是在濱河造成了極大的恐慌,老刑警劉巖晨逝,帶你破解...
    沈念sama閱讀 211,376評論 6 491
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件蛾默,死亡現場離奇詭異,居然都是意外死亡捉貌,警方通過查閱死者的電腦和手機支鸡,發(fā)現死者居然都...
    沈念sama閱讀 90,126評論 2 385
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來趁窃,“玉大人牧挣,你說我怎么就攤上這事⌒崖剑” “怎么了瀑构?”我有些...
    開封第一講書人閱讀 156,966評論 0 347
  • 文/不壞的土叔 我叫張陵,是天一觀的道長刨摩。 經常有香客問我寺晌,道長,這世上最難降的妖魔是什么澡刹? 我笑而不...
    開封第一講書人閱讀 56,432評論 1 283
  • 正文 為了忘掉前任呻征,我火速辦了婚禮,結果婚禮上罢浇,老公的妹妹穿的比我還像新娘怕犁。我一直安慰自己,他們只是感情好己莺,可當我...
    茶點故事閱讀 65,519評論 6 385
  • 文/花漫 我一把揭開白布奏甫。 她就那樣靜靜地躺著,像睡著了一般凌受。 火紅的嫁衣襯著肌膚如雪阵子。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,792評論 1 290
  • 那天胜蛉,我揣著相機與錄音挠进,去河邊找鬼色乾。 笑死,一個胖子當著我的面吹牛领突,可吹牛的內容都是我干的暖璧。 我是一名探鬼主播,決...
    沈念sama閱讀 38,933評論 3 406
  • 文/蒼蘭香墨 我猛地睜開眼君旦,長吁一口氣:“原來是場噩夢啊……” “哼澎办!你這毒婦竟也來了?” 一聲冷哼從身側響起金砍,我...
    開封第一講書人閱讀 37,701評論 0 266
  • 序言:老撾萬榮一對情侶失蹤局蚀,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后恕稠,有當地人在樹林里發(fā)現了一具尸體琅绅,經...
    沈念sama閱讀 44,143評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 36,488評論 2 327
  • 正文 我和宋清朗相戀三年鹅巍,在試婚紗的時候發(fā)現自己被綠了千扶。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 38,626評論 1 340
  • 序言:一個原本活蹦亂跳的男人離奇死亡骆捧,死狀恐怖澎羞,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情凑懂,我是刑警寧澤,帶...
    沈念sama閱讀 34,292評論 4 329
  • 正文 年R本政府宣布梧宫,位于F島的核電站接谨,受9級特大地震影響,放射性物質發(fā)生泄漏塘匣。R本人自食惡果不足惜脓豪,卻給世界環(huán)境...
    茶點故事閱讀 39,896評論 3 313
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望忌卤。 院中可真熱鬧扫夜,春花似錦、人聲如沸驰徊。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,742評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽棍厂。三九已至颗味,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間牺弹,已是汗流浹背浦马。 一陣腳步聲響...
    開封第一講書人閱讀 31,977評論 1 265
  • 我被黑心中介騙來泰國打工时呀, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人晶默。 一個月前我還...
    沈念sama閱讀 46,324評論 2 360
  • 正文 我出身青樓谨娜,卻偏偏與公主長得像,于是被迫代替她去往敵國和親磺陡。 傳聞我的和親對象是個殘疾皇子趴梢,可洞房花燭夜當晚...
    茶點故事閱讀 43,494評論 2 348