读取文件并获取字符串数组

Question

我想读一个文件并找回Strings的矢量。以下功能有效，但有更简洁或惯用的方式吗？

use std::fs::File;
use std::io::Read;

fn lines_from_file(filename: &str) -> Vec<String> {
    let mut file = match File::open(filename) {
        Ok(file) => file,
        Err(_) => panic!("no such file"),
    };
    let mut file_contents = String::new();
    file.read_to_string(&mut file_contents)
        .ok()
        .expect("failed to read!");
    let lines: Vec<String> = file_contents.split("\n")
        .map(|s: &str| s.to_string())
        .collect();
    lines
}

有些事情对我来说似乎不是最理想的：

• 两个单独的错误检查用于读取文件。
• 将整个文件读取到String，它将被丢弃。如果我只想要前N行，这将特别浪费。
• 每行制作一个&str，将被丢弃，而不是以某种方式从文件直接到每行String。

如何改进？

Answer 1

作为BurntSushi said，你可以使用the lines() iterator。但是，按原样解决您的问题：

• 你应该阅读Error Handling in Rust;那些unwrap()s应该变成?s，功能的结果成为一些合理的Result<Vec<String>, E>的E。在这里，我们重用io::Result类型别名。
• 使用lines()迭代器。您可以做的另一件事是将整个文件读入String并返回;有a lines() iterator for strings as well。
• 这个你不能做任何事情：file_contents拥有它的内容，你不能将它们分成多个拥有的Strings。你唯一能做的就是借用每一行的内容，然后将其转换成新的String。也就是说，你这样做的方式意味着你相信创造一个&str是昂贵的;事实并非如此。它实际上只是计算一对偏移量并返回它们。 &str切片实际上等同于(*const u8, usize)。

这是一个基本相同的修改版本：

use std::fs::File;
use std::io::{self, BufRead};
use std::path::Path;

fn lines_from_file<P>(filename: P) -> io::Result<io::Lines<io::BufReader<File>>>
where
    P: AsRef<Path>,
{
    let file = File::open(filename)?;
    Ok(io::BufReader::new(file).lines())
}

我做的另一个变化是：filename现在是一个通用的P: AsRef<Path>，因为这是File::open想要的，所以它将接受更多类型而不需要转换。

Answer 2

DK.'s answer非常正确并且有很好的解释。但是，你说：

读取文件并获取字符串数组

Rust数组有一个固定的长度，在编译时已知，所以我假设你的意思是“vector”。我会这样写：

use std::{
    fs::File,
    io::{prelude::*, BufReader},
    path::Path,
};

fn lines_from_file(filename: impl AsRef<Path>) -> Vec<String> {
    let file = File::open(filename).expect("no such file");
    let buf = BufReader::new(file);
    buf.lines()
        .map(|l| l.expect("Could not parse line"))
        .collect()
}

// ---

fn main() {
    let lines = lines_from_file("/etc/hosts");
    for line in lines {
        println!("{:?}", line);
    }
}

1. 与其他答案一样，使用为文件名实现AsRef的泛型类型是值得的。
2. Result::expect缩短了对Err的恐慌。
3. BufRead::lines处理多种类型的换行，而不仅仅是"\n"。
4. BufRead::lines也给你单独分配的Strings，而不是一个大的水珠。
5. 没有理由收集临时变量只是为了返回它。特别没有理由重复这种类型（Vec<String>）。

如果你想在失败时返回Result，你可以根据需要将实现压缩到一行：

use std::{
    fs::File,
    io::{self, BufRead, BufReader},
    path::Path,
};

fn lines_from_file(filename: impl AsRef<Path>) -> io::Result<Vec<String>> {
    BufReader::new(File::open(filename)?).lines().collect()
}

// ---

fn main() {
    let lines = lines_from_file("/etc/hosts").expect("Could not load lines");
    for line in lines {
        println!("{:?}", line);
    }
}