Rust中類(lèi)型轉(zhuǎn)換在錯(cuò)誤處理中的應(yīng)用小結(jié)

更新時(shí)間：2023年09月26日 11:39:20 作者：明天好，會(huì)的

隨著項(xiàng)目的進(jìn)展,關(guān)于Rust的故事又翻開(kāi)了新的一頁(yè),今天來(lái)到了服務(wù)器端的開(kāi)發(fā)場(chǎng)景,發(fā)現(xiàn)錯(cuò)誤處理中的錯(cuò)誤類(lèi)型轉(zhuǎn)換有必要分享一下,對(duì)Rust錯(cuò)誤處理相關(guān)知識(shí)感興趣的朋友一起看看吧

Rust抽象出來(lái)了Result<T,E>，T是返回值的類(lèi)型，E是錯(cuò)誤類(lèi)型。只要函數(shù)的返回值的類(lèi)型被定義為Resut<T,E>，那么作為開(kāi)發(fā)人員就有責(zé)任來(lái)處理調(diào)用這個(gè)函數(shù)可能發(fā)生的錯(cuò)誤。通過(guò)Result<T,E>，Rust其實(shí)給開(kāi)發(fā)人員指明了一條錯(cuò)誤處理的道路，使代碼更加健壯。

場(chǎng)景

服務(wù)器端處理api請(qǐng)求的框架：Rocket
服務(wù)器端處理數(shù)據(jù)持久化的框架：tokio_postgres

在api請(qǐng)求的框架中，我把返回類(lèi)型定義成了 Result<T, rocket::response::status::Custom\<String>> ，即錯(cuò)誤類(lèi)型是 rocket::response::status::Custom\<String> 。

在tokio_postgres中，直接使用 tokio_postgres::error::Error 。

即如果要處理錯(cuò)誤，就必須將 tokio_postgres::error::Error 轉(zhuǎn)換成 rocket::response::status::Custom\<String> 。那么我們從下面的原理開(kāi)始，逐一領(lǐng)略Rust的錯(cuò)誤處理方式，通過(guò)對(duì)比找到最合適的方式吧。

原理

對(duì)錯(cuò)誤的處理，Rust有3種可選的方式

使用match
使用if let
使用map_err

下面我結(jié)合場(chǎng)景，逐一演示各種方式是如何處理錯(cuò)誤的。

下面的代碼中涉及到2個(gè)模塊（文件）。 /src/routes/notes.rs 是路由層，負(fù)責(zé)將api請(qǐng)求導(dǎo)向合適的service。 /src/services/note_books.rs 是service層，負(fù)責(zé)業(yè)務(wù)邏輯和數(shù)據(jù)持久化的處理。這里的邏輯也很簡(jiǎn)單，就是route層調(diào)用service層，將數(shù)據(jù)寫(xiě)入到數(shù)據(jù)庫(kù)中。

使用match

src/routes/notes.rs

#[post("/api/notes", format = "application/json", data = "<note>")]
pub async fn post_notes(note: Json<Note>) -> Result<(), rocket::response::status::Custom<String>> {
    insert_or_update_note(&note.into_inner()).await
}

/src/services/note_book.rs

pub async fn insert_or_update_note(
    note: &Note,
) -> Result<(), rocket::response::status::Custom<String>> {
    let (client, connection) = match connect(
        "host=localhost dbname=notes_db user=postgres port=5432",
        NoTls,
    )
    .await
    {
        Ok(res) => res,
        Err(err) => {
            return Err(rocket::response::status::Custom(
                rocket::http::Status::ExpectationFailed,
                format!("{}", err),
            ));
        }
    };
    ...
    match client
        .execute(
            "insert into notes (id, title, content) values($1, $2, $3);",
            &[&get_system_seconds(), &note.title, &note.content],
        )
        .await
    {
        Ok(res) => Ok(()),
        Err(err) => Err(rocket::response::status::Custom(
            rocket::http::Status::ExpectationFailed,
            format!("{}", err),
        )),
    }
}

