KYLIN-1636 HBase benchmark, add param for drop table
Project: http://git-wip-us.apache.org/repos/asf/kylin/repo Commit: http://git-wip-us.apache.org/repos/asf/kylin/commit/d11055c5 Tree: http://git-wip-us.apache.org/repos/asf/kylin/tree/d11055c5 Diff: http://git-wip-us.apache.org/repos/asf/kylin/diff/d11055c5 Branch: refs/heads/1.5.x-HBase1.x Commit: d11055c5dd4fae00e147d565b777347dc26cee4e Parents: e767aa4 Author: lidongsjtu <lid...@apache.org> Authored: Thu May 5 13:13:25 2016 +0800 Committer: lidongsjtu <lid...@apache.org> Committed: Thu May 5 13:13:25 2016 +0800 ---------------------------------------------------------------------- .../hbase/cube/HBaseScannerBenchmark.java | 26 ++++++++++++++------ 1 file changed, 18 insertions(+), 8 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/kylin/blob/d11055c5/storage-hbase/src/main/java/org/apache/kylin/storage/hbase/cube/HBaseScannerBenchmark.java ---------------------------------------------------------------------- diff --git a/storage-hbase/src/main/java/org/apache/kylin/storage/hbase/cube/HBaseScannerBenchmark.java b/storage-hbase/src/main/java/org/apache/kylin/storage/hbase/cube/HBaseScannerBenchmark.java index 5744ea3..969c399 100644 --- a/storage-hbase/src/main/java/org/apache/kylin/storage/hbase/cube/HBaseScannerBenchmark.java +++ b/storage-hbase/src/main/java/org/apache/kylin/storage/hbase/cube/HBaseScannerBenchmark.java @@ -56,7 +56,7 @@ public class HBaseScannerBenchmark { final TableName htableName = TableName.valueOf("HBaseScannerBenchmark"); final SimpleHBaseStore simpleStore; - public HBaseScannerBenchmark(boolean createTable) throws IOException { + public HBaseScannerBenchmark() throws IOException { Builder builder = GTInfo.builder(); builder.setCodeSystem(new GTSampleCodeSystem()); DataType tint = DataType.getType("int4"); @@ -75,10 +75,6 @@ public class HBaseScannerBenchmark { gen.addMeasure(8); simpleStore = new SimpleHBaseStore(info, htableName); - - if (createTable) { - buildTable(); - } } private void buildTable() throws IOException { @@ -154,18 +150,32 @@ public class HBaseScannerBenchmark { public static void main(String[] args) throws IOException { boolean createTable = true; - if (args != null && args.length > 0) { + boolean deleteTable = true; + if (args != null && args.length > 1) { try { createTable = Boolean.parseBoolean(args[0]); } catch (Exception e) { createTable = true; } + + try { + deleteTable = Boolean.parseBoolean(args[1]); + } catch (Exception e) { + deleteTable = true; + } } + KylinConfig.setSandboxEnvIfPossible(); - HBaseScannerBenchmark benchmark = new HBaseScannerBenchmark(createTable); + HBaseScannerBenchmark benchmark = new HBaseScannerBenchmark(); + if (createTable) { + benchmark.buildTable(); + } + benchmark.testScan(); - benchmark.cleanup(); + if (deleteTable) { + benchmark.cleanup(); + } } }