mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-25 18:00:51 +00:00
72 lines
2.7 KiB
C++
72 lines
2.7 KiB
C++
|
#pragma once
|
|||
|
|
|||
|
#include <DB/Interpreters/Aggregator.h>
|
|||
|
#include <DB/IO/ReadBufferFromFile.h>
|
|||
|
#include <DB/IO/CompressedReadBuffer.h>
|
|||
|
#include <DB/DataStreams/IProfilingBlockInputStream.h>
|
|||
|
#include <DB/DataStreams/NativeBlockInputStream.h>
|
|||
|
#include <common/ClickHouseRevision.h>
|
|||
|
|
|||
|
|
|||
|
namespace DB
|
|||
|
{
|
|||
|
|
|||
|
using Poco::SharedPtr;
|
|||
|
|
|||
|
|
|||
|
/** Агрегирует поток блоков, используя заданные столбцы-ключи и агрегатные функции.
|
|||
|
* Столбцы с агрегатными функциями добавляет в конец блока.
|
|||
|
* Если final=false, агрегатные функции не финализируются, то есть, не заменяются на своё значение, а содержат промежуточное состояние вычислений.
|
|||
|
* Это необходимо, чтобы можно было продолжить агрегацию (например, объединяя потоки частично агрегированных данных).
|
|||
|
*/
|
|||
|
class AggregatingBlockInputStream : public IProfilingBlockInputStream
|
|||
|
{
|
|||
|
public:
|
|||
|
/** keys берутся из GROUP BY части запроса
|
|||
|
* Агрегатные функции ищутся везде в выражении.
|
|||
|
* Столбцы, соответствующие keys и аргументам агрегатных функций, уже должны быть вычислены.
|
|||
|
*/
|
|||
|
AggregatingBlockInputStream(BlockInputStreamPtr input_, const Aggregator::Params & params_, bool final_)
|
|||
|
: params(params_), aggregator(params), final(final_)
|
|||
|
{
|
|||
|
children.push_back(input_);
|
|||
|
}
|
|||
|
|
|||
|
String getName() const override { return "Aggregating"; }
|
|||
|
|
|||
|
String getID() const override
|
|||
|
{
|
|||
|
std::stringstream res;
|
|||
|
res << "Aggregating(" << children.back()->getID() << ", " << aggregator.getID() << ")";
|
|||
|
return res.str();
|
|||
|
}
|
|||
|
|
|||
|
protected:
|
|||
|
Block readImpl() override;
|
|||
|
|
|||
|
Aggregator::Params params;
|
|||
|
Aggregator aggregator;
|
|||
|
bool final;
|
|||
|
|
|||
|
bool executed = false;
|
|||
|
|
|||
|
/// Для чтения сброшенных во временный файл данных.
|
|||
|
struct TemporaryFileStream
|
|||
|
{
|
|||
|
ReadBufferFromFile file_in;
|
|||
|
CompressedReadBuffer compressed_in;
|
|||
|
BlockInputStreamPtr block_in;
|
|||
|
|
|||
|
TemporaryFileStream(const std::string & path)
|
|||
|
: file_in(path), compressed_in(file_in), block_in(new NativeBlockInputStream(compressed_in, ClickHouseRevision::get())) {}
|
|||
|
};
|
|||
|
std::vector<std::unique_ptr<TemporaryFileStream>> temporary_inputs;
|
|||
|
|
|||
|
/** Отсюда будем доставать готовые блоки после агрегации. */
|
|||
|
std::unique_ptr<IBlockInputStream> impl;
|
|||
|
|
|||
|
Logger * log = &Logger::get("AggregatingBlockInputStream");
|
|||
|
};
|
|||
|
|
|||
|
}
|