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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

@ -14,7 +14,8 @@ FairMQPollerNN::FairMQPollerNN(const vector<FairMQSocket*>& inputs)
fNumItems = inputs.size(); fNumItems = inputs.size();
items = new nn_pollfd[fNumItems]; 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].fd = inputs.at(i)->GetSocket(1);
items[i].events = NN_POLLIN; items[i].events = NN_POLLIN;
} }
@ -35,5 +36,6 @@ bool FairMQPollerNN::CheckInput(int index)
FairMQPollerNN::~FairMQPollerNN() 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; LOG(INFO) << "bind socket #" << fId << " on " << address;
int eid = nn_bind(fSocket, address.c_str()); int eid = nn_bind(fSocket, address.c_str());
if (eid < 0) { if (eid < 0)
{
LOG(ERROR) << "failed binding socket #" << fId << ", reason: " << nn_strerror(errno); 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; LOG(INFO) << "connect socket #" << fId << " to " << address;
int eid = nn_connect(fSocket, address.c_str()); int eid = nn_connect(fSocket, address.c_str());
if (eid < 0) { if (eid < 0)
{
LOG(ERROR) << "failed connecting socket #" << fId << ", reason: " << nn_strerror(errno); LOG(ERROR) << "failed connecting socket #" << fId << ", reason: " << nn_strerror(errno);
} }
} }
@ -62,9 +64,12 @@ size_t FairMQSocketNN::Send(FairMQMessage* msg)
{ {
void* ptr = msg->GetMessage(); void* ptr = msg->GetMessage();
int rc = nn_send(fSocket, &ptr, NN_MSG, 0); 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); LOG(ERROR) << "failed sending on socket #" << fId << ", reason: " << nn_strerror(errno);
} else { }
else
{
fBytesTx += rc; fBytesTx += rc;
++fMessagesTx; ++fMessagesTx;
static_cast<FairMQMessageNN*>(msg)->fReceiving = false; static_cast<FairMQMessageNN*>(msg)->fReceiving = false;
@ -77,9 +82,12 @@ size_t FairMQSocketNN::Receive(FairMQMessage* msg)
{ {
void* ptr = NULL; void* ptr = NULL;
int rc = nn_recv(fSocket, &ptr, NN_MSG, 0); 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); LOG(ERROR) << "failed receiving on socket #" << fId << ", reason: " << nn_strerror(errno);
} else { }
else
{
fBytesRx += rc; fBytesRx += rc;
++fMessagesRx; ++fMessagesRx;
msg->SetMessage(ptr, rc); msg->SetMessage(ptr, rc);
@ -107,7 +115,8 @@ void FairMQSocketNN::Close()
void FairMQSocketNN::SetOption(const string& option, const void* value, size_t valueSize) void FairMQSocketNN::SetOption(const string& option, const void* value, size_t valueSize)
{ {
int rc = nn_setsockopt(fSocket, NN_SOL_SOCKET, GetConstant(option), value, 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); LOG(ERROR) << "failed setting socket option, reason: " << nn_strerror(errno);
} }
} }
@ -134,14 +143,22 @@ unsigned long FairMQSocketNN::GetMessagesRx()
int FairMQSocketNN::GetConstant(const string& constant) int FairMQSocketNN::GetConstant(const string& constant)
{ {
if (constant == "sub") return NN_SUB; if (constant == "sub")
if (constant == "pub") return NN_PUB; return NN_SUB;
if (constant == "xsub") return NN_SUB; // TODO: is there XPUB, XSUB for nanomsg? if (constant == "pub")
if (constant == "xpub") return NN_PUB; return NN_PUB;
if (constant == "push") return NN_PUSH; if (constant == "xsub")
if (constant == "pull") return NN_PULL; return NN_SUB; // TODO: is there XPUB, XSUB for nanomsg?
if (constant == "snd-hwm") return NN_SNDBUF; if (constant == "xpub")
if (constant == "rcv-hwm") return NN_RCVBUF; 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; return -1;
} }

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

