forked from sqlancer/sqlancer
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathTypedExpressionGenerator.java
More file actions
79 lines (61 loc) · 2.25 KB
/
TypedExpressionGenerator.java
File metadata and controls
79 lines (61 loc) · 2.25 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
package sqlancer.common.gen;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import sqlancer.Randomly;
public abstract class TypedExpressionGenerator<E, C, T> implements ExpressionGenerator<E> {
protected List<C> columns = Collections.emptyList();
protected boolean allowAggregates;
public E generateExpression(T type) {
return generateExpression(type, 0);
}
public abstract E generateConstant(T type);
protected abstract E generateExpression(T type, int depth);
protected abstract E generateColumn(T type);
protected abstract T getRandomType();
protected abstract boolean canGenerateColumnOfType(T type);
@SuppressWarnings("unchecked") // unsafe
public <U extends TypedExpressionGenerator<E, C, T>> U setColumns(List<C> columns) {
this.columns = columns;
return (U) this;
}
public E generateLeafNode(T type) {
if (Randomly.getBooleanWithRatherLowProbability() || !canGenerateColumnOfType(type)) {
return generateConstant(type);
} else {
return generateColumn(type);
}
}
public List<E> generateExpressions(T type, int nr) {
List<E> expressions = new ArrayList<>();
for (int i = 0; i < nr; i++) {
expressions.add(generateExpression(type));
}
return expressions;
}
public List<E> generateExpressions(T type, int nr, int depth) {
List<E> expressions = new ArrayList<>();
for (int i = 0; i < nr; i++) {
expressions.add(generateExpression(type, depth));
}
return expressions;
}
public List<E> generateExpressions(int nr) {
List<E> expressions = new ArrayList<>();
for (int i = 0; i < nr; i++) {
expressions.add(generateExpression(getRandomType()));
}
return expressions;
}
// override this class to also generate ASC, DESC
public List<E> generateOrderBys() {
return generateExpressions(Randomly.smallNumber() + 1);
}
// override this class to generate aggregate functions
// public E generateHavingClause() {
// allowAggregates = true;
// E expr = generateExpression();
// allowAggregates = false;
// return expr;
// }
}