-
Notifications
You must be signed in to change notification settings - Fork 397
Expand file tree
/
Copy pathClickHouseVisitor.java
More file actions
111 lines (84 loc) · 3.94 KB
/
ClickHouseVisitor.java
File metadata and controls
111 lines (84 loc) · 3.94 KB
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package sqlancer.clickhouse;
import sqlancer.clickhouse.ast.ClickHouseAggregate;
import sqlancer.clickhouse.ast.ClickHouseAliasOperation;
import sqlancer.clickhouse.ast.ClickHouseBinaryComparisonOperation;
import sqlancer.clickhouse.ast.ClickHouseBinaryFunctionOperation;
import sqlancer.clickhouse.ast.ClickHouseBinaryLogicalOperation;
import sqlancer.clickhouse.ast.ClickHouseCastOperation;
import sqlancer.clickhouse.ast.ClickHouseColumnReference;
import sqlancer.clickhouse.ast.ClickHouseConstant;
import sqlancer.clickhouse.ast.ClickHouseExpression;
import sqlancer.clickhouse.ast.ClickHouseSelect;
import sqlancer.clickhouse.ast.ClickHouseTableReference;
import sqlancer.clickhouse.ast.ClickHouseUnaryPostfixOperation;
import sqlancer.clickhouse.ast.ClickHouseUnaryPrefixOperation;
public interface ClickHouseVisitor {
// TODO remove these default methods
default void visit(ClickHouseBinaryComparisonOperation op) {
}
default void visit(ClickHouseBinaryLogicalOperation op) {
}
default void visit(ClickHouseUnaryPrefixOperation exp) {
}
default void visit(ClickHouseUnaryPostfixOperation op) {
}
default void visit(ClickHouseConstant c) {
}
default void visit(ClickHouseSelect s, boolean inner) {
};
default void visit(ClickHouseColumnReference columnReference) {
};
default void visit(ClickHouseExpression.ClickHousePostfixText op) {
}
void visit(ClickHouseTableReference tableReference);
void visit(ClickHouseCastOperation cast);
void visit(ClickHouseAliasOperation alias);
void visit(ClickHouseExpression.ClickHouseJoin join);
void visit(ClickHouseAggregate aggregate);
void visit(ClickHouseBinaryFunctionOperation func);
default void visit(ClickHouseExpression expr) {
if (expr instanceof ClickHouseBinaryFunctionOperation) {
visit((ClickHouseBinaryFunctionOperation) expr);
} else if (expr instanceof ClickHouseBinaryComparisonOperation) {
visit((ClickHouseBinaryComparisonOperation) expr);
} else if (expr instanceof ClickHouseBinaryLogicalOperation) {
visit((ClickHouseBinaryLogicalOperation) expr);
} else if (expr instanceof ClickHouseConstant) {
visit((ClickHouseConstant) expr);
} else if (expr instanceof ClickHouseUnaryPrefixOperation) {
visit((ClickHouseUnaryPrefixOperation) expr);
} else if (expr instanceof ClickHouseSelect) {
visit((ClickHouseSelect) expr, true);
} else if (expr instanceof ClickHouseColumnReference) {
visit((ClickHouseColumnReference) expr);
} else if (expr instanceof ClickHouseTableReference) {
visit((ClickHouseTableReference) expr);
} else if (expr instanceof ClickHouseCastOperation) {
visit((ClickHouseCastOperation) expr);
} else if (expr instanceof ClickHouseExpression.ClickHouseJoin) {
visit((ClickHouseExpression.ClickHouseJoin) expr);
} else if (expr instanceof ClickHouseExpression.ClickHousePostfixText) {
visit((ClickHouseExpression.ClickHousePostfixText) expr);
} else if (expr instanceof ClickHouseAggregate) {
visit((ClickHouseAggregate) expr);
} else if (expr instanceof ClickHouseAliasOperation) {
visit((ClickHouseAliasOperation) expr);
} else if (expr instanceof ClickHouseExpression.ClickHouseJoinOnClause) {
visit((ClickHouseExpression.ClickHouseJoinOnClause) expr);
} else {
throw new AssertionError(expr);
}
}
static String asString(ClickHouseExpression expr) {
if (expr == null) {
throw new AssertionError();
}
ClickHouseToStringVisitor visitor = new ClickHouseToStringVisitor();
if (expr instanceof ClickHouseSelect) {
visitor.visit((ClickHouseSelect) expr, false);
} else {
visitor.visit(expr);
}
return visitor.get();
}
}