2022-03-03 20:27:46 +00:00
|
|
|
#include "TinyContext.h"
|
|
|
|
|
|
|
|
#include <Common/Exception.h>
|
|
|
|
#include <Coordination/KeeperDispatcher.h>
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
namespace ErrorCodes
|
|
|
|
{
|
2022-07-01 03:13:33 +00:00
|
|
|
extern const int LOGICAL_ERROR;
|
2022-03-03 20:27:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void TinyContext::setConfig(const ConfigurationPtr & config_)
|
|
|
|
{
|
|
|
|
std::lock_guard lock(keeper_dispatcher_mutex);
|
|
|
|
config = config_;
|
|
|
|
}
|
|
|
|
|
|
|
|
const Poco::Util::AbstractConfiguration & TinyContext::getConfigRef() const
|
|
|
|
{
|
|
|
|
std::lock_guard lock(keeper_dispatcher_mutex);
|
|
|
|
return config ? *config : Poco::Util::Application::instance().config();
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void TinyContext::initializeKeeperDispatcher([[maybe_unused]] bool start_async) const
|
|
|
|
{
|
|
|
|
const auto & config_ref = getConfigRef();
|
|
|
|
|
|
|
|
std::lock_guard lock(keeper_dispatcher_mutex);
|
|
|
|
|
|
|
|
if (keeper_dispatcher)
|
2022-07-01 03:13:33 +00:00
|
|
|
throw Exception(ErrorCodes::LOGICAL_ERROR, "Trying to initialize Keeper multiple times");
|
2022-03-03 20:27:46 +00:00
|
|
|
|
|
|
|
if (config_ref.has("keeper_server"))
|
|
|
|
{
|
|
|
|
keeper_dispatcher = std::make_shared<KeeperDispatcher>();
|
|
|
|
keeper_dispatcher->initialize(config_ref, true, start_async);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
std::shared_ptr<KeeperDispatcher> TinyContext::getKeeperDispatcher() const
|
2022-07-04 02:26:19 +00:00
|
|
|
{
|
|
|
|
std::lock_guard lock(keeper_dispatcher_mutex);
|
|
|
|
if (!keeper_dispatcher)
|
|
|
|
throw Exception(ErrorCodes::LOGICAL_ERROR, "Keeper must be initialized before requests");
|
|
|
|
|
|
|
|
return keeper_dispatcher;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::shared_ptr<KeeperDispatcher> TinyContext::tryGetKeeperDispatcher() const
|
2022-03-03 20:27:46 +00:00
|
|
|
{
|
|
|
|
std::lock_guard lock(keeper_dispatcher_mutex);
|
|
|
|
return keeper_dispatcher;
|
|
|
|
}
|
|
|
|
|
|
|
|
void TinyContext::shutdownKeeperDispatcher() const
|
|
|
|
{
|
|
|
|
std::lock_guard lock(keeper_dispatcher_mutex);
|
|
|
|
if (keeper_dispatcher)
|
|
|
|
{
|
|
|
|
keeper_dispatcher->shutdown();
|
|
|
|
keeper_dispatcher.reset();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void TinyContext::updateKeeperConfiguration([[maybe_unused]] const Poco::Util::AbstractConfiguration & config_)
|
|
|
|
{
|
|
|
|
std::lock_guard lock(keeper_dispatcher_mutex);
|
|
|
|
if (!keeper_dispatcher)
|
|
|
|
return;
|
|
|
|
|
|
|
|
keeper_dispatcher->updateConfiguration(config_);
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|