最近在看一個同事的代碼构回,代碼的本意是在main方法中開啟10個線程厅贪,用這10個線程來處理一批業(yè)務邏輯羞福,在某一時刻當你命令console退出的時候幻碱,這個
時候不是立即讓console退出答渔,而是需要等待10個線程把檢測狀態(tài)之后的業(yè)務邏輯執(zhí)行完之后再退出关带,這樣做是有道理的,如果強行退出會有可能造成子線程的業(yè)
務數(shù)據(jù)損壞沼撕,沒毛病吧宋雏,業(yè)務邏輯大概就是這樣。
一:現(xiàn)實場景
由于真實場景的代碼比較復雜和繁瑣务豺,為了方便演示磨总,我將同事所寫的代碼抽象一下,類似下面這樣笼沥,看好了咯~~~
1classProgram2{3privatestaticintworkThreadNums =0;45privatestaticboolisStop =false;67staticvoidMain(string[] args)8{9vartasks =newTask[10];101112for(inti =0; i <10; i++)13{14tasks[i] = Task.Factory.StartNew((obj) =>15{16Run();17}, i);18}1920//是否退出21stringinput =Console.ReadLine();2223while("Y".Equals(input, StringComparison.OrdinalIgnoreCase))24{25break;26}2728isStop =true;2930while(workThreadNums !=0)31{32Console.WriteLine("正在等待線程結束蚪燕,當前還在運行線程有:{0}", workThreadNums);3334Thread.Sleep(10);35}36Console.WriteLine("準備退出了。奔浅。馆纳。");37Console.Read();38Environment.Exit(0);39}4041staticvoidRun()42{43try44{45workThreadNums++;4647while(true)48{49if(isStop)break;5051Thread.Sleep(1000);5253//執(zhí)行業(yè)務邏輯54Console.WriteLine("我是線程:{0},正在執(zhí)行業(yè)務邏輯", Thread.CurrentThread.ManagedThreadId);55}56}57finally58{59workThreadNums--;60}61}62}
其實掃一下上面的代碼應該就知道是用來干嘛的汹桦,業(yè)務邏輯沒毛病鲁驶,基本可以實現(xiàn)剛才的業(yè)務場景,在console退出的時候可以完全確保10個線程都把自己的業(yè)
務邏輯處理完畢了舞骆。不過從美觀角度上來看钥弯,這種代碼就太low了。督禽。脆霎。一點檔次都沒有,比如存在下面兩點問題:
第一點:局部變量太多赂蠢,又是isStop又是workThreaNums绪穆,導致業(yè)務邏輯Run方法中摻雜了很多的非業(yè)務邏輯辨泳,可讀性和維護性都比較low虱岂。
第二點:main函數(shù)在退出的時候用while檢測workThreadNums是否為“0”玖院,貌似沒問題,但仔細想想這段代碼有必要嗎第岖?
接下來我把代碼跑一下难菌,可以看到這個while檢測到了在退出時的workThredNums的中間狀態(tài)“7”,有點意思吧~~~
二:代碼優(yōu)化
那上面這段代碼怎么優(yōu)化呢蔑滓?如何踢掉業(yè)務邏輯方法中的非業(yè)務代碼呢郊酒?當然應該從業(yè)務邏輯上考慮一下了,其實這個問題的核心就是兩點:
1. 如何實現(xiàn)多線程中的協(xié)作取消键袱?
2. 如何實現(xiàn)多線程整體執(zhí)行完畢通知主線程燎窘?
這種場景優(yōu)化千萬不要受到前人寫的代碼所影響,最好忘掉就更好了蹄咖,不然你會下意識的受到什么workthreadnums褐健,isstop這些變量的左右,不說廢話了澜汤,如
果你對task并發(fā)模型很熟悉的話蚜迅,你的優(yōu)化方案很快就會出來的。俊抵。谁不。
1. 協(xié)作取消:
直接用一個bool變量來判斷子線程是否退出的辦法其實是很沒有檔次的,在net 4.0中有一個類(CancellationTokenSource)專門來解決使用bool變量來判
斷的這種很low的場景徽诲,而且比bool變量具有更強大的功能刹帕,這個會在以后的文章中跟大家去講。
2. 多線程整體執(zhí)行完畢通知主線程
目前我們看到的方式是主線程通過輪詢workthreadnums這種沒有檔次的方式去做的谎替,其實這種方式本質上就是任務串行轩拨,而如果你明白task的話,你就知道
有很多的手段是執(zhí)行任務串行的院喜,比如什么ContinueWith亡蓉,WhenAll,WhenAny等等方式喷舀,所以你只需要將一組task串聯(lián)到WhenAll之后就可以了砍濒。好了,上
面就是我的解決思路硫麻,接下來看一下代碼吧:
1classProgram2{3staticvoidMain(string[] args)4{5CancellationTokenSource source =newCancellationTokenSource();67vartasks =newTask[10];89for(inti =0; i <10; i++)10{11tasks[i] = Task.Factory.StartNew((m) =>12{13Run(source.Token);14}, i);15}1617Task.WhenAll(tasks).ContinueWith((t) =>18{19Console.WriteLine("準備退出了爸邢。。拿愧。");20Console.Read();21Environment.Exit(0);22});2324stringinput =Console.ReadLine();25while("Y".Equals(input, StringComparison.OrdinalIgnoreCase))26{27source.Cancel();28}2930Console.Read();31}3233staticvoidRun(CancellationToken token)34{35while(true)36{37if(token.IsCancellationRequested)break;3839Thread.Sleep(1000);4041//執(zhí)行業(yè)務邏輯42Console.WriteLine("我是線程:{0}杠河,正在執(zhí)行業(yè)務邏輯", Thread.CurrentThread.ManagedThreadId);43}44}45}
單從代碼量上面看就縮減了17行代碼,而且業(yè)務邏輯也非常的簡單明了,然后再看業(yè)務邏輯方法Run券敌,其實你根本就不需要所謂的workThreadNums++,--
的操作唾戚,而且多線程下不用鎖的話,還容易出現(xiàn)競態(tài)的問題待诅,解決方案就是使用WhenAll等待一組Tasks完成任務叹坦,之后再串行要退出的Task任務,是不是很完美卑雁,
而協(xié)作取消的話募书,只需將取消的token傳遞給業(yè)務邏輯方法,當主線程執(zhí)行source.Cancel()方法取消的時候测蹲,子線程就會通過IsCancellationRequested感知到主
線程做了取消操作莹捡。
歡迎加入QQ群:364595326