如何为 UnixStream 和 TcpStream 创建多态类型

Question

我有一个名为 Connection 的结构，它接收 tokio::net::TcpStream 或 tokio::net::UnixStream 以与远程服务器通信。鉴于 Rust 没有构造函数，我向我的结构添加了一个静态 new() 方法，并在将流实例传递给 Connection 的新实例并将其从 new() 返回给我的最终用户之前在那里执行身份验证握手。

我的问题是如何创建一个可以设置为 UnixStream 或 TcpStream 的临时文件，以便我可以在进行握手的消息交换期间对其进行操作。由于 UnixStream 和 TcpStream 没有共同的父级，我不知道如何实现这一点：

pub struct Configuration {
  tcp_socket: Option<SocketAddr>,
  unix_socket: Option<PathBuf>,
}

pub(crate) struct Connection {
  tcp: Option<TcpStream>,
  unix: Option<UnixStream>,
}

impl Connection {

  pub(crate) async fn new(configuration: &Configuration) -> Result<Connection, Box<dyn std::error::Error>> {
    let mut stream;
    if configuration.unix_socket().is_some() {
      stream = UnixStream::connect(configuration.unix_socket().unwrap()).await?;
    } else {
      stream = TcpStream::connect(configuration.tcp_socket().unwrap()).await?;
    }
    // Handshake code goes here...

    let conn = Connection {
      tcp: Some(stream),
      unix: None,
    };

    Ok(conn)
}

Answer 1

您可以使用这样的枚举：

pub(crate) enum Connection{
  Tcp(TcpStream),
  Unix(UnixStream),
}

和 match 您可以在任何地方访问它或为具有通用功能的两个流实现一个特性。这个特征可以看作是一个泛型，它会在编译时评估它是哪个版本。

最小示例：

pub(crate) trait ExampleStream{
  fn connect<A>(address: A) -> Result<Self, Error>;
}

枚举选项最接近您目前所写的内容，因此我建议您这样做。

如何为 UnixStream 和 TcpStream 创建多态类型

How to create a polymorphic type for UnixStream and TcpStream

rust

rust-tokio