gerrit.onap Code Review - sdc.git/blob

   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.translator.services.heattotosca.impl.resourcetranslation;
  22
  23 import org.openecomp.sdc.heat.services.HeatConstants;
  24 import org.openecomp.sdc.logging.api.Logger;
  25 import org.openecomp.sdc.logging.api.LoggerFactory;
  26 import org.openecomp.sdc.tosca.datatypes.ToscaNodeType;
  27 import org.openecomp.sdc.tosca.datatypes.model.NodeTemplate;
  28 import org.openecomp.sdc.tosca.services.DataModelUtil;
  29 import org.openecomp.sdc.translator.datatypes.heattotosca.to.TranslateTo;
  30 import org.openecomp.sdc.translator.services.heattotosca.HeatToToscaUtil;
  31 import org.openecomp.sdc.translator.services.heattotosca.helper.ContrailV2VirtualMachineInterfaceHelper;
  32 import org.openecomp.sdc.translator.services.heattotosca.mapping.TranslatorHeatToToscaPropertyConverter;
  33
  34 import java.util.List;
  35 import java.util.Map;
  36
  37
  38 public class ResourceTranslationContrailV2VmInterfaceImpl extends ResourceTranslationBase {
  39   protected static Logger logger =
  40       (Logger) LoggerFactory.getLogger(ResourceTranslationContrailV2VmInterfaceImpl.class);
  41
  42   @Override
  43   protected String generateTranslatedId(TranslateTo translateTo) {
  44     if (new ContrailV2VirtualMachineInterfaceHelper().isVlanSubInterfaceResource(translateTo
  45         .getResource())) {
  46       return null;
  47     } else {
  48       return super.generateTranslatedId(translateTo);
  49     }
  50   }
  51
  52   @Override
  53   protected void translate(TranslateTo translateTo) {
  54     if (new ContrailV2VirtualMachineInterfaceHelper().isVlanSubInterfaceResource(translateTo
  55         .getResource())) {
  56       translateVlanSubInterfaceResource(translateTo);
  57     } else {
  58       translateVirtualMachineInterfaceResource(translateTo);
  59     }
  60   }
  61
  62
  63   private void translateVirtualMachineInterfaceResource(TranslateTo translateTo) {
  64     NodeTemplate nodeTemplate = new NodeTemplate();
  65     nodeTemplate.setType(ToscaNodeType.CONTRAILV2_VIRTUAL_MACHINE_INTERFACE);
  66     nodeTemplate.setProperties(TranslatorHeatToToscaPropertyConverter
  67         .getToscaPropertiesSimpleConversion(translateTo.getServiceTemplate(),translateTo.
  68                 getResourceId(),translateTo.getResource().getProperties(),
  69             nodeTemplate.getProperties(), translateTo.getHeatFileName(),
  70             translateTo.getHeatOrchestrationTemplate(), translateTo.getResource().getType(),
  71             nodeTemplate, translateTo.getContext()));
  72     String toscaVmiRefsPropertyName =
  73         HeatToToscaUtil.getToscaPropertyName(translateTo, HeatConstants.VMI_REFS_PROPERTY_NAME);
  74     if (nodeTemplate.getProperties().containsKey(toscaVmiRefsPropertyName)) {
  75       nodeTemplate.getProperties().remove(toscaVmiRefsPropertyName);
  76     }
  77
  78     handleVmiMacAddressesInProperties(translateTo, nodeTemplate);
  79
  80     new ContrailV2VirtualMachineInterfaceHelper()
  81         .connectVmiToNetwork(this, translateTo, nodeTemplate);
  82     DataModelUtil.addNodeTemplate(translateTo.getServiceTemplate(), translateTo.getTranslatedId(),
  83         nodeTemplate);
  84   }
  85
  86   private void handleVmiMacAddressesInProperties(TranslateTo translateTo,
  87                                                  NodeTemplate nodeTemplate) {
  88     String toscaVmiMacAddressesName =
  89         HeatToToscaUtil.getToscaPropertyName(translateTo, HeatConstants.VMI_MAC_ADDRESSES);
  90     String toscaVmiMacAddressesMacAddressesName =
  91         HeatToToscaUtil.getToscaPropertyName(translateTo, HeatConstants.VMI_MAC_ADDRESSES_MAC_ADDRESSES);
  92
  93     if(nodeTemplate.getProperties().containsKey(toscaVmiMacAddressesName)){
  94       Object macAddressesValue = nodeTemplate.getProperties().get(toscaVmiMacAddressesName);
  95       if(macAddressesValue instanceof Map && ((Map<String, Object>)macAddressesValue).containsKey
  96           (toscaVmiMacAddressesMacAddressesName)){
  97         updateMacAddressesMacAddressesInProperties(nodeTemplate, toscaVmiMacAddressesName,
  98             toscaVmiMacAddressesMacAddressesName,
  99             (Map<String, Object>) macAddressesValue);
 100       }
 101     }
 102   }
 103
 104   private void updateMacAddressesMacAddressesInProperties(NodeTemplate nodeTemplate,
 105                                                           String toscaVmiMacAddressesName,
 106                                                           String toscaVmiMacAddressesMacAddressesName,
 107                                                           Map<String, Object> macAddressesValue) {
 108     Object macAddressesMacAddressesValue =
 109         macAddressesValue.get(toscaVmiMacAddressesMacAddressesName);
 110     if(macAddressesMacAddressesValue instanceof List){
 111       nodeTemplate.getProperties().put(toscaVmiMacAddressesName, macAddressesMacAddressesValue);
 112     }else{
 113       nodeTemplate.getProperties().remove(toscaVmiMacAddressesName);
 114     }
 115   }
 116
 117   private void translateVlanSubInterfaceResource(TranslateTo translateTo) {
 118     new ResourceTranslationContrailV2VlanSubInterfaceImpl().translate(translateTo);
 119   }
 120
 121 }