线程池的封装,基础思想与生产者消费者的封装一样,只不过我们是将线程池封装为自动获取任务、执行任务,让用户调用相应的接口来添加任务。
在线程池的封装中,我们同样需要用到的是MutexLock、Condition、Thread这些基本的封装。
基础封装如下:
MutexLock:
1 #ifndef MUTEXLOCK_H 2 #define MUTEXLOCK_H 3 4 #include "NonCopyable.h" 5 #include <pthread.h> 6 #include <stdlib.h> 7 #include <stdio.h> 8 #define TINY_CHECK(exp)\ 9 if(!exp)\ 10 { \ 11 fprintf(stderr, "File : %s, Line : %d Exp : ["#exp"] is true, abort.\n", __FILE__, __LINE__); abort();\ 12 } 13 14 15 16 class MutexLock : NonCopyable 17 { 18 friend class Condition; 19 public: 20 MutexLock(); 21 ~MutexLock(); 22 void lock(); 23 void unlock(); 24 25 bool isLocked() const { return _isLock; } 26 pthread_mutex_t *getMutexPtr() { return &_mutex; } 27 28 private: 29 void restoreMutexStatus() 30 { _isLock = true; } 31 32 pthread_mutex_t _mutex; 33 bool _isLock; 34 }; 35 36 37 class MutexLockGuard : NonCopyable //将锁封装到MutexLockGuard中, 38 { //这样只需定义一个对象,便可 39 public: //便可自动上锁,对象销毁时自动解锁 40 MutexLockGuard(MutexLock &mutex) 41 :_mutex(mutex) 42 { _mutex.lock(); } 43 44 ~MutexLockGuard() 45 { _mutex.unlock(); } 46 47 private: 48 MutexLock &_mutex; 49 }; 50 #define MutexLockGuard(m) "ERROR" 51 52 #endif