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