Adjust DOMMountPoint to be EffectiveModelContextProvider
[mdsal.git] / dom / mdsal-dom-broker / src / main / java / org / opendaylight / mdsal / dom / broker / ShardedDOMReadTransactionAdapter.java
index 82371752e5d501b63cba4f951d679ccb29617ffe..fb3f6962d0cf1cb8859aef4a8b309b370fe7b4c7 100644 (file)
@@ -5,26 +5,22 @@
  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
  * and is available at http://www.eclipse.org/legal/epl-v10.html
  */
-
 package org.opendaylight.mdsal.dom.broker;
 
-import com.google.common.base.Function;
-import com.google.common.base.Optional;
-import com.google.common.base.Preconditions;
-import com.google.common.collect.Lists;
-import com.google.common.util.concurrent.CheckedFuture;
-import com.google.common.util.concurrent.FutureCallback;
-import com.google.common.util.concurrent.Futures;
-import com.google.common.util.concurrent.ListenableFuture;
+import static com.google.common.base.Preconditions.checkState;
+import static java.util.Objects.requireNonNull;
+
+import com.google.common.util.concurrent.FluentFuture;
 import com.google.common.util.concurrent.MoreExecutors;
 import com.google.common.util.concurrent.SettableFuture;
+import java.util.ArrayList;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;
-import javax.annotation.Nullable;
+import java.util.Optional;
+import org.eclipse.jdt.annotation.NonNull;
 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
-import org.opendaylight.mdsal.common.api.ReadFailedException;
 import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
 import org.opendaylight.mdsal.dom.api.DOMDataTreeListener;
 import org.opendaylight.mdsal.dom.api.DOMDataTreeListeningException;
@@ -40,29 +36,28 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 public class ShardedDOMReadTransactionAdapter implements DOMDataTreeReadTransaction {
-
     private static final Logger LOG = LoggerFactory.getLogger(ShardedDOMReadTransactionAdapter.class.getName());
 
-    private final List<ListenerRegistration<DOMDataTreeListener>> registrations = Lists.newArrayList();
-    private final DOMDataTreeService service;
-    private final Object txIdentifier;
+    private final List<ListenerRegistration<DOMDataTreeListener>> registrations = new ArrayList<>();
+    private final @NonNull DOMDataTreeService service;
+    private final @NonNull Object txIdentifier;
 
     private boolean finished = false;
 
     ShardedDOMReadTransactionAdapter(final Object identifier, final DOMDataTreeService service) {
-        this.service = Preconditions.checkNotNull(service);
-        this.txIdentifier = Preconditions.checkNotNull(identifier);
+        this.service = requireNonNull(service);
+        this.txIdentifier = requireNonNull(identifier);
     }
 
     @Override
     public void close() {
-        // TODO should we also cancel all read futures?
         LOG.debug("{}: Closing read transaction", txIdentifier);
         if (finished) {
             return;
         }
 
         registrations.forEach(ListenerRegistration::close);
+        // TODO should we also cancel all read futures?
         finished = true;
     }
 
@@ -72,7 +67,7 @@ public class ShardedDOMReadTransactionAdapter implements DOMDataTreeReadTransact
     }
 
     @Override
-    public CheckedFuture<Optional<NormalizedNode<?, ?>>, ReadFailedException> read(final LogicalDatastoreType store,
+    public FluentFuture<Optional<NormalizedNode<?, ?>>> read(final LogicalDatastoreType store,
             final YangInstanceIdentifier path) {
         checkRunning();
         LOG.debug("{}: Invoking read at {}:{}", txIdentifier, store, path);
@@ -83,73 +78,48 @@ public class ShardedDOMReadTransactionAdapter implements DOMDataTreeReadTransact
                     Collections.singleton(new DOMDataTreeIdentifier(store, path)), false, Collections.emptyList());
             registrations.add(reg);
         } catch (final DOMDataTreeLoopException e) {
-            // This should not happen, we are not specifying any
-            // producers when registering listener
+            // This should not happen, we are not specifying any producers when registering listener
             throw new IllegalStateException("Loop in listener and producers detected", e);
         }
 
         // After data tree change future is finished, we can close the listener registration
