Try out for comprehension for clarity

BlueBrain · Aug 3, 2023 · d7181fe · d7181fe
1 parent 3053b0b
commit d7181fe
Showing 1 changed file with 14 additions and 13 deletions.
diff --git a/...s/src/main/scala/ch/epfl/bluebrain/nexus/delta/plugins/compositeviews/CompositeSink.scala b/...s/src/main/scala/ch/epfl/bluebrain/nexus/delta/plugins/compositeviews/CompositeSink.scala
@@ -123,19 +123,20 @@ final class Batch[SinkFormat](
   }
 
   override def apply(elements: Chunk[Elem[GraphResource]]): Task[Chunk[Elem[Unit]]] =
-    query(elements)
-      .flatMap {
-        case Some(fullGraph) =>
-          elements.traverse { elem =>
-            elem.evalMapFilter { gr =>
-              replaceGraph(gr, fullGraph).flatMap(transform)
-            }
-          }
-        case None            =>
-          Task.pure(elements.map(_.drop))
-      }
-      .flatMap(sink)
-}
+    for {
+      graph       <- query(elements)
+      transformed <- graph match {
+                       case Some(fullGraph) =>
+                         elements.traverse { elem =>
+                           elem.evalMapFilter { gr =>
+                             replaceGraph(gr, fullGraph).flatMap(transform)
+                           }
+                         }
+                       case None            =>
+                         Task.pure(elements.map(_.drop))
+                     }
+      sank        <- sink(transformed)
+    } yield sank
 
 object CompositeSink {