From b55ffe89f4b237be5f79232cfddfe22bfdb87c64 Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期四, 01 四月 2021 13:23:48 +0800
Subject: [PATCH] make req/rep,sub/pub sockets sub class;

---
 src/shm_queue.h |   18 ++++++++++++++++--
 1 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/shm_queue.h b/src/shm_queue.h
index 9064f55..e9b3a1a 100644
--- a/src/shm_queue.h
+++ b/src/shm_queue.h
@@ -118,6 +118,7 @@
 {
 	typedef ShmObject<SharedQueue<MsgI>> Super;
 	typedef Super::Data Queue;
+	typedef std::function<void()> OnSend;
 	bool Write(const MsgI &buf, const int timeout_ms) { return data()->Write(buf, timeout_ms); }
 	bool Read(MsgI &buf, const int timeout_ms) { return data()->Read(buf, timeout_ms); }
 	MQId id_;
@@ -132,12 +133,25 @@
 
 	bool Recv(BHMsg &msg, const int timeout_ms);
 	bool Recv(MsgI &msg, const int timeout_ms) { return Read(msg, timeout_ms); }
-	bool Send(const MQId &remote_id, const BHMsg &msg, const int timeout_ms);
-	static bool Send(SharedMemory &shm, const MQId &remote_id, const MsgI &msg, const int timeout_ms);
+	static bool Send(SharedMemory &shm, const MQId &remote_id, const MsgI &msg, const int timeout_ms, OnSend const &onsend);
+	static bool Send(SharedMemory &shm, const MQId &remote_id, const MsgI &msg, const int timeout_ms)
+	{
+		return Send(shm, remote_id, msg, timeout_ms, []() {});
+	}
+	bool Send(const MQId &remote_id, const BHMsg &msg, const int timeout_ms, OnSend const &onsend);
+	bool Send(const MQId &remote_id, const BHMsg &msg, const int timeout_ms)
+	{
+		return Send(remote_id, msg, timeout_ms, []() {});
+	}
+	bool Send(const MQId &remote_id, const MsgI &msg, const int timeout_ms, OnSend const &onsend)
+	{
+		return Send(shm(), remote_id, msg, timeout_ms, onsend);
+	}
 	bool Send(const MQId &remote_id, const MsgI &msg, const int timeout_ms)
 	{
 		return Send(shm(), remote_id, msg, timeout_ms);
 	}
+	size_t Pending() const { return data()->size(); }
 };
 
 } // namespace bhome_shm

--
Gitblit v1.8.0