ClickHouse/dbms/include/DB/DataStreams/AggregatingBlockInputStream.h

55 lines
2.3 KiB
C
Raw Normal View History

2011-09-19 03:34:23 +00:00
#pragma once
#include <DB/Interpreters/Aggregator.h>
#include <DB/DataStreams/IProfilingBlockInputStream.h>
namespace DB
{
using Poco::SharedPtr;
/** Агрегирует поток блоков, используя заданные столбцы-ключи и агрегатные функции.
* Столбцы с агрегатными функциями добавляет в конец блока.
* Агрегатные функции не финализируются, то есть, не заменяются на своё значение, а содержат промежуточное состояние вычислений.
* Это необходимо, чтобы можно было продолжить агрегацию (например, объединяя потоки частично агрегированных данных).
*/
class AggregatingBlockInputStream : public IProfilingBlockInputStream
{
public:
AggregatingBlockInputStream(BlockInputStreamPtr input_, const ColumnNumbers & keys_, AggregateDescriptions & aggregates_,
bool with_totals_, bool separate_totals_, bool final_, size_t max_rows_to_group_by_, Limits::OverflowMode group_by_overflow_mode_)
: aggregator(new Aggregator(keys_, aggregates_, with_totals_, max_rows_to_group_by_, group_by_overflow_mode_)),
separate_totals(separate_totals_), final(final_), has_been_read(false)
2011-09-19 03:34:23 +00:00
{
2013-05-04 04:05:15 +00:00
children.push_back(input_);
2011-09-19 03:34:23 +00:00
}
2011-09-25 03:37:09 +00:00
/** keys берутся из GROUP BY части запроса
2011-09-24 20:32:41 +00:00
* Агрегатные функции ищутся везде в выражении.
* Столбцы, соответствующие keys и аргументам агрегатных функций, уже должны быть вычислены.
*/
AggregatingBlockInputStream(BlockInputStreamPtr input_, const Names & key_names, const AggregateDescriptions & aggregates,
bool with_totals_, bool separate_totals_, bool final_, size_t max_rows_to_group_by_, Limits::OverflowMode group_by_overflow_mode_);
2011-09-24 20:32:41 +00:00
2011-09-19 03:34:23 +00:00
String getName() const { return "AggregatingBlockInputStream"; }
String getID() const
{
std::stringstream res;
2013-05-04 05:20:07 +00:00
res << "Aggregating(" << children.back()->getID() << ", " << aggregator->getID() << ")";
return res.str();
}
2012-10-20 02:10:47 +00:00
protected:
Block readImpl();
2011-09-25 03:37:09 +00:00
SharedPtr<Aggregator> aggregator;
bool separate_totals;
bool final;
2011-09-19 03:34:23 +00:00
bool has_been_read;
};
}