Convert itm-impl to use mdsal-binding-util
[genius.git] / itm / itm-impl / src / main / java / org / opendaylight / genius / itm / confighelpers / ItmMonitorToggleWorker.java
index 1502ef000021b712b14a4b9846ce27c1296bff91..b603679d74a7dc156f7f10f53b33c2a9500d8fe9 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2016 Ericsson India Global Services Pvt Ltd. and others.  All rights reserved.
+ * Copyright (c) 2016, 2017 Ericsson India Global Services Pvt Ltd. and others. All rights reserved.
  *
  * This program and the accompanying materials are made available under the
  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
@@ -8,78 +8,74 @@
 package org.opendaylight.genius.itm.confighelpers;
 
 import com.google.common.util.concurrent.ListenableFuture;
-import org.opendaylight.controller.md.sal.binding.api.DataBroker;
-import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
-import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
-import org.opendaylight.genius.itm.globals.ITMConstants;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.Callable;
 import org.opendaylight.genius.itm.impl.ItmUtils;
-import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
-import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceBuilder;
-import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceKey;
+import org.opendaylight.mdsal.binding.api.DataBroker;
+import org.opendaylight.mdsal.binding.util.Datastore;
+import org.opendaylight.mdsal.binding.util.ManagedNewTransactionRunner;
+import org.opendaylight.mdsal.binding.util.ManagedNewTransactionRunnerImpl;
+import org.opendaylight.mdsal.binding.util.TypedWriteTransaction;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rev160406.IfTunnel;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rev160406.IfTunnelBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rev160406.TunnelMonitoringTypeBase;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.config.rev160406.TunnelMonitorParams;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.config.rev160406.TunnelMonitorParamsBuilder;
 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import java.util.ArrayList;
