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

问题描述

我有一个包含两个变体的枚举:

I have an enum with two variants:

enum DatabaseType {
    Memory,
    RocksDB,
}

在检查参数是 DatabaseType::Memory 还是 DatabaseType::RocksDB 的函数中，我需要什么才能设置条件 if ?

What do I need in order to make a conditional if inside a function that checks if an argument is DatabaseType::Memory or DatabaseType::RocksDB?

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

推荐答案

首先，返回并重新阅读免费的官方 Rust 书籍:Rust 编程语言，特别是 关于枚举的章节.

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

if let

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 {
        // ...
    }
}

匹配！

从 Rust 1.42.0 开始可用

matches!

This is available since Rust 1.42.0

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

另见:

• 如何在没有模式匹配的情况下比较枚举
• 从没有模式匹配的枚举中读取
• 仅通过变体而不是值比较枚举
• https://doc.rust-lang.org/std/macro.匹配.html

这篇关于如何有条件地检查枚举是一种变体还是另一种变体?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！