From 101b5cf85397ef9350aaedd12cfcf9fd3d07a565 Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期四, 20 五月 2021 12:41:51 +0800
Subject: [PATCH] refactor node center.
---
src/msg.cpp | 82 ++++++++++++++++-------------------------
1 files changed, 32 insertions(+), 50 deletions(-)
diff --git a/src/msg.cpp b/src/msg.cpp
index bb193ec..dca2044 100644
--- a/src/msg.cpp
+++ b/src/msg.cpp
@@ -16,63 +16,45 @@
* =====================================================================================
*/
#include "msg.h"
+#include "bh_util.h"
+#include "defs.h"
+#include "shm_socket.h"
-namespace bhome_msg {
-
-
-bool MsgMetaV1::Parse(const void *p)
+namespace bhome_msg
{
- assert(p);
- *this = *static_cast<const MsgMetaV1*>(p);
- return tag_ == kMsgMetaTag;
+
+ShmSocket &ShmMsg::Sender()
+{
+ static auto &mq = GetCenterInfo(shm())->mq_sender_;
+ static ShmSocket sender(mq.offset_, shm(), mq.id_);
+ return sender;
}
-void MsgMetaV1::Pack(void *p)
+int ShmMsg::Release()
{
- *static_cast<MsgMetaV1*>(p) = *this;
+ 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;
}
-bool Msg::Build(SharedMemory &shm, const MQId &src_id, const void *data, const size_t size, const bool refcount)
+void ShmMsg::Free()
{
- 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;
-
+ assert(valid());
+ shm().Dealloc(meta());
+ offset_ = 0;
+ assert(!valid());
}
-
-int Msg::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