mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-22 15:42:02 +00:00
52 lines
1.1 KiB
C++
52 lines
1.1 KiB
C++
#pragma once
|
||
|
||
#include <DB/DataStreams/IProfilingBlockInputStream.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
/** Инициализировать другой источник при первом вызове read, и затем использовать его.
|
||
* Это нужно, например, для чтения из таблицы, которая будет заполнена
|
||
* после создания объекта LazyBlockInputStream, но до первого вызова read.
|
||
*/
|
||
class LazyBlockInputStream : public IProfilingBlockInputStream
|
||
{
|
||
public:
|
||
using Generator = std::function<BlockInputStreamPtr()>;
|
||
|
||
LazyBlockInputStream(Generator generator_)
|
||
: generator(generator_) {}
|
||
|
||
String getName() const override { return "LazyBlockInputStream"; }
|
||
|
||
String getID() const override
|
||
{
|
||
std::stringstream res;
|
||
res << "Lazy(" << this << ")";
|
||
return res.str();
|
||
}
|
||
|
||
protected:
|
||
Block readImpl() override
|
||
{
|
||
if (!input)
|
||
{
|
||
input = generator();
|
||
|
||
if (!input)
|
||
return Block();
|
||
|
||
children.push_back(input);
|
||
}
|
||
|
||
return input->read();
|
||
}
|
||
|
||
private:
|
||
Generator generator;
|
||
BlockInputStreamPtr input;
|
||
};
|
||
|
||
}
|