27f4945dfa8770dab8d474990c402949fb624666
[sdc.git] /
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     // registerVersioning not implemented for DeploymentFlavorDaoZusammenImpl
39   }
40
41   @Override
42   public Collection<DeploymentFlavorEntity> list(DeploymentFlavorEntity deploymentFlavor) {
43     SessionContext context = createSessionContext();
44     ElementContext elementContext =
45         new ElementContext(deploymentFlavor.getVspId(), deploymentFlavor.getVersion().getId());
46
47     return listDeploymentFlavor(zusammenAdaptor, context, elementContext,
48         deploymentFlavor.getVspId(),
49         deploymentFlavor.getVersion());
50   }
51
52   private static Collection<DeploymentFlavorEntity> listDeploymentFlavor(
53       ZusammenAdaptor zusammenAdaptor,
54       SessionContext context,
55       ElementContext elementContext,
56       String vspId, Version version) {
57     ElementToDeploymentFlavorConvertor convertor = new ElementToDeploymentFlavorConvertor();
58     return zusammenAdaptor
59         .listElementsByName(context, elementContext, null,
60             ElementType.DeploymentFlavors.name())
61         .stream().map(elementInfo -> {
62           DeploymentFlavorEntity entity = convertor.convert(
63               elementInfo);
64           entity.setVspId(vspId);
65           entity.setVersion(version);
66           return entity;
67         })
68         .collect(Collectors.toList());
69   }
70
71
72   @Override
73   public void create(DeploymentFlavorEntity deploymentFlavor) {
74     ZusammenElement deploymentFlavorElement = deploymentFlavorToZusammen(deploymentFlavor,
75         Action.CREATE);
76     ZusammenElement deploymentFlavorElements =
77         buildStructuralElement(ElementType.DeploymentFlavors, Action.IGNORE);
78     deploymentFlavorElements.getSubElements().add(deploymentFlavorElement);
79
80     SessionContext context = createSessionContext();
81     Element savedElement = zusammenAdaptor.saveElement(context,
82         new ElementContext(deploymentFlavor.getVspId(), deploymentFlavor.getVersion().getId()),
83         deploymentFlavorElements, "Create deloymentFlavor");
84     deploymentFlavor.setId(savedElement.getSubElements().iterator().next().getElementId()
85         .getValue());
86   }
87
88   @Override
89   public void update(DeploymentFlavorEntity deploymentFlavor) {
90     ZusammenElement deploymentFlavorElement = deploymentFlavorToZusammen(deploymentFlavor,
91         Action.UPDATE);
92
93     SessionContext context = createSessionContext();
94     zusammenAdaptor.saveElement(context,
95         new ElementContext(deploymentFlavor.getVspId(), deploymentFlavor.getVersion().getId()),
96         deploymentFlavorElement, String.format("Update deloymentFlavor with id %s",
97             deploymentFlavor.getId()));
98   }
99
100   @Override
101   public DeploymentFlavorEntity get(DeploymentFlavorEntity deploymentFlavor) {
102     SessionContext context = createSessionContext();
103     ElementContext elementContext =
104         new ElementContext(deploymentFlavor.getVspId(), deploymentFlavor.getVersion().getId());
105
106     Optional<Element> element =
107         zusammenAdaptor.getElement(context, elementContext, deploymentFlavor.getId());
108
109     if (element.isPresent()) {
110       ElementToDeploymentFlavorConvertor convertor = new ElementToDeploymentFlavorConvertor();
111       DeploymentFlavorEntity entity = convertor.convert(element.get());
112       deploymentFlavor.setCompositionData(new String(FileUtils.toByteArray(element.get()
113           .getData())));
114       entity.setVspId(deploymentFlavor.getVspId());
115       entity.setVersion(deploymentFlavor.getVersion());
116       return entity;
117     }
118     return null;
119   }
120
121   @Override
122   public void delete(DeploymentFlavorEntity deploymentFlavor) {
123     ZusammenElement componentElement =
124         buildElement(new Id(deploymentFlavor.getId()), Action.DELETE);
125
126     SessionContext context = createSessionContext();
127     zusammenAdaptor.saveElement(context,
128         new ElementContext(deploymentFlavor.getVspId(), deploymentFlavor.getVersion().getId()),
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     SessionContext context = createSessionContext();
136     ElementContext elementContext = new ElementContext(vspId, version.getId());
137
138     Optional<Element> optionalElement = zusammenAdaptor.getElementByName(context,
139         elementContext, null, ElementType.DeploymentFlavors.name());
140
141     if (optionalElement.isPresent()) {
142       Element deploymentFlavorsElement = optionalElement.get();
143       Collection<Element> deploymentFlavors = deploymentFlavorsElement.getSubElements();
144
145       deploymentFlavors.forEach(deplymentFlavor -> {
146         ZusammenElement deplymentFlavorZusammenElement =
147             buildElement(deplymentFlavor.getElementId(), Action.DELETE);
148         zusammenAdaptor.saveElement(context,
149             elementContext, deplymentFlavorZusammenElement, " Delete Deplyment Flavor with id "
150                 + deplymentFlavor.getElementId());
151       });
152     }
153   }
154
155   private ZusammenElement deploymentFlavorToZusammen(DeploymentFlavorEntity deploymentFlavor,
156                                                      Action action) {
157     ZusammenElement deploymentFlavorElement =
158         buildElement(deploymentFlavor.getId() == null ? null : new Id(deploymentFlavor.getId()),
159             action);
160     Info info = new Info();
161     info.addProperty(ElementPropertyName.elementType.name(), ElementType.DeploymentFlavor);
162     info.addProperty(ElementPropertyName.compositionData.name(), deploymentFlavor
163         .getCompositionData());
164     deploymentFlavorElement.setInfo(info);
165     deploymentFlavorElement.setData(new ByteArrayInputStream(deploymentFlavor.getCompositionData()
166         .getBytes()));
167     return deploymentFlavorElement;
168   }
169
170 }