TokenStream {
let data = if let syn::Data::Struct(data) = &ast.data {
data
} else {
unimplemented!();
};
let mut flat_values = vec![];
let mut flat_types = vec![];
let mut mapped_indexes: HashMap<String, usize> = HashMap::new();
for (idx, field) in data.fields.iter().enumerate() {
let name = &field.ident;
let type_ = &field.ty;
flat_values.push(quote!(self.#name.clone()));
flat_types.push(quote!(#type_));
mapped_indexes.insert(quote!(#name), idx);
}
let name = &ast.ident;
let gen = quote! {
impl #name{
fn flat(&self) -> (#(#flat_types,)*) {
return (#(#flat_values,)*)
}
fn get_field_index(field_name: String) -> usize {
return #mapped_indexes.get(&field_name).cloned().unwrap_or(0usize)
}
}
};
gen.into()
}
Компилятор выдает следующую ошибку:
the trait ToTokens is not implemented for HashMap<std::string::String, usize>. Пробовал разворачивать и вызывать .to_string() - не помогает. Что не так с кодом?
Генерируй match с именами полей
Вот такое накодил: fn impl_flat_macro(ast: &syn::DeriveInput) -> TokenStream { let data = if let syn::Data::Struct(data) = &ast.data { data } else { unimplemented!(); }; let mut flat_values = vec![]; let mut flat_types = vec![]; let mut flat_names = vec![]; for field in data.fields.iter() { let name = &field.ident; let type_ = &field.ty; flat_values.push(quote!(self.#name.clone())); flat_types.push(quote!(#type_)); flat_names.push(quote!(#name).to_string()); } let name = &ast.ident; let gen = quote! { impl #name{ fn flat(&self) -> (#(#flat_types,)*) { return (#(#flat_values,)*) } fn get_field_index(field_name: String) -> usize{ return vec![#(#flat_names,)*].iter().position(|&r| r == field_name).unwrap_or(0usize); } } }; gen.into() }
Обсуждают сегодня