调整Request C.BHFree的位置

liuxiaolong

2021-07-20 58d904a328c0d849769b483e901a0be9426b8209

 src/msg.h

@@ -22,8 +22,6 @@
#include "proto.h"
#include "shm.h"
#include <atomic>
#include <boost/interprocess/offset_ptr.hpp>
#include <functional>
#include <stdint.h>

class ShmSocket;
@@ -44,7 +42,8 @@

   public:
      RefCount() :
          num_(1) { static_assert(std::is_pod<decltype(num_)>::value); }
          num_(1) { }
          // num_(1) { static_assert(std::is_pod<decltype(num_)>::value); }
      int Inc() { return ++num_; }
      int Dec() { return --num_; }
      int Get() { return num_.load(); }
@@ -75,20 +74,22 @@
   };
   OffsetType offset_;
   SharedMemory *pshm_;
   void *Alloc(const size_t size)

   void *Alloc(const size_t size, const void *src = nullptr)
   {
      void *p = shm().Alloc(sizeof(Meta) + size);
      if (p) {
         auto pmeta = new (p) Meta(size);
         p = pmeta + 1;
         if (src) {
            memcpy(p, src, size);
         }
      }
      return p;
   }

private:
   Meta *meta() const { return get<Meta>() - 1; }

   typedef std::function<void(void *p, int len)> ToArray;

   template <class Body>
   void *Pack(const BHMsgHead &head, const uint32_t head_len, const Body &body, const uint32_t body_len)
@@ -108,16 +109,35 @@
      }
      return addr;
   }

   void *Pack(const std::string &content)
   void *Pack(const BHMsgHead &head, const uint32_t head_len, const std::string &body_content)
   {
      void *addr = get();
      if (addr) {
         memcpy(addr, content.data(), content.size());
         meta()->size_ = content.size();
         auto p = static_cast<char *>(addr);
         auto Pack1 = [&p](uint32_t len, auto &&writer) {
            Put32(p, len);
            p += sizeof(len);
            writer(p, len);
            p += len;
         };
         Pack1(head_len, [&](void *p, int len) { head.SerializeToArray(p, len); });
         Pack1(body_content.size(), [&](void *p, int len) { memcpy(p, body_content.data(), len); });
         meta()->size_ = 4 + head_len + 4 + body_content.size();
      }
      return addr;
   }

   void *Pack(const void *src, const size_t size)
   {
      void *addr = get();
      if (addr && src) {
         memcpy(addr, src, size);
         meta()->size_ = size;
      }
      return addr;
   }

   void *Pack(const std::string &content) { return Pack(content.data(), content.size()); }

   bool Make(void *addr)
   {
@@ -166,6 +186,13 @@
      uint32_t size = sizeof(head_len) + head_len + sizeof(body_len) + body_len;
      return Make(size) && Pack(head, head_len, body, body_len);
   }
   inline bool Make(const BHMsgHead &head, const std::string &body_content)
   {
      uint32_t head_len = head.ByteSizeLong();
      uint32_t size = sizeof(head_len) + head_len + sizeof(uint32_t) + body_content.size();
      return Make(size) && Pack(head, head_len, body_content);
   }

   template <class Body>
   inline bool Fill(const BHMsgHead &head, const Body &body)
   {
@@ -175,8 +202,11 @@
      return valid() && (meta()->capacity_ >= size) && Pack(head, head_len, body, body_len);
   }

   inline bool Make(const std::string &content) { return Make(content.size()) && Pack(content); }
   inline bool Fill(const std::string &content) { return valid() && (meta()->capacity_ >= content.size()) && Pack(content); }
   inline bool Make(const void *src, const size_t size) { return Make(Alloc(size, src)); }
   inline bool Fill(const void *src, const size_t size) { return valid() && (meta()->capacity_ >= size) && Pack(src, size); }

   inline bool Make(const std::string &content) { return Make(content.data(), content.size()); }
   inline bool Fill(const std::string &content) { return Fill(content.data(), content.size()); }

   inline bool Make(const size_t size) { return Make(Alloc(size)); }