public MyStats getStats(int site) throws Exception {
MyStats s = new MyStats();
ReplicationStats rs = envs[site].getReplicationStats(StatsConfig.DEFAULT);
s.egen = rs.getElectionGen();
s.elections = rs.getElections();
s.envId = rs.getEnvId();
s.master = rs.getMaster();
ReplicationManagerStats rms = envs[site].getReplicationManagerStats(StatsConfig.DEFAULT);
s.permFailedCount = rms.getPermFailed();
return s;
}
synchronized void awaitNewmaster(Environment site, long deadline) throws Exception {
long now;
for (; ; ) {
ReplicationStats s = site.getReplicationStats(StatsConfig.DEFAULT);
// am I the master?
//
if (s.getEnvId() == s.getMaster()) {
break;
}
if ((now = System.currentTimeMillis()) >= deadline)
throw new Exception("aborted by timeout");
long waitTime = deadline - now;
wait(waitTime);
if (panic) throw new Exception("aborted by panic in DB");
}
}
@Test
public void startConductor() {
Vector<RepmgrElectionTest> workers = new Vector<RepmgrElectionTest>(NUM_WORKER_THREADS);
// Start the worker threads
for (int i = 0; i < NUM_WORKER_THREADS; i++) {
RepmgrElectionTest worker;
worker = new RepmgrElectionTest(i, i == 0 ? -1 : 0);
worker.run();
workers.add(worker);
}
// Ensure that the first site is master.
ReplicationStats rs = null;
try {
rs = workers.elementAt(0).dbenv.getReplicationStats(StatsConfig.DEFAULT);
} catch (DatabaseException dbe) {
fail("Unexpected database exception came from get replication stats." + dbe);
}
assertTrue(rs.getEnvId() == rs.getMaster());
// Stop the master
try {
workers.elementAt(0).dbenv.close();
workers.elementAt(0).dbenv = null;
workers.elementAt(0).envConfig = null;
} catch (DatabaseException dbe) {
fail("Unexpected database exception came during shutdown." + dbe);
}
// Ensure the election is completed.
try {
java.lang.Thread.sleep(5000);
} catch (InterruptedException ie) {
} // Do nothing if interrupted.
// Ensure that the site with priority = 75 is selected as new master.
try {
rs = workers.elementAt(1).dbenv.getReplicationStats(StatsConfig.DEFAULT);
} catch (DatabaseException dbe) {
fail("Unexpected database exception came from get replication stats." + dbe);
}
assertTrue(rs.getEnvId() == rs.getMaster());
// Re-start original master using the new master as bootstrap helper.
RepmgrElectionTest rejoin = new RepmgrElectionTest(0, 1);
rejoin.run();
workers.remove(0);
workers.insertElementAt(rejoin, 0);
// Close all the sites and remove their environment contents.
for (int i = 0; i < NUM_WORKER_THREADS; i++) {
try {
if (workers.elementAt(i).dbenv != null) {
workers.elementAt(i).dbenv.close();
workers.elementAt(i).dbenv = null;
workers.elementAt(i).envConfig = null;
}
} catch (DatabaseException dbe) {
fail("Unexpected database exception came during shutdown." + dbe);
}
}
}
public void run() {
String homedirName = baseDirName + threadNumber;
TestUtils.removeDir(homedirName);
try {
homedir = new File(homedirName);
homedir.mkdir();
} catch (Exception e) {
TestUtils.DEBUGOUT(
2, "Warning: initialization had a problem creating a clean directory.\n" + e);
}
try {
homedir = new File(homedirName);
} catch (NullPointerException npe) {
// can't really happen :)
}
TestUtils.DEBUGOUT(1, "Creating worker: " + threadNumber);
envConfig = new EnvironmentConfig();
envConfig.setErrorStream(TestUtils.getErrorStream());
envConfig.setErrorPrefix("RepmgrElectionTest test(" + threadNumber + ")");
envConfig.setAllowCreate(true);
envConfig.setRunRecovery(true);
envConfig.setThreaded(true);
envConfig.setInitializeLocking(true);
envConfig.setInitializeLogging(true);
envConfig.setInitializeCache(true);
envConfig.setTransactional(true);
envConfig.setTxnNoSync(true);
envConfig.setInitializeReplication(true);
envConfig.setVerboseReplication(false);
ReplicationManagerSiteConfig localConfig =
new ReplicationManagerSiteConfig(address, basePort + threadNumber);
localConfig.setLocalSite(true);
envConfig.addReplicationManagerSite(localConfig);
envConfig.setReplicationPriority(priorities[threadNumber]);
envConfig.setEventHandler(this);
envConfig.setReplicationManagerAckPolicy(ReplicationManagerAckPolicy.ALL);
if (masterThreadIndex >= 0) {
// If we already have the master, then set it as the bootstrap helper,
// otherwise, set local site as new master.
ReplicationManagerSiteConfig remoteConfig =
new ReplicationManagerSiteConfig(address, basePort + masterThreadIndex);
remoteConfig.setBootstrapHelper(true);
envConfig.addReplicationManagerSite(remoteConfig);
}
try {
dbenv = new Environment(homedir, envConfig);
} catch (FileNotFoundException e) {
fail("Unexpected FNFE in standard environment creation." + e);
} catch (DatabaseException dbe) {
fail("Unexpected database exception came from environment create." + dbe);
}
try {
// If we do not have master, then set local site as new master.
if (masterThreadIndex == -1)
dbenv.replicationManagerStart(NUM_WORKER_THREADS, ReplicationManagerStartPolicy.REP_MASTER);
else
dbenv.replicationManagerStart(NUM_WORKER_THREADS, ReplicationManagerStartPolicy.REP_CLIENT);
} catch (DatabaseException dbe) {
fail("Unexpected database exception came from replicationManagerStart." + dbe);
}
TestUtils.DEBUGOUT(1, "Started replication site: " + threadNumber);
lastSiteStarted = true;
try {
java.lang.Thread.sleep(1000 * (1 + threadNumber));
} catch (InterruptedException ie) {
}
if (masterThreadIndex != -1) {
// Wait for "Start-up done" for each client, then add next client.
ReplicationStats rs = null;
int i = 0;
do {
try {
java.lang.Thread.sleep(2000);
} catch (InterruptedException e) {
}
try {
rs = dbenv.getReplicationStats(StatsConfig.DEFAULT);
} catch (DatabaseException dbe) {
dbe.printStackTrace();
fail("Unexpected database exception came from getReplicationStats." + dbe);
}
} while (!rs.getStartupComplete() && i++ < maxLoopWait);
assertTrue(rs.getStartupComplete());
}
}