C++ 簡(jiǎn)單的任務(wù)隊(duì)列詳解

更新時(shí)間：2016年12月18日 12:29:27 投稿：jingxian

下面小編就為大家?guī)?lái)一篇C++ 簡(jiǎn)單的任務(wù)隊(duì)列詳解。小編覺(jué)得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧

任務(wù)隊(duì)列是指能夠?qū)崿F(xiàn)任務(wù)在多線程間安全傳遞的先入先出的隊(duì)列。

任務(wù)是指組合了數(shù)據(jù)和操作的對(duì)象，這里面定義為CTask類的對(duì)象。

任務(wù)的實(shí)現(xiàn)：

Task.cpp

#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;
}

Task.h

#pragma once
class CTask
{
  int* m_nCount;
public:
  CTask(int* nCount);
  ~CTask();

  void DoWork();
};

隊(duì)列的實(shí)現(xiàn)：

TaskQueue.cpp

#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;
    }
  }
}

//向任務(wù)隊(duì)列添加任務(wù)
bool CTaskQueue::Push(CTask* task)
{
  if (task == nullptr)
  {
    return false;
  }

  m_mutex.lock();
  m_taskQueue.push(task);
  m_mutex.unlock();

  return true;
}
//從任務(wù)隊(duì)列獲取任務(wù)
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();
}

TaskQueue.h

#pragma once
#include "Task.h"
#include <queue>
#include <mutex>
#include <thread>

class CTaskQueue
{
public:
  CTaskQueue();
  ~CTaskQueue();

private:
  std::queue<CTask*> m_taskQueue; //任務(wù)隊(duì)列
  std::thread m_thread;
  std::mutex m_mutex;
  bool m_bIsStart;  //線程是否開(kāi)啟

public:
  //工作線程
  void WorkThread(); 

  //向任務(wù)隊(duì)列添加任務(wù)
  bool Push(CTask* task);
  //從任務(wù)隊(duì)列獲取任務(wù)
  CTask* Pop();

  //開(kāi)啟線程
  bool Start();
  //終止線程
  void Stop();
};

測(cè)試demo：

// TaskQueue.cpp : 定義控制臺(tái)應(yīng)用程序的入口點(diǎn)。
//

#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);

  //等待線程結(jié)束
  if (thread1.joinable())
  {
    thread1.join();
  }
  if (thread2.joinable())
  {
    thread2.join();
  }

  system("pause");
  return 0;
}

以上這篇C++ 簡(jiǎn)單的任務(wù)隊(duì)列詳解就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: