1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.apache.hadoop.hbase.chaos.actions;
20
21 import java.util.List;
22
23 import org.apache.hadoop.hbase.HBaseTestingUtility;
24 import org.apache.hadoop.hbase.HRegionInfo;
25 import org.apache.hadoop.hbase.TableName;
26 import org.apache.hadoop.hbase.chaos.monkies.PolicyBasedChaosMonkey;
27 import org.apache.hadoop.hbase.client.Admin;
28
29
30
31
32 public class MoveRandomRegionOfTableAction extends Action {
33 private final long sleepTime;
34 private final TableName tableName;
35
36 public MoveRandomRegionOfTableAction(TableName tableName) {
37 this(-1, tableName);
38 }
39
40 public MoveRandomRegionOfTableAction(long sleepTime, TableName tableName) {
41 this.sleepTime = sleepTime;
42 this.tableName = tableName;
43 }
44
45 @Override
46 public void perform() throws Exception {
47 if (sleepTime > 0) {
48 Thread.sleep(sleepTime);
49 }
50
51 HBaseTestingUtility util = context.getHBaseIntegrationTestingUtility();
52 Admin admin = util.getHBaseAdmin();
53
54 LOG.info("Performing action: Move random region of table " + tableName);
55 List<HRegionInfo> regions = admin.getTableRegions(tableName);
56 if (regions == null || regions.isEmpty()) {
57 LOG.info("Table " + tableName + " doesn't have regions to move");
58 return;
59 }
60
61 HRegionInfo region = PolicyBasedChaosMonkey.selectRandomItem(
62 regions.toArray(new HRegionInfo[regions.size()]));
63 LOG.debug("Unassigning region " + region.getRegionNameAsString());
64 admin.unassign(region.getRegionName(), false);
65 if (sleepTime > 0) {
66 Thread.sleep(sleepTime);
67 }
68 }
69 }