为什么带有.chars()的.flat_map()不能与std :: io :: Lines一起使用,而是使用字符串向量?

时间:2016-11-01 00:26:06

标签: iterator rust

我试图在stdin中迭代字符。 Read.chars()方法实现了此目标,但不稳定。显而易见的替代方法是使用Read.lines()flat_map将其转换为字符迭代器。

这似乎应该有效,但不会,导致borrowed value does not live long enough错误。

use std::io::BufRead;

fn main() {
    let stdin = std::io::stdin();
    let mut lines = stdin.lock().lines();
    let mut chars = lines.flat_map(|x| x.unwrap().chars());
}

Read file character-by-character in Rust中提到了这一点,但它并没有真正解释原因。

我特别感到困惑的是,这与documentation for flat_map中的示例有何不同,后者使用flat_map.chars()应用于字符串向量。我真的不明白这应该有什么不同。我看到的主要区别是我的代码也需要调用unwrap(),但是将最后一行更改为以下代码也不起作用:

let mut chars = lines.map(|x| x.unwrap());
let mut chars = chars.flat_map(|x| x.chars());

第二行失败,因此问题似乎不是unwrap

为什么最后一行不起作用,当文档中非常相似的行没有?有没有办法让这个工作?

1 个答案:

答案 0 :(得分:7)

首先弄清楚闭包变量的类型是什么:

let mut chars = lines.flat_map(|x| {
    let () = x;
    x.unwrap().chars()
});

这表明它是Result<String, io::Error>。在unwrap ping它之后,它将是String

接下来,请查看str::chars

fn chars(&self) -> Chars

definition of Chars

pub struct Chars<'a> {
    // some fields omitted
}

由此可以看出,对字符串调用chars会返回一个迭代器,该字符串具有引用

每当我们有一个引用时,我们就知道引用不能比它借来的东西寿命更长。在这种情况下,x.unwrap()是所有者。接下来要检查的是所有权结束的位置。在这种情况下,闭包拥有String,因此在闭包结束时,值将被删除,并且任何引用都将失效。

除了代码尝试返回仍然引用该字符串的Chars。哎呀。感谢Rust,代码没有段错!

与有效示例的区别在于所有权。在这种情况下,字符串由循环外部的向量拥有,并且在消耗迭代器之前不会丢弃它们。因此,没有终身问题。

此代码真正需要的是into_chars上的String方法。迭代器可以获取值的所有权并返回字符。

不是最高效率,而是一个良好的开端:

struct IntoChars {
    s: String,
    offset: usize,
}

impl IntoChars {
    fn new(s: String) -> Self {
        IntoChars { s: s, offset: 0 }
    }
}

impl Iterator for IntoChars {
    type Item = char;

    fn next(&mut self) -> Option<Self::Item> {
        let remaining = &self.s[self.offset..];

        match remaining.chars().next() {
            Some(c) => {
                self.offset += c.len_utf8();
                Some(c)
            }
            None => None,
        }
    }
}

use std::io::BufRead;

fn main() {
    let stdin = std::io::stdin();
    let lines = stdin.lock().lines();
    let chars = lines.flat_map(|x| IntoChars::new(x.unwrap()));

    for c in chars {
        println!("{}", c);
    }
}

另见: