AuroraRuntime/Include/Aurora/IO/Buffered/BlobWriter.hpp

70 lines
1.7 KiB
C++

/***
Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.
File: BlobWriter.hpp
Date: 2021-6-10
Author: Reece
***/
#pragma once
namespace Aurora::IO::Buffered
{
struct BlobWriter : IStreamWriter
{
AU_NO_COPY_NO_MOVE(BlobWriter)
inline BlobWriter(const AuSPtr<Memory::ByteBuffer> &buffer) : buffer_(buffer) {}
inline BlobWriter() : buffer_(AuMakeShared<Memory::ByteBuffer>()) {}
inline ~BlobWriter() {}
inline virtual EStreamError IsOpen() override
{
auto pBuffer = this->buffer_;
if (!pBuffer)
{
return EStreamError::eErrorStreamNotOpen;
}
if (pBuffer->HasStreamError())
{
return EStreamError::eErrorByteBuffer;
}
return EStreamError::eErrorNone;
}
inline virtual EStreamError Write(const Memory::MemoryViewStreamRead &parameters) override
{
auto pBuffer = this->buffer_;
if (!pBuffer)
{
return EStreamError::eErrorStreamNotOpen;
}
if (pBuffer->HasStreamError())
{
return EStreamError::eErrorByteBuffer;
}
parameters.outVariable = pBuffer->Write(parameters.ptr, parameters.length);
return EStreamError::eErrorNone;
}
inline virtual void Flush() override
{
}
inline virtual void Close() override
{
AuResetMember(this->buffer_);
}
inline AuSPtr<Memory::ByteBuffer> GetBuffer()
{
return this->buffer_;
}
private:
AuSPtr<Memory::ByteBuffer> buffer_;
};
}