From ad4f3dcedab29a690c5eedbb08ba1b393917db0b Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期三, 21 四月 2021 17:39:34 +0800
Subject: [PATCH] update go api.
---
src/sendq.h | 54 ++++++++++++++++++++++++++++++++++++++++--------------
1 files changed, 40 insertions(+), 14 deletions(-)
diff --git a/src/sendq.h b/src/sendq.h
index c6f270b..bba44af 100644
--- a/src/sendq.h
+++ b/src/sendq.h
@@ -21,8 +21,11 @@
#include "defs.h"
#include "msg.h"
#include "timed_queue.h"
+#include <boost/variant2/variant.hpp>
#include <deque>
#include <functional>
+#include <list>
+#include <mutex>
#include <string>
#include <unordered_map>
@@ -36,39 +39,62 @@
public:
typedef std::string Remote;
typedef bhome_msg::MsgI MsgI;
- typedef std::function<void(const MsgI &msg)> OnMsgEvent;
+ typedef std::string Content;
+ typedef boost::variant2::variant<MsgI, Content> Data;
+ typedef std::function<void(const Data &)> OnMsgEvent;
struct MsgInfo {
- MsgI msg_;
+ Data data_;
OnMsgEvent on_expire_;
- // OnMsgEvent on_send_;
};
typedef TimedData<MsgInfo> TimedMsg;
typedef TimedMsg::TimePoint TimePoint;
typedef TimedMsg::Duration Duration;
- void Append(const MQId &id, const MsgI &msg, OnMsgEvent onExpire = OnMsgEvent())
+ template <class... Rest>
+ void Append(const MQId &id, Rest &&...rest)
{
- Append(std::string((const char *) &id, sizeof(id)), msg, onExpire);
+ Append(std::string((const char *) &id, sizeof(id)), std::forward<decltype(rest)>(rest)...);
}
+
void Append(const Remote &addr, const MsgI &msg, OnMsgEvent onExpire = OnMsgEvent())
{
- using namespace std::chrono_literals;
- Append(addr, msg, Now() + 60s, onExpire);
+ msg.AddRef();
+ AppendData(addr, Data(msg), DefaultExpire(), onExpire);
+ }
+ void Append(const Remote &addr, Content &&content, OnMsgEvent onExpire = OnMsgEvent())
+ {
+ AppendData(addr, Data(std::move(content)), DefaultExpire(), onExpire);
}
bool TrySend(bhome_shm::ShmMsgQueue &mq);
// bool empty() const { return store_.empty(); }
private:
static TimePoint Now() { return TimedMsg::Clock::now(); }
- void Append(const Remote &addr, const MsgI &msg, const TimePoint &expire, OnMsgEvent onExpire)
+ static TimePoint DefaultExpire() { return Now() + std::chrono::seconds(60); }
+ void AppendData(const Remote &addr, Data &&data, const TimePoint &expire, OnMsgEvent onExpire)
{
- msg.AddRef();
- store_[addr].emplace_back(TimedMsg(expire, MsgInfo{msg, onExpire}));
- }
- typedef std::deque<TimedMsg> MsgList;
- typedef std::unordered_map<Remote, MsgList> Store;
+ //TODO simple queue, organize later ?
- Store store_;
+ TimedMsg tmp(expire, MsgInfo{std::move(data), std::move(onExpire)});
+ std::unique_lock<std::mutex> lock(mutex_in_);
+ auto &al = in_[addr];
+ if (!al.empty()) {
+ al.front().emplace_back(std::move(tmp));
+ } else {
+ al.insert(al.begin(), Array())->emplace_back(std::move(tmp));
+ }
+ }
+ typedef std::deque<TimedMsg> Array;
+ typedef std::list<Array> ArrayList;
+ typedef std::unordered_map<Remote, ArrayList> Store;
+
+ int DoSend1Remote(bhome_shm::ShmMsgQueue &mq, const Remote &remote, Array &arr);
+ int DoSend1Remote(bhome_shm::ShmMsgQueue &mq, const Remote &remote, ArrayList &arr);
+
+ std::mutex mutex_in_;
+ std::mutex mutex_out_;
+ Store in_;
+ Store out_;
};
#endif // end of include guard: SENDQ_IWKMSK7M
--
Gitblit v1.8.0