ClickHouse/dbms/Processors/IAccumulatingTransform.h

43 lines
977 B
C++
Raw Normal View History

#pragma once
#include <Processors/IProcessor.h>
namespace DB
{
/** Has one input and one output.
* Pulls all blocks from input, and only then produce output.
* Examples: ORDER BY, GROUP BY.
*/
class IAccumulatingTransform : public IProcessor
{
protected:
InputPort & input;
OutputPort & output;
Chunk current_input_chunk;
Chunk current_output_chunk;
2019-02-07 18:51:53 +00:00
bool has_input = false;
bool finished_input = false;
bool finished_generate = false;
virtual void consume(Chunk chunk) = 0;
virtual Chunk generate() = 0;
2019-02-26 18:40:08 +00:00
/// This method can be called once per consume call. In case if some chunks are ready.
void setReadyChunk(Chunk chunk);
2019-03-04 14:56:09 +00:00
void finishConsume() { finished_input = true; }
2019-02-26 18:40:08 +00:00
public:
IAccumulatingTransform(Block input_header, Block output_header);
Status prepare() override;
void work() override;
InputPort & getInputPort() { return input; }
OutputPort & getOutputPort() { return output; }
};
}