ConcurrentStreamBuf.h

Go to the documentation of this file.

 
#pragma once
 
#include <aws/core/Core_EXPORTS.h>
#include <aws/core/auth/AWSAuthSigner.h>
#include <aws/common/array_list.h>
 
#include <mutex>
#include <condition_variable>
#include <streambuf>
#include <ios>
#include <cassert>
 
namespace Aws
{
    namespace Utils
    {
        namespace Stream
        {
            class AWS_CORE_API ConcurrentStreamBuf : public std::streambuf
            {
            public:
 
                explicit ConcurrentStreamBuf(size_t bufferLength = 4 * 1024);
 
                void SetEof();
 
            protected:
                std::streampos seekoff(std::streamoff off, std::ios_base::seekdir dir, std::ios_base::openmode which = std::ios_base::in | std::ios_base::out) override;
                std::streampos seekpos(std::streampos pos, std::ios_base::openmode which = std::ios_base::in | std::ios_base::out) override;
 
                int underflow() override;
                int overflow(int ch) override;
                int sync() override;
                std::streamsize showmanyc() override;
 
                void FlushPutArea();
 
            private:
                Aws::Vector<unsigned char> m_getArea;
                Aws::Vector<unsigned char> m_putArea;
                Aws::Vector<unsigned char> m_backbuf; // used to shuttle data from the put area to the get area
                std::mutex m_lock; // synchronize access to the common backbuffer
                std::condition_variable m_signal;
                bool m_eof;
            };
        }
    }
}