123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778 |
- //
- // echo_server_with_default.cpp
- // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
- //
- // Copyright (c) 2003-2019 Christopher M. Kohlhoff (chris at kohlhoff dot com)
- //
- // Distributed under the Boost Software License, Version 1.0. (See accompanying
- // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
- //
- #include <boost/asio/co_spawn.hpp>
- #include <boost/asio/detached.hpp>
- #include <boost/asio/io_context.hpp>
- #include <boost/asio/ip/tcp.hpp>
- #include <boost/asio/signal_set.hpp>
- #include <boost/asio/write.hpp>
- #include <cstdio>
- using boost::asio::ip::tcp;
- using boost::asio::awaitable;
- using boost::asio::co_spawn;
- using boost::asio::detached;
- using boost::asio::use_awaitable_t;
- using tcp_acceptor = use_awaitable_t<>::as_default_on_t<tcp::acceptor>;
- using tcp_socket = use_awaitable_t<>::as_default_on_t<tcp::socket>;
- namespace this_coro = boost::asio::this_coro;
- awaitable<void> echo(tcp_socket socket)
- {
- try
- {
- char data[1024];
- for (;;)
- {
- std::size_t n = co_await socket.async_read_some(boost::asio::buffer(data));
- co_await async_write(socket, boost::asio::buffer(data, n));
- }
- }
- catch (std::exception& e)
- {
- std::printf("echo Exception: %s\n", e.what());
- }
- }
- awaitable<void> listener()
- {
- auto executor = co_await this_coro::executor;
- tcp_acceptor acceptor(executor, {tcp::v4(), 55555});
- for (;;)
- {
- auto socket = co_await acceptor.async_accept();
- co_spawn(executor,
- [socket = std::move(socket)]() mutable
- {
- return echo(std::move(socket));
- },
- detached);
- }
- }
- int main()
- {
- try
- {
- boost::asio::io_context io_context(1);
- boost::asio::signal_set signals(io_context, SIGINT, SIGTERM);
- signals.async_wait([&](auto, auto){ io_context.stop(); });
- co_spawn(io_context, listener, detached);
- io_context.run();
- }
- catch (std::exception& e)
- {
- std::printf("Exception: %s\n", e.what());
- }
- }
|