2017-04-01 09:19:00 +00:00
|
|
|
#include <AggregateFunctions/AggregateFunctionFactory.h>
|
|
|
|
#include <AggregateFunctions/AggregateFunctionGroupArray.h>
|
|
|
|
#include <AggregateFunctions/Helpers.h>
|
2017-12-20 07:36:30 +00:00
|
|
|
#include <AggregateFunctions/FactoryHelpers.h>
|
|
|
|
|
2015-09-24 12:40:36 +00:00
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
2017-12-20 07:36:30 +00:00
|
|
|
namespace ErrorCodes
|
|
|
|
{
|
|
|
|
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
|
|
|
|
}
|
|
|
|
|
2015-09-24 12:40:36 +00:00
|
|
|
namespace
|
|
|
|
{
|
|
|
|
|
2017-09-15 12:16:12 +00:00
|
|
|
template <template <typename, typename> class AggregateFunctionTemplate, typename Data, typename ... TArgs>
|
2017-08-11 23:25:03 +00:00
|
|
|
static IAggregateFunction * createWithNumericOrTimeType(const IDataType & argument_type, TArgs && ... args)
|
|
|
|
{
|
2017-12-20 07:36:30 +00:00
|
|
|
if (typeid_cast<const DataTypeDate *>(&argument_type)) return new AggregateFunctionTemplate<UInt16, Data>(std::forward<TArgs>(args)...);
|
2017-08-11 23:25:03 +00:00
|
|
|
else if (typeid_cast<const DataTypeDateTime *>(&argument_type)) return new AggregateFunctionTemplate<UInt32, Data>(std::forward<TArgs>(args)...);
|
|
|
|
else return createWithNumericType<AggregateFunctionTemplate, Data, TArgs...>(argument_type, std::forward<TArgs>(args)...);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2017-08-12 00:12:43 +00:00
|
|
|
template <typename has_limit, typename ... TArgs>
|
2017-08-11 23:25:03 +00:00
|
|
|
inline AggregateFunctionPtr createAggregateFunctionGroupArrayImpl(const DataTypePtr & argument_type, TArgs ... args)
|
|
|
|
{
|
2017-08-12 00:12:43 +00:00
|
|
|
if (auto res = createWithNumericOrTimeType<GroupArrayNumericImpl, has_limit>(*argument_type, argument_type, std::forward<TArgs>(args)...))
|
2017-08-11 23:25:03 +00:00
|
|
|
return AggregateFunctionPtr(res);
|
|
|
|
|
|
|
|
if (typeid_cast<const DataTypeString *>(argument_type.get()))
|
2017-08-22 20:29:15 +00:00
|
|
|
return std::make_shared<GroupArrayGeneralListImpl<GroupArrayListNodeString, has_limit::value>>(std::forward<TArgs>(args)...);
|
2017-08-11 23:25:03 +00:00
|
|
|
|
2017-08-22 20:29:15 +00:00
|
|
|
return std::make_shared<GroupArrayGeneralListImpl<GroupArrayListNodeGeneral, has_limit::value>>(std::forward<TArgs>(args)...);
|
2017-08-11 23:25:03 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
|
2017-07-14 13:09:52 +00:00
|
|
|
static AggregateFunctionPtr createAggregateFunctionGroupArray(const std::string & name, const DataTypes & argument_types, const Array & parameters)
|
2017-06-26 12:16:29 +00:00
|
|
|
{
|
2017-12-20 07:36:30 +00:00
|
|
|
assertUnary(name, argument_types);
|
2017-06-26 12:16:29 +00:00
|
|
|
|
2017-07-10 23:30:17 +00:00
|
|
|
bool limit_size = false;
|
2017-08-11 23:25:03 +00:00
|
|
|
UInt64 max_elems = std::numeric_limits<UInt64>::max();
|
2017-07-10 23:30:17 +00:00
|
|
|
|
2017-07-11 17:43:51 +00:00
|
|
|
if (parameters.empty())
|
|
|
|
{
|
|
|
|
// no limit
|
|
|
|
}
|
2017-07-10 23:30:17 +00:00
|
|
|
else if (parameters.size() == 1)
|
|
|
|
{
|
2017-07-14 13:09:52 +00:00
|
|
|
auto type = parameters[0].getType();
|
|
|
|
if (type != Field::Types::Int64 && type != Field::Types::UInt64)
|
|
|
|
throw Exception("Parameter for aggregate function " + name + " should be positive number", ErrorCodes::BAD_ARGUMENTS);
|
2017-07-10 23:30:17 +00:00
|
|
|
|
2017-07-14 13:09:52 +00:00
|
|
|
if ((type == Field::Types::Int64 && parameters[0].get<Int64>() < 0) ||
|
|
|
|
(type == Field::Types::UInt64 && parameters[0].get<UInt64>() == 0))
|
2017-07-10 23:30:17 +00:00
|
|
|
throw Exception("Parameter for aggregate function " + name + " should be positive number", ErrorCodes::BAD_ARGUMENTS);
|
2017-07-14 13:09:52 +00:00
|
|
|
|
|
|
|
limit_size = true;
|
|
|
|
max_elems = parameters[0].get<UInt64>();
|
2017-07-10 23:30:17 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
throw Exception("Incorrect number of parameters for aggregate function " + name + ", should be 0 or 1",
|
|
|
|
ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH);
|
2017-06-26 12:16:29 +00:00
|
|
|
|
2017-07-10 23:30:17 +00:00
|
|
|
if (!limit_size)
|
2017-08-11 23:25:03 +00:00
|
|
|
return createAggregateFunctionGroupArrayImpl<std::false_type>(argument_types[0]);
|
2017-06-26 12:16:29 +00:00
|
|
|
else
|
2017-08-11 23:25:03 +00:00
|
|
|
return createAggregateFunctionGroupArrayImpl<std::true_type>(argument_types[0], max_elems);
|
2017-06-26 12:16:29 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2015-09-24 12:40:36 +00:00
|
|
|
|
|
|
|
void registerAggregateFunctionGroupArray(AggregateFunctionFactory & factory)
|
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
factory.registerFunction("groupArray", createAggregateFunctionGroupArray);
|
2015-09-24 12:40:36 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|