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

[decimalv3](function) support function width_bucket (#19806)

Browse Source
This commit is contained in:
Gabriel
2023-05-19 20:28:59 +08:00
committed by GitHub
parent 78bcc68ab8
commit 5547bbbaef
3 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
be/src/vec/functions/function_width_bucket.cpp
View File

@ -138,6 +138,15 @@ private:
} else if (which.is_decimal128()) {
_execute<ColumnDecimal128>(expr_column, min_value_column, max_value_column, num_buckets,
nested_column_column);
} else if (which.is_decimal32()) {
_execute<ColumnDecimal32>(expr_column, min_value_column, max_value_column, num_buckets,
nested_column_column);
} else if (which.is_decimal64()) {
_execute<ColumnDecimal64>(expr_column, min_value_column, max_value_column, num_buckets,
nested_column_column);
} else if (which.is_decimal128i()) {
_execute<ColumnDecimal128I>(expr_column, min_value_column, max_value_column,
num_buckets, nested_column_column);
} else if (which.is_date()) {
_execute<ColumnDate>(expr_column, min_value_column, max_value_column, num_buckets,
nested_column_column);

6
fe/fe-core/src/main/java/org/apache/doris/analysis/FunctionCallExpr.java
View File

@ -1646,9 +1646,9 @@ public class FunctionCallExpr extends Expr {
|| fnName.getFunction().equalsIgnoreCase("shuffle")
|| fnName.getFunction().equalsIgnoreCase("array_except")
|| fnName.getFunction().equalsIgnoreCase("array_contains")
|| fnName.getFunction().equalsIgnoreCase("array_position"))
&& ((args[ix].isDecimalV3())
|| (children.get(0).getType().isArrayType()
|| fnName.getFunction().equalsIgnoreCase("array_position")
|| fnName.getFunction().equalsIgnoreCase("width_bucket"))
&& (args[ix].isDecimalV3() || (children.get(0).getType().isArrayType()
&& (((ArrayType) children.get(0).getType()).getItemType().isDecimalV3())
&& (args[ix].isArrayType())
&& ((ArrayType) args[ix]).getItemType().isDecimalV3()))) {

3
gensrc/script/doris_builtins_functions.py
View File

@ -1533,6 +1533,9 @@ visible_functions = [
[['width_bucket'], 'BIGINT', ['BIGINT','BIGINT','BIGINT','BIGINT'], ''],
[['width_bucket'], 'BIGINT', ['FLOAT','FLOAT','FLOAT','BIGINT'], ''],
[['width_bucket'], 'BIGINT', ['DOUBLE','DOUBLE','DOUBLE','BIGINT'], ''],
[['width_bucket'], 'BIGINT', ['DECIMAL32','DECIMAL32','DECIMAL32','BIGINT'], ''],
[['width_bucket'], 'BIGINT', ['DECIMAL64','DECIMAL64','DECIMAL64','BIGINT'], ''],
[['width_bucket'], 'BIGINT', ['DECIMAL128','DECIMAL128','DECIMAL128','BIGINT'], ''],
[['width_bucket'], 'BIGINT', ['DECIMALV2','DECIMALV2','DECIMALV2','BIGINT'], ''],
[['width_bucket'], 'BIGINT', ['DATE','DATE','DATE','BIGINT'], ''],
[['width_bucket'], 'BIGINT', ['DATEV2','DATEV2','DATEV2','BIGINT'], ''],