neutron-vpp-mapper/src/main/java/org/opendaylight/groupbasedpolicy/neutron/vpp/mapper/processors/NeutronListener.java

   1 /*\r
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.\r
   3  *\r
   4  * This program and the accompanying materials are made available under the\r
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,\r
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html\r
   7  */\r
   8 \r
   9 package org.opendaylight.groupbasedpolicy.neutron.vpp.mapper.processors;\r
  10 \r
  11 import java.io.Closeable;\r
  12 import java.util.ArrayList;\r
  13 import java.util.Collection;\r
  14 import java.util.LinkedHashSet;\r
  15 import java.util.List;\r
  16 import java.util.Set;\r
  17 import java.util.concurrent.ExecutionException;\r
  18 \r
  19 import org.opendaylight.controller.md.sal.binding.api.ClusteredDataTreeChangeListener;\r
  20 import org.opendaylight.controller.md.sal.binding.api.DataBroker;\r
  21 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;\r
  22 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification.ModificationType;\r
  23 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;\r
  24 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;\r
  25 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;\r
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.Neutron;\r
  27 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;\r
  28 import org.opendaylight.yangtools.concepts.ListenerRegistration;\r
  29 import org.opendaylight.yangtools.yang.binding.DataObject;\r
  30 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;\r
  31 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.PathArgument;\r
  32 import org.slf4j.Logger;\r
  33 import org.slf4j.LoggerFactory;\r
  34 \r
  35 import com.google.common.annotations.VisibleForTesting;\r
  36 import com.google.common.collect.Iterators;\r
  37 import com.google.common.collect.PeekingIterator;\r
  38 \r
  39 public class NeutronListener implements ClusteredDataTreeChangeListener<Neutron>, Closeable {\r
  40 \r
  41     private static final Logger LOG = LoggerFactory.getLogger(NeutronListener.class);\r
  42 \r
  43     private final Set<MappingProvider<? extends DataObject>> dataChangeProviders = new LinkedHashSet<>();\r
  44     protected ListenerRegistration<NeutronListener> registeredListener;\r
  45 \r
  46     public NeutronListener(DataBroker dataBroker, NodeId routingNode) {\r
  47         LOG.info("Routing node chosen in ODL is {}", routingNode);\r
  48         registerHandlersAndListeners(dataBroker, routingNode);\r
  49         registeredListener = dataBroker.registerDataTreeChangeListener(new DataTreeIdentifier<>(\r
  50                 LogicalDatastoreType.CONFIGURATION, InstanceIdentifier.builder(Neutron.class).build()), this);\r
  51     }\r
  52 \r
  53     private void registerHandlersAndListeners(DataBroker dataBroker, NodeId routingNode) {\r
  54         PortHandler portHandler = new PortHandler(dataBroker, routingNode);\r
  55         dataChangeProviders.add(new PortAware(portHandler, dataBroker));\r
  56         dataChangeProviders.add(new NetworkAware(dataBroker));\r
  57     }\r
  58 \r
  59     @Override\r
  60     public void onDataTreeChanged(Collection<DataTreeModification<Neutron>> changes) {\r
  61         for (DataTreeModification<Neutron> change : changes) {\r
  62             DataObjectModification<Neutron> rootNode = change.getRootNode();\r
  63             for (MappingProvider<? extends DataObject> provider : dataChangeProviders) {\r
  64                 for (DataObjectModification<? extends DataObject> modDto : findModifiedData(provider, rootNode)) {\r
  65                     try {\r
  66                         processChangedData(modDto, modDto.getModificationType(), provider);\r
  67                     } catch (InterruptedException | ExecutionException e) {\r
  68                         LOG.error("Failed to process {} modification of node: {}. {}", modDto.getModificationType(),\r
  69                                 modDto.getIdentifier(), e.getStackTrace());\r
  70                     }\r
  71                 }\r
  72             }\r
  73         }\r
  74     }\r
  75 \r
  76     List<DataObjectModification<? extends DataObject>> findModifiedData(MappingProvider<? extends DataObject> provider,\r
  77             DataObjectModification<Neutron> rootNode) {\r
  78         List<DataObjectModification<? extends DataObject>> modDtos = new ArrayList<>();\r
  79         PeekingIterator<PathArgument> pathArgs = Iterators.peekingIterator(provider.getNeutronDtoIid()\r
  80             .getPathArguments()\r
  81             .iterator());\r
  82         DataObjectModification<? extends DataObject> modifDto = rootNode;\r
  83         while (pathArgs.hasNext()) {\r
  84             pathArgs.next();\r
  85             for (DataObjectModification<? extends DataObject> childDto : modifDto.getModifiedChildren()) {\r
  86                 if (pathArgs.hasNext() && childDto.getDataType().equals(pathArgs.peek().getType())) {\r
  87                     if (childDto.getDataType().equals(provider.getNeutronDtoIid().getTargetType())) {\r
  88                         modDtos.add(childDto);\r
  89                     } else {\r
  90                         modifDto = childDto;\r
  91                         break;\r
  92                     }\r
  93                 }\r
  94             }\r
  95         }\r
  96         return modDtos;\r
  97     }\r
  98 \r
  99     @SuppressWarnings("unchecked")\r
 100     <T extends DataObject> void processChangedData(DataObjectModification<?> dto, ModificationType m,\r
 101             MappingProvider<T> processor) throws InterruptedException, ExecutionException {\r
 102         switch (m) {\r
 103             case WRITE: {\r
 104                 if (dto.getDataBefore() != null) {\r
 105                     processor.processUpdatedNeutronDto((T) dto.getDataBefore(), (T) dto.getDataAfter());\r
 106                 } else {\r
 107                     processor.processCreatedNeutronDto((T) dto.getDataAfter());\r
 108                 }\r
 109                 break;\r
 110             }\r
 111             case SUBTREE_MODIFIED: {\r
 112                 processor.processUpdatedNeutronDto((T) dto.getDataBefore(), (T) dto.getDataAfter());\r
 113                 break;\r
 114             }\r
 115             case DELETE: {\r
 116                 processor.processDeletedNeutronDto((T) dto.getDataBefore());\r
 117                 break;\r
 118             }\r
 119         }\r
 120     }\r
 121 \r
 122     @VisibleForTesting\r
 123     void clearDataChangeProviders() {\r
 124         dataChangeProviders.clear();\r
 125     }\r
 126 \r
 127     @VisibleForTesting\r
 128     <T extends DataObject> void addDataChangeProvider(MappingProvider<T> t) {\r
 129         dataChangeProviders.add(t);\r
 130     }\r
 131 \r
 132     @Override\r
 133     public void close() {\r
 134         registeredListener.close();\r
 135     }\r
 136 }\r