HBASE-1902 Let PerformanceEvaluation support setting tableName and compress algorithm

git-svn-id: https://svn.apache.org/repos/asf/hadoop/hbase/trunk@824397 13f79535-47bb-0310-9956-ffa450edef68

CHANGES.txt

@@ -111,6 +111,8 @@ Release 0.21.0 - Unreleased
    HBASE-1899  Use scanner caching in shell count
    HBASE-1887  Update hbase trunk to latests on hadoop 0.21 branch so we can
                all test sync/append
+   HBASE-1902  Let PerformanceEvaluation support setting tableName and compress
+               algorithm (Schubert Zhang via Stack)
 
   OPTIMIZATIONS
 

src/test/org/apache/hadoop/hbase/PerformanceEvaluation.java

@@ -97,8 +97,7 @@ public class PerformanceEvaluation implements HConstants {
   public static final byte [] FAMILY_NAME = Bytes.toBytes("info");
   public static final byte [] QUALIFIER_NAME = Bytes.toBytes("data");
   
-  protected static final HTableDescriptor TABLE_DESCRIPTOR;
+  protected HTableDescriptor TABLE_DESCRIPTOR; {
-  static {
     TABLE_DESCRIPTOR = new HTableDescriptor("TestTable");
     TABLE_DESCRIPTOR.addFamily(new HColumnDescriptor(FAMILY_NAME));
   }
@@ -151,6 +150,7 @@ public class PerformanceEvaluation implements HConstants {
    */
   public PerformanceEvaluation(final HBaseConfiguration c) {
     this.conf = c;
+    setTableDesc();
   }
   
   /**
@@ -165,6 +165,16 @@ public class PerformanceEvaluation implements HConstants {
     void setStatus(final String msg) throws IOException;
   }
   
+  /**
+   * Implementations can have their status set.
+   */
+  protected void setTableDesc() {
+    TABLE_DESCRIPTOR = new HTableDescriptor(conf.get("hbase.pe.tablename", "TestTable"));
+    TABLE_DESCRIPTOR.addFamily(new HColumnDescriptor(FAMILY_NAME, 
+        3, conf.get("hbase.pe.compress", HColumnDescriptor.DEFAULT_COMPRESSION), 
+        false, true, HColumnDescriptor.DEFAULT_TTL, false));
+  }
+  
   /**
    *  This class works as the InputSplit of Performance Evaluation
    *  MapReduce InputFormat, and the Record Value of RecordReader. 
@@ -530,7 +540,7 @@ public class PerformanceEvaluation implements HConstants {
    * A test.
    * Subclass to particularize what happens per row.
    */
-  static abstract class Test {
+  abstract class Test {
     protected final Random rand = new Random(System.currentTimeMillis());
     protected final int startRow;
     protected final int perClientRunRows;
@@ -782,10 +792,30 @@ public class PerformanceEvaluation implements HConstants {
    */  
   public static byte[] generateValue(final Random r) {    
     byte [] b = new byte [ROW_LENGTH];    
-    r.nextBytes(b);
+
+    int i = 0;    
+    for(i = 0; i < (ROW_LENGTH-8); i += 8) {
+      b[i] = (byte) (65 + r.nextInt(26));
+      b[i+1] = b[i];
+      b[i+2] = b[i];
+      b[i+3] = b[i];
+      b[i+4] = b[i];
+      b[i+5] = b[i];
+      b[i+6] = b[i];
+      b[i+7] = b[i];
+    }
+    
+    byte a = (byte) (65 + r.nextInt(26));
+    for(; i < ROW_LENGTH; i++) {
+      b[i] = a;
+    }
+    
     return b;
   }
   
+  
+
+  
   static byte [] getRandomRow(final Random random, final int totalRows) {
     return format(random.nextInt(Integer.MAX_VALUE) % totalRows);
   }
@@ -897,7 +927,8 @@ public class PerformanceEvaluation implements HConstants {
     }
     System.err.println("Usage: java " + this.getClass().getName() +
         " [--master=HOST:PORT] \\");
-    System.err.println("  [--miniCluster] [--nomapred] [--rows=ROWS] <command> <nclients>");
+    System.err.println("  [--miniCluster] [--nomapred] [--rows=ROWS] [--table=NAME] [--compress=TYPE] \\"
+    System.err.println("  <command> <nclients>");
     System.err.println();
     System.err.println("Options:");
     System.err.println(" master          Specify host and port of HBase " +
@@ -907,6 +938,8 @@ public class PerformanceEvaluation implements HConstants {
     System.err.println(" nomapred        Run multiple clients using threads " +
       "(rather than use mapreduce)");
     System.err.println(" rows            Rows each client runs. Default: One million");
+    System.err.println(" table           Alternate table name. Default: 'TestTable'");
+    System.err.println(" compress        Compression type to use. Default: 'NONE'");
     System.err.println();
     System.err.println("Command:");
     System.err.println(" randomRead      Run random read test");
@@ -978,7 +1011,20 @@ public class PerformanceEvaluation implements HConstants {
           continue;
         }
         
+        final String table = "--table=";
+        if (cmd.startsWith(table)) {
+          conf.set("hbase.pe.tablename", cmd.substring(table.length()));
+          continue;
+        }
+        
+        final String compress = "--compress=";
+        if (cmd.startsWith(compress)) {
+          conf.set("hbase.pe.compress", cmd.substring(compress.length()));
+          continue;
+        }
+               
         if (COMMANDS.contains(cmd)) {
+          setTableDesc(); // update table desc
           getArgs(i + 1, args);
           runTest(cmd);
           errCode = 0;