web服务器设计(二):线程池的设计
一、设计思路
在此,设计的线程池主要创建了多个子线程来从请求队列中拿出请求并执行。同时,还需要提供插入请求的接口,用于主线程在完成读写I/O后将请求插入队列中。
同时,线程池的设计还需要考虑线程安全。在不同的线程操作队列的时候,需要对队列进行加锁。没有请求在队列的时候,为了降低CPU的利用,我们利用信号量
在队列中没有请求时挂起线程,在插入一个请求后唤醒。
线程池类的类设计如下:
设计思路:
1.在构造函数中按最大线程的的数量开启多个线程,每个线程在pthread_create的时候可以指定一个回调函数,这个函数就是work函数,而work函数里面又调用了线程池的run函数。在创建线程成功之后,需要设置线程分离。
2. run函数的主要作用是从请求队列里面拿出任务,然后执行任务。注意,在拿任务的时候需要上锁。在取出一个任务时候,首先应该减少一个信号量(sem_wait),然后再进行上锁。如果减少信号量的操作不成功,说明队列里面没有任务需要处理,就会阻塞在这里,等待append函数增加任务后唤醒。
3.append函数的主要作用是把任务放进请求队列中。append函数首先需要加锁,然后判断是不是超过了最大的任务处理数量。如果超过的话直接解锁返回false。如果在请求队列的队尾增加任务成功的话,解锁。然后增加信号量,唤醒可能在等待的run函数。
二、代码
#ifndef THREADPOOL_H
#define THREADPOOL_H
#include <list>
#include <cstdio>
#include <exception>
#include <pthread.h>
#include "locker.h"
// 线程池类,将它定义为模板类是为了代码复用,模板参数T是任务类
template<typename T>
class threadpool {
public:
/*thread_number是线程池中线程的数量,max_requests是请求队列中最多允许的、等待处理的请求的数量*/
threadpool(int thread_number = 8, int max_requests = 10000);
~threadpool();
bool append(T* request);
private:
/*工作线程运行的函数,它不断从工作队列中取出任务并执行之*/
static void* worker(void* arg);
void run();
private:
// 线程的数量
int m_thread_number;
// 描述线程池的数组,大小为m_thread_number
pthread_t * m_threads;
// 请求队列中最多允许的、等待处理的请求的数量
int m_max_requests;
// 请求队列
std::list< T* > m_workqueue;
// 保护请求队列的互斥锁
locker m_queuelocker;
// 是否有任务需要处理
sem m_queuestat;
// 是否结束线程
bool m_stop;
};
template< typename T >
threadpool< T >::threadpool(int thread_number, int max_requests) :
m_thread_number(thread_number), m_max_requests(max_requests),
m_stop(false), m_threads(NULL) {
if((thread_number <= 0) || (max_requests <= 0) ) {
throw std::exception();
}
m_threads = new pthread_t[m_thread_number];
if(!m_threads) {
throw std::exception();
}
// 创建thread_number 个线程,并将他们设置为脱离线程。
for ( int i = 0; i < thread_number; ++i ) {
printf( "create the %dth thread\n", i);
if(pthread_create(m_threads + i, NULL, worker, this ) != 0) {
delete [] m_threads;
throw std::exception();
}
// 成功的话返回0 失败的话返回错误号
if( pthread_detach( m_threads[i] ) ) {
delete [] m_threads;
throw std::exception();
}
}
}
template< typename T >
threadpool< T >::~threadpool() {
delete [] m_threads;
m_stop = true;
}
template< typename T >
bool threadpool< T >::append( T* request )
{
// 操作工作队列时一定要加锁,因为它被所有线程共享。
m_queuelocker.lock();
if ( m_workqueue.size() > m_max_requests ) {
m_queuelocker.unlock();
return false;
}
m_workqueue.push_back(request);
m_queuelocker.unlock();
m_queuestat.post();
return true;
}
template< typename T >
void* threadpool< T >::worker( void* arg )
{
threadpool* pool = ( threadpool* )arg;
pool->run();
return pool;
}
template< typename T >
void threadpool< T >::run() {
while (!m_stop) {
m_queuestat.wait();
m_queuelocker.lock();
if ( m_workqueue.empty() ) {
m_queuelocker.unlock();
continue;
}
T* request = m_workqueue.front();
m_workqueue.pop_front();
m_queuelocker.unlock();
if ( !request ) {
continue;
}
request->process();
}
}
#endif