refactor providing controllingEm to reviewers comments

ie3-institute · Jul 4, 2024 · bc214b6 · bc214b6
1 parent b37d99c
commit bc214b6
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/main/scala/edu/ie3/simona/agent/grid/GridAgentController.scala b/src/main/scala/edu/ie3/simona/agent/grid/GridAgentController.scala
@@ -202,8 +202,16 @@ class GridAgentController(
         val node = participant.getNode
 
         val controllingEm =
-          participant.getControllingEm.toScala.flatMap(em => Option(em.getUuid))
-        val emActorRef = controllingEm.flatMap(uuid => allEms.get(uuid))
+          participant.getControllingEm.toScala
+            .map(_.getUuid)
+            .map(uuid =>
+              allEms.getOrElse(
+                uuid,
+                throw new CriticalFailureException(
+                  s"EM actor with UUID $uuid not found."
+                ),
+              )
+            )
 
         val actorRef = buildParticipantActor(
           participantsConfig.requestVoltageDeviationThreshold,
@@ -212,7 +220,7 @@ class GridAgentController(
           participant,
           thermalIslandGridsByBusId,
           environmentRefs,
-          emActorRef,
+          controllingEm,
         )
         introduceAgentToEnvironment(actorRef)
         // return uuid to actorRef