1 package org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen;
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.ComponentDao;
15 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentEntity;
16 import org.openecomp.sdc.versioning.dao.types.Version;
18 import java.io.ByteArrayInputStream;
19 import java.util.Arrays;
20 import java.util.Collection;
21 import java.util.Collections;
22 import java.util.Optional;
23 import java.util.stream.Collectors;
25 public class ComponentDaoZusammenImpl implements ComponentDao {
27 private ZusammenAdaptor zusammenAdaptor;
29 public ComponentDaoZusammenImpl(ZusammenAdaptor zusammenAdaptor) {
30 this.zusammenAdaptor = zusammenAdaptor;
34 public void registerVersioning(String versionableEntityType) {
38 public Collection<ComponentEntity> list(ComponentEntity component) {
39 SessionContext context = ZusammenUtil.createSessionContext();
40 Id itemId = new Id(component.getVspId());
41 ElementContext elementContext = new ElementContext(itemId,
42 VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
43 VspZusammenUtil.getVersionTag(component.getVersion()));
45 return listComponents(zusammenAdaptor, context, elementContext, component.getVspId(),
46 component.getVersion());
49 static Collection<ComponentEntity> listComponents(ZusammenAdaptor zusammenAdaptor,
50 SessionContext context,
51 ElementContext elementContext,
52 String vspId, Version version) {
53 return zusammenAdaptor
54 .listElementsByName(context, elementContext, null, StructureElement.Components.name())
55 .stream().map(elementInfo -> mapElementInfoToComponent(vspId, version, elementInfo))
56 .collect(Collectors.toList());
59 private static ComponentEntity mapElementInfoToComponent(String vspId, Version version,
60 ElementInfo elementInfo) {
61 ComponentEntity componentEntity =
62 new ComponentEntity(vspId, version, elementInfo.getId().getValue());
63 componentEntity.setCompositionData(
64 elementInfo.getInfo().getProperty(ElementPropertyName.compositionData.name()));
65 return componentEntity;
69 public void create(ComponentEntity component) {
70 ZusammenElement componentElement = componentToZusammen(component, Action.CREATE);
71 ZusammenElement componentsElement =
72 VspZusammenUtil.buildStructuralElement(StructureElement.Components, null);
73 componentsElement.getSubElements().add(componentElement);
75 SessionContext context = ZusammenUtil.createSessionContext();
76 Id itemId = new Id(component.getVspId());
77 Optional<Element> savedElement = zusammenAdaptor.saveElement(context,
78 new ElementContext(itemId,
79 VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor)),
80 componentsElement, "Create component");
81 savedElement.ifPresent(element ->
82 component.setId(element.getSubElements().iterator().next().getElementId().getValue()));
86 public void update(ComponentEntity component) {
87 ZusammenElement componentElement = componentToZusammen(component, Action.UPDATE);
89 SessionContext context = ZusammenUtil.createSessionContext();
90 Id itemId = new Id(component.getVspId());
91 zusammenAdaptor.saveElement(context,
92 new ElementContext(itemId,
93 VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor)),
94 componentElement, String.format("Update component with id %s", component.getId()));
98 public ComponentEntity get(ComponentEntity component) {
99 SessionContext context = ZusammenUtil.createSessionContext();
100 Id itemId = new Id(component.getVspId());
101 ElementContext elementContext = new ElementContext(itemId,
102 VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
103 VspZusammenUtil.getVersionTag(component.getVersion()));
105 Optional<Element> element =
106 zusammenAdaptor.getElement(context, elementContext, component.getId());
108 if (element.isPresent()) {
109 component.setCompositionData(new String(FileUtils.toByteArray(element.get().getData())));
116 public void delete(ComponentEntity component) {
117 ZusammenElement componentElement = new ZusammenElement();
118 componentElement.setElementId(new Id(component.getId()));
119 componentElement.setAction(Action.DELETE);
121 SessionContext context = ZusammenUtil.createSessionContext();
122 Id itemId = new Id(component.getVspId());
123 zusammenAdaptor.saveElement(context,
124 new ElementContext(itemId,
125 VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor)),
126 componentElement, String.format("Delete component with id %s", component.getId()));
130 public ComponentEntity getQuestionnaireData(String vspId, Version version, String componentId) {
131 SessionContext context = ZusammenUtil.createSessionContext();
132 Id itemId = new Id(vspId);
133 ElementContext elementContext = new ElementContext(itemId,
134 VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
135 VspZusammenUtil.getVersionTag(version));
137 return getQuestionnaire(context, elementContext,
138 new ComponentEntity(vspId, version, componentId));
141 private ComponentEntity getQuestionnaire(SessionContext context, ElementContext elementContext,
142 ComponentEntity component) {
143 Optional<Element> questionnaireElement = zusammenAdaptor
144 .getElementByName(context, elementContext, new Id(component.getId()),
145 StructureElement.Questionnaire.name());
146 return questionnaireElement.map(
147 element -> element.getData() == null
149 : new String(FileUtils.toByteArray(element.getData())))
150 .map(questionnaireData -> {
151 component.setQuestionnaireData(questionnaireData);
158 public void updateQuestionnaireData(String vspId, Version version, String componentId,
159 String questionnaireData) {
160 ZusammenElement questionnaireElement =
161 componentQuestionnaireToZusammen(questionnaireData, Action.UPDATE);
163 ZusammenElement componentElement = new ZusammenElement();
164 componentElement.setAction(Action.IGNORE);
165 componentElement.setElementId(new Id(componentId));
166 componentElement.setSubElements(Collections.singletonList(questionnaireElement));
168 SessionContext context = ZusammenUtil.createSessionContext();
169 Id itemId = new Id(vspId);
170 zusammenAdaptor.saveElement(context,
171 new ElementContext(itemId,
172 VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor)),
173 componentElement, "Update component questionnaire");
177 public Collection<ComponentEntity> listQuestionnaires(String vspId, Version version) {
178 return listCompositionAndQuestionnaire(vspId, version);
182 public Collection<ComponentEntity> listCompositionAndQuestionnaire(String vspId,
184 SessionContext context = ZusammenUtil.createSessionContext();
185 Id itemId = new Id(vspId);
186 ElementContext elementContext = new ElementContext(itemId,
187 VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
188 VspZusammenUtil.getVersionTag(version));
190 Collection<ComponentEntity> components =
191 listComponents(zusammenAdaptor, context, elementContext, vspId, version);
193 components.forEach(component -> getQuestionnaire(context, elementContext, component));
198 public void deleteAll(String vspId, Version version) {
199 ZusammenElement componentsElement =
200 VspZusammenUtil.buildStructuralElement(StructureElement.Components, Action.DELETE);
202 SessionContext context = ZusammenUtil.createSessionContext();
203 Id itemId = new Id(vspId);
204 zusammenAdaptor.saveElement(context,
205 new ElementContext(itemId,
206 VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor)),
207 componentsElement, "Delete all components");
210 private ZusammenElement componentToZusammen(ComponentEntity component, Action action) {
211 ZusammenElement componentElement = buildComponentElement(component, action);
212 if (action == Action.CREATE) {
214 .setSubElements(Arrays.asList(
215 componentQuestionnaireToZusammen(component.getQuestionnaireData(), Action.CREATE),
216 VspZusammenUtil.buildStructuralElement(StructureElement.Nics, Action.CREATE),
217 VspZusammenUtil.buildStructuralElement(StructureElement.Processes, Action.CREATE),
218 VspZusammenUtil.buildStructuralElement(StructureElement.Mibs, Action.CREATE)));
220 return componentElement;
223 private ZusammenElement componentQuestionnaireToZusammen(String questionnaireData,
225 ZusammenElement questionnaireElement =
226 VspZusammenUtil.buildStructuralElement(StructureElement.Questionnaire, action);
227 questionnaireElement.setData(new ByteArrayInputStream(questionnaireData.getBytes()));
228 return questionnaireElement;
231 private ZusammenElement buildComponentElement(ComponentEntity component, Action action) {
232 ZusammenElement componentElement = new ZusammenElement();
233 componentElement.setAction(action);
234 if (component.getId() != null) {
235 componentElement.setElementId(new Id(component.getId()));
237 Info info = new Info();
238 info.addProperty(ElementPropertyName.type.name(), ElementType.Component);
239 info.addProperty(ElementPropertyName.compositionData.name(), component.getCompositionData());
240 componentElement.setInfo(info);
241 componentElement.setData(new ByteArrayInputStream(component.getCompositionData().getBytes()));
242 return componentElement;