From 3c2b6739208d961cf8b86460d7f05516d044960c Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期三, 31 三月 2021 19:13:42 +0800 Subject: [PATCH] add async recv suport; sync by waiting for async. --- src/shm_queue.cpp | 36 ++++++++---------------------------- 1 files changed, 8 insertions(+), 28 deletions(-) diff --git a/src/shm_queue.cpp b/src/shm_queue.cpp index cf4c8b4..f2fb45b 100644 --- a/src/shm_queue.cpp +++ b/src/shm_queue.cpp @@ -64,23 +64,22 @@ Remove(); } -bool ShmMsgQueue::Send(SharedMemory &shm, const MQId &remote_id, const MsgI &msg, const int timeout_ms) +bool ShmMsgQueue::Send(SharedMemory &shm, const MQId &remote_id, const MsgI &msg, const int timeout_ms, OnSend const &onsend) { Queue *remote = Find(shm, MsgQIdToName(remote_id)); - return remote && remote->Write(msg, timeout_ms, [](const MsgI &msg) { msg.AddRef(); }); + return remote && remote->Write(msg, timeout_ms, [&onsend](const MsgI &msg) { onsend(); msg.AddRef(); }); } -// bool ShmMsgQueue::Send(const MQId &remote_id, const MsgI &msg, const int timeout_ms) -// { -// Queue *remote = Find(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) +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)) { + if (Send(remote_id, msg, timeout_ms, onsend)) { return true; } else { msg.Release(shm()); @@ -89,25 +88,6 @@ return false; } -/* -bool ShmMsgQueue::Send(const MQId &remote_id, const void *data, const size_t size, const int timeout_ms) -{ - // 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. - - MsgI msg; - if(msg.BuildRequest(shm(), Id(), data, size)) { - if(Send(remote_id, msg, timeout_ms)) { - return true; - } else { - msg.Release(shm()); - } - } - return false; -} -//*/ bool ShmMsgQueue::Recv(BHMsg &msg, const int timeout_ms) { MsgI imsg; -- Gitblit v1.8.0