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 | 36 ++++++++++++++++++++++++++++++++++++ 1 files changed, 36 insertions(+), 0 deletions(-) diff --git a/src/msg.cpp b/src/msg.cpp index f180d67..c9ed60a 100644 --- a/src/msg.cpp +++ b/src/msg.cpp @@ -17,8 +17,44 @@ */ #include "msg.h" #include "bh_util.h" +#include "defs.h" +#include "socket.h" namespace bhome_msg { +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) { + 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; +} + +void ShmMsg::Free() +{ + assert(valid()); + shm().Dealloc(meta()); + offset_ = 0; + assert(!valid()); +} } // namespace bhome_msg -- Gitblit v1.8.0