mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 10:22:10 +00:00
41 lines
1.1 KiB
C++
41 lines
1.1 KiB
C++
#pragma once
|
|
#include <Processors/IAccumulatingTransform.h>
|
|
#include <Interpreters/Aggregator.h>
|
|
#include <Processors/Transforms/AggregatingTransform.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/** A pre-aggregate stream of blocks in which each block is already aggregated.
|
|
* Aggregate functions in blocks should not be finalized so that their states can be merged.
|
|
*/
|
|
class MergingAggregatedTransform : public IAccumulatingTransform
|
|
{
|
|
public:
|
|
MergingAggregatedTransform(Block header, AggregatingTransformParamsPtr params, size_t max_threads);
|
|
String getName() const override { return "MergingAggregatedTransform"; }
|
|
|
|
protected:
|
|
void consume(Chunk chunk) override;
|
|
Chunk generate() override;
|
|
|
|
private:
|
|
AggregatingTransformParamsPtr params;
|
|
Logger * log = &Logger::get("MergingAggregatedTransform");
|
|
size_t max_threads;
|
|
|
|
AggregatedDataVariants data_variants;
|
|
Aggregator::BucketToBlocks bucket_to_blocks;
|
|
|
|
UInt64 total_input_rows = 0;
|
|
UInt64 total_input_blocks = 0;
|
|
|
|
BlocksList blocks;
|
|
BlocksList::iterator next_block;
|
|
|
|
bool consume_started = false;
|
|
bool generate_started = false;
|
|
};
|
|
|
|
}
|