generics - 如何结合 std::str::lines 和 std::io::lines？

Question

我想写一个函数来解析文本，但文本可能来自外部文件或内部&str。 parse 函数可能是这样的:

fn parse(lines: GenericLinesGenerator) {
   for line in lines {
       // parse content
   }
}

...它可以像这样被调用:

use std::io::BufReader;
use std::fs::File;
let fin = BufReader::new(File::open("info.txt").expect("not found"));
parse(TransformFromIO(fin.lines()))

或

let content: &'static str = "some\nlong\ntext";
parse(TransformFromStr(content.lines()))

是否可以实现这样的parse功能？

Answer 1

两个迭代器不产生相同的值:

impl<B: BufRead> Iterator for io::Lines<B> {
    type Item = Result<String>;
}

impl<'a> Iterator for str::Lines<'a> {
    type Item = &'a str;
}

您必须以某种方式处理这种差异。最重要的区别是 io::Lines 可以失败。您的程序必须决定如何处理它；我选择了中止程序。

接下来您需要做的是接受任何可以转换为迭代器的类型，并且迭代器产生的值必须转换为您可以处理的类型。 &str 似乎是共同点。

这可以通过使用 IntoIterator 来解决和 Borrow :

use std::borrow::Borrow;
use std::fs::File;
use std::io::prelude::*;
use std::io::BufReader;

fn parse<I>(lines: I)
where
    I: IntoIterator,
    I::Item: Borrow<str>,
{
    for line in lines {
        println!("line: {}", line.borrow());
    }
}

fn main() {
    parse("alpha\nbeta\ngamma".lines());

    println!("----");

    let f = File::open("/etc/hosts").expect("Couldn't open");
    let b = BufReader::new(f);
    parse(b.lines().map(|l| l.expect("Bad line!")));
}

检查 The Rust Programming Language关于 where clauses 的部分有关特质界限的更多信息。