ZMQApp.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 <framework/pcore/zmq/utils/ZMQParent.h>
#include <framework/pcore/zmq/connections/ZMQSimpleConnection.h>
#include <daq/rfarm/event/hltsocket/HLTMainLoop.h>
#include <framework/pcore/zmq/messages/ZMQMessageFactory.h>
#include <framework/logging/Logger.h>
 
#include <boost/program_options.hpp>
#include <iostream>
 
namespace po = boost::program_options;
 
namespace Belle2 {
  template<class AInputConnection, class AOutputConnection>
  class ZMQStandardApp  {
  public:
    void initFromConsole(const std::string& description, int argc, char* argv[]);
 
    void main();
 
  protected:
    std::shared_ptr<ZMQParent> m_parent;
    std::unique_ptr<AInputConnection> m_input;
    std::unique_ptr<AOutputConnection> m_output;
    std::unique_ptr<ZMQSimpleConnection> m_monitor;
    bool m_terminate = false;
    unsigned int m_timeout = 0;
    bool m_monitorHasPriority = false;
 
    virtual void initialize();
    virtual void addOptions(po::options_description& desc);
    virtual void handleExternalSignal(EMessageTypes);
    virtual void handleInput();
    virtual void handleOutput();
    virtual void handleTimeout();
    virtual void fillMonitoringJSON(std::stringstream& buffer) const;
    bool terminated() const;
    void resetTimer();
    void pollEvent(bool pollOnInput);
 
  private:
    HLTMainLoop m_mainLoop;
    std::string m_monitoringAddress;
    int m_remainingTime = -1;
    std::chrono::system_clock::time_point m_start;
 
    bool checkTimer();
    void updateTimer();
    void handleMonitoring();
  };
}
 
#include <daq/hbasf2/apps/ZMQApp.details.h>