Refactor Register*ListenerReply classes
[controller.git] / 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     private String logContext;
34
35     DefaultShardDataTreeChangeListenerPublisher(String logContext) {
36         this.logContext = logContext;
37     }
38
39     @Override
40     public void publishChanges(final DataTreeCandidate candidate) {
41         LOG.debug("{}: publishChanges: {}", logContext, candidate);
42         processCandidateTree(candidate);
43     }
44
45     @Override
46     protected void notifyListener(AbstractDOMDataTreeChangeListenerRegistration<?> registration,
47             Collection<DataTreeCandidate> changes) {
48         LOG.debug("{}: notifyListener: listener: {}", logContext, registration.getInstance());
49         registration.getInstance().onDataTreeChanged(changes);
50     }
51
52     @Override
53     protected void registrationRemoved(final AbstractDOMDataTreeChangeListenerRegistration<?> registration) {
54         LOG.debug("Registration {} removed", registration);
55     }
56
57     @Override
58     public void registerTreeChangeListener(YangInstanceIdentifier treeId, DOMDataTreeChangeListener listener,
59             Optional<DataTreeCandidate> initialState,
60             Consumer<ListenerRegistration<DOMDataTreeChangeListener>> onRegistration) {
61         LOG.debug("{}: registerTreeChangeListener: path: {}, listener: {}", logContext, treeId, listener);
62
63         AbstractDOMDataTreeChangeListenerRegistration<org.opendaylight.mdsal.dom.api.DOMDataTreeChangeListener>
64             registration = super.registerTreeChangeListener(treeId, new ForwardingDOMDataTreeChangeListener(listener));
65
66         onRegistration.accept(
67             new org.opendaylight.controller.md.sal.dom.spi.AbstractDOMDataTreeChangeListenerRegistration<
68                     DOMDataTreeChangeListener>(listener) {
69                 @Override
70                 protected void removeRegistration() {
71                     registration.close();
72                 }
73             });
74
75         if (initialState.isPresent()) {
76             notifySingleListener(treeId, listener, initialState.get(), logContext);
77         }
78     }
79
80     static void notifySingleListener(YangInstanceIdentifier treeId, DOMDataTreeChangeListener listener,
81             DataTreeCandidate state, String logContext) {
82         LOG.debug("{}: notifySingleListener: path: {}, listener: {}", logContext, treeId, listener);
83         DefaultShardDataTreeChangeListenerPublisher publisher =
84                 new DefaultShardDataTreeChangeListenerPublisher(logContext);
85         publisher.logContext = logContext;
86         publisher.registerTreeChangeListener(treeId, listener, Optional.absent(), noop -> { });
87         publisher.publishChanges(state);
88     }
89
90     private static class ForwardingDOMDataTreeChangeListener
91             implements org.opendaylight.mdsal.dom.api.DOMDataTreeChangeListener {
92         final DOMDataTreeChangeListener delegate;
93
94         ForwardingDOMDataTreeChangeListener(DOMDataTreeChangeListener delegate) {
95             this.delegate = delegate;
96         }
97
98         @Override
99         public void onDataTreeChanged(Collection<DataTreeCandidate> changes) {
100             delegate.onDataTreeChanged(changes);
101         }
102
103         @Override
104         public String toString() {
105             return delegate.toString();
106         }
107     }
108 }