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