mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-11 02:50:50 +00:00
39 lines
641 B
C++
39 lines
641 B
C++
|
#pragma once
|
||
|
|
||
|
#include <queue>
|
||
|
#include <Processors/IAccumulatingTransform.h>
|
||
|
|
||
|
|
||
|
namespace DB
|
||
|
{
|
||
|
|
||
|
class QueueBuffer : public IAccumulatingTransform
|
||
|
{
|
||
|
private:
|
||
|
std::queue<Block> blocks;
|
||
|
public:
|
||
|
String getName() const override { return "QueueBuffer"; }
|
||
|
|
||
|
QueueBuffer(Block header)
|
||
|
: IAccumulatingTransform(header, header)
|
||
|
{
|
||
|
}
|
||
|
|
||
|
void consume(Block block) override
|
||
|
{
|
||
|
blocks.push(std::move(block));
|
||
|
}
|
||
|
|
||
|
Block generate() override
|
||
|
{
|
||
|
if (blocks.empty())
|
||
|
return {};
|
||
|
|
||
|
Block res = std::move(blocks.front());
|
||
|
blocks.pop();
|
||
|
return std::move(res);
|
||
|
}
|
||
|
};
|
||
|
|
||
|
}
|