C++ 简单的任务队列详解

#include "stdafx.h"#include "Task.h"#include <iostream>using namespace std;CTask::CTask(int* nCount){  m_nCount = nCount;}CTask::~CTask(){}void CTask::DoWork(){  (*m_nCount)++;  cout << "Count = " << *m_nCount << endl;}

#pragma onceclass CTask{  int* m_nCount;public:  CTask(int* nCount);  ~CTask();  void DoWork();};

#include "stdafx.h"#include "TaskQueue.h"CTaskQueue::CTaskQueue(){}CTaskQueue::~CTaskQueue(){}//工作线程void CTaskQueue::WorkThread(){  while (m_bIsStart)  {    if (!m_taskQueue.empty())    {      CTask* it = m_taskQueue.front();      it->DoWork();      m_taskQueue.pop();      delete it;    }  }}//向任务队列添加任务bool CTaskQueue::Push(CTask* task){  if (task == nullptr)  {    return false;  }  m_mutex.lock();  m_taskQueue.push(task);  m_mutex.unlock();  return true;}//从任务队列获取任务CTask* CTaskQueue::Pop(){  CTask* it = nullptr;  m_mutex.lock();  if (!m_taskQueue.empty())  {    it = m_taskQueue.front();    m_taskQueue.pop();  }  m_mutex.unlock();  return it;}bool CTaskQueue::Start(){  if (m_bIsStart)  {    return false;  }  m_bIsStart = true;  m_thread = std::thread(&CTaskQueue::WorkThread, this);  return true;}void CTaskQueue::Stop(){  m_bIsStart = false;  m_thread.join();}

#pragma once#include "Task.h"#include <queue>#include <mutex>#include <thread>class CTaskQueue{public:  CTaskQueue();  ~CTaskQueue();private:  std::queue<CTask*> m_taskQueue; //任务队列  std::thread m_thread;  std::mutex m_mutex;  bool m_bIsStart;  //线程是否开启public:  //工作线程  void WorkThread();   //向任务队列添加任务  bool Push(CTask* task);  //从任务队列获取任务  CTask* Pop();  //开启线程  bool Start();  //终止线程  void Stop();};

// TaskQueue.cpp : 定义控制台应用程序的入口点。//#include "stdafx.h"#include "TaskQueue.h"#include "Task.h"void MyWorkTask1(CTaskQueue* pTaskQueue, int* nCount){  for (size_t i = 0; i < 20; i++)  {      CTask* task = new CTask(nCount);    pTaskQueue->Push(task);  }}void MyWorkTask2(CTaskQueue* pTaskQueue, int* nCount){  for (size_t i = 0; i < 20; i++)  {    CTask* task = new CTask(nCount);    pTaskQueue->Push(task);  }}int _tmain(int argc, _TCHAR* argv[]){  CTaskQueue* pTaskQueue = new CTaskQueue();  pTaskQueue->Start();  int* nCount = new int(0);  std::thread thread1(&MyWorkTask1, pTaskQueue, nCount);  std::thread thread2(&MyWorkTask2, pTaskQueue, nCount);  //等待线程结束  if (thread1.joinable())  {    thread1.join();  }  if (thread2.joinable())  {    thread2.join();  }  system("pause");  return 0;}