如何将Stream :: map与返回Result的函数一起使用?

问题描述 投票:-1回答:2

我有以下代码(请参阅playground):

use futures::{stream, Future, Stream}; // 0.1.25
use std::num::ParseIntError;

fn into_many(i: i32) -> impl Stream<Item = i32, Error = ParseIntError> {
    stream::iter_ok(0..i)
}

fn convert_to_string(number: i32) -> Result<String, ParseIntError> {
    Ok(number.to_string())
}

fn main() {
    println!("start:");
    let vec = into_many(10)
        .map(|number| convert_to_string(number))
        .collect()
        .wait()
        .unwrap();

    println!("vec={:#?}", vec);

    println!("finish:");
}

它输出以下(即Vec<Result<i32, ParseIntError>>):

start:
vec=[
    Ok(
        "0"
    ),
    Ok(
        "1"
    ),
    Ok(
        "2"
    ), ...

有没有办法让它输出一个Vec<i32>,如果发生任何错误,立即停止执行并从函数返回(例如,像this example)?

注意:我确实想使用use futures::Stream; // 0.1.25,即使它对这个特定的例子没有意义。

rust future
2个回答
1
投票

以下代码(playground link)作为您问题中当前代码的修改,可获得您想要的结果:

use futures::{stream, Future, Stream}; // 0.1.25
use std::num::ParseIntError;

fn into_many(i: i32) -> impl Stream<Item = i32, Error = ParseIntError> {
    stream::iter_ok(0..i)
}

fn convert_to_string(number: i32) -> Result<String, ParseIntError> {
    Ok(number.to_string())
}

fn main() {
    println!("start:");
    let vec: Result<Vec<String>, ParseIntError> = into_many(10)
        .map(|number| convert_to_string(number))
        .collect()
        .wait()
        .unwrap()
        .into_iter()
        .collect();

    println!("vec={:#?}", vec);

    println!("finish:");
}

由于您当前的代码返回了Vec,我们可以将其转换为迭代器并将其收集到您想要的类型中。需要输入注释,以便collect知道要将迭代器收集到哪种类型。

请注意,collect特征上的Iterator方法不应与collect上的Stream方法混淆。

最后,虽然这有效,但它可能并不完全符合您的要求,因为在使用collect转换向量之前,它仍然会等待将流中的所有结果收集到向量中。我没有期货的经验所以不确定这是多么可能(它可能是但可能需要一个不太整洁的函数式编程风格的解决方案)。


0
投票

map具有返回Result的函数

不要这样做,那不是你应该使用map。相反,使用and_then

let vec = into_many(10)
    .and_then(|number| convert_to_string(number))
    .collect()
    .wait()
    .unwrap();

在潜入期货之前,您应该使用更简单的Rust概念,如OptionResult和迭代器。转移了许多概念。

也可以看看:

© www.soinside.com 2019 - 2024. All rights reserved.