天天看点

010 Rust 异步编程,使用 select 宏的条件

使用select宏

  • select中使用的Future必须实现Unpin trait和FusedFuture trait。
  • 必须实现unpin的原因是,select中使用的future不是按值获取的,而是按照可变引用获取的,通过不获取future的所有权,所以在调用select后,未完成的future可以继续使用。
  • 必须实现FusedFuture的原因,select完成后不会再轮询future,因此需要实现FusedFuture来跟踪future是否完成。
  • 同样的,对应到stream上,会有一个FusedStream trait。
use futures::{
    stream::{Stream, StreamExt, FusedStream},
    select,
};

async fn add_two_streams(
    mut s1: impl Stream<Item = u8> + FusedStream + Unpin,
    mut s2: impl Stream<Item = u8> + FusedStream + Unpin,
) -> u8 {
    let mut total = 0;

    loop {
        let item = select! {
            x = s1.next() => x,
            x = s2.next() => x,
            complete => break,
        };
        if let Some(next_num) = item {
            total += next_num;
        }
    }

    total
}      

其它

  • Fuse::terminated()允许构建一个已经终止的空的Future;
  • 当需要同时运行同一Future的多个副本时,可以使用FuturesUnordered类型。