快捷導(dǎo)航

在Rust中編寫(xiě)自定義Error的詳細(xì)代碼

更新時(shí)間：2024年01月26日 09:42:29 作者：歸云十九少

Result<T, E> 類型可以方便地用于錯(cuò)誤傳導(dǎo),Result<T, E>是模板類型,實(shí)例化后可以是各種類型,但 Rust 要求傳導(dǎo)的 Result 中的 E 是相同類型的,所以我們需要編寫(xiě)自己的 Error 類型,本文給大家介紹了在Rust中編寫(xiě)自定義Error的詳細(xì)代碼,需要的朋友可以參考下

前言

之前我們聊過(guò)，Result<T, E> 類型可以方便地用于錯(cuò)誤傳導(dǎo)，Result<T, E>是模板類型，實(shí)例化后可以是各種類型，但 Rust 要求傳導(dǎo)的 Result 中的 E 是相同類型的，或者能夠自動(dòng)轉(zhuǎn)化為相同類型。比如，下面這段代碼編譯就會(huì)報(bào)錯(cuò)。所以我們需要編寫(xiě)自己的 Error 類型，以同時(shí)包含系統(tǒng)錯(cuò)誤和具體業(yè)務(wù)錯(cuò)誤。

use std::io;
use std::fs::{File};

fn read_fs() -> io::Result<()> {
    File::open("abc.txt")?;
    
    Ok(())
}

fn user_err() -> Result<u32, String> {
    Err(String::from("test faill"))
}

fn test() -> Result<u32, String>{
    read_fs()?;
    test()?;
}

pub fn main() {
    test();
}

系統(tǒng)Error

在 Result<T, E> 中，E 表示一種錯(cuò)誤，Rust 標(biāo)準(zhǔn)庫(kù)已經(jīng)定義一系列 Error，主要是 io Error。它的定義如下：

pub struct Error {
    repr: Repr,
}

我們可以通過(guò)它的關(guān)聯(lián)函數(shù) new，from，other，配合 ErrorKind 來(lái)生成一個(gè)io Error 用于Reuslt。

自定義 Error

我們要定義一個(gè)自己的 Error 類型，它既包含系統(tǒng)的 io Error，也包含業(yè)務(wù) Error，定義如下

// 業(yè)務(wù) Error 定義
enum WorkError {
  	WorkErrorFirst,
    WorkErrorSecond,
    WorkErrorThird,
}

// 復(fù)合 Error 定義
enum MyError {
  	IoErr(std::io::Error),
    WorkErr(WorkError)
}

使用 MyError

fn read_fs() -> Result<(), MyError> {
    let ret = File::open("abc.txt");
    
    if let Err(e) = ret {
        return Err(MyError::IoErr(e));
    } else {
        println!("ret = {ret:?}");
    }
    
    Ok(())
}

fn user_err() -> Result<(), MyError> {
    Err(MyError::WorkErr(WorkError::WorkErrorSecond))
}

fn test() -> Result<(), MyError>{
    read_fs()?;
    user_err()?;
    
    Ok(())
}

pub fn main() {
    let ret = test();
    
    match ret {
        Ok(_) => {
            println!("run test success");
        }
        Err(e) => {
            println!("run test fail");
        }
    }
}

在上述實(shí)現(xiàn)中，我們通過(guò)手動(dòng)方式將 std::io::Error 轉(zhuǎn)化為 MyError。也可以通過(guò)實(shí)現(xiàn) From trait 來(lái)自動(dòng)實(shí)現(xiàn)這種轉(zhuǎn)化。

impl From<std::io::Error> for MyError {
    fn from(err: std::io::Error) -> Self {
        MyError::IoErr(err)
    }
}

// 這樣，read_fs 函數(shù)就可以簡(jiǎn)化如下：
fn read_fs() -> Result<(), MyError> {
    File::open("abc.txt")?;
    Ok(())
}

現(xiàn)在還有一個(gè)問(wèn)題，在 main 函數(shù)中，我們無(wú)法打印 e 信息。通過(guò)簡(jiǎn)單的為 WorkError 和 MyError 類型增加 #[derive(Debug)] 聲明，就可以直接打印 e 信息。

到此這篇關(guān)于在Rust中編寫(xiě)自定義Error的詳細(xì)代碼的文章就介紹到這了,更多相關(guān)Rust編寫(xiě)自定義Error內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: