From 4deeafbd502dc3c57dab8ad6ca601a38a9e7f074 Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期二, 06 四月 2021 19:10:49 +0800
Subject: [PATCH] add uni center.
---
src/msg.cpp | 214 ++++++++++++++++++++++++++++++++++++++++++-----------
1 files changed, 168 insertions(+), 46 deletions(-)
diff --git a/src/msg.cpp b/src/msg.cpp
index bb193ec..8752066 100644
--- a/src/msg.cpp
+++ b/src/msg.cpp
@@ -16,63 +16,185 @@
* =====================================================================================
*/
#include "msg.h"
+#include "bh_util.h"
-namespace bhome_msg {
-
-
-bool MsgMetaV1::Parse(const void *p)
+namespace bhome_msg
{
- assert(p);
- *this = *static_cast<const MsgMetaV1*>(p);
- return tag_ == kMsgMetaTag;
+/*TODO change msg format, header has proc info;
+reply has errer msg
+ center accept request and route.;
+//*/
+const uint32_t kMsgTag = 0xf1e2d3c4;
+const uint32_t kMsgPrefixLen = 4;
+
+inline void AddRoute(BHMsg &msg, const MQId &id) { msg.add_route()->set_mq_id(&id, sizeof(id)); }
+
+std::string RandId()
+{
+ boost::uuids::uuid id = boost::uuids::random_generator()();
+ return std::string((char *) &id, sizeof(id));
+}
+BHMsg InitMsg(MsgType type, const std::string &msgid = RandId())
+{
+ BHMsg msg;
+ msg.set_msg_id(msgid);
+ msg.set_type(type);
+ time_t tm = 0;
+ msg.set_timestamp(time(&tm));
+ return msg;
}
-void MsgMetaV1::Pack(void *p)
+BHMsg MakeRequest(const MQId &src_id, const std::string &topic, const void *data, const size_t size)
{
- *static_cast<MsgMetaV1*>(p) = *this;
+ BHMsg msg(InitMsg(kMsgTypeRequestTopic));
+ AddRoute(msg, src_id);
+ MsgRequestTopic req;
+ req.set_topic(topic);
+ req.set_data(data, size);
+ msg.set_body(req.SerializeAsString());
+ return msg;
}
-bool Msg::Build(SharedMemory &shm, const MQId &src_id, const void *data, const size_t size, const bool refcount)
+BHMsg MakeRegister(const MQId &src_id, ProcInfo info, const std::vector<std::string> &topics)
{
- if (!data || !size) {
- return false;
- }
- void *p = shm.Alloc(sizeof(MsgMetaV1) + size);
- if (!p) {
- return false;
- }
- RefCount *rc = 0;
- if (refcount) {
- rc = shm.New<RefCount>();
- if (!rc) {
- shm.Dealloc(p);
- return false;
- }
- }
- MsgMetaV1 meta;
- meta.data_size_ = size;
- meta.src_id_ = src_id;
- meta.Pack(p);
- memcpy(static_cast<char *>(p) + sizeof(meta), data, size);
- Msg(p, rc).swap(*this);
- return true;
-
+ BHMsg msg(InitMsg(kMsgTypeRegister));
+ AddRoute(msg, src_id);
+ MsgRegister reg;
+ reg.mutable_proc()->Swap(&info);
+ for (auto &t : topics) {
+ reg.add_topics(t);
+ }
+ msg.set_body(reg.SerializeAsString());
+ return msg;
}
-int Msg::Release(SharedMemory &shm)
+BHMsg MakeHeartbeat(const MQId &src_id, ProcInfo info)
{
- if (IsCounted()) {
- const int n = count_->Dec();
- if (n != 0) {
- return n;
- }
- }
- // free data
- shm.Dealloc(ptr_);
- ptr_ = 0;
- shm.Delete(count_);
- count_ = 0;
- return 0;
+ BHMsg msg(InitMsg(kMsgTypeHeartbeat));
+ AddRoute(msg, src_id);
+ MsgHeartbeat reg;
+ reg.mutable_proc()->Swap(&info);
+ msg.set_body(reg.SerializeAsString());
+ return msg;
+}
+
+BHMsg MakeReply(const std::string &src_msgid, const void *data, const size_t size)
+{
+ assert(data && size);
+ BHMsg msg(InitMsg(kMsgTypeRequestTopicReply, src_msgid));
+ MsgRequestTopicReply reply;
+ reply.set_data(data, size);
+ msg.set_body(reply.SerializeAsString());
+ return msg;
+}
+
+BHMsg MakeSubUnsub(const MQId &client, const std::vector<std::string> &topics, const MsgType sub_unsub)
+{
+ assert(sub_unsub == kMsgTypeSubscribe || sub_unsub == kMsgTypeUnsubscribe);
+ BHMsg msg(InitMsg(sub_unsub));
+ AddRoute(msg, client);
+ MsgSub subs;
+ for (auto &t : topics) {
+ subs.add_topics(t);
+ }
+ msg.set_body(subs.SerializeAsString());
+ return msg;
+}
+
+BHMsg MakeSub(const MQId &client, const std::vector<std::string> &topics) { return MakeSubUnsub(client, topics, kMsgTypeSubscribe); }
+BHMsg MakeUnsub(const MQId &client, const std::vector<std::string> &topics) { return MakeSubUnsub(client, topics, kMsgTypeUnsubscribe); }
+
+BHMsg MakePub(const std::string &topic, const void *data, const size_t size)
+{
+ assert(data && size);
+ BHMsg msg(InitMsg(kMsgTypePublish));
+ MsgPub pub;
+ pub.set_topic(topic);
+ pub.set_data(data, size);
+ msg.set_body(pub.SerializeAsString());
+ return msg;
+}
+
+BHMsg MakeQueryTopic(const MQId &client, const std::string &topic)
+{
+ BHMsg msg(InitMsg(kMsgTypeQueryTopic));
+ AddRoute(msg, client);
+ MsgQueryTopic query;
+ query.set_topic(topic);
+ msg.set_body(query.SerializeAsString());
+ return msg;
+}
+BHMsg MakeQueryTopicReply(const std::string &mqid, const std::string &msgid)
+{
+ BHMsg msg(InitMsg(kMsgTypeQueryTopicReply, msgid));
+ MsgQueryTopicReply reply;
+ reply.mutable_address()->set_mq_id(mqid);
+ msg.set_body(reply.SerializeAsString());
+ return msg;
+}
+
+void *Pack(SharedMemory &shm, const BHMsg &msg)
+{
+ uint32_t msg_size = msg.ByteSizeLong();
+ void *p = shm.Alloc(4 + msg_size);
+ if (p) {
+ Put32(p, msg_size);
+ if (!msg.SerializeToArray(static_cast<char *>(p) + kMsgPrefixLen, msg_size)) {
+ shm.Dealloc(p);
+ p = 0;
+ }
+ }
+ return p;
+}
+
+bool MsgI::Unpack(BHMsg &msg) const
+{
+ void *p = ptr_.get();
+ assert(p);
+ uint32_t msg_size = Get32(p);
+ return msg.ParseFromArray(static_cast<char *>(p) + kMsgPrefixLen, msg_size);
+}
+
+// with ref count;
+bool MsgI::MakeRC(SharedMemory &shm, const BHMsg &msg)
+{
+ void *p = Pack(shm, msg);
+ if (!p) {
+ return false;
+ }
+ RefCount *rc = shm.New<RefCount>();
+ if (!rc) {
+ shm.Dealloc(p);
+ return false;
+ }
+ MsgI(p, rc).swap(*this);
+ return true;
+}
+
+bool MsgI::Make(SharedMemory &shm, const BHMsg &msg)
+{
+ void *p = Pack(shm, msg);
+ if (!p) {
+ return false;
+ }
+ MsgI(p, 0).swap(*this);
+ return true;
+}
+
+int MsgI::Release(SharedMemory &shm)
+{
+ if (IsCounted()) {
+ const int n = count_->Dec();
+ if (n != 0) {
+ return n;
+ }
+ }
+ // free data
+ shm.Dealloc(ptr_);
+ ptr_ = 0;
+ shm.Delete(count_);
+ count_ = 0;
+ return 0;
}
} // namespace bhome_msg
--
Gitblit v1.8.0