53994efddb1bd0bec2984ed32d5e17bd0ddcd958
[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.zusammen.api.ZusammenAdaptor;
11 import org.openecomp.sdc.datatypes.model.ElementType;
12 import org.openecomp.sdc.vendorsoftwareproduct.dao.NicDao;
13 import org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen.convertor.ElementToNicConvertor;
14 import org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen.convertor.ElementToNicQuestionnaireConvertor;
15 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentEntity;
16 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.NicEntity;
17 import org.openecomp.sdc.versioning.dao.types.Version;
18 import org.openecomp.types.ElementPropertyName;
19
20 import java.io.ByteArrayInputStream;
21 import java.util.Collection;
22 import java.util.Collections;
23 import java.util.Optional;
24 import java.util.stream.Collectors;
25
26 import static org.openecomp.core.zusammen.api.ZusammenUtil.buildElement;
27 import static org.openecomp.core.zusammen.api.ZusammenUtil.buildStructuralElement;
28 import static org.openecomp.core.zusammen.api.ZusammenUtil.createSessionContext;
29
30 public class NicDaoZusammenImpl implements NicDao {
31
32   private ZusammenAdaptor zusammenAdaptor;
33
34   public NicDaoZusammenImpl(ZusammenAdaptor zusammenAdaptor) {
35     this.zusammenAdaptor = zusammenAdaptor;
36   }
37
38   @Override
39   public void registerVersioning(String versionableEntityType) {
40     // registerVersioning is not implemented for NicDaoZusammenImpl
41   }
42
43   @Override
44   public Collection<NicEntity> list(NicEntity nic) {
45     SessionContext context = createSessionContext();
46     return listNics(context, new ElementContext(nic.getVspId(), nic.getVersion().getId()), nic);
47   }
48
49   private Collection<NicEntity> listNics(SessionContext context, ElementContext elementContext,
50                                          NicEntity nic) {
51     return zusammenAdaptor.listElementsByName(context, elementContext, new Id(nic.getComponentId()),
52         ElementType.Nics.name())
53         .stream()
54         .map(new ElementToNicConvertor()::convert)
55         .map(nicEntity -> {
56           nicEntity.setComponentId(nic.getComponentId());
57           nicEntity.setVspId(nic.getVspId());
58           nicEntity.setVersion(nic.getVersion());
59           return nicEntity;
60         })
61         .collect(Collectors.toList());
62   }
63
64
65   @Override
66   public void create(NicEntity nic) {
67     ZusammenElement nicElement = nicToZusammen(nic, Action.CREATE);
68
69     ZusammenElement nicsElement = buildStructuralElement(ElementType.Nics, Action.IGNORE);
70     nicsElement.setSubElements(Collections.singletonList(nicElement));
71
72     ZusammenElement componentElement = buildElement(new Id(nic.getComponentId()), Action.IGNORE);
73     componentElement.setSubElements(Collections.singletonList(nicsElement));
74
75     SessionContext context = createSessionContext();
76     ElementContext elementContext = new ElementContext(nic.getVspId(), nic.getVersion().getId());
77
78     Element savedElement =
79         zusammenAdaptor.saveElement(context, elementContext, componentElement, "Create nic");
80     nic.setId(savedElement.getSubElements().iterator().next()
81         .getSubElements().iterator().next().getElementId().getValue());
82   }
83
84   @Override
85   public void update(NicEntity nic) {
86     ZusammenElement nicElement = nicToZusammen(nic, Action.UPDATE);
87
88     SessionContext context = createSessionContext();
89     zusammenAdaptor
90         .saveElement(context, new ElementContext(nic.getVspId(), nic.getVersion().getId()),
91             nicElement, String.format("Update nic with id %s", nic.getId()));
92   }
93
94   @Override
95   public NicEntity get(NicEntity nic) {
96     SessionContext context = createSessionContext();
97     ElementToNicConvertor convertor = new ElementToNicConvertor();
98     Optional<Element> element = zusammenAdaptor
99         .getElement(context, new ElementContext(nic.getVspId(), nic.getVersion().getId()),
100             nic.getId());
101
102     if (element.isPresent()) {
103       NicEntity entity = convertor.convert(element.get());
104       entity.setVspId(nic.getVspId());
105       entity.setVersion(nic.getVersion());
106       entity.setComponentId(nic.getComponentId());
107
108       return entity;
109     } else {
110       return null;
111     }
112   }
113
114   @Override
115   public void delete(NicEntity nic) {
116     ZusammenElement nicElement = buildElement(new Id(nic.getId()), Action.DELETE);
117
118     SessionContext context = createSessionContext();
119     zusammenAdaptor
120         .saveElement(context, new ElementContext(nic.getVspId(), nic.getVersion().getId()),
121             nicElement, String.format("Delete nic with id %s", nic.getId()));
122   }
123
124   @Override
125   public NicEntity getQuestionnaireData(String vspId, Version version, String componentId,
126                                         String nicId) {
127     SessionContext context = createSessionContext();
128
129     return getQuestionnaire(context, new ElementContext(vspId, version.getId()),
130         new NicEntity(vspId, version, componentId, nicId));
131   }
132
133   private NicEntity getQuestionnaire(SessionContext context, ElementContext elementContext,
134                                      NicEntity nic) {
135     Optional<Element> questionnaireElement = zusammenAdaptor
136         .getElementByName(context, elementContext, new Id(nic.getId()),
137             ElementType.NicQuestionnaire.name());
138     return questionnaireElement.map(new ElementToNicQuestionnaireConvertor()::convert)
139         .map(entity -> {
140           entity.setVspId(nic.getVspId());
141           entity.setVersion(nic.getVersion());
142           entity.setComponentId(nic.getComponentId());
143           return entity;
144         })
145         .orElse(null);
146   }
147
148   @Override
149   public void updateQuestionnaireData(String vspId, Version version, String componentId,
150                                       String nicId, String questionnaireData) {
151     ZusammenElement questionnaireElement =
152         nicQuestionnaireToZusammen(questionnaireData, Action.UPDATE);
153
154     ZusammenElement nicElement = buildElement(new Id(nicId), Action.IGNORE);
155     nicElement.setSubElements(Collections.singletonList(questionnaireElement));
156
157     SessionContext context = createSessionContext();
158     zusammenAdaptor.saveElement(context, new ElementContext(vspId, version.getId()), nicElement,
159         "Update nic questionnaire");
160   }
161
162   @Override
163   public Collection<NicEntity> listByVsp(String vspId, Version version) {
164     SessionContext context = createSessionContext();
165
166     Collection<ComponentEntity> components = ComponentDaoZusammenImpl
167         .listComponents(zusammenAdaptor, context, vspId, version);
168
169     ElementContext elementContext = new ElementContext(vspId, version.getId());
170     return components.stream()
171         .map(component ->
172             listNics(context, elementContext,
173                 new NicEntity(vspId, version, component.getId(), null)).stream()
174                 .map(nic -> {
175                   nic.setQuestionnaireData(getQuestionnaire(context, elementContext, nic).getQuestionnaireData());
176                   return nic;
177                 })
178                 .collect(Collectors.toList()))
179         .flatMap(Collection::stream)
180         .collect(Collectors.toList());
181   }
182
183   @Override
184   public void deleteByComponentId(String vspId, Version version, String componentId) {
185     SessionContext context = createSessionContext();
186     ElementContext elementContext = new ElementContext(vspId, version.getId());
187
188     Optional<Element> optionalElement = zusammenAdaptor.getElementByName(context,
189         elementContext, new Id(componentId), ElementType.Nics.name());
190
191     if (optionalElement.isPresent()) {
192       Element nicsElement = optionalElement.get();
193       Collection<Element> nics = nicsElement.getSubElements();
194
195       nics.forEach(nic -> {
196         ZusammenElement nicZusammenElement = buildElement(nic.getElementId(), Action.DELETE);
197         zusammenAdaptor.saveElement(context, elementContext, nicZusammenElement,
198             "Delete nic with id " + nic.getElementId());
199       });
200     }
201   }
202
203   @Override
204   public void deleteByVspId(String vspId, Version version) {
205     // deleteByVspId not implemented for NicDaoZusammenImpl
206   }
207
208   private ZusammenElement nicToZusammen(NicEntity nic, Action action) {
209     ZusammenElement nicElement = buildNicElement(nic, action);
210     if (action == Action.CREATE) {
211       nicElement.setSubElements(Collections.singletonList(
212           nicQuestionnaireToZusammen(nic.getQuestionnaireData(), Action.CREATE)));
213     }
214     return nicElement;
215   }
216
217   private ZusammenElement nicQuestionnaireToZusammen(String questionnaireData,
218                                                      Action action) {
219     ZusammenElement questionnaireElement =
220         buildStructuralElement(ElementType.NicQuestionnaire, action);
221     questionnaireElement.setData(new ByteArrayInputStream(questionnaireData.getBytes()));
222     return questionnaireElement;
223   }
224
225   private ZusammenElement buildNicElement(NicEntity nic, Action action) {
226     ZusammenElement nicElement =
227         buildElement(nic.getId() == null ? null : new Id(nic.getId()), action);
228     Info info = new Info();
229     info.addProperty(ElementPropertyName.elementType.name(), ElementType.Nic);
230     info.addProperty(ElementPropertyName.compositionData.name(), nic.getCompositionData());
231     nicElement.setInfo(info);
232     nicElement.setData(new ByteArrayInputStream(nic.getCompositionData().getBytes()));
233     return nicElement;
234   }
235 }