for comprehesion

Author: thomaschow

api/src/main/scala/ai/chronon/api/planner/MonolithJoinPlanner.scala

@@ -26,15 +26,13 @@ case class MonolithJoinPlanner(join: Join)(implicit outputPartitionSpec: Partiti
     val node = new planner.MonolithJoinNode().setJoin(join)
     val finalNode = toNode(metaData, _.setMonolithJoin(node), join)
 
-    val terminalNodeNames = Option(finalNode)
-      .map(_.metaData)
-      .map(_.name)
-      .map((name) =>
-        Map(
-          planner.Mode.BACKFILL -> name
-        ))
-      .getOrElse(Map.empty)
-      .asJava
+    val terminalNodeNames: java.util.Map[planner.Mode, String] = (
+      for {
+        fin <- Option(finalNode)
+        metaData <- Option(fin.metaData)
+        name <- Option(metaData.name)
+      } yield Map(planner.Mode.BACKFILL -> name)
+    ).getOrElse(Map.empty).asJava
     confPlan.setNodes(List(finalNode).asJava).setTerminalNodeNames(terminalNodeNames)
   }
 }