From fa95ddd1a294ebad47cabf9e149ee7d789271044 Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期三, 14 四月 2021 19:45:09 +0800 Subject: [PATCH] use 2 buf to speed up sendq; socket auto start. --- src/shm_queue.h | 19 ++++++++++++++++++- 1 files changed, 18 insertions(+), 1 deletions(-) diff --git a/src/shm_queue.h b/src/shm_queue.h index 2d5ec3a..4f544c8 100644 --- a/src/shm_queue.h +++ b/src/shm_queue.h @@ -174,14 +174,31 @@ bool TryRecv(MsgI &msg) { return data()->TryRead(msg); } template <class OnData> int TryRecvAll(OnData const &onData) { return data()->TryReadAll(onData); } - + static Queue *FindRemote(SharedMemory &shm, const MQId &remote_id); static bool Send(SharedMemory &shm, const MQId &remote_id, const MsgI &msg, const int timeout_ms, OnSend const &onsend = OnSend()); static bool TrySend(SharedMemory &shm, const MQId &remote_id, const MsgI &msg, OnSend const &onsend = OnSend()); + template <class Iter> + static int TrySendAll(SharedMemory &shm, const MQId &remote_id, const Iter begin, const Iter end, OnSend const &onsend = OnSend()) + { + Queue *remote = FindRemote(shm, remote_id); + if (remote) { + if (onsend) { + return remote->TryWrite(begin, end, [&onsend](const MsgI &msg) { onsend(); msg.AddRef(); }); + } else { + return remote->TryWrite(begin, end, [](const MsgI &msg) { msg.AddRef(); }); + } + } else { + // SetLestError(eNotFound); + return 0; + } + } template <class... Rest> bool Send(const MQId &remote_id, Rest const &...rest) { return Send(shm(), remote_id, rest...); } template <class... Rest> bool TrySend(const MQId &remote_id, Rest const &...rest) { return TrySend(shm(), remote_id, rest...); } + template <class... Rest> + int TrySendAll(const MQId &remote_id, Rest const &...rest) { return TrySendAll(shm(), remote_id, rest...); } size_t Pending() const { return data()->size(); } }; -- Gitblit v1.8.0