如何取消后台线程的执行

介绍

在使用多线程模型进行编程时,经常遇到的问题之一是,当我们关闭前台的UI线程时,后台的辅助线程仍然处于活动状态,从而导致整个应用程序无法正常退出。这时我们需要一种较安全的方式来结束后台线程的运行,这样我们可以随时结束后台线程的运行,并且在线程结束时进行相应的资源清理工作(例如将内存数据写入硬盘)。.net框架提供了一些工具来实现该功能。

目录

  • IsBackground属性
  • Abort方法
  • 轮循方式
  • 取消阻塞的线程

IsBackgound属性

Thread类提供了IsBackground属性,当线程的IsBackground属性被设置为true时,表示此线程为后台工作线程。当一个应用程序结束时,它的所有后台线程会自动的被结束执行。如果你有一个后台线程侦听Socket连接,并且正在被阻塞,那么这时候通过设置线程的IsBackground属性为True,使它自动随应用程序的结束而结束是比较合适的。但在这种情况下,线程会静悄悄的结束,它不会引发任何异常,你的线程没有机会执行一些需要的清理代码。例如,内存中的数据可能会来不及写入磁盘,从而造成丢失数据。

Abort方法

可以调用Thread类的Abort方法来强制终制线程。上调用此方法时,线程上引发ThreadAbortException,并导至线程终结,通过捕获该异常,可以执行一些资源清理代码。但这种模式也有一些问题,主要是难以知道线程上的代码执行到什么地方,所有相应的资源清理代码也难以编写。总的来说这是一种比较粗暴的终止线程执行的方法,通常来说是不推荐使用的。

轮循方式

如果后台线程将执行一个很长的计算,那么可以将计算隔成若干小段,并经常检查是否需要取消线程。.NET框架提供了CancellationTokenSource类来作为线程取消的统一模式。例如:

 public class Example
    {
        public static void Main()
        {
            CancellationTokenSource cts = new CancellationTokenSource();
            var thread = new Thread(ThreadWork);
            thread.Start(cts.Token);
            while (true)
            {
                if(Console.ReadKey().KeyChar == 'c')
                {
                    Console.WriteLine("请求取消线程的执行");
                    cts.Cancel();
                    break;
                }
            }
            Console.ReadLine();
        }

        private static void ThreadWork(object state)
        {
            CancellationToken cancellationToken = (CancellationToken)state;

            while (true)
            {
                // 检查是否取消
                if(cancellationToken.IsCancellationRequested)
                {
                    Console.WriteLine("线程已经取消了");
                    Console.WriteLine("线程的资源已经清理完成。");
                    break;
                }
                // 模拟工作
                Thread.SpinWait(500000);
                Console.WriteLine("我还在工作。");
            }
        }
    }

取消阻塞的线程

上面的示例中,后台线程会长时间进行计算,但更多的时候,线程会由于等待某个事件,从而进入阻塞状态。这个时候,实际上线程已经不再执行状态了,很明显,它没有机会去检查取消标志。 那么,该如何解决这个问题呢?CancellationToken的WaitHandle属性提供了解答。WaitHandle类有一个静态方法WaitAny,它可以同时等待多个事件,当多个事件中的任意一个有效时,线程都会从阻塞状态中返回。可以根据WaitAny方法的返回值来判断发生了什么事件,从而相应的执行代码。例子:

    public class Example
    {
        private static int Value;

        public static void Main()
        {
            var autoResetEvent = new AutoResetEvent(false);
            var cts = new CancellationTokenSource();
            var state = new { ValueAvailableEvent = autoResetEvent, CancellationToken = cts.Token };
            var threadConsumer = new Thread(ConsumerThreadWork);
            var threadProducter = new Thread(ProducterThreadWork);

            threadConsumer.Start(state);
            threadProducter.Start(state);

            while (true)
            {
                if (Console.ReadKey().KeyChar == 'c')
                {
                    Console.WriteLine("请求取消线程的执行");
                    cts.Cancel();
                    break;
                }
            }
            Console.ReadLine();

        }
        public static void ProducterThreadWork(dynamic state)
        {
            var valueAvailableEvent = (AutoResetEvent)state.ValueAvailableEvent;
            var cancellationToken = (CancellationToken)state.CancellationToken;
            var rand = new Random();
            while (!cancellationToken.IsCancellationRequested)
            {
                Value = rand.Next();
                Console.WriteLine("\r\n产生一个值{0}", Value);
                valueAvailableEvent.Set();
                Thread.Sleep(500);
            }

            Console.WriteLine("生产者线程被取消。");
        }

        public static void ConsumerThreadWork(dynamic state)
        {
            var valueAvailableEvent = (AutoResetEvent)state.ValueAvailableEvent;
            var cancellationToken = (CancellationToken)state.CancellationToken;
            var events = new[] { valueAvailableEvent, cancellationToken.WaitHandle };

            while (true)
            {
                var eventIndex = WaitHandle.WaitAny(events);
                // 处理数据
                if (eventIndex == 0)
                {
                    Console.WriteLine("处理值{0}。", Value);
                }
                // 处理取消事件
                else if (eventIndex == 1)
                {
                    Console.WriteLine("消费者线程被取消。");
                    break;
                }
            }
        }
    }

在上面的例子中,有三个线程,分别是UI线程,生产者线程和消费者线程。其中生产者线程每隔一秒产生一个有效数值,并将数据保存到Value字段中,而消费者线程等待值的产生,这个等待的过程是阻塞的。消费都线程通过WaitHandle.WaitAny方法来同时等待值有效事件或者取消事件,当任意一个事件有效时,线程都将继续,并且通过返回的值来判断发生的事件,并作相应的处理。

总结

多线程模型中的线程取消问题还是比较复杂的。Thread.IsBackground属性提供了在前台线程结束后自动结束线程的方法。Thread.Abort方法提供了一种“粗暴”的结束线程的方法。CancellationTokenSource类则是线程取消的标准模式,我们应当更多的使用这种模式。文章写的不多,基本是字数不够,代码来凑,大家伙将就的看看吧。

posted @ 2012-11-09 15:48  指南针  阅读(5348)  评论(4编辑  收藏  举报