Java PeerStruct类代码示例

import org.apache.zookeeper.test.QuorumUtil.PeerStruct; //导入依赖的package包/类
@Override
protected void setUp() throws Exception {
    LOG.info("STARTING " + getName());

    // set the snap count to something low so that we force log rollover
    // and verify that is working as part of the epoch rollover.
    SyncRequestProcessor.setSnapCount(7);

    qu = new QuorumUtil(1);
    startAll();

    for (int i = 0; i < zkClients.length; i++) {
        zkClientWatchers[i] = new CountdownWatcher();
        PeerStruct peer = qu.getPeer(i + 1);
        zkClients[i] = new ZooKeeper(
                "127.0.0.1:" + peer.clientPort,
                ClientTest.CONNECTION_TIMEOUT, zkClientWatchers[i]);
    }
    waitForClientsConnected();
}
 

import org.apache.zookeeper.test.QuorumUtil.PeerStruct; //导入依赖的package包/类
/**
 * Ensure the client is able to talk to the server.
 * 
 * @param idx the idx of the server the client is talking to
 */
private void checkClientConnected(int idx) throws Exception {
    ZooKeeper zk = getClient(idx);
    if (zk == null) {
        return;
    }
    try {
        assertNull(zk.exists("/foofoofoo-connected", false));
    } catch (ConnectionLossException e) {
        // second chance...
        // in some cases, leader change in particular, the timing is
        // very tricky to get right in order to assure that the client has
        // disconnected and reconnected. In some cases the client will
        // disconnect, then attempt to reconnect before the server is
        // back, in which case we'll see another connloss on the operation
        // in the try, this catches that case and waits for the server
        // to come back
        PeerStruct peer = qu.getPeer(idx);
        Assert.assertTrue("Waiting for server down", ClientBase.waitForServerUp(
                "127.0.0.1:" + peer.clientPort, ClientBase.CONNECTION_TIMEOUT));

        assertNull(zk.exists("/foofoofoo-connected", false));
    }
}
 

import org.apache.zookeeper.test.QuorumUtil.PeerStruct; //导入依赖的package包/类
private void shutdown(int idx) throws Exception {
    qu.shutdown(idx);

    // leader will shutdown, remaining followers will elect a new leader
    PeerStruct peer = qu.getPeer(idx);
    Assert.assertTrue("Waiting for server down", ClientBase.waitForServerDown(
            "127.0.0.1:" + peer.clientPort, ClientBase.CONNECTION_TIMEOUT));

    // if idx is the the leader then everyone will get disconnected,
    // otherwise if idx is a follower then just that client will get
    // disconnected
    if (idx == idxLeader) {
        checkClientDisconnected(idx);
        try {
            checkClientsDisconnected();
        } catch (AssertionFailedError e) {
            // the clients may or may not have already reconnected
            // to the recovered cluster, force a check, but ignore
        }
    } else {
        checkClientDisconnected(idx);
    }
}
 

import org.apache.zookeeper.test.QuorumUtil.PeerStruct; //导入依赖的package包/类
@Before
public void setUp() throws Exception {
    System.setProperty("zookeeper.admin.enableServer", "false");

    // set the snap count to something low so that we force log rollover
    // and verify that is working as part of the epoch rollover.
    SyncRequestProcessor.setSnapCount(7);

    qu = new QuorumUtil(1);
    startAll();

    for (int i = 0; i < zkClients.length; i++) {
        zkClientWatchers[i] = new CountdownWatcher();
        PeerStruct peer = qu.getPeer(i + 1);
        zkClients[i] = new ZooKeeper(
                "127.0.0.1:" + peer.clientPort,
                ClientTest.CONNECTION_TIMEOUT, zkClientWatchers[i]);
    }
    waitForClientsConnected();
}
 

import org.apache.zookeeper.test.QuorumUtil.PeerStruct; //导入依赖的package包/类
/**
 * Ensure the client is able to talk to the server.
 * 
 * @param idx the idx of the server the client is talking to
 */
private void checkClientConnected(int idx) throws Exception {
    ZooKeeper zk = getClient(idx);
    if (zk == null) {
        return;
    }
    try {
        Assert.assertNull(zk.exists("/foofoofoo-connected", false));
    } catch (ConnectionLossException e) {
        // second chance...
        // in some cases, leader change in particular, the timing is
        // very tricky to get right in order to assure that the client has
        // disconnected and reconnected. In some cases the client will
        // disconnect, then attempt to reconnect before the server is
        // back, in which case we'll see another connloss on the operation
        // in the try, this catches that case and waits for the server
        // to come back
        PeerStruct peer = qu.getPeer(idx);
        Assert.assertTrue("Waiting for server down", ClientBase.waitForServerUp(
                "127.0.0.1:" + peer.clientPort, ClientBase.CONNECTION_TIMEOUT));

        Assert.assertNull(zk.exists("/foofoofoo-connected", false));
    }
}
 

import org.apache.zookeeper.test.QuorumUtil.PeerStruct; //导入依赖的package包/类
private void shutdown(int idx) throws Exception {
    qu.shutdown(idx);

    // leader will shutdown, remaining followers will elect a new leader
    PeerStruct peer = qu.getPeer(idx);
    Assert.assertTrue("Waiting for server down", ClientBase.waitForServerDown(
            "127.0.0.1:" + peer.clientPort, ClientBase.CONNECTION_TIMEOUT));

    // if idx is the the leader then everyone will get disconnected,
    // otherwise if idx is a follower then just that client will get
    // disconnected
    if (idx == idxLeader) {
        checkClientDisconnected(idx);
        try {
            checkClientsDisconnected();
        } catch (AssertionError e) {
            // the clients may or may not have already reconnected
            // to the recovered cluster, force a check, but ignore
        }
    } else {
        checkClientDisconnected(idx);
    }
}