自己动手写线程池之线程池实现

线程池工作流

终于来到这了，前面做了若干准备工作，都是为了构建线程池。仿照NGINX线程池模型，绘制了线程池工作流程，如下图所示：

任务

定义任务基类Task，所有用户自定义Task都要继承该类，实现Exec方法。线程从任务队列中获取Task，执行Exec。

class Task
{
public:
    virtual void Exec() = 0;
    virtual ~Task() {}
};

接口

void Init(int thread_num);   // 初始化线程池中线程数量
bool AddTask(Task *task);    // 向线程池中添加任务
void Run();                  // 线程池执行任务
void WaitForAllDone();       // 等待所有任务执行完毕后退出

void Stop();                 // 中途停止线程池执行

工作线程

工作线程继承于Thread，实现Handler方法。内部while循环，每次从工作队列中取出一个任务，执行Exec方法。

void ThreadPool::WorkerThread::Handler()
{
    while (!_finished)
    {
        Task *task = _thread_pool->GetTask(this);
        if (task != NULL)
        {
            auto_ptr<Task> at_task(task);

            try
            {
                task->Exec();
            }
            catch(...)
            {
            }

            _thread_pool->Idle(this);
        }
    }

    _thread_pool->Exit(this);
}

代码

https://github.com/spch2008/threadpool/blob/v1.0/src/thread_pool.cpp