在 Rust 中,可以使用 Result 类型和 ? 运算符来处理多种错误类型。Result 是一个枚举类型,表示操作可能成功(Ok)或失败(Err)。你可以使用 ? 运算符将错误从函数中传播出去,以便在调用栈中更高层的地方处理它们。
以下是一个示例,展示了如何处理两种不同的错误类型:
use std::fs::File;
use std::io::{self, Read};
#[derive(Debug)]
enum CustomError {
IOError(io::Error),
ParseError(std::num::ParseIntError),
}
impl From<io::Error> for CustomError {
fn from(error: io::Error) -> Self {
CustomError::IOError(error)
}
}
impl From<std::num::ParseIntError> for CustomError {
fn from(error: std::num::ParseIntError) -> Self {
CustomError::ParseError(error)
}
}
fn read_file_contents(file_name: &str) -> Result<String, CustomError> {
let mut file = File::open(file_name)?;
let mut contents = String::new();
file.read_to_string(&mut contents)?;
Ok(contents)
}
fn main() {
match read_file_contents("example.txt") {
Ok(contents) => println!("File contents: {}", contents),
Err(e) => match e {
CustomError::IOError(io_error) => println!("IO error: {:?}", io_error),
CustomError::ParseError(parse_error) => println!("Parse error: {:?}", parse_error),
},
}
}
在这个示例中,我们定义了一个名为 CustomError 的枚举类型,用于表示两种不同的错误类型:IOError 和 ParseError。我们还实现了 From trait,以便可以将 io::Error 和 std::num::ParseIntError 转换为 CustomError。
read_file_contents 函数尝试打开一个文件并将其内容读取到字符串中。如果发生任何错误,它将使用 ? 运算符将错误传播出去,并在 main 函数中处理它们。在 main 函数中,我们使用 match 语句来处理 read_file_contents 返回的 Result 类型。如果操作成功,我们将打印文件内容;如果失败,我们将根据错误类型打印相应的错误消息。