详解thiserror库在Rust中的使用

tomlCopy code
[dependencies]
thiserror = "1.0"

use thiserror::Error;
// 自定义错误类型的定义
#[derive(Error, Debug)]
pub enum MyError {
    // DataNotFound 错误的描述
    #[error("data not found")]
    DataNotFound,
    // InvalidInput 错误的描述
    #[error("invalid input")]
    InvalidInput,
}
// 示例函数，展示如何使用自定义错误
fn search_data(query: &str) -> Result<(), MyError> {
    if query.is_empty() {
        // 当查询为空时，返回 InvalidInput 错误
        return Err(MyError::InvalidInput);
    }
    // 这里省略了实际的数据查询逻辑
    // ...
    // 数据未找到时返回 DataNotFound 错误
    Err(MyError::DataNotFound)
}

use std::io;
use thiserror::Error;
// 自定义错误类型的定义
#[derive(Error, Debug)]
pub enum MyError {
    // IoError 错误的描述，它包含一个嵌套的 io::Error
    #[error("I/O error occurred")]
    IoError(#[from] io::Error),
}
// 示例函数，展示如何使用嵌套的错误
fn read_file(file_path: &str) -> Result<String, MyError> {
    // 如果 fs::read_to_string 返回错误，我们使用 MyError::from 将它转换为 MyError::IoError
    std::fs::read_to_string(file_path).map_err(MyError::from)
}

use thiserror::Error;
// 自定义错误类型的定义
#[derive(Error, Debug)]
pub enum MyError {
    // FailedWithCode 的错误描述，其中 {0} 会被动态地替换为具体的代码值
    #[error("failed with code: {0}")]
    FailedWithCode(i32),
}
// 示例函数，展示如何使用动态错误消息
fn process_data(data: &str) -> Result<(), MyError> {
    let error_code = 404; // 某些计算得出的错误代码
    // 使用动态的 error_code 创建 FailedWithCode 错误
    Err(MyError::FailedWithCode(error_code))
}

use thiserror::Error;
// 模拟从其他库中导入的错误类型
#[derive(Debug, Clone)]
pub struct OtherLibError;
// 自定义错误类型的定义
#[derive(Error, Debug)]
pub enum MyError {
    // OtherError 的描述，它直接从其内部的错误类型继承
    #[error(transparent)]
    OtherError(#[from] OtherLibError),
}
// 示例函数，展示如何从其他错误类型转换
fn interface_with_other_lib() -> Result<(), MyError> {
    // 调用其他库的函数...
    // 如果那个函数返回了一个错误，我们使用 MyError::from 将它转换为 MyError::OtherError
    Err(MyError::from(OtherLibError))
}

use std::fs;
use thiserror::Error;
// 模拟从其他部分导入的解析错误类型
#[derive(Debug, Clone)]
pub struct ParseDataError;
// 自定义错误类型的定义
#[derive(Error, Debug)]
pub enum MyError {
    // IoError 错误的描述，它包含一个嵌套的 io::Error
    #[error("I/O error occurred")]
    IoError(#[from] io::Error),
    // ParseError 错误的描述，它包含一个嵌套的 ParseDataError
    #[error("failed to parse data")]
    ParseError(#[from] ParseDataError),
}
// 读取文件并尝试解析其内容
fn read_and_parse(filename: &str) -> Result<String, MyError> {
    // 读取文件内容，可能会抛出 I/O 错误
    let content = fs::read_to_string(filename)?;
    // 尝试解析内容，可能会抛出解析错误
    parse_data(&content).map_err(MyError::from)
}
// 模拟的数据解析函数，这里始终返回一个错误
fn parse_data(content: &str) -> Result<String, ParseDataError> {
    Err(ParseDataError)
}
// 主函数，展示如何使用上述错误处理逻辑
fn main() {
    match read_and_parse("data.txt") {
        Ok(data) => println!("Data: {}", data),
        Err(e) => eprintln!("Error: {}", e),
    }
}

use std::fmt;
#[derive(Debug)]
enum DataFetchError {
    HttpError(u16),
    Timeout,
    InvalidPayload,
}
impl fmt::Display for DataFetchError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::HttpError(code) => write!(f, "HTTP error with code: {}", code),
            Self::Timeout => write!(f, "Data fetching timed out"),
            Self::InvalidPayload => write!(f, "Invalid payload received"),
        }
    }
}
impl std::error::Error for DataFetchError {}
#[derive(Debug)]
enum DatabaseError {
    ConnectionFailed,
    WriteFailed(String),
}
impl fmt::Display for DatabaseError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::ConnectionFailed => write!(f, "Failed to connect to database"),
            Self::WriteFailed(reason) => write!(f, "Failed to write to database: {}", reason),
        }
    }
}
impl std::error::Error for DatabaseError {}

use thiserror::Error;
#[derive(Debug, Error)]
enum DataFetchError {
    #[error("HTTP error with code: {0}")]
    HttpError(u16),
    #[error("Data fetching timed out")]
    Timeout,
    #[error("Invalid payload received")]
    InvalidPayload,
}
#[derive(Debug, Error)]
enum DatabaseError {
    #[error("Failed to connect to database")]
    ConnectionFailed,
    #[error("Failed to write to database: {0}")]
    WriteFailed(String),
}