c++ - 为异步发送缓冲区保留内存( boost asio 套接字)

Question

我正在尝试将即发即弃 UDP 发送函数的实现从同步更改为异步。

当前简化的同步函数如下所示:

ssize_t UDPTransport::send_to(const char * buffer, size_t bufferSize) {
    return mPSocket->send_to(boost::asio::buffer(buffer, bufferSize), mOutputEndpoint);
}

我已经设置了一个 thread_group 并且 io_service::run() 设置为使用它。但是，问题是我无法保证 buffer 在此调用完成后仍然存在。我需要存储缓冲区的内容，然后知道它什么时候空闲，以便我以后可以重新使用它或删除它。下面很简单，但是如果我触发两个 send_to 调用，那么我不能保证 handle_send 会以相同的顺序被调用，我可能会 pop 仍然需要的东西!

ssize_t UDPTransport::send_to(const char * buffer, size_t bufferSize) {

    boost::asio::mutable_buffer b1 = boost::asio::buffer(buffer,bufferSize);
    mMutex.lock();
    mQueue.push(b1);

    mPSocket->async_send_to(mQueue.back(), mOutputEndpoint,
                            boost::bind(&UDPTransport::handle_send, this,
                                        boost::asio::placeholders::error,
                                        boost::asio::placeholders::bytes_transferred));

    mMutex.unlock();
    return bufferSize;
}

void UDPTransport::handle_send(const boost::system::error_code& error,
                               std::size_t bytes_transferred)
{
    mMutex.lock();
    mQueue.pop();
    mMutex.unlock();
}

什么是存储异步缓冲区的好方法，然后在不再需要时清理它？

Reading online下面可能是一种更简单的方法，但我不知道我是否相信它。为什么共享指针会决定在调用处理程序之前不取消分配自身？

ssize_t UDPTransport::send_to(const char * buffer, size_t bufferSize) 
{
    auto buf = std::make_shared<std::string>(buffer, bufferSize);
    mPSocket->async_send_to(boost::asio::buffer(*buf), mOutputEndpoint,
                            boost::bind(&UDPTransport::handle_send, this,
                                        boost::asio::placeholders::error,
                                        boost::asio::placeholders::bytes_transferred));
    return bufferSize;
}

Answer 1

我通常做的是将它包装在一个继承自 std::enable_shared_from_this<> 的类中，如下所示:

class Sender : public std::enable_shared_from_this<Sender> {
 public:
  using CompletionHandler =
      std::function<void(const boost::system::error_code& ec,
                         size_t bytes_transferred,
                         std::shared_ptr<Sender> sender)>;

  ~Sender() = default;

  template<typename... Args>
  static std::shared_ptr<Sender> Create(Args&&... args) {
    return std::shared_ptr<Sender>(new Sender(std::forward<Args>(args)...));
  }

  void AsyncSendTo(const char* buffer, size_t buffer_size,
                   CompletionHandler completion_handler) {
    data_.append(buffer, buffer_size);
    socket.async_send_to(
        boost::asio::buffer(data_), endpoint_,
        [self = shared_from_this(),
         completion_handler = std::move(completion_handler)]
        (const boost::system::error_code& ec,
         size_t bytes_transferred) mutable {
          completion_handler(ec, bytes_transferred, std::move(self));
        });
  }

 private:
  Sender() = default;
  Sender(const Sender&) = delete;
  Sender(Sender&&) = delete;
  Sender& operator=(const Sender&) = delete;
  Sender& operator=(Sender&&) = delete;

  SocketType socket_;
  EndpointType endpoint_;
  std::string data_;
}

显然，您必须保证 completion_handler的一生。但除此之外，完成处理程序将返回一个有效的 std::shared_ptr<Sender>。无论何时完成，您都可以使用发送者携带的数据做任何您需要的事情。

在您发布的示例中，buf会离开作用域并在 send_to 被销毁返回，除非您首先在 bind 中捕获它.

脚注 1:那些 std::move() s 可能需要删除，具体取决于您的编译器在涉及 lambda 时是否兼容 C++14。

脚注 2:远离 bind除非您绝对需要利用其动态特性。