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.genius.utils.hwvtep.HwvtepSouthboundConstants;
  16 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
  17 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.l2.types.rev130827.VlanId;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentation;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepLogicalSwitchRef;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepNodeName;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalPortAugmentationBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitches;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitchesKey;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.port.attributes.VlanBindings;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.port.attributes.VlanBindingsBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.port.attributes.VlanBindingsKey;
  28 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
  29 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
  30 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TpId;
  31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
  33 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  34 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
  35 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
  36 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointKey;
  38 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  39
  40 /**
  41  * Created by eaksahu on 8/8/2016.
  42  */
  43 public final class PhysicalSwitchHelper {
  44     static InstanceIdentifier<Node> dId;
  45
  46     private PhysicalSwitchHelper() {
  47
  48     }
  49
  50     public static InstanceIdentifier<Node> getPhysicalSwitchInstanceIdentifier(InstanceIdentifier<Node> iid,
  51                                                                                String switchName) {
  52         NodeId id = iid.firstKeyOf(Node.class).getNodeId();
  53         String nodeString = id.getValue() + "/physicalswitch/" + switchName;
  54         NodeId nodeId = new NodeId(new Uri(nodeString));
  55         NodeKey nodeKey = new NodeKey(nodeId);
  56         TopologyKey topoKey = new TopologyKey(HwvtepSouthboundConstants.HWVTEP_TOPOLOGY_ID);
  57         return InstanceIdentifier.builder(NetworkTopology.class)
  58                 .child(Topology.class, topoKey)
  59                 .child(Node.class, nodeKey)
  60                 .build();
  61     }
  62
  63     static List<TerminationPoint> addPhysicalSwitchTerminationPoints(InstanceIdentifier<Node> switchIid,
  64         List<String> portNames) {
  65         List<TerminationPoint> tps = new ArrayList<>();
  66         for (String portName : portNames) {
  67             tps.add(buildTerminationPointForPhysicalSwitch(switchIid, portName, getVlanBindingData(1)));
  68         }
  69         return tps;
  70     }
  71
  72     private static TerminationPoint buildTerminationPointForPhysicalSwitch(InstanceIdentifier<Node> switchIid,
  73         String portName, Map<Long, String> vlanBindingData) {
  74         TerminationPointKey tpKey = new TerminationPointKey(new TpId(portName));
  75         TerminationPointBuilder tpBuilder = new TerminationPointBuilder();
  76         tpBuilder.withKey(tpKey);
  77         tpBuilder.setTpId(tpKey.getTpId());
  78         switchIid.firstKeyOf(Node.class);
  79         HwvtepPhysicalPortAugmentationBuilder tpAugmentationBuilder =
  80                 new HwvtepPhysicalPortAugmentationBuilder();
  81         buildTerminationPoint(tpAugmentationBuilder, portName, vlanBindingData);
  82         tpBuilder.addAugmentation(tpAugmentationBuilder.build());
  83         return tpBuilder.build();
  84     }
  85
  86     private static void buildTerminationPoint(HwvtepPhysicalPortAugmentationBuilder tpAugmentationBuilder,
  87             String portName, Map<Long, String> vlanBindingData) {
  88         updatePhysicalPortId(portName, tpAugmentationBuilder);
  89         updatePort(tpAugmentationBuilder, vlanBindingData);
  90     }
  91
  92     private static void updatePhysicalPortId(String portName,
  93             HwvtepPhysicalPortAugmentationBuilder tpAugmentationBuilder) {
  94         tpAugmentationBuilder.setHwvtepNodeName(new HwvtepNodeName(portName));
  95         tpAugmentationBuilder.setHwvtepNodeDescription("");
  96     }
  97
  98     private static void updatePort(HwvtepPhysicalPortAugmentationBuilder tpAugmentationBuilder,
  99             Map<Long, String> vlanBindings) {
 100         updateVlanBindings(vlanBindings, tpAugmentationBuilder);
 101         tpAugmentationBuilder.setPhysicalPortUuid(new Uuid(UUID.randomUUID().toString()));
 102     }
 103
 104     private static void updateVlanBindings(Map<Long, String> vlanBindings,
 105             HwvtepPhysicalPortAugmentationBuilder tpAugmentationBuilder) {
 106         List<VlanBindings> vlanBindingsList = new ArrayList<>();
 107         for (Map.Entry<Long, String> vlanBindingEntry : vlanBindings.entrySet()) {
 108             Long vlanBindingKey = vlanBindingEntry.getKey();
 109             String logicalSwitch = vlanBindingEntry.getValue();
 110             if (logicalSwitch != null && vlanBindingKey != null) {
 111                 vlanBindingsList.add(createVlanBinding(vlanBindingKey, logicalSwitch));
 112             }
 113         }
 114         tpAugmentationBuilder.setVlanBindings(vlanBindingsList);
 115     }
 116
 117     private static VlanBindings createVlanBinding(Long key, String logicalSwitch) {
 118         VlanBindingsBuilder vbBuilder = new VlanBindingsBuilder();
 119         VlanBindingsKey vbKey = new VlanBindingsKey(new VlanId(key.intValue()));
 120         vbBuilder.withKey(vbKey);
 121         vbBuilder.setVlanIdKey(vbKey.getVlanIdKey());
 122         HwvtepLogicalSwitchRef hwvtepLogicalSwitchRef =
 123                 new HwvtepLogicalSwitchRef(createInstanceIdentifier(logicalSwitch));
 124         vbBuilder.setLogicalSwitchRef(hwvtepLogicalSwitchRef);
 125         return vbBuilder.build();
 126     }
 127
 128     private static InstanceIdentifier<LogicalSwitches> createInstanceIdentifier(String logicalSwitch) {
 129         NodeId id = dId.firstKeyOf(Node.class).getNodeId();
 130         NodeKey nodeKey = new NodeKey(id);
 131         return InstanceIdentifier.builder(NetworkTopology.class)
 132                 .child(Topology.class, new TopologyKey(HwvtepSouthboundConstants.HWVTEP_TOPOLOGY_ID))
 133                 .child(Node.class, nodeKey).augmentation(HwvtepGlobalAugmentation.class)
 134                 .child(LogicalSwitches.class, new LogicalSwitchesKey(new HwvtepNodeName(logicalSwitch)))
 135                 .build();
 136     }
 137
 138     private static Map<Long, String> getVlanBindingData(int mapSize) {
 139         Map<Long, String> vlanBindings = new HashMap<>();
 140         for (long i = 0; i < mapSize; i++) {
 141             i = i * 100;
 142             vlanBindings.put(i, "9227c228-6bba-4bbe-bdb8-6942768ff0f1");
 143         }
 144         return vlanBindings;
 145     }
 146
 147 }