Java ClientBase.createZKClient方法代码示例

import org.apache.zookeeper.test.ClientBase; //导入方法依赖的package包/类
/**
 * Starts a single server in replicated mode,
 * initializes its client, and waits for it
 * to be connected.
 */
private void startServer(int id, String config) throws Exception {
    peers[id] = new MainThread(id, clientPorts[id], config);
    peers[id].start();
    Assert.assertTrue("Server " + id + " is not up",
                      ClientBase.waitForServerUp("127.0.0.1:" + clientPorts[id], CONNECTION_TIMEOUT));
    Assert.assertTrue("Error- Server started in Standalone Mode!",
            peers[id].isQuorumPeerRunning());
    zkHandles[id] = ClientBase.createZKClient("127.0.0.1:" + clientPorts[id]);
    zkAdminHandles[id] = new ZooKeeperAdmin("127.0.0.1:" + clientPorts[id], CONNECTION_TIMEOUT, this);
    zkAdminHandles[id].addAuthInfo("digest", "super:test".getBytes());
}
 

import org.apache.zookeeper.test.ClientBase; //导入方法依赖的package包/类
/**
 * test the purge
 * @throws Exception an exception might be thrown here
 */
@Test
public void testPurge() throws Exception {
    tmpDir = ClientBase.createTmpDir();
    ClientBase.setupTestEnv();
    ZooKeeperServer zks = new ZooKeeperServer(tmpDir, tmpDir, 3000);
    SyncRequestProcessor.setSnapCount(100);
    final int PORT = Integer.parseInt(HOSTPORT.split(":")[1]);
    ServerCnxnFactory f = ServerCnxnFactory.createFactory(PORT, -1);
    f.startup(zks);
    Assert.assertTrue("waiting for server being up ",
            ClientBase.waitForServerUp(HOSTPORT,CONNECTION_TIMEOUT));
    ZooKeeper zk = ClientBase.createZKClient(HOSTPORT);
    try {
        for (int i = 0; i< 2000; i++) {
            zk.create("/invalidsnap-" + i, new byte[0], Ids.OPEN_ACL_UNSAFE,
                    CreateMode.PERSISTENT);
        }
    } finally {
        zk.close();
    }
    f.shutdown();
    zks.getTxnLogFactory().close();
    Assert.assertTrue("waiting for server to shutdown",
            ClientBase.waitForServerDown(HOSTPORT, CONNECTION_TIMEOUT));
    // now corrupt the snapshot
    PurgeTxnLog.purge(tmpDir, tmpDir, 3);
    FileTxnSnapLog snaplog = new FileTxnSnapLog(tmpDir, tmpDir);
    List<File> listLogs = snaplog.findNRecentSnapshots(4);
    int numSnaps = 0;
    for (File ff: listLogs) {
        if (ff.getName().startsWith("snapshot")) {
            numSnaps++;
        }
    }
    Assert.assertTrue("exactly 3 snapshots ", (numSnaps == 3));
    snaplog.close();
    zks.shutdown();
}
 

import org.apache.zookeeper.test.ClientBase; //导入方法依赖的package包/类
/**
 * Test case for https://issues.apache.org/jira/browse/ZOOKEEPER-2244
 *
 * @throws Exception
 */
