mirror of
https://github.com/boostorg/redis.git
synced 2026-01-19 16:52:08 +00:00
87 lines
2.2 KiB
C++
87 lines
2.2 KiB
C++
#include <boost/asio/co_spawn.hpp>
|
|
#include <boost/asio/consign.hpp>
|
|
|
|
#include "common.hpp"
|
|
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
#include <stdexcept>
|
|
|
|
namespace net = boost::asio;
|
|
|
|
struct run_callback {
|
|
std::shared_ptr<boost::redis::connection> conn;
|
|
boost::redis::operation op;
|
|
boost::system::error_code expected;
|
|
|
|
void operator()(boost::system::error_code const& ec) const
|
|
{
|
|
std::cout << "async_run: " << ec.message() << std::endl;
|
|
conn->cancel(op);
|
|
}
|
|
};
|
|
|
|
void run(
|
|
std::shared_ptr<boost::redis::connection> conn,
|
|
boost::redis::config cfg,
|
|
boost::system::error_code ec,
|
|
boost::redis::operation op)
|
|
{
|
|
conn->async_run(cfg, run_callback{conn, op, ec});
|
|
}
|
|
|
|
static std::string safe_getenv(const char* name, const char* default_value)
|
|
{
|
|
// MSVC doesn't like getenv
|
|
#ifdef BOOST_MSVC
|
|
#pragma warning(push)
|
|
#pragma warning(disable : 4996)
|
|
#endif
|
|
const char* res = std::getenv(name);
|
|
#ifdef BOOST_MSVC
|
|
#pragma warning(pop)
|
|
#endif
|
|
return res ? res : default_value;
|
|
}
|
|
|
|
std::string get_server_hostname() { return safe_getenv("BOOST_REDIS_TEST_SERVER", "localhost"); }
|
|
|
|
boost::redis::config make_test_config()
|
|
{
|
|
boost::redis::config cfg;
|
|
cfg.addr.host = get_server_hostname();
|
|
return cfg;
|
|
}
|
|
|
|
#ifdef BOOST_ASIO_HAS_CO_AWAIT
|
|
void run_coroutine_test(net::awaitable<void> op, std::chrono::steady_clock::duration timeout)
|
|
{
|
|
net::io_context ioc;
|
|
bool finished = false;
|
|
net::co_spawn(ioc, std::move(op), [&finished](std::exception_ptr p) {
|
|
if (p)
|
|
std::rethrow_exception(p);
|
|
finished = true;
|
|
});
|
|
ioc.run_for(timeout);
|
|
if (!finished)
|
|
throw std::runtime_error("Coroutine test did not finish");
|
|
}
|
|
#endif // BOOST_ASIO_HAS_CO_AWAIT
|
|
|
|
// Finds a value in the output of the CLIENT INFO command
|
|
// format: key1=value1 key2=value2
|
|
// TODO: duplicated
|
|
std::string_view find_client_info(std::string_view client_info, std::string_view key)
|
|
{
|
|
std::string prefix{key};
|
|
prefix += '=';
|
|
|
|
auto const pos = client_info.find(prefix);
|
|
if (pos == std::string_view::npos)
|
|
return {};
|
|
auto const pos_begin = pos + prefix.size();
|
|
auto const pos_end = client_info.find(' ', pos_begin);
|
|
return client_info.substr(pos_begin, pos_end - pos_begin);
|
|
}
|