mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-30 05:30:51 +00:00
39 lines
1016 B
C++
39 lines
1016 B
C++
#include <DB/Parsers/IAST.h>
|
|
#include <DB/Parsers/ASTIdentifier.h>
|
|
#include <DB/Parsers/ASTLiteral.h>
|
|
#include <DB/Parsers/ASTFunction.h>
|
|
#include <DB/Common/typeid_cast.h>
|
|
|
|
#include <DB/Interpreters/getClusterName.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int BAD_ARGUMENTS;
|
|
}
|
|
|
|
|
|
std::string getClusterName(const IAST & node)
|
|
{
|
|
if (const ASTIdentifier * ast_id = typeid_cast<const ASTIdentifier *>(&node))
|
|
return ast_id->name;
|
|
|
|
if (const ASTLiteral * ast_lit = typeid_cast<const ASTLiteral *>(&node))
|
|
return ast_lit->value.safeGet<String>();
|
|
|
|
if (const ASTFunction * ast_func = typeid_cast<const ASTFunction *>(&node))
|
|
{
|
|
if (!ast_func->range.first || !ast_func->range.second)
|
|
throw Exception("Illegal expression instead of cluster name.", ErrorCodes::BAD_ARGUMENTS);
|
|
|
|
return String(ast_func->range.first, ast_func->range.second);
|
|
}
|
|
|
|
throw Exception("Illegal expression instead of cluster name.", ErrorCodes::BAD_ARGUMENTS);
|
|
}
|
|
|
|
}
|