mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 17:32:32 +00:00
69 lines
2.7 KiB
C++
69 lines
2.7 KiB
C++
#include <Interpreters/InterpreterBackupQuery.h>
|
|
#include <Backups/IBackup.h>
|
|
#include <Backups/IBackupEntry.h>
|
|
#include <Backups/IRestoreTask.h>
|
|
#include <Backups/BackupFactory.h>
|
|
#include <Backups/BackupSettings.h>
|
|
#include <Backups/BackupUtils.h>
|
|
#include <Backups/RestoreSettings.h>
|
|
#include <Backups/RestoreUtils.h>
|
|
#include <Interpreters/Context.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
namespace
|
|
{
|
|
BackupMutablePtr createBackup(const BackupInfo & backup_info, const BackupSettings & backup_settings, const ContextPtr & context)
|
|
{
|
|
BackupFactory::CreateParams params;
|
|
params.open_mode = IBackup::OpenMode::WRITE;
|
|
params.context = context;
|
|
params.backup_info = backup_info;
|
|
params.base_backup_info = backup_settings.base_backup_info;
|
|
params.compression_method = backup_settings.compression_method;
|
|
params.compression_level = backup_settings.compression_level;
|
|
params.password = backup_settings.password;
|
|
return BackupFactory::instance().createBackup(params);
|
|
}
|
|
|
|
BackupMutablePtr openBackup(const BackupInfo & backup_info, const RestoreSettings & restore_settings, const ContextPtr & context)
|
|
{
|
|
BackupFactory::CreateParams params;
|
|
params.open_mode = IBackup::OpenMode::READ;
|
|
params.context = context;
|
|
params.backup_info = backup_info;
|
|
params.base_backup_info = restore_settings.base_backup_info;
|
|
params.password = restore_settings.password;
|
|
return BackupFactory::instance().createBackup(params);
|
|
}
|
|
|
|
void executeBackup(const ContextPtr & context, const ASTBackupQuery & query)
|
|
{
|
|
auto backup_settings = BackupSettings::fromBackupQuery(query);
|
|
auto backup_entries = makeBackupEntries(context, query.elements, backup_settings);
|
|
BackupMutablePtr backup = createBackup(BackupInfo::fromAST(*query.backup_name), backup_settings, context);
|
|
writeBackupEntries(backup, std::move(backup_entries), context->getSettingsRef().max_backup_threads);
|
|
}
|
|
|
|
void executeRestore(ContextMutablePtr context, const ASTBackupQuery & query)
|
|
{
|
|
auto restore_settings = RestoreSettings::fromRestoreQuery(query);
|
|
BackupPtr backup = openBackup(BackupInfo::fromAST(*query.backup_name), restore_settings, context);
|
|
auto restore_tasks = makeRestoreTasks(context, backup, query.elements, restore_settings);
|
|
executeRestoreTasks(std::move(restore_tasks), context->getSettingsRef().max_backup_threads);
|
|
}
|
|
}
|
|
|
|
BlockIO InterpreterBackupQuery::execute()
|
|
{
|
|
const auto & query = query_ptr->as<const ASTBackupQuery &>();
|
|
if (query.kind == ASTBackupQuery::BACKUP)
|
|
executeBackup(context, query);
|
|
else if (query.kind == ASTBackupQuery::RESTORE)
|
|
executeRestore(context, query);
|
|
return {};
|
|
}
|
|
|
|
}
|