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