Merge "JUnit test case for ITM"
[genius.git] / itm / itm-impl / src / main / java / org / opendaylight / genius / itm / confighelpers / ItmMonitorToggleWorker.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.genius.itm.confighelpers;
9
10 import com.google.common.util.concurrent.ListenableFuture;
11 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
12 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
13 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
14 import org.opendaylight.genius.itm.globals.ITMConstants;
15 import org.opendaylight.genius.itm.impl.ItmUtils;
16 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
17 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceBuilder;
18 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceKey;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rev160406.IfTunnel;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rev160406.IfTunnelBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rev160406.TunnelMonitoringTypeBase;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.config.rev160406.TunnelMonitorParams;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.config.rev160406.TunnelMonitorParamsBuilder;
24 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
25 import org.slf4j.Logger;
26 import org.slf4j.LoggerFactory;
27
28 import java.util.ArrayList;
29 import java.util.List;
30 import java.util.concurrent.Callable;
31
32 public class ItmMonitorToggleWorker implements Callable<List<ListenableFuture<Void>>> {
33     private static final Logger logger = LoggerFactory.getLogger(ItmMonitorToggleWorker.class) ;
34     private DataBroker dataBroker;
35     private String tzone;
36     private boolean enabled;
37     private List<HwVtep> hwVteps;
38     private  Boolean exists;
39     private Class<? extends TunnelMonitoringTypeBase> monitorProtocol = ITMConstants.DEFAULT_MONITOR_PROTOCOL;
40
41     public  ItmMonitorToggleWorker(List<HwVtep> hwVteps,String tzone,boolean enabled, Class<? extends TunnelMonitoringTypeBase> monitorProtocol, DataBroker dataBroker, Boolean exists){
42         this.dataBroker = dataBroker;
43         this.tzone = tzone;
44         this.enabled = enabled;
45         this.hwVteps = hwVteps;
46         this.exists = exists;
47         this.monitorProtocol = monitorProtocol;
48         logger.trace("ItmMonitorToggleWorker initialized with  tzone {} and toggleBoolean {}",tzone,enabled );
49         logger.debug("TunnelMonitorToggleWorker with monitor protocol = {} ",monitorProtocol);
50     }
51
52     @Override public List<ListenableFuture<Void>> call() throws Exception {
53         List<ListenableFuture<Void>> futures = new ArrayList<>() ;
54         logger.debug("Invoking Tunnel Monitor Worker tzone = {} enabled {}",tzone,enabled );
55         WriteTransaction t = dataBroker.newWriteOnlyTransaction();
56         toggleTunnelMonitoring(hwVteps,enabled,tzone,t,exists);
57         futures.add(t.submit());
58         return futures;
59     }
60
61     private void toggleTunnelMonitoring(List<HwVtep> hwVteps,Boolean enabled, String tzone, WriteTransaction t,Boolean exists) {
62         //exists means hwVteps exist for this tzone
63
64         List<String> TunnelList = ItmUtils.getTunnelsofTzone(hwVteps,tzone,dataBroker,exists);
65         if(TunnelList !=null &&!TunnelList.isEmpty()) {
66             for (String tunnel : TunnelList)
67                 toggle(tunnel, enabled,t);
68         }
69     }
70
71     private void toggle(String tunnelInterfaceName, boolean enabled, WriteTransaction t) {
72         if(tunnelInterfaceName!=null) {
73             InstanceIdentifier<Interface> trunkIdentifier = ItmUtils.buildId(tunnelInterfaceName);
74             logger.debug("TunnelMonitorToggleWorker: toggle with monitor protocol = {} ",monitorProtocol);
75             IfTunnel tunnel = new IfTunnelBuilder().setMonitorEnabled(enabled).setMonitorProtocol(monitorProtocol).build();
76             InterfaceBuilder builder = new InterfaceBuilder().setKey(new InterfaceKey(tunnelInterfaceName))
77                     .addAugmentation(IfTunnel.class, tunnel);
78             t.merge(LogicalDatastoreType.CONFIGURATION, trunkIdentifier, builder.build());
79             InstanceIdentifier<TunnelMonitorParams> iid = InstanceIdentifier.builder(TunnelMonitorParams.class).build();
80             TunnelMonitorParams protocolBuilder = new TunnelMonitorParamsBuilder().setEnabled(enabled).setMonitorProtocol(monitorProtocol).build();
81             ItmUtils.asyncUpdate(LogicalDatastoreType.OPERATIONAL,iid, protocolBuilder, dataBroker, ItmUtils.DEFAULT_CALLBACK);
82
83         }
84     }
85 }
86
87
88
89