模板队列-支持多线程

Queue.h:


// 用于缓存数据包的队列容器

#pragma once

#define WIN32_LEAN_AND_MEAN
#include <Windows.h>

#include <list>
#define DEFAULTMAXSIZE 1000000

template<class T>
class Queue
{
public:
	Queue(int nMaxLength = DEFAULTMAXSIZE);
	virtual ~Queue();

public:
	bool IsEmpty(); // 是否为空
	int Length();	// 获取当前队列大小
	const T& Pop(); // 队列出列
	bool push(const T& inElement);   // 入列
	int SetMaxLength(int nMaxLength);// 设置最大容量

private:
	int Lock();	  // 获得互斥量
	int Unlock(); // 释放互斥量
	HANDLE m_hMutex;

	int m_nMaxLength; // 最大大小
	std::list<T> m_list; // 元素
};

#include "Queue.cpp"



Queue.cpp:

#include "Queue.h"
#include <process.h>

// 默认最大为100
template<class T> Queue<T>::Queue(int nMaxLength)
{
	m_nMaxLength = nMaxLength;

	// 创建互斥量
	m_hMutex = CreateMutex(
		NULL, // 使用默认的安全属性
		FALSE,// 默认未获得拥有权
		NULL);// 匿名
}

template<class T> Queue<T>::~Queue()
{
	CloseHandle(m_hMutex);
}

// 是否为空
template<class T> bool Queue<T>::IsEmpty()
{
	Lock();
	bool bTemp = m_list.empty();
	Unlock();

	return bTemp;
}

// 获取当前队列大小
template<class T> int Queue<T>::Length()
{
	Lock();
	int nTemp = m_list.size();
	Unlock();

	return nTemp;
}

// 出列
template<class T> const T& Queue<T>::Pop()
{
	Lock();

	T Temp;
	if (!m_list.empty())
	{
		Temp = m_list.front();
		m_list.pop_front();
	}

	Unlock();

	return Temp;
}

// 是否加入成功
template<class T> bool Queue<T>::push(const T& inElement)
{
	Lock();
	// 当前达到最大容量
	if (m_list.size() > m_nMaxLength)
	{
		Unlock();
		return false;
	}

	m_list.push_back(inElement);
	Unlock();

	return true;
}

// 设置最大容量
template<class T> int Queue<T>::SetMaxLength(int nMaxLength)
{
	m_nMaxLength = nMaxLength;
	return 0;
}

// 获得互斥量
template<class T> int Queue<T>::Lock()
{
	// 等待其它线程释放互斥量
	WaitForSingleObject(m_hMutex, INFINITE);  // no time-out interval

	return 0;
}

// 释放互斥量
template<class T> int Queue<T>::Unlock()
{
	ReleaseMutex(m_hMutex);

	return 0;
}



  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
在 C++ 中实现多线程共享队列可以使用标准库中的 std::queue 和 std::mutex 来实现。具体实现步骤如下: 1. 定义一个模板类,用于表示队列中的元素类型; 2. 在队列类中定义两个 std::queue 成员变量,一个用于存储任务,另一个用于存储等待的线程; 3. 定义一个 std::mutex 对象,用于保护队列的数据; 4. 在队列类中定义 push() 和 pop() 函数,这两个函数需要使用 std::lock_guard<std::mutex> 对象来保护队列的数据; 5. 在 push() 函数中向任务队列中添加任务,并通知等待的线程有新任务加入; 6. 在 pop() 函数中从任务队列中取出一个任务并返回,如果队列为空则等待新任务的到来。 下面是一个简单的多线程共享队列的实现: ```c++ #include <queue> #include <mutex> #include <condition_variable> template<typename T> class ThreadSafeQueue { public: ThreadSafeQueue() = default; void push(const T& value) { std::lock_guard<std::mutex> lock(m_mutex); m_queue.push(value); m_condition.notify_one(); } bool pop(T& value) { std::unique_lock<std::mutex> lock(m_mutex); m_condition.wait(lock, [this]{ return !m_queue.empty(); }); if (m_queue.empty()) { return false; } value = m_queue.front(); m_queue.pop(); return true; } private: std::queue<T> m_queue; mutable std::mutex m_mutex; std::condition_variable m_condition; }; ``` 在上面的代码中,我们使用 std::unique_lock<std::mutex> 来保护队列的数据,并使用 std::condition_variable 来通知等待的线程有新任务加入。在 push() 函数中,我们先获取 std::lock_guard<std::mutex> 对象来保护队列的数据,然后向任务队列中添加任务,并通过 m_condition.notify_one() 通知等待的线程有新任务加入。在 pop() 函数中,我们获取 std::unique_lock<std::mutex> 对象来保护队列的数据,并通过 m_condition.wait() 等待新任务的到来。如果队列中有任务,则取出一个任务并返回,否则返回 false。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

苦逼的IT男

你的鼓励是我最大的创作动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值