From 9bf199a4770b08c03d553129757d960b605e598a Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期五, 14 五月 2021 18:05:21 +0800 Subject: [PATCH] add center info at fixed address in shm. --- src/msg.cpp | 33 ++++++++++++++++++++++++++++++++- 1 files changed, 32 insertions(+), 1 deletions(-) diff --git a/src/msg.cpp b/src/msg.cpp index 7ab0434..edffff1 100644 --- a/src/msg.cpp +++ b/src/msg.cpp @@ -17,9 +17,40 @@ */ #include "msg.h" #include "bh_util.h" +#include "defs.h" +#include "socket.h" namespace bhome_msg { -const uint32_t kMsgTag = 0xf1e2d3c4; +ShmSocket &ShmMsg::Sender() +{ + static auto &mq = GetCenterInfo(shm())->mq_sender_; + static ShmSocket sender(mq.offset_, shm(), mq.id_); + return sender; +} + +int ShmMsg::Release() +{ + if (!valid()) { + return 0; + } + auto n = meta()->count_.Dec(); + if (n == 0) { + int64_t free_cmd = (id() << 4) | EncodeCmd(eCmdFree); + Sender().Send(BHTopicCenterAddress(), free_cmd); + } else if (n < 0) { + LOG_FATAL() << "error double release data."; + throw std::runtime_error("double release msg."); + } + return n; +} + +void ShmMsg::Free() +{ + assert(valid()); + shm().Dealloc(meta()); + offset_ = 0; + assert(!valid()); +} } // namespace bhome_msg -- Gitblit v1.8.0