b1a0976b36b2dfbb07c99fb25a05db745ad033ee
[sdc.git] / openecomp-be / lib / openecomp-tosca-lib / src / main / java / org / openecomp / sdc / tosca / datatypes / ToscaServiceModel.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * SDC
4  * ================================================================================
5  * Copyright (C) 2017 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.openecomp.sdc.tosca.datatypes;
22
23 import org.apache.commons.collections.MapUtils;
24 import org.openecomp.core.utilities.file.FileContentHandler;
25 import org.openecomp.sdc.datatypes.model.AsdcModel;
26 import org.openecomp.sdc.tosca.datatypes.model.ServiceTemplate;
27 import org.openecomp.sdc.tosca.services.DataModelUtil;
28
29 import java.util.Collections;
30 import java.util.HashMap;
31 import java.util.Map;
32 import java.util.Optional;
33
34 /**
35  * Tosca service model.
36  */
37 public class ToscaServiceModel implements AsdcModel {
38   private FileContentHandler artifactFiles;
39   private Map<String, ServiceTemplate> serviceTemplates;
40   private String entryDefinitionServiceTemplate;
41
42   public ToscaServiceModel() {
43   }
44
45   /**
46    * Instantiates a new Tosca service model.
47    *
48    * @param artifactFiles                  the artifact files
49    * @param serviceTemplates               the service templates
50    * @param entryDefinitionServiceTemplate the entry definition service template
51    */
52   public ToscaServiceModel(FileContentHandler artifactFiles,
53                            Map<String, ServiceTemplate> serviceTemplates,
54                            String entryDefinitionServiceTemplate) {
55     this.artifactFiles = artifactFiles;
56     this.serviceTemplates = serviceTemplates;
57     this.entryDefinitionServiceTemplate = entryDefinitionServiceTemplate;
58   }
59
60   /**
61    * Gets artifact files.
62    *
63    * @return the artifact files
64    */
65   public FileContentHandler getArtifactFiles() {
66     return artifactFiles;
67   }
68
69   public void setArtifactFiles(FileContentHandler artifactFiles) {
70     this.artifactFiles = artifactFiles;
71   }
72
73   /**
74    * Gets service templates.
75    *
76    * @return the service templates
77    */
78   public Map<String, ServiceTemplate> getServiceTemplates() {
79     return Collections.unmodifiableMap(serviceTemplates);
80   }
81
82   public Optional<ServiceTemplate> getServiceTemplate(String serviceTemplateName) {
83     return MapUtils.isEmpty(this.serviceTemplates) ? Optional.empty()
84         : Optional.of(this.serviceTemplates.get(serviceTemplateName));
85   }
86
87   public void addServiceTemplate(String serviceTemplateName,
88                                  ServiceTemplate serviceTemplate) {
89     if(MapUtils.isEmpty(serviceTemplates)){
90       serviceTemplates = new HashMap<>();
91     }
92
93     serviceTemplates.put(serviceTemplateName, serviceTemplate);
94   }
95
96   /**
97    * Sets service templates.
98    *
99    * @param serviceTemplates the service templates
100    */
101   public void setServiceTemplates(Map<String, ServiceTemplate> serviceTemplates) {
102     this.serviceTemplates = serviceTemplates;
103   }
104
105   /**
106    * Gets entry definition service template.
107    *
108    * @return the entry definition service template
109    */
110   public String getEntryDefinitionServiceTemplate() {
111     return entryDefinitionServiceTemplate;
112   }
113
114   /**
115    * Sets entry definition service template.
116    *
117    * @param entryDefinitionServiceTemplate the entry definition service template
118    */
119   public void setEntryDefinitionServiceTemplate(String entryDefinitionServiceTemplate) {
120     this.entryDefinitionServiceTemplate = entryDefinitionServiceTemplate;
121   }
122
123   /**
124    * Gets cloned service model.
125    *
126    * @param toscaServiceModel the tosca service model
127    * @return the cloned service model
128    */
129   public static ToscaServiceModel getClonedServiceModel(ToscaServiceModel toscaServiceModel) {
130     return ToscaServiceModel.class.cast(DataModelUtil.getClonedObject(toscaServiceModel));
131   }
132 }