我正在尝试为 C++ 20 服务器示例创建一个客户端，该示例使用协程。
我不太确定我应该如何关闭客户端连接。据我所知，有两种方法：
一旦它准备好，这个似乎正在关闭它/没有其他事情可以像读/写操作那样做。
asio::signal_set signals(io_context, SIGINT, SIGTERM);
signals.async_wait([&](auto, auto) { io_context.stop(); });
强制关闭？
asio::post(io_context_, [this]() { socket_.close(); });
我应该使用哪一种？
#include <cstdlib>
#include <deque>
#include <iostream>
#include <thread>
#include <string>
#include <asio.hpp>
using asio::ip::tcp;
using asio::awaitable;
using asio::co_spawn;
using asio::detached;
using asio::redirect_error;
using asio::use_awaitable;
awaitable<void> connect(tcp::socket socket, const tcp::endpoint& endpoint)
{
co_await socket.async_connect(endpoint, use_awaitable);
}
int main()
{
try
{
asio::io_context io_context;
tcp::endpoint endpoint(asio::ip::make_address("127.0.0.1"), 666);
tcp::socket socket(io_context);
co_spawn(io_context, connect(std::move(socket), endpoint), detached);
io_context.run();
}
catch (std::exception& e)
{
std::cerr << "Exception: " << e.what() << "\n";
}
return 0;
}
#include <cstdlib>
#include <deque>
#include <iostream>
#include <list>
#include <memory>
#include <set>
#include <string>
#include <utility>
#include <asio/awaitable.hpp>
#include <asio/detached.hpp>
#include <asio/co_spawn.hpp>
#include <asio/io_context.hpp>
#include <asio/ip/tcp.hpp>
#include <asio/read_until.hpp>
#include <asio/redirect_error.hpp>
#include <asio/signal_set.hpp>
#include <asio/steady_timer.hpp>
#include <asio/use_awaitable.hpp>
#include <asio/write.hpp>
using asio::ip::tcp;
using asio::awaitable;
using asio::co_spawn;
using asio::detached;
using asio::redirect_error;
using asio::use_awaitable;
//----------------------------------------------------------------------
class chat_participant
{
public:
virtual ~chat_participant() = default;
virtual void deliver(const std::string& msg) = 0;
};
typedef std::shared_ptr<chat_participant> chat_participant_ptr;
//----------------------------------------------------------------------
class chat_room
{
public:
void join(chat_participant_ptr participant)
{
participants_.insert(participant);
for (const auto &msg : recent_msgs_)
participant->deliver(msg);
}
void leave(chat_participant_ptr participant)
{
participants_.erase(participant);
}
void deliver(const std::string& msg)
{
recent_msgs_.push_back(msg);
while (recent_msgs_.size() > max_recent_msgs)
recent_msgs_.pop_front();
for (const auto &participant : participants_)
participant->deliver(msg);
}
private:
std::set<chat_participant_ptr> participants_;
enum { max_recent_msgs = 100 };
std::deque<std::string> recent_msgs_;
};
//----------------------------------------------------------------------
class chat_session
: public chat_participant,
public std::enable_shared_from_this<chat_session>
{
public:
chat_session(tcp::socket socket, chat_room& room)
: socket_(std::move(socket)),
timer_(socket_.get_executor()),
room_(room)
{
timer_.expires_at(std::chrono::steady_clock::time_point::max());
}
void start()
{
room_.join(shared_from_this());
co_spawn(socket_.get_executor(),
[self = shared_from_this()]{ return self->reader(); },
detached);
co_spawn(socket_.get_executor(),
[self = shared_from_this()]{ return self->writer(); },
detached);
}
void deliver(const std::string& msg) override
{
write_msgs_.push_back(msg);
timer_.cancel_one();
}
private:
awaitable<void> reader()
{
try
{
for (std::string read_msg;;)
{
std::size_t n = co_await asio::async_read_until(socket_,
asio::dynamic_buffer(read_msg, 1024), "\n", use_awaitable);
room_.deliver(read_msg.substr(0, n));
read_msg.erase(0, n);
}
}
catch (std::exception&)
{
stop();
}
}
awaitable<void> writer()
{
try
{
while (socket_.is_open())
{
if (write_msgs_.empty())
{
asio::error_code ec;
co_await timer_.async_wait(redirect_error(use_awaitable, ec));
}
else
{
co_await asio::async_write(socket_,
asio::buffer(write_msgs_.front()), use_awaitable);
write_msgs_.pop_front();
}
}
}
catch (std::exception&)
{
stop();
}
}
void stop()
{
room_.leave(shared_from_this());
socket_.close();
timer_.cancel();
}
tcp::socket socket_;
asio::steady_timer timer_;
chat_room& room_;
std::deque<std::string> write_msgs_;
};
//----------------------------------------------------------------------
awaitable<void> listener(tcp::acceptor acceptor)
{
chat_room room;
for (;;)
{
std::make_shared<chat_session>(co_await acceptor.async_accept(use_awaitable), room)->start();
}
}
//----------------------------------------------------------------------
int main()
{
try
{
unsigned short port = 666;
asio::io_context io_context(1);
co_spawn(io_context,
listener(tcp::acceptor(io_context, { tcp::v4(), port })),
detached);
asio::signal_set signals(io_context, SIGINT, SIGTERM);
signals.async_wait([&](auto, auto) { io_context.stop(); });
io_context.run();
}
catch (std::exception& e)
{
std::cerr << "Exception: " << e.what() << "\n";
}
return 0;
}
在由 提供的示例中asio，listener在io_context线程/线程池中运行，run()在构造io_context(1 /* pool of 1 */).
侦听器将使用acceptor来侦听io_context. 该acceptor会创建一个新的chat_session为每一个新socket的连接，并把它交给了chat_room。
因此，要安全地关闭 connection，您需要post一个 lambda to asio。该asio::post会排队的拉姆达从内进行io_context线程（或多个）。
您需要提供正确io_context且socket归chat_session. 该连接必须从封闭内的io_context，如下所示：
// Where "this" is the current chat_session owning the socket
asio::post(io_context_, [this]() { socket_.close(); });
所述io_contextWIL然后关闭连接，并且还调用任何活性注册async_read/async_write所述的方法chat_session，如在C ++ 11例如：
void do_read()
{
asio::async_read(socket_,
asio::buffer(read_msg_.data(), chat_message::header_length),
/* You can provide a lambda to be called on a read / error */
[this](std::error_code ec, std::size_t /*length read*/)
{
if (!ec)
{
do_read(); // No error -> Keep on reading
}
else
{
// You'll reach this point if an active async_read was stopped
// due to an error or if you called socket_.close()
// Error -> You can close the socket here as well,
// because it is called from within the io_context
socket_.close();
}
});
}
您的第一个选项实际上会停止整个io_context. 这应该用于优雅地退出您的程序或io_context作为一个整体停止 asio 。
因此，您应该使用第二个选项来“关闭 ASIO 中的异步客户端连接”。
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句