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 为我们想要的新文件路径

相关推荐
老猿讲编程6 小时前
用示例来看C2Rust工具的使用和功能介绍
rust
金庆6 小时前
How to set_default() using config-rs crate
rust·config·set_default·valuekind
许野平8 小时前
Rust: 利用 chrono 库实现日期和字符串互相转换
开发语言·后端·rust·字符串·转换·日期·chrono
‍。。。18 小时前
使用Rust实现http/https正向代理
http·https·rust
Source.Liu18 小时前
【用Rust写CAD】第二章 第四节 函数
开发语言·rust
monkey_meng18 小时前
【Rust中的迭代器】
开发语言·后端·rust
余衫马18 小时前
Rust-Trait 特征编程
开发语言·后端·rust
monkey_meng19 小时前
【Rust中多线程同步机制】
开发语言·redis·后端·rust
hikktn1 天前
如何在 Rust 中实现内存安全:与 C/C++ 的对比分析
c语言·安全·rust
睡觉谁叫~~~1 天前
一文解秘Rust如何与Java互操作
java·开发语言·后端·rust