#pragma once

#include <DB/IO/WriteHelpers.h>
#include <DB/IO/ReadHelpers.h>

#include <DB/AggregateFunctions/IUnaryAggregateFunction.h>


namespace DB
{


struct AggregateFunctionAnyData
{
	Field value;
};


/// Берёт первое попавшееся значение
class AggregateFunctionAny : public IUnaryAggregateFunction<AggregateFunctionAnyData>
{
private:
	DataTypePtr type;
	
public:
	String getName() const { return "any"; }
	String getTypeID() const { return "any"; }

	DataTypePtr getReturnType() const
	{
		return type;
	}

	void setArgument(const DataTypePtr & argument)
	{
		type = argument;
	}


	void addOne(AggregateDataPtr place, const Field & value_) const
	{
		Data & d = data(place);

		if (!d.value.isNull())
			return;
		d.value = value_;
	}

	void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs) const
	{
		Data & d = data(place);

		if (d.value.isNull())
			d.value = data(rhs).value;
	}

	void serialize(ConstAggregateDataPtr place, WriteBuffer & buf) const
	{
		type->serializeBinary(data(place).value, buf);
	}

	void deserializeMerge(AggregateDataPtr place, ReadBuffer & buf) const
	{
		Data & d = data(place);

		Field tmp;
		type->deserializeBinary(tmp, buf);

		if (d.value.isNull())
			d.value = tmp;
	}

	Field getResult(ConstAggregateDataPtr place) const
	{
		return data(place).value;
	}
};

}