PMIxPlugin.h

/********************************************************************************
 *    Copyright (C) 2019 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *              GNU Lesser General Public Licence (LGPL) version 3,             *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
 
#ifndef FAIR_MQ_PLUGINS_PMIX
#define FAIR_MQ_PLUGINS_PMIX
 
#include "PMIx.hpp"
#include "PMIxCommands.h"
 
#include <fairmq/Plugin.h>
#include <fairmq/Version.h>
#include <FairMQLogger.h>
 
#include <string>
#include <sstream>
#include <stdexcept>
#include <string>
#include <sys/types.h>
#include <unistd.h>
#include <vector>
 
namespace fair::mq::plugins
{
 
class PMIxPlugin : public Plugin
{
  public:
    PMIxPlugin(const std::string& name,
         const Plugin::Version version,
         const std::string& maintainer,
         const std::string& homepage,
         PluginServices* pluginServices);
    ~PMIxPlugin();
 
    auto PMIxClient() const -> std::string { return fPMIxClient; };
 
  private:
    pmix::proc fProcess;
    pid_t fPid;
    std::string fPMIxClient;
    std::string fDeviceId;
    pmix::Commands fCommands;
 
    std::set<uint32_t> fStateChangeSubscribers;
    uint32_t fLastExternalController;
    bool fExitingAckedByLastExternalController;
    std::condition_variable fExitingAcked;
    std::mutex fStateChangeSubscriberMutex;
 
    DeviceState fCurrentState;
    DeviceState fLastState;
 
    auto Init() -> pmix::proc;
    auto Publish() -> void;
    auto Fence() -> void;
    auto Fence(const std::string& label) -> void;
    auto Lookup() -> void;
 
    auto SubscribeForCommands() -> void;
    auto WaitForExitingAck() -> void;
};
 
Plugin::ProgOptions PMIxProgramOptions()
{
    boost::program_options::options_description options("PMIx Plugin");
    options.add_options()
        ("pmix-dummy", boost::program_options::value<int>()->default_value(0), "Dummy.");
    return options;
}
 
REGISTER_FAIRMQ_PLUGIN(
    PMIxPlugin,                                  // Class name
    pmix,                                        // Plugin name (string, lower case chars only)
    (Plugin::Version{FAIRMQ_VERSION_MAJOR,
                     FAIRMQ_VERSION_MINOR,
                     FAIRMQ_VERSION_PATCH}),     // Version
    "FairRootGroup <fairroot@gsi.de>",           // Maintainer
    "https://github.com/FairRootGroup/FairMQ",   // Homepage
    PMIxProgramOptions                           // custom program options for the plugin
)
 
} // namespace fair::mq::plugins
 
#endif /* FAIR_MQ_PLUGINS_PMIX */