Rust - 文件输入/输出


除了读取和写入控制台之外,Rust 还允许读取和写入文件。

File 结构代表一个文件。它允许程序对文件执行读写操作。File 结构中的所有方法都返回 io::Result 枚举的变体。

下表列出了 File 结构体的常用方法 -

先生编号 模块 方法 签名 描述
1 std::fs::文件 打开() pub fn open<P: AsRef>(path: P) -> 结果 open 静态方法可用于以只读模式打开文件。
2 std::fs::文件 创造() pub fn create<P: AsRef>(path: P) -> 结果 静态方法以只写模式打开文件。如果文件已经存在,则旧内容将被销毁。否则,将创建一个新文件。
3 std::fs::remove_file 删除文件() pub fn remove_file<P: AsRef>(路径: P) -> 结果<()> 从文件系统中删除文件。不保证该文件会被立即删除。
4 std::fs::OpenOptions 附加() pub fn append(&mut self, 附加: bool) -> &mut OpenOptions 设置文件追加模式的选项。
5 std::io::写入 write_all() fn write_all(&mut self, buf: &[u8]) -> 结果 <()> 尝试将整个缓冲区写入此写入。
6 std::io::读取 读取字符串() fn read_to_string(&mut self, buf: &mut String) -> 结果 读取此源中的所有字节直到 EOF,并将它们附加到 buf。

写入文件

让我们看一个例子来了解如何写入文件。

以下程序创建一个文件“data.txt”。create() 方法用于创建文件。如果文件创建成功,该方法返回一个文件句柄。最后一行write_all函数将在新创建的文件中写入字节。如果任何操作失败,expect() 函数将返回错误消息。

use std::io::Write;
fn main() {
   let mut file = std::fs::File::create("data.txt").expect("create failed");
   file.write_all("Hello World".as_bytes()).expect("write failed");
   file.write_all("\nTutorialsPoint".as_bytes()).expect("write failed");
   println!("data written to file" );
}

输出

data written to file

从文件中读取

以下程序读取文件 data.txt 中的内容并将其打印到控制台。“打开”功能用于打开现有文件。文件的绝对或相对路径作为参数传递给 open() 函数。如果文件不存在或者由于某种原因无法访问,则 open() 函数将引发异常。如果成功,该文件的文件句柄将被分配给“file”变量。

“file”句柄的“read_to_string”函数用于将该文件的内容读入字符串变量。

use std::io::Read;

fn main(){
   let mut file = std::fs::File::open("data.txt").unwrap();
   let mut contents = String::new();
   file.read_to_string(&mut contents).unwrap();
   print!("{}", contents);
}

输出

Hello World
TutorialsPoint

删除文件

以下示例使用remove_file()函数删除文件。当发生错误时,expect() 函数会返回一条自定义消息。

use std::fs;
fn main() {
   fs::remove_file("data.txt").expect("could not remove file");
   println!("file is removed");
}

输出

file is removed

将数据附加到文件

append() 函数将数据写入文件末尾。这如下面给出的示例所示 -

use std::fs::OpenOptions;
use std::io::Write;

fn main() {
   let mut file = OpenOptions::new().append(true).open("data.txt").expect(
      "cannot open file");
   file.write_all("Hello World".as_bytes()).expect("write failed");
   file.write_all("\nTutorialsPoint".as_bytes()).expect("write failed");
   println!("file append success");
}

输出

file append success

复制文件

以下示例将文件中的内容复制到新文件。

use std::io::Read;
use std::io::Write;

fn main() {
   let mut command_line: std::env::Args = std::env::args();
   command_line.next().unwrap();
   // skip the executable file name
   // accept the source file
   let source = command_line.next().unwrap();
   // accept the destination file
   let destination = command_line.next().unwrap();
   let mut file_in = std::fs::File::open(source).unwrap();
   let mut file_out = std::fs::File::create(destination).unwrap();
   let mut buffer = [0u8; 4096];
   loop {
      let nbytes = file_in.read(&mut buffer).unwrap();
      file_out.write(&buffer[..nbytes]).unwrap();
      if nbytes < buffer.len() { break; }
   }
}

将上述程序作为main.exe data.txt datacopy.txt执行。执行文件时传递两个命令行参数 -

  • 源文件的路径
  • 目标文件