mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-20 08:40:50 +00:00
ee37f551a2
* make interpreter factory an instance * add registerInterpreter * register interpreters individually * try wire everything up * fix style * fix test
61 lines
1.7 KiB
C++
61 lines
1.7 KiB
C++
#include <Interpreters/InterpreterFactory.h>
|
|
#include <Interpreters/InterpreterBackupQuery.h>
|
|
|
|
#include <Backups/BackupsWorker.h>
|
|
#include <Columns/ColumnString.h>
|
|
#include <Columns/ColumnsNumber.h>
|
|
#include <DataTypes/DataTypeEnum.h>
|
|
#include <DataTypes/DataTypeString.h>
|
|
#include <DataTypes/DataTypeUUID.h>
|
|
#include <DataTypes/DataTypesNumber.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Processors/Sources/SourceFromSingleChunk.h>
|
|
#include <Common/logger_useful.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace
|
|
{
|
|
Block getResultRow(const BackupOperationInfo & info)
|
|
{
|
|
auto column_id = ColumnString::create();
|
|
auto column_status = ColumnInt8::create();
|
|
|
|
column_id->insert(info.id);
|
|
column_status->insert(static_cast<Int8>(info.status));
|
|
|
|
Block res_columns;
|
|
res_columns.insert(0, {std::move(column_id), std::make_shared<DataTypeString>(), "id"});
|
|
res_columns.insert(1, {std::move(column_status), std::make_shared<DataTypeEnum8>(getBackupStatusEnumValues()), "status"});
|
|
|
|
return res_columns;
|
|
}
|
|
}
|
|
|
|
BlockIO InterpreterBackupQuery::execute()
|
|
{
|
|
auto & backups_worker = context->getBackupsWorker();
|
|
auto id = backups_worker.start(query_ptr, context);
|
|
|
|
auto info = backups_worker.getInfo(id);
|
|
if (info.exception)
|
|
std::rethrow_exception(info.exception);
|
|
|
|
BlockIO res_io;
|
|
res_io.pipeline = QueryPipeline(std::make_shared<SourceFromSingleChunk>(getResultRow(info)));
|
|
return res_io;
|
|
}
|
|
|
|
void registerInterpreterBackupQuery(InterpreterFactory & factory)
|
|
{
|
|
auto create_fn = [] (const InterpreterFactory::Arguments & args)
|
|
{
|
|
return std::make_unique<InterpreterBackupQuery>(args.query, args.context);
|
|
};
|
|
factory.registerInterpreter("InterpreterBackupQuery", create_fn);
|
|
}
|
|
|
|
}
|