@ -14,7 +14,8 @@ FairMQPollerZMQ::FairMQPollerZMQ(const vector<FairMQSocket*>& inputs)
fNumItems = inputs.size(); fNumItems = inputs.size();
items = new zmq_pollitem_t[fNumItems]; 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].socket = inputs.at(i)->GetSocket();
items[i].fd = 0; items[i].fd = 0;
items[i].events = ZMQ_POLLIN; items[i].events = ZMQ_POLLIN;
@ -37,5 +38,6 @@ bool FairMQPollerZMQ::CheckInput(int index)
FairMQPollerZMQ::~FairMQPollerZMQ() 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) { if (rc != 0) {
LOG(ERROR) << "failed setting socket option, reason: " << zmq_strerror(errno); 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; LOG(INFO) << "created socket #" << fId;
@ -54,7 +62,8 @@ void FairMQSocketZMQ::Bind(const string& address)
LOG(INFO) << "bind socket #" << fId << " on " << address; LOG(INFO) << "bind socket #" << fId << " on " << address;
int rc = zmq_bind(fSocket, address.c_str()); int rc = zmq_bind(fSocket, address.c_str());
if (rc != 0) { if (rc != 0)
{
LOG(ERROR) << "failed binding socket #" << fId << ", reason: " << zmq_strerror(errno); 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; LOG(INFO) << "connect socket #" << fId << " on " << address;
int rc = zmq_connect(fSocket, address.c_str()); int rc = zmq_connect(fSocket, address.c_str());
if (rc != 0) { if (rc != 0)
{
LOG(ERROR) << "failed connecting socket #" << fId << ", reason: " << zmq_strerror(errno); 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) size_t FairMQSocketZMQ::Send(FairMQMessage* msg)
{ {
int nbytes = zmq_msg_send(static_cast<zmq_msg_t*>(msg->GetMessage()), fSocket, 0); int nbytes = zmq_msg_send(static_cast<zmq_msg_t*>(msg->GetMessage()), fSocket, 0);
if (nbytes >= 0){ if (nbytes >= 0)
{
fBytesTx += nbytes; fBytesTx += nbytes;
++fMessagesTx; ++fMessagesTx;
return nbytes; return nbytes;
} }
if (zmq_errno() == EAGAIN){ if (zmq_errno() == EAGAIN)
{
return false; return false;
} }
LOG(ERROR) << "failed sending on socket #" << fId << ", reason: " << zmq_strerror(errno); 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) size_t FairMQSocketZMQ::Receive(FairMQMessage* msg)
{ {
int nbytes = zmq_msg_recv(static_cast<zmq_msg_t*>(msg->GetMessage()), fSocket, 0); int nbytes = zmq_msg_recv(static_cast<zmq_msg_t*>(msg->GetMessage()), fSocket, 0);
if (nbytes >= 0){ if (nbytes >= 0)
{
fBytesRx += nbytes; fBytesRx += nbytes;
++fMessagesRx; ++fMessagesRx;
return nbytes; return nbytes;
} }
if (zmq_errno() == EAGAIN){ if (zmq_errno() == EAGAIN)
{
return false; return false;
} }
LOG(ERROR) << "failed receiving on socket #" << fId << ", reason: " << zmq_strerror(errno); LOG(ERROR) << "failed receiving on socket #" << fId << ", reason: " << zmq_strerror(errno);
@ -101,12 +115,14 @@ size_t FairMQSocketZMQ::Receive(FairMQMessage* msg)
void FairMQSocketZMQ::Close() void FairMQSocketZMQ::Close()
{ {
if (fSocket == NULL){ if (fSocket == NULL)
{
return; return;
} }
int rc = zmq_close(fSocket); int rc = zmq_close(fSocket);
if (rc != 0) { if (rc != 0)
{
LOG(ERROR) << "failed closing socket, reason: " << zmq_strerror(errno); 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) void FairMQSocketZMQ::SetOption(const string& option, const void* value, size_t valueSize)
{ {
int rc = zmq_setsockopt(fSocket, GetConstant(option), value, 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); LOG(ERROR) << "failed setting socket option, reason: " << zmq_strerror(errno);
} }
} }
@ -154,14 +171,22 @@ unsigned long FairMQSocketZMQ::GetMessagesRx()
int FairMQSocketZMQ::GetConstant(const string& constant) int FairMQSocketZMQ::GetConstant(const string& constant)
{ {
if (constant == "sub") return ZMQ_SUB; if (constant == "sub")
if (constant == "pub") return ZMQ_PUB; return ZMQ_SUB;
if (constant == "xsub") return ZMQ_XSUB; if (constant == "pub")
if (constant == "xpub") return ZMQ_XPUB; return ZMQ_PUB;
if (constant == "push") return ZMQ_PUSH; if (constant == "xsub")
if (constant == "pull") return ZMQ_PULL; return ZMQ_XSUB;
if (constant == "snd-hwm") return ZMQ_SNDHWM; if (constant == "xpub")
if (constant == "rcv-hwm") return ZMQ_RCVHWM; 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; return -1;
} }

View File

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

View File

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