broker_probe.hpp

Go to the documentation of this file.

/*
  ____            _
 | __ ) _ __ ___ | | _____ _ __
 |  _ \| '__/ _ \| |/ / _ \ '__|
 | |_) | | | (_) |   <  __/ |
 |____/|_|  \___/|_|\_\___|_|
  ____            _
 |  _ \ _ __ ___ | |__   ___
 | |_) | '__/ _ \| '_ \ / _ \
 |  __/| | | (_) | |_) |  __/
 |_|   |_|  \___/|_.__/ \___|
 
Small, dependency-light readiness probes shared by `mads up`'s
`ready = "broker" | "port:<n>"` and (later, on a different branch) by
`mads doctor`'s broker-reachability check. Kept outside agent.hpp/.cpp on
purpose: neither probe needs a full Mads::Agent.
 
Author(s): Paolo Bosetti
*/
#ifndef MADS_BROKER_PROBE_HPP
#define MADS_BROKER_PROBE_HPP
 
#include <chrono>
#include <filesystem>
#include <map>
#include <optional>
#include <string>
 
namespace Mads {
 
struct ProbeCurveKeys {
  std::filesystem::path key_dir;
  std::string client_key_name = "client";
  std::string server_key_name = "broker";
};
 
bool probe_broker(const std::string &uri, std::chrono::milliseconds timeout,
                  const std::optional<ProbeCurveKeys> &curve = std::nullopt);
 
bool probe_tcp_port(const std::string &host, int port,
                    std::chrono::milliseconds timeout);
 
enum class CurveProbeResult { Connected, RejectedAuth, Timeout };
 
CurveProbeResult
probe_curve_handshake(const std::string &uri,
                      const std::filesystem::path &key_dir,
                      const std::string &client_key_name,
                      const std::string &server_key_name,
                      std::chrono::milliseconds timeout);
 
std::optional<std::map<std::string, int>>
fetch_subscription_table(const std::string &sub_uri,
                         std::chrono::milliseconds timeout,
                         const std::optional<ProbeCurveKeys> &curve = std::nullopt);
 
} // namespace Mads
 
#endif // MADS_BROKER_PROBE_HPP