From 43d4e95770b0519341153202c9a535aaa8e164c5 Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期二, 01 六月 2021 14:22:20 +0800 Subject: [PATCH] refactor, remove useless code. --- src/robust.h | 33 ++----- /dev/null | 24 ------ src/shm_msg_queue.h | 6 src/shm_queue.h | 70 +---------------- utest/robust_test.cpp | 66 ---------------- src/robust.cpp | 9 ++ 6 files changed, 26 insertions(+), 182 deletions(-) diff --git a/src/robust.cpp b/src/robust.cpp index 015f97a..b4e0613 100644 --- a/src/robust.cpp +++ b/src/robust.cpp @@ -16,9 +16,16 @@ * ===================================================================================== */ #include "robust.h" -#include <chrono> #include <thread> +using namespace std::chrono; +using namespace std::chrono_literals; + +namespace +{ +void yield() { std::this_thread::sleep_for(10us); } +} // namespace + namespace robust { diff --git a/src/robust.h b/src/robust.h index 1bbe8fc..62eb0b4 100644 --- a/src/robust.h +++ b/src/robust.h @@ -23,23 +23,11 @@ #include "log.h" #include <atomic> #include <chrono> -#include <memory> -#include <mutex> -#include <string> -#include <sys/file.h> -#include <sys/ipc.h> -#include <sys/sem.h> -#include <sys/stat.h> -#include <sys/types.h> -#include <thread> #include <unistd.h> namespace robust { - -using namespace std::chrono; -using namespace std::chrono_literals; - +/* template <unsigned PowerSize = 4, class Int = int64_t> class AtomicQueue { @@ -100,16 +88,13 @@ std::atomic<size_type> tail_; AData buf[capacity]; }; +//*/ -template <class Int> -class AtomicQueue<0, Int> +class AtomicQ63 { - typedef Int Data; - typedef std::atomic<Data> AData; - static_assert(sizeof(Data) == sizeof(AData)); - public: - AtomicQueue() { memset(this, 0, sizeof(*this)); } + typedef int64_t Data; + AtomicQ63() { memset(this, 0, sizeof(*this)); } bool push(const Data d, bool try_more = false) { auto cur = buf.load(); @@ -122,13 +107,15 @@ if (r) { d = Dec(cur); } return r; } - uint32_t head() const { return 0; } - uint32_t tail() const { return 0; } private: static inline bool Empty(const Data d) { return (d & 1) == 0; } // lowest bit 1 means data ok. static inline Data Enc(const Data d) { return (d << 1) | 1; } // lowest bit 1 means data ok. static inline Data Dec(const Data d) { return d >> 1; } // lowest bit 1 means data ok. + + typedef std::atomic<Data> AData; + static_assert(sizeof(Data) == sizeof(AData)); + AData buf; }; @@ -149,7 +136,7 @@ static int GetState(Data d) { return d & MaskBits(3); } static Data Encode(Data d, State st) { return (d << 3) | st; } static Data Decode(Data d) { return d >> 3; } - static void yield() { std::this_thread::sleep_for(10us); } + typedef std::chrono::steady_clock steady_clock; typedef steady_clock::duration Duration; Duration now() { return steady_clock::now().time_since_epoch(); } diff --git a/src/shm_msg_queue.h b/src/shm_msg_queue.h index 23faa24..db5414b 100644 --- a/src/shm_msg_queue.h +++ b/src/shm_msg_queue.h @@ -29,7 +29,7 @@ { public: typedef int64_t RawData; - typedef ShmObject<SharedQ63<0>> Shmq; + typedef ShmObject<SharedQ63> Shmq; typedef Shmq::Data Queue; typedef Shmq::ShmType ShmType; typedef uint64_t MQId; @@ -43,10 +43,10 @@ ShmType &shm() const { return queue_.shm(); } int64_t AbsAddr() const { return queue_.offset(); } - bool Recv(RawData &val, const int timeout_ms) { return queue().Read(val, timeout_ms); } + // bool Recv(RawData &val, const int timeout_ms) { return queue().Read(val, timeout_ms); } bool TryRecv(RawData &val) { return queue().TryRead(val); } - bool Recv(MsgI &msg, const int timeout_ms) { return Recv(msg.OffsetRef(), timeout_ms); } + // bool Recv(MsgI &msg, const int timeout_ms) { return Recv(msg.OffsetRef(), timeout_ms); } bool TryRecv(MsgI &msg) { return TryRecv(msg.OffsetRef()); } static Queue *Find(ShmType &shm, const MQId remote); static bool TrySend(ShmType &shm, const MQInfo &remote, const RawData val); diff --git a/src/shm_queue.cpp b/src/shm_queue.cpp deleted file mode 100644 index 86f0d91..0000000 --- a/src/shm_queue.cpp +++ /dev/null @@ -1,24 +0,0 @@ -/* - * ===================================================================================== - * - * Filename: shm_queue.cpp - * - * Description: - * - * Version: 1.0 - * Created: 2021骞�03鏈�25鏃� 10鏃�34鍒�42绉� - * Revision: none - * Compiler: gcc - * - * Author: Li Chao (), - * Organization: - * - * ===================================================================================== - */ - -#include "shm_queue.h" -#include "bh_util.h" - -namespace bhome_shm -{ -} // namespace bhome_shm diff --git a/src/shm_queue.h b/src/shm_queue.h index 3a2cea6..952dae9 100644 --- a/src/shm_queue.h +++ b/src/shm_queue.h @@ -28,82 +28,20 @@ namespace bhome_shm { -template <class D> -using Circular = boost::circular_buffer<D, Allocator<D>>; - -template <class D> -class SharedQueue -{ -public: - SharedQueue(const uint32_t len, Allocator<D> const &alloc) : - queue_(len, alloc) {} - - bool Read(D &d, const int timeout_ms) - { - using namespace std::chrono; - auto end_time = steady_clock::now() + milliseconds(timeout_ms); - do { - if (TryRead(d)) { - return true; - } else { - std::this_thread::sleep_for(1ms); - } - } while (steady_clock::now() < end_time); - return false; - } - bool TryRead(D &d) - { - // bhome_shm::Guard lock(mutex_); - if (!queue_.empty()) { - d = queue_.front(); - queue_.pop_front(); - return true; - } else { - return false; - } - } - bool TryWrite(const D &d) - { - // bhome_shm::Guard lock(mutex_); - if (!queue_.full()) { - queue_.push_back(d); - return true; - } else { - return false; - } - } - -private: - Circular<D> queue_; -}; - -template <int Power = 4> +// just wrap robust::AtomicQ63 class SharedQ63 { public: template <class... T> explicit SharedQ63(T &&...t) {} // easy testing - typedef int64_t Data; - bool Read(Data &d, const int timeout_ms) - { - using namespace std::chrono; - auto end_time = steady_clock::now() + milliseconds(timeout_ms); - do { - for (int i = 0; i < 100; ++i) { - if (TryRead(d)) { - return true; - } - } - std::this_thread::sleep_for(1ms); - } while (steady_clock::now() < end_time); - return false; - } + typedef robust::AtomicQ63 AQ63; + typedef AQ63::Data Data; bool TryRead(Data &d, const bool try_more = true) { return queue_.pop(d, try_more); } bool TryWrite(const Data d, const bool try_more = true) { return queue_.push(d, try_more); } private: - robust::AtomicQueue<Power, Data> queue_; + AQ63 queue_; }; } // namespace bhome_shm diff --git a/utest/robust_test.cpp b/utest/robust_test.cpp index 3270481..a744a61 100644 --- a/utest/robust_test.cpp +++ b/utest/robust_test.cpp @@ -68,24 +68,12 @@ std::atomic<uint64_t> nwrite(0); std::atomic<uint64_t> writedone(0); -#if 1 - const int kPower = 0; - typedef AtomicQueue<kPower> Rcb; + typedef AtomicQ63 Rcb; Rcb tmp; - // BOOST_CHECK(tmp.like_empty()); BOOST_CHECK(tmp.push(1)); - if (kPower != 0) { - BOOST_CHECK(tmp.tail() == 1); - } - BOOST_CHECK(tmp.head() == 0); int64_t d; BOOST_CHECK(tmp.pop(d)); - if (kPower != 0) { - // BOOST_CHECK(tmp.like_empty()); - BOOST_CHECK(tmp.head() == 1); - BOOST_CHECK(tmp.tail() == 1); - } ShmObject<Rcb> rcb(shm, "test_rcb"); bool try_more = true; @@ -111,58 +99,6 @@ } } }; - -#else - typedef Circular<int64_t> Rcb; - ShmObject<Rcb> rcb(shm, "test_rcb", 16, shm.get_segment_manager()); - - typedef FMutex Mutex; - // typedef SemMutex Mutex; - Mutex mtx(123); - auto Writer = [&]() { - uint64_t n = 0; - while ((n = nwrite++) < nmsg) { - auto Write = [&]() { - robust::Guard<Mutex> lk(mtx); - if (rcb->full()) { - return false; - } else { - rcb->push_back(n); - return true; - } - // return rcb->push_back(n); - }; - while (!Write()) { - // MySleep(); - } - ++writedone; - } - }; - std::atomic<uint64_t> nread(0); - auto Reader = [&]() { - while (nread.load() < nmsg) { - int64_t d; - auto Read = [&]() { - robust::Guard<Mutex> lk(mtx); - if (rcb->empty()) { - return false; - } else { - d = rcb->front(); - rcb->pop_front(); - return true; - } - // return rcb->pop_front(d); - }; - if (Read()) { - ++nread; - total += d; - } else { - // MySleep(); - } - } - }; - -#endif auto status = [&]() { auto next = steady_clock::now(); -- Gitblit v1.8.0