Mirror/FairMQ
1
0
Fork 0
mirror of https://github.com/FairRootGroup/FairMQ.git synced 2025-10-15 09:31:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

First version of the shared memory transport.

Browse Source 
Use via `--transport shmem` cmd option. No pub/sub.
This commit is contained in:
Alexey Rybalchenko
2016-06-03 11:24:12 +02:00
parent 6c3b01f09c
commit a332d9fc83
39 changed files with 2121 additions and 309 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
fairmq/devices/FairMQBenchmarkSampler.cxx
View File

@@ -23,10 +23,11 @@
using namespace std;
FairMQBenchmarkSampler::FairMQBenchmarkSampler()
: fMsgSize(10000)
, fNumMsgs(0)
: fSameMessage(true)
, fMsgSize(10000)
, fMsgCounter(0)
, fMsgRate(1)
, fNumMsgs(0)
, fOutChannelName()
{
}
@@ -37,9 +38,10 @@ FairMQBenchmarkSampler::~FairMQBenchmarkSampler()
void FairMQBenchmarkSampler::InitTask()
{
fSameMessage = fConfig->GetValue<bool>("same-msg");
fMsgSize = fConfig->GetValue<int>("msg-size");
fNumMsgs = fConfig->GetValue<int>("num-msgs");
fMsgRate = fConfig->GetValue<int>("msg-rate");
fNumMsgs = fConfig->GetValue<uint64_t>("num-msgs");
fOutChannelName = fConfig->GetValue<string>("out-channel");
}
@@ -47,7 +49,7 @@ void FairMQBenchmarkSampler::Run()
{
// std::thread resetMsgCounter(&FairMQBenchmarkSampler::ResetMsgCounter, this);
int numSentMsgs = 0;
uint64_t numSentMsgs = 0;
FairMQMessagePtr baseMsg(fTransportFactory->CreateMessage(fMsgSize));
@@ -59,17 +61,37 @@ void FairMQBenchmarkSampler::Run()
while (CheckCurrentState(RUNNING))
{
FairMQMessagePtr msg(fTransportFactory->CreateMessage());
msg->Copy(baseMsg);
if (dataOutChannel.Send(msg) >= 0)
if (fSameMessage)
{
if (fNumMsgs > 0)
FairMQMessagePtr msg(fTransportFactory->CreateMessage());
msg->Copy(baseMsg);
if (dataOutChannel.Send(msg) >= 0)
{
if (++numSentMsgs >= fNumMsgs)
if (fNumMsgs > 0)
{
break;
if (numSentMsgs >= fNumMsgs)
{
break;
}
}
++numSentMsgs;
}
}
else
{
FairMQMessagePtr msg(fTransportFactory->CreateMessage(fMsgSize));
if (dataOutChannel.Send(msg) >= 0)
{
if (fNumMsgs > 0)
{
if (numSentMsgs >= fNumMsgs)
{
break;
}
}
++numSentMsgs;
}
}
@@ -82,8 +104,7 @@ void FairMQBenchmarkSampler::Run()
auto tEnd = chrono::high_resolution_clock::now();
LOG(INFO) << "Sent " << numSentMsgs << " messages, leaving RUNNING state.";
LOG(INFO) << "Sending time: " << chrono::duration<double, milli>(tEnd - tStart).count() << " ms";
LOG(INFO) << "Leaving RUNNING state. Sent " << numSentMsgs << " messages in " << chrono::duration<double, milli>(tEnd - tStart).count() << "ms.";
// resetMsgCounter.join();
}

3
fairmq/devices/FairMQBenchmarkSampler.h
View File

@@ -32,10 +32,11 @@ class FairMQBenchmarkSampler : public FairMQDevice
void ResetMsgCounter();
protected:
bool fSameMessage;
int fMsgSize;
int fNumMsgs;
int fMsgCounter;
int fMsgRate;
uint64_t fNumMsgs;
std::string fOutChannelName;
virtual void InitTask();

9
fairmq/devices/FairMQSink.cxx
View File

@@ -34,7 +34,7 @@ void FairMQSink::InitTask()
void FairMQSink::Run()
{
int numReceivedMsgs = 0;
uint64_t numReceivedMsgs = 0;
// store the channel reference to avoid traversing the map on every loop iteration
const FairMQChannel& dataInChannel = fChannels.at(fInChannelName).at(0);
@@ -43,25 +43,24 @@ void FairMQSink::Run()
while (CheckCurrentState(RUNNING))
{
std::unique_ptr<FairMQMessage> msg(fTransportFactory->CreateMessage());
FairMQMessagePtr msg(fTransportFactory->CreateMessage());
if (dataInChannel.Receive(msg) >= 0)
{
if (fNumMsgs > 0)
{
numReceivedMsgs++;
if (numReceivedMsgs >= fNumMsgs)
{
break;
}
}
numReceivedMsgs++;
}
}
auto tEnd = chrono::high_resolution_clock::now();
LOG(INFO) << "Received " << numReceivedMsgs << " messages, leaving RUNNING state.";
LOG(INFO) << "Receiving time: " << chrono::duration<double, milli>(tEnd - tStart).count() << " ms";
LOG(INFO) << "Leaving RUNNING state. Received " << numReceivedMsgs << " messages in " << chrono::duration<double, milli>(tEnd - tStart).count() << "ms.";
}
FairMQSink::~FairMQSink()