-
Notifications
You must be signed in to change notification settings - Fork 397
Expand file tree
/
Copy pathDatabendInOperation.java
More file actions
52 lines (44 loc) · 1.69 KB
/
DatabendInOperation.java
File metadata and controls
52 lines (44 loc) · 1.69 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
package sqlancer.databend.ast;
import java.util.List;
import sqlancer.common.ast.newast.NewInOperatorNode;
import sqlancer.databend.DatabendSchema;
public class DatabendInOperation extends NewInOperatorNode<DatabendExpression> implements DatabendExpression {
private final DatabendExpression leftExpr;
private final List<DatabendExpression> rightExpr;
public DatabendInOperation(DatabendExpression left, List<DatabendExpression> right, boolean isNegated) {
super(left, right, isNegated);
this.leftExpr = left;
this.rightExpr = right;
}
@Override
public DatabendSchema.DatabendDataType getExpectedType() {
return DatabendSchema.DatabendDataType.BOOLEAN;
}
@Override
public DatabendConstant getExpectedValue() {
DatabendConstant leftValue = leftExpr.getExpectedValue();
if (leftValue == null) {
return null;
}
if (leftValue.isNull()) {
return DatabendConstant.createNullConstant();
}
boolean isNull = false;
for (DatabendExpression expr : rightExpr) {
DatabendConstant rightValue = expr.getExpectedValue();
if (rightValue == null) {
return null;
}
if (rightValue.isNull()) {
isNull = true;
} else if (rightValue.isEquals(leftValue).isBoolean() && rightValue.isEquals(leftValue).asBoolean()) {
return DatabendConstant.createBooleanConstant(!isNegated());
}
}
if (isNull) {
return DatabendConstant.createNullConstant();
} else {
return DatabendConstant.createBooleanConstant(isNegated());
}
}
}