当给定的枚举不是某种变体时，如何返回None？

Question

我定义了以下枚举：

#[derive(Debug, Copy, Clone)]
struct Core;

#[derive(Debug, Copy, Clone)]
struct Mem;

#[derive(Debug, Copy, Clone)]
pub enum Atag {
    Core(Core),
    Mem(Mem),
    Cmd(&'static str),
    Unknown(u32),
    None,
}

我想在这个enum上实现一个“过滤掉”某些枚举值的函数。 我有以下内容：

impl Atag {
    /// Returns `Some` if this is a `Core` ATAG. Otherwise returns `None`.
    pub fn core(self) -> Option<Core> {
        match self {
            Atag::Core => Some(self),
            _ => None
        }
    }
}

我不知道为什么，但编译器抱怨：

error[E0532]: expected unit struct/variant or constant, found tuple variant `Atag::Core`
  --> src/main.rs:17:13
   |
17 |             Atag::Core => Some(self),
   |             ^^^^^^^^^^ not a unit struct/variant or constant
help: possible better candidate is found in another module, you can import it into scope
   |
1  | use Core;
   |

我也尝试了一种比较方法：

pub fn core(self) -> Option<Core> {
    if self == Atag::Core {
        Some(self)
    } else {
        None
    }
}

但编译器抱怨：

error[E0369]: binary operation `==` cannot be applied to type `Atag`
  --> src/main.rs:20:12
   |
20 |         if self == Atag::Core {
   |            ^^^^^^^^^^^^^^^^^^
   |
   = note: an implementation of `std::cmp::PartialEq` might be missing for `Atag`

Answer 1

我认为这只是模式匹配的限制，旨在防止意外行为。

具有Core类型的Atag的完整“定义”是Atag::Core(raw::Core) 。 显然， Core的内容与您无关，但编译器需要知道所有内容都“被占用”，因为编译器是规则的坚持者。 解决这个问题的最简单方法是使用“任何模式”， _ ，就像你匹配非Core变体一样。

impl Atag {
    /// Returns `Some` if this is a `Core` ATAG. Otherwise returns `None`.
    pub fn core(self) -> Option<Core> {
        match self {
            // The compiler now knows that a value is expected,
            // but isn't necessary for the purposes of our program.
            Atag::Core(_) => Some(self),
            _ => None
        }
    }
}

要忽略多个值，您可以使用Something::Foo(_, _) - 变量中的每个值使用一个下划线，或者忽略所有内容的Something::Foo(..) 。

请记住，与其他语言不同，Rust枚举不仅仅是“不仅仅是”不同类型的集合。 与枚举值相关联的数据是其中的一部分，就像结构的字段一样。 所以self == Atag::Core不是一个有意义的语句，因为它忽略了与Core相关的数据。 Foo(0)与Foo(12) ，即使它们都是Foo变体。

我还想指出， if let - 就我所知 - 是最接近标准if语句的选项而没有在Atag上定义一个自定义的is_core函数（假设存在match ， if let ，则是基本上没必要）。

impl Atag {
    /// Returns `Some` if this is a `Core` ATAG. Otherwise returns `None`.
    pub fn core(self) -> Option<Core> {
        if let Atag::Core(_) = self {
            Some(self)
        } else {
            None
        }
    }
}

Answer 2

我需要这样的东西来很好地将函数链接在一起。 在这种情况下，您希望返回未包装的核心类型，而不仅仅是枚举。

我还发现不使用输入更容易，因此接受了一个&self参数并返回了一个Option<&Core> 。 但你可以同时拥有两者。

Rust约定将as_X作为基于引用的转换，将into_X作为消耗该值的转换。 例如：

impl Atag {
    fn as_core(&self) -> Option<&Core> {
        if let Atag::Core(ref v) = self {
            Some(v)
        } else {
            None
        }
    }
    fn into_core(self) -> Option<Core> {
        if let Atag::Core(v) = self {
            Some(v)
        } else {
            None
        }
    }
}

fn main() {
    let c = Atag::Core(Core {});
    let m = Atag::Mem(Mem {});
    assert_eq!(c.as_core().map(|cc| "CORE_REF"), Some("CORE_REF"));
    assert_eq!(m.as_core().map(|cc| "CORE_REF"), None);
    // Consume c - we cant use it after here...
    assert_eq!(c.into_core().map(|cc| "NOM NOM CORE"), Some("NOM NOM CORE"));
    // Consume m - we cant use it after here...
    assert_eq!(m.into_core().map(|cc| "NOM NOM CORE"), None);
}