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

68 lines
2.5 KiB
C
Raw Normal View History

2014-02-27 12:49:21 +00:00
#pragma once
#include <DB/Interpreters/Aggregator.h>
#include <DB/DataStreams/IProfilingBlockInputStream.h>
#include <DB/Interpreters/ExpressionActions.h>
namespace DB
{
using Poco::SharedPtr;
/** Принимает блоки после группировки, с нефиализированными агрегатными функциями.
* Вычисляет тотальные значения в соответствии с totals_mode.
* Если нужно, вычисляет выражение из HAVING и фильтрует им строки. Отдает финализированные и отфильтрованные блоки.
*/
class TotalsHavingBlockInputStream : public IProfilingBlockInputStream
{
public:
TotalsHavingBlockInputStream(BlockInputStreamPtr input_, const Names & keys_names_,
const AggregateDescriptions & aggregates_, bool overflow_row_, ExpressionActionsPtr expression_,
const std::string & filter_column_, TotalsMode totals_mode_, double auto_include_threshold_)
: overflow_row(overflow_row_),
2014-02-27 12:49:21 +00:00
expression(expression_), filter_column_name(filter_column_), totals_mode(totals_mode_),
auto_include_threshold(auto_include_threshold_)
2014-02-27 12:49:21 +00:00
{
children.push_back(input_);
}
String getName() const override { return "TotalsHavingBlockInputStream"; }
2014-02-27 12:49:21 +00:00
String getID() const override
2014-02-27 12:49:21 +00:00
{
std::stringstream res;
res << "TotalsHavingBlockInputStream(" << children.back()->getID()
2014-02-27 12:49:21 +00:00
<< "," << filter_column_name << ")";
return res.str();
}
const Block & getTotals() override;
2014-02-27 12:49:21 +00:00
protected:
Block readImpl() override;
2014-02-27 12:49:21 +00:00
private:
bool overflow_row;
ExpressionActionsPtr expression;
String filter_column_name;
TotalsMode totals_mode;
double auto_include_threshold;
size_t passed_keys = 0;
size_t total_keys = 0;
2014-02-27 12:49:21 +00:00
/** Здесь находятся значения, не прошедшие max_rows_to_group_by.
* Они прибавляются или не прибавляются к current_totals в зависимости от totals_mode.
*/
2014-02-27 12:49:21 +00:00
Block overflow_aggregates;
/// Здесь накапливаются тотальные значения. После окончания работы, они будут помещены в IProfilingBlockInputStream::totals.
Block current_totals;
2014-02-27 12:49:21 +00:00
/// Если filter == nullptr - прибавлять все строки. Иначе - только строки, проходящие фильтр (HAVING).
void addToTotals(Block & totals, Block & block, const IColumn::Filter * filter);
2014-02-27 12:49:21 +00:00
};
}