From 34cd75f77d0ca94dbdba4e6cc9451fe4d33e78b3 Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期三, 19 五月 2021 19:14:13 +0800
Subject: [PATCH] add api BHQueryProcs.
---
src/msg.cpp | 145 ++++++++++--------------------------------------
1 files changed, 31 insertions(+), 114 deletions(-)
diff --git a/src/msg.cpp b/src/msg.cpp
index ed8adba..c9ed60a 100644
--- a/src/msg.cpp
+++ b/src/msg.cpp
@@ -17,127 +17,44 @@
*/
#include "msg.h"
#include "bh_util.h"
+#include "defs.h"
+#include "socket.h"
-namespace bhome_msg {
-
-const uint32_t kMsgTag = 0xf1e2d3c4;
-const uint32_t kMsgPrefixLen = 4;
-
-BHMsg InitMsg(MsgType type)
+namespace bhome_msg
{
- BHMsg msg;
- msg.set_type(type);
- time_t tm = 0;
- msg.set_timestamp(time(&tm));
- return msg;
+
+ShmSocket &ShmMsg::Sender()
+{
+ static auto &mq = GetCenterInfo(shm())->mq_sender_;
+ static ShmSocket sender(mq.offset_, shm(), mq.id_);
+ return sender;
}
-BHMsg MakeRequest(const MQId &src_id, const void *data, const size_t size)
+int ShmMsg::Release()
{
- assert(data && size);
- BHMsg msg(InitMsg(kMsgTypeRequest));
- msg.set_body(data, size);
- BHAddress addr;
- msg.add_route()->set_mq_id(&src_id, sizeof(src_id));
- return msg;
+ if (!valid()) {
+ return 0;
+ }
+ auto n = meta()->count_.Dec();
+ if (n == 0) {
+ if (meta()->managed_) {
+ int64_t free_cmd = (id() << 4) | EncodeCmd(eCmdFree);
+ Sender().Send(BHTopicCenterAddress(), free_cmd);
+ } else {
+ Free();
+ }
+ } else if (n < 0) {
+ LOG_FATAL() << "error double release data.";
+ throw std::runtime_error("double release msg.");
+ }
+ return n;
}
-BHMsg MakeReply(const void *data, const size_t size)
+void ShmMsg::Free()
{
- assert(data && size);
- BHMsg msg(InitMsg(kMsgTypeReply));
- msg.set_body(data, size);
- return msg;
+ assert(valid());
+ shm().Dealloc(meta());
+ offset_ = 0;
+ assert(!valid());
}
-
-BHMsg MakeSubUnsub(const std::vector<std::string> &topics, const MsgType sub_unsub)
-{
- assert(sub_unsub == kMsgTypeSubscribe || sub_unsub == kMsgTypeUnsubscribe);
- BHMsg msg(InitMsg(sub_unsub));
- DataSub subs;
- for (auto &t : topics) {
- subs.add_topics(t);
- }
- msg.set_body(subs.SerializeAsString());
- return msg;
-}
-
-BHMsg MakeSub(const std::vector<std::string> &topics) { return MakeSubUnsub(topics, kMsgTypeSubscribe); }
-BHMsg MakeUnsub(const std::vector<std::string> &topics) { return MakeSubUnsub(topics, kMsgTypeUnsubscribe); }
-
-BHMsg MakePub(const std::string &topic, const void *data, const size_t size)
-{
- assert(data && size);
- BHMsg msg(InitMsg(kMsgTypePublish));
- DataPub pub;
- pub.set_topic(topic);
- pub.set_data(data, size);
- msg.set_body(pub.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