neutron-ovsdb/src/main/java/org/opendaylight/groupbasedpolicy/neutron/ovsdb/NeutronGbpFloatingIpListener.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. 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.groupbasedpolicy.neutron.ovsdb;
  10
  11 import static com.google.common.base.Preconditions.checkNotNull;
  12
  13 import java.util.Map.Entry;
  14
  15 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  16 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  17 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  18 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
  19 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
  20 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
  21 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  22 import org.opendaylight.groupbasedpolicy.dto.EpKey;
  23 import org.opendaylight.groupbasedpolicy.neutron.ovsdb.util.NeutronOvsdbIidFactory;
  24 import org.opendaylight.groupbasedpolicy.util.DataStoreHelper;
  25 import org.opendaylight.groupbasedpolicy.util.IidFactory;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.L2BridgeDomainId;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoint.fields.L3Address;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.Endpoint;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointL3;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointL3Builder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointL3Key;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.l3endpoint.rev151217.NatAddress;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.l3endpoint.rev151217.NatAddressBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.neutron.gbp.mapper.rev150513.mappings.floating.ip.association.mappings.internal.ports.by.floating.ip.ports.InternalPortByFloatingIpPort;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.neutron.gbp.mapper.rev150513.mappings.gbp.by.neutron.mappings.endpoints.by.ports.EndpointByPort;
  37 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  38 import org.opendaylight.yangtools.yang.binding.DataObject;
  39 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  40 import org.slf4j.Logger;
  41 import org.slf4j.LoggerFactory;
  42
  43 import com.google.common.base.Optional;
  44
  45 public class NeutronGbpFloatingIpListener implements DataChangeListener, AutoCloseable {
  46
  47     private static final Logger LOG = LoggerFactory.getLogger(NeutronGbpFloatingIpListener.class);
  48     private final ListenerRegistration<DataChangeListener> gbpFloatingIpListener;
  49     private final DataBroker dataBroker;
  50
  51     public NeutronGbpFloatingIpListener(DataBroker dataBroker) {
  52         this.dataBroker = checkNotNull(dataBroker);
  53          gbpFloatingIpListener = dataBroker.registerDataChangeListener(LogicalDatastoreType.OPERATIONAL,
  54                 NeutronOvsdbIidFactory.neutronGbpInternalPortByFloatingIpIidWildcard(), this, DataChangeScope.BASE);
  55         LOG.trace("NeutronGbpFloatingIpListener started");
  56     }
  57
  58     @Override
  59     public void onDataChanged(AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
  60
  61         /*
  62          */
  63         for (Entry<InstanceIdentifier<?>, DataObject> entry : change.getCreatedData().entrySet()) {
  64             if (entry.getValue() instanceof InternalPortByFloatingIpPort) {
  65                 InternalPortByFloatingIpPort internalPortByFloatingIp = (InternalPortByFloatingIpPort) entry.getValue();
  66                 processInternalPortByFloatingIp(internalPortByFloatingIp);
  67             }
  68         }
  69
  70         /*
  71          * Updates
  72          */
  73         for (Entry<InstanceIdentifier<?>, DataObject> entry : change.getUpdatedData().entrySet()) {
  74             if (entry.getValue() instanceof InternalPortByFloatingIpPort) {
  75                 InternalPortByFloatingIpPort internalPortByFloatingIp = (InternalPortByFloatingIpPort) entry.getValue();
  76                 processInternalPortByFloatingIp(internalPortByFloatingIp);
  77             }
  78         }
  79
  80         /*
  81          * Deletions
  82          */
  83         for (InstanceIdentifier<?> iid : change.getRemovedPaths()) {
  84             /*
  85              * Remove ?
  86              */
  87         }
  88     }
  89
  90     private void processInternalPortByFloatingIp(InternalPortByFloatingIpPort internalPortByFloatingIp) {
  91         IpAddress natAddress = internalPortByFloatingIp.getFloatingIpPortIpAddress();
  92         IpAddress ipAddress = internalPortByFloatingIp.getInternalPortIpAddress();
  93
  94         ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction();
  95         Optional<EndpointByPort> optEndpointByPort = DataStoreHelper.readFromDs(
  96                 LogicalDatastoreType.OPERATIONAL,
  97                 NeutronOvsdbIidFactory.endpointByPortIid(internalPortByFloatingIp.getInternalPortId()), rTx);
  98         if (optEndpointByPort.isPresent()) {
  99             EpKey l2EpKey = new EpKey(optEndpointByPort.get().getL2Context(), optEndpointByPort.get().getMacAddress());
 100             updateEndpointNat(l2EpKey, ipAddress, natAddress);
 101         } else {
 102             LOG.error("processEpByFloatingIp: Couldn't find EP associated with {}.", internalPortByFloatingIp);
 103         }
 104     }
 105
 106     private void updateEndpointNat(EpKey l2EpKey, IpAddress ipAddress, IpAddress natAddress) {
 107         Endpoint l2Ep;
 108         EndpointL3 l3Ep;
 109         EndpointL3Key l3EpKey = null;
 110
 111         ReadWriteTransaction rwTx = dataBroker.newReadWriteTransaction();
 112
 113         Optional<Endpoint> optL2Ep = DataStoreHelper.readFromDs(LogicalDatastoreType.OPERATIONAL,
 114                 IidFactory.endpointIid((L2BridgeDomainId) l2EpKey.getL2Context(), l2EpKey.getMacAddress()), rwTx);
 115         if (optL2Ep.isPresent()) {
 116             l2Ep = optL2Ep.get();
 117         } else {
 118             LOG.error("updateEndpointNat: No Endpoint {} ", l2EpKey);
 119             return;
 120         }
 121
 122         if (l2Ep.getL3Address() == null) {
 123             LOG.error("updateEndpointNat: L2Ep {} had no IP address to translate to.", l2Ep);
 124             return;
 125         }
 126
 127         for (L3Address l3Address : l2Ep.getL3Address()) {
 128             if (l3Address.getIpAddress().equals(ipAddress)) {
 129                 l3EpKey = new EndpointL3Key(l3Address.getIpAddress(), l3Address.getL3Context());
 130                 break;
 131             }
 132         }
 133         if (l3EpKey == null) {
 134             return;
 135         }
 136         Optional<EndpointL3> optL3Ep = DataStoreHelper.readFromDs(LogicalDatastoreType.OPERATIONAL,
 137                 IidFactory.l3EndpointIid(l3EpKey.getL3Context(), l3EpKey.getIpAddress()), rwTx);
 138         if (optL3Ep.isPresent()) {
 139             l3Ep = optL3Ep.get();
 140             NatAddress nat = new NatAddressBuilder().setNatAddress(natAddress).build();
 141
 142             EndpointL3 updatedEpL3 = new EndpointL3Builder(l3Ep).addAugmentation(NatAddress.class, nat).build();
 143
 144             rwTx.put(LogicalDatastoreType.OPERATIONAL, IidFactory.l3EndpointIid(l3EpKey.getL3Context(), l3EpKey.getIpAddress()), updatedEpL3,true);
 145             boolean writeResult = DataStoreHelper.submitToDs(rwTx);
 146             if(!writeResult) {
 147                 LOG.trace("updateEndpointNat: Could not write {} to datastore.",updatedEpL3.getKey());
 148             }
 149         }
 150     }
 151
 152     @Override
 153     public void close() throws Exception {
 154         gbpFloatingIpListener.close();
 155     }
 156
 157 }