ZMQDistributor.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#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;
  };
}