From c338820e4db43ad32c20ff429a038b06bcb980f8 Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期四, 08 四月 2021 18:13:25 +0800 Subject: [PATCH] BIG change, join center,bus; now msg is head+body. --- src/shm_queue.cpp | 46 ++++++++++++++++++++++------------------------ 1 files changed, 22 insertions(+), 24 deletions(-) diff --git a/src/shm_queue.cpp b/src/shm_queue.cpp index f2fb45b..521f773 100644 --- a/src/shm_queue.cpp +++ b/src/shm_queue.cpp @@ -21,6 +21,11 @@ #include <boost/uuid/uuid_generators.hpp> #include <boost/uuid/uuid_io.hpp> +// TODO !!! add garbage collection, and handshake with gc-center. +// smart pointer will NOT work if user crash. +// problem of process release mq: client query success; server exits; client send. +// client query success; server exits; client send. + namespace bhome_shm { using namespace bhome_msg; @@ -61,7 +66,9 @@ ShmMsgQueue::~ShmMsgQueue() { - Remove(); + // It's not safe to remove, others may still holder pointers and write to it. + // TODO use smart_ptr or garbage collection. + //Remove(); } bool ShmMsgQueue::Send(SharedMemory &shm, const MQId &remote_id, const MsgI &msg, const int timeout_ms, OnSend const &onsend) @@ -69,34 +76,25 @@ Queue *remote = Find(shm, MsgQIdToName(remote_id)); return remote && remote->Write(msg, timeout_ms, [&onsend](const MsgI &msg) { onsend(); msg.AddRef(); }); } +bool ShmMsgQueue::Send(SharedMemory &shm, const MQId &remote_id, const MsgI &msg, const int timeout_ms) +{ + Queue *remote = Find(shm, MsgQIdToName(remote_id)); + return remote && remote->Write(msg, timeout_ms, [](const MsgI &msg) { msg.AddRef(); }); +} // Test shows that in the 2 cases: // 1) build msg first, then find remote queue; // 2) find remote queue first, then build msg; // 1 is about 50% faster than 2, maybe cache related. -bool ShmMsgQueue::Send(const MQId &remote_id, const BHMsg &data, const int timeout_ms, const std::function<void()> &onsend) -{ - MsgI msg; - if (msg.Make(shm(), data)) { - if (Send(remote_id, msg, timeout_ms, onsend)) { - return true; - } else { - msg.Release(shm()); - } - } - return false; -} - -bool ShmMsgQueue::Recv(BHMsg &msg, const int timeout_ms) -{ - MsgI imsg; - if (Read(imsg, timeout_ms)) { - DEFER1(imsg.Release(shm());); - return imsg.Unpack(msg); - } else { - return false; - } -} +// bool ShmMsgQueue::Recv(MsgI &imsg, BHMsgHead &head, const int timeout_ms) +// { +// if (Read(imsg, timeout_ms)) { +// // DEFER1(imsg.Release(shm());); +// return imsg.ParseHead(head); +// } else { +// return false; +// } +// } } // namespace bhome_shm -- Gitblit v1.8.0