Skip to content

Commit

Permalink
Simplifies the connect operations.
Browse files Browse the repository at this point in the history
  • Loading branch information
mzimbres committed Oct 13, 2024
1 parent 85443c8 commit 0665885
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 63 deletions.
76 changes: 14 additions & 62 deletions include/boost/redis/detail/connector.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -12,9 +12,8 @@
#include <boost/asio/compose.hpp>
#include <boost/asio/connect.hpp>
#include <boost/asio/coroutine.hpp>
#include <boost/asio/experimental/parallel_group.hpp>
#include <boost/asio/ip/tcp.hpp>
#include <boost/asio/steady_timer.hpp>
#include <boost/asio/cancel_after.hpp>
#include <string>
#include <chrono>

Expand All @@ -30,64 +29,28 @@ struct connect_op {

template <class Self>
void operator()( Self& self
, std::array<std::size_t, 2> const& order = {}
, system::error_code const& ec1 = {}
, asio::ip::tcp::endpoint const& ep= {}
, system::error_code const& ec2 = {})
, system::error_code const& ec = {}
, asio::ip::tcp::endpoint const& ep= {})
{
BOOST_ASIO_CORO_REENTER (coro)
{
ctor_->timer_.expires_after(ctor_->timeout_);

BOOST_ASIO_CORO_YIELD
asio::experimental::make_parallel_group(
[this](auto token)
{
auto f = [](system::error_code const&, auto const&) { return true; };
return asio::async_connect(*stream, *res_, f, token);
},
[this](auto token) { return ctor_->timer_.async_wait(token);}
).async_wait(
asio::experimental::wait_for_one(),
std::move(self));

if (is_cancelled(self)) {
self.complete(asio::error::operation_aborted);
return;
}

switch (order[0]) {
case 0: {
ctor_->endpoint_ = ep;
self.complete(ec1);
} break;
case 1:
{
if (ec2) {
self.complete(ec2);
} else {
self.complete(error::connect_timeout);
}
} break;

default: BOOST_ASSERT(false);
}
asio::async_connect(*stream, *res_,
[](system::error_code const&, auto const&) { return true; },
asio::cancel_after(ctor_->timeout_, std::move(self)));

ctor_->endpoint_ = ep;

// TODO: map operation_canceled into error::connect_timeout
self.complete(ec);
}
}
};

template <class Executor>
class connector {
public:
using timer_type =
asio::basic_waitable_timer<
std::chrono::steady_clock,
asio::wait_traits<std::chrono::steady_clock>,
Executor>;

connector(Executor ex)
: timer_{ex}
{}
connector(Executor) {}

void set_config(config const& cfg)
{ timeout_ = cfg.connect_timeout; }
Expand All @@ -102,28 +65,17 @@ class connector {
return asio::async_compose
< CompletionToken
, void(system::error_code)
>(connect_op<connector, Stream>{this, &stream, &res}, token, timer_);
>(connect_op<connector, Stream>{this, &stream, &res}, token);
}

std::size_t cancel(operation op)
{
switch (op) {
case operation::connect:
case operation::all:
timer_.cancel();
break;
default: /* ignore */;
}

return 0;
}
{ return 0; }

auto const& endpoint() const noexcept { return endpoint_;}

private:
template <class, class> friend struct connect_op;

timer_type timer_;
std::chrono::steady_clock::duration timeout_ = std::chrono::seconds{2};
asio::ip::tcp::endpoint endpoint_;
};
Expand Down
1 change: 0 additions & 1 deletion include/boost/redis/detail/runner.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -205,7 +205,6 @@ class runner {
using connector_type = connector<Executor>;
using handshaker_type = detail::handshaker<Executor>;
using health_checker_type = health_checker<Executor>;
using timer_type = typename connector_type::timer_type;

template <class, class, class> friend class runner_op;
template <class, class, class> friend struct hello_op;
Expand Down

0 comments on commit 0665885

Please sign in to comment.