mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 02:12:21 +00:00
29 lines
612 B
C++
29 lines
612 B
C++
#pragma once
|
|
|
|
#include <Core/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
|
|
{
|
|
private:
|
|
ReadBuffer & in;
|
|
UInt64 limit;
|
|
bool throw_exception;
|
|
std::string exception_message;
|
|
|
|
bool nextImpl() override;
|
|
|
|
public:
|
|
LimitReadBuffer(ReadBuffer & in, UInt64 limit, bool throw_exception, std::string exception_message = {});
|
|
~LimitReadBuffer() override;
|
|
};
|
|
|
|
}
|