use clang-format for FairMQ

This commit is contained in:
Alexey Rybalchenko 2014-04-10 15:20:48 +02:00
parent e80e6d4269
commit 68d51d8ed5
55 changed files with 1893 additions and 1759 deletions

54
fairmq/.clang-format Normal file
View File

@ -0,0 +1,54 @@
---
Language: Cpp
#AccessModifierOffset: -4
ConstructorInitializerIndentWidth: 4
AlignEscapedNewlinesLeft: true
AlignTrailingComments: true
AllowAllParametersOfDeclarationOnNextLine: false
AllowShortIfStatementsOnASingleLine: false
AllowShortLoopsOnASingleLine: false
AllowShortFunctionsOnASingleLine: false
AlwaysBreakTemplateDeclarations: true
# It is broken on windows. Breaks all #include "header.h"
#AlwaysBreakBeforeMultilineStrings: true
BreakBeforeBinaryOperators: false
BreakBeforeTernaryOperators: true
BreakConstructorInitializersBeforeComma: true
BinPackParameters: false
ColumnLimit: 160
ConstructorInitializerAllOnOneLineOrOnePerLine: false
ConstructorInitializerIndentWidth: 4
DerivePointerBinding: false
ExperimentalAutoDetectBinPacking: false
IndentCaseLabels: true
MaxEmptyLinesToKeep: 1
NamespaceIndentation: All
ObjCSpaceAfterProperty: false
ObjCSpaceBeforeProtocolList: false
PenaltyBreakBeforeFirstCallParameter: 1
PenaltyBreakComment: 300
PenaltyBreakString: 1000
PenaltyBreakFirstLessLess: 120
PenaltyExcessCharacter: 1000000
PenaltyReturnTypeOnItsOwnLine: 200
PointerBindsToType: true
SpacesBeforeTrailingComments: 1
Cpp11BracedListStyle: false
Standard: Cpp11
IndentWidth: 4
TabWidth: 4
UseTab: Never
BreakBeforeBraces: Allman
IndentFunctionDeclarationAfterType: true
SpacesInParentheses: false
SpacesInAngles: false
SpaceInEmptyParentheses: false
SpacesInCStyleCastParentheses: false
SpacesInContainerLiterals: true
SpaceBeforeAssignmentOperators: true
ContinuationIndentWidth: 4
CommentPragmas: '^ IWYU pragma:'
SpaceBeforeParens: ControlStatements
...

View File

