mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 16:46:47 +00:00
96 lines
2.8 KiB
C++
96 lines
2.8 KiB
C++
/********************************************************************************
|
|
* 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
|
|
{
|
|
namespace mq
|
|
{
|
|
namespace 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 plugins */
|
|
} /* namespace mq */
|
|
} /* namespace fair */
|
|
|
|
#endif /* FAIR_MQ_PLUGINS_PMIX */
|