mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-07 00:50:56 +00:00
88 lines
3.3 KiB
C++
88 lines
3.3 KiB
C++
|
#include <DB/Storages/StorageSystemMerges.h>
|
||
|
|
#include <DB/DataTypes/DataTypeString.h>
|
||
|
|
#include <DB/DataTypes/DataTypesNumberFixed.h>
|
||
|
|
#include <DB/Columns/ColumnString.h>
|
||
|
|
#include <DB/DataStreams/OneBlockInputStream.h>
|
||
|
|
#include <DB/Interpreters/Context.h>
|
||
|
|
|
||
|
|
namespace DB
|
||
|
|
{
|
||
|
|
|
||
|
|
StorageSystemMerges::StorageSystemMerges(const std::string & name, const Context & context)
|
||
|
|
: name{name}, context(context)
|
||
|
|
, columns{
|
||
|
|
{ "database", new DataTypeString },
|
||
|
|
{ "table", new DataTypeString },
|
||
|
|
{ "num_parts", new DataTypeUInt64 },
|
||
|
|
{ "result_part_name", new DataTypeString },
|
||
|
|
{ "total_size_bytes", new DataTypeUInt64 },
|
||
|
|
{ "total_size_marks", new DataTypeUInt64 },
|
||
|
|
{ "bytes_read", new DataTypeUInt64 },
|
||
|
|
{ "rows_read", new DataTypeUInt64 },
|
||
|
|
{ "bytes_written", new DataTypeUInt64 },
|
||
|
|
{ "rows_written", new DataTypeUInt64 }
|
||
|
|
}
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
StoragePtr StorageSystemMerges::create(const std::string & name, const Context & context)
|
||
|
|
{
|
||
|
|
return (new StorageSystemMerges{name, context})->thisPtr();
|
||
|
|
}
|
||
|
|
|
||
|
|
BlockInputStreams StorageSystemMerges::read(
|
||
|
|
const Names & column_names, ASTPtr query, const Settings & settings,
|
||
|
|
QueryProcessingStage::Enum & processed_stage, const size_t max_block_size, const unsigned)
|
||
|
|
{
|
||
|
|
check(column_names);
|
||
|
|
processed_stage = QueryProcessingStage::FetchColumns;
|
||
|
|
|
||
|
|
Block block;
|
||
|
|
|
||
|
|
ColumnWithNameAndType col_database{new ColumnString, new DataTypeString, "database"};
|
||
|
|
ColumnWithNameAndType col_table{new ColumnString, new DataTypeString, "table"};
|
||
|
|
ColumnWithNameAndType col_num_parts{new ColumnUInt64, new DataTypeUInt64, "num_parts"};
|
||
|
|
ColumnWithNameAndType col_result_part_name{new ColumnString, new DataTypeString, "result_part_name"};
|
||
|
|
ColumnWithNameAndType col_total_size_bytes{new ColumnUInt64, new DataTypeUInt64, "total_size_bytes"};
|
||
|
|
ColumnWithNameAndType col_total_size_marks{new ColumnUInt64, new DataTypeUInt64, "total_size_marks"};
|
||
|
|
ColumnWithNameAndType col_bytes_read{new ColumnUInt64, new DataTypeUInt64, "bytes_read"};
|
||
|
|
ColumnWithNameAndType col_rows_read{new ColumnUInt64, new DataTypeUInt64, "rows_read"};
|
||
|
|
ColumnWithNameAndType col_bytes_written{new ColumnUInt64, new DataTypeUInt64, "bytes_written"};
|
||
|
|
ColumnWithNameAndType col_rows_written{new ColumnUInt64, new DataTypeUInt64, "rows_written"};
|
||
|
|
|
||
|
|
for (const auto & merge : context.getMergeList().get())
|
||
|
|
{
|
||
|
|
const auto bytes_read = merge.bytes_read;
|
||
|
|
const auto rows_read = merge.rows_read;
|
||
|
|
const auto bytes_written = merge.bytes_written;
|
||
|
|
const auto rows_written = merge.rows_written;
|
||
|
|
|
||
|
|
col_database.column->insert(merge.database);
|
||
|
|
col_table.column->insert(merge.table);
|
||
|
|
col_num_parts.column->insert(merge.num_parts);
|
||
|
|
col_result_part_name.column->insert(merge.result_part_name);
|
||
|
|
col_total_size_bytes.column->insert(merge.total_size_bytes);
|
||
|
|
col_total_size_marks.column->insert(merge.total_size_marks);
|
||
|
|
col_bytes_read.column->insert(bytes_read);
|
||
|
|
col_rows_read.column->insert(rows_read);
|
||
|
|
col_bytes_written.column->insert(bytes_written);
|
||
|
|
col_rows_written.column->insert(rows_written);
|
||
|
|
}
|
||
|
|
|
||
|
|
block.insert(col_database);
|
||
|
|
block.insert(col_num_parts);
|
||
|
|
block.insert(col_table);
|
||
|
|
block.insert(col_total_size_bytes);
|
||
|
|
block.insert(col_result_part_name);
|
||
|
|
block.insert(col_bytes_read);
|
||
|
|
block.insert(col_total_size_marks);
|
||
|
|
block.insert(col_bytes_written);
|
||
|
|
block.insert(col_rows_read);
|
||
|
|
block.insert(col_rows_written);
|
||
|
|
|
||
|
|
return BlockInputStreams{1, new OneBlockInputStream{block}};
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
}
|