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

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，对于一些合理的E，函数的结果变成Result<Vec<String>, 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 数组有一个固定的长度，在编译时就知道了，所以我假设你的意思是“向量”。我会这样写：

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);
    }
}