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/NicDaoZusammenImpl.java

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