AtomixDocumentTree support for filtering notifications by DocumentPath
Change-Id: I3f4f616bc4f2e488e5433e44f72bcd121b564b0d
diff --git a/core/store/primitives/src/main/java/org/onosproject/store/primitives/resources/impl/AtomixDocumentTree.java b/core/store/primitives/src/main/java/org/onosproject/store/primitives/resources/impl/AtomixDocumentTree.java
index 8f0c73a..77f9b98 100644
--- a/core/store/primitives/src/main/java/org/onosproject/store/primitives/resources/impl/AtomixDocumentTree.java
+++ b/core/store/primitives/src/main/java/org/onosproject/store/primitives/resources/impl/AtomixDocumentTree.java
@@ -33,11 +33,11 @@
import org.onlab.util.Match;
import org.onlab.util.Tools;
-import org.onosproject.store.primitives.resources.impl.AtomixConsistentMapCommands.Unlisten;
import org.onosproject.store.primitives.resources.impl.AtomixDocumentTreeCommands.Clear;
import org.onosproject.store.primitives.resources.impl.AtomixDocumentTreeCommands.Get;
import org.onosproject.store.primitives.resources.impl.AtomixDocumentTreeCommands.GetChildren;
import org.onosproject.store.primitives.resources.impl.AtomixDocumentTreeCommands.Listen;
+import org.onosproject.store.primitives.resources.impl.AtomixDocumentTreeCommands.Unlisten;
import org.onosproject.store.primitives.resources.impl.AtomixDocumentTreeCommands.Update;
import org.onosproject.store.service.AsyncDocumentTree;
import org.onosproject.store.service.DocumentPath;
@@ -56,7 +56,7 @@
public class AtomixDocumentTree extends AbstractResource<AtomixDocumentTree>
implements AsyncDocumentTree<byte[]> {
- private final Map<DocumentTreeListener<byte[]>, Executor> eventListeners = new HashMap<>();
+ private final Map<DocumentTreeListener<byte[]>, InternalListener> eventListeners = new HashMap<>();
public static final String CHANGE_SUBJECT = "changeEvents";
protected AtomixDocumentTree(CopycatClient client, Properties options) {
@@ -184,21 +184,21 @@
public CompletableFuture<Void> addListener(DocumentPath path, DocumentTreeListener<byte[]> listener) {
checkNotNull(path);
checkNotNull(listener);
+ InternalListener internalListener = new InternalListener(path, listener, MoreExecutors.directExecutor());
// TODO: Support API that takes an executor
- if (isListening()) {
- eventListeners.putIfAbsent(listener, MoreExecutors.directExecutor());
- return CompletableFuture.completedFuture(null);
- } else {
+ if (!eventListeners.containsKey(listener)) {
return client.submit(new Listen(path))
- .thenRun(() -> eventListeners.put(listener, MoreExecutors.directExecutor()));
+ .thenRun(() -> eventListeners.put(listener, internalListener));
}
+ return CompletableFuture.completedFuture(null);
}
@Override
public CompletableFuture<Void> removeListener(DocumentTreeListener<byte[]> listener) {
checkNotNull(listener);
- if (eventListeners.remove(listener) != null && eventListeners.isEmpty()) {
- return client.submit(new Unlisten()).thenApply(v -> null);
+ InternalListener internalListener = eventListeners.remove(listener);
+ if (internalListener != null && eventListeners.isEmpty()) {
+ return client.submit(new Unlisten(internalListener.path)).thenApply(v -> null);
}
return CompletableFuture.completedFuture(null);
}
@@ -213,7 +213,26 @@
}
private void processTreeUpdates(List<DocumentTreeEvent<byte[]>> events) {
- events.forEach(event ->
- eventListeners.forEach((listener, executor) -> executor.execute(() -> listener.event(event))));
+ events.forEach(event -> eventListeners.values().forEach(listener -> listener.event(event)));
+ }
+
+ private class InternalListener implements DocumentTreeListener<byte[]> {
+
+ private final DocumentPath path;
+ private final DocumentTreeListener<byte[]> listener;
+ private final Executor executor;
+
+ public InternalListener(DocumentPath path, DocumentTreeListener<byte[]> listener, Executor executor) {
+ this.path = path;
+ this.listener = listener;
+ this.executor = executor;
+ }
+
+ @Override
+ public void event(DocumentTreeEvent<byte[]> event) {
+ if (event.path().isDescendentOf(path)) {
+ executor.execute(() -> listener.event(event));
+ }
+ }
}
}
diff --git a/core/store/primitives/src/main/java/org/onosproject/store/primitives/resources/impl/AtomixDocumentTreeCommands.java b/core/store/primitives/src/main/java/org/onosproject/store/primitives/resources/impl/AtomixDocumentTreeCommands.java
index a6b406c..f9f7bb1 100644
--- a/core/store/primitives/src/main/java/org/onosproject/store/primitives/resources/impl/AtomixDocumentTreeCommands.java
+++ b/core/store/primitives/src/main/java/org/onosproject/store/primitives/resources/impl/AtomixDocumentTreeCommands.java
@@ -225,11 +225,10 @@
}
@Override
- public void writeObject(BufferOutput<?> buffer, Serializer serializer) {
- }
-
- @Override
- public void readObject(BufferInput<?> buffer, Serializer serializer) {
+ public String toString() {
+ return MoreObjects.toStringHelper(getClass())
+ .add("path", path())
+ .toString();
}
}
@@ -248,11 +247,10 @@
}
@Override
- public void writeObject(BufferOutput<?> buffer, Serializer serializer) {
- }
-
- @Override
- public void readObject(BufferInput<?> buffer, Serializer serializer) {
+ public String toString() {
+ return MoreObjects.toStringHelper(getClass())
+ .add("path", path())
+ .toString();
}
}
diff --git a/core/store/primitives/src/main/java/org/onosproject/store/primitives/resources/impl/AtomixDocumentTreeState.java b/core/store/primitives/src/main/java/org/onosproject/store/primitives/resources/impl/AtomixDocumentTreeState.java
index cd725d9..06d33b5 100644
--- a/core/store/primitives/src/main/java/org/onosproject/store/primitives/resources/impl/AtomixDocumentTreeState.java
+++ b/core/store/primitives/src/main/java/org/onosproject/store/primitives/resources/impl/AtomixDocumentTreeState.java
@@ -26,7 +26,10 @@
import io.atomix.copycat.server.storage.snapshot.SnapshotWriter;
import io.atomix.resource.ResourceStateMachine;
+import java.util.Arrays;
import java.util.HashMap;
+import java.util.Iterator;
+import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Properties;
@@ -52,7 +55,7 @@
import org.slf4j.Logger;
import com.google.common.base.Throwables;
-import com.google.common.collect.ImmutableList;
+import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.collect.Queues;
@@ -64,7 +67,7 @@
implements SessionListener, Snapshottable {
private final Logger log = getLogger(getClass());
- private final Map<Long, Commit<? extends Listen>> listeners = new HashMap<>();
+ private final Map<Long, SessionListenCommits> listeners = new HashMap<>();
private AtomicLong versionCounter = new AtomicLong(0);
private final DocumentTree<TreeNodeValue> docTree = new DefaultDocumentTree<>(versionCounter::incrementAndGet);
@@ -97,25 +100,23 @@
protected void listen(Commit<? extends Listen> commit) {
Long sessionId = commit.session().id();
- if (listeners.putIfAbsent(sessionId, commit) != null) {
- commit.close();
- return;
- }
+ listeners.computeIfAbsent(sessionId, k -> new SessionListenCommits()).add(commit);
commit.session().onStateChange(
state -> {
if (state == ServerSession.State.CLOSED
|| state == ServerSession.State.EXPIRED) {
- Commit<? extends Listen> listener = listeners.remove(sessionId);
- if (listener != null) {
- listener.close();
- }
+ closeListener(commit.session().id());
}
});
}
protected void unlisten(Commit<? extends Unlisten> commit) {
+ Long sessionId = commit.session().id();
try {
- closeListener(commit.session().id());
+ SessionListenCommits listenCommits = listeners.get(sessionId);
+ if (listenCommits != null) {
+ listenCommits.remove(commit);
+ }
} finally {
commit.close();
}
@@ -261,10 +262,11 @@
result.created() ? Type.CREATED : result.newValue() == null ? Type.DELETED : Type.UPDATED,
Optional.ofNullable(result.newValue()),
Optional.ofNullable(result.oldValue()));
+
listeners.values()
- .forEach(commit -> commit.session()
- .publish(AtomixDocumentTree.CHANGE_SUBJECT,
- ImmutableList.of(event)));
+ .stream()
+ .filter(l -> event.path().isDescendentOf(l.leastCommonAncestorPath()))
+ .forEach(listener -> listener.publish(AtomixDocumentTree.CHANGE_SUBJECT, Arrays.asList(event)));
}
@Override
@@ -287,9 +289,52 @@
}
private void closeListener(Long sessionId) {
- Commit<? extends Listen> commit = listeners.remove(sessionId);
- if (commit != null) {
- commit.close();
+ SessionListenCommits listenCommits = listeners.remove(sessionId);
+ if (listenCommits != null) {
+ listenCommits.close();
+ }
+ }
+
+ private class SessionListenCommits {
+ private final List<Commit<? extends Listen>> commits = Lists.newArrayList();
+ private DocumentPath leastCommonAncestorPath;
+
+ public void add(Commit<? extends Listen> commit) {
+ commits.add(commit);
+ recomputeLeastCommonAncestor();
+ }
+
+ public void remove(Commit<? extends Unlisten> commit) {
+ // Remove the first listen commit with path matching path in unlisten commit
+ Iterator<Commit<? extends Listen>> iterator = commits.iterator();
+ while (iterator.hasNext()) {
+ Commit<? extends Listen> listenCommit = iterator.next();
+ if (listenCommit.operation().path().equals(commit.operation().path())) {
+ iterator.remove();
+ listenCommit.close();
+ }
+ }
+ recomputeLeastCommonAncestor();
+ }
+
+ public DocumentPath leastCommonAncestorPath() {
+ return leastCommonAncestorPath;
+ }
+
+ public <M> void publish(String topic, M message) {
+ commits.stream().findAny().ifPresent(commit -> commit.session().publish(topic, message));
+ }
+
+ public void close() {
+ commits.forEach(Commit::close);
+ commits.clear();
+ leastCommonAncestorPath = null;
+ }
+
+ private void recomputeLeastCommonAncestor() {
+ this.leastCommonAncestorPath = DocumentPath.leastCommonAncestor(commits.stream()
+ .map(c -> c.operation().path())
+ .collect(Collectors.toList()));
}
}
}