add environment class that can live outside manager

1 files changed, 84 insertions, 0 deletions

diff --git a/configd/src/apps/sentinel/env.cpp b/configd/src/apps/sentinel/env.cpp
new file mode 100644
index 00000000000..02df292a423
--- /dev/null
+++ b/configd/src/apps/sentinel/env.cpp
@@ -0,0 +1,84 @@
+// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+#include "env.h"
+#include <vespa/log/log.h>
+#include <vespa/config/common/exceptions.h>
+#include <vespa/vespalib/util/exceptions.h>
+#include <thread>
+#include <chrono>
+
+LOG_SETUP(".env");
+
+namespace config::sentinel {
+
+constexpr std::chrono::milliseconds CONFIG_TIMEOUT_MS(3 * 60 * 1000);
+
+Env::Env()
+  : _cfgOwner(),
+    _rpcCommandQueue(),
+    _rpcServer(),
+    _stateApi(),
+    _startMetrics(),
+    _stateServer(),
+    _statePort(0)
+{
+    _startMetrics.startedTime = vespalib::steady_clock::now();
+}
+
+Env::~Env() = default;
+
+void Env::boot(const std::string &configId) {
+    LOG(debug, "Reading configuration for ID: %s", configId.c_str());
+    _cfgOwner.subscribe(configId, CONFIG_TIMEOUT_MS);
+    if (_cfgOwner.checkForConfigUpdate()) {
+        LOG_ASSERT(_cfgOwner.hasConfig());
+    } else {
+        throw InvalidConfigException("checkForConfigUpdate() failed");
+    }
+    const auto & cfg = _cfgOwner.getConfig();
+    LOG(config, "Booting sentinel '%s' with [stateserver port %d] and [rpc port %d]",
+        configId.c_str(), cfg.port.telnet, cfg.port.rpc);
+    rpcPort(cfg.port.rpc);
+    statePort(cfg.port.telnet);
+}
+
+void Env::rpcPort(int port) {
+    if (port < 0 || port > 65535) {
+        throw vespalib::FatalException("Bad port " + std::to_string(port) + ", expected range [1, 65535]", VESPA_STRLOC);
+    }
+    if (port == 0) {
+        port = 19097; // default in config
+    }
+    if (_rpcServer && port == _rpcServer->getPort()) {
+        return; // ok already
+    }
+    _rpcServer = std::make_unique<RpcServer>(port, _rpcCommandQueue);
+}
+
+void Env::statePort(int port) {
+    if (port < 0 || port > 65535) {
+        throw vespalib::FatalException("Bad port " + std::to_string(port) + ", expected range [1, 65535]", VESPA_STRLOC);
+    }
+    if (port == 0) {
+        port = 19098;
+        const char *portString = getenv("VESPA_SENTINEL_PORT");
+        if (portString) {
+            port = strtoul(portString, nullptr, 10);
+        }
+    }
+    if (_stateServer && port == _statePort) {
+        return; // ok already
+    }
+    LOG(debug, "Config-sentinel accepts connections on port %d", port);
+    _stateServer = std::make_unique<vespalib::StateServer>(
+        port, _stateApi.myHealth, _startMetrics.producer, _stateApi.myComponents);
+    _statePort = port;
+}
+
+void Env::notifyConfigUpdated() {
+    vespalib::ComponentConfigProducer::Config current("sentinel", _cfgOwner.getGeneration(), "ok");
+    _stateApi.myComponents.addConfig(current);
+
+}
+
+}