Rust语言

关注公众号 jb51net

关闭
首页 > 软件编程 > Rust语言 > rust文件读写

rust文件读写的实现示例

作者：int8

Rust语言提供了强大的文件读写库,使得开发者可以更加方便地进行文件操作,并且其安全性可以有效避免文件操作中可能出现的风险,本文就来详细的介绍了rust文件读写的实现示例,感兴趣的可以了解一下

std::fs模块提供了结构体File，它表示一个文件。

一、打开文件

结构体File提供了open()函数
open()以只读模式打开文件，如果文件不存在，则会抛出一个错误。如果文件不可读，那么也会抛出一个错误。

范例

fn main() {
     let file = std::fs::File::open("data.txt").unwrap();
     println!("文件打开成功：{:?}", file);
}

std::fs::File的open函数没有配套的close函数，因为Rust编译器可以在文件不再被使用时自动关闭文件。

二、创建文件

结构体File提供了create() 函数
以只写模式打开文件。如果文件存在则清空旧内容；如果文件不存在则新建

范例

fn main() {
     let file = std::fs::File::create("data.txt").expect("create failed");
     println!("文件创建成功:{:?}",file);
}

三、读取文件

（一）结构体File实现了Read特性
Read特性提供了如下方法

fn read(&mut self, buf: &mut [u8]) -> Result<usize>

读取一些字节到指定的缓冲区中，返回读取的字节数。读取的字节数等于缓冲区的长度

fn read_to_end(&mut self, buf: &mut Vec<u8>) -> Result<usize>

读取所有字节，直到此源中的 EOF 为止，然后将它们追加到 buf

fn read_to_string(&mut self, buf: &mut String) -> Result<usize>

读取所有字节，直到 EOF 为止，然后将它们追加到 buf。如果读取成功则返回读取的字节数，如果读取失败则抛出错误。

实例

use std::io::prelude::*;
use std::fs;
fn main() {
    let mut buffer = [0u8; 5];
    let mut file = fs::File::open("text.txt").unwrap();
    file.read(&mut buffer).unwrap();
    println!("{:?}", buffer);
    file.read(&mut buffer).unwrap();
    println!("{:?}", buffer);
}
运行结果：
[84, 104, 105, 115, 32]
[105, 115, 32, 97, 32]

范例

data.txt内容如下

简单教程
简单编程

代码如下

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();
     println!("{}", contents);
}

运行结果如下

简单教程
简单编程

（二）快捷函数

std::fs::read_to_string

pub fn read_to_string<P: AsRef<Path>>(path: P) -> Result<String>

将文件的全部内容读取为字符串。这是使用File::open和File::read_to_string 的便捷函数

实例

use std::fs;
fn main() {
    let text = fs::read_to_string("text.txt").unwrap();
    println!("{}", text);
}
运行结果：
This is a text file.

std::fs::read

pub fn read<P: AsRef<Path>>(path: P) -> Result<Vec<u8>>

将文件的全部内容读取为字节 vector。这是使用 File::open 和 read_to_end 的便捷函数

实例

use std::fs;
fn main() {
    let content = fs::read("D:\\text.txt").unwrap();
    println!("{:?}", content);
}
运行结果：
[84, 104, 105, 115, 32, 105, 115, 32, 97, 32, 116, 101, 120, 116, 32, 102, 105, 108, 101, 46]

以上两种方式是一次性读取，十分适合Web应用的开发。但是更多情况下文件的大小可能远超内存容量。所以不常使用。

四、写入文件

（一）结构体File实现了Write特性
Write特性提供了如下方法

fn write(&mut self, buf: &[u8]) -> Result<usize>

写入一个缓冲区，返回写入的字节数。这个函数会尝试写入 buf 的全部内容，但是整个写入可能不会成功，或者写入也会产生错误。

fn write_all(&mut self, buf: &[u8]) -> Result<()>

将整个缓冲区写入File。此方法将连续调用 write，直到没有更多数据要写入或返回非 ErrorKind::Interrupted 类型的错误为止。 在成功写入整个缓冲区或发生此类错误之前，此方法将不会返回

实例

use std::io::prelude::*;
use std::fs::File;
fn main() {
    let mut file = File::create("text.txt").unwrap();
    file.write(b"FROM RUST PROGRAM").unwrap();
}

范例

use std::io::Write;
fn main() {
     let mut file = std::fs::File::create("data.txt").expect("create failed");
     file.write_all("简单教程".as_bytes()).expect("write failed");
     file.write_all("\n简单编程".as_bytes()).expect("write failed");
     println!("data written to file" );
}
编译运行，则data.txt的内容如下
简单教程
简单编程

（二）快捷函数
std::fs::write

pub fn write<P: AsRef<Path>, C: AsRef<[u8]>>(path: P, contents: C) -> Result<()>

把一个切片写入文件，如果文件不存在就会创建文件，如果文件存在就替换其内容。
这是使用 File::create 和 write_all 的便捷函数

实例

use std::fs;
fn main() {
    fs::write("text.txt", "FROM RUST PROGRAM").unwrap();
}

执行程序之后，text.txt文件的内容将会被重写为FROM RUST PROGRAM。

五、追加内容到文件末尾

Write特性并没有提供函数用于追加内容
但std::fs::OpenOptions结构体，可以配置文件打开方式。
OpenOptions提供了append()用于设置追加模式

pub fn append(&mut self, append: bool) -> &mut Self

当文件的模式设置为 追加 之后，写入文件的内容就不会代替原先的旧内容而是放在旧内容的后面。

范例

data.txt内容如下
简单教程
简单编程

代码如下
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("www.twle.cn".as_bytes()).expect("write failed");
     file.write_all("\n简单教程".as_bytes()).expect("write failed");
     file.write_all("\n简单编程".as_bytes()).expect("write failed");
     println!("数据追加成功");
}
运行结果如下
数据追加成功

打开  data.txt  文件，可以看到内容如下
简单教程
简单编程www.twle.cn
简单教程
简单编程

OpenOptions除append权限以外还有read权限和write权限，如果我们想以读写权限打开一个文件可以这样写

实例

text.txt内容如下
this is text

代码如下
use std::io::prelude::*;
use std::fs::OpenOptions;
fn main() -> std::io::Result<()> {
    let mut file = OpenOptions::new().read(true).write(true).open("text.txt")?;
    file.write(b"COVER")?;
    Ok(())
}
运行之后，text.txt文件内容将变成：
COVERis text

六、删除文件

std::fs 提供了函数 remove_file() 用于删除文件。

pub fn remove_file<P: AsRef>(path: P) -> Result<()>

注意，删除可能会失败，即使返回结果为OK，也有可能不会立即就删除。

范例

use std::fs;
fn main() {
     fs::remove_file("data.txt").expect("could not remove file");
     println!("file is removed");
}
编译运行结果如下
file is removed

打开当前目录，我们可以发现文件已经被删除了。

七、复制文件

Rust标准库没有提供任何函数用于复制一个文件为另一个新文件。
但我们可以使用上面提到的函数和方法来实现文件的复制功能。
下面的代码，我们模仿简单版本的 copy 命令

copy old_file_name new_file_name

代码如下

use std::io::Read;
use std::io::Write;
fn main() {
     let mut command_line: std::env::Args = std::env::args();
     command_line.next().unwrap();     // 跳过程序名
     // 原文件
     let source = command_line.next().unwrap();
     // 新文件
     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 data.txt data_new.txt

data.txt 为我们想要复制的原文件路径
data_new.txt 为我们想要的新文件路径

到此这篇关于rust文件读写的实现示例的文章就介绍到这了,更多相关rust文件读写内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:
阅读全文