add api BHQueryProcs.

lichao

2021-05-19 34cd75f77d0ca94dbdba4e6cc9451fe4d33e78b3

 src/msg.cpp

@@ -17,147 +17,44 @@
 */
#include "msg.h"
#include "bh_util.h"
#include "defs.h"
#include "socket.h"

namespace bhome_msg
{

const uint32_t kMsgTag = 0xf1e2d3c4;
const uint32_t kMsgPrefixLen = 4;

BHMsg InitMsg(MsgType type)
ShmSocket &ShmMsg::Sender()
{
   BHMsg msg;
   msg.set_type(type);
   time_t tm = 0;
   msg.set_timestamp(time(&tm));
   return msg;
   static auto &mq = GetCenterInfo(shm())->mq_sender_;
   static ShmSocket sender(mq.offset_, shm(), mq.id_);
   return sender;
}

BHMsg MakeRequest(const MQId &src_id, const void *data, const size_t size)
int ShmMsg::Release()
{
   assert(data && size);
   BHMsg msg(InitMsg(kMsgTypeRequest));
   msg.set_body(data, size);
   msg.add_route()->set_mq_id(&src_id, sizeof(src_id));
   return msg;
}
BHMsg MakeRequest(const MQId &src_id, const std::string &topic, const void *data, const size_t size)
{
   DataRequest req;
   req.set_topic(topic);
   req.set_data(data, size);
   const std::string &body(req.SerializeAsString());
   return MakeRequest(src_id, body.data(), body.size());
}

BHMsg MakeReply(const void *data, const size_t size)
{
   assert(data && size);
   BHMsg msg(InitMsg(kMsgTypeReply));
   DataReply reply;
   reply.set_data(data, size);
   msg.set_body(reply.SerializeAsString());
   return msg;
}

BHMsg MakeSubUnsub(const MQId &client, const std::vector<std::string> &topics, const MsgType sub_unsub)
{
   assert(sub_unsub == kMsgTypeSubscribe || sub_unsub == kMsgTypeUnsubscribe);
   BHMsg msg(InitMsg(sub_unsub));
   msg.add_route()->set_mq_id(&client, sizeof(client));
   DataSub subs;
   for (auto &t : topics) {
      subs.add_topics(t);
   if (!valid()) {
      return 0;
   }
   msg.set_body(subs.SerializeAsString());
   return msg;
}

BHMsg MakeSub(const MQId &client, const std::vector<std::string> &topics) { return MakeSubUnsub(client, topics, kMsgTypeSubscribe); }
BHMsg MakeUnsub(const MQId &client, const std::vector<std::string> &topics) { return MakeSubUnsub(client, topics, kMsgTypeUnsubscribe); }

BHMsg MakePub(const std::string &topic, const void *data, const size_t size)
{
   assert(data && size);
   BHMsg msg(InitMsg(kMsgTypePublish));
   DataPub pub;
   pub.set_topic(topic);
   pub.set_data(data, size);
   msg.set_body(pub.SerializeAsString());
   return msg;
}

BHMsg MakeQueryTopic(const std::string &topic)
{
   BHMsg msg(InitMsg(kMsgTypeQueryTopic));
   DataQueryTopic query;
   query.set_topic(topic);
   msg.set_body(query.SerializeAsString());
   return msg;
}

void *Pack(SharedMemory &shm, const BHMsg &msg)
{
   uint32_t msg_size = msg.ByteSizeLong();
   void *p = shm.Alloc(4 + msg_size);
   if (p) {
      Put32(p, msg_size);
      if (!msg.SerializeToArray(static_cast<char *>(p) + kMsgPrefixLen, msg_size)) {
         shm.Dealloc(p);
         p = 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 p;
   return n;
}

bool MsgI::Unpack(BHMsg &msg) const
void ShmMsg::Free()
{
   void *p = ptr_.get();
   assert(p);
   uint32_t msg_size = Get32(p);
   return msg.ParseFromArray(static_cast<char *>(p) + kMsgPrefixLen, msg_size);
   assert(valid());
   shm().Dealloc(meta());
   offset_ = 0;
   assert(!valid());
}

// with ref count;
bool MsgI::MakeRC(SharedMemory &shm, const BHMsg &msg)
{
   void *p = Pack(shm, msg);
   if (!p) {
      return false;
   }
   RefCount *rc = shm.New<RefCount>();
   if (!rc) {
      shm.Dealloc(p);
      return false;
   }
   MsgI(p, rc).swap(*this);
   return true;
}

bool MsgI::Make(SharedMemory &shm, const BHMsg &msg)
{
   void *p = Pack(shm, msg);
   if (!p) {
      return false;
   }
   MsgI(p, 0).swap(*this);
   return true;
}

int MsgI::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_msg