mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 00:31:14 +00:00
9a0ed2ca72
To use protobuf, run cmake as follows: cmake -DUSE_PROTOBUF=1 .. For this, protobuf library has to be installed on the system. Further changes: Clean up splitter/merger: default are N-to-1-merger and 1-to-N-splitter. Fix bug in nanomsg message deallocation. Setup proper buffer sizes for nanomsg/zeromq via cmake/bash script. chmod +x for start scripts.
137 lines
2.5 KiB
C++
137 lines
2.5 KiB
C++
/**
|
|
* FairMQMessageNN.cxx
|
|
*
|
|
* @since 2013-12-05
|
|
* @author A. Rybalchenko
|
|
*/
|
|
|
|
#include <cstring>
|
|
|
|
#include <nanomsg/nn.h>
|
|
|
|
#include "FairMQMessageNN.h"
|
|
#include "FairMQLogger.h"
|
|
|
|
FairMQMessageNN::FairMQMessageNN() :
|
|
fSize(0),
|
|
fMessage(NULL),
|
|
fReceiving(false)
|
|
{
|
|
}
|
|
|
|
FairMQMessageNN::FairMQMessageNN(size_t size)
|
|
{
|
|
fMessage = nn_allocmsg(size, 0);
|
|
if (!fMessage){
|
|
LOG(ERROR) << "failed allocating message, reason: " << nn_strerror(errno);
|
|
}
|
|
fSize = size;
|
|
fReceiving = false;
|
|
}
|
|
|
|
FairMQMessageNN::FairMQMessageNN(void* data, size_t size)
|
|
{
|
|
fMessage = nn_allocmsg(size, 0);
|
|
if (!fMessage){
|
|
LOG(ERROR) << "failed allocating message, reason: " << nn_strerror(errno);
|
|
}
|
|
memcpy (fMessage, data, size);
|
|
fSize = size;
|
|
fReceiving = false;
|
|
}
|
|
|
|
void FairMQMessageNN::Rebuild()
|
|
{
|
|
Clear();
|
|
fSize = 0;
|
|
fMessage = NULL;
|
|
fReceiving = false;
|
|
}
|
|
|
|
void FairMQMessageNN::Rebuild(size_t size)
|
|
{
|
|
Clear();
|
|
fMessage = nn_allocmsg(size, 0);
|
|
if (!fMessage){
|
|
LOG(ERROR) << "failed allocating message, reason: " << nn_strerror(errno);
|
|
}
|
|
fSize = size;
|
|
fReceiving = false;
|
|
}
|
|
|
|
void FairMQMessageNN::Rebuild(void* data, size_t size)
|
|
{
|
|
Clear();
|
|
fMessage = nn_allocmsg(size, 0);
|
|
if (!fMessage){
|
|
LOG(ERROR) << "failed allocating message, reason: " << nn_strerror(errno);
|
|
}
|
|
memcpy (fMessage, data, size);
|
|
fSize = size;
|
|
fReceiving = false;
|
|
}
|
|
|
|
void* FairMQMessageNN::GetMessage()
|
|
{
|
|
return fMessage;
|
|
}
|
|
|
|
void* FairMQMessageNN::GetData()
|
|
{
|
|
return fMessage;
|
|
}
|
|
|
|
size_t FairMQMessageNN::GetSize()
|
|
{
|
|
return fSize;
|
|
}
|
|
|
|
void FairMQMessageNN::SetMessage(void* data, size_t size)
|
|
{
|
|
fMessage = data;
|
|
fSize = size;
|
|
}
|
|
|
|
void FairMQMessageNN::Copy(FairMQMessage* msg)
|
|
{
|
|
if (fMessage){
|
|
int rc = nn_freemsg(fMessage);
|
|
if ( rc < 0 ){
|
|
LOG(ERROR) << "failed freeing message, reason: " << nn_strerror(errno);
|
|
}
|
|
}
|
|
|
|
size_t size = msg->GetSize();
|
|
|
|
fMessage = nn_allocmsg(size, 0);
|
|
if (!fMessage){
|
|
LOG(ERROR) << "failed allocating message, reason: " << nn_strerror(errno);
|
|
}
|
|
std::memcpy (fMessage, msg->GetMessage(), size);
|
|
fSize = size;
|
|
}
|
|
|
|
inline void FairMQMessageNN::Clear()
|
|
{
|
|
int rc = nn_freemsg(fMessage);
|
|
if (rc < 0) {
|
|
LOG(ERROR) << "failed freeing message, reason: " << nn_strerror(errno);
|
|
} else {
|
|
fMessage = NULL;
|
|
fSize = 0;
|
|
}
|
|
}
|
|
|
|
FairMQMessageNN::~FairMQMessageNN()
|
|
{
|
|
if(fReceiving){
|
|
int rc = nn_freemsg(fMessage);
|
|
if (rc < 0) {
|
|
LOG(ERROR) << "failed freeing message, reason: " << nn_strerror(errno);
|
|
} else {
|
|
fMessage = NULL;
|
|
fSize = 0;
|
|
}
|
|
}
|
|
}
|