30天拿下Rust之输入输出

希望睿智 2024-07-04 06:49:53

💡 如果想阅读最新的文章,或者有技术问题需要交流和沟通,可搜关注微信公众号号号“希望睿智”。

概述

在软件开发中,输入输出(I/O)是任何应用程序的基本构建模块。Rust作为一门注重安全和性能的语言,在其标准库中提供了强大的I/O功能。这些功能使得Rust程序员能够非常轻松地进行标准I/O、文件读写等操作。

标准I/O

在Rust中,标准输入通常通过std::io::Read trait实现,而标准输出则通过std::io::Write trait实现。这些trait被广泛应用在std::io模块提供的各种类型中,包括:std::io::Stdin、std::io::Stdout和std::io::Stderr。

读取标准输入

在Rust中,可以使用std::io::stdin()函数来获取标准输入流,并使用其read_line方法来读取一行文本。

在下面的示例代码中,我们首先创建了一个空的String来存储用户的输入。然后,我们使用println!宏来打印提示信息到标准输出。接着,我们调用io::stdin().read_line来读取用户输入的一行文本,并将其存储在input变量中。最后,我们使用println!宏来显示用户输入的内容。注意:read_line方法会保留输入的换行符,所以我们需要使用trim方法来移除它。

use std::io;use std::io::Read;  fn main() -> io::Result<()> {    let mut input = String::new();    println!("please input: ");    io::stdin().read_line(&mut input)?;    println!("input text is: {}", input.trim());    Ok(())}

写入标准输出

在Rust中,写入标准输出通常使用println!宏或std::io::stdout().write方法。

在下面的示例代码中,我们首先定义了一个要输出的消息字符串。然后,我们获取标准输出流,并通过调用lock方法来获取一个互斥锁的句柄。这是因为多个线程可能同时尝试写入标准输出,所以我们需要同步访问。接着,我们使用write_all方法将消息和换行符写入输出流。最后,我们调用flush方法来确保所有待处理的输出都被发送到控制台。

use std::io;  use std::io::Write;    fn main() -> io::Result<()> {    let text = "Hello World";    let stdout = io::stdout();    let mut handle = stdout.lock();    handle.write_all(text.as_bytes())?;    handle.write_all(b"\n")?;    handle.flush()?;      Ok(())  }

std::io::Stderr的用法与std::io::Stdout类似,这里就不再赘述了。

文件读写

在Rust中,文件读写通常涉及std::fs模块,该模块提供了一系列用于文件操作的功能。在Rust 1.53.0版本之前,我们还可以使用std::io::util 模块进行文件读写。但从Rust 1.53.0版本开始,std::io::util 模块已被废弃,其功能已被直接集成到std::fs和std::io中。因此,现在推荐直接使用这些模块中的函数和方法来进行文件读写。

读取文件

要读取文件的内容,可以使用std::fs::File结构体和std::io::Read trait。

在下面的示例代码中,我们调用File::open方法打开文件。如果文件不存在或无法打开,它会返回一个错误。然后,我们使用BufReader来包装File对象,这样可以高效地从文件中读取数据。read_to_string方法读取整个文件内容到一个字符串中,如果读取失败,它会返回一个错误。最后,我们使用println!打印输出了文件内容。

use std::fs::File;use std::io::{BufReader, Read};use std::path::Path;fn main() -> std::io::Result<()> {    let path = Path::new("World.txt");    let file = File::open(path)?;    let mut reader = BufReader::new(file);    let mut contents = String::new();    reader.read_to_string(&mut contents)?;    println!("content is: {}", contents);    Ok(())  }

写入文件

要写入文件,可以使用File::create或File::open带上适当的打开模式(比如:std::fs::OpenOptions)。

在下面的示例代码中,我们调用File::create方法创建一个新文件。如果文件已存在,则其内容会被清空。然后,我们使用write_all方法将字节切片写入文件。

use std::fs::File;use std::io::Write;fn main() -> std::io::Result<()> {    let path = "World.txt";    let mut file = File::create(path)?;    let text = "Hello World\nHello 霸都";    file.write_all(text.as_bytes())?;    Ok(())}

BufReader和BufWriter

BufReader和BufWriter是用于包装Read和Write接口的缓冲区结构,它们分别提供了带缓冲区的读取和写入功能。通过使用缓冲区,这些结构能够减少系统调用的次数,从而提高I/O效率,特别是在处理大量小数据块时。

BufReader

BufReader是std::io::BufReader的别名,它包装了一个实现了Read trait的对象,并在内部使用一个缓冲区来存储读取的数据。这使得连续读取操作更加高效,因为可以从内部缓冲区中快速获取数据,而不是每次都从底层对象中进行读取。

在下面的示例代码中,我们创建了一个大小为1024个字节的缓冲区,并使用BufReader的read方法来填充它。然后,我们处理读取到的数据,将其写入到标准输出。当read方法返回读取的字节数为0时,表示没有更多的数据可以读取,我们就退出循环。

use std::fs::File;use std::io::BufReader;use std::io;use std::io::Read;use std::io::Write;use std::path::Path;fn main() -> std::io::Result<()> {    let path = Path::new("World.txt");    let file = File::open(path)?;    let mut reader = BufReader::new(file);    let mut buffer = [0u8; 1024];    let stdout = io::stdout();    let mut handle = stdout.lock();    while let Ok(size) = reader.read(&mut buffer) {        if size == 0 {            break;        }        handle.write_all(&buffer[..size])?;    }    Ok(())}

BufWriter

与BufReader类似,BufWriter是std::io::BufWriter的别名,它包装了一个实现了Write trait的对象,并在内部使用一个缓冲区来存储待写入的数据。这样,当调用write或write_all方法时,数据首先被写入到内部缓冲区,直到缓冲区满或者显式地调用flush方法时,数据才会被写入到底层对象。

在下面的示例代码中,我们创建了一个BufWriter来包装File对象,并使用write_all方法将字符串写入到缓冲区。最后,我们调用flush方法来确保所有数据都被写入到文件中。在大多数情况下,不需要我们手动调用flush,因为当BufWriter对象被销毁时,其析构函数会自动调用flush。但是,在某些情况下,我们可能需要显式地刷新缓冲区以确保数据被立即写入。

use std::fs::File;use std::io::BufWriter;use std::io::Write;fn main() -> std::io::Result<()> {    let file = File::create("World.txt")?;    let mut writer = BufWriter::new(file);    let data = "Hello World";    writer.write_all(data.as_bytes())?;    writer.flush()?;    Ok(())}

0 阅读:0

希望睿智

简介:软件技术分享,一起学习,一起成长,一起进步