FairMQ/fairmq/devices/GenericFileSink.tpl

/* 
 * File:   GenericFileSink.tpl
 * Author: winckler
 *
 * Created on October 7, 2014, 7:21 PM
 */

template <typename InputPolicy, typename OutputPolicy>
GenericFileSink<InputPolicy, OutputPolicy>::GenericFileSink()
    : InputPolicy()
    , OutputPolicy()
{
}

template <typename InputPolicy, typename OutputPolicy>
GenericFileSink<InputPolicy, OutputPolicy>::~GenericFileSink()
{
}

template <typename InputPolicy, typename OutputPolicy>
void GenericFileSink<InputPolicy, OutputPolicy>::SetTransport(FairMQTransportFactory* transport)
{
    FairMQDevice::SetTransport(transport);
    // InputPolicy::SetTransport(transport);
}


template <typename InputPolicy, typename OutputPolicy>
void GenericFileSink<InputPolicy, OutputPolicy>::InitTask()
{
    InitOutputFile();
    // InputPolicy::Init();
    // OutputPolicy::Init();
}

template <typename InputPolicy, typename OutputPolicy>
void GenericFileSink<InputPolicy, OutputPolicy>::InitOutputFile()
{
    OutputPolicy::InitOutFile();
}

template <typename InputPolicy, typename OutputPolicy>
void GenericFileSink<InputPolicy, OutputPolicy>::Run()
{
    int receivedMsg = 0;

    // store the channel reference to avoid traversing the map on every loop iteration
    const FairMQChannel& inputChannel = fChannels["data-in"].at(0);

    while (CheckCurrentState(RUNNING))
    {
        std::unique_ptr<FairMQMessage> msg(fTransportFactory->CreateMessage());

        if (inputChannel.Receive(msg) > 0)
        {
            OutputPolicy::AddToFile(InputPolicy::DeSerializeMsg(msg.get()));
            receivedMsg++;
        }
    }

    MQLOG(INFO) << "Received " << receivedMsg << " messages!";
}