-
Notifications
You must be signed in to change notification settings - Fork 1
/
CheckpointFromDB.scala
73 lines (65 loc) · 2.48 KB
/
CheckpointFromDB.scala
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
/*
* Copyright 2021 ABSA Group Limited
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package za.co.absa.atum.server.model
import za.co.absa.atum.model.dto.{CheckpointDTO, MeasureDTO, MeasureResultDTO, MeasurementDTO, PartitioningDTO}
import io.circe.{DecodingFailure, Json}
import java.time.ZonedDateTime
import java.util.UUID
case class CheckpointFromDB(
idCheckpoint: Option[UUID],
checkpointName: Option[String],
author: Option[String],
measuredByAtumAgent: Option[Boolean],
measureName: Option[String],
measuredColumns: Option[Seq[String]],
measurementValue: Option[
Json
], // it's easier to convert this attribute to our `MeasurementDTO` after we received this as JSON from DB
checkpointStartTime: Option[ZonedDateTime],
checkpointEndTime: Option[ZonedDateTime]
)
object CheckpointFromDB {
def toCheckpointDTO(
partitioning: PartitioningDTO,
checkpointQueryResult: CheckpointFromDB
): Either[DecodingFailure, CheckpointDTO] = {
val measureResultOrErr = checkpointQueryResult.measurementValue.get.as[MeasureResultDTO]
measureResultOrErr match {
case Left(err) => Left(err)
case Right(measureResult) =>
Right(
CheckpointDTO(
id = checkpointQueryResult.idCheckpoint.get,
name = checkpointQueryResult.checkpointName.get,
author = checkpointQueryResult.author.get,
measuredByAtumAgent = checkpointQueryResult.measuredByAtumAgent.get,
partitioning = partitioning,
processStartTime = checkpointQueryResult.checkpointStartTime.get,
processEndTime = checkpointQueryResult.checkpointEndTime,
measurements = Set(
MeasurementDTO(
measure = MeasureDTO(
measureName = checkpointQueryResult.measureName.get,
measuredColumns = checkpointQueryResult.measuredColumns.get
),
result = measureResult
)
)
)
)
}
}
}