ZMQDistributor.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Nils Braun                                               *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <daq/hbasf2/apps/ZMQApp.h>
#include <framework/pcore/zmq/connections/ZMQLoadBalancedConnection.h>
#include <framework/pcore/zmq/connections/ZMQRawConnection.h>
#include <framework/pcore/zmq/connections/ZMQConfirmedConnection.h>
 
#include <boost/program_options.hpp>
 
namespace po = boost::program_options;
 
namespace Belle2 {
  class ZMQDistributor : public ZMQStandardApp<ZMQRawInput, ZMQLoadBalancedOutput> {
  protected:
    void initialize() final;
    void addOptions(po::options_description& desc) final;
    void handleExternalSignal(EMessageTypes type) final;
    void handleInput() final;
    void handleTimeout() final;
 
  private:
    std::string m_inputAddress;
    std::string m_outputAddress;
    unsigned int m_maximalBufferSize = 80'000'000;
    bool m_expressRecoMode = false;
    unsigned int m_stopWaitingTime = 2;
  };
 
  class ZMQInputAdapter : public ZMQStandardApp<ZMQRawInput, ZMQConfirmedOutput> {
  protected:
    void initialize() final;
    void addOptions(po::options_description& desc) final;
    void handleExternalSignal(EMessageTypes type) final;
    void handleInput() final;
    void handleTimeout() final;
 
  private:
    std::string m_inputAddress;
    std::string m_outputAddress;
    unsigned int m_maximalBufferSize = 80'000'000;
    bool m_expressRecoMode = false;
    unsigned int m_stopWaitingTime = 2;
  };
}