1 package org.opendaylight.controller.md.sal.common.impl.routing;
3 import java.util.Map.Entry;
5 import org.opendaylight.controller.md.sal.common.api.data.DataReader;
6 import org.opendaylight.yangtools.concepts.AbstractObjectRegistration;
7 import org.opendaylight.yangtools.concepts.Path;
8 import org.opendaylight.yangtools.concepts.Registration;
10 import com.google.common.base.Function;
11 import com.google.common.base.Predicate;
12 import com.google.common.collect.FluentIterable;
13 import com.google.common.collect.HashMultimap;
14 import com.google.common.collect.Multimap;
17 * Base abstract implementation of DataReadRouter, which performs
18 * a read operation on multiple data readers and then merges result.
23 public abstract class AbstractDataReadRouter<P extends Path<P>, D> implements DataReader<P, D> {
25 Multimap<P, DataReaderRegistration<P, D>> configReaders = HashMultimap.create();
26 Multimap<P, DataReaderRegistration<P, D>> operationalReaders = HashMultimap.create();
29 public D readConfigurationData(P path) {
30 FluentIterable<D> dataBits = FluentIterable //
31 .from(getReaders(configReaders, path)).transform(configurationRead(path));
32 return merge(path,dataBits);
36 public D readOperationalData(P path) {
37 FluentIterable<D> dataBits = FluentIterable //
38 .from(getReaders(operationalReaders, path)).transform(operationalRead(path));
39 return merge(path,dataBits);
44 * Merges data readed by reader instances from specified path
46 * @param path Path on which read was performed
47 * @param data Data which was returned by read operation.
48 * @return Merged result.
50 protected abstract D merge(P path,Iterable<D> data);
53 * Returns a function which performs configuration read for supplied path
56 * @return function which performs configuration read for supplied path
59 private Function<DataReader<P, D>, D> configurationRead(final P path) {
60 return new Function<DataReader<P, D>, D>() {
62 public D apply(DataReader<P, D> input) {
63 return input.readConfigurationData(path);
69 * Returns a function which performs operational read for supplied path
72 * @return function which performs operational read for supplied path
74 private Function<DataReader<P, D>, D> operationalRead(final P path) {
75 return new Function<DataReader<P, D>, D>() {
77 public D apply(DataReader<P, D> input) {
78 return input.readOperationalData(path);
86 * Register's a reader for operational data.
88 * @param path Path which is served by this reader
89 * @param reader Reader instance which is responsible for reading particular subpath.
92 public Registration<DataReader<P, D>> registerOperationalReader(P path, DataReader<P, D> reader) {
93 OperationalDataReaderRegistration<P, D> ret = new OperationalDataReaderRegistration<>(path, reader);
94 operationalReaders.put(path, ret);
98 public Registration<DataReader<P, D>> registerConfigurationReader(P path, DataReader<P, D> reader) {
99 ConfigurationDataReaderRegistration<P, D> ret = new ConfigurationDataReaderRegistration<>(path, reader);
100 configReaders.put(path, ret);
104 Iterable<DataReader<P, D>> getOperationalReaders(P path) {
105 return getReaders(operationalReaders, path);
108 Iterable<DataReader<P, D>> getConfigurationReaders(P path) {
109 return getReaders(configReaders, path);
112 private Iterable<DataReader<P, D>> getReaders(Multimap<P, DataReaderRegistration<P, D>> readerMap, P path) {
113 return FluentIterable
114 .from(readerMap.entries()) //
115 .filter(affects(path)) //
116 .transform(retrieveInstance());
119 private void removeRegistration(OperationalDataReaderRegistration<?, ?> registration) {
120 operationalReaders.remove(registration.getKey(), registration);
123 private void removeRegistration(ConfigurationDataReaderRegistration<?, ?> registration) {
124 configReaders.remove(registration.getKey(), registration);
127 private Function<? super Entry<P, DataReaderRegistration<P, D>>, DataReader<P, D>> retrieveInstance() {
128 return new Function<Entry<P, DataReaderRegistration<P, D>>, DataReader<P,D>>() {
130 public DataReader<P, D> apply(Entry<P, DataReaderRegistration<P, D>> input) {
131 return input.getValue().getInstance();
136 private Predicate<? super Entry<P, DataReaderRegistration<P, D>>> affects(final P path) {
138 return new Predicate<Entry<P, DataReaderRegistration<P, D>>>() {
141 public boolean apply(Entry<P, DataReaderRegistration<P, D>> input) {
142 final P key = input.getKey();
143 return key.contains(path) || ((P) path).contains(key);
149 @SuppressWarnings("hiding")
150 private class ConfigurationDataReaderRegistration<P extends Path<P>, D> extends DataReaderRegistration<P, D> {
152 public ConfigurationDataReaderRegistration(P key, DataReader<P, D> instance) {
153 super(key, instance);
157 protected void removeRegistration() {
158 AbstractDataReadRouter.this.removeRegistration(this);
162 @SuppressWarnings("hiding")
163 private class OperationalDataReaderRegistration<P extends Path<P>, D> extends DataReaderRegistration<P, D> {
165 public OperationalDataReaderRegistration(P key, DataReader<P, D> instance) {
166 super(key, instance);
170 protected void removeRegistration() {
171 AbstractDataReadRouter.this.removeRegistration(this);
175 private abstract static class DataReaderRegistration<P extends Path<P>, D> extends
176 AbstractObjectRegistration<DataReader<P, D>> {
184 public DataReaderRegistration(P key, DataReader<P, D> instance) {