FileQueue.h

Go to the documentation of this file.

 
#pragma once
#include "BufferedFileStream.h"
#include "IndexFile.h"
#include "catapult/functions.h"
#include <boost/filesystem/path.hpp>
 
namespace catapult { namespace io {
 
    class FileQueueWriter final : public OutputStream {
    public:
        explicit FileQueueWriter(const std::string& directory);
 
        FileQueueWriter(const std::string& directory, const std::string& indexFilename);
 
    public:
        void write(const RawBuffer& buffer) override;
        void flush() override;
 
    private:
        boost::filesystem::path m_directory;
        IndexFile m_indexFile;
        uint64_t m_indexValue;
        std::unique_ptr<BufferedOutputFileStream> m_pOutputStream;
    };
 
    class FileQueueReader final {
    public:
        explicit FileQueueReader(const std::string& directory);
 
        FileQueueReader(const std::string& directory, const std::string& readerIndexFilename, const std::string& writerIndexFilename);
 
    public:
        size_t pending() const;
 
    public:
        bool tryReadNextMessage(const consumer<std::vector<uint8_t>>& consumer);
 
        void skip(uint32_t count);
 
    private:
        bool process(const consumer<const std::string&>& processFilename);
 
    private:
        boost::filesystem::path m_directory;
        IndexFile m_readerIndexFile;
        IndexFile m_writerIndexFile;
    };
}}