Linux组件封装(五) WorkShop的封装

我们封装好了Buffer后，却不知道具体要多少个线程，一个一个线程的去关联Buffer太繁琐了。

那么，我们是不是可以讲Buffer与线程的一个队列封装在一起呢 ？

由于vector中不能存放Thread，所以，我们应在vector中添加相应的Thread *，这样，就可以完成封装了。

声明代码如下:

#ifndef WORKSHOP_H
#define WORKSHOP_H
#include "NonCopyable.h"
#include "Buffer.h"
#include <vector>
class Producer;
class Consumer;
class WorkShop : NonCopyable
{
public:
    WorkShop(size_t bufferSize, size_t producerSize, size_t consumerSize);

    ~WorkShop();
    void startWork();
private:

    size_t _bufferSize;

    Buffer _buffer;
    size_t _producerSize;
    size_t _consumerSize;

    std::vector<Producer *> _producers;
    std::vector<Consumer *> _consumers;
};


#endif  /*WORKSHOP_H*/

接下来，我们需要实现声明中的函数，构造函数中，我们需要将两个vector中的指针初始化，将该vector的大小初始化为我们得到的vector大小，将每个指针初始化为NULL， 接下来，我们需要为每个指针new一个Thread对象，这样，就完成了初始化。

同样的，当我们析构的时候，也应该将vector中的指针delete掉。

cpp代码如下:

#include "WorkShop.h"
#include "Producer.h"
#include "Consumer.h"

WorkShop::WorkShop(size_t bufferSize, size_t producerSize, size_t consumerSize)
    :_bufferSize(bufferSize),
     _buffer(_bufferSize),
     _producerSize(producerSize),
     _consumerSize(consumerSize),
     _producers(_producerSize, new Producer(_buffer)),
     _consumers(_consumerSize, new Consumer(_buffer))
{

}

WorkShop::~WorkShop()
{
    for(std::vector<Producer *>::iterator iter = _producers.begin();
        iter != _producers.end();
        ++ iter)
        delete *iter;

    for(std::vector<Consumer *>::iterator iter = _consumers.begin();
        iter != _consumers.end();
        ++ iter)
        delete *iter;
}

void WorkShop::startWork()
{
    for(std::vector<Producer *>::iterator iter = _producers.begin();
        iter != _producers.end();
        ++ iter)
        (*iter)->start();

    for(std::vector<Consumer *>::iterator iter = _consumers.begin();
        iter != _consumers.end();
        ++ iter)
        (*iter)->start();

    for(std::vector<Producer *>::iterator iter = _producers.begin();
        iter != _producers.end();
        ++ iter)
        (*iter)->join();

    for(std::vector<Consumer *>::iterator iter = _consumers.begin();
        iter != _consumers.end();
        ++ iter)
        (*iter)->join();
}

在startWork函数中， 我们先将每个线程的指针调用start()函数，将线程开启， 当我们退出进程后，再调用join()函数将每个线程关闭。

测试代码如下:

#include "WorkShop.h"

int main(int argc, const char *argv[])
{
    WorkShop a(5, 10, 15);
    a.startWork();
    return 0;
}

编译时要注意，需要的各个cpp文件在前面几篇文章中已经列出，请自行下载尝试。