Apache
/
hbase
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

HBASE-18735 Provide an option to kill a MiniHBaseCluster without waiting on shutdown 

Signed-off-by: Josh Elser <elserj@apache.org>
This commit is contained in:
Artem Ervits 2018-11-29 16:37:50 -05:00 committed by Josh Elser
parent acb58284a5
commit 349a961554
2 changed files with 64 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
hbase-server/src/test/java/org/apache/hadoop/hbase/HBaseTestingUtility.java
View File

@ -1112,9 +1112,41 @@ public class HBaseTestingUtility extends HBaseZKTestingUtility {
}
/**
* Shutdown HBase mini cluster. Does not shutdown zk or dfs if running.
* Shutdown HBase mini cluster.Does not shutdown zk or dfs if running.
* @throws java.io.IOException in case command is unsuccessful
*/
public void shutdownMiniHBaseCluster() throws IOException {
cleanup();
if (this.hbaseCluster != null) {
this.hbaseCluster.shutdown();
// Wait till hbase is down before going on to shutdown zk.
this.hbaseCluster.waitUntilShutDown();
this.hbaseCluster = null;
}
if (zooKeeperWatcher != null) {
zooKeeperWatcher.close();
zooKeeperWatcher = null;
}
}
/**
* Abruptly Shutdown HBase mini cluster. Does not shutdown zk or dfs if running.
* @throws java.io.IOException throws in case command is unsuccessful
*/
public void killMiniHBaseCluster() throws IOException {
cleanup();
if (this.hbaseCluster != null) {
getMiniHBaseCluster().killAll();
this.hbaseCluster = null;
}
if (zooKeeperWatcher != null) {
zooKeeperWatcher.close();
zooKeeperWatcher = null;
}
}
// close hbase admin, close current connection and reset MIN MAX configs for RS.
private void cleanup() throws IOException {
if (hbaseAdmin != null) {
hbaseAdmin.close();
hbaseAdmin = null;
@ -1126,16 +1158,6 @@ public class HBaseTestingUtility extends HBaseZKTestingUtility {
// unset the configuration for MIN and MAX RS to start
conf.setInt(ServerManager.WAIT_ON_REGIONSERVERS_MINTOSTART, -1);
conf.setInt(ServerManager.WAIT_ON_REGIONSERVERS_MAXTOSTART, -1);
if (this.hbaseCluster != null) {
this.hbaseCluster.shutdown();
// Wait till hbase is down before going on to shutdown zk.
this.hbaseCluster.waitUntilShutDown();
this.hbaseCluster = null;
}
if (zooKeeperWatcher != null) {
zooKeeperWatcher.close();
zooKeeperWatcher = null;
}
}
/**

35
hbase-server/src/test/java/org/apache/hadoop/hbase/TestHBaseTestingUtility.java
View File

@ -26,11 +26,7 @@ import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Random;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.FileSystem;
@ -62,6 +58,9 @@ import org.slf4j.LoggerFactory;
*/
@Category({MiscTests.class, LargeTests.class})
public class TestHBaseTestingUtility {
private static final int NUMTABLES = 1;
private static final int NUMROWS = 100;
private static final int NUMREGIONS = 10;
@ClassRule
public static final HBaseClassTestRule CLASS_RULE =
@ -471,4 +470,32 @@ public class TestHBaseTestingUtility {
htu.shutdownMiniCluster();
}
}
// This test demonstrates how long killHBTU takes vs. shutdownHBTU takes
// for realistic results, adjust NUMROWS, NUMTABLES to much larger number.
@Test
public void testKillMiniHBaseCluster() throws Exception {
HBaseTestingUtility htu = new HBaseTestingUtility();
htu.startMiniZKCluster();
try {
htu.startMiniHBaseCluster(1, 1);
TableName tableName;
byte[] FAM_NAME;
for(int i = 0; i < NUMTABLES; i++) {
tableName = TableName.valueOf(name.getMethodName() + i);
FAM_NAME = Bytes.toBytes("fam" + i);
try (Table table = htu.createMultiRegionTable(tableName, FAM_NAME, NUMREGIONS)) {
htu.loadRandomRows(table, FAM_NAME, 100, NUMROWS);
}
}
} finally {
htu.killMiniHBaseCluster();
htu.shutdownMiniZKCluster();
}
}
}