mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-04 05:22:17 +00:00
37 lines
1.0 KiB
C++
37 lines
1.0 KiB
C++
#pragma once
|
|
|
|
#include <base/types.h>
|
|
#include <IO/ReadBuffer.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/** Allows to read from another ReadBuffer no more than the specified number of bytes.
|
|
* Note that the nested ReadBuffer may read slightly more data internally to fill its buffer.
|
|
*/
|
|
class LimitReadBuffer : public ReadBuffer
|
|
{
|
|
public:
|
|
LimitReadBuffer(ReadBuffer & in_, UInt64 limit_, bool throw_exception_,
|
|
std::optional<size_t> exact_limit_, std::string exception_message_ = {});
|
|
LimitReadBuffer(std::unique_ptr<ReadBuffer> in_, UInt64 limit_, bool throw_exception_, std::optional<size_t> exact_limit_,
|
|
std::string exception_message_ = {});
|
|
~LimitReadBuffer() override;
|
|
|
|
private:
|
|
ReadBuffer * in;
|
|
bool owns_in;
|
|
|
|
UInt64 limit;
|
|
bool throw_exception;
|
|
std::optional<size_t> exact_limit;
|
|
std::string exception_message;
|
|
|
|
LimitReadBuffer(ReadBuffer * in_, bool owns, UInt64 limit_, bool throw_exception_, std::optional<size_t> exact_limit_, std::string exception_message_);
|
|
|
|
bool nextImpl() override;
|
|
};
|
|
|
|
}
|