ClickHouse/dbms/include/DB/Parsers/ASTOrderByElement.h

47 lines
1.5 KiB
C++

#pragma once
#include <DB/Parsers/IAST.h>
#include <DB/Common/Collator.h>
namespace DB
{
/** Элемент выражения, после которого стоит ASC или DESC
*/
class ASTOrderByElement : public IAST
{
public:
int direction; /// 1, если ASC, -1, если DESC
/** Collator для locale-specific сортировки строк.
* Если nullptr, то производится сортировка по байтам.
*/
std::shared_ptr<Collator> collator;
ASTOrderByElement() = default;
ASTOrderByElement(const StringRange range_, const int direction_, const std::shared_ptr<Collator> & collator_ = nullptr)
: IAST(range_), direction(direction_), collator(collator_) {}
/** Получить текст, который идентифицирует этот элемент. */
String getID() const override { return "OrderByElement"; }
ASTPtr clone() const override { return std::make_shared<ASTOrderByElement>(*this); }
protected:
void formatImpl(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const override
{
children.front()->formatImpl(settings, state, frame);
settings.ostr << (settings.hilite ? hilite_keyword : "")
<< (direction == -1 ? " DESC" : " ASC")
<< (settings.hilite ? hilite_none : "");
if (collator)
{
settings.ostr << (settings.hilite ? hilite_keyword : "") << " COLLATE " << (settings.hilite ? hilite_none : "")
<< "'" << collator->getLocale() << "'";
}
}
};
}