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 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen;
  22
  23 import com.amdocs.zusammen.adaptor.inbound.api.types.item.Element;
  24 import com.amdocs.zusammen.adaptor.inbound.api.types.item.ElementInfo;
  25 import com.amdocs.zusammen.adaptor.inbound.api.types.item.ZusammenElement;
  26 import com.amdocs.zusammen.datatypes.Id;
  27 import com.amdocs.zusammen.datatypes.SessionContext;
  28 import com.amdocs.zusammen.datatypes.item.Action;
  29 import com.amdocs.zusammen.datatypes.item.ElementContext;
  30 import com.amdocs.zusammen.datatypes.item.Info;
  31 import org.openecomp.core.zusammen.api.ZusammenAdaptor;
  32 import org.openecomp.sdc.datatypes.model.ElementType;
  33 import org.openecomp.sdc.vendorsoftwareproduct.dao.ComponentDao;
  34 import org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen.convertor.ElementToComponentConvertor;
  35 import org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen.convertor.ElementToComponentQuestionnnaireConvertor;
  36 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentEntity;
  37 import org.openecomp.sdc.versioning.dao.types.Version;
  38 import org.openecomp.types.ElementPropertyName;
  39
  40 import java.io.ByteArrayInputStream;
  41 import java.util.ArrayList;
  42 import java.util.Collection;
  43 import java.util.Collections;
  44 import java.util.Optional;
  45 import java.util.stream.Collectors;
  46
  47 import static org.openecomp.core.zusammen.api.ZusammenUtil.*;
  48
  49 public class ComponentDaoZusammenImpl implements ComponentDao {
  50
  51   private ZusammenAdaptor zusammenAdaptor;
  52
  53   public ComponentDaoZusammenImpl(ZusammenAdaptor zusammenAdaptor) {
  54     this.zusammenAdaptor = zusammenAdaptor;
  55   }
  56
  57   @Override
  58   public void registerVersioning(String versionableEntityType) {
  59     // registerVersioning is not implemented for ComponentDaoZusammenImpl
  60   }
  61
  62   @Override
  63   public Collection<ComponentEntity> list(ComponentEntity component) {
  64     SessionContext context = createSessionContext();
  65
  66     return listComponents(zusammenAdaptor, context, component.getVspId(), component.getVersion());
  67   }
  68
  69   static Collection<ComponentEntity> listComponents(ZusammenAdaptor zusammenAdaptor,
  70                                                     SessionContext context,
  71                                                     String vspId, Version version) {
  72     ElementContext elementContext = new ElementContext(vspId, version.getId());
  73
  74     Optional<ElementInfo> vspModel = zusammenAdaptor
  75         .getElementInfoByName(context, elementContext, null, ElementType.VspModel.name());
  76     if (!vspModel.isPresent()) {
  77       return new ArrayList<>();
  78     }
  79
  80     ElementToComponentConvertor convertor = new ElementToComponentConvertor();
  81     return zusammenAdaptor.listElementsByName(context, elementContext, vspModel.get().getId(),
  82         ElementType.Components.name()).stream()
  83         .map(elementInfo -> {
  84           ComponentEntity entity = convertor.convert(elementInfo);
  85           entity.setVspId(vspId);
  86           entity.setVersion(version);
  87           return entity;
  88         })
  89         .collect(Collectors.toList());
  90   }
  91
  92
  93   @Override
  94   public void create(ComponentEntity component) {
  95     ZusammenElement componentElement = componentToZusammen(component, Action.CREATE);
  96
  97     ZusammenElement componentsElement =
  98         buildStructuralElement(ElementType.Components, Action.IGNORE);
  99     componentsElement.getSubElements().add(componentElement);
 100
 101     ZusammenElement vspModel = buildStructuralElement(ElementType.VspModel, Action.IGNORE);
 102     vspModel.addSubElement(componentsElement);
 103
 104     SessionContext context = createSessionContext();
 105     Element savedVspModel = zusammenAdaptor.saveElement(context,
 106         new ElementContext(component.getVspId(), component.getVersion().getId()),
 107         vspModel, "Create component");
 108     component.setId(savedVspModel.getSubElements().iterator().next()
 109         .getSubElements().iterator().next().getElementId().getValue());
 110   }
 111
 112   @Override
 113   public void update(ComponentEntity component) {
 114     ZusammenElement componentElement = componentToZusammen(component, Action.UPDATE);
 115
 116     SessionContext context = createSessionContext();
 117     zusammenAdaptor.saveElement(context,
 118         new ElementContext(component.getVspId(), component.getVersion().getId()),
 119         componentElement, String.format("Update component with id %s", component.getId()));
 120   }
 121
 122   @Override
 123   public ComponentEntity get(ComponentEntity component) {
 124     SessionContext context = createSessionContext();
 125
 126     Optional<Element> element =
 127         zusammenAdaptor.getElement(context,
 128             new ElementContext(component.getVspId(), component.getVersion().getId()),
 129             component.getId());
 130
 131     if (element.isPresent()) {
 132       ComponentEntity entity = new ElementToComponentConvertor().convert(element.get());
 133       entity.setVspId(component.getVspId());
 134       entity.setVersion(component.getVersion());
 135       return entity;
 136     }
 137     return null;
 138   }
 139
 140   @Override
 141   public void delete(ComponentEntity component) {
 142     ZusammenElement componentElement = buildElement(new Id(component.getId()), Action.DELETE);
 143
 144     SessionContext context = createSessionContext();
 145     zusammenAdaptor.saveElement(context,
 146         new ElementContext(component.getVspId(), component.getVersion().getId()),
 147         componentElement, String.format("Delete component with id %s", component.getId()));
 148   }
 149
 150   @Override
 151   public ComponentEntity getQuestionnaireData(String vspId, Version version, String componentId) {
 152     SessionContext context = createSessionContext();
 153
 154     return getQuestionnaire(context, new ElementContext(vspId, version.getId()),
 155         new ComponentEntity(vspId, version, componentId));
 156   }
 157
 158   private ComponentEntity getQuestionnaire(SessionContext context, ElementContext elementContext,
 159                                            ComponentEntity component) {
 160     Optional<Element> questionnaireElement = zusammenAdaptor
 161         .getElementByName(context, elementContext, new Id(component.getId()),
 162             ElementType.ComponentQuestionnaire.name());
 163     return questionnaireElement.map(new ElementToComponentQuestionnnaireConvertor()::convert)
 164         .map(entity -> {
 165           entity.setVspId(component.getVspId());
 166           entity.setVersion(component.getVersion());
 167           return entity;
 168         })
 169         .orElse(null);
 170   }
 171
 172   @Override
 173   public void updateQuestionnaireData(String vspId, Version version, String componentId,
 174                                       String questionnaireData) {
 175     ZusammenElement questionnaireElement =
 176         componentQuestionnaireToZusammen(questionnaireData, Action.UPDATE);
 177
 178     ZusammenElement componentElement = buildElement(new Id(componentId), Action.IGNORE);
 179     componentElement.setSubElements(Collections.singletonList(questionnaireElement));
 180
 181     SessionContext context = createSessionContext();
 182     zusammenAdaptor.saveElement(context, new ElementContext(vspId, version.getId()),
 183         componentElement, "Update component questionnaire");
 184   }
 185
 186   @Override
 187   public Collection<ComponentEntity> listQuestionnaires(String vspId, Version version) {
 188     return listCompositionAndQuestionnaire(vspId, version);
 189   }
 190
 191   @Override
 192   public Collection<ComponentEntity> listCompositionAndQuestionnaire(String vspId,
 193                                                                      Version version) {
 194     SessionContext context = createSessionContext();
 195
 196     Collection<ComponentEntity> components =
 197         listComponents(zusammenAdaptor, context, vspId, version);
 198
 199     ElementContext elementContext = new ElementContext(vspId, version.getId());
 200     components.forEach(component -> component.setQuestionnaireData(
 201         getQuestionnaire(context, elementContext, component).getQuestionnaireData()));
 202     return components;
 203   }
 204
 205   @Override
 206   public void deleteAll(String vspId, Version version) {
 207     SessionContext context = createSessionContext();
 208     ElementContext elementContext = new ElementContext(vspId, version.getId());
 209
 210     Collection<ElementInfo> vspModelSubs = zusammenAdaptor
 211         .listElementsByName(context, elementContext, null, ElementType.VspModel.name());
 212
 213     Optional<ElementInfo> componentsElement = vspModelSubs.stream()
 214         .filter(elementInfo -> elementInfo.getInfo() != null
 215             && ElementType.Components.name().equals(elementInfo.getInfo().getName()))
 216         .findFirst();
 217     if (!componentsElement.isPresent()) {
 218       return;
 219     }
 220
 221     ZusammenElement components = buildElement(componentsElement.get().getId(), Action.IGNORE);
 222     components.setSubElements(componentsElement.get().getSubElements().stream()
 223         .map(component -> buildElement(component.getId(), Action.DELETE))
 224         .collect(Collectors.toList()));
 225
 226     zusammenAdaptor.saveElement(context, elementContext, components, "Delete all components");
 227   }
 228
 229   private ZusammenElement componentToZusammen(ComponentEntity component, Action action) {
 230     ZusammenElement componentElement = buildComponentElement(component, action);
 231
 232     if (action == Action.CREATE) {
 233       ZusammenElement mibs = buildStructuralElement(ElementType.Mibs, Action.CREATE);
 234       mibs.addSubElement(buildStructuralElement(ElementType.SNMP_TRAP, Action.CREATE));
 235       mibs.addSubElement(buildStructuralElement(ElementType.VES_EVENTS, Action.CREATE));
 236       mibs.addSubElement(buildStructuralElement(ElementType.SNMP_POLL, Action.CREATE));
 237
 238       componentElement.addSubElement(mibs);
 239       componentElement.addSubElement(
 240           componentQuestionnaireToZusammen(component.getQuestionnaireData(), Action.CREATE));
 241       componentElement.addSubElement(buildStructuralElement(ElementType.Nics, Action.CREATE));
 242       componentElement.addSubElement(buildStructuralElement(ElementType.Processes, Action.CREATE));
 243       componentElement.addSubElement(buildStructuralElement(ElementType.Computes, Action.CREATE));
 244       componentElement.addSubElement(buildStructuralElement(ElementType.Images, Action.CREATE));
 245     }
 246     return componentElement;
 247   }
 248
 249   private ZusammenElement componentQuestionnaireToZusammen(String questionnaireData,
 250                                                            Action action) {
 251     ZusammenElement questionnaireElement =
 252         buildStructuralElement(ElementType.ComponentQuestionnaire, action);
 253     questionnaireElement.setData(new ByteArrayInputStream(questionnaireData.getBytes()));
 254     return questionnaireElement;
 255   }
 256
 257   private ZusammenElement buildComponentElement(ComponentEntity component, Action action) {
 258     ZusammenElement componentElement =
 259         buildElement(component.getId() == null ? null : new Id(component.getId()), action);
 260     Info info = new Info();
 261     info.addProperty(ElementPropertyName.elementType.name(), ElementType.Component);
 262     info.addProperty(ElementPropertyName.compositionData.name(), component.getCompositionData());
 263     componentElement.setInfo(info);
 264     componentElement.setData(new ByteArrayInputStream(component.getCompositionData().getBytes()));
 265     return componentElement;
 266   }
 267
 268
 269 }