mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-29 19:12:03 +00:00
100 lines
3.4 KiB
C++
100 lines
3.4 KiB
C++
#include "DictionaryFactory.h"
|
|
|
|
#include <memory>
|
|
#include "DictionarySourceFactory.h"
|
|
#include "DictionaryStructure.h"
|
|
#include "getDictionaryConfigurationFromAST.h"
|
|
#include <Interpreters/Context.h>
|
|
#include <Common/logger_useful.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int LOGICAL_ERROR;
|
|
extern const int EXCESSIVE_ELEMENT_IN_CONFIG;
|
|
extern const int UNKNOWN_ELEMENT_IN_CONFIG;
|
|
}
|
|
|
|
void DictionaryFactory::registerLayout(const std::string & layout_type, LayoutCreateFunction create_layout, bool is_layout_complex)
|
|
{
|
|
auto it = registered_layouts.find(layout_type);
|
|
if (it != registered_layouts.end())
|
|
throw Exception(ErrorCodes::LOGICAL_ERROR, "DictionaryFactory: the layout name '{}' is not unique", layout_type);
|
|
|
|
RegisteredLayout layout { .layout_create_function = create_layout, .is_layout_complex = is_layout_complex };
|
|
registered_layouts.emplace(layout_type, std::move(layout));
|
|
}
|
|
|
|
DictionaryPtr DictionaryFactory::create(
|
|
const std::string & name,
|
|
const Poco::Util::AbstractConfiguration & config,
|
|
const std::string & config_prefix,
|
|
ContextPtr global_context,
|
|
bool created_from_ddl) const
|
|
{
|
|
Poco::Util::AbstractConfiguration::Keys keys;
|
|
const auto & layout_prefix = config_prefix + ".layout";
|
|
config.keys(layout_prefix, keys);
|
|
if (keys.size() != 1)
|
|
throw Exception(ErrorCodes::EXCESSIVE_ELEMENT_IN_CONFIG,
|
|
"{}: element dictionary.layout should have exactly one child element",
|
|
name);
|
|
|
|
const DictionaryStructure dict_struct{config, config_prefix};
|
|
|
|
DictionarySourcePtr source_ptr = DictionarySourceFactory::instance().create(
|
|
name, config, config_prefix + ".source", dict_struct, global_context, config.getString(config_prefix + ".database", ""), created_from_ddl);
|
|
LOG_TRACE(&Poco::Logger::get("DictionaryFactory"), "Created dictionary source '{}' for dictionary '{}'", source_ptr->toString(), name);
|
|
|
|
const auto & layout_type = keys.front();
|
|
|
|
{
|
|
const auto found = registered_layouts.find(layout_type);
|
|
if (found != registered_layouts.end())
|
|
{
|
|
const auto & layout_creator = found->second.layout_create_function;
|
|
auto result = layout_creator(name, dict_struct, config, config_prefix, std::move(source_ptr), global_context, created_from_ddl);
|
|
if (config.hasProperty(config_prefix + ".comment"))
|
|
result->setDictionaryComment(config.getString(config_prefix + ".comment"));
|
|
|
|
return result;
|
|
}
|
|
}
|
|
|
|
throw Exception(ErrorCodes::UNKNOWN_ELEMENT_IN_CONFIG,
|
|
"{}: unknown dictionary layout type: {}",
|
|
name,
|
|
layout_type);
|
|
}
|
|
|
|
DictionaryPtr DictionaryFactory::create(const std::string & name, const ASTCreateQuery & ast, ContextPtr global_context) const
|
|
{
|
|
auto configuration = getDictionaryConfigurationFromAST(ast, global_context);
|
|
return DictionaryFactory::create(name, *configuration, "dictionary", global_context, true);
|
|
}
|
|
|
|
bool DictionaryFactory::isComplex(const std::string & layout_type) const
|
|
{
|
|
auto it = registered_layouts.find(layout_type);
|
|
|
|
if (it == registered_layouts.end())
|
|
{
|
|
throw Exception(ErrorCodes::UNKNOWN_ELEMENT_IN_CONFIG,
|
|
"Unknown dictionary layout type: {}",
|
|
layout_type);
|
|
}
|
|
|
|
return it->second.is_layout_complex;
|
|
}
|
|
|
|
|
|
DictionaryFactory & DictionaryFactory::instance()
|
|
{
|
|
static DictionaryFactory ret;
|
|
return ret;
|
|
}
|
|
|
|
}
|