CountDownLatch作用阻塞一個或多個線程等待其他線程完成操作。
定義初始化的時候茫经,需要傳入一個正數(shù)來初始化計數(shù)器(0也可以屿附,但這樣定義沒有實際意義)。有兩個方法countDown()用于遞減計數(shù)器,await()方法阻塞當前線程,直到計數(shù)器遞減為0祟霍。
CountDownLatch通常用于多個線程之間的協(xié)調(diào)工作杏头。
假設有如下情節(jié):
同時獲取5張表的數(shù)據(jù)并一同返回
為了讓cpu更好的得到利用,程序執(zhí)行效率更高沸呐,使用多線程來完成醇王。
public class CountDownLatchExample {
// 循環(huán)次數(shù)
private static final int FOR_NUMBER = 5;
public static void main(String[] args) {
// 查詢數(shù)據(jù)
for (int i = 0; i < FOR_NUMBER; i++) {
new Thread(() -> {
System.out.println("查詢第:\t" + Thread.currentThread().getName() + "張表數(shù)據(jù)完成!");
}, String.valueOf(i)).start();
}
System.out.println("查詢完畢");
}
}
我們看一下執(zhí)行結(jié)果是否是我們想要的結(jié)果。
可以看到崭添,還有數(shù)據(jù)沒查詢完成他就體檢進行查詢完畢的操作了寓娩。那如果在實際開發(fā)過程中,就等于數(shù)據(jù)還沒處理完成就返回用戶數(shù)據(jù)了呼渣。這并不是我們想要的結(jié)果棘伴。
那么剛才也有說CountDownLatch是阻塞一個或多個線程等待其他線程完成操作,那么我們試一下屁置。
public class CountDownLatchExample {
// 循環(huán)五次
private static final int FOR_NUMBER = 5;
// 實例化定義一個CountDownLatch需要減少的總次數(shù)
private static CountDownLatch countDownLatch = new CountDownLatch(5);
public static void main(String[] args) {
// 查詢數(shù)據(jù)
for (int i = 0; i < FOR_NUMBER; i++) {
new Thread(() -> {
System.out.println("查詢第:\t" + Thread.currentThread().getName() + "張表數(shù)據(jù)完成!");
// 執(zhí)行完 查詢 然后進行遞減操作 每次減1
countDownLatch.countDown();
}, String.valueOf(i)).start();
}
try {
countDownLatch.await(); // await()方法起到阻塞的作用焊夸,直到計數(shù)器值等于0
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("查詢完畢");
}
}
上邊我們定義了一個CountDownLatch里邊值寫入的是5表明我們總共需要減少的次數(shù),再每次執(zhí)行完查詢操作完成后進行減1操作蓝角,最后在執(zhí)行完畢之前進行await()方法阻塞阱穗,計數(shù)器為0則才放過往下走。我們看一下就結(jié)果是否跟想象的一樣使鹅。
OKK!非常完美的達到了預期的效果揪阶。
總體CountDownLatch的作用以及使用就沒有了,很簡單患朱,大家可以多多嘗試鲁僚。在開發(fā)中用到非常多。
補充:
CountDownLatch的await()有重載方法await(long timeout, TimeUnit unit), timeout則是設置最大等待時間,unit則是時間類型蕴茴,如果超過這個時間程序則將繼續(xù)執(zhí)行劝评,不再阻塞。