mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 19:02:04 +00:00
68 lines
2.1 KiB
C++
68 lines
2.1 KiB
C++
#include "StorageSystemNamedCollections.h"
|
|
|
|
#include <Common/FieldVisitorToString.h>
|
|
#include <DataTypes/DataTypeString.h>
|
|
#include <DataTypes/DataTypeMap.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Interpreters/ProfileEventsExt.h>
|
|
#include <Access/Common/AccessType.h>
|
|
#include <Access/Common/AccessFlags.h>
|
|
#include <Access/ContextAccess.h>
|
|
#include <Columns/ColumnMap.h>
|
|
#include <Common/NamedCollections/NamedCollections.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
NamesAndTypesList StorageSystemNamedCollections::getNamesAndTypes()
|
|
{
|
|
return {
|
|
{"name", std::make_shared<DataTypeString>()},
|
|
{"collection", std::make_shared<DataTypeMap>(std::make_shared<DataTypeString>(), std::make_shared<DataTypeString>())},
|
|
};
|
|
}
|
|
|
|
StorageSystemNamedCollections::StorageSystemNamedCollections(const StorageID & table_id_)
|
|
: IStorageSystemOneBlock(table_id_)
|
|
{
|
|
}
|
|
|
|
void StorageSystemNamedCollections::fillData(MutableColumns & res_columns, ContextPtr context, const SelectQueryInfo &) const
|
|
{
|
|
const auto & access = context->getAccess();
|
|
|
|
NamedCollectionUtils::loadIfNot();
|
|
|
|
auto collections = NamedCollectionFactory::instance().getAll();
|
|
for (const auto & [name, collection] : collections)
|
|
{
|
|
if (!access->isGranted(AccessType::SHOW_NAMED_COLLECTIONS, name))
|
|
continue;
|
|
|
|
res_columns[0]->insert(name);
|
|
|
|
auto * column_map = typeid_cast<ColumnMap *>(res_columns[1].get());
|
|
|
|
auto & offsets = column_map->getNestedColumn().getOffsets();
|
|
auto & tuple_column = column_map->getNestedData();
|
|
auto & key_column = tuple_column.getColumn(0);
|
|
auto & value_column = tuple_column.getColumn(1);
|
|
|
|
size_t size = 0;
|
|
for (const auto & key : collection->getKeys())
|
|
{
|
|
key_column.insertData(key.data(), key.size());
|
|
if (access->isGranted(AccessType::SHOW_NAMED_COLLECTIONS_SECRETS))
|
|
value_column.insert(collection->get<String>(key));
|
|
else
|
|
value_column.insert("[HIDDEN]");
|
|
size++;
|
|
}
|
|
|
|
offsets.push_back(offsets.back() + size);
|
|
}
|
|
}
|
|
|
|
}
|