From 77a6c3512a44dfe6540dde71946e6484fe4f173f Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期一, 10 五月 2021 16:05:28 +0800
Subject: [PATCH] test lock code.

---
 src/sendq.cpp |   99 +++++++++++++++++++++++++++++++++++++------------
 1 files changed, 75 insertions(+), 24 deletions(-)

diff --git a/src/sendq.cpp b/src/sendq.cpp
index 290e9d1..c0d5afd 100644
--- a/src/sendq.cpp
+++ b/src/sendq.cpp
@@ -16,47 +16,98 @@
  * =====================================================================================
  */
 #include "sendq.h"
-#include "shm_queue.h"
+#include "shm_msg_queue.h"
 #include <chrono>
 
-bool SendQ::TrySend(bhome_shm::ShmMsgQueue &mq)
+using namespace bhome_shm;
+
+int SendQ::DoSend1Remote(ShmMsgQueue &mq, const Remote remote, Array &arr)
 {
-	auto FirstNotExpired = [](MsgList &l) {
+	auto FirstNotExpired = [](Array &l) {
 		auto Less = [](const TimedMsg &msg, const TimePoint &tp) { return msg.expire() < tp; };
 		return std::lower_bound(l.begin(), l.end(), Now(), Less);
 	};
 
-	auto SendOneRemote = [&](const Remote &remote, MsgList &msg_list) {
-		auto pos = FirstNotExpired(msg_list);
-		for (auto it = msg_list.begin(); it != pos; ++it) {
-			auto &info = it->data();
-			if (info.on_expire_) {
-				info.on_expire_(info.msg_);
-			}
-			info.msg_.Release(mq.shm());
+	auto pos = FirstNotExpired(arr);
+	for (auto it = arr.begin(); it != pos; ++it) {
+		auto &info = it->data();
+		if (info.on_expire_) {
+			info.on_expire_(info.data_);
 		}
+		if (info.data_.index() == 0) {
+			boost::variant2::get<0>(info.data_).Release();
+		}
+	}
 
-		//TODO maybe use TrySendAll ?
-		while (pos != msg_list.end() && mq.TrySend(*(MQId *) remote.data(), pos->data().msg_)) {
-			auto &msg = pos->data().msg_;
-			if (msg.IsCounted()) {
-				msg.Release(mq.shm());
+	auto SendData = [&](Data &d) {
+		bool r = false;
+		if (d.index() == 0) {
+			auto &msg = boost::variant2::get<0>(pos->data().data_);
+			r = mq.TrySend(remote, msg);
+			if (r) {
+				msg.Release();
 			}
-			++pos;
+		} else {
+			auto &content = boost::variant2::get<1>(pos->data().data_);
+			MsgI msg;
+			if (msg.Make(content)) {
+				DEFER1(msg.Release(););
+				r = mq.TrySend(remote, msg);
+			}
 		}
-		msg_list.erase(msg_list.begin(), pos);
+		return r;
 	};
 
-	if (!store_.empty()) {
-		auto rec = store_.begin();
+	while (pos != arr.end() && SendData(pos->data().data_)) {
+		++pos;
+	}
+
+	int nprocessed = std::distance(arr.begin(), pos);
+	arr.erase(arr.begin(), pos);
+	return nprocessed;
+}
+
+int SendQ::DoSend1Remote(ShmMsgQueue &mq, const Remote remote, ArrayList &al)
+{
+	int nsend = 0;
+	auto AllSent = [&](Array &arr) {
+		nsend += DoSend1Remote(mq, 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)
+{
+	std::unique_lock<std::mutex> lock(mutex_out_);
+	size_t nsend = 0;
+	if (!out_.empty()) {
+		auto rec = out_.begin();
 		do {
-			SendOneRemote(rec->first, rec->second);
+			nsend += DoSend1Remote(mq, rec->first, rec->second);
 			if (rec->second.empty()) {
-				rec = store_.erase(rec);
+				rec = out_.erase(rec);
 			} else {
 				++rec;
 			}
-		} while (rec != store_.end());
+		} while (rec != out_.end());
 	}
-	return !store_.empty();
+
+	auto Collect = [&]() {
+		std::unique_lock<std::mutex> lock(mutex_in_);
+		if (out_.empty()) {
+			out_.swap(in_);
+		} else if (nsend == 0) { // remote blocked
+			for (auto &kv : in_) {
+				auto &to = out_[kv.first];
+				to.splice(to.end(), kv.second);
+			}
+			in_.clear();
+		}
+	};
+
+	Collect();
+
+	return !out_.empty();
 }
\ No newline at end of file

--
Gitblit v1.8.0