Question

有没有办法提取枚举中的元素数量？

简单示例（使用虚构的number_of_elements方法）：

enum FooBar { A = 0, B, C, };

println!("Number of items: {}", FooBar.number_of_elements());
// "Number of items: 3"

在C中我通常会......

enum FooBar { A = 0, B, C, };
#define FOOBAR_NUMBER_OF_ITEMS (C + 1)

然而，与此相当的Rust不起作用：

enum FooBar { A = 0, B, C, };
const FOOBAR_NUMBER_OF_ITEMS: usize = (C as usize) + 1;

// Raises an error:
//     unimplemented constant expression: enum variants

在枚举中包含最后一项是非常不方便的，因为如果不考虑所有成员，匹配的枚举将会出错。

enum FooBar { A = 0, B, C, FOOBAR_NUMBER_OF_ITEMS, };

有没有办法将枚举中的项目数作为常量值？

注意：即使这与问题没有直接关系，我之所以想要这个功能，是因为我正在使用builder-pattern构建一系列只运行一次的行动。出于这个原因，我可以使用枚举大小的固定大小数组。

Answer 1

您可以使用新的程序宏（在撰写本答案之日起2周内稳定）：

extern crate proc_macro;
extern crate syn;
#[macro_use]
extern crate quote;

use proc_macro::TokenStream;

#[proc_macro_derive(EnumVariantCount)]
pub fn derive_enum_variant_count(input: TokenStream) -> TokenStream {
    let syn_item: syn::DeriveInput = syn::parse(input).unwrap();
    let len = match syn_item.data {
        syn::Data::Enum(enum_item) => enum_item.variants.len(),
        _ => panic!("EnumVariantCount only works on Enums"),
    };
    let expanded =quote! {
const LENGTH: usize = #len;
    };
    expanded.into()
}

让读者知道这个派生宏可以在同一个模块中多次使用。

要使用宏，只需将#[derive(EnumVariantCount)]附加到枚举即可。现在应该有一个名为LENGTH的全局常量。

如何将枚举中的元素（变体）数量作为常量值？

1 个答案: