mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 11:22:12 +00:00
4e76629aaf
- lots of static_cast - add safe_cast - types adjustments - config - IStorage::read/watch - ... - some TODO's (to convert types in future) P.S. That was quite a journey... v2: fixes after rebase v3: fix conflicts after #42308 merged Signed-off-by: Azat Khuzhin <a.khuzhin@semrush.com>
120 lines
4.5 KiB
C++
120 lines
4.5 KiB
C++
#include <Storages/System/StorageSystemQuotaLimits.h>
|
|
#include <DataTypes/DataTypeString.h>
|
|
#include <DataTypes/DataTypesNumber.h>
|
|
#include <DataTypes/DataTypeNullable.h>
|
|
#include <Columns/ColumnString.h>
|
|
#include <Columns/ColumnsNumber.h>
|
|
#include <Columns/ColumnNullable.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Access/AccessControl.h>
|
|
#include <Access/Quota.h>
|
|
#include <Access/Common/AccessFlags.h>
|
|
#include <base/range.h>
|
|
#include <boost/range/algorithm_ext/push_back.hpp>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace
|
|
{
|
|
void addValue(IColumn & out_column, NullMap & out_column_null_map, QuotaValue value, const QuotaTypeInfo & type_info)
|
|
{
|
|
out_column_null_map.push_back(false);
|
|
if (type_info.output_as_float)
|
|
static_cast<ColumnFloat64 &>(out_column).getData().push_back(double(value) / type_info.output_denominator);
|
|
else
|
|
static_cast<ColumnUInt64 &>(out_column).getData().push_back(value / type_info.output_denominator);
|
|
}
|
|
|
|
void addValue(IColumn & out_column, NullMap & out_column_null_map, std::optional<QuotaValue> value, const QuotaTypeInfo & type_info)
|
|
{
|
|
if (value)
|
|
addValue(out_column, out_column_null_map, *value, type_info);
|
|
else
|
|
{
|
|
out_column_null_map.push_back(true);
|
|
out_column.insertDefault();
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
NamesAndTypesList StorageSystemQuotaLimits::getNamesAndTypes()
|
|
{
|
|
NamesAndTypesList names_and_types{
|
|
{"quota_name", std::make_shared<DataTypeString>()},
|
|
{"duration", std::make_shared<DataTypeUInt32>()},
|
|
{"is_randomized_interval", std::make_shared<DataTypeUInt8>()},
|
|
};
|
|
|
|
for (auto quota_type : collections::range(QuotaType::MAX))
|
|
{
|
|
const auto & type_info = QuotaTypeInfo::get(quota_type);
|
|
String column_name = "max_" + type_info.name;
|
|
DataTypePtr data_type;
|
|
if (type_info.output_as_float)
|
|
data_type = std::make_shared<DataTypeFloat64>();
|
|
else
|
|
data_type = std::make_shared<DataTypeUInt64>();
|
|
names_and_types.push_back({column_name, std::make_shared<DataTypeNullable>(data_type)});
|
|
}
|
|
|
|
return names_and_types;
|
|
}
|
|
|
|
|
|
void StorageSystemQuotaLimits::fillData(MutableColumns & res_columns, ContextPtr context, const SelectQueryInfo &) const
|
|
{
|
|
/// If "select_from_system_db_requires_grant" is enabled the access rights were already checked in InterpreterSelectQuery.
|
|
const auto & access_control = context->getAccessControl();
|
|
if (!access_control.doesSelectFromSystemDatabaseRequireGrant())
|
|
context->checkAccess(AccessType::SHOW_QUOTAS);
|
|
|
|
std::vector<UUID> ids = access_control.findAll<Quota>();
|
|
|
|
size_t column_index = 0;
|
|
auto & column_quota_name = assert_cast<ColumnString &>(*res_columns[column_index++]);
|
|
auto & column_duration = assert_cast<ColumnUInt32 &>(*res_columns[column_index++]).getData();
|
|
auto & column_is_randomized_interval = assert_cast<ColumnUInt8 &>(*res_columns[column_index++]).getData();
|
|
|
|
IColumn * column_max[static_cast<size_t>(QuotaType::MAX)];
|
|
NullMap * column_max_null_map[static_cast<size_t>(QuotaType::MAX)];
|
|
for (auto quota_type : collections::range(QuotaType::MAX))
|
|
{
|
|
auto quota_type_i = static_cast<size_t>(quota_type);
|
|
column_max[quota_type_i] = &assert_cast<ColumnNullable &>(*res_columns[column_index]).getNestedColumn();
|
|
column_max_null_map[quota_type_i] = &assert_cast<ColumnNullable &>(*res_columns[column_index++]).getNullMapData();
|
|
}
|
|
|
|
auto add_row = [&](const String & quota_name, const Quota::Limits & limits)
|
|
{
|
|
column_quota_name.insertData(quota_name.data(), quota_name.length());
|
|
column_duration.push_back(static_cast<UInt32>(limits.duration.count()));
|
|
column_is_randomized_interval.push_back(limits.randomize_interval);
|
|
|
|
for (auto quota_type : collections::range(QuotaType::MAX))
|
|
{
|
|
auto quota_type_i = static_cast<size_t>(quota_type);
|
|
const auto & type_info = QuotaTypeInfo::get(quota_type);
|
|
addValue(*column_max[quota_type_i], *column_max_null_map[quota_type_i], limits.max[quota_type_i], type_info);
|
|
}
|
|
};
|
|
|
|
auto add_rows = [&](const String & quota_name, const std::vector<Quota::Limits> & all_limits)
|
|
{
|
|
for (const auto & limits : all_limits)
|
|
add_row(quota_name, limits);
|
|
};
|
|
|
|
for (const auto & id : ids)
|
|
{
|
|
auto quota = access_control.tryRead<Quota>(id);
|
|
if (!quota)
|
|
continue;
|
|
|
|
add_rows(quota->getName(), quota->all_limits);
|
|
}
|
|
}
|
|
}
|