【muduo】base篇--CountDownLatch（倒计时计数器）

136 阅读 0 评论 90 点赞

我是靠谱客的博主烂漫微笑，这篇文章主要介绍【muduo】base篇--CountDownLatch（倒计时计数器），现在分享给大家，希望可以做个参考。

一、倒计时计数器的作用

CountDownLatch是一个同步辅助类，在完成一组正在其他线程中执行的操作之前，它允许一个或多个线程一直等待。假设我们周末要去旅游，出游前需要提前订好机票、巴士和酒店，都订好后就可以出发了，这个用代码怎么实现？可以用CountDownLatch。

主线程发起多个子线程，等这些子线程各自都完成一定的任务之后，主线程才继续执行。通常用于主线程等待多个子线程完成初始化。
主线程发起多个子线程，子线程都等待主线程，主线程完成其他一些任务之后通知所有子线程开始执行。通常用于多个子线程等待主线程发出“起跑”命令。

二、CountDownLatch源码

【执行过程】：

1. 首先，创建一个计数器，设置 count 的初始值，程序中的数值设置为3；

2. 在执行了线程 D 之后，调用 countDownLatch.wait()方法，将会进入阻塞状态，直到 countDownLatch 的 count 参数值为 0；

3. 在其他线程里，调用 countDownLatch.countDown() 方法，调用该方法会将计数值减 1；

4. 当其他线程中的 countDown() 方法把计数值变成 0 时，等待线程里的 countDownLatch.wait() 立即退出，执行下面的代码。

复制代码

#include <muduo/base/CountDownLatch.h>

using namespace muduo;

/******************************************************************** 
Description : 注意声明的顺序，mutex_要先于condition_构造。
*********************************************************************/
CountDownLatch::CountDownLatch(int count)
  : mutex_(),
    condition_(mutex_),
    count_(count)
{
}

/******************************************************************** 
Description : 主线程调用wait进入阻塞状态，等待计数器变为0。
*********************************************************************/
void CountDownLatch::wait()
{
  // 调用wait前先加锁，调用后自动解锁
  MutexLockGuard lock(mutex_);
  while (count_ > 0)
  {
    condition_.wait(); 
  }
}

/******************************************************************** 
Description:子线程调用countDown将计数器减一，计数器变为0唤醒所有线程。
*********************************************************************/
void CountDownLatch::countDown()
{
  MutexLockGuard lock(mutex_);
  --count_;
  // 倒计时计数为0，唤醒所有线程
  if (count_ == 0)
  {
    condition_.notifyAll();
  }
}

/******************************************************************** 
Description : 获取倒计时计数器的值。
*********************************************************************/
int CountDownLatch::getCount() const
{
  MutexLockGuard lock(mutex_);
  return count_;
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#include <muduo/base/CountDownLatch.h>

using namespace muduo;

/******************************************************************** 
Description : 注意声明的顺序，mutex_要先于condition_构造。
*********************************************************************/
CountDownLatch::CountDownLatch(int count)
  : mutex_(),
    condition_(mutex_),
    count_(count)
{
}

/******************************************************************** 
Description : 主线程调用wait进入阻塞状态，等待计数器变为0。
*********************************************************************/
void CountDownLatch::wait()
{
  // 调用wait前先加锁，调用后自动解锁
  MutexLockGuard lock(mutex_);
  while (count_ > 0)
  {
    condition_.wait(); 
  }
}

/******************************************************************** 
Description:子线程调用countDown将计数器减一，计数器变为0唤醒所有线程。
*********************************************************************/
void CountDownLatch::countDown()
{
  MutexLockGuard lock(mutex_);
  --count_;
  // 倒计时计数为0，唤醒所有线程
  if (count_ == 0)
  {
    condition_.notifyAll();
  }
}

/******************************************************************** 
Description : 获取倒计时计数器的值。
*********************************************************************/
int CountDownLatch::getCount() const
{
  MutexLockGuard lock(mutex_);
  return count_;
}