org.apache.zookeeper.server.ZooKeeperServer#closeSession ( )源码实例Demo

@Test(timeout=15000)
public void testHandleSessionExpiration() throws Exception {
  ActiveStandbyElectorCallback cb = cbs[0];
  byte[] appData = appDatas[0];
  ActiveStandbyElector elector = electors[0];
  
  // Let the first elector become active
  elector.ensureParentZNode();
  elector.joinElection(appData);
  ZooKeeperServer zks = getServer(serverFactory);
  ActiveStandbyElectorTestUtil.waitForActiveLockData(null,
      zks, PARENT_DIR, appData);
  Mockito.verify(cb, Mockito.timeout(1000)).becomeActive();
  checkFatalsAndReset();
  
  LOG.info("========================== Expiring session");
  zks.closeSession(elector.getZKSessionIdForTests());

  // Should enter neutral mode when disconnected
  Mockito.verify(cb, Mockito.timeout(1000)).enterNeutralMode();

  // Should re-join the election and regain active
  ActiveStandbyElectorTestUtil.waitForActiveLockData(null,
      zks, PARENT_DIR, appData);
  Mockito.verify(cb, Mockito.timeout(1000)).becomeActive();
  checkFatalsAndReset();
  
  LOG.info("========================== Quitting election");
  elector.quitElection(false);
  ActiveStandbyElectorTestUtil.waitForActiveLockData(null,
      zks, PARENT_DIR, null);

  // Double check that we don't accidentally re-join the election
  // due to receiving the "expired" event.
  Thread.sleep(1000);
  Mockito.verify(cb, Mockito.never()).becomeActive();
  ActiveStandbyElectorTestUtil.waitForActiveLockData(null,
      zks, PARENT_DIR, null);

  checkFatalsAndReset();
}
 

@Test(timeout=15000)
public void testHandleSessionExpirationOfStandby() throws Exception {
  // Let elector 0 be active
  electors[0].ensureParentZNode();
  electors[0].joinElection(appDatas[0]);
  ZooKeeperServer zks = getServer(serverFactory);
  ActiveStandbyElectorTestUtil.waitForActiveLockData(null,
      zks, PARENT_DIR, appDatas[0]);
  Mockito.verify(cbs[0], Mockito.timeout(1000)).becomeActive();
  checkFatalsAndReset();
  
  // Let elector 1 be standby
  electors[1].joinElection(appDatas[1]);
  ActiveStandbyElectorTestUtil.waitForElectorState(null, electors[1],
      State.STANDBY);
  
  LOG.info("========================== Expiring standby's session");
  zks.closeSession(electors[1].getZKSessionIdForTests());

  // Should enter neutral mode when disconnected
  Mockito.verify(cbs[1], Mockito.timeout(1000)).enterNeutralMode();

  // Should re-join the election and go back to STANDBY
  ActiveStandbyElectorTestUtil.waitForElectorState(null, electors[1],
      State.STANDBY);
  checkFatalsAndReset();
  
  LOG.info("========================== Quitting election");
  electors[1].quitElection(false);

  // Double check that we don't accidentally re-join the election
  // by quitting elector 0 and ensuring elector 1 doesn't become active
  electors[0].quitElection(false);
  
  // due to receiving the "expired" event.
  Thread.sleep(1000);
  Mockito.verify(cbs[1], Mockito.never()).becomeActive();
  ActiveStandbyElectorTestUtil.waitForActiveLockData(null,
      zks, PARENT_DIR, null);

  checkFatalsAndReset();
}
 

@Test(timeout=15000)
public void testHandleSessionExpiration() throws Exception {
  ActiveStandbyElectorCallback cb = cbs[0];
  byte[] appData = appDatas[0];
  ActiveStandbyElector elector = electors[0];
  
  // Let the first elector become active
  elector.ensureParentZNode();
  elector.joinElection(appData);
  ZooKeeperServer zks = getServer(serverFactory);
  ActiveStandbyElectorTestUtil.waitForActiveLockData(null,
      zks, PARENT_DIR, appData);
  Mockito.verify(cb, Mockito.timeout(1000)).becomeActive();
  checkFatalsAndReset();
  
  LOG.info("========================== Expiring session");
  zks.closeSession(elector.getZKSessionIdForTests());

  // Should enter neutral mode when disconnected
  Mockito.verify(cb, Mockito.timeout(1000)).enterNeutralMode();

  // Should re-join the election and regain active
  ActiveStandbyElectorTestUtil.waitForActiveLockData(null,
      zks, PARENT_DIR, appData);
  Mockito.verify(cb, Mockito.timeout(1000)).becomeActive();
  checkFatalsAndReset();
  
  LOG.info("========================== Quitting election");
  elector.quitElection(false);
  ActiveStandbyElectorTestUtil.waitForActiveLockData(null,
      zks, PARENT_DIR, null);

  // Double check that we don't accidentally re-join the election
  // due to receiving the "expired" event.
  Thread.sleep(1000);
  Mockito.verify(cb, Mockito.never()).becomeActive();
  ActiveStandbyElectorTestUtil.waitForActiveLockData(null,
      zks, PARENT_DIR, null);

  checkFatalsAndReset();
}
 

@Test(timeout=15000)
public void testHandleSessionExpirationOfStandby() throws Exception {
  // Let elector 0 be active
  electors[0].ensureParentZNode();
  electors[0].joinElection(appDatas[0]);
  ZooKeeperServer zks = getServer(serverFactory);
  ActiveStandbyElectorTestUtil.waitForActiveLockData(null,
      zks, PARENT_DIR, appDatas[0]);
  Mockito.verify(cbs[0], Mockito.timeout(1000)).becomeActive();
  checkFatalsAndReset();
  
  // Let elector 1 be standby
  electors[1].joinElection(appDatas[1]);
  ActiveStandbyElectorTestUtil.waitForElectorState(null, electors[1],
      State.STANDBY);
  
  LOG.info("========================== Expiring standby's session");
  zks.closeSession(electors[1].getZKSessionIdForTests());

  // Should enter neutral mode when disconnected
  Mockito.verify(cbs[1], Mockito.timeout(1000)).enterNeutralMode();

  // Should re-join the election and go back to STANDBY
  ActiveStandbyElectorTestUtil.waitForElectorState(null, electors[1],
      State.STANDBY);
  checkFatalsAndReset();
  
  LOG.info("========================== Quitting election");
  electors[1].quitElection(false);

  // Double check that we don't accidentally re-join the election
  // by quitting elector 0 and ensuring elector 1 doesn't become active
  electors[0].quitElection(false);
  
  // due to receiving the "expired" event.
  Thread.sleep(1000);
  Mockito.verify(cbs[1], Mockito.never()).becomeActive();
  ActiveStandbyElectorTestUtil.waitForActiveLockData(null,
      zks, PARENT_DIR, null);

  checkFatalsAndReset();
}