Java AsyncCallback.DataCallback方法代码示例

import org.apache.zookeeper.AsyncCallback; //导入方法依赖的package包/类
@Test
public void testNotCloseZkWhenPending() throws Exception {
  ZooKeeper mockedZK = mock(ZooKeeper.class);
  Exchanger<AsyncCallback.DataCallback> exchanger = new Exchanger<>();
  doAnswer(i -> {
    exchanger.exchange(i.getArgument(2));
    return null;
  }).when(mockedZK).getData(anyString(), anyBoolean(),
    any(AsyncCallback.DataCallback.class), any());
  doAnswer(i -> null).when(mockedZK).close();
  when(mockedZK.getState()).thenReturn(ZooKeeper.States.CONNECTED);
  RO_ZK.zookeeper = mockedZK;
  CompletableFuture<byte[]> future = RO_ZK.get(PATH);
  AsyncCallback.DataCallback callback = exchanger.exchange(null);
  // 2 * keep alive time to ensure that we will not close the zk when there are pending requests
  Thread.sleep(6000);
  assertNotNull(RO_ZK.zookeeper);
  verify(mockedZK, never()).close();
  callback.processResult(Code.OK.intValue(), PATH, null, DATA, null);
  assertArrayEquals(DATA, future.get());
  // now we will close the idle connection.
  waitForIdleConnectionClosed();
  verify(mockedZK, times(1)).close();
}
 

import org.apache.zookeeper.AsyncCallback; //导入方法依赖的package包/类
@Override
public void performBackgroundOperation(final OperationAndData<Void> operationAndData) throws Exception
{
    try
    {
        final TimeTrace trace = client.getZookeeperClient().startTracer("GetDataBuilderImpl-Background");
        AsyncCallback.DataCallback callback = new AsyncCallback.DataCallback()
        {
            @Override
            public void processResult(int rc, String path, Object ctx, byte[] data, Stat stat)
            {
                watching.commitWatcher(rc, false);
                trace.commit();
                CuratorEvent event = new CuratorEventImpl(client, CuratorEventType.GET_CONFIG, rc, path, null, ctx, stat, data, null, null, null, null);
                client.processBackgroundOperation(operationAndData, event);
            }
        };
        if ( watching.isWatched() )
        {
            client.getZooKeeper().getConfig(true, callback, backgrounding.getContext());
        }
        else
        {
            client.getZooKeeper().getConfig(watching.getWatcher(ZooDefs.CONFIG_NODE), callback, backgrounding.getContext());
        }
    }
    catch ( Throwable e )
    {
        backgrounding.checkError(e, watching);
    }
}
 

import org.apache.zookeeper.AsyncCallback; //导入方法依赖的package包/类
@Override
public void performBackgroundOperation(final OperationAndData<Void> data) throws Exception
{
    try
    {
        final TimeTrace trace = client.getZookeeperClient().startTracer("ReconfigBuilderImpl-Background");
        AsyncCallback.DataCallback callback = new AsyncCallback.DataCallback()
        {
            @Override
            public void processResult(int rc, String path, Object ctx, byte[] bytes, Stat stat)
            {
                trace.commit();
                if ( (responseStat != null) && (stat != null) )
                {
                    DataTree.copyStat(stat, responseStat);
                }
                CuratorEvent event = new CuratorEventImpl(client, CuratorEventType.RECONFIG, rc, path, null, ctx, stat, bytes, null, null, null, null);
                client.processBackgroundOperation(data, event);
            }
        };
        client.getZooKeeper().reconfig(joining, leaving, newMembers, fromConfig, callback, backgrounding.getContext());
    }
    catch ( Throwable e )
    {
        backgrounding.checkError(e, null);
    }
}
 

import org.apache.zookeeper.AsyncCallback; //导入方法依赖的package包/类
/**
 * This test verifies that when the session id is reused, and the original
 * client is disconnected, but not session closed, that the server
 * will remove ephemeral nodes created by the original session.
 */
@Test
public void testSession()
    throws IOException, InterruptedException, KeeperException
{
    DisconnectableZooKeeper zk = createClient();
    zk.create("/e", new byte[0], Ids.OPEN_ACL_UNSAFE,
                    CreateMode.EPHEMERAL);
    LOG.info("zk with session id 0x" + Long.toHexString(zk.getSessionId())
            + " was destroyed!");

    // disconnect the client by killing the socket, not sending the
    // session disconnect to the server as usual. This allows the test
    // to verify disconnect handling
    zk.disconnect();

    Stat stat = new Stat();
    startSignal = new CountDownLatch(1);
    zk = new DisconnectableZooKeeper(HOSTPORT, CONNECTION_TIMEOUT,
            new MyWatcher("testSession"), zk.getSessionId(),
            zk.getSessionPasswd());
    startSignal.await();

    LOG.info("zk with session id 0x" + Long.toHexString(zk.getSessionId())
             + " was created!");
    zk.getData("/e", false, stat);
    LOG.info("After get data /e");
    zk.close();

    zk = createClient();
    Assert.assertEquals(null, zk.exists("/e", false));
    LOG.info("before close zk with session id 0x"
            + Long.toHexString(zk.getSessionId()) + "!");
    zk.close();
    try {
        zk.getData("/e", false, stat);
        Assert.fail("Should have received a SessionExpiredException");
    } catch(KeeperException.SessionExpiredException e) {}
    
    AsyncCallback.DataCallback cb = new AsyncCallback.DataCallback() {
        String status = "not done";
        public void processResult(int rc, String p, Object c, byte[] b, Stat s) {
            synchronized(this) { status = KeeperException.Code.get(rc).toString(); this.notify(); }
        }
       public String toString() { return status; }
    };
    zk.getData("/e", false, cb, null);
    synchronized(cb) {
        if (cb.toString().equals("not done")) {
            cb.wait(1000);
        }
    }
    Assert.assertEquals(KeeperException.Code.SESSIONEXPIRED.toString(), cb.toString());        
}
 

