BufferedFileStream.h

Go to the documentation of this file.

 
#pragma once
#include "RawFile.h"
#include "Stream.h"
#include <vector>
 
namespace catapult { namespace io {
 
    constexpr size_t Default_Stream_Buffer_Size = 4096;
 
    class BufferedOutputFileStream final : public OutputStream {
    public:
        BufferedOutputFileStream(RawFile&& rawFile, size_t bufferSize = Default_Stream_Buffer_Size);
 
    public:
        void write(const RawBuffer& buffer) override;
 
        void flush() override;
 
    private:
        RawFile m_rawFile;
        std::vector<uint8_t> m_buffer;
        size_t m_bufferPosition; // position of next write
    };
 
    class BufferedInputFileStream final : public InputStream {
    public:
        BufferedInputFileStream(RawFile&& rawFile, size_t bufferSize = Default_Stream_Buffer_Size);
 
    public:
        bool eof() const override;
 
        void read(const MutableRawBuffer& buffer) override;
 
    private:
        RawFile m_rawFile;
        std::vector<uint8_t> m_buffer;
        size_t m_bufferPosition; // position of next read
        // during reading m_numBytesInBuffer mostly will be m_buffer.size() and might be smaller near the end of file
        size_t m_numBytesInBuffer;
    };
}}