/*
|
* =====================================================================================
|
*
|
* Filename: msg.cpp
|
*
|
* Description:
|
*
|
* Version: 1.0
|
* Created: 2021年03月24日 16时48分42秒
|
* Revision: none
|
* Compiler: gcc
|
*
|
* Author: Li Chao (),
|
* Organization:
|
*
|
* =====================================================================================
|
*/
|
#include "msg.h"
|
#include "bh_util.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::Pack(SharedMemory &shm,
|
const uint32_t head_len, const ToArray &headToArray,
|
const uint32_t body_len, const ToArray &bodyToArray)
|
{
|
void *addr = shm.Alloc(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 = static_cast<char *>(ptr_.get());
|
assert(p);
|
uint32_t msg_size = Get32(p);
|
p += 4;
|
return head.ParseFromArray(p, msg_size);
|
}
|
|
// with ref count;
|
bool MsgI::MakeRC(SharedMemory &shm, void *p)
|
{
|
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, void *p)
|
{
|
if (!p) {
|
return false;
|
}
|
MsgI(p, 0).swap(*this);
|
return true;
|
}
|
|
bool MsgI::EnableRefCount(SharedMemory &shm)
|
{
|
if (!IsCounted()) {
|
count_ = shm.New<RefCount>();
|
}
|
return IsCounted();
|
}
|
|
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
|
