f87721f11859a450144369eef8a2664d538667fd
[ccsdk/cds.git] /
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.controllerblueprints.service.enhancer
18
19 import com.att.eelf.configuration.EELFLogger
20 import com.att.eelf.configuration.EELFManager
21 import org.onap.ccsdk.apps.controllerblueprints.core.BluePrintError
22 import org.onap.ccsdk.apps.controllerblueprints.core.BluePrintException
23 import org.onap.ccsdk.apps.controllerblueprints.core.data.ArtifactType
24 import org.onap.ccsdk.apps.controllerblueprints.core.data.NodeTemplate
25 import org.onap.ccsdk.apps.controllerblueprints.core.data.NodeType
26 import org.onap.ccsdk.apps.controllerblueprints.core.format
27 import org.onap.ccsdk.apps.controllerblueprints.core.interfaces.BluePrintNodeTemplateEnhancer
28 import org.onap.ccsdk.apps.controllerblueprints.core.interfaces.BluePrintRepoService
29 import org.onap.ccsdk.apps.controllerblueprints.core.interfaces.BluePrintTypeEnhancerService
30 import org.onap.ccsdk.apps.controllerblueprints.core.service.BluePrintContext
31 import org.springframework.beans.factory.config.ConfigurableBeanFactory
32 import org.springframework.context.annotation.Scope
33 import org.springframework.stereotype.Service
34
35 @Service
36 @Scope(value = ConfigurableBeanFactory.SCOPE_PROTOTYPE)
37 open class BluePrintNodeTemplateEnhancerImpl(private val bluePrintRepoService: BluePrintRepoService,
38                                              private val bluePrintTypeEnhancerService: BluePrintTypeEnhancerService)
39     : BluePrintNodeTemplateEnhancer {
40
41     private val log: EELFLogger = EELFManager.getInstance().getLogger(BluePrintNodeTemplateEnhancerImpl::class.toString())
42
43     lateinit var bluePrintContext: BluePrintContext
44     lateinit var error: BluePrintError
45
46     override fun enhance(bluePrintContext: BluePrintContext, error: BluePrintError, name: String, nodeTemplate: NodeTemplate) {
47         log.info("Enhancing NodeTemplate($name)")
48         this.bluePrintContext = bluePrintContext
49         this.error = error
50
51         val nodeTypeName = nodeTemplate.type
52         // Get NodeType from Repo and Update Service Template
53         val nodeType = populateNodeType(nodeTypeName)
54
55         // Enrich NodeType
56         bluePrintTypeEnhancerService.enhanceNodeType(bluePrintContext, error, nodeTypeName, nodeType)
57
58         //Enrich Node Template Artifacts
59         enhanceNodeTemplateArtifactDefinition(name, nodeTemplate)
60     }
61
62
63     open fun populateNodeType(nodeTypeName: String): NodeType {
64
65         val nodeType = bluePrintContext.serviceTemplate.nodeTypes?.get(nodeTypeName)
66                 ?: bluePrintRepoService.getNodeType(nodeTypeName)
67                 ?: throw BluePrintException(format("Couldn't get NodeType({}) from repo.", nodeTypeName))
68         bluePrintContext.serviceTemplate.nodeTypes?.put(nodeTypeName, nodeType)
69         return nodeType
70     }
71
72     open fun enhanceNodeTemplateArtifactDefinition(nodeTemplateName: String, nodeTemplate: NodeTemplate) {
73
74         nodeTemplate.artifacts?.forEach { artifactDefinitionName, artifactDefinition ->
75             val artifactTypeName = artifactDefinition.type
76                     ?: throw BluePrintException(format("Artifact type is missing for NodeTemplate({}) artifact({})", nodeTemplateName, artifactDefinitionName))
77
78             // Populate Artifact Type
79             populateArtifactType(artifactTypeName)
80         }
81     }
82
83     open fun populateArtifactType(artifactTypeName: String): ArtifactType {
84         val artifactType = bluePrintContext.serviceTemplate.artifactTypes?.get(artifactTypeName)
85                 ?: bluePrintRepoService.getArtifactType(artifactTypeName)
86                 ?: throw BluePrintException(format("Couldn't get ArtifactType({}) from repo.", artifactTypeName))
87         bluePrintContext.serviceTemplate.artifactTypes?.put(artifactTypeName, artifactType)
88         return artifactType
89     }
90
91 }