天天看點

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類型。