NETVIRT-1630 migrate to md-sal APIs
[netvirt.git] / bgpmanager / impl / src / main / java / org / opendaylight / netvirt / bgpmanager / BgpUtil.java
index 6f61b3a6702614e9173251e0204ecb3905c0c496..a412f8c654206d890cd02016cab75c19b23c2e86 100755 (executable)
@@ -7,29 +7,30 @@
  */
 package org.opendaylight.netvirt.bgpmanager;
 
-import static java.util.stream.Collectors.toList;
-
-import com.google.common.base.Optional;
 import java.net.Inet4Address;
 import java.net.Inet6Address;
 import java.net.InetAddress;
 import java.util.ArrayList;
-import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
+import java.util.Optional;
 import java.util.concurrent.BlockingQueue;
+import java.util.concurrent.ExecutionException;
 import java.util.concurrent.LinkedBlockingQueue;
 import javax.annotation.PostConstruct;
 import javax.annotation.PreDestroy;
 import javax.inject.Inject;
 import javax.inject.Singleton;
-import org.opendaylight.controller.md.sal.binding.api.DataBroker;
-import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
+import org.opendaylight.genius.datastoreutils.SingleTransactionDataBroker;
 import org.opendaylight.genius.mdsalutil.MDSALUtil;
 import org.opendaylight.genius.mdsalutil.NwConstants;
 import org.opendaylight.genius.utils.batching.ActionableResource;
 import org.opendaylight.genius.utils.batching.ActionableResourceImpl;
 import org.opendaylight.genius.utils.batching.DefaultBatchHandler;
 import org.opendaylight.genius.utils.batching.ResourceBatchingManager;
+import org.opendaylight.mdsal.binding.api.DataBroker;
+import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
 import org.opendaylight.netvirt.bgpmanager.thrift.gen.af_afi;
 import org.opendaylight.netvirt.bgpmanager.thrift.gen.af_safi;
 import org.opendaylight.netvirt.bgpmanager.thrift.gen.encap_type;
@@ -41,18 +42,18 @@ import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev1509
 import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.EncapType;
 import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.LayerType;
 import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.bgp.DcgwTepList;
-import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.bgp.Vrfs;
-import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.bgp.VrfsKey;
+import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.bgp.VrfsContainer;
 import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.bgp.dcgw.tep.list.DcgwTep;
 import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.bgp.dcgw.tep.list.DcgwTepKey;
