private Long getWriteRequestMetric(String tableName, ServerName serverName) {
Map<String, Long> serverMap = tableServerWriteRequestMap.get(tableName);
if (serverMap == null) {
serverMap = new HashMap<>();
tableServerWriteRequestMap.put(tableName, serverMap);
}
Long writeRequest = serverMap.get(serverName.getServerName());
if (writeRequest == null) {
writeRequest = 0L;
serverMap.put(serverName.getServerName(), writeRequest);
}
return writeRequest;
}
protected void waitForMoving(HRegionInfo hRegionInfo, ServerName serverName) throws Exception {
Map<byte[], HServerLoad.RegionLoad> regionsLoad = null;
for (int i = 0; i < MAX_WAIT_ITERATION; i++) {
HServerLoad load = admin.getClusterStatus().getLoad(serverName);
regionsLoad = load.getRegionsLoad();
for (byte[] regionName : regionsLoad.keySet()) {
if (Arrays.equals(regionName, hRegionInfo.getRegionName())) return;
}
admin.move(hRegionInfo.getEncodedNameAsBytes(), serverName.getServerName().getBytes());
Thread.sleep(WAIT_INTERVAL);
}
System.out.println("hRegionInfo = " + Bytes.toString(hRegionInfo.getRegionName()));
for (Map.Entry<byte[], HServerLoad.RegionLoad> entry : regionsLoad.entrySet()) {
System.out.println(
"regionsLoad = " + Bytes.toString(entry.getKey()) + " - " + entry.getValue());
}
Assert.fail(Util.getMethodName() + " failed");
}
protected void move(HRegionInfo regionInfo, ServerName serverName) throws Exception {
admin.move(regionInfo.getEncodedName().getBytes(), serverName.getServerName().getBytes());
waitForMoving(regionInfo, serverName);
}