ProcessMonitor.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Anselm Baur, Nils Braun                                  *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <framework/pcore/ProcHelper.h>
#include <framework/pcore/zmq/sockets/ZMQClient.h>
#include <framework/pcore/zmq/utils/StreamHelper.h>
 
#include <string>
#include <map>
 
namespace Belle2 {
  class ProcessMonitor {
  public:
    void subscribe(const std::string& pubSocketAddress, const std::string& subSocketAddress,
                   const std::string& controlSocketAddress);
    void waitForRunningInput(int timeout);
    void waitForRunningOutput(int timeout);
    void waitForRunningWorker(int timeout);
    void killProcesses(unsigned int timeout);
 
    void initialize(unsigned int requestedNumberOfWorkers);
 
    void terminate();
 
    void reset();
 
    void checkMulticast(int timeout = 0);
    void checkChildProcesses();
    void checkSignals(int g_signalReceived);
 
    bool hasEnded() const;
 
    unsigned int needMoreWorkers() const;
 
    bool hasWorkers() const;
 
  private:
    ZMQClient m_client;
 
    unsigned int m_requestedNumberOfWorkers = 0;
    std::map<int, ProcType> m_processList;
    bool m_hasEnded = false;
 
    StreamHelper m_streamer;
 
    bool m_receivedStatistics = false;
 
    unsigned int processesWithType(const ProcType& procType) const;
 
    template <class ASocket>
    void processMulticast(const ASocket& socket);
  };
} // namespace Belle2