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

[Function](Exec) Support windows function cume_dist (#30997)

Browse Source
This commit is contained in:
nanfeng
2024-02-10 11:24:58 +08:00
committed by yiguolei
parent f65844fae4
commit e68019c10a
16 changed files with 450 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fe/fe-core/src/main/java/org/apache/doris/catalog/BuiltinWindowFunctions.java
View File

@ -17,6 +17,7 @@
package org.apache.doris.catalog;
import org.apache.doris.nereids.trees.expressions.functions.window.CumeDist;
import org.apache.doris.nereids.trees.expressions.functions.window.DenseRank;
import org.apache.doris.nereids.trees.expressions.functions.window.FirstValue;
import org.apache.doris.nereids.trees.expressions.functions.window.Lag;
@ -46,7 +47,8 @@ public class BuiltinWindowFunctions implements FunctionHelper {
window(Ntile.class, "ntile"),
window(PercentRank.class, "percent_rank"),
window(Rank.class, "rank"),
window(RowNumber.class, "row_number")
window(RowNumber.class, "row_number"),
window(CumeDist.class, "cume_dist")
);
public static final BuiltinWindowFunctions INSTANCE = new BuiltinWindowFunctions();

8
fe/fe-core/src/main/java/org/apache/doris/catalog/FunctionSet.java
View File

@ -1608,6 +1608,14 @@ public class FunctionSet<T> {
null,
"",
"", true));
// vec cume_dist
addBuiltin(AggregateFunction.createAnalyticBuiltin("cume_dist",
Lists.<Type>newArrayList(), Type.DOUBLE, Type.VARCHAR,
"",
"",
null,
"",
"", true));
//vec row_number
addBuiltin(AggregateFunction.createAnalyticBuiltin("row_number",
new ArrayList<Type>(), Type.BIGINT, Type.BIGINT,

13
fe/fe-core/src/main/java/org/apache/doris/nereids/rules/analysis/WindowFunctionChecker.java
View File

@ -26,6 +26,7 @@ import org.apache.doris.nereids.trees.expressions.WindowFrame;
import org.apache.doris.nereids.trees.expressions.WindowFrame.FrameBoundType;
import org.apache.doris.nereids.trees.expressions.WindowFrame.FrameBoundary;
import org.apache.doris.nereids.trees.expressions.WindowFrame.FrameUnitsType;
import org.apache.doris.nereids.trees.expressions.functions.window.CumeDist;
import org.apache.doris.nereids.trees.expressions.functions.window.DenseRank;
import org.apache.doris.nereids.trees.expressions.functions.window.FirstOrLastValue;
import org.apache.doris.nereids.trees.expressions.functions.window.FirstValue;
@ -387,6 +388,18 @@ public class WindowFunctionChecker extends DefaultExpressionVisitor<Expression,
return rowNumber;
}
/**
* required WindowFrame: (RANGE, UNBOUNDED PRECEDING, CURRENT ROW)
*/
@Override
public CumeDist visitCumeDist(CumeDist cumeDist, Void ctx) {
WindowFrame requiredFrame = new WindowFrame(FrameUnitsType.RANGE,
FrameBoundary.newPrecedingBoundary(), FrameBoundary.newCurrentRowBoundary());
checkAndCompleteWindowFrame(requiredFrame, cumeDist.getName());
return cumeDist;
}
/**
* required WindowFrame: (ROWS, UNBOUNDED PRECEDING, CURRENT ROW)
*/

61
fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/functions/window/CumeDist.java Normal file
View File

@ -0,0 +1,61 @@
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
package org.apache.doris.nereids.trees.expressions.functions.window;
import org.apache.doris.catalog.FunctionSignature;
import org.apache.doris.nereids.trees.expressions.Expression;
import org.apache.doris.nereids.trees.expressions.functions.AlwaysNotNullable;
import org.apache.doris.nereids.trees.expressions.shape.LeafExpression;
import org.apache.doris.nereids.trees.expressions.visitor.ExpressionVisitor;
import org.apache.doris.nereids.types.DoubleType;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableList;
import java.util.List;
/**
* Window function: cume_dist()
*/
public class CumeDist extends WindowFunction implements AlwaysNotNullable, LeafExpression {
public CumeDist() {
super("cume_dist");
}
@Override
public List<FunctionSignature> getSignatures() {
return ImmutableList.of(FunctionSignature.ret(DoubleType.INSTANCE).args());
}
@Override
public FunctionSignature searchSignature(List<FunctionSignature> signatures) {
return signatures.get(0);
}
@Override
public CumeDist withChildren(List<Expression> children) {
Preconditions.checkArgument(children.size() == 0);
return new CumeDist();
}
@Override
public <R, C> R accept(ExpressionVisitor<R, C> visitor, C context) {
return visitor.visitCumeDist(this, context);
}
}

5
fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/visitor/WindowFunctionVisitor.java
View File

@ -17,6 +17,7 @@
package org.apache.doris.nereids.trees.expressions.visitor;
import org.apache.doris.nereids.trees.expressions.functions.window.CumeDist;
import org.apache.doris.nereids.trees.expressions.functions.window.DenseRank;
import org.apache.doris.nereids.trees.expressions.functions.window.FirstValue;
import org.apache.doris.nereids.trees.expressions.functions.window.Lag;
@ -68,4 +69,8 @@ public interface WindowFunctionVisitor<R, C> {
default R visitRowNumber(RowNumber rowNumber, C context) {
return visitWindowFunction(rowNumber, context);
}
default R visitCumeDist(CumeDist cumeDist, C context) {
return visitWindowFunction(cumeDist, context);
}
}