mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-13 18:02:24 +00:00
70d1adfe4b
* save format string for NetException * format exceptions * format exceptions 2 * format exceptions 3 * format exceptions 4 * format exceptions 5 * format exceptions 6 * fix * format exceptions 7 * format exceptions 8 * Update MergeTreeIndexGin.cpp * Update AggregateFunctionMap.cpp * Update AggregateFunctionMap.cpp * fix
52 lines
1.4 KiB
C++
52 lines
1.4 KiB
C++
#include <memory>
|
|
#include <AggregateFunctions/AggregateFunctionFactory.h>
|
|
#include <AggregateFunctions/AggregateFunctionAvg.h>
|
|
#include <AggregateFunctions/Helpers.h>
|
|
#include <AggregateFunctions/FactoryHelpers.h>
|
|
|
|
namespace DB
|
|
{
|
|
struct Settings;
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
|
|
}
|
|
|
|
namespace
|
|
{
|
|
bool allowType(const DataTypePtr& type) noexcept
|
|
{
|
|
const WhichDataType t(type);
|
|
return t.isInt() || t.isUInt() || t.isFloat() || t.isDecimal();
|
|
}
|
|
|
|
AggregateFunctionPtr createAggregateFunctionAvg(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
|
|
{
|
|
assertNoParameters(name, parameters);
|
|
assertUnary(name, argument_types);
|
|
|
|
const DataTypePtr& data_type = argument_types[0];
|
|
|
|
if (!allowType(data_type))
|
|
throw Exception(ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT, "Illegal type {} of argument for aggregate function {}",
|
|
data_type->getName(), name);
|
|
|
|
AggregateFunctionPtr res;
|
|
|
|
if (isDecimal(data_type))
|
|
res.reset(createWithDecimalType<AggregateFunctionAvg>(
|
|
*data_type, argument_types, getDecimalScale(*data_type)));
|
|
else
|
|
res.reset(createWithNumericType<AggregateFunctionAvg>(*data_type, argument_types));
|
|
|
|
return res;
|
|
}
|
|
}
|
|
|
|
void registerAggregateFunctionAvg(AggregateFunctionFactory & factory)
|
|
{
|
|
factory.registerFunction("avg", createAggregateFunctionAvg, AggregateFunctionFactory::CaseInsensitive);
|
|
}
|
|
}
|