-
Notifications
You must be signed in to change notification settings - Fork 397
Expand file tree
/
Copy pathMaterializeVisitor.java
More file actions
127 lines (101 loc) · 5.2 KB
/
MaterializeVisitor.java
File metadata and controls
127 lines (101 loc) · 5.2 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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
package sqlancer.materialize;
import java.util.List;
import sqlancer.materialize.MaterializeSchema.MaterializeColumn;
import sqlancer.materialize.MaterializeSchema.MaterializeDataType;
import sqlancer.materialize.ast.MaterializeAggregate;
import sqlancer.materialize.ast.MaterializeBetweenOperation;
import sqlancer.materialize.ast.MaterializeBinaryLogicalOperation;
import sqlancer.materialize.ast.MaterializeCastOperation;
import sqlancer.materialize.ast.MaterializeColumnValue;
import sqlancer.materialize.ast.MaterializeConstant;
import sqlancer.materialize.ast.MaterializeExpression;
import sqlancer.materialize.ast.MaterializeFunction;
import sqlancer.materialize.ast.MaterializeInOperation;
import sqlancer.materialize.ast.MaterializeLikeOperation;
import sqlancer.materialize.ast.MaterializeOrderByTerm;
import sqlancer.materialize.ast.MaterializePOSIXRegularExpression;
import sqlancer.materialize.ast.MaterializePostfixOperation;
import sqlancer.materialize.ast.MaterializePostfixText;
import sqlancer.materialize.ast.MaterializePrefixOperation;
import sqlancer.materialize.ast.MaterializeSelect;
import sqlancer.materialize.ast.MaterializeSelect.MaterializeFromTable;
import sqlancer.materialize.ast.MaterializeSelect.MaterializeSubquery;
import sqlancer.materialize.ast.MaterializeSimilarTo;
import sqlancer.materialize.gen.MaterializeExpressionGenerator;
public interface MaterializeVisitor {
void visit(MaterializeConstant constant);
void visit(MaterializePostfixOperation op);
void visit(MaterializeColumnValue c);
void visit(MaterializePrefixOperation op);
void visit(MaterializeSelect op);
void visit(MaterializeOrderByTerm op);
void visit(MaterializeFunction f);
void visit(MaterializeCastOperation cast);
void visit(MaterializeBetweenOperation op);
void visit(MaterializeInOperation op);
void visit(MaterializePostfixText op);
void visit(MaterializeAggregate op);
void visit(MaterializeSimilarTo op);
void visit(MaterializePOSIXRegularExpression op);
void visit(MaterializeFromTable from);
void visit(MaterializeSubquery subquery);
void visit(MaterializeBinaryLogicalOperation op);
void visit(MaterializeLikeOperation op);
default void visit(MaterializeExpression expression) {
if (expression instanceof MaterializeConstant) {
visit((MaterializeConstant) expression);
} else if (expression instanceof MaterializePostfixOperation) {
visit((MaterializePostfixOperation) expression);
} else if (expression instanceof MaterializeColumnValue) {
visit((MaterializeColumnValue) expression);
} else if (expression instanceof MaterializePrefixOperation) {
visit((MaterializePrefixOperation) expression);
} else if (expression instanceof MaterializeSelect) {
visit((MaterializeSelect) expression);
} else if (expression instanceof MaterializeOrderByTerm) {
visit((MaterializeOrderByTerm) expression);
} else if (expression instanceof MaterializeFunction) {
visit((MaterializeFunction) expression);
} else if (expression instanceof MaterializeCastOperation) {
visit((MaterializeCastOperation) expression);
} else if (expression instanceof MaterializeBetweenOperation) {
visit((MaterializeBetweenOperation) expression);
} else if (expression instanceof MaterializeInOperation) {
visit((MaterializeInOperation) expression);
} else if (expression instanceof MaterializeAggregate) {
visit((MaterializeAggregate) expression);
} else if (expression instanceof MaterializePostfixText) {
visit((MaterializePostfixText) expression);
} else if (expression instanceof MaterializeSimilarTo) {
visit((MaterializeSimilarTo) expression);
} else if (expression instanceof MaterializePOSIXRegularExpression) {
visit((MaterializePOSIXRegularExpression) expression);
} else if (expression instanceof MaterializeFromTable) {
visit((MaterializeFromTable) expression);
} else if (expression instanceof MaterializeSubquery) {
visit((MaterializeSubquery) expression);
} else if (expression instanceof MaterializeLikeOperation) {
visit((MaterializeLikeOperation) expression);
} else {
throw new AssertionError(expression);
}
}
static String asString(MaterializeExpression expr) {
MaterializeToStringVisitor visitor = new MaterializeToStringVisitor();
visitor.visit(expr);
return visitor.get();
}
static String asExpectedValues(MaterializeExpression expr) {
MaterializeExpectedValueVisitor v = new MaterializeExpectedValueVisitor();
v.visit(expr);
return v.get();
}
static String getExpressionAsString(MaterializeGlobalState globalState, MaterializeDataType type,
List<MaterializeColumn> columns) {
MaterializeExpression expression = MaterializeExpressionGenerator.generateExpression(globalState, columns,
type);
MaterializeToStringVisitor visitor = new MaterializeToStringVisitor();
visitor.visit(expression);
return visitor.get();
}
}