mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-24 00:22:29 +00:00
88 lines
2.6 KiB
C++
88 lines
2.6 KiB
C++
#pragma once
|
||
|
||
#include <list>
|
||
#include <sstream>
|
||
|
||
#include <Poco/SharedPtr.h>
|
||
|
||
#include <DB/Core/Types.h>
|
||
#include <DB/Core/Exception.h>
|
||
#include <DB/Core/ErrorCodes.h>
|
||
#include <DB/Parsers/StringRange.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
using Poco::SharedPtr;
|
||
|
||
|
||
/** Элемент синтаксического дерева (в дальнейшем - направленного ациклического графа с элементами семантики)
|
||
*/
|
||
class IAST
|
||
{
|
||
public:
|
||
typedef std::vector<SharedPtr<IAST> > ASTs;
|
||
ASTs children;
|
||
StringRange range;
|
||
|
||
/** Идентификатор части выражения. Используется при интерпретации, чтобы вычислять не всё выражение сразу,
|
||
* а по частям (например, сначала WHERE, потом фильтрация, потом всё остальное).
|
||
*/
|
||
unsigned part_id;
|
||
|
||
|
||
IAST() : range(NULL, NULL), part_id(0) {}
|
||
IAST(StringRange range_) : range(range_), part_id(0) {}
|
||
virtual ~IAST() {}
|
||
|
||
/** Получить каноническое имя столбца, если элемент является столбцом */
|
||
virtual String getColumnName() { throw Exception("Trying to get name of not a column", ErrorCodes::NOT_A_COLUMN); }
|
||
|
||
/** Получить алиас, если он есть, или каноническое имя столбца; если элемент является столбцом */
|
||
virtual String getAlias() { return getColumnName(); }
|
||
|
||
/** Получить текст, который идентифицирует этот элемент. */
|
||
virtual String getID() = 0;
|
||
|
||
/** Получить глубокую копию дерева. */
|
||
virtual SharedPtr<IAST> clone() const = 0;
|
||
|
||
/** Получить текст, который идентифицирует этот элемент и всё поддерево.
|
||
* Обычно он содержит идентификатор элемента и getTreeID от всех детей.
|
||
*/
|
||
String getTreeID()
|
||
{
|
||
std::stringstream s;
|
||
s << getID();
|
||
|
||
if (!children.empty())
|
||
{
|
||
s << "(";
|
||
for (ASTs::iterator it = children.begin(); it != children.end(); ++it)
|
||
{
|
||
if (it != children.begin())
|
||
s << ", ";
|
||
s << (*it)->getTreeID();
|
||
}
|
||
s << ")";
|
||
}
|
||
|
||
return s.str();
|
||
}
|
||
|
||
void dumpTree(std::ostream & ostr, size_t indent = 0)
|
||
{
|
||
String indent_str(indent, '-');
|
||
ostr << indent_str << getID() << ", " << this << ", part_id = " << part_id << std::endl;
|
||
for (ASTs::iterator it = children.begin(); it != children.end(); ++it)
|
||
(*it)->dumpTree(ostr, indent + 1);
|
||
}
|
||
};
|
||
|
||
|
||
typedef SharedPtr<IAST> ASTPtr;
|
||
typedef std::vector<ASTPtr> ASTs;
|
||
|
||
}
|