elanmanager/impl/src/test/java/org/opendaylight/netvirt/elan/l2gw/nodehandlertest/PhysicalSwitchHelper.java

   1 /*
   2  * Copyright (c) 2016 Ericsson India Global Services Pvt Ltd. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.netvirt.elan.l2gw.nodehandlertest;
   9
  10 import java.util.ArrayList;
  11 import java.util.HashMap;
  12 import java.util.List;
  13 import java.util.Map;
  14 import java.util.UUID;
  15 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  16 import org.opendaylight.genius.utils.hwvtep.HwvtepSouthboundConstants;
  17 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
  18 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.l2.types.rev130827.VlanId;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentation;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepLogicalSwitchRef;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepNodeName;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalPortAugmentation;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalPortAugmentationBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitches;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitchesKey;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.port.attributes.VlanBindings;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.port.attributes.VlanBindingsBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.port.attributes.VlanBindingsKey;
  30 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
  31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
  32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TpId;
  33 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  34 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
  35 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  36 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
  37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
  38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointKey;
  40 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  41 import org.slf4j.Logger;
  42 import org.slf4j.LoggerFactory;
  43
  44 /**
  45  * Created by eaksahu on 8/8/2016.
  46  */
  47 public final class PhysicalSwitchHelper {
  48     static Logger LOG = LoggerFactory.getLogger(PhysicalSwitchHelper.class);
  49
  50     static InstanceIdentifier<Node> dId;
  51
  52     private PhysicalSwitchHelper() { }
  53
  54     public static InstanceIdentifier<Node> getPhysicalSwitchInstanceIdentifier(InstanceIdentifier<Node> iid,
  55                                                                                String switchName) {
  56         NodeId id = iid.firstKeyOf(Node.class).getNodeId();
  57         String nodeString = id.getValue() + "/physicalswitch/" + switchName;
  58         NodeId nodeId = new NodeId(new Uri(nodeString));
  59         NodeKey nodeKey = new NodeKey(nodeId);
  60         TopologyKey topoKey = new TopologyKey(HwvtepSouthboundConstants.HWVTEP_TOPOLOGY_ID);
  61         return InstanceIdentifier.builder(NetworkTopology.class)
  62                 .child(Topology.class, topoKey)
  63                 .child(Node.class, nodeKey)
  64                 .build();
  65     }
  66
  67     public static List<TerminationPoint> addPhysicalSwitchTerminationPoints(InstanceIdentifier<Node> switchIid,
  68                                                                             WriteTransaction transaction,
  69                                                                             List<String> portNames) {
  70         List<TerminationPoint> tps = new ArrayList<>();
  71         for (String portName : portNames) {
  72             tps.add(buildTerminationPointForPhysicalSwitch(switchIid, portName, transaction, getVlanBindingData(1)));
  73         }
  74         return tps;
  75     }
  76
  77     public static TerminationPoint buildTerminationPointForPhysicalSwitch(InstanceIdentifier<Node> switchIid,
  78                                                                           String portName, WriteTransaction transaction,
  79                                                                           Map<Long, String> vlanBindingData) {
  80         TerminationPointKey tpKey = new TerminationPointKey(new TpId(portName));
  81         TerminationPointBuilder tpBuilder = new TerminationPointBuilder();
  82         tpBuilder.setKey(tpKey);
  83         tpBuilder.setTpId(tpKey.getTpId());
  84         switchIid.firstKeyOf(Node.class);
  85         InstanceIdentifier<TerminationPoint> tpPath = switchIid.child(TerminationPoint.class,
  86                 new TerminationPointKey(new TpId(portName)));
  87         HwvtepPhysicalPortAugmentationBuilder tpAugmentationBuilder =
  88                 new HwvtepPhysicalPortAugmentationBuilder();
  89         buildTerminationPoint(tpAugmentationBuilder, portName, vlanBindingData);
  90         tpBuilder.addAugmentation(HwvtepPhysicalPortAugmentation.class, tpAugmentationBuilder.build());
  91         return tpBuilder.build();
  92     }
  93
  94     public static void buildTerminationPoint(HwvtepPhysicalPortAugmentationBuilder tpAugmentationBuilder,
  95                                              String portName, Map<Long, String> vlanBindingData) {
  96         updatePhysicalPortId(portName, tpAugmentationBuilder);
  97         updatePort(portName, tpAugmentationBuilder, vlanBindingData);
  98     }
  99
 100     public static void updatePhysicalPortId(String portName,
 101                                             HwvtepPhysicalPortAugmentationBuilder tpAugmentationBuilder) {
 102         tpAugmentationBuilder.setHwvtepNodeName(new HwvtepNodeName(portName));
 103         tpAugmentationBuilder.setHwvtepNodeDescription("");
 104     }
 105
 106     public static void updatePort(String portName, HwvtepPhysicalPortAugmentationBuilder tpAugmentationBuilder,
 107                                   Map<Long, String> vlanBindings) {
 108         updateVlanBindings(vlanBindings, tpAugmentationBuilder);
 109         tpAugmentationBuilder.setPhysicalPortUuid(new Uuid(UUID.randomUUID().toString()));
 110     }
 111
 112     public static void updateVlanBindings(Map<Long, String> vlanBindings,
 113                                           HwvtepPhysicalPortAugmentationBuilder tpAugmentationBuilder) {
 114         List<VlanBindings> vlanBindingsList = new ArrayList<>();
 115         for (Map.Entry<Long, String> vlanBindingEntry : vlanBindings.entrySet()) {
 116             Long vlanBindingKey = vlanBindingEntry.getKey();
 117             String logicalSwitch = vlanBindingEntry.getValue();
 118             if (logicalSwitch != null && vlanBindingKey != null) {
 119                 vlanBindingsList.add(createVlanBinding(vlanBindingKey, logicalSwitch));
 120             }
 121         }
 122         tpAugmentationBuilder.setVlanBindings(vlanBindingsList);
 123     }
 124
 125     public static VlanBindings createVlanBinding(Long key, String logicalSwitch) {
 126         VlanBindingsBuilder vbBuilder = new VlanBindingsBuilder();
 127         VlanBindingsKey vbKey = new VlanBindingsKey(new VlanId(key.intValue()));
 128         vbBuilder.setKey(vbKey);
 129         vbBuilder.setVlanIdKey(vbKey.getVlanIdKey());
 130         HwvtepLogicalSwitchRef hwvtepLogicalSwitchRef =
 131                 new HwvtepLogicalSwitchRef(createInstanceIdentifier(logicalSwitch));
 132         vbBuilder.setLogicalSwitchRef(hwvtepLogicalSwitchRef);
 133         return vbBuilder.build();
 134     }
 135
 136     public static InstanceIdentifier<LogicalSwitches> createInstanceIdentifier(String logicalSwitch) {
 137         NodeId id = dId.firstKeyOf(Node.class).getNodeId();
 138         NodeKey nodeKey = new NodeKey(id);
 139         InstanceIdentifier<LogicalSwitches> iid = null;
 140         iid = InstanceIdentifier.builder(NetworkTopology.class)
 141                 .child(Topology.class, new TopologyKey(HwvtepSouthboundConstants.HWVTEP_TOPOLOGY_ID))
 142                 .child(Node.class, nodeKey).augmentation(HwvtepGlobalAugmentation.class)
 143                 .child(LogicalSwitches.class, new LogicalSwitchesKey(new HwvtepNodeName(logicalSwitch)))
 144                 .build();
 145         return iid;
 146     }
 147
 148     public static Map<Long, String> getVlanBindingData(int mapSize) {
 149         Map<Long, String> vlanBindings = new HashMap<>();
 150         for (Integer i = 0; i < mapSize; i++) {
 151             i = i * 100;
 152             vlanBindings.put(Long.valueOf(i), "9227c228-6bba-4bbe-bdb8-6942768ff0f1");
 153         }
 154         return vlanBindings;
 155     }
 156
 157 }