ClickHouse/src/AggregateFunctions/AggregateFunctionSum.cpp

81 lines
2.9 KiB
C++
Raw Normal View History

#include <AggregateFunctions/AggregateFunctionFactory.h>
#include <AggregateFunctions/AggregateFunctionSum.h>
#include <AggregateFunctions/Helpers.h>
#include <AggregateFunctions/FactoryHelpers.h>
2015-09-24 12:40:36 +00:00
namespace DB
{
struct Settings;
2020-02-25 18:10:48 +00:00
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
}
2015-09-24 12:40:36 +00:00
namespace
{
template <typename T>
2018-09-11 18:42:06 +00:00
struct SumSimple
{
/// @note It uses slow Decimal128 (cause we need such a variant). sumWithOverflow is faster for Decimal32/64
2021-09-10 11:49:22 +00:00
using ResultType = std::conditional_t<is_decimal<T>,
std::conditional_t<std::is_same_v<T, Decimal256>, Decimal256, Decimal128>,
NearestFieldType<T>>;
2018-09-11 18:42:06 +00:00
using AggregateDataType = AggregateFunctionSumData<ResultType>;
using Function = AggregateFunctionSum<T, ResultType, AggregateDataType, AggregateFunctionTypeSum>;
2018-09-11 18:42:06 +00:00
};
template <typename T>
2018-09-11 18:42:06 +00:00
struct SumSameType
{
using ResultType = T;
using AggregateDataType = AggregateFunctionSumData<ResultType>;
using Function = AggregateFunctionSum<T, ResultType, AggregateDataType, AggregateFunctionTypeSumWithOverflow>;
2018-09-11 18:42:06 +00:00
};
template <typename T>
2018-09-11 18:42:06 +00:00
struct SumKahan
{
using ResultType = Float64;
using AggregateDataType = AggregateFunctionSumKahanData<ResultType>;
using Function = AggregateFunctionSum<T, ResultType, AggregateDataType, AggregateFunctionTypeSumKahan>;
2018-09-11 18:42:06 +00:00
};
template <typename T> using AggregateFunctionSumSimple = typename SumSimple<T>::Function;
template <typename T> using AggregateFunctionSumWithOverflow = typename SumSameType<T>::Function;
template <typename T> using AggregateFunctionSumKahan =
2021-09-10 11:49:22 +00:00
std::conditional_t<is_decimal<T>, typename SumSimple<T>::Function, typename SumKahan<T>::Function>;
template <template <typename> class Function>
AggregateFunctionPtr createAggregateFunctionSum(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
2015-09-24 12:40:36 +00:00
{
assertNoParameters(name, parameters);
assertUnary(name, argument_types);
2015-09-24 12:40:36 +00:00
2018-09-11 18:42:06 +00:00
AggregateFunctionPtr res;
2022-04-15 23:15:40 +00:00
const DataTypePtr & data_type = argument_types[0];
2018-09-11 18:42:06 +00:00
if (isDecimal(data_type))
2019-02-11 19:26:32 +00:00
res.reset(createWithDecimalType<Function>(*data_type, *data_type, argument_types));
2018-09-11 18:42:06 +00:00
else
2019-02-11 19:26:32 +00:00
res.reset(createWithNumericType<Function>(*data_type, argument_types));
2015-09-24 12:40:36 +00:00
if (!res)
2018-09-11 18:42:06 +00:00
throw Exception("Illegal type " + argument_types[0]->getName() + " of argument for aggregate function " + name,
ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT);
return res;
2015-09-24 12:40:36 +00:00
}
}
void registerAggregateFunctionSum(AggregateFunctionFactory & factory)
{
factory.registerFunction("sum", createAggregateFunctionSum<AggregateFunctionSumSimple>, AggregateFunctionFactory::CaseInsensitive);
factory.registerFunction("sumWithOverflow", createAggregateFunctionSum<AggregateFunctionSumWithOverflow>);
factory.registerFunction("sumKahan", createAggregateFunctionSum<AggregateFunctionSumKahan>);
2015-09-24 12:40:36 +00:00
}
}