-import java.util.List;
-import java.util.concurrent.Callable;
+public class ItmMonitorToggleWorker implements Callable<List<? extends ListenableFuture<?>>> {
 
+    private static final Logger LOG = LoggerFactory.getLogger(ItmMonitorToggleWorker.class);
 
-public class ItmMonitorToggleWorker implements Callable<List<ListenableFuture<Void>>> {
-    private static final Logger logger = LoggerFactory.getLogger(ItmMonitorToggleWorker.class) ;
-    private DataBroker dataBroker;
-    private String tzone;
-    private boolean enabled;
-    private List<HwVtep> hwVteps;
-    private  Boolean exists;
-    private Class<? extends TunnelMonitoringTypeBase> monitorProtocol = ITMConstants.DEFAULT_MONITOR_PROTOCOL;
+    private final DataBroker dataBroker;
+    private final String tzone;
+    private final boolean enabled;
+    private final Class<? extends TunnelMonitoringTypeBase> monitorProtocol;
+    private final ManagedNewTransactionRunner txRunner;
 
-    public  ItmMonitorToggleWorker(List<HwVtep> hwVteps,String tzone,boolean enabled, Class<? extends TunnelMonitoringTypeBase> monitorProtocol, DataBroker dataBroker, Boolean exists){
+    public ItmMonitorToggleWorker(String tzone, boolean enabled,
+            Class<? extends TunnelMonitoringTypeBase> monitorProtocol, DataBroker dataBroker) {
         this.dataBroker = dataBroker;
         this.tzone = tzone;
         this.enabled = enabled;
-        this.hwVteps = hwVteps;
-        this.exists = exists;
         this.monitorProtocol = monitorProtocol;
-        logger.trace("ItmMonitorToggleWorker initialized with  tzone {} and toggleBoolean {}",tzone,enabled );
-        logger.debug("TunnelMonitorToggleWorker with monitor protocol = {} ",monitorProtocol);
+        this.txRunner = new ManagedNewTransactionRunnerImpl(dataBroker);
+        LOG.trace("ItmMonitorToggleWorker initialized with  tzone {} and toggleBoolean {}",tzone,enabled);
+        LOG.debug("TunnelMonitorToggleWorker with monitor protocol = {} ",monitorProtocol);
     }
 
-    @Override public List<ListenableFuture<Void>> call() throws Exception {
-        List<ListenableFuture<Void>> futures = new ArrayList<>() ;
-        logger.debug("Invoking Tunnel Monitor Worker tzone = {} enabled {}",tzone,enabled );
-        WriteTransaction t = dataBroker.newWriteOnlyTransaction();
-        toggleTunnelMonitoring(hwVteps,enabled,tzone,t,exists);
-        futures.add(t.submit());
-        return futures;
+    @Override
+    public List<? extends ListenableFuture<?>> call() {
+        LOG.debug("ItmMonitorToggleWorker invoked with tzone = {} enabled {}",tzone,enabled);
+        return toggleTunnelMonitoring();
     }
 
-    private void toggleTunnelMonitoring(List<HwVtep> hwVteps,Boolean enabled, String tzone, WriteTransaction t,Boolean exists) {
-        //exists means hwVteps exist for this tzone
-
-        List<String> TunnelList = ItmUtils.getTunnelsofTzone(hwVteps,tzone,dataBroker,exists);
-        if(TunnelList !=null &&!TunnelList.isEmpty()) {
-            for (String tunnel : TunnelList)
-                toggle(tunnel, enabled,t);
-        }
+    private List<? extends ListenableFuture<?>> toggleTunnelMonitoring() {
+        List<String> tunnelList = ItmUtils.getInternalTunnelInterfaces(dataBroker);
+        LOG.debug("toggleTunnelMonitoring: TunnelList size {}", tunnelList.size());
+        InstanceIdentifier<TunnelMonitorParams> iid = InstanceIdentifier.create(TunnelMonitorParams.class);
+        TunnelMonitorParams monitorParams = new TunnelMonitorParamsBuilder()
+                .setEnabled(enabled).setMonitorProtocol(monitorProtocol).build();
+        LOG.debug("toggleTunnelMonitoring: Updating Operational DS");
+        List<ListenableFuture<?>> futures = new ArrayList<>();
+        futures.add(txRunner.callWithNewWriteOnlyTransactionAndSubmit(Datastore.OPERATIONAL,
+            tx -> tx.merge(iid, monitorParams)));
+        futures.add(txRunner.callWithNewWriteOnlyTransactionAndSubmit(Datastore.CONFIGURATION,
+            tx -> tunnelList.forEach(tunnel -> toggle(tunnel, tx))));
+        return futures;
     }
 
-    private void toggle(String tunnelInterfaceName, boolean enabled, WriteTransaction t) {
-        if(tunnelInterfaceName!=null) {
-            InstanceIdentifier<Interface> trunkIdentifier = ItmUtils.buildId(tunnelInterfaceName);
-            logger.debug("TunnelMonitorToggleWorker: toggle with monitor protocol = {} ",monitorProtocol);
-            IfTunnel tunnel = new IfTunnelBuilder().setMonitorEnabled(enabled).setMonitorProtocol(monitorProtocol).build();
-            InterfaceBuilder builder = new InterfaceBuilder().setKey(new InterfaceKey(tunnelInterfaceName))
-                    .addAugmentation(IfTunnel.class, tunnel);
-            t.merge(LogicalDatastoreType.CONFIGURATION, trunkIdentifier, builder.build());
+    private void toggle(String tunnelInterfaceName, TypedWriteTransaction<Datastore.Configuration> tx) {
+        if (tunnelInterfaceName != null) {
+            InstanceIdentifier<IfTunnel> trunkIdentifier = ItmUtils.buildTunnelId(tunnelInterfaceName);
+            LOG.debug("TunnelMonitorToggleWorker: tunnelInterfaceName: {}, monitorProtocol = {}, "
+                    + "monitorEnable = {} ",tunnelInterfaceName, monitorProtocol, enabled);
+            IfTunnel tunnel =
+                    new IfTunnelBuilder().setMonitorEnabled(enabled).setMonitorProtocol(monitorProtocol).build();
+            tx.merge(trunkIdentifier, tunnel);
         }
     }
 }
-
-
-
-
-