fixup! WIP: Rewrite to store loaders _and_ loaded configrethink

1 files changed, 26 insertions, 15 deletions

diff --git a/src/config/builder.rs b/src/config/builder.rs
index 9e3cce9..3748921 100644
--- a/src/config/builder.rs
+++ b/src/config/builder.rs
@@ -92,21 +92,32 @@ impl<'source> AsyncConfigBuilder<'source> {
         Config::build_from_async_builder(self).await
     }
 
-    pub(crate) async fn reload(&self) -> Result<Vec<ConfigObject<'source>>, SourceError> {
-        use futures::stream::FuturesOrdered;
-        use futures::stream::StreamExt;
-        use futures::FutureExt;
-        use itertools::Itertools;
+    pub(crate) async fn reload(&'source self) -> Result<Vec<ConfigObject<'source>>, SourceError> {
+        async fn do_load<'source>(builders: &'source Vec<Box<dyn AsyncConfigSource<'source>>>) -> Result<Vec<ConfigObject<'source>>, SourceError> {
+            let mut v = Vec::with_capacity(builders.len());
+            for cs in builders.iter() {
+                v.push(cs.load().await?);
+            }
+            Ok(v)
+        }
 
-        self.overwrites_builders
-            .iter()
-            .map(|cs| async move { cs.load().await }.boxed())
-            .chain(self.layers_builders.iter().map(|cs| async move { cs.load().await }.boxed()))
-            .chain(self.defaults_builders.iter().map(|cs| async move { cs.load().await }.boxed()))
-            .collect::<FuturesOrdered<_>>()
-            .collect::<Vec<Result<ConfigObject<'source>, SourceError>>>()
-            .await
-            .into_iter()
-            .collect()
+        let overwrites = do_load(&self.overwrites_builders);
+        let layers = do_load(&self.layers_builders);
+        let defaults = do_load(&self.defaults_builders);
+
+        let (mut overwrites, mut layers, mut defaults) =
+            futures::try_join!(overwrites, layers, defaults)?;
+
+        let mut v = Vec::with_capacity({
+            self.layers_builders.len()
+                + self.defaults_builders.len()
+                + self.overwrites_builders.len()
+        });
+
+        v.append(&mut overwrites);
+        v.append(&mut layers);
+        v.append(&mut defaults);
+
+        Ok(v)
     }
 }