#include #include #include #include #include #include #include namespace DB { namespace ErrorCodes { } StorageSystemStoragePolicies::StorageSystemStoragePolicies(const StorageID & table_id_) : IStorage(table_id_) { StorageInMemoryMetadata storage_metadata; storage_metadata.setColumns( ColumnsDescription({ {"policy_name", std::make_shared()}, {"volume_name", std::make_shared()}, {"volume_priority", std::make_shared()}, {"disks", std::make_shared(std::make_shared())}, {"volume_type", std::make_shared()}, {"max_data_part_size", std::make_shared()}, {"move_factor", std::make_shared()}, {"prefer_not_to_merge", std::make_shared()} })); // TODO: Add string column with custom volume-type-specific options setInMemoryMetadata(storage_metadata); } Pipe StorageSystemStoragePolicies::read( const Names & column_names, const StorageMetadataPtr & metadata_snapshot, SelectQueryInfo & /*query_info*/, ContextPtr context, QueryProcessingStage::Enum /*processed_stage*/, const size_t /*max_block_size*/, const unsigned /*num_streams*/) { check(metadata_snapshot, column_names); MutableColumnPtr col_policy_name = ColumnString::create(); MutableColumnPtr col_volume_name = ColumnString::create(); MutableColumnPtr col_priority = ColumnUInt64::create(); MutableColumnPtr col_disks = ColumnArray::create(ColumnString::create()); MutableColumnPtr col_volume_type = ColumnString::create(); MutableColumnPtr col_max_part_size = ColumnUInt64::create(); MutableColumnPtr col_move_factor = ColumnFloat32::create(); MutableColumnPtr col_prefer_not_to_merge = ColumnUInt8::create(); for (const auto & [policy_name, policy_ptr] : context->getPoliciesMap()) { const auto & volumes = policy_ptr->getVolumes(); for (size_t i = 0; i != volumes.size(); ++i) { col_policy_name->insert(policy_name); col_volume_name->insert(volumes[i]->getName()); col_priority->insert(i + 1); Array disks; disks.reserve(volumes[i]->getDisks().size()); for (const auto & disk_ptr : volumes[i]->getDisks()) disks.push_back(disk_ptr->getName()); col_disks->insert(disks); col_volume_type->insert(volumeTypeToString(volumes[i]->getType())); col_max_part_size->insert(volumes[i]->max_data_part_size); col_move_factor->insert(policy_ptr->getMoveFactor()); col_prefer_not_to_merge->insert(volumes[i]->areMergesAvoided() ? 1 : 0); } } Columns res_columns; res_columns.emplace_back(std::move(col_policy_name)); res_columns.emplace_back(std::move(col_volume_name)); res_columns.emplace_back(std::move(col_priority)); res_columns.emplace_back(std::move(col_disks)); res_columns.emplace_back(std::move(col_volume_type)); res_columns.emplace_back(std::move(col_max_part_size)); res_columns.emplace_back(std::move(col_move_factor)); res_columns.emplace_back(std::move(col_prefer_not_to_merge)); UInt64 num_rows = res_columns.at(0)->size(); Chunk chunk(std::move(res_columns), num_rows); return Pipe(std::make_shared(metadata_snapshot->getSampleBlock(), std::move(chunk))); } }