@Test(timeout = 120000)
public void testRestartZooKeeperServer() throws Exception {
    final int clientPorts[] = new int[SERVER_COUNT];
    StringBuilder sb = new StringBuilder();
    String server;

    for (int i = 0; i < SERVER_COUNT; i++) {
        clientPorts[i] = PortAssignment.unique();
        server = "server." + i + "=127.0.0.1:" + PortAssignment.unique()
                + ":" + PortAssignment.unique() + ":participant;127.0.0.1:"
                + clientPorts[i];
        sb.append(server + "\n");
    }
    String currentQuorumCfgSection = sb.toString();
    MainThread mt[] = new MainThread[SERVER_COUNT];

    for (int i = 0; i < SERVER_COUNT; i++) {
        mt[i] = new MainThread(i, clientPorts[i], currentQuorumCfgSection,
                false);
        mt[i].start();
    }

    // ensure server started
    for (int i = 0; i < SERVER_COUNT; i++) {
        Assert.assertTrue("waiting for server " + i + " being up",
                ClientBase.waitForServerUp("127.0.0.1:" + clientPorts[i],
                        CONNECTION_TIMEOUT));
    }

    ZooKeeper zk = ClientBase.createZKClient("127.0.0.1:" + clientPorts[0]);

    String zNodePath="/serverRestartTest";
    String data = "originalData";
    zk.create(zNodePath, data.getBytes(), Ids.OPEN_ACL_UNSAFE,
            CreateMode.PERSISTENT);
    zk.close();

    /**
     * stop two servers out of three and again start them
     */
    mt[0].shutdown();
    mt[1].shutdown();
    mt[0].start();
    mt[1].start();
    // ensure server started
    for (int i = 0; i < SERVER_COUNT; i++) {
        Assert.assertTrue("waiting for server " + i + " being up",
                ClientBase.waitForServerUp("127.0.0.1:" + clientPorts[i],
                        CONNECTION_TIMEOUT));
    }
    zk = ClientBase.createZKClient("127.0.0.1:" + clientPorts[0]);

    byte[] dataBytes = zk.getData(zNodePath, null, null);
    String receivedData = new String(dataBytes);
    assertEquals(data, receivedData);

    for (int i = 0; i < SERVER_COUNT; i++) {
        mt[i].shutdown();
    }
}
 

import org.apache.zookeeper.test.ClientBase; //导入方法依赖的package包/类
private void verifyQuorumConfig(int sid, List<String> joiningServers, List<String> leavingServers) throws Exception {
    ZooKeeper zk = ClientBase.createZKClient("127.0.0.1:" + clientPorts.get(sid));
    ReconfigTest.testNormalOperation(zk, zk);
    ReconfigTest.testServerHasConfig(zk, joiningServers, leavingServers);
    zk.close();
}
 

import org.apache.zookeeper.test.ClientBase; //导入方法依赖的package包/类
/**
 * Tests purge when logs are rolling or a new snapshot is created, then
 * these newer files should alse be excluded in the current cycle.
 *
 * For frequent snapshotting, configured SnapCount to 30. There are three
 * threads which will create 1000 znodes each and simultaneously do purge
 * call
 */
@Test
public void testPurgeWhenLogRollingInProgress() throws Exception {
    tmpDir = ClientBase.createTmpDir();
    ClientBase.setupTestEnv();
    ZooKeeperServer zks = new ZooKeeperServer(tmpDir, tmpDir, 3000);
    SyncRequestProcessor.setSnapCount(30);
    final int PORT = Integer.parseInt(HOSTPORT.split(":")[1]);
    ServerCnxnFactory f = ServerCnxnFactory.createFactory(PORT, -1);
    f.startup(zks);
    Assert.assertTrue("waiting for server being up ",
            ClientBase.waitForServerUp(HOSTPORT, CONNECTION_TIMEOUT));
    final ZooKeeper zk = ClientBase.createZKClient(HOSTPORT);
    final CountDownLatch doPurge = new CountDownLatch(1);
    final CountDownLatch purgeFinished = new CountDownLatch(1);
    final AtomicBoolean opFailed = new AtomicBoolean(false);
    new Thread() {
        public void run() {
            try {
                doPurge.await(OP_TIMEOUT_IN_MILLIS / 2,
                        TimeUnit.MILLISECONDS);
                PurgeTxnLog.purge(tmpDir, tmpDir, 3);
            } catch (IOException ioe) {
                LOG.error("Exception when purge", ioe);
                opFailed.set(true);
            } catch (InterruptedException ie) {
                LOG.error("Exception when purge", ie);
                opFailed.set(true);
            } finally {
                purgeFinished.countDown();
            }
        };
    }.start();
    final int thCount = 3;
    List<String> znodes = manyClientOps(zk, doPurge, thCount,
            "/invalidsnap");
    Assert.assertTrue("Purging is not finished!", purgeFinished.await(
            OP_TIMEOUT_IN_MILLIS, TimeUnit.MILLISECONDS));
    Assert.assertFalse("Purging failed!", opFailed.get());
    for (String znode : znodes) {
        try {
            zk.getData(znode, false, null);
        } catch (Exception ke) {
            LOG.error("Unexpected exception when visiting znode!", ke);
            Assert.fail("Unexpected exception when visiting znode!");
        }
    }
    zk.close();
    f.shutdown();
    zks.shutdown();
    zks.getTxnLogFactory().close();
}