通過(guò)上面的代碼我們可以讀出一下內(nèi)容：

在service層定義了route層相同的錯(cuò)誤類(lèi)型
在service層將持久層的錯(cuò)誤轉(zhuǎn)換成了route層的錯(cuò)誤類(lèi)型
使用match的代碼量還是比較大

使用if let

/src/services/note_book.rs

pub async fn insert_or_update_note(
    note: &Note,
) -> Result<(), rocket::response::status::Custom<String>> {
    if let Ok((client, connection)) = connect(
        "host=localhost dbname=notes_db user=postgres port=5432",
        NoTls,
    )
    .await
    {
        ...
        if let Ok(res) = client
            .execute(
                "insert into notes (id, title, content) values($1, $2, $3);",
                &[&get_system_seconds(), &note.title, &note.content],
            )
            .await
        {
            Ok(())
        } else {
            Err(rocket::response::status::Custom(
                rocket::http::Status::ExpectationFailed,
                format!("{}", "unknown error"),
            ))
        }
    } else {
        Err(rocket::response::status::Custom(
            rocket::http::Status::ExpectationFailed,
            format!("{}", "unknown error"),
        ))
    }
}

src/routes/notes.rs

#[post("/api/notes", format = "application/json", data = "<note>")]
pub async fn post_notes(note: Json<Note>) -> Result<(), rocket::response::status::Custom<String>> {
    insert_or_update_note(&note.into_inner()).await
}

使用了 if let ... ，代碼更加的別扭，并且在else分支中，拿不到具體的錯(cuò)誤信息。

其實(shí)，不難看出，我們的目標(biāo)是將api的請(qǐng)求，經(jīng)過(guò)route層和service層，將數(shù)據(jù)寫(xiě)入到數(shù)據(jù)中。但這其中的錯(cuò)誤處理代碼的干擾就特別大，甚至要有邏輯嵌套現(xiàn)象。這種代碼的已經(jīng)離初衷比較遠(yuǎn)了，是否有更加簡(jiǎn)潔的方式，使代碼能夠最大限度的還原邏輯本身，把錯(cuò)誤處理的噪音降到最低呢？答案肯定是有的。那就是map_err

map_err

map_err是Result上的一個(gè)方法，專(zhuān)門(mén)用于錯(cuò)誤的轉(zhuǎn)換。下面的代碼經(jīng)過(guò)了map_err的改寫(xiě)，看上去是不是清爽了不少啊。/src/services/note_book.rs

pub async fn insert_or_update_note(
    note: &Note,
) -> Result<(), rocket::response::status::Custom<String>> {
    let (client, connection) = connect(
        "host=localhost dbname=notes_db user=postgres port=5432",
        NoTls,
    )
    .await
    .map_err(|err| {
        rocket::response::status::Custom(
            rocket::http::Status::ExpectationFailed,
            format!("{}", err),
        )
    })?;
    ...
    let _ = client
        .execute(
            "insert into notes (id, title, content) values($1, $2, $3);",
            &[&get_system_seconds(), &note.title, &note.content],
        )
        .await
        .map_err(|err| {
            rocket::response::status::Custom(
                rocket::http::Status::ExpectationFailed,
                format!("{}", err),
            )
        })?;
    Ok(())
}

src/routes/notes.rs

#[post("/api/notes", format = "application/json", data = "<note>")]
pub async fn post_notes(note: Json<Note>) -> Result<(), rocket::response::status::Custom<String>> {
    insert_or_update_note(&note.into_inner()).await
}

經(jīng)過(guò)map_err改寫(xiě)后的代碼，代碼的邏輯流程基本上還原了邏輯本身，但是map_err要額外占4行代碼，且錯(cuò)誤對(duì)象的初始化代碼存在重復(fù)。在實(shí)際的工程項(xiàng)目中，service層的處理函數(shù)可能是成百上千，如果再乘以4，那多出來(lái)的代碼量也不少啊，這會(huì)給后期的維護(hù)帶來(lái)不小的壓力。

