如何查明通过 Boost ASIO TCP 套接字连接发送了多少数据？

Question

我有一些指向 boost::asio::io_service、boost::asio::ip::tcp::endpoint、boost::asio: 的共享指针:ip::tcp::acceptor 和 boost::asio::ip::tcp::socket。我接受用户连接并将套接字的shared_ptr传递给我的某个类。它发挥作用。

现在我想要的很简单——统计我的流量。我想获取有关该连接期间发送和接收的数据量的信息。如何从接受的 Boost ASIO TCP 套接字获取此类信息？

Answer 1

假设您使用异步方法，为 async_read 提供的完成处理程序将指示接收到的字节数。类似地，提供给async_write的完成处理程序将指示写入的字节数。将正在运行的计数器维护为类的成员是很简单的，在该类中将方法绑定为前面描述的完成处理程序。

#include <boost/asio.hpp>

#include <iostream>

class Socket
{
public:
    Socket(
            boost::asio::io_service& io_service
          ) :
        _socket( io_service ),
        _counter( 0 )
    {

    }

    void readHandler(
            const boost::system::error_code& error,
            std::size_t bytes_transferred
            )
    {
        _counter += bytes_transferred;
    }

    void writeHHandler(
            const boost::system::error_code& error,
            std::size_t bytes_transferred
            )
    {
        _counter += bytes_transferred;
    }

private:
    boost::asio::ip::tcp::socket _socket;
    std::size_t _counter;
};

int
main()
{
    boost::asio::io_service io_service;
    Socket foo( io_service );
}