mirrors/TarsCloud_TarsCpp
1
0
Fork 0
mirror of https://gitee.com/TarsCloud/TarsCpp.git synced 2024-12-22 22:16:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

tc_thread queue add front & pop_front

Browse Source
This commit is contained in:
ruanshudong 2020-04-06 21:30:19 +08:00
parent 94f8cf8d18
commit b4529532dc
2 changed files with 38 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
util/include/util/tc_cas_queue.h
View File

@ -34,7 +34,7 @@ public:
typedef D queue_type;
/**
* @brief 从头部获取数据, 没有数据则等待.
* @brief 从头部获取数据, 没有数据抛异常
*
* @param t
* @return bool: true, 获取了数据, false, 无数据

37
util/include/util/tc_thread_queue.h
View File

@ -48,6 +48,14 @@ public:
typedef D queue_type;
/**
* @brief 从头部获取数据, 没有数据抛异常
*
* @param t
* @return bool: true, 获取了数据, false, 无数据
*/
T front();
/**
* @brief 从头部获取数据, 没有数据则等待.
*
@ -60,6 +68,13 @@ public:
*/
bool pop_front(T& t, size_t millsecond = 0, bool wait = true);
/**
* @brief 从头部获取数据.
*
* @return bool: true, 获取了数据, false, 无数据
*/
bool pop_front();
/**
* @brief 通知等待在队列上面的线程都醒过来
*/
@ -148,6 +163,13 @@ protected:
mutable std::mutex _mutex;
};
template<typename T, typename D> T TC_ThreadQueue<T, D>::front()
{
std::unique_lock<std::mutex> lock(_mutex);
return _queue.front();
}
template<typename T, typename D> bool TC_ThreadQueue<T, D>::pop_front(T& t, size_t millsecond, bool wait)
{
if(wait) {
@ -198,6 +220,21 @@ template<typename T, typename D> bool TC_ThreadQueue<T, D>::pop_front(T& t, size
}
template<typename T, typename D> bool TC_ThreadQueue<T, D>::pop_front()
{
std::unique_lock<std::mutex> lock(_mutex);
if (_queue.empty())
{
return false;
}
_queue.pop_front();
assert(_size > 0);
--_size;
return true;
}
template<typename T, typename D> void TC_ThreadQueue<T, D>::notifyT()
{
std::unique_lock<std::mutex> lock(_mutex);