give sizeHint

yoohaemin · Jun 9, 2024 · 05d4c39 · 05d4c39
1 parent d5533c8
commit 05d4c39
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/zquery/src/main/scala/decrel/reify/zquery.scala b/zquery/src/main/scala/decrel/reify/zquery.scala
@@ -62,8 +62,9 @@ trait zquery[R] extends bifunctor.module[ZQuery[R, +*, +*]] {
         val deduplicated = requests.distinctBy(_.id)
 
         batchExecute(deduplicated.map(_.id)).flatMap { results =>
-          val resultsMap = mutable.Map
-            .newBuilder[In, Exit[E, Out]]
+          val mapBuilder = mutable.Map.newBuilder[In, Exit[E, Out]]
+          mapBuilder.sizeHint(results)
+          val resultsMap = mapBuilder
             .addAll(results.view.map(pair => pair._1 -> Exit.succeed(pair._2)))
             .result()
             .withDefault(in =>