mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 08:41:16 +00:00
fc94342db8
- Split INITIALIZING state into Init+Bind+Connect - Remove PAUSE state - Convert state/transitions to enum classes (CamelCase) - Transition to a state only once previous handler is complete - Add CompleteInit transition to notify Initializing state that config updates are complete - Deprecate WaitForEndOfState(transition) in favor of WaitForState(state)/WaitForNextState() - Update tests/plugins to new APIs - Deprecate CheckCurrentState() in favor of NewStatePending()
131 lines
3.4 KiB
C++
131 lines
3.4 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2015-2017 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH *
|
|
* *
|
|
* This software is distributed under the terms of the *
|
|
* GNU Lesser General Public Licence (LGPL) version 3, *
|
|
* copied verbatim in the file "LICENSE" *
|
|
********************************************************************************/
|
|
|
|
#ifndef FAIR_MQ_TEST_POLLIN_H
|
|
#define FAIR_MQ_TEST_POLLIN_H
|
|
|
|
#include <FairMQDevice.h>
|
|
#include <FairMQLogger.h>
|
|
#include <options/FairMQProgOptions.h>
|
|
#include <thread>
|
|
|
|
namespace fair
|
|
{
|
|
namespace mq
|
|
{
|
|
namespace test
|
|
{
|
|
|
|
using namespace std;
|
|
|
|
class PollIn : public FairMQDevice
|
|
{
|
|
public:
|
|
PollIn()
|
|
: fPollType(0)
|
|
{}
|
|
|
|
protected:
|
|
auto InitTask() -> void override
|
|
{
|
|
fPollType = fConfig->GetValue<int>("poll-type");
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(200));
|
|
}
|
|
|
|
auto Run() -> void override
|
|
{
|
|
vector<FairMQChannel*> chans;
|
|
|
|
chans.push_back(&fChannels.at("data1").at(0));
|
|
chans.push_back(&fChannels.at("data2").at(0));
|
|
|
|
FairMQPollerPtr poller = nullptr;
|
|
|
|
if (fPollType == 0)
|
|
{
|
|
poller = NewPoller(chans);
|
|
}
|
|
else if (fPollType == 1)
|
|
{
|
|
poller = NewPoller("data1", "data2");
|
|
}
|
|
else
|
|
{
|
|
LOG(error) << "wrong poll type provided: " << fPollType;
|
|
}
|
|
|
|
bool arrived1 = false;
|
|
bool arrived2 = false;
|
|
bool bothArrived = false;
|
|
|
|
FairMQMessagePtr msg1(NewMessage());
|
|
FairMQMessagePtr msg2(NewMessage());
|
|
|
|
while (!bothArrived)
|
|
{
|
|
poller->Poll(100);
|
|
|
|
if (fPollType == 0)
|
|
{
|
|
if (poller->CheckInput(0))
|
|
{
|
|
LOG(debug) << "CheckInput(0) triggered";
|
|
if (Receive(msg1, "data1", 0) >= 0)
|
|
{
|
|
arrived1 = true;
|
|
}
|
|
}
|
|
|
|
if (poller->CheckInput(1))
|
|
{
|
|
LOG(debug) << "CheckInput(1) triggered";
|
|
if (Receive(msg2, "data2", 0) >= 0)
|
|
{
|
|
arrived2 = true;
|
|
}
|
|
}
|
|
}
|
|
else if (fPollType == 1)
|
|
{
|
|
if (poller->CheckInput("data1", 0))
|
|
{
|
|
LOG(debug) << "CheckInput(\"data1\", 0) triggered";
|
|
if (Receive(msg1, "data1", 0) >= 0)
|
|
{
|
|
arrived1 = true;
|
|
}
|
|
}
|
|
|
|
if (poller->CheckInput("data2", 0))
|
|
{
|
|
LOG(debug) << "CheckInput(\"data2\", 0) triggered";
|
|
if (Receive(msg2, "data2", 0) >= 0)
|
|
{
|
|
arrived2 = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (arrived1 && arrived2)
|
|
{
|
|
bothArrived = true;
|
|
LOG(info) << "POLL test successfull";
|
|
}
|
|
}
|
|
};
|
|
|
|
private:
|
|
int fPollType;
|
|
};
|
|
|
|
} // namespace test
|
|
} // namespace mq
|
|
} // namespace fair
|
|
|
|
#endif /* FAIR_MQ_TEST_POLLIN_H */
|