import org.apache.zookeeper.AsyncCallback; //导入方法依赖的package包/类
/**
 * This test verifies that when the session id is reused, and the original
 * client is disconnected, but not session closed, that the server
 * will remove ephemeral nodes created by the original session.
 */
@Test
public void testSession()
    throws IOException, InterruptedException, KeeperException
{
    DisconnectableZooKeeper zk = createClient();
    zk.create("/e", new byte[0], Ids.OPEN_ACL_UNSAFE,
                    CreateMode.EPHEMERAL);
    LOG.info("zk with session id 0x" + Long.toHexString(zk.getSessionId())
            + " was destroyed!");

    // disconnect the client by killing the socket, not sending the
    // session disconnect to the server as usual. This allows the test
    // to verify disconnect handling
    zk.disconnect();

    Stat stat = new Stat();
    startSignal = new CountDownLatch(1);
    zk = new DisconnectableZooKeeper(HOSTPORT, CONNECTION_TIMEOUT,
            new MyWatcher("testSession"), zk.getSessionId(),
            zk.getSessionPasswd());
    startSignal.await();

    LOG.info("zk with session id 0x" + Long.toHexString(zk.getSessionId())
             + " was created!");
    zk.getData("/e", false, stat);
    LOG.info("After get data /e");
    zk.close();

    zk = createClient();
    Assert.assertEquals(null, zk.exists("/e", false));
    LOG.info("before close zk with session id 0x"
            + Long.toHexString(zk.getSessionId()) + "!");
    zk.close();
    try {
        zk.getData("/e", false, stat);
        Assert.fail("Should have received a SessionExpiredException");
    } catch(KeeperException.SessionExpiredException e) {}

    AsyncCallback.DataCallback cb = new AsyncCallback.DataCallback() {
        String status = "not done";
        public void processResult(int rc, String p, Object c, byte[] b, Stat s) {
            synchronized(this) { status = KeeperException.Code.get(rc).toString(); this.notify(); }
        }
       public String toString() { return status; }
    };
    zk.getData("/e", false, cb, null);
    synchronized(cb) {
        if (cb.toString().equals("not done")) {
            cb.wait(1000);
        }
    }
    Assert.assertEquals(KeeperException.Code.SESSIONEXPIRED.toString(), cb.toString());
}
 

import org.apache.zookeeper.AsyncCallback; //导入方法依赖的package包/类
/**
 * This test verifies that when the session id is reused, and the original
 * client is disconnected, but not session closed, that the server
 * will remove ephemeral nodes created by the original session.
 */
@Test
public void testSession()
    throws IOException, InterruptedException, KeeperException
{
    DisconnectableZooKeeper zk = createClient();
    zk.create("/e", new byte[0], Ids.OPEN_ACL_UNSAFE,
                    CreateMode.EPHEMERAL);
    LOG.info("zk with session id 0x" + Long.toHexString(zk.getSessionId())
            + " was destroyed!");

    // disconnect the client by killing the socket, not sending the
    // session disconnect to the server as usual. This allows the test
    // to verify disconnect handling
    zk.disconnect();

    Stat stat = new Stat();
    startSignal = new CountDownLatch(1);
    zk = new DisconnectableZooKeeper(HOSTPORT, CONNECTION_TIMEOUT, this,
                           zk.getSessionId(),
                           zk.getSessionPasswd());
    startSignal.await();

    LOG.info("zk with session id 0x" + Long.toHexString(zk.getSessionId())
             + " was created!");
    zk.getData("/e", false, stat);
    LOG.info("After get data /e");
    zk.close();

    zk = createClient();
    assertEquals(null, zk.exists("/e", false));
    LOG.info("before close zk with session id 0x"
            + Long.toHexString(zk.getSessionId()) + "!");
    zk.close();
    try {
        zk.getData("/e", false, stat);
        Assert.fail("Should have received a SessionExpiredException");
    } catch(KeeperException.SessionExpiredException e) {}
    
    AsyncCallback.DataCallback cb = new AsyncCallback.DataCallback() {
        String status = "not done";
        public void processResult(int rc, String p, Object c, byte[] b, Stat s) {
            synchronized(this) { status = KeeperException.Code.get(rc).toString(); this.notify(); }
        }
       public String toString() { return status; }
    };
    zk.getData("/e", false, cb, null);
    synchronized(cb) {
        if (cb.toString().equals("not done")) {
            cb.wait(1000);
        }
    }
    Assert.assertEquals(KeeperException.Code.SESSIONEXPIRED.toString(), cb.toString());        
}
 

import org.apache.zookeeper.AsyncCallback; //导入方法依赖的package包/类
@Override
public void getData(String path, Watcher watcher, AsyncCallback.DataCallback cb, Object ctx) {
    delegate.getData(path, watcher, cb, ctx);
}
 

import org.apache.zookeeper.AsyncCallback; //导入方法依赖的package包/类
void getData(String path, Watcher watcher, AsyncCallback.DataCallback cb, Object ctx);