@ -13,11 +13,10 @@
#include "FairMQBenchmarkSampler.h"
#include "FairMQLogger.h"
FairMQBenchmarkSampler::FairMQBenchmarkSampler() :
fEventSize(10000),
fEventRate(1),
fEventCounter(0)
FairMQBenchmarkSampler::FairMQBenchmarkSampler()
: fEventSize(10000)
, fEventRate(1)
, fEventCounter(0)
{
}
@ -41,7 +40,8 @@ void FairMQBenchmarkSampler::Run()
void* buffer = operator new[](fEventSize);
FairMQMessage* base_msg = fTransportFactory->CreateMessage(buffer, fEventSize);
while ( fState == RUNNING ) {
while (fState == RUNNING)
{
FairMQMessage* msg = fTransportFactory->CreateMessage();
msg->Copy(base_msg);
@ -49,7 +49,8 @@ void FairMQBenchmarkSampler::Run()
--fEventCounter;
while (fEventCounter == 0) {
while (fEventCounter == 0)
{
boost::this_thread::sleep(boost::posix_time::milliseconds(1));
}
@ -67,11 +68,15 @@ void FairMQBenchmarkSampler::Run()
void FairMQBenchmarkSampler::ResetEventCounter()
{
while ( true ) {
try {
while (true)
{
try
{
fEventCounter = fEventRate / 100;
boost::this_thread::sleep(boost::posix_time::milliseconds(10));
} catch (boost::thread_interrupted&) {
}
catch (boost::thread_interrupted&)
{
break;
}
}
@ -90,7 +95,8 @@ void FairMQBenchmarkSampler::Log(int intervalInMs)
t0 = get_timestamp();
while (true) {
while (true)
{
boost::this_thread::sleep(boost::posix_time::milliseconds(intervalInMs));
t1 = get_timestamp();
@ -113,7 +119,8 @@ void FairMQBenchmarkSampler::Log(int intervalInMs)
void FairMQBenchmarkSampler::SetProperty(const int key, const string& value, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
default:
FairMQDevice::SetProperty(key, value, slot);
break;
@ -122,7 +129,8 @@ void FairMQBenchmarkSampler::SetProperty(const int key, const string& value, con
string FairMQBenchmarkSampler::GetProperty(const int key, const string& default_ /*= ""*/, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
default:
return FairMQDevice::GetProperty(key, default_, slot);
}
@ -130,7 +138,8 @@ string FairMQBenchmarkSampler::GetProperty(const int key, const string& default_
void FairMQBenchmarkSampler::SetProperty(const int key, const int value, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
case EventSize:
fEventSize = value;
break;
@ -145,7 +154,8 @@ void FairMQBenchmarkSampler::SetProperty(const int key, const int value, const i
int FairMQBenchmarkSampler::GetProperty(const int key, const int default_ /*= 0*/, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
case EventSize:
return fEventSize;
case EventRate:

View File

@ -19,7 +19,8 @@
class FairMQBenchmarkSampler : public FairMQDevice
{
public:
enum {
enum
{
InputFile = FairMQDevice::Last,
EventRate,
EventSize,

View File

@ -13,7 +13,6 @@
#include "FairMQBuffer.h"
#include "FairMQLogger.h"
FairMQBuffer::FairMQBuffer()
{
}
@ -25,12 +24,14 @@ void FairMQBuffer::Run()
boost::thread rateLogger(boost::bind(&FairMQDevice::LogSocketRates, this));
bool received = false;
while ( fState == RUNNING ) {
while (fState == RUNNING)
{
FairMQMessage* msg = fTransportFactory->CreateMessage();
received = fPayloadInputs->at(0)->Receive(msg);
if (received) {
if (received)
{
fPayloadOutputs->at(0)->Send(msg);
received = false;
}
@ -45,4 +46,3 @@ void FairMQBuffer::Run()
FairMQBuffer::~FairMQBuffer()
{
}

View File

@ -10,12 +10,12 @@
#include "FairMQDevice.h"
class FairMQBuffer : public FairMQDevice
{
public:
FairMQBuffer();
virtual ~FairMQBuffer();
protected:
virtual void Run();
};

View File

@ -7,7 +7,6 @@
#include "FairMQConfigurable.h"
FairMQConfigurable::FairMQConfigurable()
{
}

View File

@ -15,7 +15,8 @@ using std::string;
class FairMQConfigurable
{
public:
enum {
enum
{
Last = 1
};
FairMQConfigurable();

View File

@ -11,12 +11,13 @@
#include "FairMQDevice.h"
#include "FairMQLogger.h"
FairMQDevice::FairMQDevice() :
fNumIoThreads(1),
FairMQDevice::FairMQDevice()
: fNumIoThreads(1)
,
// fPayloadContext(NULL),
fPayloadInputs(new vector<FairMQSocket*>()),
fPayloadOutputs(new vector<FairMQSocket*>()),
fLogIntervalInMs(1000)
fPayloadInputs(new vector<FairMQSocket*>())
, fPayloadOutputs(new vector<FairMQSocket*>())
, fLogIntervalInMs(1000)
{
}
@ -73,7 +74,6 @@ void FairMQDevice::InitInput()
} catch (std::out_of_range& e) {
}
}
}
void FairMQDevice::InitOutput()
{
@ -96,7 +96,6 @@ void FairMQDevice::InitOutput()
} catch (std::out_of_range& e) {
}
}
}
void FairMQDevice::Run()
{
@ -109,7 +108,8 @@ void FairMQDevice::Pause()
// Method for setting properties represented as a string.
void FairMQDevice::SetProperty(const int key, const string& value, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
case Id:
fId = value;
break;
@ -146,7 +146,8 @@ void FairMQDevice::SetProperty(const int key, const string& value, const int slo
// Method for setting properties represented as an integer.
void FairMQDevice::SetProperty(const int key, const int value, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
case NumIoThreads:
fNumIoThreads = value;
break;
@ -184,7 +185,8 @@ void FairMQDevice::SetProperty(const int key, const int value, const int slot/*=
// Method for getting properties represented as an string.
string FairMQDevice::GetProperty(const int key, const string& default_ /*= ""*/, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
case Id:
return fId;
case InputAddress:
@ -207,7 +209,8 @@ string FairMQDevice::GetProperty(const int key, const string& default_/*= ""*/,
// Method for getting properties represented as an integer.
int FairMQDevice::GetProperty(const int key, const int default_ /*= 0*/, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
case NumIoThreads:
return fNumIoThreads;
case LogIntervalInMs:
@ -260,14 +263,16 @@ void FairMQDevice::LogSocketRates()
// End of temp stuff
int i = 0;
for ( vector<FairMQSocket*>::iterator itr = fPayloadInputs->begin(); itr != fPayloadInputs->end(); itr++ ) {
for (vector<FairMQSocket*>::iterator itr = fPayloadInputs->begin(); itr != fPayloadInputs->end(); itr++)
{
bytesInput[i] = (*itr)->GetBytesRx();
messagesInput[i] = (*itr)->GetMessagesRx();
++i;
}
i = 0;
for ( vector<FairMQSocket*>::iterator itr = fPayloadOutputs->begin(); itr != fPayloadOutputs->end(); itr++ ) {
for (vector<FairMQSocket*>::iterator itr = fPayloadOutputs->begin(); itr != fPayloadOutputs->end(); itr++)
{
bytesOutput[i] = (*itr)->GetBytesTx();
messagesOutput[i] = (*itr)->GetMessagesTx();
++i;
@ -275,15 +280,18 @@ void FairMQDevice::LogSocketRates()
t0 = get_timestamp();
while ( true ) {
try {
while (true)
{
try
{
t1 = get_timestamp();
timeSinceLastLog_ms = (t1 - t0) / 1000.0L;
i = 0;
for ( vector<FairMQSocket*>::iterator itr = fPayloadInputs->begin(); itr != fPayloadInputs->end(); itr++ ) {
for (vector<FairMQSocket*>::iterator itr = fPayloadInputs->begin(); itr != fPayloadInputs->end(); itr++)
{
bytesInputNew[i] = (*itr)->GetBytesRx();
megabytesPerSecondInput[i] = ((double)(bytesInputNew[i] - bytesInput[i]) / (1024. * 1024.)) / (double)timeSinceLastLog_ms * 1000.;
bytesInput[i] = bytesInputNew[i];
@ -318,7 +326,8 @@ void FairMQDevice::LogSocketRates()
messagesPerSecondOutput[i] = (double)(messagesOutputNew[i] - messagesOutput[i]) / (double)timeSinceLastLog_ms * 1000.;
messagesOutput[i] = messagesOutputNew[i];
LOG(DEBUG) << "#" << fId << "." << (*itr)->GetId() << ": " << messagesPerSecondOutput[i] << " msg/s, " << megabytesPerSecondOutput[i] << " MB/s";
LOG(DEBUG) << "#" << fId << "." << (*itr)->GetId() << ": " << messagesPerSecondOutput[i] << " msg/s, " << megabytesPerSecondOutput[i]
<< " MB/s";
// Temp stuff for process termination
// if ( !sentSomething && messagesPerSecondOutput[i] > 0 ) {
@ -347,7 +356,9 @@ void FairMQDevice::LogSocketRates()
t0 = t1;
boost::this_thread::sleep(boost::posix_time::milliseconds(fLogIntervalInMs));
} catch (boost::thread_interrupted&) {
}
catch (boost::thread_interrupted&)
{
cout << "rateLogger interrupted" << endl;
break;
}
@ -378,12 +389,14 @@ void FairMQDevice::ListenToCommands()
void FairMQDevice::Shutdown()
{
LOG(INFO) << ">>>>>>> closing inputs <<<<<<<";
for( vector<FairMQSocket*>::iterator itr = fPayloadInputs->begin(); itr != fPayloadInputs->end(); itr++ ) {
for (vector<FairMQSocket*>::iterator itr = fPayloadInputs->begin(); itr != fPayloadInputs->end(); itr++)
{
(*itr)->Close();
}
LOG(INFO) << ">>>>>>> closing outputs <<<<<<<";
for( vector<FairMQSocket*>::iterator itr = fPayloadOutputs->begin(); itr != fPayloadOutputs->end(); itr++ ) {
for (vector<FairMQSocket*>::iterator itr = fPayloadOutputs->begin(); itr != fPayloadOutputs->end(); itr++)
{
(*itr)->Close();
}
@ -393,11 +406,13 @@ void FairMQDevice::Shutdown()
FairMQDevice::~FairMQDevice()
{
for( vector<FairMQSocket*>::iterator itr = fPayloadInputs->begin(); itr != fPayloadInputs->end(); itr++ ) {
for (vector<FairMQSocket*>::iterator itr = fPayloadInputs->begin(); itr != fPayloadInputs->end(); itr++)
{
delete (*itr);
}
for( vector<FairMQSocket*>::iterator itr = fPayloadOutputs->begin(); itr != fPayloadOutputs->end(); itr++ ) {
for (vector<FairMQSocket*>::iterator itr = fPayloadOutputs->begin(); itr != fPayloadOutputs->end(); itr++)
{
delete (*itr);
}
@ -406,4 +421,3 @@ FairMQDevice::~FairMQDevice()
delete fPayloadInputs;
delete fPayloadOutputs;
}

View File

@ -25,7 +25,8 @@ using std::endl;
class FairMQDevice : public FairMQStateMachine, public FairMQConfigurable
{
public:
enum {
enum
{
Id = FairMQConfigurable::Last,
NumIoThreads,
NumInputs,

View File

@ -34,7 +34,8 @@ std::ostringstream& FairMQLogger::Log(int type)
std::strftime(mbstr, 100, "%H:%M:%S", std::localtime(&t));
string type_str;
switch (type) {
switch (type)
{
case DEBUG:
type_str = "\033[01;34mDEBUG\033[0m";
break;
@ -50,7 +51,9 @@ std::ostringstream& FairMQLogger::Log(int type)
break;
}
os << "[\033[01;36m" << mbstr << "\033[0m]" << "[" << type_str << "]" << " ";
os << "[\033[01;36m" << mbstr << "\033[0m]"
<< "[" << type_str << "]"
<< " ";
return os;
}

View File

@ -21,12 +21,17 @@ using std::ostringstream;
class FairMQLogger
{
public:
enum {
DEBUG, INFO, ERROR, STATE
enum
{
DEBUG,
INFO,
ERROR,
STATE
};
FairMQLogger();
virtual ~FairMQLogger();
ostringstream& Log(int type);
private:
ostringstream os;
};
@ -35,7 +40,6 @@ typedef unsigned long long timestamp_t;
timestamp_t get_timestamp();
#define LOG(type) \
FairMQLogger().Log(FairMQLogger::type)
#define LOG(type) FairMQLogger().Log(FairMQLogger::type)
#endif /* FAIRMQLOGGER_H_ */

View File

@ -12,7 +12,6 @@
#include "FairMQMerger.h"
#include "FairMQPoller.h"
FairMQMerger::FairMQMerger()
{
}
@ -31,16 +30,20 @@ void FairMQMerger::Run()
bool received = false;
while ( fState == RUNNING ) {
while (fState == RUNNING)
{
FairMQMessage* msg = fTransportFactory->CreateMessage();
poller->Poll(100);
for(int i = 0; i < fNumInputs; i++) {
if (poller->CheckInput(i)){
for (int i = 0; i < fNumInputs; i++)
{
if (poller->CheckInput(i))
{
received = fPayloadInputs->at(i)->Receive(msg);
}
if (received) {
if (received)
{
fPayloadOutputs->at(0)->Send(msg);
received = false;
}
@ -54,4 +57,3 @@ void FairMQMerger::Run()
rateLogger.interrupt();
rateLogger.join();
}

View File

@ -10,12 +10,12 @@
#include "FairMQDevice.h"
class FairMQMerger : public FairMQDevice
{
public:
FairMQMerger();
virtual ~FairMQMerger();
protected:
virtual void Run();
};

View File

@ -10,7 +10,6 @@
#include <cstddef> // for size_t
class FairMQMessage
{
public:

View File

@ -15,7 +15,6 @@ class FairMQPoller
virtual bool CheckInput(int index) = 0;
virtual ~FairMQPoller() {};
};
#endif /* FAIRMQPOLLER_H_ */

View File

@ -27,7 +27,8 @@ void FairMQProxy::Run()
FairMQMessage* msg = fTransportFactory->CreateMessage();
while ( fState == RUNNING ) {
while (fState == RUNNING)
{
fPayloadInputs->at(0)->Receive(msg);
fPayloadOutputs->at(0)->Send(msg);
}

View File

@ -10,12 +10,12 @@
#include "FairMQDevice.h"
class FairMQProxy : public FairMQDevice
{
public:
FairMQProxy();
virtual ~FairMQProxy();
protected:
virtual void Run();
};

View File

@ -21,7 +21,8 @@ void FairMQSink::Run()
boost::thread rateLogger(boost::bind(&FairMQDevice::LogSocketRates, this));
while ( fState == RUNNING ) {
while (fState == RUNNING)
{
FairMQMessage* msg = fTransportFactory->CreateMessage();
fPayloadInputs->at(0)->Receive(msg);
@ -36,4 +37,3 @@ void FairMQSink::Run()
FairMQSink::~FairMQSink()
{
}

View File

@ -10,12 +10,12 @@
#include "FairMQDevice.h"
class FairMQSink : public FairMQDevice
{
public:
FairMQSink();
virtual ~FairMQSink();
protected:
virtual void Run();
};

View File

@ -11,7 +11,6 @@
#include "FairMQLogger.h"
#include "FairMQSplitter.h"
FairMQSplitter::FairMQSplitter()
{
}
@ -29,15 +28,18 @@ void FairMQSplitter::Run()
bool received = false;
int direction = 0;
while ( fState == RUNNING ) {
while (fState == RUNNING)
{
FairMQMessage* msg = fTransportFactory->CreateMessage();
received = fPayloadInputs->at(0)->Receive(msg);
if (received) {
if (received)
{
fPayloadOutputs->at(direction)->Send(msg);
direction++;
if (direction >= fNumOutputs) {
if (direction >= fNumOutputs)
{
direction = 0;
}
received = false;

View File

@ -10,12 +10,12 @@
#include "FairMQDevice.h"
class FairMQSplitter : public FairMQDevice
{
public:
FairMQSplitter();
virtual ~FairMQSplitter();
protected:
virtual void Run();
};

3
fairmq/apply-clang-format.sh Executable file
View File

@ -0,0 +1,3 @@
#!/bin/bash
find . -type f \( -iname "*.h" ! -iname "*.pb.h" -o -iname "*.cxx" \) -execdir clang-format -i {} \;

View File

@ -12,17 +12,18 @@
#include "FairMQMessageNN.h"
#include "FairMQLogger.h"
FairMQMessageNN::FairMQMessageNN() :
fSize(0),
fMessage(NULL),
fReceiving(false)
FairMQMessageNN::FairMQMessageNN()
: fSize(0)
, fMessage(NULL)
, fReceiving(false)
{
}
FairMQMessageNN::FairMQMessageNN(size_t size)
{
fMessage = nn_allocmsg(size, 0);
if (!fMessage){
if (!fMessage)
{
LOG(ERROR) << "failed allocating message, reason: " << nn_strerror(errno);
}
fSize = size;
@ -32,7 +33,8 @@ FairMQMessageNN::FairMQMessageNN(size_t size)
FairMQMessageNN::FairMQMessageNN(void* data, size_t size)
{
fMessage = nn_allocmsg(size, 0);
if (!fMessage){
if (!fMessage)
{
LOG(ERROR) << "failed allocating message, reason: " << nn_strerror(errno);
}
memcpy(fMessage, data, size);
@ -52,7 +54,8 @@ void FairMQMessageNN::Rebuild(size_t size)
{
Clear();
fMessage = nn_allocmsg(size, 0);
if (!fMessage){
if (!fMessage)
{
LOG(ERROR) << "failed allocating message, reason: " << nn_strerror(errno);
}
fSize = size;
@ -63,7 +66,8 @@ void FairMQMessageNN::Rebuild(void* data, size_t size)
{
Clear();
fMessage = nn_allocmsg(size, 0);
if (!fMessage){
if (!fMessage)
{
LOG(ERROR) << "failed allocating message, reason: " << nn_strerror(errno);
}
memcpy(fMessage, data, size);
@ -94,9 +98,11 @@ void FairMQMessageNN::SetMessage(void* data, size_t size)
void FairMQMessageNN::Copy(FairMQMessage* msg)
{
if (fMessage){
if (fMessage)
{
int rc = nn_freemsg(fMessage);
if ( rc < 0 ){
if (rc < 0)
{
LOG(ERROR) << "failed freeing message, reason: " << nn_strerror(errno);
}
}
@ -104,7 +110,8 @@ void FairMQMessageNN::Copy(FairMQMessage* msg)
size_t size = msg->GetSize();
fMessage = nn_allocmsg(size, 0);
if (!fMessage){
if (!fMessage)
{
LOG(ERROR) << "failed allocating message, reason: " << nn_strerror(errno);
}
std::memcpy(fMessage, msg->GetMessage(), size);
@ -114,9 +121,12 @@ void FairMQMessageNN::Copy(FairMQMessage* msg)
inline void FairMQMessageNN::Clear()
{
int rc = nn_freemsg(fMessage);
if (rc < 0) {
if (rc < 0)
{
LOG(ERROR) << "failed freeing message, reason: " << nn_strerror(errno);
} else {
}
else
{
fMessage = NULL;
fSize = 0;
}
@ -124,11 +134,15 @@ inline void FairMQMessageNN::Clear()
FairMQMessageNN::~FairMQMessageNN()
{
if(fReceiving){
if (fReceiving)
{
int rc = nn_freemsg(fMessage);
if (rc < 0) {
if (rc < 0)
{
LOG(ERROR) << "failed freeing message, reason: " << nn_strerror(errno);
} else {
}
else
{
fMessage = NULL;
fSize = 0;
}

View File

@ -12,7 +12,6 @@
#include "FairMQMessage.h"
class FairMQMessageNN : public FairMQMessage
{
public:

View File

@ -14,7 +14,8 @@ FairMQPollerNN::FairMQPollerNN(const vector<FairMQSocket*>& inputs)
fNumItems = inputs.size();
items = new nn_pollfd[fNumItems];
for (int i = 0; i < fNumItems; i++) {
for (int i = 0; i < fNumItems; i++)
{
items[i].fd = inputs.at(i)->GetSocket(1);
items[i].events = NN_POLLIN;
}
@ -35,5 +36,6 @@ bool FairMQPollerNN::CheckInput(int index)
FairMQPollerNN::~FairMQPollerNN()
{
if (items != NULL) delete [] items;
if (items != NULL)
delete[] items;
}

View File

@ -43,7 +43,8 @@ void FairMQSocketNN::Bind(const string& address)
LOG(INFO) << "bind socket #" << fId << " on " << address;
int eid = nn_bind(fSocket, address.c_str());
if (eid < 0) {
if (eid < 0)
{
LOG(ERROR) << "failed binding socket #" << fId << ", reason: " << nn_strerror(errno);
}
}
@ -53,7 +54,8 @@ void FairMQSocketNN::Connect(const string& address)
LOG(INFO) << "connect socket #" << fId << " to " << address;
int eid = nn_connect(fSocket, address.c_str());
if (eid < 0) {
if (eid < 0)
{
LOG(ERROR) << "failed connecting socket #" << fId << ", reason: " << nn_strerror(errno);
}
}
@ -62,9 +64,12 @@ size_t FairMQSocketNN::Send(FairMQMessage* msg)
{
void* ptr = msg->GetMessage();
int rc = nn_send(fSocket, &ptr, NN_MSG, 0);
if (rc < 0) {
if (rc < 0)
{
LOG(ERROR) << "failed sending on socket #" << fId << ", reason: " << nn_strerror(errno);
} else {
}
else
{
fBytesTx += rc;
++fMessagesTx;
static_cast<FairMQMessageNN*>(msg)->fReceiving = false;
@ -77,9 +82,12 @@ size_t FairMQSocketNN::Receive(FairMQMessage* msg)
{
void* ptr = NULL;
int rc = nn_recv(fSocket, &ptr, NN_MSG, 0);
if (rc < 0) {
if (rc < 0)
{
LOG(ERROR) << "failed receiving on socket #" << fId << ", reason: " << nn_strerror(errno);
} else {
}
else
{
fBytesRx += rc;
++fMessagesRx;
msg->SetMessage(ptr, rc);
@ -107,7 +115,8 @@ void FairMQSocketNN::Close()
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) {
if (rc < 0)
{
LOG(ERROR) << "failed setting socket option, reason: " << nn_strerror(errno);
}
}
@ -134,14 +143,22 @@ unsigned long FairMQSocketNN::GetMessagesRx()
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;
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;
}

View File

@ -14,7 +14,6 @@
#include "FairMQSocket.h"
class FairMQSocketNN : public FairMQSocket
{
public:

View File

@ -26,7 +26,6 @@ class FairMQTransportFactoryNN : public FairMQTransportFactory
virtual FairMQSocket* CreateSocket(const string& type, int num, int numIoThreads);
virtual FairMQPoller* CreatePoller(const vector<FairMQSocket*>& inputs);
virtual ~FairMQTransportFactoryNN() {};
};

View File

@ -15,11 +15,10 @@
#include "FairMQBinSampler.h"
#include "FairMQLogger.h"
FairMQBinSampler::FairMQBinSampler() :
fEventSize(10000),
fEventRate(1),
fEventCounter(0)
FairMQBinSampler::FairMQBinSampler()
: fEventSize(10000)
, fEventRate(1)
, fEventCounter(0)
{
}
@ -44,11 +43,13 @@ void FairMQBinSampler::Run()
LOG(DEBUG) << "Message size: " << fEventSize * sizeof(Content) << " bytes.";
while ( fState == RUNNING ) {
while (fState == RUNNING)
{
Content* payload = new Content[fEventSize];
for (int i = 0; i < fEventSize; ++i) {
for (int i = 0; i < fEventSize; ++i)
{
(&payload[i])->x = rand() % 100 + 1;
(&payload[i])->y = rand() % 100 + 1;
(&payload[i])->z = rand() % 100 + 1;
@ -64,7 +65,8 @@ void FairMQBinSampler::Run()
--fEventCounter;
while (fEventCounter == 0) {
while (fEventCounter == 0)
{
boost::this_thread::sleep(boost::posix_time::milliseconds(1));
}
@ -81,11 +83,15 @@ void FairMQBinSampler::Run()
void FairMQBinSampler::ResetEventCounter()
{
while ( true ) {
try {
while (true)
{
try
{
fEventCounter = fEventRate / 100;
boost::this_thread::sleep(boost::posix_time::milliseconds(10));
} catch (boost::thread_interrupted&) {
}
catch (boost::thread_interrupted&)
{
break;
}
}
@ -104,7 +110,8 @@ void FairMQBinSampler::Log(int intervalInMs)
t0 = get_timestamp();
while (true) {
while (true)
{
boost::this_thread::sleep(boost::posix_time::milliseconds(intervalInMs));
t1 = get_timestamp();
@ -127,7 +134,8 @@ void FairMQBinSampler::Log(int intervalInMs)
void FairMQBinSampler::SetProperty(const int key, const string& value, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
default:
FairMQDevice::SetProperty(key, value, slot);
break;
@ -136,7 +144,8 @@ void FairMQBinSampler::SetProperty(const int key, const string& value, const int
string FairMQBinSampler::GetProperty(const int key, const string& default_ /*= ""*/, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
default:
return FairMQDevice::GetProperty(key, default_, slot);
}
@ -144,7 +153,8 @@ string FairMQBinSampler::GetProperty(const int key, const string& default_/*= ""
void FairMQBinSampler::SetProperty(const int key, const int value, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
case EventSize:
fEventSize = value;
break;
@ -159,7 +169,8 @@ void FairMQBinSampler::SetProperty(const int key, const int value, const int slo
int FairMQBinSampler::GetProperty(const int key, const int default_ /*= 0*/, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
case EventSize:
return fEventSize;
case EventRate:

View File

@ -12,7 +12,8 @@
#include "FairMQDevice.h"
struct Content {
struct Content
{
double a;
double b;
int x;
@ -23,7 +24,8 @@ struct Content {
class FairMQBinSampler : public FairMQDevice
{
public:
enum {
enum
{
InputFile = FairMQDevice::Last,
EventRate,
EventSize,

View File

@ -21,7 +21,8 @@ void FairMQBinSink::Run()
boost::thread rateLogger(boost::bind(&FairMQDevice::LogSocketRates, this));
while ( fState == RUNNING ) {
while (fState == RUNNING)
{
FairMQMessage* msg = fTransportFactory->CreateMessage();
fPayloadInputs->at(0)->Receive(msg);

View File

@ -10,7 +10,8 @@
#include "FairMQDevice.h"
struct Content {
struct Content
{
double a;
double b;
int x;
@ -23,6 +24,7 @@ class FairMQBinSink: public FairMQDevice
public:
FairMQBinSink();
virtual ~FairMQBinSink();
protected:
virtual void Run();
};

View File

@ -15,11 +15,10 @@
#include "payload.pb.h"
FairMQProtoSampler::FairMQProtoSampler() :
fEventSize(10000),
fEventRate(1),
fEventCounter(0)
FairMQProtoSampler::FairMQProtoSampler()
: fEventSize(10000)
, fEventRate(1)
, fEventCounter(0)
{
}
@ -42,11 +41,13 @@ void FairMQProtoSampler::Run()
srand(time(NULL));
while ( fState == RUNNING ) {
while (fState == RUNNING)
{
sampler::Payload p;
for (int i = 0; i < fEventSize; ++i) {
for (int i = 0; i < fEventSize; ++i)
{
sampler::Content* content = p.add_data();
content->set_x(rand() % 100 + 1);
@ -68,7 +69,8 @@ void FairMQProtoSampler::Run()
--fEventCounter;
while (fEventCounter == 0) {
while (fEventCounter == 0)
{
boost::this_thread::sleep(boost::posix_time::milliseconds(1));
}
@ -84,11 +86,15 @@ void FairMQProtoSampler::Run()
void FairMQProtoSampler::ResetEventCounter()
{
while ( true ) {
try {
while (true)
{
try
{
fEventCounter = fEventRate / 100;
boost::this_thread::sleep(boost::posix_time::milliseconds(10));
} catch (boost::thread_interrupted&) {
}
catch (boost::thread_interrupted&)
{
break;
}
}
@ -107,7 +113,8 @@ void FairMQProtoSampler::Log(int intervalInMs)
t0 = get_timestamp();
while (true) {
while (true)
{
boost::this_thread::sleep(boost::posix_time::milliseconds(intervalInMs));
t1 = get_timestamp();
@ -130,7 +137,8 @@ void FairMQProtoSampler::Log(int intervalInMs)
void FairMQProtoSampler::SetProperty(const int key, const string& value, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
default:
FairMQDevice::SetProperty(key, value, slot);
break;
@ -139,7 +147,8 @@ void FairMQProtoSampler::SetProperty(const int key, const string& value, const i
string FairMQProtoSampler::GetProperty(const int key, const string& default_ /*= ""*/, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
default:
return FairMQDevice::GetProperty(key, default_, slot);
}
@ -147,7 +156,8 @@ string FairMQProtoSampler::GetProperty(const int key, const string& default_/*=
void FairMQProtoSampler::SetProperty(const int key, const int value, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
case EventSize:
fEventSize = value;
break;
@ -162,7 +172,8 @@ void FairMQProtoSampler::SetProperty(const int key, const int value, const int s
int FairMQProtoSampler::GetProperty(const int key, const int default_ /*= 0*/, const int slot /*= 0*/)
{
switch (key) {
switch (key)
{
case EventSize:
return fEventSize;
case EventRate:

View File

@ -15,7 +15,8 @@
class FairMQProtoSampler : public FairMQDevice
{
public:
enum {
enum
{
InputFile = FairMQDevice::Last,
EventRate,
EventSize,

View File

@ -23,7 +23,8 @@ void FairMQProtoSink::Run()
boost::thread rateLogger(boost::bind(&FairMQDevice::LogSocketRates, this));
while ( fState == RUNNING ) {
while (fState == RUNNING)
{
FairMQMessage* msg = fTransportFactory->CreateMessage();
fPayloadInputs->at(0)->Receive(msg);

View File

@ -10,7 +10,8 @@
#include "FairMQDevice.h"
struct Content {
struct Content
{
double a;
double b;
int x;
@ -23,6 +24,7 @@ class FairMQProtoSink: public FairMQDevice
public:
FairMQProtoSink();
virtual ~FairMQProtoSink();
protected:
virtual void Run();
};

View File

@ -22,7 +22,6 @@ using std::cin;
using std::endl;
using std::stringstream;
FairMQBenchmarkSampler sampler;
static void s_signal_handler(int signal)
@ -48,10 +47,10 @@ static void s_catch_signals (void)
int main(int argc, char** argv)
{
if ( argc != 9 ) {
if (argc != 9)
{
cout << "Usage: bsampler ID eventSize eventRate numIoTreads\n"
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n"
<< endl;
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n" << endl;
return 1;
}
@ -90,10 +89,8 @@ int main(int argc, char** argv)
sampler.SetProperty(FairMQBenchmarkSampler::NumInputs, 0);
sampler.SetProperty(FairMQBenchmarkSampler::NumOutputs, 1);
sampler.ChangeState(FairMQBenchmarkSampler::INIT);
sampler.SetProperty(FairMQBenchmarkSampler::OutputSocketType, argv[i], 0);
++i;
int outputSndBufSize;
@ -105,13 +102,10 @@ int main(int argc, char** argv)
sampler.SetProperty(FairMQBenchmarkSampler::OutputAddress, argv[i], 0);
++i;
sampler.ChangeState(FairMQBenchmarkSampler::SETOUTPUT);
sampler.ChangeState(FairMQBenchmarkSampler::SETINPUT);
sampler.ChangeState(FairMQBenchmarkSampler::RUN);
char ch;
cin.get(ch);
@ -120,4 +114,3 @@ int main(int argc, char** argv)
return 0;
}

View File

@ -22,7 +22,6 @@ using std::cin;
using std::endl;
using std::stringstream;
FairMQBinSampler sampler;
static void s_signal_handler(int signal)
@ -48,10 +47,10 @@ static void s_catch_signals (void)
int main(int argc, char** argv)
{
if ( argc != 9 ) {
if (argc != 9)
{
cout << "Usage: bsampler ID eventSize eventRate numIoTreads\n"
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n"
<< endl;
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n" << endl;
return 1;
}
@ -90,10 +89,8 @@ int main(int argc, char** argv)
sampler.SetProperty(FairMQBinSampler::NumInputs, 0);
sampler.SetProperty(FairMQBinSampler::NumOutputs, 1);
sampler.ChangeState(FairMQBinSampler::INIT);
sampler.SetProperty(FairMQBinSampler::OutputSocketType, argv[i], 0);
++i;
int outputSndBufSize;
@ -105,13 +102,10 @@ int main(int argc, char** argv)
sampler.SetProperty(FairMQBinSampler::OutputAddress, argv[i], 0);
++i;
sampler.ChangeState(FairMQBinSampler::SETOUTPUT);
sampler.ChangeState(FairMQBinSampler::SETINPUT);
sampler.ChangeState(FairMQBinSampler::RUN);
char ch;
cin.get(ch);
@ -120,4 +114,3 @@ int main(int argc, char** argv)
return 0;
}

View File

@ -22,7 +22,6 @@ using std::cin;
using std::endl;
using std::stringstream;
FairMQBinSink sink;
static void s_signal_handler(int signal)
@ -48,10 +47,10 @@ static void s_catch_signals (void)
int main(int argc, char** argv)
{
if ( argc != 7 ) {
if (argc != 7)
{
cout << "Usage: sink \tID numIoTreads\n"
<< "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n"
<< endl;
<< "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n" << endl;
return 1;
}
@ -80,10 +79,8 @@ int main(int argc, char** argv)
sink.SetProperty(FairMQBinSink::NumInputs, 1);
sink.SetProperty(FairMQBinSink::NumOutputs, 0);
sink.ChangeState(FairMQBinSink::INIT);
sink.SetProperty(FairMQBinSink::InputSocketType, argv[i], 0);
++i;
int inputRcvBufSize;
@ -95,12 +92,10 @@ int main(int argc, char** argv)
sink.SetProperty(FairMQBinSink::InputAddress, argv[i], 0);
++i;
sink.ChangeState(FairMQBinSink::SETOUTPUT);
sink.ChangeState(FairMQBinSink::SETINPUT);
sink.ChangeState(FairMQBinSink::RUN);
char ch;
cin.get(ch);
@ -109,4 +104,3 @@ int main(int argc, char** argv)
return 0;
}

View File

@ -22,7 +22,6 @@ using std::cin;
using std::endl;
using std::stringstream;
FairMQBuffer buffer;
static void s_signal_handler(int signal)
@ -48,7 +47,8 @@ static void s_catch_signals (void)
int main(int argc, char** argv)
{
if ( argc != 11 ) {
if (argc != 11)
{
cout << "Usage: buffer \tID numIoTreads\n"
<< "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n"
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n" << endl;
@ -79,10 +79,8 @@ int main(int argc, char** argv)
buffer.SetProperty(FairMQBuffer::NumInputs, 1);
buffer.SetProperty(FairMQBuffer::NumOutputs, 1);
buffer.ChangeState(FairMQBuffer::INIT);
buffer.SetProperty(FairMQBuffer::InputSocketType, argv[i], 0);
++i;
int inputRcvBufSize;
@ -105,13 +103,10 @@ int main(int argc, char** argv)
buffer.SetProperty(FairMQBuffer::OutputAddress, argv[i], 0);
++i;
buffer.ChangeState(FairMQBuffer::SETOUTPUT);
buffer.ChangeState(FairMQBuffer::SETINPUT);
buffer.ChangeState(FairMQBuffer::RUN);
char ch;
cin.get(ch);
@ -120,4 +115,3 @@ int main(int argc, char** argv)
return 0;
}

View File

@ -22,7 +22,6 @@ using std::cin;
using std::endl;
using std::stringstream;
FairMQMerger merger;
static void s_signal_handler(int signal)
@ -48,13 +47,13 @@ static void s_catch_signals (void)
int main(int argc, char** argv)
{
if ( argc < 16 || (argc - 8) % 4 != 0 ) {
if (argc < 16 || (argc - 8) % 4 != 0)
{
cout << "Usage: merger \tID numIoTreads numInputs\n"
<< "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n"
<< "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n"
<< "\t\t...\n"
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n"
<< argc << " arguments provided" << endl;
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n" << argc << " arguments provided" << endl;
return 1;
}
@ -89,7 +88,8 @@ int main(int argc, char** argv)
merger.ChangeState(FairMQMerger::INIT);
for (int iInput = 0; iInput < numInputs; iInput++ ) {
for (int iInput = 0; iInput < numInputs; iInput++)
{
merger.SetProperty(FairMQMerger::InputSocketType, argv[i], iInput);
++i;
int inputRcvBufSize;
@ -117,7 +117,6 @@ int main(int argc, char** argv)
merger.ChangeState(FairMQMerger::SETINPUT);
merger.ChangeState(FairMQMerger::RUN);
char ch;
cin.get(ch);
@ -126,4 +125,3 @@ int main(int argc, char** argv)
return 0;
}

View File

@ -22,7 +22,6 @@ using std::cin;
using std::endl;
using std::stringstream;
FairMQProtoSampler sampler;
static void s_signal_handler(int signal)
@ -48,10 +47,10 @@ static void s_catch_signals (void)
int main(int argc, char** argv)
{
if ( argc != 9 ) {
if (argc != 9)
{
cout << "Usage: bsampler ID eventSize eventRate numIoTreads\n"
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n"
<< endl;
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n" << endl;
return 1;
}
@ -90,10 +89,8 @@ int main(int argc, char** argv)
sampler.SetProperty(FairMQProtoSampler::NumInputs, 0);
sampler.SetProperty(FairMQProtoSampler::NumOutputs, 1);
sampler.ChangeState(FairMQProtoSampler::INIT);
sampler.SetProperty(FairMQProtoSampler::OutputSocketType, argv[i], 0);
++i;
int outputSndBufSize;
@ -105,13 +102,10 @@ int main(int argc, char** argv)
sampler.SetProperty(FairMQProtoSampler::OutputAddress, argv[i], 0);
++i;
sampler.ChangeState(FairMQProtoSampler::SETOUTPUT);
sampler.ChangeState(FairMQProtoSampler::SETINPUT);
sampler.ChangeState(FairMQProtoSampler::RUN);
char ch;
cin.get(ch);
@ -120,4 +114,3 @@ int main(int argc, char** argv)
return 0;
}

View File

@ -22,7 +22,6 @@ using std::cin;
using std::endl;
using std::stringstream;
FairMQProtoSink sink;
static void s_signal_handler(int signal)
@ -48,10 +47,10 @@ static void s_catch_signals (void)
int main(int argc, char** argv)
{
if ( argc != 7 ) {
if (argc != 7)
{
cout << "Usage: sink \tID numIoTreads\n"
<< "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n"
<< endl;
<< "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n" << endl;
return 1;
}
@ -80,10 +79,8 @@ int main(int argc, char** argv)
sink.SetProperty(FairMQProtoSink::NumInputs, 1);
sink.SetProperty(FairMQProtoSink::NumOutputs, 0);
sink.ChangeState(FairMQProtoSink::INIT);
sink.SetProperty(FairMQProtoSink::InputSocketType, argv[i], 0);
++i;
int inputRcvBufSize;
@ -95,12 +92,10 @@ int main(int argc, char** argv)
sink.SetProperty(FairMQProtoSink::InputAddress, argv[i], 0);
++i;
sink.ChangeState(FairMQProtoSink::SETOUTPUT);
sink.ChangeState(FairMQProtoSink::SETINPUT);
sink.ChangeState(FairMQProtoSink::RUN);
char ch;
cin.get(ch);
@ -109,4 +104,3 @@ int main(int argc, char** argv)
return 0;
}

View File

@ -22,7 +22,6 @@ using std::cin;
using std::endl;
using std::stringstream;
FairMQProxy proxy;
static void s_signal_handler(int signal)
@ -48,7 +47,8 @@ static void s_catch_signals (void)
int main(int argc, char** argv)
{
if ( argc != 11 ) {
if (argc != 11)
{
cout << "Usage: proxy \tID numIoTreads\n"
<< "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n"
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n" << endl;
@ -80,10 +80,8 @@ int main(int argc, char** argv)
proxy.SetProperty(FairMQProxy::NumInputs, 1);
proxy.SetProperty(FairMQProxy::NumOutputs, 1);
proxy.ChangeState(FairMQProxy::INIT);
proxy.SetProperty(FairMQProxy::InputSocketType, argv[i], 0);
++i;
int inputRcvBufSize;
@ -106,12 +104,10 @@ int main(int argc, char** argv)
proxy.SetProperty(FairMQProxy::OutputAddress, argv[i], 0);
++i;
proxy.ChangeState(FairMQProxy::SETOUTPUT);
proxy.ChangeState(FairMQProxy::SETINPUT);
proxy.ChangeState(FairMQProxy::RUN);
char ch;
cin.get(ch);
@ -120,4 +116,3 @@ int main(int argc, char** argv)
return 0;
}

View File

@ -22,7 +22,6 @@ using std::cin;
using std::endl;
using std::stringstream;
FairMQSink sink;
static void s_signal_handler(int signal)
@ -48,10 +47,10 @@ static void s_catch_signals (void)
int main(int argc, char** argv)
{
if ( argc != 7 ) {
if (argc != 7)
{
cout << "Usage: sink \tID numIoTreads\n"
<< "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n"
<< endl;
<< "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n" << endl;
return 1;
}
@ -80,10 +79,8 @@ int main(int argc, char** argv)
sink.SetProperty(FairMQSink::NumInputs, 1);
sink.SetProperty(FairMQSink::NumOutputs, 0);
sink.ChangeState(FairMQSink::INIT);
sink.SetProperty(FairMQSink::InputSocketType, argv[i], 0);
++i;
int inputRcvBufSize;
@ -95,12 +92,10 @@ int main(int argc, char** argv)
sink.SetProperty(FairMQSink::InputAddress, argv[i], 0);
++i;
sink.ChangeState(FairMQSink::SETOUTPUT);
sink.ChangeState(FairMQSink::SETINPUT);
sink.ChangeState(FairMQSink::RUN);
char ch;
cin.get(ch);
@ -109,4 +104,3 @@ int main(int argc, char** argv)
return 0;
}

View File

@ -22,7 +22,6 @@ using std::cin;
using std::endl;
using std::stringstream;
FairMQSplitter splitter;
static void s_signal_handler(int signal)
@ -48,13 +47,13 @@ static void s_catch_signals (void)
int main(int argc, char** argv)
{
if ( argc < 16 || (argc - 8) % 4 != 0 ) {
if (argc < 16 || (argc - 8) % 4 != 0)
{
cout << "Usage: splitter \tID numIoTreads numOutputs\n"
<< "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n"
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n"
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n"
<< "\t\t..."
<< argc << " arguments provided" << endl;
<< "\t\t..." << argc << " arguments provided" << endl;
return 1;
}
@ -101,7 +100,8 @@ int main(int argc, char** argv)
++i;
int outputSndBufSize;
for (int iOutput = 0; iOutput < numOutputs; iOutput++) {
for (int iOutput = 0; iOutput < numOutputs; iOutput++)
{
splitter.SetProperty(FairMQSplitter::OutputSocketType, argv[i], iOutput);
++i;
stringstream(argv[i]) >> outputSndBufSize;
@ -117,7 +117,6 @@ int main(int argc, char** argv)
splitter.ChangeState(FairMQSplitter::SETINPUT);
splitter.ChangeState(FairMQSplitter::RUN);
char ch;
cin.get(ch);
@ -126,4 +125,3 @@ int main(int argc, char** argv)
return 0;
}

View File

@ -12,12 +12,14 @@
FairMQContextZMQ::FairMQContextZMQ(int numIoThreads)
{
fContext = zmq_ctx_new();
if (fContext == NULL){
if (fContext == NULL)
{
LOG(ERROR) << "failed creating context, reason: " << zmq_strerror(errno);
}
int rc = zmq_ctx_set(fContext, ZMQ_IO_THREADS, numIoThreads);
if (rc != 0){
if (rc != 0)
{
LOG(ERROR) << "failed configuring context, reason: " << zmq_strerror(errno);
}
}
@ -34,12 +36,14 @@ void* FairMQContextZMQ::GetContext()
void FairMQContextZMQ::Close()
{
if (fContext == NULL){
if (fContext == NULL)
{
return;
}
int rc = zmq_ctx_destroy(fContext);
if (rc != 0) {
if (rc != 0)
{
LOG(ERROR) << "failed closing context, reason: " << zmq_strerror(errno);
}

View File

@ -11,11 +11,11 @@
#include "FairMQMessageZMQ.h"
#include "FairMQLogger.h"
FairMQMessageZMQ::FairMQMessageZMQ()
{
int rc = zmq_msg_init(&fMessage);
if (rc != 0) {
if (rc != 0)
{
LOG(ERROR) << "failed initializing message, reason: " << zmq_strerror(errno);
}
}
@ -23,7 +23,8 @@ FairMQMessageZMQ::FairMQMessageZMQ()
FairMQMessageZMQ::FairMQMessageZMQ(size_t size)
{
int rc = zmq_msg_init_size(&fMessage, size);
if (rc != 0) {
if (rc != 0)
{
LOG(ERROR) << "failed initializing message with size, reason: " << zmq_strerror(errno);
}
}
@ -31,7 +32,8 @@ FairMQMessageZMQ::FairMQMessageZMQ(size_t size)
FairMQMessageZMQ::FairMQMessageZMQ(void* data, size_t size)
{
int rc = zmq_msg_init_data(&fMessage, data, size, &CleanUp, NULL);
if (rc != 0) {
if (rc != 0)
{
LOG(ERROR) << "failed initializing message with data, reason: " << zmq_strerror(errno);
}
}
@ -40,7 +42,8 @@ void FairMQMessageZMQ::Rebuild()
{
CloseMessage();
int rc = zmq_msg_init(&fMessage);
if (rc != 0) {
if (rc != 0)
{
LOG(ERROR) << "failed initializing message, reason: " << zmq_strerror(errno);
}
}
@ -49,7 +52,8 @@ void FairMQMessageZMQ::Rebuild(size_t size)
{
CloseMessage();
int rc = zmq_msg_init_size(&fMessage, size);
if (rc != 0) {
if (rc != 0)
{
LOG(ERROR) << "failed initializing message with size, reason: " << zmq_strerror(errno);
}
}
@ -58,7 +62,8 @@ void FairMQMessageZMQ::Rebuild(void* data, size_t size)
{
CloseMessage();
int rc = zmq_msg_init_data(&fMessage, data, size, &CleanUp, NULL);
if (rc != 0) {
if (rc != 0)
{
LOG(ERROR) << "failed initializing message with data, reason: " << zmq_strerror(errno);
}
}
@ -94,12 +99,12 @@ void FairMQMessageZMQ::Copy(FairMQMessage* msg)
inline void FairMQMessageZMQ::CloseMessage()
{
int rc = zmq_msg_close(&fMessage);
if (rc != 0) {
if (rc != 0)
{
LOG(ERROR) << "failed closing message, reason: " << zmq_strerror(errno);
}
}
void FairMQMessageZMQ::CleanUp(void* data, void* hint)
{
free(data);
@ -108,7 +113,8 @@ void FairMQMessageZMQ::CleanUp(void* data, void* hint)
FairMQMessageZMQ::~FairMQMessageZMQ()
{
int rc = zmq_msg_close(&fMessage);
if (rc != 0) {
if (rc != 0)
{
LOG(ERROR) << "failed closing message with data, reason: " << zmq_strerror(errno);
}
}

View File

@ -14,7 +14,6 @@
#include "FairMQMessage.h"
class FairMQMessageZMQ : public FairMQMessage
{
public:

View File

@ -14,7 +14,8 @@ FairMQPollerZMQ::FairMQPollerZMQ(const vector<FairMQSocket*>& inputs)
fNumItems = inputs.size();
items = new zmq_pollitem_t[fNumItems];
for (int i = 0; i < fNumItems; i++) {
for (int i = 0; i < fNumItems; i++)
{
items[i].socket = inputs.at(i)->GetSocket();
items[i].fd = 0;
items[i].events = ZMQ_POLLIN;
@ -37,5 +38,6 @@ bool FairMQPollerZMQ::CheckInput(int index)
FairMQPollerZMQ::~FairMQPollerZMQ()
{
if (items != NULL) delete [] items;
if (items != NULL)
delete[] items;
}

View File

@ -39,6 +39,14 @@ FairMQSocketZMQ::FairMQSocketZMQ(const string& type, int num, int numIoThreads)
if (rc != 0) {
LOG(ERROR) << "failed setting socket option, reason: " << zmq_strerror(errno);
}
if (type == "sub")
{
rc = zmq_setsockopt(fSocket, ZMQ_SUBSCRIBE, NULL, 0);
if (rc != 0)
{
LOG(ERROR) << "failed setting socket option, reason: " << zmq_strerror(errno);
}
}
LOG(INFO) << "created socket #" << fId;
@ -54,7 +62,8 @@ void FairMQSocketZMQ::Bind(const string& address)
LOG(INFO) << "bind socket #" << fId << " on " << address;
int rc = zmq_bind(fSocket, address.c_str());
if (rc != 0) {
if (rc != 0)
{
LOG(ERROR) << "failed binding socket #" << fId << ", reason: " << zmq_strerror(errno);
}
}
@ -64,7 +73,8 @@ void FairMQSocketZMQ::Connect(const string& address)
LOG(INFO) << "connect socket #" << fId << " on " << address;
int rc = zmq_connect(fSocket, address.c_str());
if (rc != 0) {
if (rc != 0)
{
LOG(ERROR) << "failed connecting socket #" << fId << ", reason: " << zmq_strerror(errno);
}
}
@ -72,12 +82,14 @@ void FairMQSocketZMQ::Connect(const string& address)
size_t FairMQSocketZMQ::Send(FairMQMessage* msg)
{
int nbytes = zmq_msg_send(static_cast<zmq_msg_t*>(msg->GetMessage()), fSocket, 0);
if (nbytes >= 0){
if (nbytes >= 0)
{
fBytesTx += nbytes;
++fMessagesTx;
return nbytes;
}
if (zmq_errno() == EAGAIN){
if (zmq_errno() == EAGAIN)
{
return false;
}
LOG(ERROR) << "failed sending on socket #" << fId << ", reason: " << zmq_strerror(errno);
@ -87,12 +99,14 @@ size_t FairMQSocketZMQ::Send(FairMQMessage* msg)
size_t FairMQSocketZMQ::Receive(FairMQMessage* msg)
{
int nbytes = zmq_msg_recv(static_cast<zmq_msg_t*>(msg->GetMessage()), fSocket, 0);
if (nbytes >= 0){
if (nbytes >= 0)
{
fBytesRx += nbytes;
++fMessagesRx;
return nbytes;
}
if (zmq_errno() == EAGAIN){
if (zmq_errno() == EAGAIN)
{
return false;
}
LOG(ERROR) << "failed receiving on socket #" << fId << ", reason: " << zmq_strerror(errno);
@ -101,12 +115,14 @@ size_t FairMQSocketZMQ::Receive(FairMQMessage* msg)
void FairMQSocketZMQ::Close()
{
if (fSocket == NULL){
if (fSocket == NULL)
{
return;
}
int rc = zmq_close(fSocket);
if (rc != 0) {
if (rc != 0)
{
LOG(ERROR) << "failed closing socket, reason: " << zmq_strerror(errno);
}
@ -127,7 +143,8 @@ int FairMQSocketZMQ::GetSocket(int nothing)
void FairMQSocketZMQ::SetOption(const string& option, const void* value, size_t valueSize)
{
int rc = zmq_setsockopt(fSocket, GetConstant(option), value, valueSize);
if (rc < 0) {
if (rc < 0)
{
LOG(ERROR) << "failed setting socket option, reason: " << zmq_strerror(errno);
}
}
@ -154,14 +171,22 @@ unsigned long FairMQSocketZMQ::GetMessagesRx()
int FairMQSocketZMQ::GetConstant(const string& constant)
{
if (constant == "sub") return ZMQ_SUB;
if (constant == "pub") return ZMQ_PUB;
if (constant == "xsub") return ZMQ_XSUB;
if (constant == "xpub") return ZMQ_XPUB;
if (constant == "push") return ZMQ_PUSH;
if (constant == "pull") return ZMQ_PULL;
if (constant == "snd-hwm") return ZMQ_SNDHWM;
if (constant == "rcv-hwm") return ZMQ_RCVHWM;
if (constant == "sub")
return ZMQ_SUB;
if (constant == "pub")
return ZMQ_PUB;
if (constant == "xsub")
return ZMQ_XSUB;
if (constant == "xpub")
return ZMQ_XPUB;
if (constant == "push")
return ZMQ_PUSH;
if (constant == "pull")
return ZMQ_PULL;
if (constant == "snd-hwm")
return ZMQ_SNDHWM;
if (constant == "rcv-hwm")
return ZMQ_RCVHWM;
return -1;
}

View File

@ -15,7 +15,6 @@
#include "FairMQSocket.h"
#include "FairMQContextZMQ.h"
class FairMQSocketZMQ : public FairMQSocket
{
public:

View File

@ -27,7 +27,6 @@ class FairMQTransportFactoryZMQ : public FairMQTransportFactory
virtual FairMQSocket* CreateSocket(const string& type, int num, int numIoThreads);
virtual FairMQPoller* CreatePoller(const vector<FairMQSocket*>& inputs);
virtual ~FairMQTransportFactoryZMQ() {};
};