C#多线程(8):线程完成数

内容纲要

解决一个问题

假如,程序需要向一个 Web 发送 5 次请求,受网路波动影响,有一定几率请求失败。如果失败了,就需要重试。

示例代码如下:

    class Program
    {
        private static int count = 0;
        static void Main(string[] args)
        {
            for (int i = 0; i < 5; i++)
                new Thread(HttpRequest).Start();            // 创建线程

            // 用于不断向另一个线程发送信号
            while (count < 5)
            {
                Thread.Sleep(100);
            }
            Console.WriteLine("任务执行完毕");
        }

        // 模拟网络请求
        public static void HttpRequest()
        {
            Console.WriteLine("开始一个任务");
            // 随机生成一个数,如果为偶数,则模拟请求失败
            bool isSuccess = (new Random().Next(0, 10)) % 2 == 0;

            // ... ...模拟请求 HTTP
            Thread.Sleep(TimeSpan.FromSeconds(2));

            // 请求失败则重试
            if (!isSuccess)
            {
                Console.WriteLine($"请求失败,count={count}");
                new Thread(() =>
                {
                    HttpRequest();
                }).Start();
                return;
            }
            // 完成一次任务,+1
            Interlocked.Add(ref count,1);
            Console.WriteLine($"完成任务,count={count}");
        }
    }

代码太糟糕了,但我们可以使用 CountdownEvent 类来改造它。

CountdownEvent 类

表示在计数变为零时处于有信号状态的同步基元。

也就是说,设定一个计数器,每个线程完成后,就会减去 1 ,当计数器为 0 时,代表所有线程都已经完成了任务。

构造函数和方法

CountdownEvent 类的构造函数如下:

构造函数 说明
CountdownEvent(Int32) 使用指定计数初始化 CountdownEvent 类的新实例。

CountdownEvent 类的常用方法如下:

方法 说明
AddCount() 将 CountdownEvent 的当前计数加 1。
AddCount(Int32) 将 CountdownEvent 的当前计数增加指定值。
Reset() 将 CurrentCount 重置为 InitialCount 的值。
Reset(Int32) 将 InitialCount 属性重新设置为指定值。
Signal() 向 CountdownEvent 注册信号,同时减小 CurrentCount 的值。
Signal(Int32) 向 CountdownEvent 注册多个信号,同时将 CurrentCount 的值减少指定数量。
TryAddCount() 增加一个 CurrentCount 的尝试。
TryAddCount(Int32) 增加指定值的 CurrentCount 的尝试。
Wait() 阻止当前线程,直到设置了 CountdownEvent 为止。
Wait(CancellationToken) 阻止当前线程,直到设置了 CountdownEvent 为止,同时观察 CancellationToken。
Wait(Int32) 阻止当前线程,直到设置了 CountdownEvent 为止,同时使用 32 位带符号整数测量超时。
Wait(Int32, CancellationToken) 阻止当前线程,直到设置了 CountdownEvent 为止,并使用 32 位带符号整数测量超时,同时观察 CancellationToken。
Wait(TimeSpan) 阻止当前线程,直到设置了 CountdownEvent 为止,同时使用 TimeSpan 测量超时。
Wait(TimeSpan, CancellationToken) 阻止当前线程,直到设置了 CountdownEvent 为止,并使用 TimeSpan 测量超时,同时观察 CancellationToken。

API 比较多,没事,我们来慢慢了解它。

示例

我们来编写一个场景代码,一个有五件事,需要完成,分别派出 5 个人去实现。

.Wait(); 用在一个线程中,这个线程将等待其它完成都完成任务后,才能继续往下执行。

Signal(); 用于工作线程中,向 CountdownEvent 对象发送信号,告知线程已经完成任务,然后 CountdownEvent.CurrentCount 将减去 1。

当计数器为 0 时,阻塞的线程将恢复执行。

代码示例如下:

    class Program
    {
        // 手头上有 5 件事
        private static CountdownEvent countd = new CountdownEvent(5);
        static void Main(string[] args)
        {
            Console.WriteLine("开始交待任务");
            // 同时叫 5 个人,去做 5 件事
            for (int i = 0; i < 5; i++)
            {
                Thread thread = new Thread(DoOne);
                thread.Name = $"{i}";
                thread.Start();
            }

            // 等他们都完成事情
            countd.Wait();

            Console.WriteLine("任务完成,线程退出");
            Console.ReadKey();
        }

        public static void DoOne()
        {
            int n = new Random().Next(0, 10);
            // 模拟要 n 秒才能完成
            Thread.Sleep(TimeSpan.FromSeconds(n));
            // 完成了,减去一件事
            countd.Signal();
            Console.WriteLine($"    {Thread.CurrentThread.Name}完成一件事了");
        }
    }

示例很简单,每个线程在完成自己的任务时,需要调用 Signal() 方法,使得计数器减去1。

.Wait(); 可以等待所有的任务完成。

需要注意的是,如果不调用 Signal() 或者计数器一直不为0,那么 Wait() 将无限等待。

当然,Wait() 可以设置等待时间,

另外我们也看到了常用方法中有 AddCount()Reset()等。

这个类的等待控制方式比较宽松,Wait() 后,到底什么时候才能执行,全凭其它线程自觉。

如果发现线程执行任务失败,我们可以不调用 Signal() 或者 使用 AddCount() 来增加次数,进行重试

点赞

发表评论

电子邮件地址不会被公开。 必填项已用*标注