Linux组件封装(五) WorkShop的封装
我们封装好了Buffer后,却不知道具体要多少个线程,一个一个线程的去关联Buffer太繁琐了。
那么,我们是不是可以讲Buffer与线程的一个队列封装在一起呢 ?
由于vector中不能存放Thread,所以,我们应在vector中添加相应的Thread *,这样,就可以完成封装了。
声明代码如下:
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代码如下:
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, NULL), 11 _consumers(_consumerSize, NULL) 12 { 13 for(std::vector<Producer *>::iterator iter = _producers.begin(); 14 iter != _producers.end(); 15 ++ iter) 16 *iter = new Producer(_buffer); 17 18 for(std::vector<Consumer *>::iterator iter = _consumers.begin(); 19 iter != _consumers.end(); 20 ++ iter) 21 *iter = new Consumer(_buffer); 22 } 23 24 WorkShop::~WorkShop() 25 { 26 for(std::vector<Producer *>::iterator iter = _producers.begin(); 27 iter != _producers.end(); 28 ++ iter) 29 delete *iter; 30 31 for(std::vector<Consumer *>::iterator iter = _consumers.begin(); 32 iter != _consumers.end(); 33 ++ iter) 34 delete *iter; 35 } 36 37 void WorkShop::startWork() 38 { 39 for(std::vector<Producer *>::iterator iter = _producers.begin(); 40 iter != _producers.end(); 41 ++ iter) 42 (*iter)->start(); 43 44 for(std::vector<Consumer *>::iterator iter = _consumers.begin(); 45 iter != _consumers.end(); 46 ++ iter) 47 (*iter)->start(); 48 49 for(std::vector<Producer *>::iterator iter = _producers.begin(); 50 iter != _producers.end(); 51 ++ iter) 52 (*iter)->join(); 53 54 for(std::vector<Consumer *>::iterator iter = _consumers.begin(); 55 iter != _consumers.end(); 56 ++ iter) 57 (*iter)->join(); 58 }
在startWork函数中, 我们先将每个线程的指针调用start()函数,将线程开启, 当我们退出进程后,再调用join()函数将每个线程关闭。
测试代码如下:
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文件在前面几篇文章中已经列出,请自行下载尝试。
郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。