opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/DefaultShardDataTreeChangeListenerPublisher.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 package org.opendaylight.controller.cluster.datastore;
   9
  10 import com.google.common.base.Optional;
  11 import java.util.Collection;
  12 import java.util.function.Consumer;
  13 import javax.annotation.concurrent.NotThreadSafe;
  14 import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeChangeListener;
  15 import org.opendaylight.mdsal.dom.spi.AbstractDOMDataTreeChangeListenerRegistration;
  16 import org.opendaylight.mdsal.dom.spi.store.AbstractDOMStoreTreeChangePublisher;
  17 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  18 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  19 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidate;
  20 import org.slf4j.Logger;
  21 import org.slf4j.LoggerFactory;
  22
  23 /**
  24  * Default implementation of ShardDataTreeChangeListenerPublisher that directly generates and publishes
  25  * notifications for DataTreeChangeListeners.
  26  *
  27  * @author Thomas Pantelis
  28  */
  29 @NotThreadSafe
  30 final class DefaultShardDataTreeChangeListenerPublisher extends AbstractDOMStoreTreeChangePublisher
  31         implements ShardDataTreeChangeListenerPublisher {
  32     private static final Logger LOG = LoggerFactory.getLogger(DefaultShardDataTreeChangeListenerPublisher.class);
  33
  34     @Override
  35     public void publishChanges(final DataTreeCandidate candidate, String logContext) {
  36         processCandidateTree(candidate);
  37     }
  38
  39     @Override
  40     protected void notifyListener(AbstractDOMDataTreeChangeListenerRegistration<?> registration,
  41             Collection<DataTreeCandidate> changes) {
  42         registration.getInstance().onDataTreeChanged(changes);
  43     }
  44
  45     @Override
  46     protected void registrationRemoved(final AbstractDOMDataTreeChangeListenerRegistration<?> registration) {
  47         LOG.debug("Registration {} removed", registration);
  48     }
  49
  50     @Override
  51     public void registerTreeChangeListener(YangInstanceIdentifier treeId, DOMDataTreeChangeListener listener,
  52             Optional<DataTreeCandidate> initialState,
  53             Consumer<ListenerRegistration<DOMDataTreeChangeListener>> onRegistration) {
  54         AbstractDOMDataTreeChangeListenerRegistration<org.opendaylight.mdsal.dom.api.DOMDataTreeChangeListener>
  55             registration = super.registerTreeChangeListener(treeId,
  56                 (org.opendaylight.mdsal.dom.api.DOMDataTreeChangeListener)changes ->
  57                     listener.onDataTreeChanged(changes));
  58
  59         onRegistration.accept(
  60             new org.opendaylight.controller.md.sal.dom.spi.AbstractDOMDataTreeChangeListenerRegistration<
  61                     DOMDataTreeChangeListener>(listener) {
  62                 @Override
  63                 protected void removeRegistration() {
  64                     registration.close();
  65                 }
  66             });
  67
  68         if (initialState.isPresent()) {
  69             notifySingleListener(treeId, listener, initialState.get());
  70         }
  71     }
  72
  73     static void notifySingleListener(YangInstanceIdentifier treeId, DOMDataTreeChangeListener listener,
  74             DataTreeCandidate state) {
  75         DefaultShardDataTreeChangeListenerPublisher publisher = new DefaultShardDataTreeChangeListenerPublisher();
  76         publisher.registerTreeChangeListener(treeId, listener, Optional.absent(), noop -> { });
  77         publisher.publishChanges(state, "");
  78     }
  79 }