gerrit.onap Code Review - ccsdk/cds.git/blob

   1 /*
   2  * Copyright © 2017-2018 AT&T Intellectual Property.
   3  *
   4  * Modifications Copyright © 2019 IBM, Bell Canada.
   5  *
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *     http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  */
  18
  19 package org.onap.ccsdk.apps.blueprintsprocessor.functions.resource.resolution.processor
  20
  21 import org.onap.ccsdk.apps.blueprintsprocessor.functions.python.executor.BlueprintJythonService
  22 import org.onap.ccsdk.apps.blueprintsprocessor.functions.resource.resolution.CapabilityResourceSource
  23 import org.onap.ccsdk.apps.controllerblueprints.core.BluePrintConstants
  24 import org.onap.ccsdk.apps.controllerblueprints.core.BluePrintProcessorException
  25 import org.onap.ccsdk.apps.controllerblueprints.core.interfaces.BluePrintScriptsService
  26 import org.onap.ccsdk.apps.controllerblueprints.core.utils.JacksonUtils
  27 import org.onap.ccsdk.apps.controllerblueprints.resource.dict.ResourceAssignment
  28 import org.slf4j.LoggerFactory
  29 import org.springframework.context.ApplicationContext
  30 import org.springframework.stereotype.Service
  31 import java.io.File
  32
  33 @Service("resource-assignment-processor-capability")
  34 open class CapabilityResourceAssignmentProcessor(private var applicationContext: ApplicationContext,
  35                                                  private val bluePrintScriptsService: BluePrintScriptsService,
  36                                                  private val bluePrintJythonService: BlueprintJythonService) :
  37         ResourceAssignmentProcessor() {
  38
  39     companion object {
  40         const val CAPABILITY_TYPE_KOTLIN_COMPONENT = "KOTLIN-COMPONENT"
  41         const val CAPABILITY_TYPE_JAVA_COMPONENT = "JAVA-COMPONENT"
  42         const val CAPABILITY_TYPE_JYTHON_COMPONENT = "JYTHON-COMPONENT"
  43     }
  44
  45     override fun getName(): String {
  46         return "resource-assignment-processor-capability"
  47     }
  48
  49     override fun process(resourceAssignment: ResourceAssignment) {
  50
  51         val resourceDefinition = resourceDictionaries[resourceAssignment.dictionaryName]
  52                 ?: throw BluePrintProcessorException("couldn't get resource definition for ${resourceAssignment.dictionaryName}")
  53
  54         val resourceSource = resourceDefinition.sources[resourceAssignment.dictionarySource]
  55                 ?: throw BluePrintProcessorException("couldn't get resource definition " +
  56                         "${resourceAssignment.dictionaryName} source(${resourceAssignment.dictionarySource})")
  57
  58         val resourceSourceProps = checkNotNull(resourceSource.properties) { "failed to get $resourceSource properties" }
  59         /**
  60          * Get the Capability Resource Source Info from Property Definitions.
  61          */
  62         val capabilityResourceSourceProperty = JacksonUtils
  63                 .getInstanceFromMap(resourceSourceProps, CapabilityResourceSource::class.java)
  64
  65         val instanceType = capabilityResourceSourceProperty.type
  66         val instanceName = capabilityResourceSourceProperty.instanceName
  67
  68         var componentResourceAssignmentProcessor: ResourceAssignmentProcessor? = null
  69
  70         when (instanceType) {
  71             CAPABILITY_TYPE_KOTLIN_COMPONENT -> {
  72                 componentResourceAssignmentProcessor = getKotlinResourceAssignmentProcessorInstance(instanceName,
  73                         capabilityResourceSourceProperty.instanceDependencies)
  74             }
  75             CAPABILITY_TYPE_JAVA_COMPONENT -> {
  76                 // Initialize Capability Resource Assignment Processor
  77                 componentResourceAssignmentProcessor = applicationContext.getBean(instanceName, ResourceAssignmentProcessor::class.java)
  78             }
  79             CAPABILITY_TYPE_JYTHON_COMPONENT -> {
  80                 val content = getJythonContent(instanceName)
  81                 componentResourceAssignmentProcessor = getJythonResourceAssignmentProcessorInstance(instanceName,
  82                         content, capabilityResourceSourceProperty.instanceDependencies)
  83             }
  84         }
  85
  86         checkNotNull(componentResourceAssignmentProcessor) { "failed to get capability resource assignment processor($instanceName)" }
  87
  88         // Assign Current Blueprint runtime and ResourceDictionaries
  89         componentResourceAssignmentProcessor.raRuntimeService = raRuntimeService
  90         componentResourceAssignmentProcessor.resourceDictionaries = resourceDictionaries
  91
  92         // Invoke componentResourceAssignmentProcessor
  93         componentResourceAssignmentProcessor.apply(resourceAssignment)
  94     }
  95
  96     override fun recover(runtimeException: RuntimeException, resourceAssignment: ResourceAssignment) {
  97
  98         TODO("To Implement")
  99     }
 100
 101     private fun getKotlinResourceAssignmentProcessorInstance(scriptClassName: String,
 102                                                              instanceNames: List<String>? = null): ResourceAssignmentProcessor {
 103         var scriptPropertyInstances: MutableMap<String, Any>? = null
 104
 105         if (instanceNames != null && instanceNames.isNotEmpty()) {
 106             scriptPropertyInstances = hashMapOf()
 107             instanceNames.forEach {
 108                 scriptPropertyInstances[it] = applicationContext.getBean(it)
 109                         ?: throw BluePrintProcessorException("couldn't get the dependency instance($it)")
 110             }
 111         }
 112
 113         return getKotlinResourceAssignmentProcessorInstance(scriptClassName, scriptPropertyInstances)
 114
 115     }
 116
 117     fun getKotlinResourceAssignmentProcessorInstance(scriptClassName: String,
 118                                                      scriptPropertyInstances: MutableMap<String, Any>? = null):
 119             ResourceAssignmentProcessor {
 120
 121         val resourceAssignmentProcessor = bluePrintScriptsService
 122                 .scriptInstance<ResourceAssignmentProcessor>(raRuntimeService.bluePrintContext(),
 123                         scriptClassName, false)
 124
 125         // Add additional Instance
 126         if (scriptPropertyInstances != null) {
 127             resourceAssignmentProcessor.scriptPropertyInstances = scriptPropertyInstances
 128         }
 129
 130         return resourceAssignmentProcessor
 131     }
 132
 133     private fun getJythonContent(instanceName: String): String {
 134         val absolutePath = raRuntimeService.bluePrintContext().rootPath
 135                 .plus(File.separator)
 136                 .plus(BluePrintConstants.TOSCA_SCRIPTS_JYTHON_DIR)
 137                 .plus(File.separator)
 138                 .plus("$instanceName.py")
 139
 140         return JacksonUtils.getContent(absolutePath)
 141
 142     }
 143
 144     /**
 145      * getJythonResourceAssignmentProcessorInstance Purpose: prepare the jython
 146      * executor component as a resource assignment processor
 147      *
 148      * @param pythonClassName String
 149      * @param content String
 150      * @param dependencyInstances List<String>
 151      * @return resourceAssignmentProcessor ResourceAssignmentProcessor
 152      */
 153     private fun getJythonResourceAssignmentProcessorInstance(pythonClassName: String, content: String,
 154                                                              dependencyInstances: List<String>?):
 155             ResourceAssignmentProcessor {
 156         val jythonContextInstance: MutableMap<String, Any> = hashMapOf()
 157         jythonContextInstance["log"] = LoggerFactory.getLogger(pythonClassName)
 158         jythonContextInstance["raRuntimeService"] = raRuntimeService
 159         dependencyInstances?.forEach { instanceName ->
 160             jythonContextInstance[instanceName] = applicationContext.getBean(instanceName)
 161         }
 162
 163         return getJythonResourceAssignmentProcessorInstance(pythonClassName, content, jythonContextInstance)
 164     }
 165
 166     fun getJythonResourceAssignmentProcessorInstance(pythonClassName: String, content: String,
 167                                                      dependencyInstances: MutableMap<String, Any>):
 168             ResourceAssignmentProcessor {
 169
 170         val resourceAssignmentProcessor = bluePrintJythonService
 171                 .jythonInstance<ResourceAssignmentProcessor>(raRuntimeService.bluePrintContext(), pythonClassName,
 172                         content, dependencyInstances)
 173
 174         // Add additional Instance
 175         if (dependencyInstances != null) {
 176             resourceAssignmentProcessor.scriptPropertyInstances = dependencyInstances
 177         }
 178
 179         return resourceAssignmentProcessor
 180     }
 181 }