分析.NET中的CountDownLatch類
最近筆者在用.Net寫程序時(shí)遇到一個(gè)問(wèn)題:有N個(gè)互不相關(guān)的任務(wù)要在線程池中跑,但有一個(gè)線程要等待N個(gè)任務(wù)完成之后才能繼續(xù)。而這個(gè)N是個(gè)未知數(shù),可能會(huì) 很大(因此才會(huì)想到使用線程池而不是手動(dòng)去new一個(gè)therad)。翻了翻.Net類庫(kù)的文檔,發(fā)現(xiàn)一個(gè)叫WaitHandle的類。這個(gè)類的用法挺有 意思,需要為每個(gè)線程創(chuàng)建一個(gè)WaitHandle對(duì)象并把它們放在一個(gè)數(shù)組中,然后用WaitHandle類中的WaitAll方法來(lái)等待這些 WaitHandle被調(diào)用Set方法。(代碼就不寫了,可以參考MSDN http://msdn.microsoft.com/zh-cn/library /system.threading.waithandle.aspx)
雖然覺得這有點(diǎn)復(fù)雜,但還是試了試。當(dāng)程序運(yùn)行時(shí),碰到了一個(gè)問(wèn) 題,如果WaitHandle數(shù)組超過(guò)64個(gè)元素之后,WaitHandle對(duì)象的WatiAll方法罷工了。后來(lái)為了程序能運(yùn)行,只得想了一個(gè)笨辦法: 先創(chuàng)建兩個(gè)WaitHandle對(duì)象放在數(shù)組,然后用循環(huán)兩個(gè)兩個(gè)地運(yùn)行任務(wù)。代碼的思路大概是下面這樣:
WaitHandle[] handles = new WaitHandle[]{ |
雖然代碼這么寫比較復(fù)雜,但至少可以保證運(yùn)行時(shí)不會(huì)出問(wèn)題。但這么寫代碼顯然并不KISS!于是問(wèn)了問(wèn)高手,說(shuō)有個(gè) RegisterWaitForSingleObject方法,但一看這個(gè)方法的參數(shù)列表就夠讓人暈的了。有點(diǎn)懷念Java了,記得Java中有個(gè) CountDownLatch類,創(chuàng)建類的時(shí)候賦一個(gè)初始值X,然后主線程中調(diào)用await,線程池中跑的線程調(diào)用countDown方法。就可以實(shí)現(xiàn)主 線程等待X次countDown方法調(diào)用之后繼續(xù)。這樣既沒有64個(gè)WaitHandle的限制,也不用去研究那個(gè) RegisterWaitForSingleObject方法。不過(guò)問(wèn)題在于.Net中并沒有這么一個(gè)東西,只能自己動(dòng)手了。
class CountDownLatch { |
有了這個(gè)東西,上面的代碼可以改的更少一些。
|
而對(duì)于任務(wù)的代碼來(lái)說(shuō),在結(jié)尾處吧對(duì)WaitHandle的Set方法的調(diào)用改為對(duì)CountDownLatch類的CountDown方法的調(diào)用即可。
最后,我想說(shuō)的是其實(shí)沒必要把思路都拘束在.Net上或者Java上,相互借鑒會(huì)讓思路更開闊一些。不過(guò)有句心里話想說(shuō)的就是,其實(shí)Java的類庫(kù)在某些方面做的比.Net好一些。
【編輯推薦】