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