1b9eb4cb9f
The init.hh contains some bits that only main.cc needs. Some of its forward declarations are neede by neither the headers itself, nor the main.cc that includes it. Signed-off-by: Pavel Emelyanov <xemul@scylladb.com> Closes scylladb/scylladb#25110
127 lines
3.9 KiB
C++
127 lines
3.9 KiB
C++
/*
|
|
* Copyright (C) 2015-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
|
|
*/
|
|
#pragma once
|
|
|
|
#include <any>
|
|
|
|
#include <seastar/core/sstring.hh>
|
|
#include <seastar/core/future.hh>
|
|
#include <seastar/core/distributed.hh>
|
|
#include <seastar/core/abort_source.hh>
|
|
#include "utils/log.hh"
|
|
#include "utils/s3/creds.hh"
|
|
#include "seastarx.hh"
|
|
#include <boost/program_options.hpp>
|
|
#include <yaml-cpp/yaml.h>
|
|
|
|
namespace db {
|
|
class extensions;
|
|
class config;
|
|
}
|
|
|
|
namespace gms {
|
|
class inet_address;
|
|
}
|
|
|
|
extern logging::logger startlog;
|
|
|
|
class bad_configuration_error : public std::exception {};
|
|
|
|
[[nodiscard]] std::set<gms::inet_address> get_seeds_from_db_config(const db::config& cfg,
|
|
gms::inet_address broadcast_address,
|
|
bool fail_on_lookup_error);
|
|
|
|
class service_set {
|
|
public:
|
|
service_set();
|
|
~service_set();
|
|
template<typename... Args>
|
|
service_set(seastar::sharded<Args>&... args)
|
|
: service_set()
|
|
{
|
|
(..., add(args));
|
|
}
|
|
template<typename T>
|
|
void add(seastar::sharded<T>& t) {
|
|
add(std::any{std::addressof(t)});
|
|
}
|
|
template<typename T>
|
|
seastar::sharded<T>& find() const {
|
|
return *std::any_cast<seastar::sharded<T>*>(find(typeid(seastar::sharded<T>*)));
|
|
}
|
|
private:
|
|
void add(std::any);
|
|
std::any find(const std::type_info&) const;
|
|
|
|
class impl;
|
|
std::unique_ptr<impl> _impl;
|
|
};
|
|
|
|
/**
|
|
* Very simplistic config registry. Allows hooking in a config object
|
|
* to the "main" sequence.
|
|
*/
|
|
class configurable {
|
|
public:
|
|
configurable() {
|
|
// We auto register. Not that like cycle is assumed to be forever
|
|
// and scope should be managed elsewhere.
|
|
register_configurable(*this);
|
|
}
|
|
virtual ~configurable()
|
|
{}
|
|
// Hook to add command line options and/or add main config options
|
|
virtual void append_options(db::config&, boost::program_options::options_description_easy_init&)
|
|
{};
|
|
// Called after command line is parsed and db/config populated.
|
|
// Hooked config can for example take this opportunity to load any file(s).
|
|
virtual future<> initialize(const boost::program_options::variables_map&) {
|
|
return make_ready_future();
|
|
}
|
|
virtual future<> initialize(const boost::program_options::variables_map& map, const db::config& cfg, db::extensions& exts) {
|
|
return initialize(map);
|
|
}
|
|
|
|
/**
|
|
* State of initiating system for optional
|
|
* notification callback to objects created by
|
|
* `initialize`
|
|
*/
|
|
enum class system_state {
|
|
started,
|
|
stopped,
|
|
};
|
|
|
|
using notify_func = std::function<future<>(system_state)>;
|
|
|
|
/** Hooks should override this to allow adding a notification function to the startup sequence. */
|
|
virtual future<notify_func> initialize_ex(const boost::program_options::variables_map& map, const db::config& cfg, db::extensions& exts) {
|
|
return initialize(map, cfg, exts).then([] { return notify_func{}; });
|
|
}
|
|
|
|
virtual future<notify_func> initialize_ex(const boost::program_options::variables_map& map, const db::config& cfg, db::extensions& exts, const service_set&) {
|
|
return initialize_ex(map, cfg, exts);
|
|
}
|
|
|
|
class notify_set {
|
|
public:
|
|
future<> notify_all(system_state);
|
|
private:
|
|
friend class configurable;
|
|
std::vector<notify_func> _listeners;
|
|
};
|
|
|
|
// visible for testing
|
|
static std::vector<std::reference_wrapper<configurable>>& configurables();
|
|
static future<notify_set> init_all(const boost::program_options::variables_map&, const db::config&, db::extensions&, const service_set& = {});
|
|
static future<notify_set> init_all(const db::config&, db::extensions&, const service_set& = {});
|
|
static void append_all(db::config&, boost::program_options::options_description_easy_init&);
|
|
private:
|
|
static void register_configurable(configurable &);
|
|
};
|