gerrit.onap Code Review - sdc.git/blob

   1 /*
   2  * Copyright © 2016-2017 European Support Limited
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package org.openecomp.sdc.translator.services.heattotosca.impl.resourcetranslation;
  18
  19 import org.openecomp.sdc.common.togglz.ToggleableFeature;
  20 import org.openecomp.sdc.heat.services.HeatConstants;
  21 import org.openecomp.sdc.tosca.datatypes.ToscaNodeType;
  22 import org.openecomp.sdc.tosca.datatypes.model.NodeTemplate;
  23 import org.openecomp.sdc.tosca.services.DataModelUtil;
  24 import org.openecomp.sdc.translator.datatypes.heattotosca.to.TranslateTo;
  25 import org.openecomp.sdc.translator.services.heattotosca.HeatToToscaUtil;
  26 import org.openecomp.sdc.translator.services.heattotosca.helper.ContrailV2VirtualMachineInterfaceHelper;
  27 import org.openecomp.sdc.translator.services.heattotosca.mapping.TranslatorHeatToToscaPropertyConverter;
  28
  29 import java.util.List;
  30 import java.util.Map;
  31
  32
  33 public class ResourceTranslationContrailV2VmInterfaceImpl extends ResourceTranslationBase {
  34
  35   @Override
  36   protected String generateTranslatedId(TranslateTo translateTo) {
  37     if (!(new ContrailV2VirtualMachineInterfaceHelper().isVlanSubInterfaceResource(translateTo
  38         .getResource())) || ToggleableFeature.VLAN_TAGGING.isActive()) {
  39       return super.generateTranslatedId(translateTo);
  40     }
  41
  42     return null;
  43   }
  44
  45   @Override
  46   protected void translate(TranslateTo translateTo) {
  47     if (new ContrailV2VirtualMachineInterfaceHelper().isVlanSubInterfaceResource(translateTo
  48         .getResource())) {
  49       translateVlanSubInterfaceResource(translateTo);
  50     } else {
  51       translateVirtualMachineInterfaceResource(translateTo);
  52     }
  53   }
  54
  55
  56   private void translateVirtualMachineInterfaceResource(TranslateTo translateTo) {
  57     NodeTemplate nodeTemplate = new NodeTemplate();
  58     nodeTemplate.setType(ToscaNodeType.CONTRAILV2_VIRTUAL_MACHINE_INTERFACE);
  59     nodeTemplate.setProperties(TranslatorHeatToToscaPropertyConverter
  60         .getToscaPropertiesSimpleConversion(translateTo.getServiceTemplate(),translateTo.
  61                 getResourceId(),translateTo.getResource().getProperties(),
  62             nodeTemplate.getProperties(), translateTo.getHeatFileName(),
  63             translateTo.getHeatOrchestrationTemplate(), translateTo.getResource().getType(),
  64             nodeTemplate, translateTo.getContext()));
  65     String toscaVmiRefsPropertyName =
  66         HeatToToscaUtil.getToscaPropertyName(translateTo, HeatConstants.VMI_REFS_PROPERTY_NAME);
  67     if (nodeTemplate.getProperties().containsKey(toscaVmiRefsPropertyName)) {
  68       nodeTemplate.getProperties().remove(toscaVmiRefsPropertyName);
  69     }
  70
  71     handleVmiMacAddressesInProperties(translateTo, nodeTemplate);
  72
  73     new ContrailV2VirtualMachineInterfaceHelper()
  74         .connectVmiToNetwork(this, translateTo, nodeTemplate);
  75     DataModelUtil.addNodeTemplate(translateTo.getServiceTemplate(), translateTo.getTranslatedId(),
  76         nodeTemplate);
  77   }
  78
  79   private void handleVmiMacAddressesInProperties(TranslateTo translateTo,
  80                                                  NodeTemplate nodeTemplate) {
  81     String toscaVmiMacAddressesName =
  82         HeatToToscaUtil.getToscaPropertyName(translateTo, HeatConstants.VMI_MAC_ADDRESSES);
  83     String toscaVmiMacAddressesMacAddressesName =
  84         HeatToToscaUtil.getToscaPropertyName(translateTo, HeatConstants.VMI_MAC_ADDRESSES_MAC_ADDRESSES);
  85
  86     if(nodeTemplate.getProperties().containsKey(toscaVmiMacAddressesName)){
  87       Object macAddressesValue = nodeTemplate.getProperties().get(toscaVmiMacAddressesName);
  88       if(macAddressesValue instanceof Map && ((Map<String, Object>)macAddressesValue).containsKey
  89           (toscaVmiMacAddressesMacAddressesName)){
  90         updateMacAddressesMacAddressesInProperties(nodeTemplate, toscaVmiMacAddressesName,
  91             toscaVmiMacAddressesMacAddressesName,
  92             (Map<String, Object>) macAddressesValue);
  93       }
  94     }
  95   }
  96
  97   private void updateMacAddressesMacAddressesInProperties(NodeTemplate nodeTemplate,
  98                                                           String toscaVmiMacAddressesName,
  99                                                           String toscaVmiMacAddressesMacAddressesName,
 100                                                           Map<String, Object> macAddressesValue) {
 101     Object macAddressesMacAddressesValue =
 102         macAddressesValue.get(toscaVmiMacAddressesMacAddressesName);
 103     if(macAddressesMacAddressesValue instanceof List){
 104       nodeTemplate.getProperties().put(toscaVmiMacAddressesName, macAddressesMacAddressesValue);
 105     }else{
 106       nodeTemplate.getProperties().remove(toscaVmiMacAddressesName);
 107     }
 108   }
 109
 110   private void translateVlanSubInterfaceResource(TranslateTo translateTo) {
 111     new ResourceTranslationContrailV2VlanSubInterfaceImpl().translate(translateTo);
 112   }
 113
 114 }