mirror of
https://github.com/boostorg/redis.git
synced 2026-02-08 23:12:14 +00:00
107e56145d
- Improvements in the buffer interface. - Parser can deal with bulk strings that contain the separator. - Test improvements. - Removes the parser from the async op.
84 lines
2.3 KiB
C++
84 lines
2.3 KiB
C++
/* Copyright (c) 2019 Marcelo Zimbres Silva (mzimbres at gmail dot com)
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
#include <aedis/aedis.hpp>
|
|
|
|
namespace aedis {
|
|
|
|
struct initiate_async_receive {
|
|
using executor_type = aedis::net::system_executor;
|
|
|
|
std::string::const_iterator pbegin;
|
|
std::string::const_iterator pend;
|
|
|
|
executor_type get_executor() noexcept
|
|
{ return aedis::net::system_executor(); }
|
|
|
|
template <
|
|
class ReadHandler,
|
|
class MutableBufferSequence>
|
|
void operator()(
|
|
ReadHandler&& handler,
|
|
MutableBufferSequence const& buffers)
|
|
{
|
|
boost::system::error_code ec;
|
|
if (std::size(buffers) == 0) {
|
|
handler(ec, 0);
|
|
return;
|
|
}
|
|
|
|
auto begin = boost::asio::buffer_sequence_begin(buffers);
|
|
auto end = boost::asio::buffer_sequence_end(buffers);
|
|
//std::cout << "Buffers size: " << std::size(buffers) << std::endl;
|
|
|
|
std::size_t transferred = 0;
|
|
while (begin != end) {
|
|
//std::cout << "Buffer size: " << std::ssize(*begin) << std::endl;
|
|
auto const min = std::min(std::ssize(*begin), pend - pbegin);
|
|
std::copy(pbegin, pbegin + min, static_cast<char*>(begin->data()));
|
|
std::advance(pbegin, min);
|
|
transferred += min;
|
|
++begin;
|
|
}
|
|
|
|
handler(ec, transferred);
|
|
}
|
|
};
|
|
|
|
template <class Executor>
|
|
struct test_stream {
|
|
std::string payload;
|
|
|
|
using executor_type = Executor;
|
|
|
|
template<
|
|
class MutableBufferSequence,
|
|
class ReadHandler =
|
|
aedis::net::default_completion_token_t<executor_type>
|
|
>
|
|
auto async_read_some(
|
|
MutableBufferSequence const& buffers,
|
|
ReadHandler&& handler = net::default_completion_token_t<executor_type>{})
|
|
{
|
|
return aedis::net::async_initiate<ReadHandler,
|
|
void (boost::system::error_code, std::size_t)>(
|
|
initiate_async_receive
|
|
{std::cbegin(payload), std::cend(payload)},
|
|
handler, buffers);
|
|
}
|
|
|
|
executor_type get_executor() noexcept
|
|
{ return aedis::net::system_executor(); }
|
|
|
|
template<class Executor1>
|
|
struct rebind_executor {
|
|
using other = test_stream<Executor1>;
|
|
};
|
|
};
|
|
|
|
}
|