ClickHouse/src/Parsers/ASTAssignment.h

43 lines
1.0 KiB
C++

#pragma once
#include <Parsers/IAST.h>
namespace DB
{
/// Part of the ALTER UPDATE statement of the form: column = expr
class ASTAssignment : public IAST
{
public:
String column_name;
ASTPtr expression() const
{
return children.at(0);
}
String getID(char delim) const override { return "Assignment" + (delim + column_name); }
ASTPtr clone() const override
{
auto res = std::make_shared<ASTAssignment>(*this);
res->children = { expression()->clone() };
return res;
}
protected:
void formatImpl(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const override
{
settings.ostr << (settings.hilite ? hilite_identifier : "");
settings.writeIdentifier(column_name);
settings.ostr << (settings.hilite ? hilite_none : "");
settings.ostr << (settings.hilite ? hilite_operator : "") << " = " << (settings.hilite ? hilite_none : "");
expression()->formatImpl(settings, state, frame);
}
};
}