ClickHouse/dbms/include/DB/AggregateFunctions/AggregateFunctionAny.h
2012-10-24 19:09:42 +00:00

78 lines
1.2 KiB
C++

#pragma once
#include <DB/IO/WriteHelpers.h>
#include <DB/IO/ReadHelpers.h>
#include <DB/AggregateFunctions/IUnaryAggregateFunction.h>
namespace DB
{
/// Берёт первое попавшееся значение
class AggregateFunctionAny : public IUnaryAggregateFunction
{
private:
bool got;
Field value;
DataTypePtr type;
public:
AggregateFunctionAny() : got(false) {}
String getName() const { return "any"; }
String getTypeID() const { return "any"; }
AggregateFunctionPlainPtr cloneEmpty() const
{
AggregateFunctionAny * res = new AggregateFunctionAny;
res->type = type;
return res;
}
DataTypePtr getReturnType() const
{
return type;
}
void setArgument(const DataTypePtr & argument)
{
type = argument;
}
void addOne(const Field & value_)
{
if (got)
return;
got = true;
value = value_;
}
void merge(const IAggregateFunction & rhs)
{
if (!got)
value = static_cast<const AggregateFunctionAny &>(rhs).value;
}
void serialize(WriteBuffer & buf) const
{
type->serializeBinary(value, buf);
}
void deserializeMerge(ReadBuffer & buf)
{
Field tmp;
type->deserializeBinary(tmp, buf);
if (!got)
value = tmp;
}
Field getResult() const
{
return value;
}
};
}