support HashMap and BTreeMap

1 files changed, 5 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
index a5253d3..5e3653d 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -413,6 +413,11 @@ impl TypeTransform {
                 let inner = inner.convert(quote!(x));
                 quote! {#arg.into_iter().map(|x| #inner).collect()}
             }
+            TypeTransform::CollectionMap(key, value) => {
+                let key = key.convert(quote!(k));
+                let value = value.convert(quote!(v));
+                quote! {#arg.into_iter().map(|(k,v)| (#key, #value)).collect()}
+            }
             TypeTransform::NoOp => arg,
         }
     }