move ref count into msg meta, only 1 poinetr now.

lichao

2021-04-23 dc86ace85e437ecb8a2e728e4dce36d02bbb8a6e

 src/msg.cpp

@@ -16,63 +16,80 @@
 * =====================================================================================
 */
#include "msg.h"
#include "bh_util.h"

namespace bhome_shm {
	

bool MsgMetaV1::Parse(const void *p)
namespace bhome_msg
{
    assert(p);
    *this = *static_cast<const MsgMetaV1*>(p);
    return tag_ == kMsgMetaTag;
/*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)
{
   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());
}

void MsgMetaV1::Pack(void *p)
void *MsgI::Pack(SharedMemory &shm,
                 const uint32_t head_len, const ToArray &headToArray,
                 const uint32_t body_len, const ToArray &bodyToArray)
{
    *static_cast<MsgMetaV1*>(p) = *this;
   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 Msg::Build(SharedMemory &shm, const MQId &src_id, const void *data, const size_t size, const bool refcount)
bool MsgI::ParseHead(BHMsgHead &head) const
{
    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;

   auto p = get<char>();
   assert(p);
   uint32_t msg_size = Get32(p);
   p += 4;
   return head.ParseFromArray(p, msg_size);
}

int Msg::Release(SharedMemory &shm) 
bool MsgI::Make(SharedMemory &shm, void *p)
{
    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;
   if (!p) {
      return false;
   }
   MsgI(p).swap(*this);
   return true;
}

} // namespace bhome_shm
int MsgI::Release(SharedMemory &shm)
{
   if (!valid()) {
      return 0;
   }
   auto n = meta()->count_.Dec();
   if (n == 0) {
      Free(shm);
   }
   return n;
}

} // namespace bhome_msg