在Rust中,錯誤處理是通過使用Result類型和自定義錯誤類型來實現的。Result是一個枚舉類型,表示操作可能成功(Ok)或失敗(Err)。當操作失敗時,可以返回一個包含錯誤信息的Err變體。以下是如何在Rust中使用Result類型處理錯誤的示例:
use std::fmt;
#[derive(Debug)]
enum CustomError {
IOError(std::io::Error),
ParseError(std::num::ParseIntError),
}
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),
}
}
}
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)
}
}
這里我們定義了一個名為CustomError的枚舉類型,它包含兩種錯誤:IOError和ParseError。我們還實現了fmt::Display和From trait,以便可以方便地打印錯誤信息和將標準錯誤類型轉換為CustomError。
use std::fs::File;
use std::io::Read;
fn read_file_contents(filename: &str) -> Result<String, CustomError> {
let mut file = File::open(filename)?;
let mut contents = String::new();
file.read_to_string(&mut contents)?;
Ok(contents)
}
在這個函數中,我們使用了兩次?操作符來處理錯誤。如果File::open或file.read_to_string操作失敗,它們會返回一個Err變體,其中包含有關錯誤的信息。?操作符會將這些錯誤傳播給調用者,這樣我們就可以根據需要處理它們。
fn main() {
match read_file_contents("example.txt") {
Ok(contents) => println!("File contents: {}", contents),
Err(e) => eprintln!("Error: {}", e),
}
}
在main函數中,我們使用match語句來處理read_file_contents函數返回的Result。如果操作成功,我們打印文件內容;如果操作失敗,我們打印錯誤信息。