mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 16:46:47 +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.
149 lines
3.2 KiB
C++
149 lines
3.2 KiB
C++
/**
|
|
* FairMQSocketNN.cxx
|
|
*
|
|
* @since 2012-12-05
|
|
* @author A. Rybalchenko
|
|
*/
|
|
|
|
#include <sstream>
|
|
|
|
#include "FairMQSocketNN.h"
|
|
#include "FairMQMessageNN.h"
|
|
#include "FairMQLogger.h"
|
|
|
|
FairMQSocketNN::FairMQSocketNN(const string& type, int num) :
|
|
fBytesTx(0),
|
|
fBytesRx(0),
|
|
fMessagesTx(0),
|
|
fMessagesRx(0)
|
|
{
|
|
stringstream id;
|
|
id << type << "." << num;
|
|
fId = id.str();
|
|
|
|
fSocket = nn_socket (AF_SP, GetConstant(type));
|
|
if (type == "sub") {
|
|
nn_setsockopt(fSocket, NN_SUB, NN_SUB_SUBSCRIBE, NULL, 0);
|
|
}
|
|
|
|
LOG(INFO) << "created socket #" << fId;
|
|
}
|
|
|
|
string FairMQSocketNN::GetId()
|
|
{
|
|
return fId;
|
|
}
|
|
|
|
void FairMQSocketNN::Bind(const string& address)
|
|
{
|
|
LOG(INFO) << "bind socket #" << fId << " on " << address;
|
|
|
|
int eid = nn_bind(fSocket, address.c_str());
|
|
if (eid < 0) {
|
|
LOG(ERROR) << "failed binding socket #" << fId << ", reason: " << nn_strerror(errno);
|
|
}
|
|
}
|
|
|
|
void FairMQSocketNN::Connect(const string& address)
|
|
{
|
|
LOG(INFO) << "connect socket #" << fId << " to " << address;
|
|
|
|
int eid = nn_connect(fSocket, address.c_str());
|
|
if (eid < 0) {
|
|
LOG(ERROR) << "failed connecting socket #" << fId << ", reason: " << nn_strerror(errno);
|
|
}
|
|
}
|
|
|
|
size_t FairMQSocketNN::Send(FairMQMessage* msg)
|
|
{
|
|
void* ptr = msg->GetMessage();
|
|
int rc = nn_send(fSocket, &ptr, NN_MSG, 0);
|
|
if (rc < 0) {
|
|
LOG(ERROR) << "failed sending on socket #" << fId << ", reason: " << nn_strerror(errno);
|
|
} else {
|
|
fBytesTx += rc;
|
|
++fMessagesTx;
|
|
static_cast<FairMQMessageNN*>(msg)->fReceiving = false;
|
|
}
|
|
|
|
return rc;
|
|
}
|
|
|
|
size_t FairMQSocketNN::Receive(FairMQMessage* msg)
|
|
{
|
|
void* ptr = NULL;
|
|
int rc = nn_recv(fSocket, &ptr, NN_MSG, 0);
|
|
if (rc < 0) {
|
|
LOG(ERROR) << "failed receiving on socket #" << fId << ", reason: " << nn_strerror(errno);
|
|
} else {
|
|
fBytesRx += rc;
|
|
++fMessagesRx;
|
|
msg->SetMessage(ptr, rc);
|
|
static_cast<FairMQMessageNN*>(msg)->fReceiving = true;
|
|
}
|
|
|
|
return rc;
|
|
}
|
|
|
|
void* FairMQSocketNN::GetSocket()
|
|
{
|
|
return NULL; // dummy method to comply with the interface. functionality not possible in zeromq.
|
|
}
|
|
|
|
int FairMQSocketNN::GetSocket(int nothing)
|
|
{
|
|
return fSocket;
|
|
}
|
|
|
|
void FairMQSocketNN::Close()
|
|
{
|
|
nn_close(fSocket);
|
|
}
|
|
|
|
void FairMQSocketNN::SetOption(const string& option, const void* value, size_t valueSize)
|
|
{
|
|
int rc = nn_setsockopt(fSocket, NN_SOL_SOCKET, GetConstant(option), value, valueSize);
|
|
if (rc < 0) {
|
|
LOG(ERROR) << "failed setting socket option, reason: " << nn_strerror(errno);
|
|
}
|
|
}
|
|
|
|
unsigned long FairMQSocketNN::GetBytesTx()
|
|
{
|
|
return fBytesTx;
|
|
}
|
|
|
|
unsigned long FairMQSocketNN::GetBytesRx()
|
|
{
|
|
return fBytesRx;
|
|
}
|
|
|
|
unsigned long FairMQSocketNN::GetMessagesTx()
|
|
{
|
|
return fMessagesTx;
|
|
}
|
|
|
|
unsigned long FairMQSocketNN::GetMessagesRx()
|
|
{
|
|
return fMessagesRx;
|
|
}
|
|
|
|
int FairMQSocketNN::GetConstant(const string& constant)
|
|
{
|
|
if (constant == "sub") return NN_SUB;
|
|
if (constant == "pub") return NN_PUB;
|
|
if (constant == "xsub") return NN_SUB; // TODO: is there XPUB, XSUB for nanomsg?
|
|
if (constant == "xpub") return NN_PUB;
|
|
if (constant == "push") return NN_PUSH;
|
|
if (constant == "pull") return NN_PULL;
|
|
if (constant == "snd-hwm") return NN_SNDBUF;
|
|
if (constant == "rcv-hwm") return NN_RCVBUF;
|
|
|
|
return -1;
|
|
}
|
|
|
|
FairMQSocketNN::~FairMQSocketNN()
|
|
{
|
|
Close();
|
|
}
|