2018-05-23 20:19:33 +00:00
|
|
|
#include <Processors/ISimpleTransform.h>
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
ISimpleTransform::ISimpleTransform(Block input_header, Block output_header)
|
|
|
|
: IProcessor({std::move(input_header)}, {std::move(output_header)}),
|
|
|
|
input(inputs.front()), output(outputs.front())
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ISimpleTransform::Status ISimpleTransform::prepare()
|
|
|
|
{
|
2019-02-07 18:51:53 +00:00
|
|
|
/// Check can output.
|
2018-05-23 20:19:33 +00:00
|
|
|
|
2019-02-07 18:51:53 +00:00
|
|
|
if (output.isFinished())
|
2018-05-23 20:19:33 +00:00
|
|
|
{
|
2019-02-07 18:51:53 +00:00
|
|
|
input.close();
|
|
|
|
return Status::Finished;
|
2018-05-23 20:19:33 +00:00
|
|
|
}
|
|
|
|
|
2019-02-07 18:51:53 +00:00
|
|
|
if (!output.canPush())
|
2018-05-23 20:19:33 +00:00
|
|
|
{
|
2019-02-07 18:51:53 +00:00
|
|
|
input.setNotNeeded();
|
|
|
|
return Status::PortFull;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Output if has data.
|
|
|
|
if (transformed)
|
|
|
|
{
|
|
|
|
output.push(std::move(current_block));
|
|
|
|
has_input = false;
|
2018-05-23 20:19:33 +00:00
|
|
|
transformed = false;
|
|
|
|
}
|
|
|
|
|
2019-02-07 18:51:53 +00:00
|
|
|
/// Check can input.
|
|
|
|
if (!has_input)
|
2018-05-24 02:52:21 +00:00
|
|
|
{
|
2019-02-07 18:51:53 +00:00
|
|
|
if (input.isFinished())
|
|
|
|
{
|
|
|
|
output.finish();
|
|
|
|
return Status::Finished;
|
|
|
|
}
|
|
|
|
|
|
|
|
input.setNeeded();
|
|
|
|
|
|
|
|
if (!input.hasData())
|
|
|
|
return Status::NeedData;
|
|
|
|
|
|
|
|
current_block = input.pull();
|
|
|
|
has_input = true;
|
2018-05-24 02:52:21 +00:00
|
|
|
}
|
2018-05-23 20:19:33 +00:00
|
|
|
|
2019-02-07 18:51:53 +00:00
|
|
|
/// Now transform.
|
|
|
|
return Status::Ready;
|
2018-05-23 20:19:33 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void ISimpleTransform::work()
|
|
|
|
{
|
|
|
|
transform(current_block);
|
|
|
|
transformed = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|