Sync Integ to Master
[sdc.git] / common-be / src / main / java / org / openecomp / sdc / be / datatypes / elements / CompositionDataDefinition.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.be.datatypes.elements;
22
23 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
24 import org.openecomp.sdc.be.datatypes.tosca.ToscaDataDefinition;
25
26 import java.util.HashMap;
27 import java.util.Map;
28
29 public class CompositionDataDefinition extends ToscaDataDefinition {
30         
31         public Map<String, ComponentInstanceDataDefinition> getComponentInstances() {
32                 return (Map<String, ComponentInstanceDataDefinition>) getToscaPresentationValue(JsonPresentationFields.COMPONENT_INSTANCES);
33         }
34         public void setComponentInstances(Map<String, ComponentInstanceDataDefinition> componentInstances) {
35                 setToscaPresentationValue(JsonPresentationFields.COMPONENT_INSTANCES, componentInstances);
36         }
37         public Map<String, RelationshipInstDataDefinition> getRelations() {
38                 return (Map<String, RelationshipInstDataDefinition>) getToscaPresentationValue(JsonPresentationFields.RELATIONS);
39         }
40         public void setRelations(Map<String, RelationshipInstDataDefinition> relations) {
41                 setToscaPresentationValue(JsonPresentationFields.RELATIONS, relations);
42         }
43         public void addInstance( String key, ComponentInstanceDataDefinition instance ){
44                 Map<String, ComponentInstanceDataDefinition> instances = getComponentInstances();
45                 if (instances == null ){
46                         instances = new HashMap<>();
47                         setToscaPresentationValue(JsonPresentationFields.COMPONENT_INSTANCES, instances );
48                 }
49                 instances.put(key, instance);
50         }
51         
52         public void addRelation( String key, RelationshipInstDataDefinition relation ){
53                 Map<String, RelationshipInstDataDefinition> relations = getRelations();
54                 if (relations == null ){
55                         relations = new HashMap<>();
56                         setToscaPresentationValue(JsonPresentationFields.RELATIONS, relations );
57                 }
58                 relations.put(key, relation);
59         }
60 }