From 58d904a328c0d849769b483e901a0be9426b8209 Mon Sep 17 00:00:00 2001 From: liuxiaolong <liuxiaolong@aiotlink.com> Date: 星期二, 20 七月 2021 20:20:44 +0800 Subject: [PATCH] 调整Request C.BHFree的位置 --- src/sendq.cpp | 35 ++++++++++++++++++++++++++++------- 1 files changed, 28 insertions(+), 7 deletions(-) diff --git a/src/sendq.cpp b/src/sendq.cpp index 1eaefe6..2a772b0 100644 --- a/src/sendq.cpp +++ b/src/sendq.cpp @@ -21,7 +21,26 @@ using namespace bhome_shm; -int SendQ::DoSend1Remote(ShmMsgQueue &mq, const Remote remote, Array &arr) +void SendQ::AppendData(const MQInfo &mq, const Data data, const TimePoint &expire, OnMsgEvent onExpire) +{ + TimedMsg tmp(expire, MsgInfo{mq, data, std::move(onExpire)}); + std::unique_lock<std::mutex> lock(mutex_in_); + + try { + auto &al = in_[mq.id_]; + if (!al.empty()) { + al.front().emplace_back(std::move(tmp)); + } else { + al.insert(al.begin(), Array())->emplace_back(std::move(tmp)); + } + count_in_.Count1(); + } catch (std::exception &e) { + LOG_ERROR() << "sendq error: " << e.what(); + throw e; + } +} + +int SendQ::DoSend1Remote(const Remote remote, Array &arr) { auto FirstNotExpired = [](Array &l) { auto Less = [](const TimedMsg &msg, const TimePoint &tp) { return msg.expire() < tp; }; @@ -35,8 +54,8 @@ info.on_expire_(info.data_); } } - - while (pos != arr.end() && mq.TrySend(remote, pos->data().data_)) { + auto TrySend1 = [this](MsgInfo const &info) { return ShmMsgQueue::TrySend(shm_, info.mq_, info.data_); }; + while (pos != arr.end() && TrySend1(pos->data())) { ++pos; } @@ -45,25 +64,26 @@ return nprocessed; } -int SendQ::DoSend1Remote(ShmMsgQueue &mq, const Remote remote, ArrayList &al) +int SendQ::DoSend1Remote(const Remote remote, ArrayList &al) { int nsend = 0; auto AllSent = [&](Array &arr) { - nsend += DoSend1Remote(mq, remote, arr); + nsend += DoSend1Remote(remote, arr); return arr.empty(); }; for (auto it = al.begin(); it != al.end() && AllSent(*it); it = al.erase(it)) {} return nsend; } -bool SendQ::TrySend(ShmMsgQueue &mq) +bool SendQ::TrySend() { std::unique_lock<std::mutex> lock(mutex_out_); + size_t nsend = 0; if (!out_.empty()) { auto rec = out_.begin(); do { - nsend += DoSend1Remote(mq, rec->first, rec->second); + nsend += DoSend1Remote(rec->first, rec->second); if (rec->second.empty()) { rec = out_.erase(rec); } else { @@ -71,6 +91,7 @@ } } while (rec != out_.end()); } + count_out_.Count(nsend); auto Collect = [&]() { std::unique_lock<std::mutex> lock(mutex_in_); -- Gitblit v1.8.0