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

Merge pull request #22762 from vdimir/create-aggregate-func-settings

Browse Source
This commit is contained in:
Vladimir 2021-05-26 14:32:14 +03:00 committed by GitHub
commit 6650c67b01
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
127 changed files with 257 additions and 76 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/AggregateFunctions/AggregateFunctionAggThrow.cpp
View File

@ -11,6 +11,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -105,7 +106,7 @@ public:
void registerAggregateFunctionAggThrow(AggregateFunctionFactory & factory)
{
factory.registerFunction("aggThrow", [](const std::string & name, const DataTypes & argument_types, const Array & parameters)
factory.registerFunction("aggThrow", [](const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
Float64 throw_probability = 1.0;
if (parameters.size() == 1)

15
src/AggregateFunctions/AggregateFunctionAny.cpp
View File

@ -1,28 +1,27 @@
#include <AggregateFunctions/AggregateFunctionFactory.h>
#include <AggregateFunctions/HelpersMinMaxAny.h>
#include <AggregateFunctions/FactoryHelpers.h>
#include "registerAggregateFunctions.h"
namespace DB
{
struct Settings;
namespace
{
AggregateFunctionPtr createAggregateFunctionAny(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionAny(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings * settings)
{
return AggregateFunctionPtr(createAggregateFunctionSingleValue<AggregateFunctionsSingleValue, AggregateFunctionAnyData>(name, argument_types, parameters));
return AggregateFunctionPtr(createAggregateFunctionSingleValue<AggregateFunctionsSingleValue, AggregateFunctionAnyData>(name, argument_types, parameters, settings));
}
AggregateFunctionPtr createAggregateFunctionAnyLast(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionAnyLast(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings * settings)
{
return AggregateFunctionPtr(createAggregateFunctionSingleValue<AggregateFunctionsSingleValue, AggregateFunctionAnyLastData>(name, argument_types, parameters));
return AggregateFunctionPtr(createAggregateFunctionSingleValue<AggregateFunctionsSingleValue, AggregateFunctionAnyLastData>(name, argument_types, parameters, settings));
}
AggregateFunctionPtr createAggregateFunctionAnyHeavy(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionAnyHeavy(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings * settings)
{
return AggregateFunctionPtr(createAggregateFunctionSingleValue<AggregateFunctionsSingleValue, AggregateFunctionAnyHeavyData>(name, argument_types, parameters));
return AggregateFunctionPtr(createAggregateFunctionSingleValue<AggregateFunctionsSingleValue, AggregateFunctionAnyHeavyData>(name, argument_types, parameters, settings));
}
}

1
src/AggregateFunctions/AggregateFunctionArgMinMax.h
View File

@ -8,6 +8,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionArray.cpp
View File

@ -5,6 +5,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionArray.h
View File

@ -9,6 +9,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

5
src/AggregateFunctions/AggregateFunctionAvg.cpp
View File

@ -3,10 +3,11 @@
#include <AggregateFunctions/AggregateFunctionAvg.h>
#include <AggregateFunctions/Helpers.h>
#include <AggregateFunctions/FactoryHelpers.h>
#include "registerAggregateFunctions.h"
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
@ -20,7 +21,7 @@ bool allowType(const DataTypePtr& type) noexcept
return t.isInt() || t.isUInt() || t.isFloat() || t.isDecimal();
}
AggregateFunctionPtr createAggregateFunctionAvg(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionAvg(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertUnary(name, argument_types);

1
src/AggregateFunctions/AggregateFunctionAvg.h
View File

@ -12,6 +12,7 @@
namespace DB
{
struct Settings;
template <typename T>
using DecimalOrVectorCol = std::conditional_t<IsDecimalNumber<T>, ColumnDecimal<T>, ColumnVector<T>>;

6
src/AggregateFunctions/AggregateFunctionAvgWeighted.cpp
View File

@ -4,10 +4,11 @@
#include <AggregateFunctions/AggregateFunctionAvgWeighted.h>
#include <AggregateFunctions/Helpers.h>
#include <AggregateFunctions/FactoryHelpers.h>
#include "registerAggregateFunctions.h"
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
@ -60,7 +61,8 @@ static IAggregateFunction * create(const IDataType & first_type, const IDataType
#undef LINE
}
AggregateFunctionPtr createAggregateFunctionAvgWeighted(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr
createAggregateFunctionAvgWeighted(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertBinary(name, argument_types);

2
src/AggregateFunctions/AggregateFunctionAvgWeighted.h
View File

@ -5,6 +5,8 @@
namespace DB
{
struct Settings;
template <typename T>
using AvgWeightedFieldType = std::conditional_t<IsDecimalNumber<T>,
std::conditional_t<std::is_same_v<T, Decimal256>, Decimal256, Decimal128>,

4
src/AggregateFunctions/AggregateFunctionBitwise.cpp
View File

@ -7,6 +7,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
@ -16,7 +18,7 @@ namespace
{
template <template <typename> class Data>
AggregateFunctionPtr createAggregateFunctionBitwise(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionBitwise(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertUnary(name, argument_types);

1
src/AggregateFunctions/AggregateFunctionBitwise.h
View File

@ -12,6 +12,7 @@
namespace DB
{
struct Settings;
template <typename T>

3
src/AggregateFunctions/AggregateFunctionBoundingRatio.cpp
View File

@ -6,6 +6,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -15,7 +16,7 @@ namespace ErrorCodes
namespace
{
AggregateFunctionPtr createAggregateFunctionRate(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionRate(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertBinary(name, argument_types);

1
src/AggregateFunctions/AggregateFunctionBoundingRatio.h
View File

@ -11,6 +11,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

5
src/AggregateFunctions/AggregateFunctionCategoricalInformationValue.cpp
View File

@ -8,6 +8,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -21,8 +22,8 @@ namespace
AggregateFunctionPtr createAggregateFunctionCategoricalIV(
const std::string & name,
const DataTypes & arguments,
const Array & params
)
const Array & params,
const Settings *)
{
assertNoParameters(name, params);

1
src/AggregateFunctions/AggregateFunctionCategoricalInformationValue.h
View File

@ -13,6 +13,7 @@
namespace DB
{
struct Settings;
template <typename T = UInt64>
class AggregateFunctionCategoricalIV final : public IAggregateFunctionHelper<AggregateFunctionCategoricalIV<T>>

1
src/AggregateFunctions/AggregateFunctionCombinatorFactory.cpp
View File

@ -5,6 +5,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionCombinatorFactory.h
View File

@ -9,6 +9,7 @@
namespace DB
{
struct Settings;
/** Create aggregate function combinator by matching suffix in aggregate function name.
*/

3
src/AggregateFunctions/AggregateFunctionCount.cpp
View File

@ -6,6 +6,7 @@
namespace DB
{
struct Settings;
AggregateFunctionPtr AggregateFunctionCount::getOwnNullAdapter(
const AggregateFunctionPtr &, const DataTypes & types, const Array & params, const AggregateFunctionProperties & /*properties*/) const
@ -16,7 +17,7 @@ AggregateFunctionPtr AggregateFunctionCount::getOwnNullAdapter(
namespace
{
AggregateFunctionPtr createAggregateFunctionCount(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionCount(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertArityAtMost<1>(name, argument_types);

1
src/AggregateFunctions/AggregateFunctionCount.h
View File

@ -13,6 +13,7 @@
namespace DB
{
struct Settings;
struct AggregateFunctionCountData
{

4
src/AggregateFunctions/AggregateFunctionDeltaSum.cpp
View File

@ -7,6 +7,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -20,7 +21,8 @@ namespace
AggregateFunctionPtr createAggregateFunctionDeltaSum(
const String & name,
const DataTypes & arguments,
const Array & params)
const Array & params,
const Settings *)
{
assertNoParameters(name, params);

1
src/AggregateFunctions/AggregateFunctionDeltaSum.h
View File

@ -15,6 +15,7 @@
namespace DB
{
struct Settings;
template <typename T>
struct AggregationFunctionDeltaSumData

3
src/AggregateFunctions/AggregateFunctionDeltaSumTimestamp.cpp
View File

@ -20,7 +20,8 @@ namespace
AggregateFunctionPtr createAggregateFunctionDeltaSumTimestamp(
const String & name,
const DataTypes & arguments,
const Array & params)
const Array & params,
const Settings *)
{
assertNoParameters(name, params);

2
src/AggregateFunctions/AggregateFunctionDistinct.cpp
View File

@ -6,6 +6,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;

1
src/AggregateFunctions/AggregateFunctionDistinct.h
View File

@ -13,6 +13,7 @@
namespace DB
{
struct Settings;
template <typename T>

3
src/AggregateFunctions/AggregateFunctionEntropy.cpp
View File

@ -7,6 +7,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -16,7 +17,7 @@ namespace ErrorCodes
namespace
{
AggregateFunctionPtr createAggregateFunctionEntropy(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionEntropy(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
if (argument_types.empty())

1
src/AggregateFunctions/AggregateFunctionEntropy.h
View File

@ -14,6 +14,7 @@
namespace DB
{
struct Settings;
/** Calculates Shannon Entropy, using HashMap and computing empirical distribution function.
* Entropy is measured in bits (base-2 logarithm is used).

4
src/AggregateFunctions/AggregateFunctionFactory.cpp
View File

@ -23,6 +23,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -137,7 +138,8 @@ AggregateFunctionPtr AggregateFunctionFactory::getImpl(
if (!out_properties.returns_default_when_only_null && has_null_arguments)
return nullptr;
return found.creator(name, argument_types, parameters);
const Settings * settings = query_context ? &query_context->getSettingsRef() : nullptr;
return found.creator(name, argument_types, parameters, settings);
}
/// Combinators of aggregate functions.

3
src/AggregateFunctions/AggregateFunctionFactory.h
View File

@ -14,6 +14,7 @@
namespace DB
{
struct Settings;
class Context;
class IDataType;
@ -26,7 +27,7 @@ using DataTypes = std::vector<DataTypePtr>;
* Parameters are for "parametric" aggregate functions.
* For example, in quantileWeighted(0.9)(x, weight), 0.9 is "parameter" and x, weight are "arguments".
*/
using AggregateFunctionCreator = std::function<AggregateFunctionPtr(const String &, const DataTypes &, const Array &)>;
using AggregateFunctionCreator = std::function<AggregateFunctionPtr(const String &, const DataTypes &, const Array &, const Settings *)>;
struct AggregateFunctionWithProperties
{

1
src/AggregateFunctions/AggregateFunctionForEach.cpp
View File

@ -6,6 +6,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionForEach.h
View File

@ -13,6 +13,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

5
src/AggregateFunctions/AggregateFunctionGroupArray.cpp
View File

@ -9,6 +9,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -49,7 +50,7 @@ inline AggregateFunctionPtr createAggregateFunctionGroupArrayImpl(const DataType
}
AggregateFunctionPtr createAggregateFunctionGroupArray(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionGroupArray(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertUnary(name, argument_types);
@ -83,7 +84,7 @@ AggregateFunctionPtr createAggregateFunctionGroupArray(const std::string & name,
return createAggregateFunctionGroupArrayImpl<GroupArrayTrait<true, Sampler::NONE>>(argument_types[0], max_elems);
}
AggregateFunctionPtr createAggregateFunctionGroupArraySample(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionGroupArraySample(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertUnary(name, argument_types);

2
src/AggregateFunctions/AggregateFunctionGroupArray.h
View File

@ -26,6 +26,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int TOO_LARGE_ARRAY_SIZE;

4
src/AggregateFunctions/AggregateFunctionGroupArrayInsertAt.cpp
View File

@ -7,6 +7,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
@ -15,7 +17,7 @@ namespace ErrorCodes
namespace
{
AggregateFunctionPtr createAggregateFunctionGroupArrayInsertAt(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionGroupArrayInsertAt(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertBinary(name, argument_types);

1
src/AggregateFunctions/AggregateFunctionGroupArrayInsertAt.h
View File

@ -20,6 +20,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

3
src/AggregateFunctions/AggregateFunctionGroupArrayMoving.cpp
View File

@ -10,6 +10,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -57,7 +58,7 @@ inline AggregateFunctionPtr createAggregateFunctionMovingImpl(const std::string
}
template <template <typename, typename> class Function>
AggregateFunctionPtr createAggregateFunctionMoving(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionMoving(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertUnary(name, argument_types);

1
src/AggregateFunctions/AggregateFunctionGroupArrayMoving.h
View File

@ -22,6 +22,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

6
src/AggregateFunctions/AggregateFunctionGroupBitmap.cpp
View File

@ -9,6 +9,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
@ -33,7 +35,7 @@ namespace
}
template <template <typename> class Data>
AggregateFunctionPtr createAggregateFunctionBitmap(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionBitmap(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertUnary(name, argument_types);
@ -57,7 +59,7 @@ namespace
// Additional aggregate functions to manipulate bitmaps.
template <template <typename, typename> class AggregateFunctionTemplate>
AggregateFunctionPtr
createAggregateFunctionBitmapL2(const std::string & name, const DataTypes & argument_types, const Array & parameters)
createAggregateFunctionBitmapL2(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertUnary(name, argument_types);

1
src/AggregateFunctions/AggregateFunctionGroupBitmap.h
View File

@ -11,6 +11,7 @@
namespace DB
{
struct Settings;
/// Counts bitmap operation on numbers.
template <typename T, typename Data>
class AggregateFunctionBitmap final : public IAggregateFunctionDataHelper<Data, AggregateFunctionBitmap<T, Data>>

1
src/AggregateFunctions/AggregateFunctionGroupBitmapData.h
View File

@ -16,6 +16,7 @@
namespace DB
{
struct Settings;
enum BitmapKind
{

3
src/AggregateFunctions/AggregateFunctionGroupUniqArray.cpp
View File

@ -9,6 +9,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -72,7 +73,7 @@ inline AggregateFunctionPtr createAggregateFunctionGroupUniqArrayImpl(const std:
}
AggregateFunctionPtr createAggregateFunctionGroupUniqArray(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionGroupUniqArray(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertUnary(name, argument_types);

1
src/AggregateFunctions/AggregateFunctionGroupUniqArray.h
View File

@ -23,6 +23,7 @@
namespace DB
{
struct Settings;
template <typename T>

3
src/AggregateFunctions/AggregateFunctionHistogram.cpp
View File

@ -8,6 +8,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -22,7 +23,7 @@ namespace ErrorCodes
namespace
{
AggregateFunctionPtr createAggregateFunctionHistogram(const std::string & name, const DataTypes & arguments, const Array & params)
AggregateFunctionPtr createAggregateFunctionHistogram(const std::string & name, const DataTypes & arguments, const Array & params, const Settings *)
{
if (params.size() != 1)
throw Exception("Function " + name + " requires single parameter: bins count", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH);

1
src/AggregateFunctions/AggregateFunctionHistogram.h
View File

@ -26,6 +26,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionIf.cpp
View File

@ -6,6 +6,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionIf.h
View File

@ -8,6 +8,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

4
src/AggregateFunctions/AggregateFunctionMLMethod.cpp
View File

@ -16,6 +16,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int BAD_ARGUMENTS;
@ -29,7 +31,7 @@ namespace
using FuncLogisticRegression = AggregateFunctionMLMethod<LinearModelData, NameLogisticRegression>;
template <class Method>
AggregateFunctionPtr
createAggregateFunctionMLMethod(const std::string & name, const DataTypes & argument_types, const Array & parameters)
createAggregateFunctionMLMethod(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
if (parameters.size() > 4)
throw Exception(

2
src/AggregateFunctions/AggregateFunctionMLMethod.h
View File

@ -11,6 +11,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int LOGICAL_ERROR;

3
src/AggregateFunctions/AggregateFunctionMannWhitney.cpp
View File

@ -12,11 +12,12 @@ extern const int NOT_IMPLEMENTED;
namespace DB
{
struct Settings;
namespace
{
AggregateFunctionPtr createAggregateFunctionMannWhitneyUTest(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionMannWhitneyUTest(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertBinary(name, argument_types);

1
src/AggregateFunctions/AggregateFunctionMannWhitney.h
View File

@ -23,6 +23,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

9
src/AggregateFunctions/AggregateFunctionMax.cpp
View File

@ -6,18 +6,19 @@
namespace DB
{
struct Settings;
namespace
{
AggregateFunctionPtr createAggregateFunctionMax(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionMax(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings * settings)
{
return AggregateFunctionPtr(createAggregateFunctionSingleValue<AggregateFunctionsSingleValue, AggregateFunctionMaxData>(name, argument_types, parameters));
return AggregateFunctionPtr(createAggregateFunctionSingleValue<AggregateFunctionsSingleValue, AggregateFunctionMaxData>(name, argument_types, parameters, settings));
}
AggregateFunctionPtr createAggregateFunctionArgMax(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionArgMax(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings * settings)
{
return AggregateFunctionPtr(createAggregateFunctionArgMinMax<AggregateFunctionMaxData>(name, argument_types, parameters));
return AggregateFunctionPtr(createAggregateFunctionArgMinMax<AggregateFunctionMaxData>(name, argument_types, parameters, settings));
}
}

16
src/AggregateFunctions/AggregateFunctionMaxIntersections.cpp
View File

@ -7,6 +7,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
@ -32,11 +34,17 @@ namespace
void registerAggregateFunctionsMaxIntersections(AggregateFunctionFactory & factory)
{
factory.registerFunction("maxIntersections", [](const std::string & name, const DataTypes & argument_types, const Array & parameters)
{ return createAggregateFunctionMaxIntersections(AggregateFunctionIntersectionsKind::Count, name, argument_types, parameters); });
factory.registerFunction("maxIntersections",
[](const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
return createAggregateFunctionMaxIntersections(AggregateFunctionIntersectionsKind::Count, name, argument_types, parameters);
});
factory.registerFunction("maxIntersectionsPosition", [](const std::string & name, const DataTypes & argument_types, const Array & parameters)
{ return createAggregateFunctionMaxIntersections(AggregateFunctionIntersectionsKind::Position, name, argument_types, parameters); });
factory.registerFunction("maxIntersectionsPosition",
[](const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
return createAggregateFunctionMaxIntersections(AggregateFunctionIntersectionsKind::Position, name, argument_types, parameters);
});
}
}

1
src/AggregateFunctions/AggregateFunctionMaxIntersections.h
View File

@ -19,6 +19,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionMerge.cpp
View File

@ -6,6 +6,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

2
src/AggregateFunctions/AggregateFunctionMerge.h
View File

@ -9,6 +9,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;

9
src/AggregateFunctions/AggregateFunctionMin.cpp
View File

@ -6,18 +6,19 @@
namespace DB
{
struct Settings;
namespace
{
AggregateFunctionPtr createAggregateFunctionMin(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionMin(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings * settings)
{
return AggregateFunctionPtr(createAggregateFunctionSingleValue<AggregateFunctionsSingleValue, AggregateFunctionMinData>(name, argument_types, parameters));
return AggregateFunctionPtr(createAggregateFunctionSingleValue<AggregateFunctionsSingleValue, AggregateFunctionMinData>(name, argument_types, parameters, settings));
}
AggregateFunctionPtr createAggregateFunctionArgMin(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionArgMin(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings * settings)
{
return AggregateFunctionPtr(createAggregateFunctionArgMinMax<AggregateFunctionMinData>(name, argument_types, parameters));
return AggregateFunctionPtr(createAggregateFunctionArgMinMax<AggregateFunctionMinData>(name, argument_types, parameters, settings));
}
}

2
src/AggregateFunctions/AggregateFunctionMinMaxAny.h
View File

@ -15,6 +15,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;

1
src/AggregateFunctions/AggregateFunctionNothing.h
View File

@ -8,6 +8,7 @@
namespace DB
{
struct Settings;
/** Aggregate function that takes arbitrary number of arbitrary arguments and does nothing.

1
src/AggregateFunctions/AggregateFunctionNull.cpp
View File

@ -9,6 +9,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionNull.h
View File

@ -12,6 +12,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionOrFill.cpp
View File

@ -6,6 +6,7 @@
namespace DB
{
struct Settings;
namespace
{

1
src/AggregateFunctions/AggregateFunctionOrFill.h
View File

@ -11,6 +11,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

3
src/AggregateFunctions/AggregateFunctionQuantile.cpp
View File

@ -8,6 +8,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -83,7 +84,7 @@ static constexpr bool supportBigInt()
}
template <template <typename, bool> class Function>
AggregateFunctionPtr createAggregateFunctionQuantile(const std::string & name, const DataTypes & argument_types, const Array & params)
AggregateFunctionPtr createAggregateFunctionQuantile(const std::string & name, const DataTypes & argument_types, const Array & params, const Settings *)
{
/// Second argument type check doesn't depend on the type of the first one.
Function<void, true>::assertSecondArg(argument_types);

1
src/AggregateFunctions/AggregateFunctionQuantile.h
View File

@ -29,6 +29,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

3
src/AggregateFunctions/AggregateFunctionRankCorrelation.cpp
View File

@ -12,11 +12,12 @@ extern const int NOT_IMPLEMENTED;
namespace DB
{
struct Settings;
namespace
{
AggregateFunctionPtr createAggregateFunctionRankCorrelation(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionRankCorrelation(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertBinary(name, argument_types);
assertNoParameters(name, parameters);

1
src/AggregateFunctions/AggregateFunctionRankCorrelation.h
View File

@ -18,6 +18,7 @@
namespace DB
{
struct Settings;
struct RankCorrelationData : public StatisticalSample<Float64, Float64>

1
src/AggregateFunctions/AggregateFunctionResample.cpp
View File

@ -6,6 +6,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionResample.h
View File

@ -9,6 +9,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

4
src/AggregateFunctions/AggregateFunctionRetention.cpp
View File

@ -7,6 +7,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
@ -15,7 +17,7 @@ namespace ErrorCodes
namespace
{
AggregateFunctionPtr createAggregateFunctionRetention(const std::string & name, const DataTypes & arguments, const Array & params)
AggregateFunctionPtr createAggregateFunctionRetention(const std::string & name, const DataTypes & arguments, const Array & params, const Settings *)
{
assertNoParameters(name, params);

2
src/AggregateFunctions/AggregateFunctionRetention.h
View File

@ -17,6 +17,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;

3
src/AggregateFunctions/AggregateFunctionSequenceMatch.cpp
View File

@ -10,6 +10,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -23,7 +24,7 @@ namespace
{
template <template <typename, typename> class AggregateFunction, template <typename> class Data>
AggregateFunctionPtr createAggregateFunctionSequenceBase(const std::string & name, const DataTypes & argument_types, const Array & params)
AggregateFunctionPtr createAggregateFunctionSequenceBase(const std::string & name, const DataTypes & argument_types, const Array & params, const Settings *)
{
if (params.size() != 1)
throw Exception{"Aggregate function " + name + " requires exactly one parameter.",

1
src/AggregateFunctions/AggregateFunctionSequenceMatch.h
View File

@ -15,6 +15,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

8
src/AggregateFunctions/AggregateFunctionSimpleLinearRegression.cpp
View File

@ -4,10 +4,12 @@
#include <AggregateFunctions/FactoryHelpers.h>
#include <Core/TypeListNumber.h>
#include "registerAggregateFunctions.h"
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
@ -19,8 +21,8 @@ namespace
AggregateFunctionPtr createAggregateFunctionSimpleLinearRegression(
const String & name,
const DataTypes & arguments,
const Array & params
)
const Array & params,
const Settings *)
{
assertNoParameters(name, params);
assertBinary(name, arguments);

1
src/AggregateFunctions/AggregateFunctionSimpleLinearRegression.h
View File

@ -13,6 +13,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionSimpleState.cpp
View File

@ -3,6 +3,7 @@
namespace DB
{
struct Settings;
namespace
{
class AggregateFunctionCombinatorSimpleState final : public IAggregateFunctionCombinator

1
src/AggregateFunctions/AggregateFunctionSimpleState.h
View File

@ -7,6 +7,7 @@
namespace DB
{
struct Settings;
/** Not an aggregate function, but an adapter of aggregate functions.
* Aggregate functions with the `SimpleState` suffix is almost identical to the corresponding ones,

1
src/AggregateFunctions/AggregateFunctionState.cpp
View File

@ -7,6 +7,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionState.h
View File

@ -8,6 +8,7 @@
namespace DB
{
struct Settings;
/** Not an aggregate function, but an adapter of aggregate functions,

5
src/AggregateFunctions/AggregateFunctionStatistics.cpp
View File

@ -7,6 +7,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -17,7 +18,7 @@ namespace
{
template <template <typename> class FunctionTemplate>
AggregateFunctionPtr createAggregateFunctionStatisticsUnary(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionStatisticsUnary(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertUnary(name, argument_types);
@ -31,7 +32,7 @@ AggregateFunctionPtr createAggregateFunctionStatisticsUnary(const std::string &
}
template <template <typename, typename> class FunctionTemplate>
AggregateFunctionPtr createAggregateFunctionStatisticsBinary(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionStatisticsBinary(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertBinary(name, argument_types);

1
src/AggregateFunctions/AggregateFunctionStatistics.h
View File

@ -12,6 +12,7 @@
namespace DB
{
struct Settings;
namespace detail
{

5
src/AggregateFunctions/AggregateFunctionStatisticsSimple.cpp
View File

@ -7,6 +7,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -17,7 +18,7 @@ namespace
{
template <template <typename> class FunctionTemplate>
AggregateFunctionPtr createAggregateFunctionStatisticsUnary(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionStatisticsUnary(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertUnary(name, argument_types);
@ -36,7 +37,7 @@ AggregateFunctionPtr createAggregateFunctionStatisticsUnary(const std::string &
}
template <template <typename, typename> class FunctionTemplate>
AggregateFunctionPtr createAggregateFunctionStatisticsBinary(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionStatisticsBinary(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertBinary(name, argument_types);

1
src/AggregateFunctions/AggregateFunctionStatisticsSimple.h
View File

@ -30,6 +30,7 @@
namespace DB
{
struct Settings;
enum class StatisticsFunctionKind
{

3
src/AggregateFunctions/AggregateFunctionStudentTTest.cpp
View File

@ -14,6 +14,7 @@ namespace ErrorCodes
namespace DB
{
struct Settings;
namespace
{
@ -56,7 +57,7 @@ struct StudentTTestData : public TTestMoments<Float64>
}
};
AggregateFunctionPtr createAggregateFunctionStudentTTest(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionStudentTTest(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertBinary(name, argument_types);
assertNoParameters(name, parameters);

4
src/AggregateFunctions/AggregateFunctionSum.cpp
View File

@ -7,6 +7,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
@ -50,7 +52,7 @@ template <typename T> using AggregateFunctionSumKahan =
template <template <typename> class Function>
AggregateFunctionPtr createAggregateFunctionSum(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionSum(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertUnary(name, argument_types);

1
src/AggregateFunctions/AggregateFunctionSum.h
View File

@ -15,6 +15,7 @@
namespace DB
{
struct Settings;
/// Uses addOverflow method (if available) to avoid UB for sumWithOverflow()
///

6
src/AggregateFunctions/AggregateFunctionSumCount.cpp
View File

@ -6,6 +6,9 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
@ -19,7 +22,8 @@ bool allowType(const DataTypePtr& type) noexcept
return t.isInt() || t.isUInt() || t.isFloat() || t.isDecimal();
}
AggregateFunctionPtr createAggregateFunctionSumCount(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr
createAggregateFunctionSumCount(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertNoParameters(name, parameters);
assertUnary(name, argument_types);

4
src/AggregateFunctions/AggregateFunctionSumMap.cpp
View File

@ -9,6 +9,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
@ -75,7 +77,7 @@ auto parseArguments(const std::string & name, const DataTypes & arguments)
// function template that allows to choose the aggregate function variant that
// accepts either normal arguments or tuple argument.
template<template <bool tuple_argument> typename MappedFunction>
AggregateFunctionPtr createAggregateFunctionMap(const std::string & name, const DataTypes & arguments, const Array & params)
AggregateFunctionPtr createAggregateFunctionMap(const std::string & name, const DataTypes & arguments, const Array & params, const Settings *)
{
auto [keys_type, values_types, tuple_argument] = parseArguments(name, arguments);

1
src/AggregateFunctions/AggregateFunctionSumMap.h
View File

@ -21,6 +21,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/AggregateFunctionTTest.h
View File

@ -23,6 +23,7 @@ extern "C"
namespace DB
{
struct Settings;
class ReadBuffer;
class WriteBuffer;

3
src/AggregateFunctions/AggregateFunctionTopK.cpp
View File

@ -12,6 +12,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -58,7 +59,7 @@ static IAggregateFunction * createWithExtraTypes(const DataTypePtr & argument_ty
template <bool is_weighted>
AggregateFunctionPtr createAggregateFunctionTopK(const std::string & name, const DataTypes & argument_types, const Array & params)
AggregateFunctionPtr createAggregateFunctionTopK(const std::string & name, const DataTypes & argument_types, const Array & params, const Settings *)
{
if (!is_weighted)
{

1
src/AggregateFunctions/AggregateFunctionTopK.h
View File

@ -17,6 +17,7 @@
namespace DB
{
struct Settings;
template <typename T>

5
src/AggregateFunctions/AggregateFunctionUniq.cpp
View File

@ -14,6 +14,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -29,7 +30,7 @@ namespace
* It differs, for example, in that it uses a trivial hash function, since `uniq` of many arguments first hashes them out itself.
*/
template <typename Data, typename DataForVariadic>
AggregateFunctionPtr createAggregateFunctionUniq(const std::string & name, const DataTypes & argument_types, const Array & params)
AggregateFunctionPtr createAggregateFunctionUniq(const std::string & name, const DataTypes & argument_types, const Array & params, const Settings *)
{
assertNoParameters(name, params);
@ -73,7 +74,7 @@ AggregateFunctionPtr createAggregateFunctionUniq(const std::string & name, const
}
template <bool is_exact, template <typename> class Data, typename DataForVariadic>
AggregateFunctionPtr createAggregateFunctionUniq(const std::string & name, const DataTypes & argument_types, const Array & params)
AggregateFunctionPtr createAggregateFunctionUniq(const std::string & name, const DataTypes & argument_types, const Array & params, const Settings *)
{
assertNoParameters(name, params);

1
src/AggregateFunctions/AggregateFunctionUniq.h
View File

@ -28,6 +28,7 @@
namespace DB
{
struct Settings;
/// uniq

14
src/AggregateFunctions/AggregateFunctionUniqCombined.cpp
View File

@ -13,6 +13,8 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
@ -133,8 +135,16 @@ namespace
void registerAggregateFunctionUniqCombined(AggregateFunctionFactory & factory)
{
using namespace std::placeholders;
factory.registerFunction("uniqCombined", std::bind(createAggregateFunctionUniqCombined, false, _1, _2, _3)); // NOLINT
factory.registerFunction("uniqCombined64", std::bind(createAggregateFunctionUniqCombined, true, _1, _2, _3)); // NOLINT
factory.registerFunction("uniqCombined",
[](const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
return createAggregateFunctionUniqCombined(false, name, argument_types, parameters);
});
factory.registerFunction("uniqCombined64",
[](const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
return createAggregateFunctionUniqCombined(true, name, argument_types, parameters);
});
}
}

1
src/AggregateFunctions/AggregateFunctionUniqCombined.h
View File

@ -21,6 +21,7 @@
namespace DB
{
struct Settings;
namespace detail
{
/** Hash function for uniqCombined/uniqCombined64 (based on Ret).

3
src/AggregateFunctions/AggregateFunctionUniqUpTo.cpp
View File

@ -10,6 +10,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
@ -24,7 +25,7 @@ namespace
constexpr UInt8 uniq_upto_max_threshold = 100;
AggregateFunctionPtr createAggregateFunctionUniqUpTo(const std::string & name, const DataTypes & argument_types, const Array & params)
AggregateFunctionPtr createAggregateFunctionUniqUpTo(const std::string & name, const DataTypes & argument_types, const Array & params, const Settings *)
{
UInt8 threshold = 5; /// default value

1
src/AggregateFunctions/AggregateFunctionUniqUpTo.h
View File

@ -25,6 +25,7 @@
namespace DB
{
struct Settings;
/** Counts the number of unique values up to no more than specified in the parameter.

3
src/AggregateFunctions/AggregateFunctionWelchTTest.cpp
View File

@ -14,6 +14,7 @@ namespace ErrorCodes
namespace DB
{
struct Settings;
namespace
{
@ -53,7 +54,7 @@ struct WelchTTestData : public TTestMoments<Float64>
}
};
AggregateFunctionPtr createAggregateFunctionWelchTTest(const std::string & name, const DataTypes & argument_types, const Array & parameters)
AggregateFunctionPtr createAggregateFunctionWelchTTest(const std::string & name, const DataTypes & argument_types, const Array & parameters, const Settings *)
{
assertBinary(name, argument_types);
assertNoParameters(name, parameters);

16
src/AggregateFunctions/AggregateFunctionWindowFunnel.cpp
View File

@ -1,7 +1,8 @@
#include <AggregateFunctions/AggregateFunctionFactory.h>
#include <AggregateFunctions/AggregateFunctionWindowFunnel.h>
#include <AggregateFunctions/Helpers.h>
#include <AggregateFunctions/FactoryHelpers.h>
#include <AggregateFunctions/Helpers.h>
#include <Core/Settings.h>
#include <DataTypes/DataTypeDate.h>
#include <DataTypes/DataTypeDateTime.h>
@ -10,18 +11,29 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
extern const int UNKNOWN_AGGREGATE_FUNCTION;
}
namespace
{
template <template <typename> class Data>
AggregateFunctionPtr createAggregateFunctionWindowFunnel(const std::string & name, const DataTypes & arguments, const Array & params)
AggregateFunctionPtr
createAggregateFunctionWindowFunnel(const std::string & name, const DataTypes & arguments, const Array & params, const Settings * settings)
{
if (settings == nullptr || !settings->allow_experimental_funnel_functions)
{
throw Exception(
"Aggregate function " + name + " is experimental. Set `allow_experimental_funnel_functions` setting to enable it",
ErrorCodes::UNKNOWN_AGGREGATE_FUNCTION);
}
if (params.empty())
throw Exception{"Aggregate function " + name + " requires at least one parameter: <window>, [option, [option, ...]]", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH};

1
src/AggregateFunctions/AggregateFunctionWindowFunnel.h
View File

@ -13,6 +13,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

1
src/AggregateFunctions/FactoryHelpers.h
View File

@ -7,6 +7,7 @@
namespace DB
{
struct Settings;
namespace ErrorCodes
{

Some files were not shown because too many files have changed in this diff Show More