mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-24 18:50:49 +00:00
54 lines
1.3 KiB
C++
54 lines
1.3 KiB
C++
#pragma once
|
|
|
|
#include <Poco/Logger.h>
|
|
#include <DataStreams/IBlockInputStream.h>
|
|
#include <Interpreters/SubqueryForSet.h>
|
|
|
|
|
|
namespace Poco { class Logger; }
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/** Returns the data from the stream of blocks without changes, but
|
|
* in the `readPrefix` function or before reading the first block
|
|
* initializes all the passed sets.
|
|
*/
|
|
class CreatingSetsBlockInputStream : public IBlockInputStream
|
|
{
|
|
public:
|
|
CreatingSetsBlockInputStream(
|
|
const BlockInputStreamPtr & input,
|
|
const SubqueriesForSets & subqueries_for_sets_,
|
|
const Context & context_);
|
|
|
|
String getName() const override { return "CreatingSets"; }
|
|
|
|
Block getHeader() const override { return children.back()->getHeader(); }
|
|
|
|
/// Takes `totals` only from the main source, not from subquery sources.
|
|
Block getTotals() override;
|
|
|
|
protected:
|
|
Block readImpl() override;
|
|
void readPrefixImpl() override;
|
|
|
|
private:
|
|
SubqueriesForSets subqueries_for_sets;
|
|
const Context & context;
|
|
bool created = false;
|
|
|
|
SizeLimits network_transfer_limits;
|
|
|
|
size_t rows_to_transfer = 0;
|
|
size_t bytes_to_transfer = 0;
|
|
|
|
using Logger = Poco::Logger;
|
|
Logger * log = &Logger::get("CreatingSetsBlockInputStream");
|
|
|
|
void createAll();
|
|
void createOne(SubqueryForSet & subquery);
|
|
};
|
|
|
|
}
|