From 8967e7f2f8b94dc032135707e16c8a9f233d0db6 Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期四, 03 六月 2021 13:32:16 +0800 Subject: [PATCH] rafactor, remove old todo, add some err msg. --- src/msg.cpp | 84 +++++++++++++++--------------------------- 1 files changed, 30 insertions(+), 54 deletions(-) diff --git a/src/msg.cpp b/src/msg.cpp index 9883246..3546424 100644 --- a/src/msg.cpp +++ b/src/msg.cpp @@ -16,63 +16,39 @@ * ===================================================================================== */ #include "msg.h" +#include "bh_util.h" +#include "defs.h" +#include "shm_socket.h" -namespace bhome_shm { - - -bool MsgMetaV1::Parse(const void *p) +namespace bhome_msg { - assert(p); - *this = *static_cast<const MsgMetaV1*>(p); - return tag_ == kMsgMetaTag; + +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); + DefaultSender(shm()).Send(BHTopicCenterAddress(shm()), free_cmd); + } else { + Free(); + } + } else if (n < 0) { + // ns_log::GetTrace(); + LOG_FATAL() << "double release msg."; + throw std::runtime_error("double release msg."); + } + return n; } -void MsgMetaV1::Pack(void *p) +void ShmMsg::Free() { - *static_cast<MsgMetaV1*>(p) = *this; + assert(valid()); + shm().Dealloc(meta()); + offset_ = 0; + assert(!valid()); } - -bool Msg::Build(SharedMemory &shm, const MQId &src_id, const void *data, const size_t size, const bool refcount) -{ - 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; - -} - -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_shm +} // namespace bhome_msg -- Gitblit v1.8.0