温馨提示×

rust thiserror如何处理多种错误类型

小樊
81
2024-11-29 03:55:32
栏目: 编程语言

在 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 的枚举类型,用于表示两种不同的错误类型:IOErrorParseError。我们还实现了 From trait,以便可以将 io::Errorstd::num::ParseIntError 转换为 CustomError

read_file_contents 函数尝试打开一个文件并将其内容读取到字符串中。如果发生任何错误,它将使用 ? 运算符将错误传播出去,并在 main 函数中处理它们。在 main 函数中,我们使用 match 语句来处理 read_file_contents 返回的 Result 类型。如果操作成功,我们将打印文件内容;如果失败,我们将根据错误类型打印相应的错误消息。

0