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.yangtools.concepts.ListenerRegistration;\r
  28 import org.opendaylight.yangtools.yang.binding.DataObject;\r
  29 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;\r
  30 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.PathArgument;\r
  31 import org.slf4j.Logger;\r
  32 import org.slf4j.LoggerFactory;\r
  33 \r
  34 import com.google.common.annotations.VisibleForTesting;\r
  35 import com.google.common.collect.Iterators;\r
  36 import com.google.common.collect.PeekingIterator;\r
  37 \r
  38 public class NeutronListener implements ClusteredDataTreeChangeListener<Neutron>, Closeable {\r
  39 \r
  40     private static final Logger LOG = LoggerFactory.getLogger(NeutronListener.class);\r
  41 \r
  42     private final Set<MappingProvider<? extends DataObject>> dataChangeProviders = new LinkedHashSet<>();\r
  43     protected ListenerRegistration<NeutronListener> registeredListener;\r
  44 \r
  45     public NeutronListener(DataBroker dataBroker) {\r
  46         registerHandlersAndListeners(dataBroker);\r
  47         registeredListener = dataBroker.registerDataTreeChangeListener(new DataTreeIdentifier<>(\r
  48                 LogicalDatastoreType.CONFIGURATION, InstanceIdentifier.builder(Neutron.class).build()), this);\r
  49     }\r
  50 \r
  51     private void registerHandlersAndListeners(DataBroker dataBroker) {\r
  52         PortHandler portHandler = new PortHandler(dataBroker);\r
  53         dataChangeProviders.add(new PortAware(portHandler, dataBroker));\r
  54         dataChangeProviders.add(new NetworkAware(dataBroker));\r
  55     }\r
  56 \r
  57     @Override\r
  58     public void onDataTreeChanged(Collection<DataTreeModification<Neutron>> changes) {\r
  59         for (DataTreeModification<Neutron> change : changes) {\r
  60             DataObjectModification<Neutron> rootNode = change.getRootNode();\r
  61             for (MappingProvider<? extends DataObject> provider : dataChangeProviders) {\r
  62                 for (DataObjectModification<? extends DataObject> modDto : findModifiedData(provider, rootNode)) {\r
  63                     try {\r
  64                         processChangedData(modDto, modDto.getModificationType(), provider);\r
  65                     } catch (InterruptedException | ExecutionException e) {\r
  66                         LOG.error("Failed to process {} modification of node: {}. {}", modDto.getModificationType(),\r
  67                                 modDto.getIdentifier(), e.getStackTrace());\r
  68                     }\r
  69                 }\r
  70             }\r
  71         }\r
  72     }\r
  73 \r
  74     List<DataObjectModification<? extends DataObject>> findModifiedData(MappingProvider<? extends DataObject> provider,\r
  75             DataObjectModification<Neutron> rootNode) {\r
  76         List<DataObjectModification<? extends DataObject>> modDtos = new ArrayList<>();\r
  77         PeekingIterator<PathArgument> pathArgs = Iterators.peekingIterator(provider.getNeutronDtoIid()\r
  78             .getPathArguments()\r
  79             .iterator());\r
  80         DataObjectModification<? extends DataObject> modifDto = rootNode;\r
  81         while (pathArgs.hasNext()) {\r
  82             pathArgs.next();\r
  83             for (DataObjectModification<? extends DataObject> childDto : modifDto.getModifiedChildren()) {\r
  84                 if (pathArgs.hasNext() && childDto.getDataType().equals(pathArgs.peek().getType())) {\r
  85                     if (childDto.getDataType().equals(provider.getNeutronDtoIid().getTargetType())) {\r
  86                         modDtos.add(childDto);\r
  87                     } else {\r
  88                         modifDto = childDto;\r
  89                         break;\r
  90                     }\r
  91                 }\r
  92             }\r
  93         }\r
  94         return modDtos;\r
  95     }\r
  96 \r
  97     @SuppressWarnings("unchecked")\r
  98     <T extends DataObject> void processChangedData(DataObjectModification<?> dto, ModificationType m,\r
  99             MappingProvider<T> processor) throws InterruptedException, ExecutionException {\r
 100         switch (m) {\r
 101             case WRITE: {\r
 102                 if (dto.getDataBefore() != null) {\r
 103                     processor.processUpdatedNeutronDto((T) dto.getDataBefore(), (T) dto.getDataAfter());\r
 104                 } else {\r
 105                     processor.processCreatedNeutronDto((T) dto.getDataAfter());\r
 106                 }\r
 107                 break;\r
 108             }\r
 109             case SUBTREE_MODIFIED: {\r
 110                 processor.processUpdatedNeutronDto((T) dto.getDataBefore(), (T) dto.getDataAfter());\r
 111                 break;\r
 112             }\r
 113             case DELETE: {\r
 114                 processor.processDeletedNeutronDto((T) dto.getDataBefore());\r
 115                 break;\r
 116             }\r
 117         }\r
 118     }\r
 119 \r
 120     @VisibleForTesting\r
 121     void clearDataChangeProviders() {\r
 122         dataChangeProviders.clear();\r
 123     }\r
 124 \r
 125     @VisibleForTesting\r
 126     <T extends DataObject> void addDataChangeProvider(MappingProvider<T> t) {\r
 127         dataChangeProviders.add(t);\r
 128     }\r
 129 \r
 130     @Override\r
 131     public void close() {\r
 132         registeredListener.close();\r
 133     }\r
 134 }\r