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