C++ 简单的任务队列详解
任务队列是指能够实现任务在多线程间安全传递的先入先出的队列。
任务是指组合了数据和操作的对象,这里面定义为CTask类的对象。
任务的实现:
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();
};
队列的实现:
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;
}
}
}
//向任务队列添加任务
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();
}
TaskQueue.h
#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();
};
测试demo:
// 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;
}
以上这篇C++ 简单的任务队列详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。
以上是 C++ 简单的任务队列详解 的全部内容, 来源链接: utcz.com/z/337248.html