ClickHouse/dbms/include/DB/DataStreams/MergingAggregatedBlockInputStream.h
2013-05-27 14:02:55 +00:00

50 lines
1.5 KiB
C++

#pragma once
#include <DB/Interpreters/Aggregator.h>
#include <DB/DataStreams/IProfilingBlockInputStream.h>
namespace DB
{
using Poco::SharedPtr;
/** Доагрегирует поток блоков, в котором каждый блок уже агрегирован.
* Агрегатные функции в блоках не должны быть финализированы, чтобы их состояния можно было объединить.
* Сам тоже не финализирует агрегатные функции.
*/
class MergingAggregatedBlockInputStream : public IProfilingBlockInputStream
{
public:
MergingAggregatedBlockInputStream(BlockInputStreamPtr input_, const ColumnNumbers & keys_, const AggregateDescriptions & aggregates_, bool with_totals_)
: aggregator(new Aggregator(keys_, aggregates_, with_totals_)), has_been_read(false)
{
children.push_back(input_);
}
MergingAggregatedBlockInputStream(BlockInputStreamPtr input_, const Names & keys_names_, const AggregateDescriptions & aggregates_, bool with_totals_)
: aggregator(new Aggregator(keys_names_, aggregates_, with_totals_)), has_been_read(false)
{
children.push_back(input_);
}
String getName() const { return "MergingAggregatedBlockInputStream"; }
String getID() const
{
std::stringstream res;
res << "MergingAggregated(" << children.back()->getID() << ", " << aggregator->getID() << ")";
return res.str();
}
protected:
Block readImpl();
private:
SharedPtr<Aggregator> aggregator;
bool has_been_read;
};
}