Fixing issues with ARP integration of VpnService
[vpnservice.git] / itm / itm-impl / src / main / java / org / opendaylight / vpnservice / itm / listeners / TransportZoneListener.java
1 /*
2  * Copyright (c) 2015 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
9 package org.opendaylight.vpnservice.itm.listeners;
10
11 import java.math.BigInteger;
12 import java.util.ArrayList;
13 import java.util.List;
14 import java.util.Map;
15 import java.util.Set;
16 import java.util.concurrent.ConcurrentHashMap;
17
18 import com.google.common.base.Optional;
19 import com.google.common.util.concurrent.CheckedFuture;
20 import com.google.common.util.concurrent.ListenableFuture;
21
22 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
23 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
24 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpPrefix;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.itm.op.rev150701.tunnels.DPNTEPsInfo;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.itm.op.rev150701.tunnels.dpn.teps.info.TunnelEndPoints;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.itm.rev150701.TransportZones;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.itm.rev150701.TransportZonesBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.itm.rev150701.transport.zones.TransportZone;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.itm.rev150701.transport.zones.transport.zone.Subnets;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.itm.rev150701.transport.zones.transport.zone.subnets.Vteps;
34 import org.opendaylight.vpnservice.datastoreutils.AsyncDataTreeChangeListenerBase;
35 import org.opendaylight.vpnservice.datastoreutils.DataStoreJobCoordinator;
36 import org.opendaylight.vpnservice.itm.impl.ITMManager;
37 import org.opendaylight.vpnservice.itm.impl.ItmUtils;
38 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
39 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
40 import org.opendaylight.vpnservice.itm.confighelpers.ItmTepAddWorker ;
41 import org.opendaylight.vpnservice.itm.confighelpers.ItmTepRemoveWorker;
42 import org.slf4j.Logger;
43 import org.slf4j.LoggerFactory;
44
45 /**
46  * This class listens for interface creation/removal/update in Configuration DS.
47  * This is used to handle interfaces for base of-ports.
48  */
49 public class TransportZoneListener extends AsyncDataTreeChangeListenerBase<TransportZone, TransportZoneListener> implements AutoCloseable{
50     private static final Logger LOG = LoggerFactory.getLogger(TransportZoneListener.class);
51     private DataBroker dataBroker;
52     private ITMManager itmManager;
53
54     public TransportZoneListener(final DataBroker dataBroker) {
55         super(TransportZone.class, TransportZoneListener.class);
56         this.dataBroker = dataBroker;
57         initializeTZNode(dataBroker);
58     }
59
60     public void setItmManager(ITMManager itmManager) {
61         this.itmManager = itmManager;
62     }
63
64     private void initializeTZNode(DataBroker db) {
65         ReadWriteTransaction transaction = db.newReadWriteTransaction();
66         InstanceIdentifier<TransportZones> path = InstanceIdentifier.create(TransportZones.class);
67         CheckedFuture<Optional<TransportZones>, ReadFailedException> tzones =
68                         transaction.read(LogicalDatastoreType.CONFIGURATION,path);
69         try {
70             if (!tzones.get().isPresent()) {
71                 TransportZonesBuilder tzb = new TransportZonesBuilder();
72                 transaction.put(LogicalDatastoreType.CONFIGURATION,path,tzb.build());
73                 transaction.submit();
74             } else {
75                 transaction.cancel();
76             }
77         } catch (Exception e) {
78             LOG.error("Error initializing TransportZones {}",e);
79         }
80     }
81
82     @Override
83     public void close() throws Exception {
84         LOG.info("tzChangeListener Closed");
85     }
86     @Override
87     protected InstanceIdentifier<TransportZone> getWildCardPath() {
88         return InstanceIdentifier.create(TransportZones.class).child(TransportZone.class);
89     }
90
91     @Override
92     protected TransportZoneListener getDataTreeChangeListener() {
93         return TransportZoneListener.this;
94     }
95
96     @Override
97     protected void remove(InstanceIdentifier<TransportZone> key, TransportZone tzOld) {
98         LOG.debug("Received Transport Zone Remove Event: {}, {}", key, tzOld);
99         List<DPNTEPsInfo> opDpnList = createDPNTepInfo(tzOld);
100         LOG.trace("Delete: Invoking deleteTunnels in ItmManager with DpnList {}", opDpnList);
101         if(opDpnList.size()>0) {
102             LOG.trace("Delete: Invoking ItmManager");
103            // itmManager.deleteTunnels(opDpnList);
104             DataStoreJobCoordinator coordinator = DataStoreJobCoordinator.getInstance();
105             ItmTepRemoveWorker removeWorker = new ItmTepRemoveWorker(opDpnList, dataBroker);
106             coordinator.enqueueJob(tzOld.getZoneName(), removeWorker);
107         }
108     }
109
110     @Override
111     protected void update(InstanceIdentifier<TransportZone> key, TransportZone tzOld, TransportZone tzNew) {
112         LOG.debug("Received Transport Zone Update Event: {}, {}, {}", key, tzOld, tzNew);
113         if( !(tzOld.equals(tzNew))) {
114            add(key, tzNew);
115         }
116     }
117
118     @Override
119     protected void add(InstanceIdentifier<TransportZone> key, TransportZone tzNew) {
120         LOG.debug("Received Transport Zone Add Event: {}, {}", key, tzNew);
121         List<DPNTEPsInfo> opDpnList = createDPNTepInfo(tzNew);
122         LOG.trace("Add: Operational dpnTepInfo - Before invoking ItmManager {}", opDpnList);
123         if(opDpnList.size()>0) {
124           LOG.trace("Add: Invoking ItmManager with DPN List {} " , opDpnList);
125           //itmManager.build_all_tunnels(opDpnList);
126           DataStoreJobCoordinator coordinator = DataStoreJobCoordinator.getInstance();
127           ItmTepAddWorker addWorker = new ItmTepAddWorker(opDpnList,dataBroker);
128           coordinator.enqueueJob(tzNew.getZoneName(), addWorker);
129       }
130     }
131
132     private List<DPNTEPsInfo> createDPNTepInfo(TransportZone transportZone){
133
134         Map<BigInteger, List<TunnelEndPoints>> mapDPNToTunnelEndpt = new ConcurrentHashMap<>();
135         List<DPNTEPsInfo> dpnTepInfo = new ArrayList<DPNTEPsInfo>();
136        // List<TransportZone> transportZoneList = transportZones.getTransportZone();
137        // for(TransportZone transportZone : transportZoneList) {
138             String zone_name = transportZone.getZoneName();
139             String tunnel_type = transportZone.getTunnelType();
140             LOG.trace("Transport Zone_name: {}", zone_name);
141             List<Subnets> subnetsList = transportZone.getSubnets();
142             if(subnetsList!=null){
143                 for (Subnets subnet : subnetsList) {
144                     IpPrefix ipPrefix = subnet.getPrefix();
145                     IpAddress gatewayIP = subnet.getGatewayIp();
146                     int vlanID = subnet.getVlanId();
147                     LOG.trace("IpPrefix: {}, gatewayIP: {}, vlanID: {} ", ipPrefix, gatewayIP, vlanID);
148                     List<Vteps> vtepsList = subnet.getVteps();
149                     for (Vteps vteps : vtepsList) {
150                         BigInteger dpnID = vteps.getDpnId();
151                         String port = vteps.getPortname();
152                         IpAddress ipAddress = vteps.getIpAddress();
153                         LOG.trace("DpnID: {}, port: {}, ipAddress: {}", dpnID, port, ipAddress);
154                     TunnelEndPoints tunnelEndPoints = ItmUtils.createTunnelEndPoints(dpnID, ipAddress, port, vlanID, ipPrefix, gatewayIP, zone_name, tunnel_type);
155                         List<TunnelEndPoints> tunnelEndPointsList = mapDPNToTunnelEndpt.get(dpnID);
156                         if (tunnelEndPointsList != null) {
157                             LOG.trace("Existing DPN info list in the Map: {} ", dpnID);
158                             tunnelEndPointsList.add(tunnelEndPoints);
159                         } else {
160                             LOG.trace("Adding new DPN info list to the Map: {} ", dpnID);
161                             tunnelEndPointsList = new ArrayList<TunnelEndPoints>();
162                             tunnelEndPointsList.add(tunnelEndPoints);
163                             mapDPNToTunnelEndpt.put(dpnID, tunnelEndPointsList);
164                         }
165                     }
166                 }
167             }
168         //}
169         if(mapDPNToTunnelEndpt.size()>0){
170             Set<BigInteger> keys = mapDPNToTunnelEndpt.keySet();
171             LOG.trace("List of dpns in the Map: {} ", keys);
172             for(BigInteger key: keys){
173                 DPNTEPsInfo newDpnTepsInfo = ItmUtils.createDPNTepInfo(key, mapDPNToTunnelEndpt.get(key));
174                 dpnTepInfo.add(newDpnTepsInfo);
175             }
176         }
177         return dpnTepInfo;
178     }
179 }