e2624728178c381a0c4d0f4cfa235e9c2d860b40
[vid.git] / vid-app-common / src / main / java / org / onap / vid / dal / AsyncInstantiationRepository.kt
1 /*-
2  * ============LICENSE_START=======================================================
3  * VID
4  * ================================================================================
5  * Copyright (C) 2017 - 2019 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.onap.vid.dal
22
23 import org.onap.portalsdk.core.domain.support.DomainVo
24 import org.onap.portalsdk.core.service.DataAccessService
25 import org.onap.vid.dao.JobRequest
26 import org.onap.vid.exceptions.GenericUncheckedException
27 import org.onap.vid.exceptions.NotFoundException
28 import org.onap.vid.job.Job
29 import org.onap.vid.model.JobAuditStatus
30 import org.onap.vid.model.ResourceInfo
31 import org.onap.vid.model.ServiceInfo
32 import org.onap.vid.model.serviceInstantiation.ServiceInstantiation
33 import org.onap.vid.utils.DaoUtils
34 import org.springframework.beans.factory.annotation.Autowired
35 import org.springframework.stereotype.Repository
36 import java.sql.Timestamp
37 import java.time.LocalDateTime
38 import java.util.*
39 import kotlin.collections.HashMap
40
41 @Repository
42 class AsyncInstantiationRepository @Autowired constructor(val dataAccessService:DataAccessService) {
43
44     fun addJobRequest(jobUuid: UUID, request:ServiceInstantiation) {
45         save(JobRequest(jobUuid, request))
46     }
47
48     fun getJobRequest(jobUuid: UUID):ServiceInstantiation? {
49         return getSingleItem(JobRequest::class.java, "JOB_ID", jobUuid).request
50     }
51
52     fun saveResourceInfo(resource:ResourceInfo) {
53         save(resource)
54     }
55
56     fun getResourceInfoByRootJobId(rootJobId: UUID): Map<String, ResourceInfo> {
57         val resourceInfoList:List<ResourceInfo> = getResultList(ResourceInfo::class.java, "ROOT_JOB_ID", rootJobId)
58
59         if (resourceInfoList.isEmpty()) {
60             throw GenericUncheckedException("Failed to retrieve resource info with rootJobId " + rootJobId + " from ResourceInfo table. no resource found")
61         }
62         return resourceInfoList.fold(HashMap(), { accumulator, item ->
63             accumulator.put(item.trackById, item); accumulator})
64     }
65
66     fun getResourceInfoByTrackId(trackById: String):ResourceInfo {
67         return getSingleItem(ResourceInfo::class.java, "TRACK_BY_ID", trackById)
68     }
69
70     fun saveServiceInfo(serviceInfo: ServiceInfo) {
71         save(serviceInfo)
72     }
73
74     fun getServiceInfoByJobId(jobUuid: UUID): ServiceInfo {
75         return getSingleItem(ServiceInfo::class.java, "jobId", jobUuid)
76     }
77
78     fun getServiceInfoByTemplateIdAndJobStatus(templateId: UUID, jobStatus: Job.JobStatus): List<ServiceInfo> {
79         return getResultList(ServiceInfo::class.java, mapOf("templateId" to templateId, "jobStatus" to jobStatus), "AND")
80     }
81
82     fun getAllServicesInfo(): List<ServiceInfo> {
83         return dataAccessService.getList(ServiceInfo::class.java, filterByCreationDateAndNotDeleted(), orderByCreatedDateAndStatus(), null) as List<ServiceInfo>
84     }
85
86     private fun filterByCreationDateAndNotDeleted(): String {
87         val minus3Months = LocalDateTime.now().minusMonths(3)
88         val filterDate = Timestamp.valueOf(minus3Months)
89         return filterServicesByNotHiddenAndNotDeleted() +
90                 "   and created >= '" + filterDate + "' "
91     }
92
93     private fun filterInstantiatedServiceByServiceModelId(serviceModelUuid: UUID): String {
94         return filterServicesByNotHiddenAndNotDeleted() +
95                 " and SERVICE_MODEL_ID = '$serviceModelUuid'" +
96                 " and ACTION  = 'INSTANTIATE'"
97     }
98
99     private fun filterServicesByNotHiddenAndNotDeleted(): String {
100         return " WHERE" +
101                 "   hidden = false" +
102                 "   and deleted_at is null" // don't fetch deleted
103     }
104
105
106     private fun orderByCreatedDateAndStatus(): String {
107         return " createdBulkDate DESC ,\n" +
108                 "  (CASE jobStatus\n" +
109                 "   WHEN 'COMPLETED' THEN 0\n" +
110                 "   WHEN 'FAILED' THEN 0\n" +
111                 "   WHEN 'COMPLETED_WITH_ERRORS' THEN 0\n" +
112                 "   WHEN 'IN_PROGRESS' THEN 1\n" +
113                 "   WHEN 'PAUSE' THEN 2\n" +
114                 "   WHEN 'PENDING' THEN 3\n" +
115                 "   WHEN 'STOPPED' THEN 3 END),\n" +
116                 "  statusModifiedDate "
117     }
118
119     fun getAuditStatuses(jobUUID: UUID, source: JobAuditStatus.SourceStatus): List<JobAuditStatus> {
120         // order by ORDINAL.
121         // CREATED_DATE is kept for backward compatibility: when all Ordinals are zero
122         return getResultList(JobAuditStatus::class.java, mapOf("SOURCE" to source, "JOB_ID" to jobUUID), "AND", " ORDINAL, CREATED_DATE ")
123     }
124
125     fun addJobAudiStatus(jobAuditStatus:JobAuditStatus) {
126         save(jobAuditStatus)
127     }
128
129     private fun <T: DomainVo> save(item:T) {
130         dataAccessService.saveDomainObject(item, DaoUtils.getPropsMap())
131     }
132
133     private fun <T> getSingleItem(className:Class<T>, filterKey:String, filterValue:Any): T {
134         val resultList:List<T> = getResultList(className, filterKey, filterValue)
135         if (resultList.size < 1) {
136             throw NotFoundException("Failed to retrieve $className with $filterKey $filterValue from table. no resource found")
137         }else if (resultList.size > 1) {
138             throw GenericUncheckedException("Failed to retrieve $className with $filterKey $filterValue from table. found more than 1 resources")
139         }
140         return resultList[0]
141     }
142
143     private fun <T> getResultList(className:Class<T>, filterKey:String, filterValue:Any): List<T> {
144         return getResultList(className, mapOf(filterKey to filterValue), "AND", null)
145     }
146
147     private fun <T> getResultList(className:Class<T>, filters: Map<String, Any>, conditionType: String): List<T> {
148         return getResultList(className, filters, conditionType, null)
149     }
150
151     private fun <T> getResultList(className:Class<T>, filters: Map<String, Any>, conditionType: String, orderBy: String?): List<T> {
152         var condition:String = filters
153                 .map{f -> f.key + " = '" + f.value + "'"}
154                 .joinToString(" $conditionType ")
155         return dataAccessService.getList(className, " WHERE $condition", orderBy, null) as List<T>
156     }
157
158     fun listInstantiatedServicesByServiceModelId(serviceModelId: UUID): List<ServiceInfo> =
159             dataAccessService.getList(ServiceInfo::class.java, filterInstantiatedServiceByServiceModelId(serviceModelId), orderByCreatedDateAndStatus(), null) as List<ServiceInfo>;
160 }