From cab831748a2a9cc18b7f18f3b5e14a4374b7ab68 Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期一, 17 五月 2021 18:34:26 +0800
Subject: [PATCH] socket send using abs addr, avoid shm find by id.

---
 src/sendq.cpp |   33 ++++++++++++++++++++++++++++++++-
 1 files changed, 32 insertions(+), 1 deletions(-)

diff --git a/src/sendq.cpp b/src/sendq.cpp
index 1eaefe6..f1e5918 100644
--- a/src/sendq.cpp
+++ b/src/sendq.cpp
@@ -21,6 +21,24 @@
 
 using namespace bhome_shm;
 
+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));
+		}
+	} catch (std::exception &e) {
+		LOG_ERROR() << "sendq error: " << e.what();
+		throw e;
+	}
+}
+
 int SendQ::DoSend1Remote(ShmMsgQueue &mq, const Remote remote, Array &arr)
 {
 	auto FirstNotExpired = [](Array &l) {
@@ -36,7 +54,7 @@
 		}
 	}
 
-	while (pos != arr.end() && mq.TrySend(remote, pos->data().data_)) {
+	while (pos != arr.end() && mq.TrySend(pos->data().mq_, pos->data().data_)) {
 		++pos;
 	}
 
@@ -59,6 +77,8 @@
 bool SendQ::TrySend(ShmMsgQueue &mq)
 {
 	std::unique_lock<std::mutex> lock(mutex_out_);
+	// if (TooFast()) { return false; }
+
 	size_t nsend = 0;
 	if (!out_.empty()) {
 		auto rec = out_.begin();
@@ -89,3 +109,14 @@
 
 	return !out_.empty();
 }
+
+bool SendQ::TooFast()
+{
+	auto cur = NowSec();
+	if (cur > last_time_) {
+		last_time_ = cur;
+		count_ = 0;
+	}
+
+	return ++count_ > 1000 * 100;
+} // not accurate in multi-thread.
\ No newline at end of file

--
Gitblit v1.8.0