ClickHouse/dbms/include/DB/DataStreams/OneBlockInputStream.h

46 lines
745 B
C
Raw Normal View History

2012-05-08 11:19:00 +00:00
#pragma once
#include <Poco/SharedPtr.h>
#include <DB/DataStreams/IProfilingBlockInputStream.h>
namespace DB
{
using Poco::SharedPtr;
/** Поток блоков, из которого можно прочитать один блок.
*/
class OneBlockInputStream : public IProfilingBlockInputStream
{
public:
OneBlockInputStream(const Block & block_) : block(block_), has_been_read(false) {}
2012-05-08 11:19:00 +00:00
String getName() const override { return "OneBlockInputStream"; }
2012-10-20 02:10:47 +00:00
String getID() const override
{
std::stringstream res;
res << this;
return res.str();
}
2012-10-20 02:10:47 +00:00
protected:
Block readImpl() override
2012-05-08 11:19:00 +00:00
{
if (has_been_read)
return Block();
has_been_read = true;
return block;
}
private:
Block block;
bool has_been_read;
};
}