File: cpp20_resolve_with_sentinel.cpp

package info (click to toggle)
boost1.90 1.90.0-5
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 593,168 kB
  • sloc: cpp: 4,190,642; xml: 196,648; python: 34,618; ansic: 23,145; asm: 5,468; sh: 3,776; makefile: 1,162; perl: 1,020; sql: 728; ruby: 676; yacc: 478; java: 77; lisp: 24; csh: 6
file content (77 lines) | stat: -rw-r--r-- 2,412 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
/* Copyright (c) 2018-2022 Marcelo Zimbres Silva (mzimbres@gmail.com)
 *
 * Distributed under the Boost Software License, Version 1.0. (See
 * accompanying file LICENSE.txt)
 */

#include <boost/redis/connection.hpp>

#include <boost/asio/consign.hpp>
#include <boost/asio/detached.hpp>
#include <boost/asio/redirect_error.hpp>
#include <boost/asio/use_awaitable.hpp>

#include <iostream>

#if defined(BOOST_ASIO_HAS_CO_AWAIT)

namespace asio = boost::asio;
using endpoints = asio::ip::tcp::resolver::results_type;
using boost::redis::request;
using boost::redis::response;
using boost::redis::ignore_t;
using boost::redis::config;
using boost::redis::address;
using boost::redis::connection;

auto redir(boost::system::error_code& ec) { return asio::redirect_error(asio::use_awaitable, ec); }

// For more info see
// - https://redis.io/docs/manual/sentinel.
// - https://redis.io/docs/reference/sentinel-clients.
auto resolve_master_address(std::vector<address> const& addresses) -> asio::awaitable<address>
{
   request req;
   req.push("SENTINEL", "get-master-addr-by-name", "mymaster");
   req.push("QUIT");

   auto conn = std::make_shared<connection>(co_await asio::this_coro::executor);

   response<std::optional<std::array<std::string, 2>>, ignore_t> resp;
   for (auto addr : addresses) {
      boost::system::error_code ec;
      config cfg;
      cfg.addr = addr;
      // TODO: async_run and async_exec should be lauched in
      // parallel here so we can wait for async_run completion
      // before eventually calling it again.
      conn->async_run(cfg, asio::consign(asio::detached, conn));
      co_await conn->async_exec(req, resp, redir(ec));
      conn->cancel();
      if (!ec && std::get<0>(resp))
         co_return address{
            std::get<0>(resp).value().value().at(0),
            std::get<0>(resp).value().value().at(1)};
   }

   co_return address{};
}

auto co_main(config cfg) -> asio::awaitable<void>
{
   // A list of sentinel addresses from which only one is responsive.
   // This simulates sentinels that are down.
   std::vector<address> const addresses{
      address{"foo", "26379"},
      address{"bar", "26379"},
      cfg.addr
   };

   auto const ep = co_await resolve_master_address(addresses);

   std::clog << "Host: " << ep.host << "\n"
             << "Port: " << ep.port << "\n"
             << std::flush;
}

#endif  // defined(BOOST_ASIO_HAS_CO_AWAIT)