Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,11 @@
public abstract class CuratorFrameworkBase implements CuratorFramework {
abstract NamespaceImpl getNamespaceImpl();

/**
* Return the underlying client which is the one constructed from {@link org.apache.curator.framework.CuratorFrameworkFactory}.
*/
public abstract CuratorFramework client();

@Override
public final CuratorFramework nonNamespaceView() {
return usingNamespace(null);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -254,6 +254,11 @@ public void clearWatcherReferences(Watcher watcher) {
// NOP
}

@Override
public CuratorFramework client() {
return this;
}

@Override
public CuratorFrameworkState getState() {
return state.get();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,11 @@ public DelegatingCuratorFramework(CuratorFrameworkBase client) {
this.client = client;
}

@Override
public final CuratorFramework client() {
return client.client();
}

@Override
public CuratorFrameworkState getState() {
return client.getState();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
import org.apache.curator.framework.api.BackgroundCallback;
import org.apache.curator.framework.api.CuratorClosedException;
import org.apache.curator.framework.api.CuratorEventType;
import org.apache.curator.framework.imps.CuratorFrameworkBase;
import org.apache.curator.framework.listen.Listenable;
import org.apache.curator.framework.listen.StandardListenerManager;
import org.apache.curator.framework.state.ConnectionStateListener;
Expand Down Expand Up @@ -80,7 +81,8 @@ public PersistentWatcher(CuratorFramework client, String basePath, boolean recur
public void start() {
Preconditions.checkState(state.compareAndSet(State.LATENT, State.STARTED), "Already started");
client.getConnectionStateListenable().addListener(connectionStateListener);
client.getCuratorListenable().addListener(((ignored, event) -> {
// This could be a namespaced facade which does not support getCuratorListenable.
((CuratorFrameworkBase) client).client().getCuratorListenable().addListener(((ignored, event) -> {
if (event.getType() == CuratorEventType.CLOSING) {
onClientClosed();
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,45 @@ public void testConnectionLost() throws Exception {
internalTest(false);
}

@Test
public void testNamespacedWatching() throws Exception {
BlockingQueue<WatchedEvent> events = new LinkedBlockingQueue<>();

try (CuratorFramework client = CuratorFrameworkFactory.newClient(
server.getConnectString(), timing.session(), timing.connection(), new RetryOneTime(1))) {
client.start();
// given: connected curator client
client.blockUntilConnected();

// given: started persistent watcher under namespaced facade
PersistentWatcher persistentWatcher = new PersistentWatcher(client.usingNamespace("top"), "/main", true);
persistentWatcher.getListenable().addListener(events::add);
persistentWatcher.start();

// when: create paths
client.create().forPath("/top/main");
client.create().forPath("/top/main/a");

// then: receive node watch events
WatchedEvent event1 = events.poll(5, TimeUnit.SECONDS);
assertNotNull(event1);
assertEquals(Watcher.Event.EventType.NodeCreated, event1.getType());
assertEquals("/main", event1.getPath());

WatchedEvent event2 = events.poll(5, TimeUnit.SECONDS);
assertNotNull(event2);
assertEquals(Watcher.Event.EventType.NodeCreated, event2.getType());
assertEquals("/main/a", event2.getPath());
}

// when: curator client closed
// then: listener get Closed notification
WatchedEvent event = events.poll(5, TimeUnit.SECONDS);
assertNotNull(event);
assertEquals(Watcher.Event.EventType.None, event.getType());
assertEquals(Watcher.Event.KeeperState.Closed, event.getState());
}

@Test
public void testConcurrentClientClose() throws Exception {
BlockingQueue<WatchedEvent> events = new LinkedBlockingQueue<>();
Expand Down