ClickHouse/dbms/src/Interpreters/getClusterName.cpp

#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);
}

}