From db322f33ba13592f2492317e3f1a070454c97059 Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期四, 13 五月 2021 19:34:46 +0800 Subject: [PATCH] center alloc all msgs. --- src/msg.cpp | 74 ++++++++----------------------------- 1 files changed, 16 insertions(+), 58 deletions(-) diff --git a/src/msg.cpp b/src/msg.cpp index ba844da..a4777d2 100644 --- a/src/msg.cpp +++ b/src/msg.cpp @@ -17,79 +17,37 @@ */ #include "msg.h" #include "bh_util.h" +#include "socket.h" namespace bhome_msg { -/*TODO change msg format, header has proc info; -reply has errer msg - center accept request and route.; -//*/ -const uint32_t kMsgTag = 0xf1e2d3c4; -void *MsgI::Alloc(SharedMemory &shm, const size_t size) +ShmSocket &ShmMsg::Sender() { - void *p = shm.Alloc(sizeof(Meta) + size); - if (p) { - auto pmeta = new (p) Meta; - p = pmeta + 1; - } - return p; -} -void MsgI::Free(SharedMemory &shm) -{ - assert(valid()); - shm.Dealloc(meta()); - ptr_ = nullptr; - assert(!valid()); + static ShmSocket sender(shm(), false, BHGlobalSenderAddress(), 16); + return sender; } -void *MsgI::Pack(SharedMemory &shm, - const uint32_t head_len, const ToArray &headToArray, - const uint32_t body_len, const ToArray &bodyToArray) -{ - void *addr = Alloc(shm, sizeof(head_len) + head_len + sizeof(body_len) + body_len); - if (addr) { - auto p = static_cast<char *>(addr); - auto Pack1 = [&p](auto len, auto &writer) { - Put32(p, len); - p += sizeof(len); - writer(p, len); - p += len; - }; - Pack1(head_len, headToArray); - Pack1(body_len, bodyToArray); - } - return addr; -} - -bool MsgI::ParseHead(BHMsgHead &head) const -{ - auto p = get<char>(); - assert(p); - uint32_t msg_size = Get32(p); - p += 4; - return head.ParseFromArray(p, msg_size); -} - -bool MsgI::Make(SharedMemory &shm, void *p) -{ - if (!p) { - return false; - } - MsgI(p).swap(*this); - return true; -} - -int MsgI::Release(SharedMemory &shm) +int ShmMsg::Release() { if (!valid()) { return 0; } auto n = meta()->count_.Dec(); if (n == 0) { - Free(shm); + int64_t free_cmd = (id() << 4) | EncodeCmd(eCmdFree); + Sender().Send(BHTopicCenterAddress(), free_cmd); + } else if (n < 0) { + throw -123; } return n; } +void ShmMsg::Free() +{ + assert(valid()); + shm().Dealloc(meta()); + offset_ = 0; + assert(!valid()); +} } // namespace bhome_msg -- Gitblit v1.8.0