mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 09:22:05 +00:00
94 lines
3.2 KiB
C++
94 lines
3.2 KiB
C++
#include <base/defines.h>
|
|
#include <Parsers/ASTExpressionList.h>
|
|
#include <Parsers/ASTFunction.h>
|
|
#include <Parsers/ASTIdentifier.h>
|
|
#include <Parsers/ASTStatisticDeclaration.h>
|
|
#include <Parsers/formatAST.h>
|
|
#include <Parsers/parseQuery.h>
|
|
#include <Parsers/ParserCreateQuery.h>
|
|
#include <Poco/Logger.h>
|
|
#include <Storages/extractKeyExpressionList.h>
|
|
#include <Storages/StatisticsDescription.h>
|
|
#include <Storages/ColumnsDescription.h>
|
|
|
|
#include <Common/logger_useful.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int INCORRECT_QUERY;
|
|
extern const int LOGICAL_ERROR;
|
|
};
|
|
|
|
StatisticType stringToType(String type)
|
|
{
|
|
if (type == "tdigest")
|
|
return TDigest;
|
|
throw Exception(ErrorCodes::INCORRECT_QUERY, "Unknown statistic type: {}. We only support statistic type `tdigest` right now.", type);
|
|
}
|
|
|
|
String StatisticDescription::getTypeName() const
|
|
{
|
|
if (type == TDigest)
|
|
return "tdigest";
|
|
throw Exception(ErrorCodes::INCORRECT_QUERY, "Unknown statistic type: {}. We only support statistic type `tdigest` right now.", type);
|
|
}
|
|
|
|
std::vector<StatisticDescription> StatisticDescription::getStatisticsFromAST(const ASTPtr & definition_ast, const ColumnsDescription & columns)
|
|
{
|
|
const auto * stat_definition = definition_ast->as<ASTStatisticDeclaration>();
|
|
if (!stat_definition)
|
|
throw Exception(ErrorCodes::LOGICAL_ERROR, "Cannot create statistic from non ASTStatisticDeclaration AST");
|
|
|
|
std::vector<StatisticDescription> stats;
|
|
stats.reserve(stat_definition->columns->children.size());
|
|
for (const auto & column_ast : stat_definition->columns->children)
|
|
{
|
|
StatisticDescription stat;
|
|
stat.type = stringToType(Poco::toLower(stat_definition->type));
|
|
String column_name = column_ast->as<ASTIdentifier &>().name();
|
|
|
|
if (!columns.hasPhysical(column_name))
|
|
throw Exception(ErrorCodes::INCORRECT_QUERY, "Incorrect column name {}", column_name);
|
|
|
|
const auto & column = columns.getPhysical(column_name);
|
|
stat.column_name = column.name;
|
|
|
|
auto function_node = std::make_shared<ASTFunction>();
|
|
function_node->name = "STATISTIC";
|
|
function_node->arguments = std::make_shared<ASTExpressionList>();
|
|
function_node->arguments->children.push_back(std::make_shared<ASTIdentifier>(stat_definition->type));
|
|
function_node->children.push_back(function_node->arguments);
|
|
|
|
stat.ast = function_node;
|
|
|
|
stats.push_back(stat);
|
|
}
|
|
|
|
if (stats.empty())
|
|
throw Exception(ErrorCodes::INCORRECT_QUERY, "Empty statistic column list");
|
|
|
|
return stats;
|
|
}
|
|
|
|
String queryToString(const IAST & query);
|
|
|
|
StatisticDescription StatisticDescription::getStatisticFromColumnDeclaration(const ASTColumnDeclaration & column)
|
|
{
|
|
const auto & stat_type_list_ast = column.stat_type->as<ASTFunction &>().arguments;
|
|
if (stat_type_list_ast->children.size() != 1)
|
|
throw Exception(ErrorCodes::INCORRECT_QUERY, "We expect only one statistic type for column {}", queryToString(column));
|
|
const auto & stat_type = stat_type_list_ast->children[0]->as<ASTFunction &>().name;
|
|
|
|
StatisticDescription stat;
|
|
stat.type = stringToType(Poco::toLower(stat_type));
|
|
stat.column_name = column.name;
|
|
stat.ast = column.stat_type;
|
|
|
|
return stat;
|
|
}
|
|
|
|
}
|