rust 如何有条件地检查枚举是一个变体还是另一个变体?

clj7thdc  于 2023-01-09  发布在  其他
关注(0)|答案(2)|浏览(125)

我有一个枚举,它有两个变体:

enum DatabaseType {
    Memory,
    RocksDB,
}

要在一个函数中生成条件if来检查参数是DatabaseType::Memory还是DatabaseType::RocksDB,我需要什么?

fn initialize(datastore: DatabaseType) -> Result<V, E> {
    if /* Memory */ {
        //..........
    } else if /* RocksDB */ {
        //..........
    }
}
xmd2e60i

xmd2e60i1#

首先看看免费的官方Rust书籍The Rust Programming Language,特别是the chapter on enums

一个月

fn initialize(datastore: DatabaseType) {
    match datastore {
        DatabaseType::Memory => {
            // ...
        }
        DatabaseType::RocksDB => {
            // ...
        }
    }
}

一个月一个月

fn initialize(datastore: DatabaseType) {
    if let DatabaseType::Memory = datastore {
        // ...
    } else {
        // ...
    }
}

==

#[derive(PartialEq)]
enum DatabaseType {
    Memory,
    RocksDB,
}

fn initialize(datastore: DatabaseType) {
    if DatabaseType::Memory == datastore {
        // ...
    } else {
        // ...
    }
}

matches!

此功能自Rust 1.42.0起可用

fn initialize(datastore: DatabaseType) {
    if matches!(datastore, DatabaseType::Memory) {
        // ...
    } else {
        // ...
    }
}

另见:

db2dz4w8

db2dz4w82#

// A simple example that runs in rust 1.58:
enum Aap {
    Noot(i32, char),
    Mies(String, f64),
}

fn main() {
    let aap: Aap = Aap::Noot(42, 'q');
    let noot: Aap = Aap::Mies(String::from("noot"), 422.0);
    println!("{}", doe(aap));
    println!("{}", doe(noot));
}

fn doe(a: Aap) -> i32 {
    match a {
        Aap::Noot(i, _) => i,
        Aap::Mies(_, f) => f as i32,
    }
}

相关问题