时间戳

首页 > 解决方案 > 长度未知时如何读取数据？

c++ - 长度未知时如何读取数据？

问题描述

我目前正在使用 boost::asio 跨网络连接读取数据，但选择了一种我认为无效的模式：

auto listener::read(std::function<void(std::error_code ec, packet packet)> callback) noexcept -> void {
  m_buffer.resize(1);
  m_buffer.shrink_to_fit();

  asio::async_read(*m_socket, asio::buffer(m_buffer), asio::transfer_exactly(1),
                   [&, callback](std::error_code ec, std::size_t length) {
                     const auto available = m_socket->available();
                     packet tmp;
                     tmp.resize(available);
                     asio::async_read(*m_socket, asio::buffer(tmp), asio::transfer_exactly(available));
                     tmp.insert(tmp.begin(), std::make_move_iterator(m_buffer.begin()),
                                std::make_move_iterator(m_buffer.end()));
                     callback(ec, std::move(tmp));
                   });
}

（packet是std::vector<unsigned char>）

我不确定如何在没有临时的情况下创建它。一开始我无法调整大小m_buffer，因为我不知道有多少数据即将到来。我尝试m_buffer仅在 lambda 中使用调整大小来匹配available+ 1，但我最终丢失了存储在m_buffer.

当期望一个未知长度的数据包时，有没有更有效的方法来做到这一点？

标签： c++c++11boost-asio

解决方案

首先，你不能这样做：

 asio::async_read(*m_socket, asio::buffer(tmp), asio::transfer_exactly(available)); //[1]
 tmp.insert(tmp.begin(), std::make_move_iterator(m_buffer.begin()),
                            std::make_move_iterator(m_buffer.end())); // [2]

在 [1] 中异步操作开始。async_read立即返回。然后我们有两个并发操作，第一个插入到tmp，第二个（异步操作）填充tmp一些数据。您可以使用同步操作：asio::read而不是asio::async_read，前者是阻塞功能，因此insert仅在读取数据时执行。

如果您不想使用连接向量、创建临时对象等，您可以使用boost::asio::dynamic_buffer：

struct listener {
 vector<char> m_buffer;
 // others members
};

void listener::read(std::function<void(std::error_code ec, packet p)> callback) 
{
  boost::asio::async_read(m_socket, boost::asio::dynamic_buffer(m_buffer), boost::asio::transfer_exactly(1),
                                                 ^^^^^^^^^^^^^
             [&, callback](std::error_code ec, std::size_t length) 
             {
                 const auto available = m_socket.available();
                 boost::asio::async_read(m_socket, boost::asio::dynamic_buffer(m_buffer), 
                                                                ^^^^^^^^^^^^^
                      boost::asio::transfer_exactly(available),
                   [this,callback](const boost::system::error_code& ec, size_t)
                   {
                      callback(ec, std::move(m_buffer));
                   });
            });
}

m_buffer由异步操作自动增加。你不手动做。如您所见，我添加了新的处理程序 -> where is called callback(ec,move(m_buffer))。当调用此处理程序时，我们知道读取操作结束。

推荐阅读