-import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.bgp.vrfs.AddressFamiliesVrf;
+import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.bgp.vrfscontainer.Vrfs;
+import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.bgp.vrfscontainer.VrfsKey;
+import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.bgp.vrfscontainer.vrfs.AddressFamiliesVrf;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rev160406.TunnelTypeBase;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rev160406.TunnelTypeMplsOverGre;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.DpnEndpoints;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.dpn.endpoints.DPNTEPsInfo;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.DcGatewayIpList;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.ElanInstances;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstance;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstanceKey;
@@ -143,7 +144,7 @@ public class BgpUtil implements AutoCloseable {
     }
 
     public <T extends DataObject> void update(final InstanceIdentifier<T> path, final T data) {
-        ActionableResource actResource = new ActionableResourceImpl(path.toString());
+        ActionableResourceImpl actResource = new ActionableResourceImpl(path.toString());
         actResource.setAction(ActionableResource.UPDATE);
         actResource.setInstanceIdentifier(path);
         actResource.setInstance(data);
@@ -151,7 +152,7 @@ public class BgpUtil implements AutoCloseable {
     }
 
     public <T extends DataObject> void write(final InstanceIdentifier<T> path, final T data) {
-        ActionableResource actResource = new ActionableResourceImpl(path.toString());
+        ActionableResourceImpl actResource = new ActionableResourceImpl(path.toString());
         actResource.setAction(ActionableResource.CREATE);
         actResource.setInstanceIdentifier(path);
         actResource.setInstance(data);
@@ -159,7 +160,7 @@ public class BgpUtil implements AutoCloseable {
     }
 
     public <T extends DataObject> void delete(final InstanceIdentifier<T> path) {
-        ActionableResource actResource = new ActionableResourceImpl(path.toString());
+        ActionableResourceImpl actResource = new ActionableResourceImpl(path.toString());
         actResource.setAction(ActionableResource.DELETE);
         actResource.setInstanceIdentifier(path);
         actResource.setInstance(null);
@@ -199,8 +200,13 @@ public class BgpUtil implements AutoCloseable {
 
     public VpnInstanceOpDataEntry getVpnInstanceOpData(String rd)  {
         InstanceIdentifier<VpnInstanceOpDataEntry> id = getVpnInstanceOpDataIdentifier(rd);
-        Optional<VpnInstanceOpDataEntry> vpnInstanceOpData = MDSALUtil.read(dataBroker,
-                LogicalDatastoreType.OPERATIONAL, id);
+        Optional<VpnInstanceOpDataEntry> vpnInstanceOpData = Optional.empty();
+        try {
+            vpnInstanceOpData = SingleTransactionDataBroker.syncReadOptional(dataBroker,
+                    LogicalDatastoreType.OPERATIONAL, id);
+        } catch (ExecutionException | InterruptedException e) {
+            LOG.error("Exception while reading VpnInstanceOpDataEntry DS for the Vpn Rd {}", rd, e);
+        }
         if (vpnInstanceOpData.isPresent()) {
             return vpnInstanceOpData.get();
         }
@@ -214,8 +220,13 @@ public class BgpUtil implements AutoCloseable {
 
     private String getElanNamefromRd(String rd)  {
         InstanceIdentifier<EvpnRdToNetwork> id = getEvpnRdToNetworkIdentifier(rd);
-        Optional<EvpnRdToNetwork> evpnRdToNetworkOpData = MDSALUtil.read(dataBroker,
-                LogicalDatastoreType.CONFIGURATION, id);
+        Optional<EvpnRdToNetwork> evpnRdToNetworkOpData = Optional.empty();
+        try {
+            evpnRdToNetworkOpData = SingleTransactionDataBroker.syncReadOptional(dataBroker,
+                    LogicalDatastoreType.CONFIGURATION, id);
+        } catch (ExecutionException | InterruptedException e) {
+            LOG.error("Exception while reading EvpnRdToNetwork DS for the Vpn Rd {}", rd, e);
+        }
         if (evpnRdToNetworkOpData.isPresent()) {
             return evpnRdToNetworkOpData.get().getNetworkId();
         }
@@ -268,8 +279,15 @@ public class BgpUtil implements AutoCloseable {
     }
 
     public String getVpnNameFromRd(String rd) {
-        VpnInstanceOpDataEntry vpnInstanceOpData = getVpnInstanceOpData(rd);
-        return vpnInstanceOpData != null ? vpnInstanceOpData.getVpnInstanceName() : null;
+        final Map<String, String> rdtoVpnMap = new HashMap<>();
+        if (rdtoVpnMap.get(rd) != null) {
+            return rdtoVpnMap.get(rd);
+        } else {
+            VpnInstanceOpDataEntry vpnInstanceOpData = getVpnInstanceOpData(rd);
+            String vpnName = vpnInstanceOpData != null ? vpnInstanceOpData.getVpnInstanceName() : null;
+            rdtoVpnMap.put(rd, vpnName);
+            return vpnName;
+        }
     }
 
     /** get the vrf with the RouterDistinguisher pass in param.
@@ -279,8 +297,15 @@ public class BgpUtil implements AutoCloseable {
     public Vrfs getVrfFromRd(String rd) {
         Vrfs vrfs = null;
         KeyedInstanceIdentifier<Vrfs, VrfsKey> id = InstanceIdentifier.create(Bgp.class)
+                .child(VrfsContainer.class)
                 .child(Vrfs.class, new VrfsKey(rd));
-        Optional<Vrfs> vrfsFromDs = MDSALUtil.read(dataBroker, LogicalDatastoreType.CONFIGURATION, id);
+        Optional<Vrfs> vrfsFromDs = Optional.empty();
+        try {
+            vrfsFromDs = SingleTransactionDataBroker.syncReadOptional(dataBroker, LogicalDatastoreType.CONFIGURATION,
+                    id);
+        } catch (ExecutionException | InterruptedException e) {
+            LOG.error("Exception while reading BGP VRF table for the Vpn Rd {}", rd, e);
+        }
         if (vrfsFromDs.isPresent()) {
             vrfs = vrfsFromDs.get();
         }
@@ -293,9 +318,9 @@ public class BgpUtil implements AutoCloseable {
      */
     public static LayerType getLayerType(AddressFamiliesVrf adf) {
         LayerType layerTypeValue = null;
-        if (adf.getSafi() == af_safi.SAFI_EVPN.getValue()) {
+        if (adf.getSafi().intValue() == af_safi.SAFI_EVPN.getValue()) {
             layerTypeValue = LayerType.LAYER2;
-        } else if (adf.getSafi() == af_safi.SAFI_MPLS_VPN.getValue()) {
+        } else if (adf.getSafi().intValue() == af_safi.SAFI_MPLS_VPN.getValue()) {
             layerTypeValue = LayerType.LAYER3;
         }
         return layerTypeValue;
@@ -334,8 +359,13 @@ public class BgpUtil implements AutoCloseable {
     public BfdConfig getBfdConfig() {
         InstanceIdentifier<BfdConfig> id =
                 InstanceIdentifier.builder(BfdConfig.class).build();
-        Optional<BfdConfig> bfdConfigOptional = MDSALUtil.read(dataBroker,
-                LogicalDatastoreType.CONFIGURATION, id);
+        Optional<BfdConfig> bfdConfigOptional = Optional.empty();
+        try {
+            bfdConfigOptional = SingleTransactionDataBroker.syncReadOptional(dataBroker,
+                    LogicalDatastoreType.CONFIGURATION, id);
+        } catch (ExecutionException | InterruptedException e) {
+            LOG.error("Exception while reading BfdConfig", e);
+        }
         if (bfdConfigOptional.isPresent()) {
             return bfdConfigOptional.get();
         }
@@ -345,8 +375,13 @@ public class BgpUtil implements AutoCloseable {
     public DcgwTepList getDcgwTepConfig() {
         InstanceIdentifier<DcgwTepList> id =
                 InstanceIdentifier.builder(Bgp.class).child(DcgwTepList.class).build();
-        Optional<DcgwTepList> dcgwTepListOptional = MDSALUtil.read(dataBroker,
-                LogicalDatastoreType.CONFIGURATION, id);
+        Optional<DcgwTepList> dcgwTepListOptional = Optional.empty();
+        try {
+            dcgwTepListOptional = SingleTransactionDataBroker.syncReadOptional(dataBroker,
+                    LogicalDatastoreType.CONFIGURATION, id);
+        } catch (ExecutionException | InterruptedException e) {
+            LOG.error("getDcgwTepConfig: Exception while reading DcgwTepList", e);
+        }
         if (dcgwTepListOptional.isPresent()) {
             return dcgwTepListOptional.get();
         }
@@ -358,8 +393,13 @@ public class BgpUtil implements AutoCloseable {
                 InstanceIdentifier.builder(Bgp.class)
                         .child(DcgwTepList.class)
                         .child(DcgwTep.class, new DcgwTepKey(dcgwIp)).build();
-        Optional<DcgwTep> tepListOptional = MDSALUtil.read(dataBroker,
-                LogicalDatastoreType.CONFIGURATION, id);
+        Optional<DcgwTep> tepListOptional = Optional.empty();
+        try {
+            tepListOptional = SingleTransactionDataBroker.syncReadOptional(dataBroker,
+                    LogicalDatastoreType.CONFIGURATION, id);
+        } catch (ExecutionException | InterruptedException e) {
+            LOG.error("Exception while reading DcgwTep for the IP {}", dcgwIp, e);
+        }
         if (tepListOptional.isPresent()) {
             return tepListOptional.get().getTepIps();
         }
@@ -377,25 +417,7 @@ public class BgpUtil implements AutoCloseable {
         }
     }
 
-    private List<String> getDcGwIps() {
-        InstanceIdentifier<DcGatewayIpList> dcGatewayIpListid =
-                InstanceIdentifier.builder(DcGatewayIpList.class).build();
-        DcGatewayIpList dcGatewayIpListConfig =
-                MDSALUtil.read(dataBroker, LogicalDatastoreType.CONFIGURATION, dcGatewayIpListid).orNull();
-        if (dcGatewayIpListConfig == null) {
-            return Collections.EMPTY_LIST;
-        }
-        return dcGatewayIpListConfig.getDcGatewayIp()
-                .stream()
-                .filter(dcGwIp -> dcGwIp.getTunnnelType().equals(TunnelTypeMplsOverGre.class))
-                .map(dcGwIp -> String.valueOf(dcGwIp.getIpAddress().getIpv4Address())).sorted()
-                .collect(toList());
-    }
-
-
-    public void removeOrUpdateLBGroups(String tepIp, int addRemoveOrUpdate, boolean isTunnelUp) {
-        LOG.debug("removing bucket towards DCGW {}", tepIp);
-        List<String> availableDcGws = getDcGwIps();
+    public void removeOrUpdateLBGroups(String tepIp, int addRemoveOrUpdate) {
         getDpnTEPsInfos(dataBroker).forEach(dpnInfo -> {
             if (NwConstants.MOD_FLOW == addRemoveOrUpdate) {
                 LOG.debug("Updating bucket in DPN {}", dpnInfo.getDPNID());
@@ -403,8 +425,8 @@ public class BgpUtil implements AutoCloseable {
                 LOG.debug("Deleting groups in DPN {}", dpnInfo.getDPNID());
             }
             Class<? extends TunnelTypeBase> tunType = TunnelTypeMplsOverGre.class;
-            fibManager.programDcGwLoadBalancingGroup(availableDcGws, dpnInfo.getDPNID(),
-                    tepIp, addRemoveOrUpdate, isTunnelUp, tunType);
+            fibManager.programDcGwLoadBalancingGroup(dpnInfo.getDPNID(),
+                    tepIp, addRemoveOrUpdate, false, tunType);
         });
     }
 }