blob: d93c10b6e391aeb746faf0025c0e7fd70136ad6c (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
#pragma once
#include <Core/Names.h>
#include <Parsers/IAST.h>
namespace DB
{
/** PROJECTION SELECT query
*/
class ASTProjectionSelectQuery : public IAST
{
public:
enum class Expression : uint8_t
{
WITH,
SELECT,
GROUP_BY,
ORDER_BY,
};
/** Get the text that identifies this element. */
String getID(char) const override { return "ProjectionSelectQuery"; }
ASTPtr clone() const override;
ASTPtr & refSelect() { return getExpression(Expression::SELECT); }
ASTPtr with() const { return getExpression(Expression::WITH); }
ASTPtr select() const { return getExpression(Expression::SELECT); }
ASTPtr groupBy() const { return getExpression(Expression::GROUP_BY); }
ASTPtr orderBy() const { return getExpression(Expression::ORDER_BY); }
/// Set/Reset/Remove expression.
void setExpression(Expression expr, ASTPtr && ast);
ASTPtr getExpression(Expression expr, bool clone = false) const
{
auto it = positions.find(expr);
if (it != positions.end())
return clone ? children[it->second]->clone() : children[it->second];
return {};
}
ASTPtr cloneToASTSelect() const;
protected:
void formatImpl(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const override;
private:
std::unordered_map<Expression, size_t> positions;
ASTPtr & getExpression(Expression expr);
};
}
|
