Update agent/src/main/scala/za/co/absa/atum/agent/dispatcher/Dispatch…

…er.scala Co-authored-by: Adam Cervenka <adam.cervenka@absa.africa>
AbsaOSS · Oct 12, 2023 · 4666055 · 4666055
1 parent 6d8f769
commit 4666055
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agent/src/main/scala/za/co/absa/atum/agent/dispatcher/Dispatcher.scala b/agent/src/main/scala/za/co/absa/atum/agent/dispatcher/Dispatcher.scala
@@ -20,7 +20,7 @@ import za.co.absa.atum.model.Partitioning
 import za.co.absa.atum.model.dto.{AtumContextDTO, CheckpointDTO}
 
 trait Dispatcher {
-  def fetchAtumContext(partitioning: Partitioning, parentPartitioning: Option[Partitioning]): Option[AtumContextDTO]
+  def getOrCreateAtumContext(partitioning: Partitioning, parentPartitioning: Option[Partitioning]): AtumContextDTO
 
   def saveCheckpoint(checkpoint: CheckpointDTO): Unit
 }