Use collect for decoding more collection types.

rust-lang · Oct 5, 2023 · 2db1d59 · 2db1d59
1 parent 5f69ca6
commit 2db1d59
Showing 1 changed file with 6 additions and 40 deletions.
diff --git a/compiler/rustc_serialize/src/serialize.rs b/compiler/rustc_serialize/src/serialize.rs
@@ -578,13 +578,7 @@ where
 {
     fn decode(d: &mut D) -> BTreeMap<K, V> {
         let len = d.read_usize();
-        let mut map = BTreeMap::new();
-        for _ in 0..len {
-            let key = Decodable::decode(d);
-            let val = Decodable::decode(d);
-            map.insert(key, val);
-        }
-        map
+        (0..len).map(|_| (Decodable::decode(d), Decodable::decode(d))).collect()
     }
 }
 
@@ -606,11 +600,7 @@ where
 {
     fn decode(d: &mut D) -> BTreeSet<T> {
         let len = d.read_usize();
-        let mut set = BTreeSet::new();
-        for _ in 0..len {
-            set.insert(Decodable::decode(d));
-        }
-        set
+        (0..len).map(|_| Decodable::decode(d)).collect()
     }
 }
 
@@ -637,14 +627,7 @@ where
 {
     fn decode(d: &mut D) -> HashMap<K, V, S> {
         let len = d.read_usize();
-        let state = Default::default();
-        let mut map = HashMap::with_capacity_and_hasher(len, state);
-        for _ in 0..len {
-            let key = Decodable::decode(d);
-            let val = Decodable::decode(d);
-            map.insert(key, val);
-        }
-        map
+        (0..len).map(|_| (Decodable::decode(d), Decodable::decode(d))).collect()
     }
 }
 
@@ -668,12 +651,7 @@ where
 {
     fn decode(d: &mut D) -> HashSet<T, S> {
         let len = d.read_usize();
-        let state = Default::default();
-        let mut set = HashSet::with_capacity_and_hasher(len, state);
-        for _ in 0..len {
-            set.insert(Decodable::decode(d));
-        }
-        set
+        (0..len).map(|_| Decodable::decode(d)).collect()
     }
 }
 
@@ -700,14 +678,7 @@ where
 {
     fn decode(d: &mut D) -> indexmap::IndexMap<K, V, S> {
         let len = d.read_usize();
-        let state = Default::default();
-        let mut map = indexmap::IndexMap::with_capacity_and_hasher(len, state);
-        for _ in 0..len {
-            let key = Decodable::decode(d);
-            let val = Decodable::decode(d);
-            map.insert(key, val);
-        }
-        map
+        (0..len).map(|_| (Decodable::decode(d), Decodable::decode(d))).collect()
     }
 }
 
@@ -731,12 +702,7 @@ where
 {
     fn decode(d: &mut D) -> indexmap::IndexSet<T, S> {
         let len = d.read_usize();
-        let state = Default::default();
-        let mut set = indexmap::IndexSet::with_capacity_and_hasher(len, state);
-        for _ in 0..len {
-            set.insert(Decodable::decode(d));
-        }
-        set
+        (0..len).map(|_| Decodable::decode(d)).collect()
     }
 }