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 / ComputeDaoZusammenImpl.java
index aa4bab6..0f92c95 100644 (file)
@@ -2,7 +2,6 @@ package org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen;
 
 
 import com.amdocs.zusammen.adaptor.inbound.api.types.item.Element;
-import com.amdocs.zusammen.adaptor.inbound.api.types.item.ElementInfo;
 import com.amdocs.zusammen.adaptor.inbound.api.types.item.ZusammenElement;
 import com.amdocs.zusammen.datatypes.Id;
 import com.amdocs.zusammen.datatypes.SessionContext;
@@ -11,11 +10,13 @@ import com.amdocs.zusammen.datatypes.item.ElementContext;
 import com.amdocs.zusammen.datatypes.item.Info;
 import org.openecomp.core.utilities.file.FileUtils;
 import org.openecomp.core.zusammen.api.ZusammenAdaptor;
-import org.openecomp.core.zusammen.api.ZusammenUtil;
+import org.openecomp.sdc.datatypes.model.ElementType;
 import org.openecomp.sdc.vendorsoftwareproduct.dao.ComputeDao;
+import org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen.convertor.ElementToComputeConvertor;
 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentEntity;
 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComputeEntity;
 import org.openecomp.sdc.versioning.dao.types.Version;
+import org.openecomp.types.ElementPropertyName;
 
 import java.io.ByteArrayInputStream;
 import java.util.Collection;
@@ -23,6 +24,10 @@ import java.util.Collections;
 import java.util.Optional;
 import java.util.stream.Collectors;
 
