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

add serialization API

Browse Source
This commit is contained in:
winckler
2016-03-24 15:59:33 +01:00
parent 9a340fb7f6
commit 4c50409af5
5 changed files with 218 additions and 281 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
fairmq/devices/GenericFileSink.h
View File

@@ -41,55 +41,48 @@
template <typename T, typename U>
class GenericFileSink : public FairMQDevice, public T, public U
{
protected:
typedef T deserialization_type;
typedef U sink_type;
public:
typedef T input_policy;
typedef U sink_type;
GenericFileSink()
: deserialization_type()
, sink_type()
: FairMQDevice(), input_policy(), sink_type()
{}
virtual ~GenericFileSink()
{}
template <typename... Args>
void SetTransport(Args... args)
{
FairMQDevice::SetTransport(std::forward<Args>(args)...);
}
template <typename... Args>
void InitInputContainer(Args... args)
template<typename... Args>
void InitInputData(Args&&... args)
{
deserialization_type::InitContainer(std::forward<Args>(args)...);
input_policy::Create(std::forward<Args>(args)...);
}
protected:
using input_policy::fInput;
virtual void InitTask()
{
sink_type::InitOutputFile();
}
typedef typename input_policy::deserialization_type deserializer_type;
virtual void Run()
{
int receivedMsg = 0;
// store the channel reference to avoid traversing the map on every loop iteration
const FairMQChannel& inputChannel = fChannels.at("data-in").at(0);
while (CheckCurrentState(RUNNING))
{
std::unique_ptr<FairMQMessage> msg(fTransportFactory->CreateMessage());
if (inputChannel.Receive(msg) > 0)
if (Receive<deserializer_type>(fInput, "data-in") > 0)
{
sink_type::AddToFile(deserialization_type::DeserializeMsg(msg.get()));
U::Serialize(fInput);// add fInput to file
receivedMsg++;
}
}
MQLOG(INFO) << "Received " << receivedMsg << " messages!";
LOG(INFO) << "Received " << receivedMsg << " messages!";
}
};