如何在 ASIO 中关闭异步客户端连接？

Question

我正在尝试为 C++ 20 服务器示例创建一个客户端，该示例使用协程。

我不太确定应该如何关闭客户端连接。据我所知，有两种方法：

#1

一旦 ready/there 与 read/write 操作一样无事可做，这似乎将其关闭。

asio::signal_set signals(io_context, SIGINT, SIGTERM);
signals.async_wait([&](auto, auto) { io_context.stop(); });

#2

强制关闭？

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;
}

Answer 1

在 asio 提供的示例中，listener 在 io_context thread/thread-pool 中运行，它由 run() 启动并给定一个线程-构建 io_context(1 /* pool of 1 */).

时的池大小

侦听器将使用 acceptor 来侦听来自 io_context 的新连接。 acceptor 将为每个新的 socket 连接创建一个新的 chat_session，并将其交给 chat_room。

因此，要安全地关闭连接，您需要post一个lambda到asio。 asio::post 将从 io_context 个线程中对 lambda 进行排队。

您需要提供正确的 io_context 和 chat_session 拥有的 socket。连接必须从在中关闭 io_context 如下：

// Where "this" is the current chat_session owning the socket
asio::post(io_context_, [this]() { socket_.close(); });

然后 io_context 将关闭连接并调用 chat_session 的任何活动注册的 async_read / async_write 方法，例如 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。这应该用于优雅地退出你的程序或停止整个 asio io_context。

因此您应该使用第二个选项来“关闭 ASIO 中的异步客户端连接”.

如何在 ASIO 中关闭异步客户端连接？

How to close async client connection in ASIO?

c++

asio

#1

#2

客户端代码（未完成）

服务器代码