当前位置：首页 > news >正文

Rust 文件I/O操作实战：高效处理文件系统

news 2026/5/6 22:03:53

Rust 文件I/O操作实战：高效处理文件系统

文件I/O的重要性

文件I/O是编程中最基本的操作之一，几乎所有的应用程序都会涉及到文件的读写操作。无论是配置文件、日志文件，还是数据存储，文件I/O都是不可避免的。在Rust中，文件I/O操作既安全又高效，提供了丰富的API来处理各种文件系统操作。

基本文件操作

读取文件

use std::fs::File; use std::io::{self, Read}; fn main() -> io::Result<()> { // 打开文件 let mut file = File::open("hello.txt")?; // 读取文件内容 let mut contents = String::new(); file.read_to_string(&mut contents)?; println!("文件内容: {}", contents); Ok(()) }

写入文件

use std::fs::File; use std::io::{self, Write}; fn main() -> io::Result<()> { // 创建或截断文件 let mut file = File::create("hello.txt")?; // 写入内容 file.write_all(b"Hello, World!")?; println!("文件写入成功"); Ok(()) }

追加内容

use std::fs::OpenOptions; use std::io::{self, Write}; fn main() -> io::Result<()> { // 打开文件，设置为追加模式 let mut file = OpenOptions::new() .append(true) .create(true) .open("hello.txt")?; // 追加内容 writeln!(file, "追加的内容")?; println!("内容追加成功"); Ok(()) }

异步文件I/O

在处理大量文件或需要非阻塞操作时，异步文件I/O是一个很好的选择。Rust的tokio库提供了异步文件I/O的支持。

异步读取文件

use tokio::fs::File; use tokio::io::{self, AsyncReadExt}; #[tokio::main] async fn main() -> io::Result<()> { // 打开文件 let mut file = File::open("hello.txt").await?; // 读取文件内容 let mut contents = String::new(); file.read_to_string(&mut contents).await?; println!("文件内容: {}", contents); Ok(()) }

异步写入文件

use tokio::fs::File; use tokio::io::{self, AsyncWriteExt}; #[tokio::main] async fn main() -> io::Result<()> { // 创建或截断文件 let mut file = File::create("hello.txt").await?; // 写入内容 file.write_all(b"Hello, World!").await?; println!("文件写入成功"); Ok(()) }

文件元数据

获取文件元数据

use std::fs::File; use std::os::unix::fs::MetadataExt; fn main() -> std::io::Result<()> { let file = File::open("hello.txt")?; let metadata = file.metadata()?; println!("文件大小: {} 字节", metadata.len()); println!("是否是文件: {}", metadata.is_file()); println!("是否是目录: {}", metadata.is_dir()); println!("权限: {:?}", metadata.permissions()); // Unix特有的元数据 println!("inode: {}", metadata.ino()); println!("设备ID: {}", metadata.dev()); Ok(()) }

检查文件是否存在

use std::path::Path; fn main() { let path = Path::new("hello.txt"); if path.exists() { println!("文件存在"); } else { println!("文件不存在"); } if path.is_file() { println!("是文件"); } else if path.is_dir() { println!("是目录"); } }

目录操作

创建目录

use std::fs; fn main() -> std::io::Result<()> { // 创建单个目录 fs::create_dir("test")?; // 创建嵌套目录 fs::create_dir_all("test/nested/dir")?; println!("目录创建成功"); Ok(()) }

读取目录内容

use std::fs; fn main() -> std::io::Result<()> { let entries = fs::read_dir(".")?; for entry in entries { let entry = entry?; let path = entry.path(); if path.is_file() { println!("文件: {:?}", path.file_name().unwrap()); } else if path.is_dir() { println!("目录: {:?}", path.file_name().unwrap()); } } Ok(()) }

删除文件和目录

use std::fs; fn main() -> std::io::Result<()> { // 删除文件 fs::remove_file("hello.txt")?; // 删除空目录 fs::remove_dir("test")?; // 删除目录及其所有内容 fs::remove_dir_all("test")?; println!("删除成功"); Ok(()) }

复制文件

use std::fs; fn main() -> std::io::Result<()> { fs::copy("source.txt", "destination.txt")?; println!("文件复制成功"); Ok(()) }

移动文件

use std::fs; fn main() -> std::io::Result<()> { fs::rename("old_name.txt", "new_name.txt")?; println!("文件移动成功"); Ok(()) }

高级文件操作

内存映射

内存映射是一种将文件内容映射到内存的技术，可以提高大文件的读写性能。

use memmap2::Mmap; use std::fs::File; fn main() -> std::io::Result<()> { let file = File::open("large_file.txt")?; let mmap = unsafe { Mmap::map(&file)? }; // 直接从内存中读取数据 let content = &mmap[..100]; println!("文件内容: {}", String::from_utf8_lossy(content)); Ok(()) }

读取大文件

对于大文件，我们可以使用缓冲区来分块读取，避免一次性加载整个文件到内存。

use std::fs::File; use std::io::{self, BufReader, Read}; fn main() -> io::Result<()> { let file = File::open("large_file.txt")?; let mut reader = BufReader::new(file); let mut buffer = [0; 1024]; // 1KB缓冲区 loop { let bytes_read = reader.read(&mut buffer)?; if bytes_read == 0 { break; } // 处理读取的数据 println!("读取了 {} 字节", bytes_read); } Ok(()) }

写入大文件

use std::fs::File; use std::io::{self, BufWriter, Write}; fn main() -> io::Result<()> { let file = File::create("large_file.txt")?; let mut writer = BufWriter::new(file); for i in 0..1000000 { writeln!(writer, "Line {}", i)?; } // 确保所有数据都写入磁盘 writer.flush()?; println!("大文件写入成功"); Ok(()) }

实用应用

配置文件管理

use serde::{Deserialize, Serialize}; use std::fs; #[derive(Debug, Deserialize, Serialize)] struct Config { server: ServerConfig, database: DatabaseConfig, } #[derive(Debug, Deserialize, Serialize)] struct ServerConfig { host: String, port: u16, } #[derive(Debug, Deserialize, Serialize)] struct DatabaseConfig { url: String, username: String, password: String, } fn read_config() -> std::io::Result<Config> { let content = fs::read_to_string("config.toml")?; let config: Config = toml::from_str(&content)?; Ok(config) } fn write_config(config: &Config) -> std::io::Result<()> { let content = toml::to_string_pretty(config)?; fs::write("config.toml", content)?; Ok(()) } fn main() -> std::io::Result<()> { // 读取配置 let mut config = read_config()?; println!("当前配置: {:?}", config); // 修改配置 config.server.port = 8080; // 写入配置 write_config(&config)?; println!("配置已更新"); Ok(()) }

日志文件管理

use log::{debug, error, info, trace, warn}; use simplelog::{ColorChoice, Config, LevelFilter, TermLogger, TerminalMode}; use std::fs::OpenOptions; use std::io::Write; use std::path::Path; fn setup_logger() { TermLogger::init( LevelFilter::Debug, Config::default(), TerminalMode::Mixed, ColorChoice::Auto, ) .unwrap(); } fn log_to_file(message: &str) -> std::io::Result<()> { let log_file = Path::new("app.log"); let mut file = OpenOptions::new() .append(true) .create(true) .open(log_file)?; writeln!(file, "{}", message)?; Ok(()) } fn main() -> std::io::Result<()> { setup_logger(); info!("应用启动"); log_to_file("应用启动")?; debug!("调试信息"); log_to_file("调试信息")?; warn!("警告信息"); log_to_file("警告信息")?; error!("错误信息"); log_to_file("错误信息")?; info!("应用退出"); log_to_file("应用退出")?; Ok(()) }

CSV文件处理

use csv::{ReaderBuilder, WriterBuilder}; use serde::{Deserialize, Serialize}; #[derive(Debug, Deserialize, Serialize)] struct Person { name: String, age: u32, city: String, } fn read_csv() -> std::io::Result<()> { let file = std::fs::File::open("people.csv")?; let mut reader = ReaderBuilder::new().from_reader(file); for result in reader.deserialize::<Person>() { match result { Ok(person) => println!("{:?}", person), Err(e) => eprintln!("错误: {}", e), } } Ok(()) } fn write_csv() -> std::io::Result<()> { let file = std::fs::File::create("people.csv")?; let mut writer = WriterBuilder::new().from_writer(file); let people = vec![ Person { name: "Alice".to_string(), age: 30, city: "New York".to_string() }, Person { name: "Bob".to_string(), age: 25, city: "London".to_string() }, Person { name: "Charlie".to_string(), age: 35, city: "Paris".to_string() }, ]; for person in people { writer.serialize(person)?; } writer.flush()?; Ok(()) } fn main() -> std::io::Result<()> { write_csv()?; read_csv()?; Ok(()) }