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