ClickHouse/dbms/Processors/IAccumulatingTransform.h
Ivan 97f2a2213e
Move all folders inside /dbms one level up (#9974)
* Move some code outside dbms/src folder
* Fix paths
2020-04-02 02:51:21 +03:00

43 lines
977 B
C++

#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;
bool has_input = false;
bool finished_input = false;
bool finished_generate = false;
virtual void consume(Chunk chunk) = 0;
virtual Chunk generate() = 0;
/// This method can be called once per consume call. In case if some chunks are ready.
void setReadyChunk(Chunk chunk);
void finishConsume() { finished_input = true; }
public:
IAccumulatingTransform(Block input_header, Block output_header);
Status prepare() override;
void work() override;
InputPort & getInputPort() { return input; }
OutputPort & getOutputPort() { return output; }
};
}