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/ComponentDaoZusammenImpl.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.ElementInfo;
   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.zusammen.api.ZusammenAdaptor;
  12 import org.openecomp.sdc.datatypes.model.ElementType;
  13 import org.openecomp.sdc.vendorsoftwareproduct.dao.ComponentDao;
  14 import org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen.convertor.ElementToComponentConvertor;
  15 import org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen.convertor.ElementToComponentQuestionnnaireConvertor;
  16 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentEntity;
  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.ArrayList;
  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 ComponentDaoZusammenImpl implements ComponentDao {
  30
  31   private ZusammenAdaptor zusammenAdaptor;
  32
  33   public ComponentDaoZusammenImpl(ZusammenAdaptor zusammenAdaptor) {
  34     this.zusammenAdaptor = zusammenAdaptor;
  35   }
  36
  37   @Override
  38   public void registerVersioning(String versionableEntityType) {
  39     // registerVersioning is not implemented for ComponentDaoZusammenImpl
  40   }
  41
  42   @Override
  43   public Collection<ComponentEntity> list(ComponentEntity component) {
  44     SessionContext context = createSessionContext();
  45
  46     return listComponents(zusammenAdaptor, context, component.getVspId(), component.getVersion());
  47   }
  48
  49   static Collection<ComponentEntity> listComponents(ZusammenAdaptor zusammenAdaptor,
  50                                                     SessionContext context,
  51                                                     String vspId, Version version) {
  52     ElementContext elementContext = new ElementContext(vspId, version.getId());
  53
  54     Optional<ElementInfo> vspModel = zusammenAdaptor
  55         .getElementInfoByName(context, elementContext, null, ElementType.VspModel.name());
  56     if (!vspModel.isPresent()) {
  57       return new ArrayList<>();
  58     }
  59
  60     ElementToComponentConvertor convertor = new ElementToComponentConvertor();
  61     return zusammenAdaptor.listElementsByName(context, elementContext, vspModel.get().getId(),
  62         ElementType.Components.name()).stream()
  63         .map(elementInfo -> {
  64           ComponentEntity entity = convertor.convert(elementInfo);
  65           entity.setVspId(vspId);
  66           entity.setVersion(version);
  67           return entity;
  68         })
  69         .collect(Collectors.toList());
  70   }
  71
  72
  73   @Override
  74   public void create(ComponentEntity component) {
  75     ZusammenElement componentElement = componentToZusammen(component, Action.CREATE);
  76
  77     ZusammenElement componentsElement =
  78         buildStructuralElement(ElementType.Components, Action.IGNORE);
  79     componentsElement.getSubElements().add(componentElement);
  80
  81     ZusammenElement vspModel = buildStructuralElement(ElementType.VspModel, Action.IGNORE);
  82     vspModel.addSubElement(componentsElement);
  83
  84     SessionContext context = createSessionContext();
  85     Element savedVspModel = zusammenAdaptor.saveElement(context,
  86         new ElementContext(component.getVspId(), component.getVersion().getId()),
  87         vspModel, "Create component");
  88     component.setId(savedVspModel.getSubElements().iterator().next()
  89         .getSubElements().iterator().next().getElementId().getValue());
  90   }
  91
  92   @Override
  93   public void update(ComponentEntity component) {
  94     ZusammenElement componentElement = componentToZusammen(component, Action.UPDATE);
  95
  96     SessionContext context = createSessionContext();
  97     zusammenAdaptor.saveElement(context,
  98         new ElementContext(component.getVspId(), component.getVersion().getId()),
  99         componentElement, String.format("Update component with id %s", component.getId()));
 100   }
 101
 102   @Override
 103   public ComponentEntity get(ComponentEntity component) {
 104     SessionContext context = createSessionContext();
 105
 106     Optional<Element> element =
 107         zusammenAdaptor.getElement(context,
 108             new ElementContext(component.getVspId(), component.getVersion().getId()),
 109             component.getId());
 110
 111     if (element.isPresent()) {
 112       ComponentEntity entity = new ElementToComponentConvertor().convert(element.get());
 113       entity.setVspId(component.getVspId());
 114       entity.setVersion(component.getVersion());
 115       return entity;
 116     }
 117     return null;
 118   }
 119
 120   @Override
 121   public void delete(ComponentEntity component) {
 122     ZusammenElement componentElement = buildElement(new Id(component.getId()), Action.DELETE);
 123
 124     SessionContext context = createSessionContext();
 125     zusammenAdaptor.saveElement(context,
 126         new ElementContext(component.getVspId(), component.getVersion().getId()),
 127         componentElement, String.format("Delete component with id %s", component.getId()));
 128   }
 129
 130   @Override
 131   public ComponentEntity getQuestionnaireData(String vspId, Version version, String componentId) {
 132     SessionContext context = createSessionContext();
 133
 134     return getQuestionnaire(context, new ElementContext(vspId, version.getId()),
 135         new ComponentEntity(vspId, version, componentId));
 136   }
 137
 138   private ComponentEntity getQuestionnaire(SessionContext context, ElementContext elementContext,
 139                                            ComponentEntity component) {
 140     Optional<Element> questionnaireElement = zusammenAdaptor
 141         .getElementByName(context, elementContext, new Id(component.getId()),
 142             ElementType.ComponentQuestionnaire.name());
 143     return questionnaireElement.map(new ElementToComponentQuestionnnaireConvertor()::convert)
 144         .map(entity -> {
 145           entity.setVspId(component.getVspId());
 146           entity.setVersion(component.getVersion());
 147           return entity;
 148         })
 149         .orElse(null);
 150   }
 151
 152   @Override
 153   public void updateQuestionnaireData(String vspId, Version version, String componentId,
 154                                       String questionnaireData) {
 155     ZusammenElement questionnaireElement =
 156         componentQuestionnaireToZusammen(questionnaireData, Action.UPDATE);
 157
 158     ZusammenElement componentElement = buildElement(new Id(componentId), Action.IGNORE);
 159     componentElement.setSubElements(Collections.singletonList(questionnaireElement));
 160
 161     SessionContext context = createSessionContext();
 162     zusammenAdaptor.saveElement(context, new ElementContext(vspId, version.getId()),
 163         componentElement, "Update component questionnaire");
 164   }
 165
 166   @Override
 167   public Collection<ComponentEntity> listQuestionnaires(String vspId, Version version) {
 168     return listCompositionAndQuestionnaire(vspId, version);
 169   }
 170
 171   @Override
 172   public Collection<ComponentEntity> listCompositionAndQuestionnaire(String vspId,
 173                                                                      Version version) {
 174     SessionContext context = createSessionContext();
 175
 176     Collection<ComponentEntity> components =
 177         listComponents(zusammenAdaptor, context, vspId, version);
 178
 179     ElementContext elementContext = new ElementContext(vspId, version.getId());
 180     components.forEach(component -> component.setQuestionnaireData(
 181         getQuestionnaire(context, elementContext, component).getQuestionnaireData()));
 182     return components;
 183   }
 184
 185   @Override
 186   public void deleteAll(String vspId, Version version) {
 187     SessionContext context = createSessionContext();
 188     ElementContext elementContext = new ElementContext(vspId, version.getId());
 189
 190     Collection<ElementInfo> vspModelSubs = zusammenAdaptor
 191         .listElementsByName(context, elementContext, null, ElementType.VspModel.name());
 192
 193     Optional<ElementInfo> componentsElement = vspModelSubs.stream()
 194         .filter(elementInfo -> elementInfo.getInfo() != null
 195             && ElementType.Components.name().equals(elementInfo.getInfo().getName()))
 196         .findFirst();
 197     if (!componentsElement.isPresent()) {
 198       return;
 199     }
 200
 201     ZusammenElement components = buildElement(componentsElement.get().getId(), Action.IGNORE);
 202     components.setSubElements(componentsElement.get().getSubElements().stream()
 203         .map(component -> buildElement(component.getId(), Action.DELETE))
 204         .collect(Collectors.toList()));
 205
 206     zusammenAdaptor.saveElement(context, elementContext, components, "Delete all components");
 207   }
 208
 209   private ZusammenElement componentToZusammen(ComponentEntity component, Action action) {
 210     ZusammenElement componentElement = buildComponentElement(component, action);
 211
 212     if (action == Action.CREATE) {
 213       ZusammenElement mibs = buildStructuralElement(ElementType.Mibs, Action.CREATE);
 214       mibs.addSubElement(buildStructuralElement(ElementType.SNMP_TRAP, Action.CREATE));
 215       mibs.addSubElement(buildStructuralElement(ElementType.VES_EVENTS, Action.CREATE));
 216       mibs.addSubElement(buildStructuralElement(ElementType.SNMP_POLL, Action.CREATE));
 217
 218       componentElement.addSubElement(mibs);
 219       componentElement.addSubElement(
 220           componentQuestionnaireToZusammen(component.getQuestionnaireData(), Action.CREATE));
 221       componentElement.addSubElement(buildStructuralElement(ElementType.Nics, Action.CREATE));
 222       componentElement.addSubElement(buildStructuralElement(ElementType.Processes, Action.CREATE));
 223       componentElement.addSubElement(buildStructuralElement(ElementType.Computes, Action.CREATE));
 224       componentElement.addSubElement(buildStructuralElement(ElementType.Images, Action.CREATE));
 225     }
 226     return componentElement;
 227   }
 228
 229   private ZusammenElement componentQuestionnaireToZusammen(String questionnaireData,
 230                                                            Action action) {
 231     ZusammenElement questionnaireElement =
 232         buildStructuralElement(ElementType.ComponentQuestionnaire, action);
 233     questionnaireElement.setData(new ByteArrayInputStream(questionnaireData.getBytes()));
 234     return questionnaireElement;
 235   }
 236
 237   private ZusammenElement buildComponentElement(ComponentEntity component, Action action) {
 238     ZusammenElement componentElement =
 239         buildElement(component.getId() == null ? null : new Id(component.getId()), action);
 240     Info info = new Info();
 241     info.addProperty(ElementPropertyName.elementType.name(), ElementType.Component);
 242     info.addProperty(ElementPropertyName.compositionData.name(), component.getCompositionData());
 243     componentElement.setInfo(info);
 244     componentElement.setData(new ByteArrayInputStream(component.getCompositionData().getBytes()));
 245     return componentElement;
 246   }
 247
 248
 249 }