【线程同步】使用 C++ 实现 waitgroup

最新推荐文章于 2024-07-29 21:54:30 发布

JunJie_1107

最新推荐文章于 2024-07-29 21:54:30 发布

阅读量779

点赞数

分类专栏： C/C++ 服务器开发文章标签： c++ 开发语言

本文链接：https://blog.youkuaiyun.com/weixin_43869898/article/details/125925887

版权

本文使用独占锁、原子类型、条件变量，实现线程间同步，API接口模仿Go语言中的 WaitGroup

#ifndef __WAITGROUP_H_
#define __WAITGROUP_H_

#include <mutex>
#include <atomic>
#include <condition_variable>

namespace algo {
   

class WaitGroup {
   

public:
    WaitGroup() {
   
        m_counter = 0;
    }

    void Add(int count = 1) {
   
        m_counter += count;
    }

    void Done() {
   
        m_counter--;
        if (m_counter.load() <= 0) {
   
            m_c