1370a479676e0035ec0bd99fedfdcf6aafcc1768
[ccsdk/cds.git] / ms / blueprintsprocessor / functions / resource-resolution / src / main / kotlin / org / onap / ccsdk / apps / blueprintsprocessor / functions / resource / resolution / processor / CapabilityResourceAssignmentProcessor.kt
1 /*
2  * Copyright © 2017-2018 AT&T Intellectual Property.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.onap.ccsdk.apps.blueprintsprocessor.functions.resource.resolution.processor
18
19 import org.onap.ccsdk.apps.blueprintsprocessor.functions.resource.resolution.CapabilityResourceSource
20 import org.onap.ccsdk.apps.controllerblueprints.core.BluePrintProcessorException
21 import org.onap.ccsdk.apps.controllerblueprints.core.interfaces.BluePrintScriptsService
22 import org.onap.ccsdk.apps.controllerblueprints.core.utils.JacksonUtils
23 import org.onap.ccsdk.apps.controllerblueprints.resource.dict.ResourceAssignment
24 import org.springframework.context.ApplicationContext
25 import org.springframework.stereotype.Service
26
27 @Service("resource-assignment-processor-capability")
28 open class CapabilityResourceAssignmentProcessor(private var applicationContext: ApplicationContext,
29                                                  private val bluePrintScriptsService: BluePrintScriptsService) :
30         ResourceAssignmentProcessor() {
31
32     companion object {
33         const val CAPABILITY_TYPE_KOTLIN_COMPONENT = "KOTLIN-COMPONENT"
34         const val CAPABILITY_TYPE_JAVA_COMPONENT = "JAVA-COMPONENT"
35         const val CAPABILITY_TYPE_JYTHON_COMPONENT = "JYTHON-COMPONENT"
36     }
37
38     override fun getName(): String {
39         return "resource-assignment-processor-capability"
40     }
41
42     override fun process(resourceAssignment: ResourceAssignment) {
43
44         val resourceDefinition = resourceDictionaries[resourceAssignment.dictionaryName]
45                 ?: throw BluePrintProcessorException("couldn't get resource definition for ${resourceAssignment.dictionaryName}")
46
47         val resourceSource = resourceDefinition.sources[resourceAssignment.dictionarySource]
48                 ?: throw BluePrintProcessorException("couldn't get resource definition " +
49                         "${resourceAssignment.dictionaryName} source(${resourceAssignment.dictionarySource})")
50
51         val resourceSourceProps = checkNotNull(resourceSource.properties) { "failed to get $resourceSource properties" }
52         /**
53          * Get the Capability Resource Source Info from Property Definitions.
54          */
55         val capabilityResourceSourceProperty = JacksonUtils
56                 .getInstanceFromMap(resourceSourceProps, CapabilityResourceSource::class.java)
57
58         val instanceType = capabilityResourceSourceProperty.type
59         val instanceName = capabilityResourceSourceProperty.instanceName
60
61         var componentResourceAssignmentProcessor: ResourceAssignmentProcessor? = null
62
63         when (instanceType) {
64             CAPABILITY_TYPE_KOTLIN_COMPONENT -> {
65                 componentResourceAssignmentProcessor = getKotlinResourceAssignmentProcessorInstance(instanceName,
66                         capabilityResourceSourceProperty.instanceDependencies)
67             }
68             CAPABILITY_TYPE_JAVA_COMPONENT -> {
69                 // Initialize Capability Resource Assignment Processor
70                 componentResourceAssignmentProcessor = applicationContext.getBean(instanceName, ResourceAssignmentProcessor::class.java)
71             }
72             CAPABILITY_TYPE_JYTHON_COMPONENT -> {
73                 TODO(" No implementation")
74             }
75         }
76
77         checkNotNull(componentResourceAssignmentProcessor) { "failed to get capability resource assignment processor($instanceName)" }
78
79         // Assign Current Blueprint runtime and ResourceDictionaries
80         componentResourceAssignmentProcessor.raRuntimeService = raRuntimeService
81         componentResourceAssignmentProcessor.resourceDictionaries = resourceDictionaries
82
83         // Invoke componentResourceAssignmentProcessor
84         componentResourceAssignmentProcessor.apply(resourceAssignment)
85     }
86
87     override fun recover(runtimeException: RuntimeException, resourceAssignment: ResourceAssignment) {
88
89         TODO("To Implement")
90     }
91
92     private fun getKotlinResourceAssignmentProcessorInstance(scriptClassName: String,
93                                                              instanceNames: List<String>? = null): ResourceAssignmentProcessor {
94         var scriptPropertyInstances: MutableMap<String, Any>? = null
95
96         if (instanceNames != null && instanceNames.isNotEmpty()) {
97             scriptPropertyInstances = hashMapOf()
98             instanceNames.forEach {
99                 scriptPropertyInstances[it] = applicationContext.getBean(it)
100                         ?: throw BluePrintProcessorException("couldn't get the dependency instance($it)")
101             }
102         }
103
104         return getKotlinResourceAssignmentProcessorInstance(scriptClassName, scriptPropertyInstances)
105
106     }
107
108     fun getKotlinResourceAssignmentProcessorInstance(scriptClassName: String,
109                                                      scriptPropertyInstances: MutableMap<String, Any>? = null):
110             ResourceAssignmentProcessor {
111
112         val resourceAssignmentProcessor = bluePrintScriptsService
113                 .scriptInstance<ResourceAssignmentProcessor>(raRuntimeService.bluePrintContext(),
114                         scriptClassName, false)
115
116         // Add additional Instance
117         if (scriptPropertyInstances != null) {
118             resourceAssignmentProcessor.scriptPropertyInstances = scriptPropertyInstances
119         }
120
121         return resourceAssignmentProcessor
122     }
123 }