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

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