ClickHouse/dbms/src/Storages/System/StorageSystemMerges.cpp

80 lines
3.5 KiB
C++

#include <Storages/System/StorageSystemMerges.h>
#include <DataTypes/DataTypeString.h>
#include <DataTypes/DataTypesNumber.h>
#include <DataTypes/DataTypeArray.h>
#include <Columns/ColumnString.h>
#include <DataStreams/OneBlockInputStream.h>
#include <Interpreters/Context.h>
#include <Storages/MergeTree/MergeList.h>
namespace DB
{
StorageSystemMerges::StorageSystemMerges(const std::string & name)
: name{name}
, columns{
{ "database", std::make_shared<DataTypeString>() },
{ "table", std::make_shared<DataTypeString>() },
{ "elapsed", std::make_shared<DataTypeFloat64>() },
{ "progress", std::make_shared<DataTypeFloat64>() },
{ "num_parts", std::make_shared<DataTypeUInt64>() },
{ "source_part_names", std::make_shared<DataTypeArray>(std::make_shared<DataTypeString>()) },
{ "result_part_name", std::make_shared<DataTypeString>() },
{ "total_size_bytes_compressed", std::make_shared<DataTypeUInt64>() },
{ "total_size_marks", std::make_shared<DataTypeUInt64>() },
{ "bytes_read_uncompressed", std::make_shared<DataTypeUInt64>() },
{ "rows_read", std::make_shared<DataTypeUInt64>() },
{ "bytes_written_uncompressed", std::make_shared<DataTypeUInt64>() },
{ "rows_written", std::make_shared<DataTypeUInt64>() },
{ "columns_written", std::make_shared<DataTypeUInt64>() },
{ "memory_usage", std::make_shared<DataTypeUInt64>() },
{ "thread_number", std::make_shared<DataTypeUInt64>() },
}
{
}
StoragePtr StorageSystemMerges::create(const std::string & name)
{
return make_shared(name);
}
BlockInputStreams StorageSystemMerges::read(
const Names & column_names,
const ASTPtr & query,
const Context & context,
QueryProcessingStage::Enum & processed_stage,
const size_t max_block_size,
const unsigned)
{
check(column_names);
processed_stage = QueryProcessingStage::FetchColumns;
Block block = getSampleBlock();
for (const auto & merge : context.getMergeList().get())
{
size_t i = 0;
block.getByPosition(i++).column->insert(merge.database);
block.getByPosition(i++).column->insert(merge.table);
block.getByPosition(i++).column->insert(merge.elapsed);
block.getByPosition(i++).column->insert(std::min(1., merge.progress)); /// little cheat
block.getByPosition(i++).column->insert(merge.num_parts);
block.getByPosition(i++).column->insert(merge.source_part_names);
block.getByPosition(i++).column->insert(merge.result_part_name);
block.getByPosition(i++).column->insert(merge.total_size_bytes_compressed);
block.getByPosition(i++).column->insert(merge.total_size_marks);
block.getByPosition(i++).column->insert(merge.bytes_read_uncompressed);
block.getByPosition(i++).column->insert(merge.rows_read);
block.getByPosition(i++).column->insert(merge.bytes_written_uncompressed);
block.getByPosition(i++).column->insert(merge.rows_written);
block.getByPosition(i++).column->insert(merge.columns_written);
block.getByPosition(i++).column->insert(merge.memory_usage);
block.getByPosition(i++).column->insert(merge.thread_number);
}
return BlockInputStreams{1, std::make_shared<OneBlockInputStream>(block)};
}
}