关闭 websocket 并释放文件描述符的最快方法

Question

我正在使用 boost monster 来建立 websocket 连接，我的一个进程可能有大量的流/连接，在终止进程时，我使用以下方法在析构函数中调用阻塞关闭每个 websocket：

                    if ( m_ws.next_layer().next_layer().is_open() )
                    {
                        boost::system::error_code ec;
                        m_ws.close(boost::beast::websocket::close_code::normal, ec);
                    }

拥有大量 websocket，使得终止进程长时间阻塞，有没有办法强制终止（删除）连接并更快地释放底层资源？提前致谢。

Answer 1

正如我在评论中告诉您的那样，关闭连接将是套接字上的快速操作，但它不会花费时间并阻塞线程。在你的例子中，我不知道你的程序做了多少工作来关闭每个套接字，但请记住，如果你的主线程结束，这意味着进程已经结束，SO将释放它一直在使用的所有资源，在不关闭每个套接字的情况下，我使用此技术，WebSocket 客户端会检测连接的结束，但当协议出现问题或远程端点突然断开连接时，我会关闭套接字。

共享代码以查看代码正在执行哪些其他活动可能会很有用。

顺便跟大家分享一下我的代码，我关闭websocket没有问题：

void wscnx::close(beast::websocket::close_code reason, const boost::system::error_code &ec)
{
    // std::cout << "\nwscnx::close\n";
    if (is_ssl && m_wss->is_open())
                     m_wss->close(beast::websocket::normal);
    else if (!is_ssl && m_ws->is_open())
        m_wss->close(beast::websocket::normal);
    
    if (!m_server.expired())
        m_server.lock()->cnx_closed(m_cnx_info, ec);
}

就我而言，我使用的是异步方法读取和同步方法
编写时，我没有使用异步方法来编写，以避免出现两个同时写入操作的情况。

另外，重要的是要说我正在使用异步方式接受新连接。

接受套接字连接的代码，您可以在其中设置写入和读取的 TIMES OUTS，而不是使用计时器。

void wscnx::accept_tcp()
{   
    m_ws->set_option(
        websocket::stream_base::timeout::suggested(
            beast::role_type::server));

    m_ws->set_option(websocket::stream_base::decorator(
        [](websocket::response_type &res)
        {
            res.set(http::field::server,
                    std::string(BOOST_BEAST_VERSION_STRING) +
                        " websocket-server-async");
        }));

    //std::cout << "wscnx::[" << m_id << "]:: TCP async_handshake\n";
    m_ws->async_accept(
        [self{shared_from_this()}](const boost::system::error_code &ec)
        {
            if (ec)
            {
                self->close(beast::websocket::close_code::protocol_error, ec);
                return;
            }
            // self->read_tcp();
            self->read();
        });
}

要读取的代码：

void wscnx::read()
{
    if (!is_ssl && !m_ws->is_open())
        return;
    else if (is_ssl  && !m_wss->is_open())
        return;

    auto f_read = [self{shared_from_this()}](const boost::system::error_code &ec, std::size_t bytes_transferred)
                 {
                    boost::ignore_unused(bytes_transferred);

                    // This indicates that the session was closed
                    if (ec == websocket::error::closed)
                    {
                        self->close(beast::websocket::close_code::normal, ec);
                        return;
                    }

                    if (ec)
                    {
                        self->close(beast::websocket::close_code::abnormal, ec);
                        return;
                    }

                    std::string data = beast::buffers_to_string(self->m_rx_buffer.data());
                    self->m_rx_buffer.consume(bytes_transferred);

                    if (!self->m_server.expired())
                    {
                        std::string_view vdata(data.c_str());
                        self->m_server.lock()->on_data_rx(self->m_cnx_info.id, vdata, self->cnx_info());
                    }
                    self->read(); 
                };//lambda

    if (!is_ssl)
        m_ws->async_read(m_rx_buffer, f_read);
    else
        m_wss->async_read(m_rx_buffer, f_read);
}

要写入的代码：

void wscnx::write(std::string_view data, bool close_on_write)
{
    std::unique_lock<std::mutex> u_lock(m_mtx_write);
    if ( (!is_ssl && !m_ws->is_open()) || (is_ssl && !m_wss->is_open()))
        return;

    boost::system::error_code ec;
    size_t bytes_transferred{0};
    if (is_ssl)
        bytes_transferred = m_wss->write(net::buffer(data), ec);
    else
        bytes_transferred = m_ws->write(net::buffer(data), ec);
    
    boost::ignore_unused(bytes_transferred);

    // This indicates that the session was closed
    if (ec == websocket::error::closed)
    {
        // std::cout << "[wscnx::[" << self->m_id << "]::on wirite] Error: " << ec.message() << "\n";
        close(beast::websocket::close_code::normal, ec);
        return;
    }

    if (ec)
    {
        // std::cout << "[wscnx::[" << self->m_id << "]::on wirite] Error READING: " << ec.message() << "\n";
        close(beast::websocket::close_code::abnormal, ec);
        return;
    }

    if (close_on_write)
        close(beast::websocket::close_code::normal, ec);
}

如果您想查看整个代码，这是链接。该项目仍处于开发阶段，但已经有效。