openecomp-be/lib/openecomp-healing-lib/openecomp-sdc-healing-impl/src/main/java/org/openecomp/sdc/healing/healers/SubEntitiesQuestionnaireHealer.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2017 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.healing.healers;
  22
  23 import org.openecomp.core.utilities.json.JsonSchemaDataGenerator;
  24 import org.openecomp.sdc.common.utils.SdcCommon;
  25 import org.openecomp.sdc.healing.interfaces.Healer;
  26 import org.openecomp.sdc.logging.context.impl.MdcDataDebugMessage;
  27 import org.openecomp.sdc.vendorsoftwareproduct.dao.ComponentDao;
  28 import org.openecomp.sdc.vendorsoftwareproduct.dao.ComponentDaoFactory;
  29 import org.openecomp.sdc.vendorsoftwareproduct.dao.NetworkDao;
  30 import org.openecomp.sdc.vendorsoftwareproduct.dao.NetworkDaoFactory;
  31 import org.openecomp.sdc.vendorsoftwareproduct.dao.NicDao;
  32 import org.openecomp.sdc.vendorsoftwareproduct.dao.NicDaoFactory;
  33 import org.openecomp.sdc.vendorsoftwareproduct.dao.VendorSoftwareProductDao;
  34 import org.openecomp.sdc.vendorsoftwareproduct.dao.VendorSoftwareProductDaoFactory;
  35 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentEntity;
  36 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.CompositionEntity;
  37 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.NetworkEntity;
  38 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.NicEntity;
  39 import org.openecomp.sdc.vendorsoftwareproduct.services.schemagenerator.SchemaGenerator;
  40 import org.openecomp.sdc.vendorsoftwareproduct.types.composition.CompositionEntityType;
  41 import org.openecomp.sdc.vendorsoftwareproduct.types.schemagenerator.SchemaTemplateContext;
  42 import org.openecomp.sdc.versioning.dao.types.Version;
  43
  44 import java.util.Collection;
  45 import java.util.Map;
  46 import java.util.Objects;
  47
  48 public class SubEntitiesQuestionnaireHealer implements Healer {
  49   private static Version version00 = new Version(0, 0);
  50   private MdcDataDebugMessage mdcDataDebugMessage = new MdcDataDebugMessage();
  51
  52   private static final VendorSoftwareProductDao vendorSoftwareProductDao =
  53       VendorSoftwareProductDaoFactory.getInstance().createInterface();
  54   private static ComponentDao componentDao = ComponentDaoFactory.getInstance().createInterface();
  55   private static NicDao nicDao = NicDaoFactory.getInstance().createInterface();
  56   private static NetworkDao networkDao = NetworkDaoFactory.getInstance().createInterface();
  57
  58   @Override
  59   public Object heal(Map<String, Object> healingParams) throws Exception {
  60
  61
  62     mdcDataDebugMessage.debugEntryMessage(null);
  63
  64     String vspId = (String) healingParams.get(SdcCommon.VSP_ID);
  65     Version version = version00.equals(healingParams.get(SdcCommon.VERSION)) ? new Version
  66         (0, 1)
  67         : (Version) healingParams.get(SdcCommon.VERSION);
  68
  69     Collection<ComponentEntity> componentEntities =
  70
  71         componentDao.listCompositionAndQuestionnaire(vspId, version);
  72     networkDao.list(new NetworkEntity(vspId, version,null));
  73
  74     Collection<NicEntity> nicEntities = vendorSoftwareProductDao.listNicsByVsp(vspId, version);
  75
  76     healCompositionEntityQuestionnaire(componentEntities, version, CompositionEntityType.component);
  77     healCompositionEntityQuestionnaire(nicEntities, version, CompositionEntityType.nic);
  78
  79     mdcDataDebugMessage.debugExitMessage(null);
  80     return new Object();
  81   }
  82
  83
  84   private void healCompositionEntityQuestionnaire(Collection
  85                                                       compositionEntities,
  86                                                   Version newVersion, CompositionEntityType type) {
  87
  88
  89     mdcDataDebugMessage.debugEntryMessage(null);
  90
  91     for (Object entity : compositionEntities) {
  92       CompositionEntity compositionEntity = (CompositionEntity) entity;
  93       if (Objects.isNull(compositionEntity.getQuestionnaireData())) {
  94         compositionEntity.setVersion(newVersion);
  95         updateNullQuestionnaire(compositionEntity, type);
  96       }
  97     }
  98
  99     mdcDataDebugMessage.debugExitMessage(null);
 100   }
 101
 102   private void updateNullQuestionnaire(CompositionEntity entity,
 103                                        CompositionEntityType type) {
 104
 105
 106     mdcDataDebugMessage.debugEntryMessage(null);
 107
 108     entity.setQuestionnaireData(
 109         new JsonSchemaDataGenerator(SchemaGenerator
 110             .generate(SchemaTemplateContext.questionnaire, type,
 111                 null)).generateData());
 112
 113     switch (type) {
 114       case component:
 115         ComponentEntity component = (ComponentEntity) entity;
 116         componentDao.updateQuestionnaireData(component.getVspId(), component
 117             .getVersion(), component.getId(), component.getQuestionnaireData());
 118         break;
 119
 120       case nic:
 121         NicEntity nic = (NicEntity) entity;
 122         nicDao.updateQuestionnaireData(nic.getVspId(), nic.getVersion(), nic.getComponentId(),
 123             nic.getId(), nic.getQuestionnaireData());
 124         break;
 125     }
 126     mdcDataDebugMessage.debugExitMessage(null);
 127   }
 128
 129 }