mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 01:12:12 +00:00
43 lines
923 B
C++
43 lines
923 B
C++
#pragma once
|
|
|
|
#include <DataStreams/IBlockInputStream.h>
|
|
#include <Interpreters/Aggregator.h>
|
|
#include <Core/ColumnNumbers.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class ExpressionActions;
|
|
|
|
|
|
/** Takes blocks after grouping, with non-finalized aggregate functions.
|
|
* Calculates all subsets of columns and aggreagetes over them.
|
|
*/
|
|
class CubeBlockInputStream : public IBlockInputStream
|
|
{
|
|
private:
|
|
using ExpressionActionsPtr = std::shared_ptr<ExpressionActions>;
|
|
using AggregateColumns = std::vector<ColumnRawPtrs>;
|
|
public:
|
|
CubeBlockInputStream(
|
|
const BlockInputStreamPtr & input_, const Aggregator::Params & params_);
|
|
|
|
String getName() const override { return "Cube"; }
|
|
|
|
Block getHeader() const override;
|
|
|
|
protected:
|
|
Block readImpl() override;
|
|
|
|
private:
|
|
Aggregator aggregator;
|
|
ColumnNumbers keys;
|
|
UInt32 mask = 0;
|
|
Block source_block;
|
|
Block zero_block;
|
|
bool is_data_read = false;
|
|
};
|
|
|
|
}
|