diff options
Diffstat (limited to 'mingling_macros/src/chain.rs')
| -rw-r--r-- | mingling_macros/src/chain.rs | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/mingling_macros/src/chain.rs b/mingling_macros/src/chain.rs index a7be233..9ef4ae0 100644 --- a/mingling_macros/src/chain.rs +++ b/mingling_macros/src/chain.rs @@ -125,10 +125,18 @@ pub fn chain_attr(item: TokenStream) -> TokenStream { let chain_entry = quote! { #struct_name => #previous_type, }; + let chain_exist_entry = quote! { + id if id == std::any::TypeId::of::<#previous_type>() => true, + }; let mut chains = crate::CHAINS.lock().unwrap(); - let entry = chain_entry.to_string(); - if !chains.contains(&entry) { - chains.push(entry); + let mut chain_exist = crate::CHAINS_EXIST.lock().unwrap(); + let chain_entry = chain_entry.to_string(); + let chain_exist_entry = chain_exist_entry.to_string(); + if !chains.contains(&chain_entry) { + chains.push(chain_entry); + } + if !chains.contains(&chain_exist_entry) { + chain_exist.push(chain_exist_entry); } expanded.into() |