温馨提示×

rust thiserror怎样实现自定义错误

小樊
81
2024-11-29 04:19:33
栏目: 编程语言

在Rust中,你可以通过实现std::error::Errorstd::fmt::Display trait来创建自定义错误类型。下面是一个简单的例子:

use std::fmt;
use std::error;

// 定义一个自定义错误类型
#[derive(Debug)]
pub enum CustomError {
    IoError(std::io::Error),
    ParseError(std::num::ParseIntError),
    OtherError(String),
}

// 实现From trait,以便可以从标准错误中转换
impl From<std::io::Error> for CustomError {
    fn from(err: std::io::Error) -> CustomError {
        CustomError::IoError(err)
    }
}

impl From<std::num::ParseIntError> for CustomError {
    fn from(err: std::num::ParseIntError) -> CustomError {
        CustomError::ParseError(err)
    }
}

// 实现Error trait
impl error::Error for CustomError {
    fn source(&self) -> Option<&(dyn error::Error + 'static)> {
        match *self {
            CustomError::IoError(ref err) => Some(err),
            CustomError::ParseError(ref err) => Some(err),
            CustomError::OtherError(ref _msg) => None,
        }
    }
}

// 实现Display trait
impl fmt::Display for CustomError {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match *self {
            CustomError::IoError(ref err) => write!(f, "IO error: {}", err),
            CustomError::ParseError(ref err) => write!(f, "Parse error: {}", err),
            CustomError::OtherError(ref msg) => write!(f, "Other error: {}", msg),
        }
    }
}

// 示例函数,返回自定义错误
fn parse_number(s: &str) -> Result<i32, CustomError> {
    s.parse::<i32>().map_err(CustomError::from)
}

fn main() {
    match parse_number("123") {
        Ok(num) => println!("Parsed number: {}", num),
        Err(e) => println!("Error: {}", e),
    }
}

在这个例子中,我们定义了一个名为CustomError的枚举类型,它包含了三种可能的错误:IoErrorParseErrorOtherError。我们还实现了From trait,以便可以从标准错误中转换到自定义错误。最后,我们实现了ErrorDisplay trait,以便可以方便地使用和显示自定义错误。

0