4

我正在尝试过滤掉 struct 类型的所有字段bool。但是syn::Type枚举似乎没有案例,或者我错误地阅读了定义:

pub enum Type {
    Array(TypeArray),
    BareFn(TypeBareFn),
    Group(TypeGroup),
    ImplTrait(TypeImplTrait),
    Infer(TypeInfer),
    Macro(TypeMacro),
    Never(TypeNever),
    Paren(TypeParen),
    Path(TypePath),
    Ptr(TypePtr),
    Reference(TypeReference),
    Slice(TypeSlice),
    TraitObject(TypeTraitObject),
    Tuple(TypeTuple),
    Verbatim(TokenStream),
    // some variants omitted
}

我查看syn::Types了源代码,以检查省略了哪些变体,但这并没有让我更进一步。这是我到目前为止所拥有的:

#[proc_macro_derive(Creator)]
pub fn derive_creator(_item: TokenStream) -> TokenStream {
    let item = parse_macro_input!(_item as syn::DeriveInput);
    let item_ident = item.ident;

    let fields = if let syn::Data::Struct(syn::DataStruct {
        fields: syn::Fields::Named(syn::FieldsNamed { ref named, .. }),
        ..
    }) = item.data
    {
        named
    } else {
        panic!("You can derive Creator only on a struct!")
    };

    let bool_fields = fields.iter().filter(|field| 
        match field.ty {
            // case when field type is bool => true
            _ => false
        }
    );
    unimplemented!()
}

我走错路了吗?或者这根本不可能?还是我错过了什么?

4

2 回答 2

6

我觉得可能有一种更清洁的方式(无需克隆和分配字符串),但过去我做过类似的事情:

match field.ty {
    Type::Path(type_path) if type_path.clone().into_token_stream().to_string() == "bool" => {
        true
    }
    _ => false
}

您也许可以定义bool一次类型,然后比较它是否相等:

let bool_ty = Type::Path(TypePath {
    qself: None,
    path: Path::from(Ident::new("bool", Span::call_site())),
});

if field.ty == bool_ty {
    // it's a bool
}

但我不确定跨度的差异是否会影响平等。Span似乎没有实现PartialEq,所以我猜这没关系。*


*欢迎编辑澄清这一点。

于 2021-04-01T15:00:40.307 回答
1

另一种方式:

    let bool_fields = fields.iter().filter(|field| {
        if let Type::Path(tp) = &field.ty {
            let segments = &tp.path.segments;
            // Check len to avoid "bool::SomeOtherType".
            if segments.len() == 1 {
                return segments[0].ident == "bool";
            }
        }
        false
    });

记住:

值得指出的是,这会按名称检查是否相等。类型别名、单字段元组或 core::primitive::bool 将不匹配,尽管它们在类型解析后都是 bool。– 用户2722968

于 2021-04-01T15:09:19.853 回答