使用countDownLatch 实现 三体动漫中航天服检查流程

#include <boost/foreach.hpp>                 //-lboost_iostreams
#include "muduo/base/CountDownLatch.h"
#include "muduo/base/Thread.h"
#include <iostream>
#include <vector>
#include <string>
#include <vector>

using namespace std;

class TestFun
{
    
public:
    TestFun(int numThreads,vector<string>* p) : latch_(numThreads),pVector(p)
      {
          for (int i = 0; i < numThreads; ++i)
        {
              threads_.emplace_back(new muduo::Thread(std::bind(&TestFun::threadFunc,this),p->at(i)));
        }                
      }

      void run()
    {        
        for (auto& thr : threads_)
        {
              thr->start();
        }
        printf("all threads started\n");

        printf("waiting for count down latch\n");
        printf("in run:%d->Count:%d\n",muduo::CurrentThread::tid(),latch_.getCount());

        latch_.wait();                                  //会原子unlock mutex并进入等待，wait执行完毕会自动重新加锁
    }

    void threadFunc()
      {
        printf("tid=%d,开始检查:%s\n",muduo::CurrentThread::tid(),muduo::CurrentThread::name());
           latch_.countDown();                              //必须在while前，启动一个减1,0会唤醒wait()启动完毕
        printf("%d->Count:%d\n",muduo::CurrentThread::tid(),latch_.getCount());
        bool running = true;
           while (running)
        {
            sleep(1);              
              running = false;
        }    
    
        printf("tid=%d,检查完毕:%s \n",muduo::CurrentThread::tid(),muduo::CurrentThread::name());
    }  


    void joinAll()
    {
        for (auto& thr : threads_)
        {
          thr->join();
        }

        printf("ALL thread finish\n");
    }


private:  
    muduo::CountDownLatch                       latch_;
    std::vector<std::unique_ptr<muduo::Thread>> threads_;
    vector<string>*                             pVector;  
};

int main()
{
    vector<string> vec;
    vec.emplace_back("1.外骨骼关节轴承");
    vec.emplace_back("2.辅助背包");
    vec.emplace_back("3.防护层");
    vec.emplace_back("4.隔热层");
    vec.emplace_back("5.气密限制层");    
    //BOOST_FOREACH(std::string s,vec) cout<<s<<endl;

    printf("Main: pid=%d, tid=%d\n", ::getpid(), muduo::CurrentThread::tid());
    
    
      TestFun t(vec.size(),&vec);
      t.run();
      t.joinAll();    

    printf("舱外航天服全模块检查完毕\n");  

   
    return 0;
}

 

#include <boost/foreach.hpp>                 //-lboost_iostreams
#include "muduo/base/CountDownLatch.h"
#include "muduo/base/Thread.h"
#include <iostream>
#include <vector>
#include <string>
#include <vector>

using namespace std;

#define THREAD_NUM   2

muduo::CountDownLatch  order_latch(1);
muduo::CountDownLatch  answer_latch(THREAD_NUM);

void workerThread()
{
  printf("work thread pid = %d->Wait\n" , muduo::CurrentThread::tid());
  order_latch.wait();                                                    //等待发令
  printf("work thread pid = %d->Start\n", muduo::CurrentThread::tid());  //收到起跑
    
  bool running = true;
  while (running)
  {
    sleep(1);            
    answer_latch.countDown();                                            //结束 
    running = false;
  }    
    
  printf("work thread pid = %d->Stop\n" , muduo::CurrentThread::tid());  //收到起跑
}

int main()
{    
  char c[10];
    vector<unique_ptr<muduo::Thread>> threads;  

    for (int i = 0; i < THREAD_NUM; i++)
  {
    sprintf(c,"%d",i);
    auto p = new muduo::Thread(workerThread,string(c));
    threads.emplace_back(p);
    p->start();    
  }  

  sleep(1);

  printf("order thread pid = %d->Start\n", muduo::CurrentThread::tid());
  order_latch.countDown();                                               //发起跑令

  printf("order thread pid = %d->Wait\n", muduo::CurrentThread::tid());
  answer_latch.wait();                                                   

  printf("order thread pid = %d->Stop\n", muduo::CurrentThread::tid());  //结束

  for (auto& thr : threads)
  {
    thr->join();
  }

  printf("All thread finish\n");  
  return 0;
}

 

备注:编译需链接 muduo库和boost库

-lmuduo_base -lboost_iostreams