database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[Feature] Support Array Type compare function for nereids planner (#31701)

Browse Source 
Support Array Type compare function for nereids planner
This commit is contained in:
koarz
2024-03-20 20:11:57 +08:00
committed by yiguolei
parent 6d076f9947
commit 8e3d28b93a
6 changed files with 163 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
fe/fe-core/src/main/java/org/apache/doris/analysis/ArrayLiteral.java
View File

@ -99,7 +99,14 @@ public class ArrayLiteral extends LiteralExpr {
@Override
public int compareLiteral(LiteralExpr expr) {
return 0;
int size = Math.min(expr.getChildren().size(), this.children.size());
for (int i = 0; i < size; i++) {
if (((LiteralExpr) (this.getChild(i))).compareTo((LiteralExpr) (expr.getChild(i))) != 0) {
return ((LiteralExpr) (this.getChild(i))).compareTo((LiteralExpr) (expr.getChild(i)));
}
}
return this.children.size() > expr.getChildren().size() ? 1 :
(this.children.size() == expr.getChildren().size() ? 0 : -1);
}
@Override

2
fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/ComparisonPredicate.java
View File

@ -62,7 +62,7 @@ public abstract class ComparisonPredicate extends BinaryOperator {
@Override
public void checkLegalityBeforeTypeCoercion() {
children().forEach(c -> {
if (c.getDataType().isComplexType()) {
if (c.getDataType().isComplexType() && !c.getDataType().isArrayType()) {
throw new AnalysisException("comparison predicate could not contains complex type: " + this.toSql());
}
});

3
fe/fe-core/src/main/java/org/apache/doris/nereids/util/TypeCoercionUtils.java
View File

@ -1635,6 +1635,9 @@ public class TypeCoercionUtils {
}
private static boolean supportCompare(DataType dataType) {
if (dataType.isArrayType()) {
return true;
}
if (!(dataType instanceof PrimitiveType)) {
return false;
}