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();
+        }
     }
 
 }

Reply via email to