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

[Feature](function) support random int from specific range (#28076)

Browse Source 
mysql> select rand(-20, -10);
+------------------+
| random(-20, -10) |
+------------------+
|              -13 |
+------------------+
1 row in set (0.10 sec)
This commit is contained in:
zclllyybb
2023-12-08 10:15:25 +08:00
committed by GitHub
parent e75d91c91b
commit 25b90eb782
8 changed files with 160 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/functions/scalar/Random.java
View File

@ -18,6 +18,7 @@
package org.apache.doris.nereids.trees.expressions.functions.scalar;
import org.apache.doris.catalog.FunctionSignature;
import org.apache.doris.nereids.exceptions.AnalysisException;
import org.apache.doris.nereids.trees.expressions.Expression;
import org.apache.doris.nereids.trees.expressions.functions.ExplicitlyCastableSignature;
import org.apache.doris.nereids.trees.expressions.functions.Nondeterministic;
@ -39,7 +40,8 @@ public class Random extends ScalarFunction
public static final List<FunctionSignature> SIGNATURES = ImmutableList.of(
FunctionSignature.ret(DoubleType.INSTANCE).args(),
FunctionSignature.ret(DoubleType.INSTANCE).args(BigIntType.INSTANCE)
FunctionSignature.ret(DoubleType.INSTANCE).args(BigIntType.INSTANCE),
FunctionSignature.ret(BigIntType.INSTANCE).args(BigIntType.INSTANCE, BigIntType.INSTANCE)
);
/**
@ -58,6 +60,17 @@ public class Random extends ScalarFunction
Preconditions.checkState(arg instanceof Literal, "The param of rand function must be literal");
}
/**
* constructor with 2 argument.
*/
public Random(Expression lchild, Expression rchild) {
super("random", lchild, rchild);
// align with original planner behavior, refer to:
// org/apache/doris/analysis/Expr.getBuiltinFunction()
Preconditions.checkState(lchild instanceof Literal && rchild instanceof Literal,
"The param of rand function must be literal");
}
/**
* custom compute nullable.
*/
@ -80,13 +93,14 @@ public class Random extends ScalarFunction
*/
@Override
public Random withChildren(List<Expression> children) {
Preconditions.checkArgument(children.size() == 0
|| children.size() == 1);
if (children.isEmpty() && arity() == 0) {
return this;
} else {
if (children.isEmpty()) {
return new Random();
} else if (children.size() == 1) {
return new Random(children.get(0));
} else if (children.size() == 2) {
return new Random(children.get(0), children.get(1));
}
throw new AnalysisException("random function only accept 0-2 arguments");
}
@Override