mirror/oatpp
2
0
Fork 0
mirror of https://github.com/oatpp/oatpp.git synced 2025-03-31 18:30:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Introduce stream write callbacks

Browse Source
This commit is contained in:
lganzzzo 2019-07-13 20:14:55 +03:00
parent d94cddfa98
commit 6b99725db5
2 changed files with 146 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
src/oatpp/core/data/stream/Stream.cpp
View File

@ -27,6 +27,9 @@
namespace oatpp { namespace data{ namespace stream {
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// ConsistentOutputStream
oatpp::async::Action ConsistentOutputStream::suggestOutputStreamAction(data::v_io_size ioResult) {
if(ioResult > 0) {
@ -86,8 +89,37 @@ data::v_io_size ConsistentOutputStream::writeAsString(bool value) {
return write("false", 5);
}
}
// Functions
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// DefaultWriteCallback
DefaultWriteCallback::DefaultWriteCallback(OutputStream* stream)
: m_stream(stream)
{}
data::v_io_size DefaultWriteCallback::write(const void *data, data::v_io_size count) {
return writeExactSizeData(m_stream, data, count);
}
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// DefaultAsyncWriteCallback
DefaultAsyncWriteCallback::DefaultAsyncWriteCallback(OutputStream* stream)
: m_stream(stream)
{}
oatpp::async::Action DefaultAsyncWriteCallback::writeAsyncInline(oatpp::async::AbstractCoroutine* coroutine,
const void*& currBufferPtr,
data::v_io_size& bytesLeft,
oatpp::async::Action&& nextAction)
{
return writeExactSizeDataAsyncInline(coroutine, m_stream, currBufferPtr, bytesLeft, std::forward<oatpp::async::Action>(nextAction));
}
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Other functions
ConsistentOutputStream& operator << (ConsistentOutputStream& s, const oatpp::String& str) {
if(str) {
@ -193,10 +225,11 @@ ConsistentOutputStream& operator << (ConsistentOutputStream& s, bool value) {
}
oatpp::data::v_io_size transfer(const std::shared_ptr<InputStream>& fromStream,
const std::shared_ptr<OutputStream>& toStream,
oatpp::data::v_io_size transferSize,
void* buffer,
oatpp::data::v_io_size bufferSize) {
const std::shared_ptr<OutputStream>& toStream,
oatpp::data::v_io_size transferSize,
void* buffer,
oatpp::data::v_io_size bufferSize)
{
oatpp::data::v_io_size progress = 0;
@ -227,7 +260,8 @@ oatpp::data::v_io_size transfer(const std::shared_ptr<InputStream>& fromStream,
oatpp::async::CoroutineStarter transferAsync(const std::shared_ptr<InputStream>& fromStream,
const std::shared_ptr<OutputStream>& toStream,
oatpp::data::v_io_size transferSize,
const std::shared_ptr<oatpp::data::buffer::IOBuffer>& buffer) {
const std::shared_ptr<oatpp::data::buffer::IOBuffer>& buffer)
{
class TransferCoroutine : public oatpp::async::Coroutine<TransferCoroutine> {
private:

105
src/oatpp/core/data/stream/Stream.hpp
View File

@ -290,8 +290,113 @@ ConsistentOutputStream& operator << (ConsistentOutputStream& s, v_float32 value)
ConsistentOutputStream& operator << (ConsistentOutputStream& s, v_float64 value);
ConsistentOutputStream& operator << (ConsistentOutputStream& s, bool value);
/**
* Callback for stream write operation.
*/
class WriteCallback {
public:
/**
* Default virtual destructor.
*/
virtual ~WriteCallback() = default;
/**
* Write callback.
* @param data - pointer to data.
* @param count - size of the data in bytes.
* @return - &id:oatpp::data::v_io_size;.
*/
virtual data::v_io_size write(const void *data, data::v_io_size count) = 0;
};
/**
* Callback for stream asynchronous write operation.
*/
class AsyncWriteCallback {
public:
/**
* Default virtual destructor.
*/
virtual ~AsyncWriteCallback() = default;
/**
* Async-Inline write callback.
* @param coroutine - caller coroutine.
* @param currBufferPtr - pointer to current data position.
* @param bytesLeft - how much bytes left to write.
* @param nextAction - next action when write finished.
* @return - &id:oatpp::async::Action;.
*/
virtual oatpp::async::Action writeAsyncInline(oatpp::async::AbstractCoroutine* coroutine,
const void*& currBufferPtr,
data::v_io_size& bytesLeft,
oatpp::async::Action&& nextAction) = 0;
};
/**
* Default callback for stream write operation. <br>
* Uses &l:writeExactSizeData (); method underhood.
*/
class DefaultWriteCallback : public WriteCallback {
private:
OutputStream* m_stream;
public:
/**
* Constructor.
* @param stream - stream to write to.
*/
DefaultWriteCallback(OutputStream* stream);
/**
* Write callback.
* @param data - pointer to data.
* @param count - size of the data in bytes.
* @return - &id:oatpp::data::v_io_size;.
*/
data::v_io_size write(const void *data, data::v_io_size count) override;
};
/**
* Default callback for stream asynchronous write operation.
* Uses &l:writeExactSizeDataAsyncInline (); method underhood.
*/
class DefaultAsyncWriteCallback : public AsyncWriteCallback {
private:
OutputStream* m_stream;
public:
/**
* Constructor.
* @param stream - stream to write to.
*/
DefaultAsyncWriteCallback(OutputStream* stream);
/**
* Async-Inline write callback.
* @param coroutine - caller coroutine.
* @param currBufferPtr - pointer to current data position.
* @param bytesLeft - how much bytes left to write.
* @param nextAction - next action when write finished.
* @return - &id:oatpp::async::Action;.
*/
oatpp::async::Action writeAsyncInline(oatpp::async::AbstractCoroutine* coroutine,
const void*& currBufferPtr,
data::v_io_size& bytesLeft,
oatpp::async::Action&& nextAction) override;
};
/**
* Error of Asynchronous stream transfer.
*/
class AsyncTransferError : public oatpp::async::Error {
public:
/**
* Constructor.
* @param what
*/
AsyncTransferError(const char* what) : oatpp::async::Error(what) {}
};