ClickHouse/src/IO/Lz4InflatingReadBuffer.h

#pragma once

#include <IO/BufferWithOwnMemory.h>
#include <IO/CompressionMethod.h>
#include <IO/ReadBuffer.h>

#include <lz4.h>
#include <lz4frame.h>


namespace DB
{

class Lz4InflatingReadBuffer : public BufferWithOwnMemory<ReadBuffer>
{
public:
    Lz4InflatingReadBuffer(
        std::unique_ptr<ReadBuffer> in_,
        size_t buf_size = DBMS_DEFAULT_BUFFER_SIZE,
        char * existing_memory = nullptr,
        size_t alignment = 0);

    ~Lz4InflatingReadBuffer() override;

private:
    bool nextImpl() override;

    std::unique_ptr<ReadBuffer> in;

    LZ4F_dctx* dctx;

    void * in_data;
    void * out_data;

    size_t in_available;
    size_t out_available;

    bool eof_flag = false;
};

}