Merge pull request #287 from k-nasa/optimizing_allocations

Optimizing allocations using Stream::size_hint

src/collections/binary_heap/extend.rs

@@ -10,9 +10,9 @@ impl<T: Ord> Extend<T> for BinaryHeap<T> {
         stream: S,
     ) -> Pin<Box<dyn Future<Output = ()> + 'a>> {
         let stream = stream.into_stream();
-        //TODO: Add this back in when size_hint is added to Stream/StreamExt
-        //let (lower_bound, _) = stream.size_hint();
-        //self.reserve(lower_bound);
+
+        self.reserve(stream.size_hint().0);
+
         Box::pin(stream.for_each(move |item| self.push(item)))
     }
 }

src/collections/hash_map/extend.rs

@@ -23,13 +23,12 @@ where
         // hint lower bound if the map is empty. Otherwise reserve half the hint (rounded up), so
         // the map will only resize twice in the worst case.
 
-        //TODO: Add this back in when size_hint is added to Stream/StreamExt
-        //let reserve = if self.is_empty() {
-        //    stream.size_hint().0
-        //} else {
-        //    (stream.size_hint().0 + 1) / 2
-        //};
-        //self.reserve(reserve);
+        let additional = if self.is_empty() {
+            stream.size_hint().0
+        } else {
+            (stream.size_hint().0 + 1) / 2
+        };
+        self.reserve(additional);
 
         Box::pin(stream.for_each(move |(k, v)| {
             self.insert(k, v);

src/collections/hash_set/extend.rs

@@ -26,13 +26,12 @@ where
         // hint lower bound if the map is empty. Otherwise reserve half the hint (rounded up), so
         // the map will only resize twice in the worst case.
 
-        //TODO: Add this back in when size_hint is added to Stream/StreamExt
-        //let reserve = if self.is_empty() {
-        //    stream.size_hint().0
-        //} else {
-        //    (stream.size_hint().0 + 1) / 2
-        //};
-        //self.reserve(reserve);
+        let additional = if self.is_empty() {
+            stream.size_hint().0
+        } else {
+            (stream.size_hint().0 + 1) / 2
+        };
+        self.reserve(additional);
 
         Box::pin(stream.for_each(move |item| {
             self.insert(item);

src/collections/linked_list/extend.rs

@@ -10,9 +10,6 @@ impl<T> Extend<T> for LinkedList<T> {
         stream: S,
     ) -> Pin<Box<dyn Future<Output = ()> + 'a>> {
         let stream = stream.into_stream();
-        //TODO: Add this back in when size_hint is added to Stream/StreamExt
-        //let (lower_bound, _) = stream.size_hint();
-        //self.reserve(lower_bound);
         Box::pin(stream.for_each(move |item| self.push_back(item)))
     }
 }

src/collections/vec_deque/extend.rs

@@ -10,9 +10,9 @@ impl<T> Extend<T> for VecDeque<T> {
         stream: S,
     ) -> Pin<Box<dyn Future<Output = ()> + 'a>> {
         let stream = stream.into_stream();
-        //TODO: Add this back in when size_hint is added to Stream/StreamExt
-        //let (lower_bound, _) = stream.size_hint();
-        //self.reserve(lower_bound);
+
+        self.reserve(stream.size_hint().0);
+
         Box::pin(stream.for_each(move |item| self.push_back(item)))
     }
 }

src/string/extend.rs

@@ -10,9 +10,8 @@ impl Extend<char> for String {
         stream: S,
     ) -> Pin<Box<dyn Future<Output = ()> + 'a>> {
         let stream = stream.into_stream();
-        //TODO: Add this back in when size_hint is added to Stream/StreamExt
-        // let (lower_bound, _) = stream.size_hint();
-        // self.reserve(lower_bound);
+
+        self.reserve(stream.size_hint().0);
 
         Box::pin(stream.for_each(move |c| self.push(c)))
     }

src/vec/extend.rs

@@ -9,9 +9,9 @@ impl<T> Extend<T> for Vec<T> {
         stream: S,
     ) -> Pin<Box<dyn Future<Output = ()> + 'a>> {
         let stream = stream.into_stream();
-        //TODO: Add this back in when size_hint is added to Stream/StreamExt
-        //let (lower_bound, _) = stream.size_hint();
-        //self.reserve(lower_bound);
+
+        self.reserve(stream.size_hint().0);
+
         Box::pin(stream.for_each(move |item| self.push(item)))
     }
 }