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 " WHERE" +
  90                 "   hidden = false" +
  91                 "   and deleted_at is null" +  // don't fetch deleted
  92
  93                 "   and created >= '" + filterDate + "' "
  94     }
  95
  96     private fun orderByCreatedDateAndStatus(): String {
  97         return " createdBulkDate DESC ,\n" +
  98                 "  (CASE jobStatus\n" +
  99                 "   WHEN 'COMPLETED' THEN 0\n" +
 100                 "   WHEN 'FAILED' THEN 0\n" +
 101                 "   WHEN 'COMPLETED_WITH_ERRORS' THEN 0\n" +
 102                 "   WHEN 'IN_PROGRESS' THEN 1\n" +
 103                 "   WHEN 'PAUSE' THEN 2\n" +
 104                 "   WHEN 'PENDING' THEN 3\n" +
 105                 "   WHEN 'STOPPED' THEN 3 END),\n" +
 106                 "  statusModifiedDate "
 107     }
 108
 109     fun getAuditStatuses(jobUUID: UUID, source: JobAuditStatus.SourceStatus): List<JobAuditStatus> {
 110         // order by ORDINAL.
 111         // CREATED_DATE is kept for backward compatibility: when all Ordinals are zero
 112         return getResultList(JobAuditStatus::class.java, mapOf("SOURCE" to source, "JOB_ID" to jobUUID), "AND", " ORDINAL, CREATED_DATE ")
 113     }
 114
 115     fun addJobAudiStatus(jobAuditStatus:JobAuditStatus) {
 116         save(jobAuditStatus)
 117     }
 118
 119     private fun <T: DomainVo> save(item:T) {
 120         dataAccessService.saveDomainObject(item, DaoUtils.getPropsMap())
 121     }
 122
 123     private fun <T> getSingleItem(className:Class<T>, filterKey:String, filterValue:Any): T {
 124         val resultList:List<T> = getResultList(className, filterKey, filterValue)
 125         if (resultList.size < 1) {
 126             throw NotFoundException("Failed to retrieve $className with $filterKey $filterValue from table. no resource found")
 127         }else if (resultList.size > 1) {
 128             throw GenericUncheckedException("Failed to retrieve $className with $filterKey $filterValue from table. found more than 1 resources")
 129         }
 130         return resultList[0]
 131     }
 132
 133     private fun <T> getResultList(className:Class<T>, filterKey:String, filterValue:Any): List<T> {
 134         return getResultList(className, mapOf(filterKey to filterValue), "AND", null)
 135     }
 136
 137     private fun <T> getResultList(className:Class<T>, filters: Map<String, Any>, conditionType: String): List<T> {
 138         return getResultList(className, filters, conditionType, null)
 139     }
 140
 141     private fun <T> getResultList(className:Class<T>, filters: Map<String, Any>, conditionType: String, orderBy: String?): List<T> {
 142         var condition:String = filters
 143                 .map{f -> f.key + " = '" + f.value + "'"}
 144                 .joinToString(" $conditionType ")
 145         return dataAccessService.getList(className, " WHERE $condition", orderBy, null) as List<T>
 146     }
 147 }