mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-21 16:00:50 +00:00
34 lines
866 B
C++
34 lines
866 B
C++
#pragma once
|
|
|
|
#include <DB/IO/ReadBuffer.h>
|
|
#include <DB/DataStreams/IProfilingBlockInputStream.h>
|
|
#include <memory>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/** Provides reading from a Buffer, taking exclusive ownership over it's lifetime,
|
|
* simplifies usage of ReadBufferFromFile (no need to manage buffer lifetime) etc.
|
|
*/
|
|
class OwningBufferBlockInputStream : public IProfilingBlockInputStream
|
|
{
|
|
public:
|
|
OwningBufferBlockInputStream(const BlockInputStreamPtr & stream, std::unique_ptr<ReadBuffer> buffer)
|
|
: stream{stream}, buffer{std::move(buffer)}
|
|
{
|
|
children.push_back(stream);
|
|
}
|
|
|
|
private:
|
|
Block readImpl() override { return stream->read(); }
|
|
|
|
String getName() const override { return "OwningBufferBlockInputStream"; }
|
|
|
|
String getID() const override { return "OwningBuffer(" + stream->getID() + ")"; }
|
|
|
|
BlockInputStreamPtr stream;
|
|
std::unique_ptr<ReadBuffer> buffer;
|
|
};
|
|
|
|
}
|