可重载的 boost::asio::basic_stream_socket

Question

开发网络应用程序时，我有一个 Connection 类来管理在网络上发送和接收消息。我正在使用 boost::asio。

我现在想让 Connection 类处理通过 TCP 和本地 UNIX 流套接字的连接。然而，boost 的模板设计让我感到困惑。 AFAICT，local::stream_protocol::socket 和 ip::tcp::socket 之间没有共享基类。

我将如何创建一个封装网络语义的连接，以便其他代码不必处理所使用协议的细节？

IE 我想实现类似的东西：

class Connection() {
  Connection(ip::tcp::endpoint& ep);
  Connection(local::stream_protocol::endpoint& ep);

  void send(Buffer& buf);
}

我将如何实现这一点？

Answer 1

经过一番思考，我目前的解决方案是将Connection的send和recv函数虚拟化，并创建一个Connection的模板子类，大致是：

template <typename Protocol>
class ConnectionImpl : public Connection {
    typedef typename Protocol::socket Socket;
    typedef typename Protocol::endpoint EndPoint;

    Socket _socket;
public:
    ConnectionImpl(boost::asio::io_service& ioSvc, const EndPoint& addr) 
        : Connection(ioSvc), _socket(ioSvc) { 
        _socket.connect(addr);
    }

    void trySend() {
        // Initiate async send on _socket here
    }

    void tryRead() {
        // Initiate async recv on _socket here
    }
}

有没有办法避免是否需要子类化和使用虚函数？

Answer 2

AFAICT，之间没有共享基类
local::stream_protocol::socket 和 ip::tcp::socket。

所有套接字对象都没有明确地指定基类，文档很好地描述了原理

不包括 BSD 套接字 API 的不安全和容易出错的方面。为了
例如，使用 int 来表示所有套接字就缺乏类型安全性。
Boost.Asio 中的套接字表示对每个套接字使用不同的类型
协议，例如对于 TCP，使用 ip::tcp::socket，对于 UDP，使用 ip::tcp::socket
使用 ip::udp::socket

Answer 3

请改用 boost::asio:generic::stream_protocol::socket 。当您调用 async_connect()/connect() 时，它将从远程端点提取系列和协议，然后将它们传递给 socket() 系统调用以创建正确的套接字。

boost::asio::generic::stream_protocol::socket socket_{io_service};

if (use_unix_socket) {
    boost::asio::local::stream_protocol::endpoint unix_endpoint{"/tmp/socketpath.sock"};
    socket_.async_connect(unix_endpoint, [](boost::system::error_code ec){

    }};
}
else {
    boost::asio::ip::tcp::endpoint tcp_endpoint{...};
    socket_.async_connect(tcp_endpoint, [](boost::system::error_code ec){

    }};
}

还有来自 boost::asio::basic_socket 的代码：

template <typename ConnectHandler>
  BOOST_ASIO_INITFN_RESULT_TYPE(ConnectHandler,
      void (boost::system::error_code))
  async_connect(const endpoint_type& peer_endpoint,
      BOOST_ASIO_MOVE_ARG(ConnectHandler) handler)
  {
    // If you get an error on the following line it means that your handler does
    // not meet the documented type requirements for a ConnectHandler.
    BOOST_ASIO_CONNECT_HANDLER_CHECK(ConnectHandler, handler) type_check;

    if (!is_open())
    {
      boost::system::error_code ec;
      const protocol_type protocol = peer_endpoint.protocol();
      if (this->get_service().open(this->get_implementation(), protocol, ec))
      {
        detail::async_result_init<
          ConnectHandler, void (boost::system::error_code)> init(
            BOOST_ASIO_MOVE_CAST(ConnectHandler)(handler));

        this->get_io_service().post(
            boost::asio::detail::bind_handler(
              BOOST_ASIO_MOVE_CAST(BOOST_ASIO_HANDLER_TYPE(
                ConnectHandler, void (boost::system::error_code)))(
                  init.handler), ec));

        return init.result.get();
      }
    }

    return this->get_service().async_connect(this->get_implementation(),
        peer_endpoint, BOOST_ASIO_MOVE_CAST(ConnectHandler)(handler));
  }