thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-24 08:32:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Set function divide as suitable for short-circuit in case of Nullable(Decimal)

Browse Source
This commit is contained in:
Pavel Kruglov 2021-08-17 16:24:28 +03:00
parent 98eb619b4b
commit bc853d250f
5 changed files with 26 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Functions/FunctionBinaryArithmetic.h
View File

@ -958,7 +958,7 @@ public:
bool isSuitableForShortCircuitArgumentsExecution(const DataTypesWithConstInfo & arguments) const override
{
return ((IsOperation<Op>::div_int || IsOperation<Op>::modulo) && !arguments[1].is_const)
|| (IsOperation<Op>::div_floating && (isDecimal(arguments[0].type) || isDecimal(arguments[1].type)));
|| (IsOperation<Op>::div_floating && (isDecimalOrNullableDecimal(arguments[0].type) || isDecimalOrNullableDecimal(arguments[1].type)));
}
DataTypePtr getReturnTypeImpl(const DataTypes & arguments) const override

10
src/Functions/FunctionHelpers.cpp
View File

@ -303,4 +303,14 @@ NullPresence getNullPresense(const ColumnsWithTypeAndName & args)
return res;
}
bool isDecimalOrNullableDecimal(const DataTypePtr & type)
{
WhichDataType which(type);
if (which.isDecimal())
return true;
if (!which.isNullable())
return false;
return isDecimal(assert_cast<const DataTypeNullable *>(type.get())->getNestedType());
}
}

2
src/Functions/FunctionHelpers.h
View File

@ -3,6 +3,7 @@
#include <Common/typeid_cast.h>
#include <Common/assert_cast.h>
#include <DataTypes/IDataType.h>
#include <DataTypes/DataTypeNullable.h>
#include <Columns/IColumn.h>
#include <Columns/ColumnArray.h>
#include <Columns/ColumnConst.h>
@ -171,4 +172,5 @@ struct NullPresence
NullPresence getNullPresense(const ColumnsWithTypeAndName & args);
bool isDecimalOrNullableDecimal(const DataTypePtr & type);
}

8
tests/queries/0_stateless/01822_short_circuit.reference
View File

@ -1395,3 +1395,11 @@ Decimal32
21.00
14.00
10.50
0.00
42.00
21.00
14.00
10.50
\N
\N
\N

5
tests/queries/0_stateless/01822_short_circuit.sql
View File

@ -123,3 +123,8 @@ select if(number > 0, intDiv(42, number), 1) from numbers(5);
select if(number > 0, intDiv(42, number), 1) from numbers(5);
select if(number > 0, 42 / toDecimal32(number, 2), 0) from numbers(5);
select if(number = 0, 0, toDecimal32(42, 2) / number) from numbers(5);
select if(isNull(x), Null, 42 / x) from (select CAST(materialize(Null), 'Nullable(Decimal32(2))') as x);
select if(isNull(x), Null, x / 0) from (select CAST(materialize(Null), 'Nullable(Decimal32(2))') as x);
select if(isNull(x), Null, intDiv(42, x)) from (select CAST(materialize(Null), 'Nullable(Int64)') as x);