mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-24 16:42:05 +00:00
82 lines
2.6 KiB
C++
82 lines
2.6 KiB
C++
#pragma once
|
|
|
|
#include <DB/AggregateFunctions/AggregateFunctionsMinMaxAny.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
|
|
/// Возможные значения параметров шаблонов см. в AggregateFunctionsMinMaxAny.h
|
|
template <typename ResultData, typename ValueData>
|
|
struct AggregateFunctionsArgMinMaxData
|
|
{
|
|
using ResultData_t = ResultData;
|
|
using ValueData_t = ValueData;
|
|
|
|
ResultData result; // аргумент, при котором достигается минимальное/максимальное значение value.
|
|
ValueData value; // значение, для которого считается минимум/максимум.
|
|
};
|
|
|
|
/// Возвращает первое попавшееся значение arg для минимального/максимального value. Пример: argMax(arg, value).
|
|
template <typename Data>
|
|
class AggregateFunctionsArgMinMax final : public IAggregateFunctionHelper<Data>
|
|
{
|
|
private:
|
|
DataTypePtr type_res;
|
|
DataTypePtr type_val;
|
|
|
|
public:
|
|
String getName() const override { return (0 == strcmp(Data::ValueData_t::name(), "min")) ? "argMin" : "argMax"; }
|
|
|
|
DataTypePtr getReturnType() const override
|
|
{
|
|
return type_res;
|
|
}
|
|
|
|
void setArguments(const DataTypes & arguments) override
|
|
{
|
|
if (arguments.size() != 2)
|
|
throw Exception("Aggregate function " + getName() + " requires exactly two arguments.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH);
|
|
|
|
type_res = arguments[0];
|
|
type_val = arguments[1];
|
|
}
|
|
|
|
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num) const override
|
|
{
|
|
if (this->data(place).value.changeIfBetter(*columns[1], row_num))
|
|
this->data(place).result.change(*columns[0], row_num);
|
|
}
|
|
|
|
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs) const override
|
|
{
|
|
if (this->data(place).value.changeIfBetter(this->data(rhs).value))
|
|
this->data(place).result.change(this->data(rhs).result);
|
|
}
|
|
|
|
void serialize(ConstAggregateDataPtr place, WriteBuffer & buf) const override
|
|
{
|
|
this->data(place).result.write(buf, *type_res.get());
|
|
this->data(place).value.write(buf, *type_val.get());
|
|
}
|
|
|
|
void deserializeMerge(AggregateDataPtr place, ReadBuffer & buf) const override
|
|
{
|
|
Data rhs; /// Для строчек не очень оптимально, так как может делаться одна лишняя аллокация.
|
|
|
|
rhs.result.read(buf, *type_res.get());
|
|
rhs.value.read(buf, *type_val.get());
|
|
|
|
if (this->data(place).value.changeIfBetter(rhs.value))
|
|
this->data(place).result.change(rhs.result);
|
|
}
|
|
|
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
|
{
|
|
this->data(place).result.insertResultInto(to);
|
|
}
|
|
};
|
|
|
|
}
|