为什么我的迭代器中的字符串被连接起来?

时间:2017-02-07 23:58:41

标签: iterator rust type-inference

我最初的目标是获取一个单词列表,每行一个,并将它们放在HashSet中,同时放弃注释行并正确地引发I / O错误。给定文件" stopwords.txt":

a
# this is actually a comment
of
the
this

我设法使代码编译如下:

fn stopword_set() -> io::Result<HashSet<String>> {
    let words = Result::from_iter(
        BufReader::new(File::open("stopwords.txt")?)
                .lines()
                .filter(|r| match r {
                    &Ok(ref l) => !l.starts_with('#'),
                    _ => true
                }));
    Ok(HashSet::from_iter(words))
}

fn main() {
    let set = stopword_set().unwrap();
    println!("{:?}", set);
    assert_eq!(set.len(), 4);
}

此处还有一个playground,它也会创建上面的文件。

我希望在程序结束时有一组4个字符串。令我惊讶的是,该函数实际上返回一个包含单个字符串的集合,其中所有单词都连接在一起:

{"aofthethis"}
thread 'main' panicked at 'assertion failed: `(left == right)` (left: `1`, right: `4`)'

FromIterator文档中的一条建议的引导下,我删除了对from_iter的所有调用,并使用了collect代替Playground)解决了这个问题。

fn stopword_set() -> io::Result<HashSet<String>> {
    BufReader::new(File::open("stopwords.txt")?)
            .lines()
            .filter(|r| match r {
                &Ok(ref l) => !l.starts_with('#'),
                _ => true
            }).collect()
}

为什么以前拨打from_iter会导致意外推断,而collect()的工作方式与预期一致?

1 个答案:

答案 0 :(得分:8)

更简单的再现:

use std::collections::HashSet;
use std::iter::FromIterator;

fn stopword_set() -> Result<HashSet<String>, u8> {
    let input: Vec<Result<_, u8>> = vec![Ok("foo".to_string()), Ok("bar".to_string())];
    let words = Result::from_iter(input.into_iter());
    Ok(HashSet::from_iter(words))
}

fn main() {
    let set = stopword_set().unwrap();
    println!("{:?}", set);
    assert_eq!(set.len(), 2);
}

问题在于,我们从迭代器中收集两次。 words的类型为Result<_, u8>。但是,Result 实现Iterator本身,所以当我们在最后调用from_iter时,编译器会看到Ok类型必须由于方法签名而为String。向后工作,您可以从String的迭代器构造Strings,这就是编译器选择的内容。

删除第二个from_iter可以解决它:

fn stopword_set() -> Result<HashSet<String>, u8> {
    let input: Vec<Result<_, u8>> = vec![Ok("foo".to_string()), Ok("bar".to_string())];
    Result::from_iter(input.into_iter())
}

或者原着:

fn stopword_set() -> io::Result<HashSet<String>> {
    Result::from_iter(
        BufReader::new(File::open("stopwords.txt")?)
                .lines()
                .filter(|r| match r {
                    &Ok(ref l) => !l.starts_with('#'),
                    _ => true
                }))
}

当然,我通常建议使用collect,因为我更喜欢链接:

fn stopword_set() -> io::Result<HashSet<String>> {
    BufReader::new(File::open("stopwords.txt")?)
        .lines()
        .filter(|r| match r {
            &Ok(ref l) => !l.starts_with('#'),
            _ => true,
        })
        .collect()
}