Rust如何在Linux中处理文件I/O

在Rust中处理文件I/O：核心模块与基础操作

说到Rust中的文件操作，绕不开标准库里的两个核心模块：std::fs和std::io。前者负责文件系统层面的操作，后者则提供了通用的输入输出接口。两者结合，构成了Rust文件处理的基础。下面，我们就通过几个典型的场景，来看看如何用它们完成日常工作。

1. 引入所需的模块

动手之前，先把必要的工具“请”进来。这一步很关键，能让你后续的代码清晰且高效。

use std::fs::{File, OpenOptions};
use std::io::{self, Read, Write};
use std::path::Path;

2. 创建新文件并写入数据

想创建一个文件并往里写点东西？OpenOptions是你的好帮手。它用起来非常灵活，可以精确控制文件的打开模式。

fn main() -> io::Result<()> {
    let path = Path::new("example.txt");
    let mut file = OpenOptions::new()
        .write(true)
        .create(true)
        .open(&path)?;

    file.write_all(b"Hello, world!")?;
    Ok(())
}

看这段代码：.create(true)确保了如果文件不存在就创建它，而.write(true)则打开了写入权限。最后用write_all一次性写入字节数据，干净利落。

3. 读取文件全部内容

读取文件内容也是家常便饭。思路通常是打开文件，准备一个缓冲区，然后把数据读进去。这里选择read_to_end，意味着一次性读取整个文件，适合处理体积不大的文件。

fn main() -> io::Result<()> {
    let path = Path::new("example.txt");
    let mut file = File::open(&path)?;
    let mut contents = Vec::new();

    file.read_to_end(&mut contents)?;
    println!("File contents: {:?}", String::from_utf8_lossy(&contents));
    Ok(())
}

注意String::from_utf8_lossy的用法，它能将字节向量（可能包含无效UTF-8序列）优雅地转换为字符串，对于处理来源不确定的文本数据非常实用。

4. 向现有文件追加数据

有时候，我们不想覆盖旧文件，而是在末尾添加新内容。这时候，就需要把OpenOptions的.append(true)模式用起来了。

fn main() -> io::Result<()> {
    let path = Path::new("example.txt");
    let mut file = OpenOptions::new()
        .write(true)
        .append(true)
        .open(&path)?;

    file.write_all(b"\nAppended text")?;
    Ok(())
}

与创建模式的关键区别就在.append(true)上。它保证所有写入操作都从文件末尾开始，完美避免了意外覆盖原有数据。

5. 逐行读取文件

对于需要按行处理的大文件，一次性读入内存就不太合适了。更优雅的方式是使用BufReader进行缓冲，然后逐行迭代。

use std::io::{self, BufRead};

fn main() -> io::Result<()> {
    let path = Path::new("example.txt");
    let file = File::open(&path)?;
    let reader = io::BufReader::new(file);

    for line in reader.lines() {
        println!("Line: {}", line?);
    }
    Ok(())
}

这里，BufReader::lines()方法返回一个迭代器，每次产生一个io::Result。这种方式内存友好，是处理日志文件、配置文件等场景的标配。

总结与提醒

以上几个示例，基本覆盖了日常文件I/O操作的核心场景。你可以像搭积木一样，根据实际需求组合这些操作。另外，不知你注意到没有，所有示例函数的返回类型都是io::Result<()>，并且大量使用了?操作符进行错误传播。这正是Rust错误处理的精髓所在——它强制你直面可能的I/O错误。在实际项目中，你很可能需要根据业务逻辑，对这些错误进行更精细的捕获和处理，而不是简单地向上传播。这算是Rust给你的一份“甜蜜的负担”吧，习惯了之后，你会发现代码的健壮性大大提升了。