在异步 Function 参数中声明相关的特征类型 Object

Question

我想要一个 function 异步处理可变数量的（ Sink ， Stream ）元组。

use futures::channel::mpsc;
use futures::{Sink, Stream, SinkExt, StreamExt};

async fn foo(v: Vec<(Box<dyn Sink<Error = std::io::Error>>, Box<dyn Stream<Item = u8>>)>) {
    for (mut tx, mut rx) in v {
        let _ = tx.send(0);
        let _ = rx.next().await;
    }
}

#[tokio::main]
pub async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let (tx, mut rx) = mpsc::channel(32);
    foo(vec![(Box::new(tx), Box::new(rx))]).await;
    
    Ok(())
}

但我得到这个编译错误：

error[E0107]: wrong number of type arguments: expected 1, found 0
 --> src/main.rs:4:30
  |
4 | async fn foo(v: Vec<(Box<dyn Sink<Error = std::io::Error>>, Box<dyn Stream<Item = u8>>)>) {
  |                              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^ expected 1 type argument

编译器本身提示我以这种方式声明特征 object 的关联类型。 我不确定它为什么不接受它。

Answer 1

编译器希望您指定Sink的“类型参数”。 这不是错误类型，而是发送到接收器的项目的类型，如Sink<Foo> 。 您将u8指定为 stream 的类型，并且在一个和另一个之间发送不变的值，因此您可能需要一个Sink<u8> 。

一旦你这样做了，编译器接下来会抱怨你需要指定Error关联类型（这次是真实的）。 但是，如果您指定std::io::Error ，则从main()对foo()的调用将无法编译，因为mpsc::Sender的Sink实现将其自己的mpsc::SendError指定为错误类型。

最后，接收器和 stream 都需要固定，以便它们可以跨越等待点。 这是通过使用Pin<Box<...>>而不是Box<...>和Box::pin(...)而不是Box::new(...)来完成的。

通过上述更改，编译后的版本如下所示：

use futures::channel::mpsc;
use futures::{Sink, SinkExt, Stream, StreamExt};
use std::pin::Pin;

async fn foo(
    v: Vec<(
        Pin<Box<dyn Sink<u8, Error = mpsc::SendError>>>,
        Pin<Box<dyn Stream<Item = u8>>>,
    )>,
) {
    for (mut tx, mut rx) in v {
        let _ = tx.send(0);
        let _ = rx.next().await;
    }
}

#[tokio::main]
pub async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let (tx, rx) = mpsc::channel(32);
    foo(vec![(Box::pin(tx), Box::pin(rx))]).await;

    Ok(())
}