gerrit.onap Code Review - sdc.git/blob

   1 package org.openecomp.sdc.healing.healers;
   2
   3 import org.apache.commons.collections.MapUtils;
   4 import org.openecomp.core.model.dao.ServiceModelDao;
   5 import org.openecomp.core.model.dao.ServiceModelDaoFactory;
   6 import org.openecomp.core.model.types.ServiceElement;
   7 import org.openecomp.sdc.common.togglz.ToggleableFeature;
   8 import org.openecomp.sdc.common.utils.SdcCommon;
   9 import org.openecomp.sdc.healing.interfaces.Healer;
  10 import org.openecomp.sdc.logging.context.impl.MdcDataDebugMessage;
  11 import org.openecomp.sdc.tosca.datatypes.ToscaNodeType;
  12 import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
  13 import org.openecomp.sdc.tosca.datatypes.model.NodeTemplate;
  14 import org.openecomp.sdc.tosca.datatypes.model.ServiceTemplate;
  15 import org.openecomp.sdc.tosca.services.DataModelUtil;
  16 import org.openecomp.sdc.tosca.services.ToscaAnalyzerService;
  17 import org.openecomp.sdc.tosca.services.impl.ToscaAnalyzerServiceImpl;
  18 import org.openecomp.sdc.translator.services.heattotosca.globaltypes.GlobalTypesServiceTemplates;
  19 import org.openecomp.sdc.versioning.dao.types.Version;
  20
  21 import java.util.Arrays;
  22 import java.util.List;
  23 import java.util.Map;
  24 import java.util.Objects;
  25 import java.util.Optional;
  26
  27 public class ForwarderCapabilityHealer implements Healer {
  28
  29   private MdcDataDebugMessage mdcDataDebugMessage = new MdcDataDebugMessage();
  30
  31   private final ServiceModelDao<ToscaServiceModel, ServiceElement> serviceModelDao;
  32   private static ToscaAnalyzerService toscaAnalyzerService = new ToscaAnalyzerServiceImpl();
  33   private static final String FORWARDER_CAPABILITY_ID = "Forwarder";
  34   private static final String UNDERSCORE = "_";
  35
  36   public ForwarderCapabilityHealer() {
  37     this.serviceModelDao =
  38         ServiceModelDaoFactory.getInstance().createInterface();
  39   }
  40
  41   public ForwarderCapabilityHealer(ServiceModelDao<ToscaServiceModel, ServiceElement>
  42                                        serviceModelDao,
  43                                    ToscaAnalyzerService toscaAnalyzerService){
  44     this.serviceModelDao = serviceModelDao;
  45     this.toscaAnalyzerService = toscaAnalyzerService;
  46   }
  47
  48   @Override
  49   public Object heal(Map<String, Object> healingParams) throws Exception {
  50     String vspId = (String) healingParams.get(SdcCommon.VSP_ID);
  51     Version version = (Version) healingParams.get(SdcCommon.VERSION);
  52
  53     if(!ToggleableFeature.FORWARDER_CAPABILITY.isActive()) {
  54       return Optional.empty();
  55     }
  56
  57     ToscaServiceModel serviceModel =
  58         serviceModelDao.getServiceModel(vspId, version);
  59
  60     if (Objects.isNull(serviceModel)
  61         || MapUtils.isEmpty(serviceModel.getServiceTemplates())) {
  62       return Optional.empty();
  63     }
  64
  65     addForwarderCapabilityToServiceModel(serviceModel);
  66     serviceModelDao.deleteAll(vspId, version);
  67     serviceModelDao.storeServiceModel(vspId, version, serviceModel);
  68
  69     return Optional.of(serviceModel);
  70   }
  71
  72   private void addForwarderCapabilityToServiceModel(ToscaServiceModel serviceModel) {
  73     serviceModel.getServiceTemplates().entrySet().stream().filter(serviceTemplateEntry -> Objects
  74         .nonNull(serviceTemplateEntry.getValue()))
  75         .forEach(serviceTemplateEntry -> handleServiceTemplate(serviceTemplateEntry.getValue(),
  76             serviceModel));
  77
  78     handleGlobalTypes(serviceModel);
  79   }
  80
  81   private void handleGlobalTypes(ToscaServiceModel serviceModel) {
  82     Map<String, ServiceTemplate> globalTypesServiceTemplates =
  83         GlobalTypesServiceTemplates.getGlobalTypesServiceTemplates();
  84
  85     if (MapUtils.isEmpty(globalTypesServiceTemplates)) {
  86       return;
  87     }
  88
  89     globalTypesServiceTemplates.entrySet()
  90         .stream()
  91         .filter(globalTypesServiceTemplateEntry -> Objects.nonNull
  92             (globalTypesServiceTemplateEntry.getValue()))
  93         .forEach(globalTypesServiceTemplateEntry -> serviceModel.addServiceTemplate
  94             (globalTypesServiceTemplateEntry.getKey(), globalTypesServiceTemplateEntry.getValue()));
  95   }
  96
  97   private void handleServiceTemplate(ServiceTemplate serviceTemplate,
  98                                      ToscaServiceModel toscaServiceModel) {
  99     if (Objects.isNull(serviceTemplate.getTopology_template())
 100         || MapUtils.isEmpty(serviceTemplate.getTopology_template().getNode_templates())) {
 101       return;
 102     }
 103
 104     Map<String, NodeTemplate> nodeTemplates =
 105         serviceTemplate.getTopology_template().getNode_templates();
 106
 107     nodeTemplates.entrySet().stream()
 108         .filter(nodeTemplateEntry ->
 109             toscaAnalyzerService.isTypeOf(nodeTemplateEntry.getValue(),
 110                 ToscaNodeType.NATIVE_NETWORK_PORT, serviceTemplate, toscaServiceModel))
 111         .forEach(nodeTemplateEntry ->
 112             addForwarderToSubstitutionMappings(nodeTemplateEntry.getKey(), serviceTemplate)
 113         );
 114   }
 115
 116   private void addForwarderToSubstitutionMappings(String portNodeTemplateId,
 117                                                   ServiceTemplate serviceTemplate) {
 118     if (Objects.isNull(serviceTemplate.getTopology_template())
 119         || Objects.isNull(serviceTemplate.getTopology_template().getSubstitution_mappings())) {
 120       return;
 121     }
 122
 123     List<String> substitutionMappingCapabilityList =
 124         Arrays.asList(portNodeTemplateId, FORWARDER_CAPABILITY_ID);
 125
 126     DataModelUtil.addSubstitutionMappingCapability(
 127         serviceTemplate,
 128         FORWARDER_CAPABILITY_ID + UNDERSCORE + portNodeTemplateId,
 129         substitutionMappingCapabilityList);
 130
 131   }
 132 }