使用Task的一些知识优化了一下同事的多线程协作取消的一串代码

最近在看一个同事的代码,代码的本意是在main方法中开启10个线程,用这10个线程来处理一批业务逻辑,在某一时刻当你命令console退出的时候,这个

时候不是立即让console退出,而是需要等待10个线程把检测状态之后的业务逻辑执行完之后再退出,这样做是有道理的,如果强行退出会有可能造成子线程的业

务数据损坏,没毛病吧,业务逻辑大概就是这样。

 

一:现实场景

由于真实场景的代码比较复杂和繁琐,为了方便演示,我将同事所写的代码抽象一下,类似下面这样,看好了咯~~~

复制代码

 1 class Program 2     { 3         private static int workThreadNums = 0; 4  5         private static bool isStop = false; 6  7         static void Main(string[] args) 8         { 9             var tasks = new Task[10];10 11 12             for (int i = 0; i < 10; i++)13             {14                 tasks[i] = Task.Factory.StartNew((obj) =>15                 {16                     Run();17                 }, i);18             }19 20             //是否退出21             string input = Console.ReadLine();22 23             while ("Y".Equals(input, StringComparison.OrdinalIgnoreCase))24             {25                 break;26             }27 28             isStop = true;29 30             while (workThreadNums != 0)31             {32                 Console.WriteLine("正在等待线程结束,当前还在运行线程有:{0}", workThreadNums);33 34                 Thread.Sleep(10);35             }36             Console.WriteLine("准备退出了。。。");37             Console.Read();38             Environment.Exit(0);39         }40 41         static void Run()42         {43             try44             {45                 workThreadNums++;46 47                 while (true)48                 {49                     if (isStop) break;50 51                     Thread.Sleep(1000);52 53                     //执行业务逻辑54                     Console.WriteLine("我是线程:{0},正在执行业务逻辑", Thread.CurrentThread.ManagedThreadId);55                 }56             }57             finally58             {59                 workThreadNums--;60             }61         }62     }

复制代码

 

      其实扫一下上面的代码应该就知道是用来干嘛的,业务逻辑没毛病,基本可以实现刚才的业务场景,在console退出的时候可以完全确保10个线程都把自己的业

务逻辑处理完毕了。不过从美观角度上来看,这种代码就太low了。。。一点档次都没有,比如存在下面两点问题:

 

第一点:局部变量太多,又是isStop又是workThreaNums,导致业务逻辑Run方法中掺杂了很多的非业务逻辑,可读性和维护性都比较low。

第二点:main函数在退出的时候用while检测workThreadNums是否为“0”,貌似没问题,但仔细想想这段代码有必要吗?

 

接下来我把代码跑一下,可以看到这个while检测到了在退出时的workThredNums的中间状态“7”,有点意思吧~~~

 

二:代码优化

  那上面这段代码怎么优化呢?如何踢掉业务逻辑方法中的非业务代码呢?当然应该从业务逻辑上考虑一下了,其实这个问题的核心就是两点:

 

1. 如何实现多线程中的协作取消?

2. 如何实现多线程整体执行完毕通知主线程?

 

这种场景优化千万不要受到前人写的代码所影响,最好忘掉就更好了,不然你会下意识的受到什么workthreadnums,isstop这些变量的左右,不说废话了,如

果你对task并发模型很熟悉的话,你的优化方案很快就会出来的。。。

 

1. 协作取消:

    直接用一个bool变量来判断子线程是否退出的办法其实是很没有档次的,在net 4.0中有一个类(CancellationTokenSource)专门来解决使用bool变量来判

断的这种很low的场景,而且比bool变量具有更强大的功能,这个会在以后的文章中跟大家去讲。

 

2. 多线程整体执行完毕通知主线程

    目前我们看到的方式是主线程通过轮询workthreadnums这种没有档次的方式去做的,其实这种方式本质上就是任务串行,而如果你明白task的话,你就知道

有很多的手段是执行任务串行的,比如什么ContinueWith,WhenAll,WhenAny等等方式,所以你只需要将一组task串联到WhenAll之后就可以了。好了,上

面就是我的解决思路,接下来看一下代码吧:

复制代码

 1  class Program 2     { 3         static void Main(string[] args) 4         { 5             CancellationTokenSource source = new CancellationTokenSource(); 6  7             var tasks = new Task[10]; 8  9             for (int i = 0; i < 10; i++)10             {11                 tasks[i] = Task.Factory.StartNew((m) =>12                 {13                     Run(source.Token);14                 }, i);15             }16 17             Task.WhenAll(tasks).ContinueWith((t) =>18             {19                 Console.WriteLine("准备退出了。。。");20                 Console.Read();21                 Environment.Exit(0);22             });23 24             string input = Console.ReadLine();25             while ("Y".Equals(input, StringComparison.OrdinalIgnoreCase))26             {27                 source.Cancel();28             }29 30             Console.Read();31         }32 33         static void Run(CancellationToken token)34         {35             while (true)36             {37                 if (token.IsCancellationRequested) break;38 39                 Thread.Sleep(1000);40 41                 //执行业务逻辑42                 Console.WriteLine("我是线程:{0},正在执行业务逻辑", Thread.CurrentThread.ManagedThreadId);43             }44         }45     }

复制代码

 

       单从代码量上面看就缩减了17行代码,而且业务逻辑也非常的简单明了,然后再看业务逻辑方法Run,其实你根本就不需要所谓的workThreadNums++,--

的操作,而且多线程下不用锁的话,还容易出现竞态的问题,解决方案就是使用WhenAll等待一组Tasks完成任务,之后再串行要退出的Task任务,是不是很完美,

而协作取消的话,只需将取消的token传递给业务逻辑方法,当主线程执行source.Cancel()方法取消的时候,子线程就会通过IsCancellationRequested感知到主

线程做了取消操作。

欢迎加入技术QQ群:364595326

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章