快捷導(dǎo)航

Rust中Strum庫實現(xiàn)枚舉

更新時間：2026年06月17日 09:07:49 作者：Rust研習(xí)社

很多Rust新手都知道可以通過派生宏來擴展枚舉的能力,但其實 Rust 社區(qū)有一個非常流行的 Enum 增強庫——Strum,通過它提供的宏能幫我們解決開發(fā)過程中很多的痛點,感興趣的可以了解一下

當前痛點

在 Rust 中，枚舉是一個非常強大的特性，很多語言里的枚舉只是幾個整數(shù)常量：

enum Color { RED, GREEN, BLUE };

而 Rust 的枚舉不僅可以攜帶數(shù)據(jù)，還能用于狀態(tài)機、錯誤處理、協(xié)議建模等場景。但還是不夠方便，比如：

枚舉轉(zhuǎn)字符串
字符串轉(zhuǎn)枚舉
遍歷所有枚舉值
...

這些功能如果自己手寫，那么就需要手寫很多樣板代碼，說實話挺麻煩的。而這時候就是強大的宏出場的時候了，接下來我們將通過一些使用場景來介紹 Strum 的用法。

首先，我們先在 Cargo.toml 添加上 strum 依賴：

[dependencies]
strum = { version = "0.28", features = ["derive"] }

場景一：枚舉轉(zhuǎn)字符串

有這么一個枚舉：

enum Status { 
    Pending, 
    Running, 
    Finished, 
}

如果需要將枚舉轉(zhuǎn)換成字符串，那么一般是通過實現(xiàn) Display 特征來實現(xiàn)：

use std::fmt;

enum Status {
    Pending,
    Running,
    Finished,
}

impl fmt::Display for Status {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Status::Pending => write!(f, "pending"),
            Status::Running => write!(f, "running"),
            Status::Finished => write!(f, "finished"),
        }
    }
}

fn main() {
    println!("{}", Status::Pending);
    println!("{}", Status::Running);
    println!("{}", Status::Finished);
}

而使用 Strum 的 Display 派生宏來實現(xiàn)則非常簡單：

use strum::Display;

#[derive(Display)]
enum Status {
    #[strum(serialize = "pending")]
    Pending,
    #[strum(serialize = "running")]
    Running,
    #[strum(serialize = "finished")]
    Finished,
}

fn main() {
    println!("{}", Status::Pending);
    println!("{}", Status::Running);
    println!("{}", Status::Finished);
}

本質(zhì)上就是 Strum 自動幫我們實現(xiàn)了 Display 特征，說白了就是自動生成這些樣板代碼。

場景二：字符串轉(zhuǎn)枚舉

依舊是這個枚舉：

enum Status {
    Pending,
    Running,
    Finished,
}

假設(shè)我們現(xiàn)在需要將 pending 字符串轉(zhuǎn)換為對應(yīng)的枚舉，那么可以通過實現(xiàn) FromStr 特征來實現(xiàn)：

use std::str::FromStr;

#[derive(Debug)]
enum Status {
    Pending,
    Running,
    Finished,
}

impl FromStr for Status {
    type Err = String;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            "pending" => Ok(Status::Pending),
            "running" => Ok(Status::Running),
            "finished" => Ok(Status::Finished),
            _ => Err("invalid".into()),
        }
    }
}

fn main() {
    println!("{:?}", "pending".parse::<Status>());
    println!("{:?}", "running".parse::<Status>());
    println!("{:?}", "finished".parse::<Status>());
}

這里可以使用 Strum 的 EnumString 派生宏來進行簡化：

use strum::EnumString;

#[derive(Debug, EnumString)]
enum Status {
    #[strum(serialize = "pending")]
    Pending,
    #[strum(serialize = "running")]
    Running,
    #[strum(serialize = "finished")]
    Finished,
}

fn main() {
    println!("{:?}", "pending".parse::<Status>());
    println!("{:?}", "running".parse::<Status>());
    println!("{:?}", "finished".parse::<Status>());
}

場景三：遍歷所有枚舉值

一般情況下，如果我們想遍歷枚舉值，得專門構(gòu)建出一個枚舉數(shù)組來使用，比如：

fn all_status() -> Vec<Status> {
    vec![
        Status::Running,
        Status::Finished,
        Status::Cancelled,
    ]
}

其實這樣挺麻煩的，萬一有新增或減少枚舉值，還得同步進行修改。而使用 Strum 的 EnumIter 派生宏就能解決這個問題：

use strum::{EnumIter, IntoEnumIterator};

#[derive(Debug, EnumIter)]
enum Status {
    Pending,
    Running,
    Finished,
}

fn main() {
    for status in Status::iter() {
        println!("{:?}", status);
    }
}