mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 16:46:47 +00:00
91 lines
5.0 KiB
C++
91 lines
5.0 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2017-2018 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" *
|
|
********************************************************************************/
|
|
|
|
#include "Config.h"
|
|
|
|
#include <fairmq/JSONParser.h>
|
|
#include <fairmq/SuboptParser.h>
|
|
|
|
#include <vector>
|
|
|
|
using namespace std;
|
|
|
|
namespace fair
|
|
{
|
|
namespace mq
|
|
{
|
|
namespace plugins
|
|
{
|
|
|
|
Config::Config(const string& name, const Plugin::Version version, const string& maintainer, const string& homepage, PluginServices* pluginServices)
|
|
: Plugin(name, version, maintainer, homepage, pluginServices)
|
|
{
|
|
SubscribeToDeviceStateChange([&](DeviceState newState) {
|
|
if (newState == DeviceState::InitializingDevice) {
|
|
string idForParser;
|
|
|
|
if (PropertyExists("config-key")) {
|
|
idForParser = GetProperty<string>("config-key");
|
|
} else if (PropertyExists("id")) {
|
|
idForParser = GetProperty<string>("id");
|
|
}
|
|
|
|
if (idForParser != "") {
|
|
try {
|
|
if (PropertyExists("mq-config")) {
|
|
LOG(debug) << "mq-config: Using default JSON parser";
|
|
SetProperties(JSONParser(GetProperty<string>("mq-config"), idForParser));
|
|
} else if (PropertyExists("channel-config")) {
|
|
LOG(debug) << "channel-config: Parsing channel configuration";
|
|
SetProperties(SuboptParser(GetProperty<vector<string>>("channel-config"), idForParser));
|
|
} else {
|
|
LOG(warn) << "fair::mq::plugins::Config: no channels configuration provided via --mq-config or --channel-config";
|
|
}
|
|
} catch (exception& e) {
|
|
LOG(error) << e.what();
|
|
}
|
|
} else {
|
|
LOG(warn) << "No device ID or config-key provided for the configuration parser.";
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
Plugin::ProgOptions ConfigPluginProgramOptions()
|
|
{
|
|
namespace po = boost::program_options;
|
|
auto pluginOptions = po::options_description{"FairMQ device options"};
|
|
pluginOptions.add_options()
|
|
("id", po::value<string >()->default_value(""), "Device ID.")
|
|
("io-threads", po::value<int >()->default_value(1), "Number of I/O threads.")
|
|
("transport", po::value<string >()->default_value("zeromq"), "Transport ('zeromq'/'shmem').")
|
|
("network-interface", po::value<string >()->default_value("default"), "Network interface to bind on (e.g. eth0, ib0..., default will try to detect the interface of the default route).")
|
|
("init-timeout", po::value<int >()->default_value(120), "Timeout for the initialization in seconds (when expecting dynamic initialization).")
|
|
("max-run-time", po::value<uint64_t >()->default_value(0), "Maximum runtime for the Running state handler, after which state will change to Ready (in seconds, 0 for no limit).")
|
|
("print-channels", po::value<bool >()->implicit_value(true), "Print registered channel endpoints in a machine-readable format (<channel name>:<min num subchannels>:<max num subchannels>)")
|
|
("shm-segment-size", po::value<size_t >()->default_value(2000000000), "Shared memory: size of the shared memory segment (in bytes).")
|
|
("shm-throw-bad-alloc", po::value<bool >()->default_value(true), "Throw a fair::mq::MessageBadAlloc if cannot allocate a message (retry if false).")
|
|
("shm-monitor", po::value<bool >()->default_value(true), "Shared memory: run monitor daemon.")
|
|
("ofi-size-hint", po::value<size_t >()->default_value(0), "EXPERIMENTAL: OFI size hint for the allocator.")
|
|
("rate", po::value<float >()->default_value(0.), "Rate for conditional run loop (Hz).")
|
|
("session", po::value<string >()->default_value("default"), "Session name.")
|
|
("config-key", po::value<string >(), "Use provided value instead of device id for fetching the configuration from JSON file.")
|
|
("mq-config", po::value<string >(), "JSON input as file.")
|
|
("channel-config", po::value<vector<string>>()->multitoken()->composing(), "Configuration of single or multiple channel(s) by comma separated key=value list");
|
|
return pluginOptions;
|
|
}
|
|
|
|
Config::~Config()
|
|
{
|
|
UnsubscribeFromDeviceStateChange();
|
|
}
|
|
|
|
} /* namespace plugins */
|
|
} /* namespace mq */
|
|
} /* namespace fair */
|