HBASE-11211 LoadTestTool option for specifying number of regions per server

This commit is contained in:
Andrew Purtell 2014-05-22 18:49:14 -07:00
parent b168b8b2d5
commit fd94fcde58
2 changed files with 20 additions and 5 deletions

View File

@ -132,11 +132,12 @@ import org.apache.zookeeper.ZooKeeper.States;
public class HBaseTestingUtility extends HBaseCommonTestingUtility { public class HBaseTestingUtility extends HBaseCommonTestingUtility {
private MiniZooKeeperCluster zkCluster = null; private MiniZooKeeperCluster zkCluster = null;
public static final String REGIONS_PER_SERVER_KEY = "hbase.test.regions-per-server";
/** /**
* The default number of regions per regionserver when creating a pre-split * The default number of regions per regionserver when creating a pre-split
* table. * table.
*/ */
private static int DEFAULT_REGIONS_PER_SERVER = 5; public static final int DEFAULT_REGIONS_PER_SERVER = 5;
/** /**
* Set if we were passed a zkCluster. If so, we won't shutdown zk as * Set if we were passed a zkCluster. If so, we won't shutdown zk as
@ -3196,10 +3197,11 @@ public class HBaseTestingUtility extends HBaseCommonTestingUtility {
throw new IllegalStateException("No live regionservers"); throw new IllegalStateException("No live regionservers");
} }
totalNumberOfRegions = numberOfServers * DEFAULT_REGIONS_PER_SERVER; int regionsPerServer = conf.getInt(REGIONS_PER_SERVER_KEY, DEFAULT_REGIONS_PER_SERVER);
totalNumberOfRegions = numberOfServers * regionsPerServer;
LOG.info("Number of live regionservers: " + numberOfServers + ", " + LOG.info("Number of live regionservers: " + numberOfServers + ", " +
"pre-splitting table into " + totalNumberOfRegions + " regions " + "pre-splitting table into " + totalNumberOfRegions + " regions " +
"(default regions per server: " + DEFAULT_REGIONS_PER_SERVER + ")"); "(default regions per server: " + regionsPerServer + ")");
byte[][] splits = new RegionSplitter.HexStringSplit().split( byte[][] splits = new RegionSplitter.HexStringSplit().split(
totalNumberOfRegions); totalNumberOfRegions);

View File

@ -130,7 +130,8 @@ public class LoadTestTool extends AbstractHBaseTool {
protected static final String OPT_ZK_PARENT_NODE = "zk_root"; protected static final String OPT_ZK_PARENT_NODE = "zk_root";
protected static final String OPT_SKIP_INIT = "skip_init"; protected static final String OPT_SKIP_INIT = "skip_init";
protected static final String OPT_INIT_ONLY = "init_only"; protected static final String OPT_INIT_ONLY = "init_only";
private static final String NUM_TABLES = "num_tables"; protected static final String NUM_TABLES = "num_tables";
protected static final String OPT_REGIONS_PER_SERVER = "regions_per_server";
protected static final String OPT_BATCHUPDATE = "batchupdate"; protected static final String OPT_BATCHUPDATE = "batchupdate";
protected static final String OPT_UPDATE = "update"; protected static final String OPT_UPDATE = "update";
@ -178,6 +179,7 @@ public class LoadTestTool extends AbstractHBaseTool {
private int verifyPercent; private int verifyPercent;
private int numTables = 1; private int numTables = 1;
private int regionsPerServer = HBaseTestingUtility.DEFAULT_REGIONS_PER_SERVER;
private String superUser; private String superUser;
@ -292,6 +294,10 @@ public class LoadTestTool extends AbstractHBaseTool {
+ "tool will load n table parallely. -tn parameter value becomes " + "tool will load n table parallely. -tn parameter value becomes "
+ "table name prefix. Each table name is in format <tn>_1...<tn>_n"); + "table name prefix. Each table name is in format <tn>_1...<tn>_n");
addOptWithArg(OPT_REGIONS_PER_SERVER,
"A positive integer number. When a number n is specified, load test "
+ "tool will create the test table with n regions per server");
addOptWithArg(OPT_ENCRYPTION, OPT_ENCRYPTION_USAGE); addOptWithArg(OPT_ENCRYPTION, OPT_ENCRYPTION_USAGE);
} }
@ -399,9 +405,16 @@ public class LoadTestTool extends AbstractHBaseTool {
} }
numTables = 1; numTables = 1;
if(cmd.hasOption(NUM_TABLES)) { if (cmd.hasOption(NUM_TABLES)) {
numTables = parseInt(cmd.getOptionValue(NUM_TABLES), 1, Short.MAX_VALUE); numTables = parseInt(cmd.getOptionValue(NUM_TABLES), 1, Short.MAX_VALUE);
} }
regionsPerServer = HBaseTestingUtility.DEFAULT_REGIONS_PER_SERVER;
if (cmd.hasOption(OPT_REGIONS_PER_SERVER)) {
regionsPerServer = parseInt(cmd.getOptionValue(OPT_REGIONS_PER_SERVER), 1,
Integer.MAX_VALUE);
conf.setInt(HBaseTestingUtility.REGIONS_PER_SERVER_KEY, regionsPerServer);
}
System.out.println("Regions per server: " + regionsPerServer);
} }
private void parseColumnFamilyOptions(CommandLine cmd) { private void parseColumnFamilyOptions(CommandLine cmd) {