那是否還有改進(jìn)的空間呢？答案是Yes。

Rust為我們提供了From<T> trait，用于類(lèi)型轉(zhuǎn)換。它定義了從一種類(lèi)型T到另一種類(lèi)型Self的轉(zhuǎn)換方法。我覺(jué)得這是Rust語(yǔ)言設(shè)計(jì)亮點(diǎn)之一。

但是，Rust有一個(gè)顯示，即實(shí)現(xiàn)From<T> trait的結(jié)構(gòu)，必須有一個(gè)在當(dāng)前的crate中，也就是說(shuō)我們不能直接通過(guò)From<T>來(lái)實(shí)現(xiàn)從 tokio_postgres::error::Error 到 rocket::response::status::Custom<String> 。也就是說(shuō)下面的代碼編譯器會(huì)報(bào)錯(cuò)。

impl From<tokio_postgres::Error> for rocket::response::status::Custom<String> {}

報(bào)錯(cuò)如下：

32 | impl From<tokio_postgres::Error> for rocket::response::status::Custom<String> {}
| ^^^^^---------------------------^^^^^----------------------------------------
| | | |
| | | `rocket::response::status::Custom` is not defined in the current crate
| | `tokio_postgres::Error` is not defined in the current crate
| impl doesn't use only types from inside the current crate

因此，我們要定義一個(gè)類(lèi)型 MyError 作為中間類(lèi)型來(lái)轉(zhuǎn)換一下。/src/models.rs

pub struct MyError {
    pub message: String,
}
impl From<tokio_postgres::Error> for MyError {
    fn from(err: Error) -> Self {
        Self {
            message: format!("{}", err),
        }
    }
}
impl From<MyError> for rocket::response::status::Custom<String> {
    fn from(val: MyError) -> Self {
        status::Custom(Status::ExpectationFailed, val.message)
    }
}

/src/services/note_book.rs

pub async fn insert_or_update_note(
    note: &Note,
) -> Result<(), rocket::response::status::Custom<String>> {
    let (client, connection) = connect(
        "host=localhost dbname=notes_db user=postgres port=5432",
        NoTls,
    )
    .await
    .map_err(MyError::from)?;
    ...
    let _ = client
        .execute(
            "insert into notes (id, title, content) values($1, $2, $3);",
            &[&get_system_seconds(), &note.title, &note.content],
        )
        .await
        .map_err(MyError::from)?;
    Ok(())
}

src/routes/notes.rs

#[post("/api/notes", format = "application/json", data = "<note>")]
pub async fn post_notes(note: Json<Note>) -> Result<(), rocket::response::status::Custom<String>> {
    insert_or_update_note(&note.into_inner()).await
}

而 MyError 到 rocket::response::status::Custom<String> 之間的轉(zhuǎn)換是隱式的，由編譯器來(lái)完成。因此我們的錯(cuò)誤類(lèi)型的轉(zhuǎn)換最終縮短為 map_err(|err|MyError::from(err)) ，再簡(jiǎn)寫(xiě)為 map_err(MyError::from)。

關(guān)于錯(cuò)誤處理中的類(lèi)型轉(zhuǎn)換應(yīng)用解析就到這里。通過(guò)分析這個(gè)過(guò)程，我們可以看到，在設(shè)計(jì)模塊時(shí)，我們應(yīng)該確定一種錯(cuò)誤類(lèi)型，就像tokio_postgres庫(kù)一樣，只暴露了tokio_postgress::error::Error一種錯(cuò)誤類(lèi)型。這種設(shè)計(jì)既方便我們?cè)谠O(shè)計(jì)模塊時(shí)處理錯(cuò)誤轉(zhuǎn)換，也方便其我們的模塊在被調(diào)用時(shí)，其它代碼進(jìn)行錯(cuò)誤處理。

到此這篇關(guān)于Rust中類(lèi)型轉(zhuǎn)換在錯(cuò)誤處理中的應(yīng)用解析的文章就介紹到這了,更多相關(guān)Rust錯(cuò)誤處理內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: