From 7285f5762bd3f1be94884730d9a28dd406f98fc5 Mon Sep 17 00:00:00 2001 From: wangzhengquan <wangzhengquan85@126.com> Date: 星期三, 06 一月 2021 18:12:05 +0800 Subject: [PATCH] update --- src/queue/lock_free_queue.h | 112 ++++++++++++++++++++++++++++++++++++-------------------- 1 files changed, 72 insertions(+), 40 deletions(-) diff --git a/src/queue/lock_free_queue.h b/src/queue/lock_free_queue.h index 84c885c..6a610dc 100644 --- a/src/queue/lock_free_queue.h +++ b/src/queue/lock_free_queue.h @@ -7,6 +7,7 @@ #include "sem_util.h" #include "logger_factory.h" #include "shm_allocator.h" +#include "px_sem_util.h" // default Queue size #define LOCK_FREE_Q_DEFAULT_SIZE 16 @@ -75,11 +76,13 @@ { private: - int slots; - int items; + sem_t slots; + sem_t items; + + public: - int mutex; + sem_t mutex; LockFreeQueue(size_t qsize = LOCK_FREE_Q_DEFAULT_SIZE); /// @brief destructor of the class. @@ -150,9 +153,14 @@ LockFreeQueue<ELEM_T, Allocator, Q_TYPE>::LockFreeQueue(size_t qsize): reference(0), m_qImpl(qsize) { // std::cout << "LockFreeQueue init reference=" << reference << std::endl; - slots = SemUtil::get(IPC_PRIVATE, qsize); - items = SemUtil::get(IPC_PRIVATE, 0); - mutex = SemUtil::get(IPC_PRIVATE, 1); + if (sem_init(&slots, 1, qsize) == -1) + err_exit(errno, "LockFreeQueue sem_init"); + if (sem_init(&items, 1, 0) == -1) + err_exit(errno, "LockFreeQueue sem_init"); + if (sem_init(&mutex, 1, 1) == -1) + err_exit(errno, "LockFreeQueue sem_init"); + + } template < @@ -162,9 +170,15 @@ LockFreeQueue<ELEM_T, Allocator, Q_TYPE>::~LockFreeQueue() { // LoggerFactory::getLogger()->debug("LockFreeQueue desctroy"); - SemUtil::remove(slots); - SemUtil::remove(items); - SemUtil::remove(mutex); + if(sem_destroy(&slots) == -1) { + err_exit(errno, "LockFreeQueue sem_destroy"); + } + if(sem_destroy(&items) == -1) { + err_exit(errno, "LockFreeQueue sem_destroy"); + } + if(sem_destroy(&mutex) == -1) { + err_exit(errno, "LockFreeQueue sem_destroy"); + } } template < @@ -201,16 +215,15 @@ template <typename T, typename AT> class Q_TYPE> bool LockFreeQueue<ELEM_T, Allocator, Q_TYPE>::push(const ELEM_T &a_data) { -LoggerFactory::getLogger()->debug("==================LockFreeQueue push before\n"); - if (SemUtil::dec(slots) == -1) { +// LoggerFactory::getLogger()->debug("==================LockFreeQueue push before\n"); + if (sem_wait(&slots) == -1) { err_msg(errno, "LockFreeQueue push"); return false; } - + if ( m_qImpl.push(a_data) ) { - - SemUtil::inc(items); -LoggerFactory::getLogger()->debug("==================LockFreeQueue push after\n"); + sem_post(&items); +// LoggerFactory::getLogger()->debug("==================LockFreeQueue push after\n"); return true; } return false; @@ -223,7 +236,7 @@ template <typename T, typename AT> class Q_TYPE> bool LockFreeQueue<ELEM_T, Allocator, Q_TYPE>::push_nowait(const ELEM_T &a_data) { - if (SemUtil::dec_nowait(slots) == -1) { + if (sem_trywait(&slots) == -1) { if (errno == EAGAIN) return false; else { @@ -234,7 +247,7 @@ } if ( m_qImpl.push(a_data)) { - SemUtil::inc(items); + sem_post(&items); return true; } return false; @@ -245,22 +258,38 @@ typename ELEM_T, typename Allocator, template <typename T, typename AT> class Q_TYPE> -bool LockFreeQueue<ELEM_T, Allocator, Q_TYPE>::push_timeout(const ELEM_T &a_data, const struct timespec * timeout) +bool LockFreeQueue<ELEM_T, Allocator, Q_TYPE>::push_timeout(const ELEM_T &a_data, const struct timespec * ts) { + // int tmp_sec; + // struct timespec timeout; + // if (clock_gettime(CLOCK_REALTIME, &timeout) == -1) + // err_exit(errno, "clock_gettime"); + // timeout.tv_nsec += ts->tv_nsec; + // tmp_sec = timeout.tv_nsec / 10e9; + // timeout.tv_nsec = timeout.tv_nsec - tmp_sec * 10e9; + // timeout.tv_sec += ts->tv_sec + tmp_sec; -LoggerFactory::getLogger()->debug("==================LockFreeQueue push_timeout before\n"); - if (SemUtil::dec_timeout(slots, timeout) == -1) { - if (errno == EAGAIN) + struct timespec timeout = PXSemUtil::calc_sem_timeout(ts); + // LoggerFactory::getLogger()->debug("==================LockFreeQueue push_timeout before tv_sec=%d, tv_nsec=%ld", + // timeout.tv_sec, timeout.tv_nsec); + + while (sem_timedwait(&slots, &timeout) == -1) { + // LoggerFactory::getLogger()->debug("==================LockFreeQueue push_timeout before tv_sec=%d, tv_nsec=%ld, ETIMEDOUT=%d, errno=%d\n", + // timeout.tv_sec, timeout.tv_nsec, ETIMEDOUT, errno); + + if(errno == ETIMEDOUT) return false; + else if(errno == EINTR) + continue; else { - err_msg(errno, "LockFreeQueue push_timeout"); + LoggerFactory::getLogger()->error(errno, "LockFreeQueue push_timeout"); return false; } } if (m_qImpl.push(a_data)){ - SemUtil::inc(items); -LoggerFactory::getLogger()->debug("==================LockFreeQueue push_timeout after\n"); + sem_post(&items); +// LoggerFactory::getLogger()->debug("==================LockFreeQueue push_timeout after\n"); return true; } return false; @@ -277,19 +306,18 @@ bool LockFreeQueue<ELEM_T, Allocator, Q_TYPE>::pop(ELEM_T &a_data) { -LoggerFactory::getLogger()->debug("==================LockFreeQueue pop before\n"); - if (SemUtil::dec(items) == -1) { - err_msg(errno, "LockFreeQueue pop"); +// LoggerFactory::getLogger()->debug("==================LockFreeQueue pop before\n"); + if (sem_wait(&items) == -1) { + LoggerFactory::getLogger()->error(errno, "LockFreeQueue pop"); return false; } if (m_qImpl.pop(a_data)) { - SemUtil::inc(slots); -LoggerFactory::getLogger()->debug("==================LockFreeQueue pop after\n"); + sem_post(&slots); +// LoggerFactory::getLogger()->debug("==================LockFreeQueue pop after\n"); return true; } return false; - } template < @@ -298,21 +326,20 @@ template <typename T, typename AT> class Q_TYPE> bool LockFreeQueue<ELEM_T, Allocator, Q_TYPE>::pop_nowait(ELEM_T &a_data) { - if (SemUtil::dec_nowait(items) == -1) { + if (sem_trywait(&items) == -1) { if (errno == EAGAIN) return false; else { - err_msg(errno, "LockFreeQueue pop_nowait"); + LoggerFactory::getLogger()->error(errno, "LockFreeQueue pop_nowait"); return false; } } if (m_qImpl.pop(a_data)) { - SemUtil::inc(slots); + sem_post(&slots); return true; } return false; - } @@ -320,21 +347,26 @@ typename ELEM_T, typename Allocator, template <typename T, typename AT> class Q_TYPE> -bool LockFreeQueue<ELEM_T, Allocator, Q_TYPE>::pop_timeout(ELEM_T &a_data, struct timespec * timeout) +bool LockFreeQueue<ELEM_T, Allocator, Q_TYPE>::pop_timeout(ELEM_T &a_data, struct timespec * ts) { -LoggerFactory::getLogger()->debug("==================LockFreeQueue pop_timeout before\n"); - if (SemUtil::dec_timeout(items, timeout) == -1) { +// LoggerFactory::getLogger()->debug("==================LockFreeQueue pop_timeout before\n"); + + struct timespec timeout = PXSemUtil::calc_sem_timeout(ts); + + while (sem_timedwait(&items, &timeout) == -1) { if (errno == EAGAIN) return false; + else if(errno == EINTR) + continue; else { - // err_msg(errno, "LockFreeQueue pop_timeout"); + LoggerFactory::getLogger()->error(errno, "LockFreeQueue pop_timeout"); return false; } } if (m_qImpl.pop(a_data)) { - SemUtil::inc(slots); -LoggerFactory::getLogger()->debug("==================LockFreeQueue pop_timeout after\n"); + sem_post(&slots); +// LoggerFactory::getLogger()->debug("==================LockFreeQueue pop_timeout after\n"); return true; } return false; -- Gitblit v1.8.0