2
0
mirror of https://github.com/boostorg/redis.git synced 2026-01-30 08:02:39 +00:00
Files
redis/examples/echo_server.cpp
2022-01-07 03:14:20 +01:00

66 lines
1.5 KiB
C++

#include <aedis/aedis.hpp>
#include <iostream>
#include "user_session.hpp"
namespace net = aedis::net;
using aedis::resp3::client_base;
using tcp_acceptor = aedis::net::use_awaitable_t<>::as_default_on_t<aedis::net::ip::tcp::acceptor>;
class my_redis_client : public client_base_type {
private:
void on_event(response_id id) override
{
if (auto session = id.session.lock()) {
session->on_event(id);
} else {
std::cout << "Session expired." << std::endl;
}
}
public:
my_redis_client(net::any_io_executor ex)
: client_base_type(ex)
{}
};
net::awaitable<void> listener()
{
auto ex = co_await net::this_coro::executor;
tcp_acceptor acceptor(ex, {net::ip::tcp::v4(), 55555});
auto rclient = std::make_shared<my_redis_client>(ex);
rclient->start();
std::string resp; // TODO: Use a shared_ptr.
for (;;) {
auto socket = co_await acceptor.async_accept();
auto session = std::make_shared<user_session>(std::move(socket), rclient);
response_id id{command::ping, &resp, session};
auto filler = [id](auto& req, auto const& msg)
{ req.push(id, msg); };
session->start(filler);
}
}
int main()
{
try {
net::io_context io_context(1);
net::signal_set signals(io_context, SIGINT, SIGTERM);
signals.async_wait([&](auto, auto){ io_context.stop(); });
co_spawn(io_context, listener(), net::detached);
io_context.run();
} catch (std::exception& e) {
std::cerr << e.what() << std::endl;
}
}
/// \example echo_server.cpp