rust 处理错误,不使用 try catch, 而是使用 Result<T, E>。
简单的处理rust错误
在各种关于rust错误处理的文档中,为了解释清楚其背后的机制,看着内容很多,不好理解。
比如我们写一个方法,读取文件内容:
fn read_file_to_string(file_path: String) -> Result<String, io::Error>{
let mut file = File::open(file_path)?;
let mut contents = String::new();
file.read_to_string(&mut contents)?;
Ok(contents)
}
上面的代码,当文件不存在的时候,也可以很好的返回异常信息。
调用代码:
let r = read_file_to_string(r"d:\1111.txt".to_string());
match r {
Ok(str) => println!("OK: {str}"),
Err(e) => println!("Error: {e}"),
};
如果文件不存在,会输出信息:
这个异常处理的过程不复杂,分为三步:
-
自定义的函数要返回Result<T,E>,
-
返回Result的函数时,后面加上问号,
-
在最上层,使用match处理结果。
但是这样是不够的,如果在一个大项目中,我们很难找到是哪个文件缺失了,rust不像c#那样,可以很容易的获取到出现问题的代码行数、类和方法名等。
最直观的方法是,在异常信息里,带上文件名。
自定义错误,带上文件名
rust自定义错误分为三步:
1)定义错误类型
2)实现Error特征(trait)
3) 实现Display特征
自定义错误的类型是enum, 和其他语言相比,这有点奇怪。 代码如下:
// 定义自定义错误类型
#[derive(Debug)]
pub enum MyError {
FileOpenError(String),
ParseError(String),
Common(String),
}
// 实现Error特质
impl Error for MyError {}
// 实现Display特质以便打印错误信息
impl fmt::Display for MyError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
MyError::FileOpenError(msg) => write!(f, "Failed to open file: {}", msg),
MyError::ParseError(msg) => write!(f, "Parse error: {}", msg),
MyError::Common(msg) => write!(f, "Other error: {}", msg),
}
}
}
这时,读取文件的函数代码要改成这样:
fn read_file_to_string(file_path: String) -> Result<String, MyError>{
let r = File::open(file_path.clone());
match r {
Ok(mut file) => {
let mut contents = String::new();
let r2 = file.read_to_string(&mut contents);
match r2 {
Ok(size) => return Ok( contents),
Err(e) => return Err(MyError::Common(format!("{e} 文件: {file_path}"))),
}
},
Err(e) => {
return Err(MyError::FileOpenError(format!("{e} 文件: {file_path}")));
},
}
}
代码变得很啰嗦,好在能比较好的显示错误了:
自定义错误的三部曲,虽然有点长,但是这是项目的公共代码,还是可以忍受的。读取文件的代码,和 c#比起来,真的太罗嗦了。
简化通用异常处理
读取文件内容的函数,代码罗嗦的原因是,异常类型通过问号匹配到自定义的MyError很麻烦。
这里我们采用一种更通用的方式,来处理异常:
1) 重新定义异常类型,并且提供其他异常向自定义异常转换的方法
custom_error.rs:
use std::error::Error;
use std::fmt;
use std::fmt::Display;
// 自定义错误类型,包含文件路径信息
#[derive(Debug)]
pub struct MyError {
msg: String,
source: String,
}
// 为自定义错误类型实现Error trait
impl Error for MyError {}
// 实现Display trait,以便于打印错误信息
impl fmt::Display for MyError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{:?}: {}", self.source, self.msg)
}
}
pub fn convert_error(msg:String, err: String) -> MyError {
MyError {
msg: msg ,
source: err.to_string(),
}
}
// 定义一个新的trait
pub trait MyErrorExtension<T> {
fn ex_err(self, msg:&String)-> Result<T, MyError>;
}
// 为Result<T,E>类型实现MyExtension trait
impl<T,E:Display> MyErrorExtension<T> for Result<T,E> {
fn ex_err(self, msg:&String) -> Result<T, MyError> {
match self {
Ok(t) => Ok(t),
Err(e) => Err(MyError{msg:msg.to_string(), source: e.to_string()}),
}
}
}
2) 定义带有通用异常处理能力的函数的示例:
fn read_file_to_string(file_path: String) -> Result<String, MyError>{
let context_info = format!("文件路径: {file_path}");
fs::metadata(&file_path).ex_err(&context_info)?;
let mut file = File::open(&file_path).ex_err( &context_info)?;
let mut contents = String::new();
file.read_to_string(&mut contents).ex_err(&context_info)?;
Ok(contents)
}
以打开文件的方法为例,原本的调用是:
let mut file = File::open(&file_path)?;
新的调用,后面附加了重要的上下文信息,并且把异常类型转换为MyError:
let mut file = File::open(&file_path).ex_err( &context_info)?;
通过扩展方法ex_err, 达到了我们的目的。