ToB企服应用市场:ToB评测及商务社交产业平台

标题: [项目][WebServer][ThreadPool]详细讲解 [打印本页]
作者: 王柳    时间: 2024-9-29 02:19
标题: [项目][WebServer][ThreadPool]详细讲解
  1. static const int THREAD_POOL_NUM = 10;
  3. // 单例模式
  4. class ThreadPool
  5. {
  6. public:
  7.     static ThreadPool *GetInstance(int num = THREAD_POOL_NUM)
  8.     {
  9.         static pthread_mutex_t sMtx = PTHREAD_MUTEX_INITIALIZER;
  10.         if (_tp == nullptr)
  11.         {
  12.             pthread_mutex_lock(&sMtx);
  13.             if (_tp == nullptr) // 双重判断，以防线程安全问题
  14.             {
  15.                 _tp = new ThreadPool(num);
  16.                 _tp->Init();
  17.             }
  18.             pthread_mutex_unlock(&sMtx);
  19.         }
  21.         return _tp;
  22.     }
  24.     // static使该成员函数没有this指针，因为线程执行的函数只能有一个void*参数
  25.     static void *ThreadRoutine(void *args)
  26.     {
  27.         ThreadPool *tp = (ThreadPool *)args;
  29.         while(true)
  30.         {
  31.             Task task;
  32.             tp->Lock();
  33.             while(tp->TaskQueueIsEmpty()) // while防止伪唤醒
  34.             {
  35.                 tp->ThreadWait();
  36.             }
  37.             tp->Pop(task);
  38.             tp->Unlock(); // 注意，不要在临界资源区内处理任务哦~
  39.             task.ProcessOn();
  40.         }
  41.     }
  43.     bool Init()
  44.     {
  45.         for (int i = 0; i < _num; i++)
  46.         {
  47.             pthread_t tid;
  48.             if (pthread_create(&tid, nullptr, ThreadRoutine, this) != 0)
  49.             {
  50.                 LOG(FATAL, "Create ThreadPool Error");
  51.                 return false;
  52.             }
  53.         }
  54.         LOG(INFO, "Create ThreadPool Success");
  55.         
  56.         return true;
  57.     }
  59.     void Push(const Task& task) // in
  60.     {
  61.         Lock();
  62.         _taskQueue.push(task); // 任务队列为临界资源，操作要加锁
  63.         Unlock();
  65.         ThreadWakeUp();
  66.     }
  68.     void Pop(Task& task) // out
  69.     {
  70.         task = _taskQueue.front();
  71.         _taskQueue.pop();
  72.     }
  74.     void ThreadWait()
  75.     {
  76.         pthread_cond_wait(&_cond, &_mtx);
  77.     }
  79.     void ThreadWakeUp()
  80.     {
  81.         pthread_cond_signal(&_cond);
  82.     }
  84.     bool TaskQueueIsEmpty()
  85.     {
  86.         return !_taskQueue.size();
  87.     }
  89.     void Lock()
  90.     {
  91.         pthread_mutex_lock(&_mtx);
  92.     }
  94.     void Unlock()
  95.     {
  96.         pthread_mutex_unlock(&_mtx);
  97.     }
  99.     bool IsStop()
  100.     {
  101.         return _stop;
  102.     }
  104.     ~ThreadPool()
  105.     {
  106.         pthread_mutex_destroy(&_mtx);
  107.         pthread_cond_destroy(&_cond);
  108.     }
  109. private:
  110.     ThreadPool(int num = THREAD_POOL_NUM)
  111.         : _num(num), _stop(false)
  112.     {
  113.         pthread_mutex_init(&_mtx, nullptr);
  114.         pthread_cond_init(&_cond, nullptr);
  115.     }
  117.     ThreadPool(const ThreadPool &) = delete;
  118. private:
  119.     int _num;
  120.     bool _stop;
  121.     std::queue<Task> _taskQueue;
  122.     pthread_mutex_t _mtx;
  123.     pthread_cond_t _cond;
  124.     static ThreadPool *_tp;
  125. };
  127. ThreadPool* ThreadPool::_tp = nullptr;
复制代码
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。



欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/) Powered by Discuz! X3.4