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

FairMQ: Add test for example/advanced/Region.

Browse Source 
Also fix a regression in nanomsg transport.
This commit is contained in:
Alexey Rybalchenko
2017-11-28 15:37:41 +01:00
parent 15479a58d1
commit 5b091f7eec
9 changed files with 90 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
examples/advanced/Region/FairMQExampleRegionSampler.cxx
View File

@@ -23,6 +23,8 @@ using namespace std;
FairMQExampleRegionSampler::FairMQExampleRegionSampler()
: fMsgSize(10000)
, fMaxIterations(0)
, fNumIterations(0)
, fRegion(nullptr)
, fNumUnackedMsgs(0)
{
@@ -31,11 +33,18 @@ FairMQExampleRegionSampler::FairMQExampleRegionSampler()
void FairMQExampleRegionSampler::InitTask()
{
fMsgSize = fConfig->GetValue<int>("msg-size");
fMaxIterations = fConfig->GetValue<uint64_t>("max-iterations");
fRegion = FairMQUnmanagedRegionPtr(NewUnmanagedRegionFor("data",
0,
10000000,
[this](void* data, size_t size) { --fNumUnackedMsgs; } // callback to be called when message buffers no longer needed by transport
[this](void* data, size_t size) { // callback to be called when message buffers no longer needed by transport
--fNumUnackedMsgs;
if (fMaxIterations > 0)
{
LOG(DEBUG) << "Received ack";
}
}
));
}
@@ -50,6 +59,12 @@ bool FairMQExampleRegionSampler::ConditionalRun()
if (Send(msg, "data", 0) > 0)
{
++fNumUnackedMsgs;
if (fMaxIterations > 0 && ++fNumIterations >= fMaxIterations)
{
LOG(INFO) << "Configured maximum number of iterations reached. Leaving RUNNING state.";
return false;
}
}
return true;