Incorporate the ECOMP SDC Artefact Generator code
[aai/babel.git] / src / main / java / org / onap / aai / babel / xml / generator / model / VfModule.java
1 /**
2  * ============LICENSE_START=======================================================
3  * org.onap.aai
4  * ================================================================================
5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
6  * Copyright © 2017-2018 European Software Marketing Ltd.
7  * ================================================================================
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  *       http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  * ============LICENSE_END=========================================================
20  */
21 package org.onap.aai.babel.xml.generator.model;
22
23 import java.util.List;
24 import org.onap.aai.babel.xml.generator.types.Cardinality;
25 import org.onap.aai.babel.xml.generator.types.Model;
26
27 @Model(widget = Widget.Type.VFMODULE, cardinality = Cardinality.UNBOUNDED, dataDeleteFlag = true)
28 public class VfModule extends Resource {
29
30     Widget vserver = null;
31     boolean addlintf = false;
32     boolean addvolume = false;
33
34     List<String> members;
35
36     public void setMembers(List<String> members) {
37         this.members = members;
38     }
39
40     /**
41      * Adds Widget.
42      *
43      * @param widget the widget
44      * @return the boolean
45      */
46     @Override
47     public boolean addWidget(Widget widget) {
48         if (widget.memberOf(members)) {
49             if (vserver == null && widget.getId().equals(new VServerWidget().getId())) {
50                 addVserverWidget(widget);
51             } else if (widget.getId().equals(new LIntfWidget().getId())) {
52                 return addLIntfWidget(widget);
53             } else if (widget.getId().equals(new VolumeWidget().getId())) {
54                 addVolumeWidget(widget);
55                 return true;
56             }
57             if (widget.getId().equals(new OamNetwork().getId())) {
58                 return false;
59             }
60             return widgets.add(widget);
61         }
62         return false;
63     }
64
65     private void addVolumeWidget(Widget widget) {
66         if (vserver != null) {
67             vserver.addWidget(widget);
68         } else {
69             addvolume = true;
70         }
71     }
72
73     /**
74      * @param widget
75      * @return
76      */
77     private boolean addLIntfWidget(Widget widget) {
78         if (vserver != null) {
79             vserver.addWidget(widget);
80             return true;
81         } else {
82             addlintf = true;
83             return false;
84         }
85     }
86
87     private void addVserverWidget(Widget widget) {
88         vserver = widget;
89         if (addlintf) {
90             vserver.addWidget(new LIntfWidget());
91         }
92         if (addvolume) {
93             vserver.addWidget(new VolumeWidget());
94         }
95     }
96
97     @Override
98     public int hashCode() {
99         return getModelNameVersionId().hashCode();
100     }
101
102     @Override
103     public boolean equals(Object obj) {
104         if (obj instanceof Resource) {
105             return getModelNameVersionId().equals(((Resource) obj).getModelNameVersionId());
106         }
107         return false;
108     }
109 }