Fixes #120 - added value parameters

AbsaOSS · Mar 19, 2024 · 57a1fd7 · 57a1fd7
1 parent 52449d3
commit 57a1fd7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/server/src/main/scala/za/co/absa/atum/server/Main.scala b/server/src/main/scala/za/co/absa/atum/server/Main.scala
@@ -18,7 +18,7 @@ package za.co.absa.atum.server
 
 import za.co.absa.atum.server.api.controller._
 import za.co.absa.atum.server.api.database.{PostgresDatabaseProvider, TransactorProvider}
-import za.co.absa.atum.server.api.database.runs.functions.{CreatePartitioningIfNotExists, WriteCheckpoint}
+import za.co.absa.atum.server.api.database.runs.functions.{CreatePartitioningIfNotExists, WriteCheckpoint, GetPartitioningMeasures, GetPartitioningAdditionalData}
 import za.co.absa.atum.server.api.http.Server
 import za.co.absa.atum.server.api.repository.{CheckpointRepositoryImpl, PartitioningRepositoryImpl}
 import za.co.absa.atum.server.api.service.{CheckpointServiceImpl, PartitioningServiceImpl}
@@ -41,6 +41,8 @@ object Main extends ZIOAppDefault with Server {
         PartitioningRepositoryImpl.layer,
         CheckpointRepositoryImpl.layer,
         CreatePartitioningIfNotExists.layer,
+        GetPartitioningMeasures.layer,
+        GetPartitioningAdditionalData.layer,
         WriteCheckpoint.layer,
         PostgresDatabaseProvider.layer,
         TransactorProvider.layer,

diff --git a/server/src/main/scala/za/co/absa/atum/server/api/repository/PartitioningRepository.scala b/server/src/main/scala/za/co/absa/atum/server/api/repository/PartitioningRepository.scala
@@ -89,8 +89,7 @@ class PartitioningRepositoryImpl(createPartitioningIfNotExistsFn: CreatePartitio
 }
 
 object PartitioningRepositoryImpl {
-//  To ask about the behavior
-  val layer: URLayer[CreatePartitioningIfNotExists, PartitioningRepository] = ZLayer {
+  val layer: URLayer[CreatePartitioningIfNotExists with GetPartitioningMeasures with GetPartitioningAdditionalData, PartitioningRepository] = ZLayer {
     for {
       createPartitioningIfNotExists <- ZIO.service[CreatePartitioningIfNotExists]
       getPartitioningMeasures <- ZIO.service[GetPartitioningMeasures]