-        Futures.addCallback(initialDataTreeChangeFuture, new FutureCallback<Optional<NormalizedNode<?, ?>>>() {
-            @Override
-            public void onSuccess(@Nullable final Optional<NormalizedNode<?, ?>> result) {
-                reg.close();
-            }
-
-            @Override
-            public void onFailure(final Throwable throwable) {
-                reg.close();
-            }
-        }, MoreExecutors.directExecutor());
-
-        return Futures.makeChecked(initialDataTreeChangeFuture, ReadFailedException.MAPPER);
+        initialDataTreeChangeFuture.addListener(reg::close, MoreExecutors.directExecutor());
+        return FluentFuture.from(initialDataTreeChangeFuture);
     }
 
     @Override
-    public CheckedFuture<Boolean, ReadFailedException> exists(final LogicalDatastoreType store,
-            final YangInstanceIdentifier path) {
+    public FluentFuture<Boolean> exists(final LogicalDatastoreType store, final YangInstanceIdentifier path) {
         checkRunning();
         LOG.debug("{}: Invoking exists at {}:{}", txIdentifier, store, path);
-        final Function<Optional<NormalizedNode<?, ?>>, Boolean> transform =
-            optionalNode -> optionalNode.isPresent() ? Boolean.TRUE : Boolean.FALSE;
-        final ListenableFuture<Boolean> existsResult = Futures.transform(read(store, path), transform,
-            MoreExecutors.directExecutor());
-        return Futures.makeChecked(existsResult, ReadFailedException.MAPPER);
+        return read(store, path).transform(Optional::isPresent, MoreExecutors.directExecutor());
     }
 
     private void checkRunning() {
-        Preconditions.checkState(!finished, "Transaction is already closed");
+        checkState(!finished, "Transaction is already closed");
     }
 
-    static class ReadShardedListener implements DOMDataTreeListener {
-
+    static final class ReadShardedListener implements DOMDataTreeListener {
         private final SettableFuture<Optional<NormalizedNode<?, ?>>> readResultFuture;
 
         ReadShardedListener(final SettableFuture<Optional<NormalizedNode<?, ?>>> future) {
-            this.readResultFuture = Preconditions.checkNotNull(future);
+            this.readResultFuture = requireNonNull(future);
         }
 
         @Override
         public void onDataTreeChanged(final Collection<DataTreeCandidate> changes,
                 final Map<DOMDataTreeIdentifier, NormalizedNode<?, ?>> subtrees) {
-            Preconditions.checkState(changes.size() == 1 && subtrees.size() == 1,
+            checkState(changes.size() == 1 && subtrees.size() == 1,
                     "DOMDataTreeListener registered exactly on one subtree");
-
-            for (final DataTreeCandidate change : changes) {
-                if (change.getRootNode().getModificationType().equals(ModificationType.UNMODIFIED)) {
-                    readResultFuture.set(Optional.absent());
-                    return;
-                }
-            }
-
-            for (final NormalizedNode<?, ?> initialState : subtrees.values()) {
-                readResultFuture.set(Optional.of(initialState));
+            if (changes.iterator().next().getRootNode().getModificationType().equals(ModificationType.UNMODIFIED)) {
+                readResultFuture.set(Optional.empty());
+            } else {
+                readResultFuture.set(Optional.of(subtrees.values().iterator().next()));
             }
         }
 
         @Override
         public void onDataTreeFailed(final Collection<DOMDataTreeListeningException> causes) {
-            // TODO If we get just one exception, we don't need to do
-            // chaining
+            // TODO If we get just one exception, we don't need to do chaining
 
             // We chain all exceptions and return aggregated one
             readResultFuture.setException(new DOMDataTreeListeningException("Aggregated DOMDataTreeListening exception",