我们封装好了Buffer后,却不知道具体要多少个线程,一个一个线程的去关联Buffer太繁琐了。
那么,我们是不是可以讲Buffer与线程的一个队列封装在一起呢 ?
由于vector中不能存放Thread,所以,我们应在vector中添加相应的Thread *,这样,就可以完成封装了。
声明代码如下:
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
1 #ifndef WORKSHOP_H 2 #define WORKSHOP_H 3 #include "NonCopyable.h" 4 #include "Buffer.h" 5 #include <vector> 6 class Producer; 7 class Consumer; 8 class WorkShop : NonCopyable 9 { 10 public: 11 WorkShop(size_t bufferSize, size_t producerSize, size_t consumerSize); 12 13 ~WorkShop(); 14 void startWork(); 15 private: 16 17 size_t _bufferSize; 18 19 Buffer _buffer; 20 size_t _producerSize; 21 size_t _consumerSize; 22 23 std::vector<Producer *> _producers; 24 std::vector<Consumer *> _consumers; 25 }; 26 27 28 #endif /*WORKSHOP_H*/
接下来,我们需要实现声明中的函数,构造函数中,我们需要将两个vector中的指针初始化,将该vector的大小初始化为我们得到的vector大小,将每个指针初始化为NULL, 接下来,我们需要为每个指针new一个Thread对象,这样,就完成了初始化。
同样的,当我们析构的时候,也应该将vector中的指针delete掉。
cpp代码如下:
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
1 #include "WorkShop.h" 2 #include "Producer.h" 3 #include "Consumer.h" 4 5 WorkShop::WorkShop(size_t bufferSize, size_t producerSize, size_t consumerSize) 6 :_bufferSize(bufferSize), 7 _buffer(_bufferSize), 8 _producerSize(producerSize), 9 _consumerSize(consumerSize), 10 _producers(_producerSize, new Producer(_buffer)), 11 _consumers(_consumerSize, new Consumer(_buffer)) 12 { 13 14 } 15 16 WorkShop::~WorkShop() 17 { 18 for(std::vector<Producer *>::iterator iter = _producers.begin(); 19 iter != _producers.end(); 20 ++ iter) 21 delete *iter; 22 23 for(std::vector<Consumer *>::iterator iter = _consumers.begin(); 24 iter != _consumers.end(); 25 ++ iter) 26 delete *iter; 27 } 28 29 void WorkShop::startWork() 30 { 31 for(std::vector<Producer *>::iterator iter = _producers.begin(); 32 iter != _producers.end(); 33 ++ iter) 34 (*iter)->start(); 35 36 for(std::vector<Consumer *>::iterator iter = _consumers.begin(); 37 iter != _consumers.end(); 38 ++ iter) 39 (*iter)->start(); 40 41 for(std::vector<Producer *>::iterator iter = _producers.begin(); 42 iter != _producers.end(); 43 ++ iter) 44 (*iter)->join(); 45 46 for(std::vector<Consumer *>::iterator iter = _consumers.begin(); 47 iter != _consumers.end(); 48 ++ iter) 49 (*iter)->join(); 50 }
在startWork函数中, 我们先将每个线程的指针调用start()函数,将线程开启, 当我们退出进程后,再调用join()函数将每个线程关闭。
测试代码如下:
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
1 #include "WorkShop.h" 2 3 int main(int argc, const char *argv[]) 4 { 5 WorkShop a(5, 10, 15); 6 a.startWork(); 7 return 0; 8 }
编译时要注意,需要的各个cpp文件在前面几篇文章中已经列出,请自行下载尝试。