[SDC] Onboarding 1710 rebase.
[sdc.git] / openecomp-be / lib / openecomp-sdc-vendor-software-product-lib / openecomp-sdc-vendor-software-product-core / src / main / java / org / openecomp / sdc / vendorsoftwareproduct / dao / impl / zusammen / DeploymentFlavorDaoZusammenImpl.java
1 package org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen;
2
3 import com.amdocs.zusammen.adaptor.inbound.api.types.item.Element;
4 import com.amdocs.zusammen.adaptor.inbound.api.types.item.ElementInfo;
5 import com.amdocs.zusammen.adaptor.inbound.api.types.item.ZusammenElement;
6 import com.amdocs.zusammen.datatypes.Id;
7 import com.amdocs.zusammen.datatypes.SessionContext;
8 import com.amdocs.zusammen.datatypes.item.Action;
9 import com.amdocs.zusammen.datatypes.item.ElementContext;
10 import com.amdocs.zusammen.datatypes.item.Info;
11 import org.openecomp.core.utilities.file.FileUtils;
12 import org.openecomp.core.zusammen.api.ZusammenAdaptor;
13 import org.openecomp.core.zusammen.api.ZusammenUtil;
14 import org.openecomp.sdc.vendorsoftwareproduct.dao.DeploymentFlavorDao;
15 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.DeploymentFlavorEntity;
16 import org.openecomp.sdc.versioning.dao.types.Version;
17
18 import java.io.ByteArrayInputStream;
19 import java.util.Collection;
20 import java.util.Optional;
21 import java.util.stream.Collectors;
22
23 public class DeploymentFlavorDaoZusammenImpl implements DeploymentFlavorDao {
24
25   private ZusammenAdaptor zusammenAdaptor;
26
27   public DeploymentFlavorDaoZusammenImpl(ZusammenAdaptor zusammenAdaptor) {
28     this.zusammenAdaptor = zusammenAdaptor;
29   }
30
31   @Override
32   public void registerVersioning(String versionableEntityType) {
33   }
34
35   @Override
36   public Collection<DeploymentFlavorEntity> list(DeploymentFlavorEntity deploymentFlavor) {
37     SessionContext context = ZusammenUtil.createSessionContext();
38     Id itemId = new Id(deploymentFlavor.getVspId());
39     ElementContext elementContext = new ElementContext(itemId,
40         VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
41         VspZusammenUtil.getVersionTag(deploymentFlavor.getVersion()));
42
43     return listDeploymentFlavor(zusammenAdaptor, context, elementContext, deploymentFlavor.getVspId(),
44         deploymentFlavor.getVersion());
45   }
46
47   static Collection<DeploymentFlavorEntity> listDeploymentFlavor(ZusammenAdaptor zusammenAdaptor,
48                                                     SessionContext context,
49                                                     ElementContext elementContext,
50                                                     String vspId, Version version) {
51     return zusammenAdaptor
52         .listElementsByName(context, elementContext, null, StructureElement.DeploymentFlavors.name())
53         .stream().map(elementInfo -> mapElementInfoToComponent(vspId, version, elementInfo))
54         .collect(Collectors.toList());
55   }
56
57   private static DeploymentFlavorEntity mapElementInfoToComponent(String vspId, Version version,
58                                                            ElementInfo elementInfo) {
59     DeploymentFlavorEntity deploymentFlavorEntity =
60         new DeploymentFlavorEntity(vspId, version, elementInfo.getId().getValue());
61     deploymentFlavorEntity.setCompositionData(
62         elementInfo.getInfo().getProperty(ElementPropertyName.compositionData.name()));
63     return deploymentFlavorEntity;
64   }
65
66   @Override
67   public void create(DeploymentFlavorEntity deploymentFlavor) {
68     ZusammenElement deploymentFlavorElement = deploymentFlavorToZusammen(deploymentFlavor,
69         Action.CREATE);
70     ZusammenElement deploymentFlavorElements =
71         VspZusammenUtil.buildStructuralElement(StructureElement.DeploymentFlavors, null);
72     deploymentFlavorElements.getSubElements().add(deploymentFlavorElement);
73
74     SessionContext context = ZusammenUtil.createSessionContext();
75     Id itemId = new Id(deploymentFlavor.getVspId());
76     Optional<Element> savedElement = zusammenAdaptor.saveElement(context,
77         new ElementContext(itemId,
78             VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor)),
79         deploymentFlavorElements, "Create deloymentFlavor");
80     savedElement.ifPresent(element ->
81         deploymentFlavor.setId(element.getSubElements().iterator().next().getElementId()
82             .getValue()));
83   }
84
85   @Override
86   public void update(DeploymentFlavorEntity deploymentFlavor) {
87     ZusammenElement deploymentFlavorElement = deploymentFlavorToZusammen(deploymentFlavor,
88         Action.UPDATE);
89
90     SessionContext context = ZusammenUtil.createSessionContext();
91     Id itemId = new Id(deploymentFlavor.getVspId());
92     zusammenAdaptor.saveElement(context,
93         new ElementContext(itemId,
94             VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor)),
95         deploymentFlavorElement, String.format("Update deloymentFlavor with id %s",
96             deploymentFlavor.getId()));
97   }
98
99   @Override
100   public DeploymentFlavorEntity get(DeploymentFlavorEntity deploymentFlavor) {
101     SessionContext context = ZusammenUtil.createSessionContext();
102     Id itemId = new Id(deploymentFlavor.getVspId());
103     ElementContext elementContext = new ElementContext(itemId,
104         VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
105         VspZusammenUtil.getVersionTag(deploymentFlavor.getVersion()));
106
107     Optional<Element> element =
108         zusammenAdaptor.getElement(context, elementContext, deploymentFlavor.getId());
109
110     if (element.isPresent()) {
111       deploymentFlavor.setCompositionData(new String(FileUtils.toByteArray(element.get()
112           .getData())));
113       return deploymentFlavor;
114     }
115     return null;
116   }
117
118   @Override
119   public void delete(DeploymentFlavorEntity deploymentFlavor) {
120     ZusammenElement componentElement = new ZusammenElement();
121     componentElement.setElementId(new Id(deploymentFlavor.getId()));
122     componentElement.setAction(Action.DELETE);
123
124     SessionContext context = ZusammenUtil.createSessionContext();
125     Id itemId = new Id(deploymentFlavor.getVspId());
126     zusammenAdaptor.saveElement(context,
127         new ElementContext(itemId,
128             VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor)),
129         componentElement, String.format("Delete deloymentFlavor with id %s",
130             deploymentFlavor.getId()));
131   }
132
133   @Override
134   public void deleteAll(String vspId, Version version) {
135     ZusammenElement deploymentFlavorsElement =
136         VspZusammenUtil.buildStructuralElement(StructureElement.DeploymentFlavors, Action.DELETE);
137
138     SessionContext context = ZusammenUtil.createSessionContext();
139     Id itemId = new Id(vspId);
140     zusammenAdaptor.saveElement(context,
141         new ElementContext(itemId,
142             VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor)),
143         deploymentFlavorsElement, "Delete all deploymentFlavors");
144   }
145
146   private ZusammenElement deploymentFlavorToZusammen(DeploymentFlavorEntity deploymentFlavor,
147                                                      Action action) {
148     ZusammenElement deploymentFlavorElement = buildDeploymentFlavorElement
149         (deploymentFlavor, action);
150
151     return deploymentFlavorElement;
152   }
153
154   /*private ZusammenElement deplymentFlavorQuestionnaireToZusammen(String questionnaireData,
155                                                            Action action) {
156     ZusammenElement questionnaireElement =
157         VspZusammenUtil.buildStructuralElement(StructureElement.Questionnaire, action);
158     questionnaireElement.setData(new ByteArrayInputStream(questionnaireData.getBytes()));
159     return questionnaireElement;
160   }*/
161
162   private ZusammenElement buildDeploymentFlavorElement(DeploymentFlavorEntity deploymentFlavor,
163                                                        Action action) {
164     ZusammenElement deploymentFlavorElement = new ZusammenElement();
165     deploymentFlavorElement.setAction(action);
166     if (deploymentFlavor.getId() != null) {
167       deploymentFlavorElement.setElementId(new Id(deploymentFlavor.getId()));
168     }
169     Info info = new Info();
170     info.addProperty(ElementPropertyName.type.name(), ElementType.DeploymentFlavor);
171     info.addProperty(ElementPropertyName.compositionData.name(), deploymentFlavor
172         .getCompositionData());
173     deploymentFlavorElement.setInfo(info);
174     deploymentFlavorElement.setData(new ByteArrayInputStream(deploymentFlavor.getCompositionData()
175         .getBytes()));
176     return deploymentFlavorElement;
177   }
178
179 }