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

70 lines
2.2 KiB
C++
Raw Normal View History

#include <DataStreams/OneBlockInputStream.h>
#include <Storages/System/StorageSystemDisks.h>
namespace DB
{
namespace ErrorCodes
{
}
StorageSystemDisks::StorageSystemDisks(const std::string & name_)
2019-08-14 15:20:52 +00:00
: name(name_)
{
setColumns(ColumnsDescription(
{
{"name", std::make_shared<DataTypeString>()},
{"path", std::make_shared<DataTypeString>()},
{"free_space", std::make_shared<DataTypeUInt64>()},
{"total_space", std::make_shared<DataTypeUInt64>()},
{"keep_free_space", std::make_shared<DataTypeUInt64>()},
}));
}
BlockInputStreams StorageSystemDisks::read(
2019-08-14 15:20:52 +00:00
const Names & column_names,
const SelectQueryInfo & /*query_info*/,
const Context & context,
QueryProcessingStage::Enum /*processed_stage*/,
const size_t /*max_block_size*/,
const unsigned /*num_streams*/)
{
check(column_names);
MutableColumnPtr col_name_mut = ColumnString::create();
MutableColumnPtr col_path_mut = ColumnString::create();
MutableColumnPtr col_free_mut = ColumnUInt64::create();
MutableColumnPtr col_total_mut = ColumnUInt64::create();
MutableColumnPtr col_keep_mut = ColumnUInt64::create();
const auto & disk_selector = context.getDiskSelector();
2019-08-14 15:20:52 +00:00
for (const auto & [disk_name, disk_ptr] : disk_selector.getDisksMap())
{
2019-08-14 15:20:52 +00:00
col_name_mut->insert(disk_name);
col_path_mut->insert(disk_ptr->getPath());
col_free_mut->insert(disk_ptr->getAvailableSpace());
col_total_mut->insert(disk_ptr->getTotalSpace());
col_keep_mut->insert(disk_ptr->getKeepingFreeSpace());
}
ColumnPtr col_name = std::move(col_name_mut);
ColumnPtr col_path = std::move(col_path_mut);
ColumnPtr col_free = std::move(col_free_mut);
ColumnPtr col_total = std::move(col_total_mut);
ColumnPtr col_keep = std::move(col_keep_mut);
Block res = getSampleBlock().cloneEmpty();
size_t col_num = 0;
res.getByPosition(col_num++).column = col_name;
res.getByPosition(col_num++).column = col_path;
res.getByPosition(col_num++).column = col_free;
res.getByPosition(col_num++).column = col_total;
res.getByPosition(col_num++).column = col_keep;
return BlockInputStreams(1, std::make_shared<OneBlockInputStream>(res));
}
}