#include #include #include #include #include #include #include #include #include #include #include #include "config_core.h" #include "DatabaseFactory.h" #include #if USE_MYSQL #include #endif namespace DB { namespace ErrorCodes { extern const int UNKNOWN_ELEMENT_IN_AST; extern const int BAD_ARGUMENTS; extern const int UNKNOWN_DATABASE_ENGINE; extern const int CANNOT_CREATE_DATABASE; } DatabasePtr DatabaseFactory::get( const String & database_name, const String & metadata_path, const ASTStorage * engine_define, Context & context) { try { Poco::File(metadata_path).createDirectory(); return getImpl(database_name, metadata_path, engine_define, context); } catch (...) { Poco::File metadata_dir(metadata_path); if (metadata_dir.exists()) metadata_dir.remove(true); throw; } } DatabasePtr DatabaseFactory::getImpl( const String & database_name, const String & metadata_path, const ASTStorage * engine_define, Context & context) { String engine_name = engine_define->engine->name; if (engine_name != "MySQL" && engine_name != "Lazy" && engine_define->engine->arguments) throw Exception("Database engine " + engine_name + " cannot have arguments", ErrorCodes::BAD_ARGUMENTS); if (engine_define->engine->parameters || engine_define->partition_by || engine_define->primary_key || engine_define->order_by || engine_define->sample_by || engine_define->settings) throw Exception("Database engine " + engine_name + " cannot have parameters, primary_key, order_by, sample_by, settings", ErrorCodes::UNKNOWN_ELEMENT_IN_AST); if (engine_name == "Ordinary") return std::make_shared(database_name, metadata_path, context); else if (engine_name == "Atomic") return std::make_shared(database_name, metadata_path, context); else if (engine_name == "Memory") return std::make_shared(database_name); else if (engine_name == "Dictionary") return std::make_shared(database_name); #if USE_MYSQL else if (engine_name == "MySQL") { const ASTFunction * engine = engine_define->engine; if (!engine->arguments || engine->arguments->children.size() != 4) throw Exception("MySQL Database require mysql_hostname, mysql_database_name, mysql_username, mysql_password arguments.", ErrorCodes::BAD_ARGUMENTS); const auto & arguments = engine->arguments->children; const auto & host_name_and_port = arguments[0]->as()->value.safeGet(); const auto & database_name_in_mysql = arguments[1]->as()->value.safeGet(); const auto & mysql_user_name = arguments[2]->as()->value.safeGet(); const auto & mysql_user_password = arguments[3]->as()->value.safeGet(); try { const auto & [remote_host_name, remote_port] = parseAddress(host_name_and_port, 3306); auto mysql_pool = mysqlxx::Pool(database_name_in_mysql, remote_host_name, mysql_user_name, mysql_user_password, remote_port); auto mysql_database = std::make_shared( context, database_name, metadata_path, engine_define, database_name_in_mysql, std::move(mysql_pool)); mysql_database->empty(context); /// test database is works fine. return mysql_database; } catch (...) { const auto & exception_message = getCurrentExceptionMessage(true); throw Exception("Cannot create MySQL database, because " + exception_message, ErrorCodes::CANNOT_CREATE_DATABASE); } } #endif else if (engine_name == "Lazy") { const ASTFunction * engine = engine_define->engine; if (!engine->arguments || engine->arguments->children.size() != 1) throw Exception("Lazy database require cache_expiration_time_seconds argument", ErrorCodes::BAD_ARGUMENTS); const auto & arguments = engine->arguments->children; const auto cache_expiration_time_seconds = arguments[0]->as()->value.safeGet(); return std::make_shared(database_name, metadata_path, cache_expiration_time_seconds, context); } throw Exception("Unknown database engine: " + engine_name, ErrorCodes::UNKNOWN_DATABASE_ENGINE); } }