feat: add indexmap_with_hasher and indexset_with_hasher macros

Author: iajoiner

src/macros.rs

@@ -1,3 +1,38 @@
+/// Create an [`IndexMap`][crate::IndexMap] from a list of key-value pairs and a custom hasher
+///
+/// ## Example
+///
+/// ```
+/// use indexmap::indexmap_with_hasher;
+/// use std::collections::hash_map::RandomState;
+///
+/// let map = indexmap_with_hasher!{
+///     RandomState;
+///     "a" => 1,
+///     "b" => 2,
+/// };
+/// assert_eq!(map["a"], 1);
+/// assert_eq!(map["b"], 2);
+/// assert_eq!(map.get("c"), None);
+///
+/// // "a" is the first key
+/// assert_eq!(map.keys().next(), Some(&"a"));
+/// ```
+#[macro_export]
+macro_rules! indexmap_with_hasher {
+    ($S:ty; $($key:expr => $value:expr,)+) => { $crate::indexmap_with_hasher!($S; $($key => $value),+) };
+    ($S:ty; $($key:expr => $value:expr),*) => {{
+        const CAP: usize = <[()]>::len(&[$({ stringify!($key); }),*]);
+        #[allow(unused_mut)]
+        // Specify your custom S (must implement Default) as the hasher:
+        let mut map = $crate::IndexMap::<_, _, $S>::with_capacity_and_hasher(CAP, <$S>::default());
+        $(
+            map.insert($key, $value);
+        )*
+        map
+    }};
+}
+
 #[cfg(feature = "std")]
 #[cfg_attr(docsrs, doc(cfg(feature = "std")))]
 #[macro_export]
@@ -35,6 +70,41 @@ macro_rules! indexmap {
     };
 }
 
+/// Create an [`IndexSet`][crate::IndexSet] from a list of key-value pairs and a custom hasher
+///
+/// ## Example
+///
+/// ```
+/// use indexmap::indexset_with_hasher;
+/// use std::collections::hash_map::RandomState;
+///
+/// let set = indexset_with_hasher!{
+///     RandomState;
+///     "a",
+///     "b",
+/// };
+/// assert!(set.contains("a"));
+/// assert!(set.contains("b"));
+/// assert!(!set.contains("c"));
+///
+/// // "a" is the first value
+/// assert_eq!(set.iter().next(), Some(&"a"));
+/// ```
+#[macro_export]
+macro_rules! indexset_with_hasher {
+    ($S:ty; $($value:expr,)+) => { $crate::indexset_with_hasher!($S; $($value),+) };
+    ($S:ty; $($value:expr),*) => {{
+        const CAP: usize = <[()]>::len(&[$({ stringify!($value); }),*]);
+        #[allow(unused_mut)]
+        // Specify your custom S (must implement Default) as the hasher:
+        let mut set = $crate::IndexSet::<_, $S>::with_capacity_and_hasher(CAP, <$S>::default());
+        $(
+            set.insert($value);
+        )*
+        set
+    }};
+}
+
 #[cfg(feature = "std")]
 #[cfg_attr(docsrs, doc(cfg(feature = "std")))]
 #[macro_export]