在Rust中,你可以通过实现std::error::Error
和std::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
的枚举类型,它包含了三种可能的错误:IoError
、ParseError
和OtherError
。我们还实现了From
trait,以便可以从标准错误中转换到自定义错误。最后,我们实现了Error
和Display
trait,以便可以方便地使用和显示自定义错误。