Use mapChunksZIO on streams instead of mapZIO (#2046)

ghostdogpr · Dec 18, 2023 · cf62c7b · cf62c7b
1 parent 9dd2287
commit cf62c7b
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/core/src/main/scala/caliban/execution/Executor.scala b/core/src/main/scala/caliban/execution/Executor.scala
@@ -239,12 +239,11 @@ object Executor {
           case ReducedStep.ListStep(steps, areItemsNullable) => makeListQuery(steps, areItemsNullable)
           case ReducedStep.StreamStep(stream)                =>
             ZQuery
-              .environment[R]
-              .map(env =>
+              .environmentWith[R](env =>
                 ResponseValue.StreamValue(
-                  stream
-                    .mapZIO(loop(_).catchAll(_ => ZQuery.succeed(NullValue)).run)
-                    .provideEnvironment(env)
+                  stream.mapChunksZIO { chunk =>
+                    collectAll(chunk, isTopLevelField)(loop(_).catchAll(_ => ZQuery.succeed(NullValue))).run
+                  }.provideEnvironment(env)
                 )
               )
           case ReducedStep.DeferStep(obj, nextSteps, path)   =>