001/** 002 * Licensed to the Apache Software Foundation (ASF) under one 003 * or more contributor license agreements. See the NOTICE file 004 * distributed with this work for additional information 005 * regarding copyright ownership. The ASF licenses this file 006 * to you under the Apache License, Version 2.0 (the 007 * "License"); you may not use this file except in compliance 008 * with the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, software 013 * distributed under the License is distributed on an "AS IS" BASIS, 014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 015 * See the License for the specific language governing permissions and 016 * limitations under the License. 017 */ 018package org.apache.hadoop.hbase.chaos.actions; 019 020import java.io.IOException; 021import java.util.concurrent.ThreadLocalRandom; 022 023import org.apache.hadoop.hbase.HBaseTestingUtility; 024import org.apache.hadoop.hbase.TableName; 025import org.apache.hadoop.hbase.client.Admin; 026import org.slf4j.Logger; 027import org.slf4j.LoggerFactory; 028 029 030public class SplitAllRegionOfTableAction extends Action { 031 private static final Logger LOG = 032 LoggerFactory.getLogger(SplitAllRegionOfTableAction.class); 033 private static final int DEFAULT_MAX_SPLITS = 3; 034 private static final String MAX_SPLIT_KEY = "hbase.chaosmonkey.action.maxFullTableSplits"; 035 036 private final TableName tableName; 037 private int maxFullTableSplits = DEFAULT_MAX_SPLITS; 038 private int splits = 0; 039 040 public SplitAllRegionOfTableAction(TableName tableName) { 041 this.tableName = tableName; 042 } 043 044 045 public void init(ActionContext context) throws IOException { 046 super.init(context); 047 this.maxFullTableSplits = getConf().getInt(MAX_SPLIT_KEY, DEFAULT_MAX_SPLITS); 048 } 049 050 @Override 051 public void perform() throws Exception { 052 HBaseTestingUtility util = context.getHBaseIntegrationTestingUtility(); 053 Admin admin = util.getAdmin(); 054 // Don't try the split if we're stopping 055 if (context.isStopping()) { 056 return; 057 } 058 059 060 // Don't always split. This should allow splitting of a full table later in the run 061 if (ThreadLocalRandom.current().nextDouble() 062 < (((double) splits) / ((double) maxFullTableSplits)) / ((double) 2)) { 063 splits++; 064 LOG.info("Performing action: Split all regions of " + tableName); 065 admin.split(tableName); 066 } else { 067 LOG.info("Skipping split of all regions."); 068 } 069 } 070}