mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 00:31:14 +00:00
7fda980710
Organize sockets as a map of vectors of FairMQChannels. Update FairMQStateMachine by removing SETTINGINPUT, SETTINGOUTPUT, BIND and CONNECT states and by adding INITIALIZING_TASK, RESETTING_TASK and RESETTING_DEVICE states. Run states functions in their own thread.
93 lines
2.6 KiB
C++
93 lines
2.6 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH *
|
|
* *
|
|
* This software is distributed under the terms of the *
|
|
* GNU Lesser General Public Licence version 3 (LGPL) version 3, *
|
|
* copied verbatim in the file "LICENSE" *
|
|
********************************************************************************/
|
|
|
|
/*
|
|
* File: GenericFileSink.h
|
|
* Author: winckler
|
|
*
|
|
* Created on October 7, 2014, 6:06 PM
|
|
*/
|
|
|
|
#ifndef GENERICFILESINK_H
|
|
#define GENERICFILESINK_H
|
|
|
|
#include "FairMQDevice.h"
|
|
#include <boost/thread.hpp>
|
|
#include <boost/bind.hpp>
|
|
#include "FairMQLogger.h"
|
|
|
|
/*********************************************************************
|
|
* -------------- NOTES -----------------------
|
|
* All policies must have a default constructor
|
|
* Function to define in (parent) policy classes :
|
|
*
|
|
* -------- INPUT POLICY --------
|
|
* InputPolicy::InitContainer(...)
|
|
* CONTAINER_TYPE InputPolicy::DeSerializeMsg(FairMQMessage* msg)
|
|
*
|
|
*
|
|
* -------- OUTPUT POLICY --------
|
|
* OutputPolicy::AddToFile(CONTAINER_TYPE);
|
|
* OutputPolicy::InitOutputFile()
|
|
**********************************************************************/
|
|
|
|
#include "FairMQDevice.h"
|
|
|
|
template <typename InputPolicy, typename OutputPolicy>
|
|
class GenericFileSink : public FairMQDevice, public InputPolicy, public OutputPolicy
|
|
{
|
|
public:
|
|
GenericFileSink()
|
|
: InputPolicy()
|
|
, OutputPolicy()
|
|
{}
|
|
|
|
virtual ~GenericFileSink()
|
|
{}
|
|
|
|
void SetTransport(FairMQTransportFactory* transport)
|
|
{
|
|
FairMQDevice::SetTransport(transport);
|
|
}
|
|
|
|
template <typename... Args>
|
|
void InitInputContainer(Args... args)
|
|
{
|
|
InputPolicy::InitContainer(std::forward<Args>(args)...);
|
|
}
|
|
|
|
protected:
|
|
virtual void InitTask()
|
|
{
|
|
OutputPolicy::InitOutputFile();
|
|
}
|
|
|
|
virtual void Run()
|
|
{
|
|
int receivedMsg = 0;
|
|
|
|
while (GetCurrentState() == RUNNING)
|
|
{
|
|
FairMQMessage* msg = fTransportFactory->CreateMessage();
|
|
|
|
if (fChannels["data-in"].at(0).Receive(msg) > 0)
|
|
{
|
|
OutputPolicy::AddToFile(InputPolicy::DeSerializeMsg(msg));
|
|
receivedMsg++;
|
|
}
|
|
|
|
delete msg;
|
|
}
|
|
|
|
MQLOG(INFO) << "Received " << receivedMsg << " messages!";
|
|
}
|
|
};
|
|
|
|
#endif /* GENERICFILESINK_H */
|
|
|