ClickHouse/dbms/IO/IReadableWriteBuffer.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

33 lines
693 B
C++

#pragma once
#include <memory>
#include <IO/ReadBuffer.h>
namespace DB
{
struct IReadableWriteBuffer
{
/// At the first time returns getReadBufferImpl(). Next calls return nullptr.
inline std::shared_ptr<ReadBuffer> tryGetReadBuffer()
{
if (!can_reread)
return nullptr;
can_reread = false;
return getReadBufferImpl();
}
virtual ~IReadableWriteBuffer() {}
protected:
/// Creates read buffer from current write buffer.
/// Returned buffer points to the first byte of original buffer.
/// Original stream becomes invalid.
virtual std::shared_ptr<ReadBuffer> getReadBufferImpl() = 0;
bool can_reread = true;
};
}