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.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;
 
  19 import java.io.ByteArrayInputStream;
 
  20 import java.util.Collection;
 
  21 import java.util.Optional;
 
  22 import java.util.stream.Collectors;
 
  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;
 
  28 public class DeploymentFlavorDaoZusammenImpl implements DeploymentFlavorDao {
 
  30   private ZusammenAdaptor zusammenAdaptor;
 
  32   public DeploymentFlavorDaoZusammenImpl(ZusammenAdaptor zusammenAdaptor) {
 
  33     this.zusammenAdaptor = zusammenAdaptor;
 
  37   public void registerVersioning(String versionableEntityType) {
 
  41   public Collection<DeploymentFlavorEntity> list(DeploymentFlavorEntity deploymentFlavor) {
 
  42     SessionContext context = createSessionContext();
 
  43     ElementContext elementContext =
 
  44         new ElementContext(deploymentFlavor.getVspId(), deploymentFlavor.getVersion().getId());
 
  46     return listDeploymentFlavor(zusammenAdaptor, context, elementContext,
 
  47         deploymentFlavor.getVspId(),
 
  48         deploymentFlavor.getVersion());
 
  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(
 
  63           entity.setVspId(vspId);
 
  64           entity.setVersion(version);
 
  67         .collect(Collectors.toList());
 
  72   public void create(DeploymentFlavorEntity deploymentFlavor) {
 
  73     ZusammenElement deploymentFlavorElement = deploymentFlavorToZusammen(deploymentFlavor,
 
  75     ZusammenElement deploymentFlavorElements =
 
  76         buildStructuralElement(ElementType.DeploymentFlavors, Action.IGNORE);
 
  77     deploymentFlavorElements.getSubElements().add(deploymentFlavorElement);
 
  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()
 
  88   public void update(DeploymentFlavorEntity deploymentFlavor) {
 
  89     ZusammenElement deploymentFlavorElement = deploymentFlavorToZusammen(deploymentFlavor,
 
  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()));
 
 100   public DeploymentFlavorEntity get(DeploymentFlavorEntity deploymentFlavor) {
 
 101     SessionContext context = createSessionContext();
 
 102     ElementContext elementContext =
 
 103         new ElementContext(deploymentFlavor.getVspId(), deploymentFlavor.getVersion().getId());
 
 105     Optional<Element> element =
 
 106         zusammenAdaptor.getElement(context, elementContext, deploymentFlavor.getId());
 
 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()
 
 113       entity.setVspId(deploymentFlavor.getVspId());
 
 114       entity.setVersion(deploymentFlavor.getVersion());
 
 121   public void delete(DeploymentFlavorEntity deploymentFlavor) {
 
 122     ZusammenElement componentElement =
 
 123         buildElement(new Id(deploymentFlavor.getId()), Action.DELETE);
 
 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()));
 
 133   public void deleteAll(String vspId, Version version) {
 
 134     SessionContext context = createSessionContext();
 
 135     ElementContext elementContext = new ElementContext(vspId, version.getId());
 
 137     Optional<Element> optionalElement = zusammenAdaptor.getElementByName(context,
 
 138         elementContext, null, ElementType.DeploymentFlavors.name());
 
 140     if (optionalElement.isPresent()) {
 
 141       Element deploymentFlavorsElement = optionalElement.get();
 
 142       Collection<Element> deploymentFlavors = deploymentFlavorsElement.getSubElements();
 
 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());
 
 154   /*private ZusammenElement deplymentFlavorQuestionnaireToZusammen(String questionnaireData,
 
 156     ZusammenElement questionnaireElement =
 
 157         VspbuildStructuralElement(ElementType.Questionnaire, action);
 
 158     questionnaireElement.setData(new ByteArrayInputStream(questionnaireData.getBytes()));
 
 159     return questionnaireElement;
 
 162   private ZusammenElement deploymentFlavorToZusammen(DeploymentFlavorEntity deploymentFlavor,
 
 164     ZusammenElement deploymentFlavorElement =
 
 165         buildElement(deploymentFlavor.getId() == null ? null : new Id(deploymentFlavor.getId()),
 
 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()
 
 174     return deploymentFlavorElement;