-
Notifications
You must be signed in to change notification settings - Fork 397
Expand file tree
/
Copy pathClickHouseAggregate.java
More file actions
81 lines (62 loc) · 3.2 KB
/
ClickHouseAggregate.java
File metadata and controls
81 lines (62 loc) · 3.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
package sqlancer.clickhouse.ast;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
import com.clickhouse.client.ClickHouseDataType;
import sqlancer.Randomly;
import sqlancer.clickhouse.ClickHouseSchema;
public class ClickHouseAggregate extends ClickHouseExpression {
private final ClickHouseAggregate.ClickHouseAggregateFunction func;
private final ClickHouseExpression expr;
public enum ClickHouseAggregateFunction {
AVG(ClickHouseDataType.Int8, ClickHouseDataType.Int16, ClickHouseDataType.Int32, ClickHouseDataType.Int64,
ClickHouseDataType.UInt8, ClickHouseDataType.UInt16, ClickHouseDataType.UInt32,
ClickHouseDataType.UInt64, ClickHouseDataType.Float32, ClickHouseDataType.Float64),
COUNT(ClickHouseDataType.Int8, ClickHouseDataType.Int16, ClickHouseDataType.Int32, ClickHouseDataType.Int64,
ClickHouseDataType.UInt8, ClickHouseDataType.UInt16, ClickHouseDataType.UInt32,
ClickHouseDataType.UInt64, ClickHouseDataType.Float32, ClickHouseDataType.Float64,
ClickHouseDataType.String),
MAX, MIN,
SUM(ClickHouseDataType.Int8, ClickHouseDataType.Int16, ClickHouseDataType.Int32, ClickHouseDataType.Int64,
ClickHouseDataType.UInt8, ClickHouseDataType.UInt16, ClickHouseDataType.UInt32,
ClickHouseDataType.UInt64, ClickHouseDataType.Float32, ClickHouseDataType.Float64);
private ClickHouseDataType[] supportedReturnTypes;
ClickHouseAggregateFunction(ClickHouseDataType... supportedReturnTypes) {
this.supportedReturnTypes = supportedReturnTypes.clone();
}
public static ClickHouseAggregateFunction getRandom() {
return Randomly.fromOptions(values());
}
public static ClickHouseAggregateFunction getRandom(ClickHouseDataType type) {
return Randomly.fromOptions(values());
}
public ClickHouseDataType getType(ClickHouseDataType returnType) {
return returnType;
}
public boolean supportsReturnType(ClickHouseDataType returnType) {
return Arrays.asList(supportedReturnTypes).stream().anyMatch(t -> t == returnType)
|| supportedReturnTypes.length == 0;
}
public static List<ClickHouseAggregateFunction> getAggregates(ClickHouseDataType type) {
return Arrays.asList(values()).stream().filter(p -> p.supportsReturnType(type))
.collect(Collectors.toList());
}
public ClickHouseSchema.ClickHouseLancerDataType getRandomReturnType() {
if (supportedReturnTypes.length == 0) {
return ClickHouseSchema.ClickHouseLancerDataType.getRandom();
} else {
return new ClickHouseSchema.ClickHouseLancerDataType(Randomly.fromOptions(supportedReturnTypes));
}
}
}
public ClickHouseAggregate(ClickHouseExpression expr, ClickHouseAggregateFunction func) {
this.expr = expr;
this.func = func;
}
public ClickHouseAggregate.ClickHouseAggregateFunction getFunc() {
return func;
}
public ClickHouseExpression getExpr() {
return expr;
}
}