+import static org.openecomp.core.zusammen.api.ZusammenUtil.buildElement;
+import static org.openecomp.core.zusammen.api.ZusammenUtil.buildStructuralElement;
+import static org.openecomp.core.zusammen.api.ZusammenUtil.createSessionContext;
+
 public class ComputeDaoZusammenImpl implements ComputeDao {
 
   private ZusammenAdaptor zusammenAdaptor;
@@ -37,83 +42,78 @@ public class ComputeDaoZusammenImpl implements ComputeDao {
 
   @Override
   public Collection<ComputeEntity> list(ComputeEntity compute) {
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(compute.getVspId());
-    ElementContext elementContext = new ElementContext(itemId,
-        VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
-        VspZusammenUtil.getVersionTag(compute.getVersion()));
+    SessionContext context = createSessionContext();
+    ElementContext elementContext =
+        new ElementContext(compute.getVspId(), compute.getVersion().getId());
 
     return listComputes(context, elementContext, compute);
   }
 
   private Collection<ComputeEntity> listComputes(SessionContext context,
-                                            ElementContext elementContext, ComputeEntity compute) {
+                                                 ElementContext elementContext,
+                                                 ComputeEntity compute) {
+    ElementToComputeConvertor convertor = new ElementToComputeConvertor();
     return zusammenAdaptor
         .listElementsByName(context, elementContext, new Id(compute.getComponentId()),
-            StructureElement.Computes.name())
-        .stream().map(elementInfo -> mapElementInfoToCompute(
-            compute.getVspId(), compute.getVersion(), compute.getComponentId(), elementInfo))
+            ElementType.Computes.name())
+        .stream().map(elementInfo -> convertor.convert(elementInfo))
+        .map(computeEntity -> {
+          computeEntity.setComponentId(compute.getComponentId());
+          computeEntity.setVspId(compute.getVspId());
+          computeEntity.setVersion(compute.getVersion());
+          return computeEntity;
+        })
         .collect(Collectors.toList());
   }
 
-  private static ComputeEntity mapElementInfoToCompute(String vspId, Version version,
-                                                     String componentId, ElementInfo elementInfo) {
-    ComputeEntity componentEntity =
-        new ComputeEntity(vspId, version, componentId, elementInfo.getId().getValue());
-    componentEntity.setCompositionData(
-        elementInfo.getInfo().getProperty(ElementPropertyName.compositionData.name()));
-    return componentEntity;
-  }
-
   @Override
   public void create(ComputeEntity compute) {
     ZusammenElement computeElement = computeToZusammen(compute, Action.CREATE);
 
-    ZusammenElement computesElement =
-        VspZusammenUtil.buildStructuralElement(StructureElement.Computes, null);
+    ZusammenElement computesElement = buildStructuralElement(ElementType.Computes, Action.IGNORE);
     computesElement.setSubElements(Collections.singletonList(computeElement));
 
-    ZusammenElement componentElement = new ZusammenElement();
-    componentElement.setElementId(new Id(compute.getComponentId()));
-    componentElement.setAction(Action.IGNORE);
+    ZusammenElement componentElement =
+        buildElement(new Id(compute.getComponentId()), Action.IGNORE);
     componentElement.setSubElements(Collections.singletonList(computesElement));
 
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(compute.getVspId());
-    ElementContext elementContext = new ElementContext(itemId,
-        VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor));
+    SessionContext context = createSessionContext();
+    ElementContext elementContext =
+        new ElementContext(compute.getVspId(), compute.getVersion().getId());
 
-    Optional<Element> savedElement =
+    Element savedElement =
         zusammenAdaptor.saveElement(context, elementContext, componentElement, "Create compute");
-    savedElement.ifPresent(element ->
-        compute.setId(element.getSubElements().iterator().next()
-            .getSubElements().iterator().next().getElementId().getValue()));
+    compute.setId(savedElement.getSubElements().iterator().next()
+        .getSubElements().iterator().next().getElementId().getValue());
   }
 
   @Override
   public void update(ComputeEntity compute) {
     ZusammenElement computeElement = computeToZusammen(compute, Action.UPDATE);
 
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(compute.getVspId());
-    ElementContext elementContext = new ElementContext(itemId,
-        VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor));
+    SessionContext context = createSessionContext();
+    ElementContext elementContext =
+        new ElementContext(compute.getVspId(), compute.getVersion().getId());
     zusammenAdaptor.saveElement(context, elementContext, computeElement,
         String.format("Update compute with id %s", compute.getId()));
   }
 
   @Override
   public ComputeEntity get(ComputeEntity compute) {
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(compute.getVspId());
-    ElementContext elementContext = new ElementContext(itemId,
-        VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
-        VspZusammenUtil.getVersionTag(compute.getVersion()));
-    Optional<Element> element = zusammenAdaptor.getElement(context, elementContext, compute.getId());
+    SessionContext context = createSessionContext();
+    ElementContext elementContext =
+        new ElementContext(compute.getVspId(), compute.getVersion().getId());
+    Optional<Element> element =
+        zusammenAdaptor.getElement(context, elementContext, compute.getId());
 
     if (element.isPresent()) {
-      compute.setCompositionData(new String(FileUtils.toByteArray(element.get().getData())));
-      return compute;
+
+      ElementToComputeConvertor convertor = new ElementToComputeConvertor();
+      ComputeEntity entity = convertor.convert(element.get());
+      entity.setVspId(compute.getVspId());
+      entity.setVersion(compute.getVersion());
+      entity.setComponentId(compute.getComponentId());
+      return entity;
     } else {
       return null;
     }
@@ -121,36 +121,30 @@ public class ComputeDaoZusammenImpl implements ComputeDao {
 
   @Override
   public void delete(ComputeEntity compute) {
-    ZusammenElement computeElement = new ZusammenElement();
-    computeElement.setElementId(new Id(compute.getId()));
-    computeElement.setAction(Action.DELETE);
-
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(compute.getVspId());
-    ElementContext elementContext = new ElementContext(itemId,
-        VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor));
+    ZusammenElement computeElement = buildElement(new Id(compute.getId()), Action.DELETE);
+
+    SessionContext context = createSessionContext();
+    ElementContext elementContext =
+        new ElementContext(compute.getVspId(), compute.getVersion().getId());
     zusammenAdaptor.saveElement(context, elementContext, computeElement,
         String.format("Delete compute with id %s", compute.getId()));
   }
 
   @Override
   public ComputeEntity getQuestionnaireData(String vspId, Version version, String componentId,
-                                        String computeId) {
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(vspId);
-    ElementContext elementContext = new ElementContext(itemId,
-        VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
-        VspZusammenUtil.getVersionTag(version));
+                                            String computeId) {
+    SessionContext context = createSessionContext();
+    ElementContext elementContext = new ElementContext(vspId, version.getId());
 
     return getQuestionnaire(context, elementContext,
         new ComputeEntity(vspId, version, componentId, computeId));
   }
 
   private ComputeEntity getQuestionnaire(SessionContext context, ElementContext elementContext,
-                                     ComputeEntity compute) {
+                                         ComputeEntity compute) {
     Optional<Element> questionnaireElement = zusammenAdaptor
         .getElementByName(context, elementContext, new Id(compute.getId()),
-            StructureElement.Questionnaire.name());
+            ElementType.ComputeQuestionnaire.name());
     return questionnaireElement.map(
         element -> element.getData() == null
             ? null
@@ -168,29 +162,22 @@ public class ComputeDaoZusammenImpl implements ComputeDao {
     ZusammenElement questionnaireElement =
         computeQuestionnaireToZusammen(questionnaireData, Action.UPDATE);
 
-    ZusammenElement computeElement = new ZusammenElement();
-    computeElement.setAction(Action.IGNORE);
-    computeElement.setElementId(new Id(computeId));
+    ZusammenElement computeElement = buildElement(new Id(computeId), Action.IGNORE);
     computeElement.setSubElements(Collections.singletonList(questionnaireElement));
 
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(vspId);
-    ElementContext elementContext = new ElementContext(itemId,
-        VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor));
+    SessionContext context = createSessionContext();
+    ElementContext elementContext = new ElementContext(vspId, version.getId());
     zusammenAdaptor.saveElement(context, elementContext, computeElement, "Update compute "
         + "questionnaire");
   }
 
   @Override
   public Collection<ComputeEntity> listByVsp(String vspId, Version version) {
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(vspId);
-    ElementContext elementContext = new ElementContext(itemId,
-        VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
-        VspZusammenUtil.getVersionTag(version));
+    SessionContext context = createSessionContext();
+    ElementContext elementContext = new ElementContext(vspId, version.getId());
 
     Collection<ComponentEntity> components = ComponentDaoZusammenImpl
-        .listComponents(zusammenAdaptor, context, elementContext, vspId, version);
+        .listComponents(zusammenAdaptor, context, vspId, version);
 
     return components.stream()
         .map(component ->
@@ -204,15 +191,23 @@ public class ComputeDaoZusammenImpl implements ComputeDao {
 
   @Override
   public void deleteAll(String vspId, Version version) {
-    ZusammenElement computesElement =
-        VspZusammenUtil.buildStructuralElement(StructureElement.Computes, Action.DELETE);
 
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(vspId);
-    zusammenAdaptor.saveElement(context,
-        new ElementContext(itemId,
-            VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor)),
-        computesElement, "Delete all computes");
+    SessionContext context = createSessionContext();
+    ElementContext elementContext = new ElementContext(vspId, version.getId());
+
+    Optional<Element> optionalElement = zusammenAdaptor.getElementByName(context,
+        elementContext, null, ElementType.Computes.name());
+
+    if (optionalElement.isPresent()) {
+      Element computesElement = optionalElement.get();
+      Collection<Element> computes = computesElement.getSubElements();
+
+      computes.forEach(compute -> {
+        ZusammenElement computeElement = buildElement(compute.getElementId(), Action.DELETE);
+        zusammenAdaptor.saveElement(context, elementContext, computeElement,
+            "Delete compute with id " + compute.getElementId());
+      });
+    }
   }
 
   private ZusammenElement computeToZusammen(ComputeEntity compute, Action action) {
@@ -225,27 +220,21 @@ public class ComputeDaoZusammenImpl implements ComputeDao {
   }
 
   private ZusammenElement computeQuestionnaireToZusammen(String questionnaireData,
-                                                     Action action) {
+                                                         Action action) {
     ZusammenElement questionnaireElement =
-        VspZusammenUtil.buildStructuralElement(StructureElement.Questionnaire, action);
+        buildStructuralElement(ElementType.ComputeQuestionnaire, action);
     questionnaireElement.setData(new ByteArrayInputStream(questionnaireData.getBytes()));
     return questionnaireElement;
   }
 
   private ZusammenElement buildComputeElement(ComputeEntity compute, Action action) {
-    ZusammenElement computeElement = new ZusammenElement();
-    computeElement.setAction(action);
-    if (compute.getId() != null) {
-      computeElement.setElementId(new Id(compute.getId()));
-    }
+    ZusammenElement computeElement =
+        buildElement(compute.getId() == null ? null : new Id(compute.getId()), action);
     Info info = new Info();
-    info.addProperty(ElementPropertyName.type.name(), ElementType.Compute);
+    info.addProperty(ElementPropertyName.elementType.name(), ElementType.Compute);
     info.addProperty(ElementPropertyName.compositionData.name(), compute.getCompositionData());
     computeElement.setInfo(info);
     computeElement.setData(new ByteArrayInputStream(compute.getCompositionData().getBytes()));
     return computeElement;
   }
-
-
-
 }