svn commit: r1542324 - in /hbase/branches/0.96/hbase-it/src/test/java/org/apache/hadoop/hbase/chaos: actions/ChangeCompressionAction.java factories/SlowDeterministicMonkeyFactory.java

[jx...@apache.org]

Author: jxiang
Date: Fri Nov 15 16:56:35 2013
New Revision: 1542324

URL: http://svn.apache.org/r1542324
Log:
HBASE-9947 Add CM action for online compression algorithm change

Added:
    hbase/branches/0.96/hbase-it/src/test/java/org/apache/hadoop/hbase/chaos/actions/ChangeCompressionAction.java   (with props)
Modified:
    hbase/branches/0.96/hbase-it/src/test/java/org/apache/hadoop/hbase/chaos/factories/SlowDeterministicMonkeyFactory.java

Added: hbase/branches/0.96/hbase-it/src/test/java/org/apache/hadoop/hbase/chaos/actions/ChangeCompressionAction.java
URL: http://svn.apache.org/viewvc/hbase/branches/0.96/hbase-it/src/test/java/org/apache/hadoop/hbase/chaos/actions/ChangeCompressionAction.java?rev=1542324&view=auto
==============================================================================
--- hbase/branches/0.96/hbase-it/src/test/java/org/apache/hadoop/hbase/chaos/actions/ChangeCompressionAction.java (added)
+++ hbase/branches/0.96/hbase-it/src/test/java/org/apache/hadoop/hbase/chaos/actions/ChangeCompressionAction.java Fri Nov 15 16:56:35 2013
@@ -0,0 +1,81 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.hadoop.hbase.chaos.actions;
+
+import java.io.IOException;
+import java.util.Random;
+
+import org.apache.hadoop.hbase.HColumnDescriptor;
+import org.apache.hadoop.hbase.HTableDescriptor;
+import org.apache.hadoop.hbase.client.HBaseAdmin;
+import org.apache.hadoop.hbase.io.compress.Compression.Algorithm;
+import org.apache.hadoop.hbase.util.Bytes;
+
+/**
+ * Action that changes the compression algorithm on a column family from a list of tables.
+ */
+public class ChangeCompressionAction extends Action {
+  private final byte[] tableName;
+  private final String tableNameString;
+
+  private HBaseAdmin admin;
+  private Random random;
+
+  public ChangeCompressionAction(String tableName) {
+    tableNameString = tableName;
+    this.tableName = Bytes.toBytes(tableName);
+    this.random = new Random();
+  }
+
+  @Override
+  public void init(ActionContext context) throws IOException {
+    super.init(context);
+    this.admin = context.getHBaseIntegrationTestingUtility().getHBaseAdmin();
+  }
+
+  @Override
+  public void perform() throws Exception {
+    HTableDescriptor tableDescriptor = admin.getTableDescriptor(tableName);
+    HColumnDescriptor[] columnDescriptors = tableDescriptor.getColumnFamilies();
+
+    if (columnDescriptors == null || columnDescriptors.length == 0) {
+      return;
+    }
+
+    // Possible compression algorithms. If an algorithm is not supported,
+    // modifyTable will fail, so there is no harm.
+    Algorithm[] possibleAlgos = Algorithm.values();
+
+    // Since not every compression algorithm is supported,
+    // let's use the same algorithm for all column families.
+    Algorithm algo = possibleAlgos[random.nextInt(possibleAlgos.length)];
+
+    LOG.debug("Performing action: Changing compression algorithms on "
+      + tableNameString + " to " + algo);
+    for (HColumnDescriptor descriptor : columnDescriptors) {
+      if (random.nextBoolean()) {
+        descriptor.setCompactionCompressionType(algo);
+      } else {
+        descriptor.setCompressionType(algo);
+      }
+    }
+
+    admin.modifyTable(tableName, tableDescriptor);
+  }
+}

Propchange: hbase/branches/0.96/hbase-it/src/test/java/org/apache/hadoop/hbase/chaos/actions/ChangeCompressionAction.java
------------------------------------------------------------------------------
    svn:eol-style = native

Modified: hbase/branches/0.96/hbase-it/src/test/java/org/apache/hadoop/hbase/chaos/factories/SlowDeterministicMonkeyFactory.java
URL: http://svn.apache.org/viewvc/hbase/branches/0.96/hbase-it/src/test/java/org/apache/hadoop/hbase/chaos/factories/SlowDeterministicMonkeyFactory.java?rev=1542324&r1=1542323&r2=1542324&view=diff
==============================================================================
--- hbase/branches/0.96/hbase-it/src/test/java/org/apache/hadoop/hbase/chaos/factories/SlowDeterministicMonkeyFactory.java (original)
+++ hbase/branches/0.96/hbase-it/src/test/java/org/apache/hadoop/hbase/chaos/factories/SlowDeterministicMonkeyFactory.java Fri Nov 15 16:56:35 2013
@@ -21,6 +21,7 @@ package org.apache.hadoop.hbase.chaos.fa
 import org.apache.hadoop.hbase.chaos.actions.Action;
 import org.apache.hadoop.hbase.chaos.actions.AddColumnAction;
 import org.apache.hadoop.hbase.chaos.actions.BatchRestartRsAction;
+import org.apache.hadoop.hbase.chaos.actions.ChangeCompressionAction;
 import org.apache.hadoop.hbase.chaos.actions.ChangeEncodingAction;
 import org.apache.hadoop.hbase.chaos.actions.ChangeVersionsAction;
 import org.apache.hadoop.hbase.chaos.actions.CompactRandomRegionOfTableAction;
@@ -69,6 +70,7 @@ public class SlowDeterministicMonkeyFact
         new AddColumnAction(tableName),
         new RemoveColumnAction(tableName, columnFamilies),
         new ChangeEncodingAction(tableName),
+        new ChangeCompressionAction(tableName),
         new ChangeVersionsAction(tableName)
     };