mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-01 14:10:50 +00:00
48 lines
779 B
C++
48 lines
779 B
C++
#include <Processors/ISource.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
ISource::ISource(Block header)
|
|
: IProcessor({}, {std::move(header)}), output(outputs.front())
|
|
{
|
|
}
|
|
|
|
ISource::Status ISource::prepare()
|
|
{
|
|
if (finished)
|
|
{
|
|
output.finish();
|
|
return Status::Finished;
|
|
}
|
|
|
|
/// Check can output.
|
|
if (output.isFinished())
|
|
return Status::Finished;
|
|
|
|
if (!output.canPush())
|
|
return Status::PortFull;
|
|
|
|
if (!has_input)
|
|
return Status::Ready;
|
|
|
|
output.push(std::move(current_block));
|
|
has_input = false;
|
|
|
|
/// Now, we pushed to output, and it must be full.
|
|
return Status::PortFull;
|
|
}
|
|
|
|
void ISource::work()
|
|
{
|
|
current_block = generate();
|
|
if (!current_block)
|
|
finished = true;
|
|
else
|
|
has_input = true;
|
|
}
|
|
|
|
}
|
|
|