diff --git a/hbase-server/src/main/java/org/apache/hadoop/hbase/master/AssignmentManager.java b/hbase-server/src/main/java/org/apache/hadoop/hbase/master/AssignmentManager.java index 0d68382..1d184bf 100644 --- a/hbase-server/src/main/java/org/apache/hadoop/hbase/master/AssignmentManager.java +++ b/hbase-server/src/main/java/org/apache/hadoop/hbase/master/AssignmentManager.java @@ -848,7 +848,7 @@ public class AssignmentManager extends ZooKeeperListener { case M_ZK_REGION_OFFLINE: // Insert in RIT and resend to the regionserver - regionStates.updateRegionState(rt, State.PENDING_OPEN); + regionStates.updateRegionState(rt, State.OFFLINE); final RegionState rsOffline = regionStates.getRegionState(regionInfo); this.executorService.submit( new EventHandler(server, EventType.M_MASTER_RECOVERY) { @@ -858,7 +858,7 @@ public class AssignmentManager extends ZooKeeperListener { try { RegionPlan plan = new RegionPlan(regionInfo, null, sn); addPlan(encodedName, plan); - assign(rsOffline, false, false); + assign(rsOffline, true, false); } finally { lock.unlock(); } @@ -1577,6 +1577,7 @@ public class AssignmentManager extends ZooKeeperListener { /** * Use care with forceNewPlan. It could cause double assignment. */ + @VisibleForTesting public void assign(HRegionInfo region, boolean setOfflineInZK, boolean forceNewPlan) { if (isDisabledorDisablingRegionInRIT(region)) { @@ -2054,7 +2055,7 @@ public class AssignmentManager extends ZooKeeperListener { * @param setOfflineInZK * @param forceNewPlan */ - private void assign(RegionState state, + public void assign(RegionState state, boolean setOfflineInZK, final boolean forceNewPlan) { long startTime = EnvironmentEdgeManager.currentTime(); try { diff --git a/hbase-server/src/test/java/org/apache/hadoop/hbase/master/TestAssignmentManager.java b/hbase-server/src/test/java/org/apache/hadoop/hbase/master/TestAssignmentManager.java index 0dd1ff2..d585756 100644 --- a/hbase-server/src/test/java/org/apache/hadoop/hbase/master/TestAssignmentManager.java +++ b/hbase-server/src/test/java/org/apache/hadoop/hbase/master/TestAssignmentManager.java @@ -859,6 +859,7 @@ public class TestAssignmentManager { } } + /** * Mocked load balancer class used in the testcase to make sure that the testcase waits until * random assignment is called and the gate variable is set to true. @@ -929,6 +930,22 @@ public class TestAssignmentManager { am.shutdown(); } + @Test(timeout = 600000) + public void testAssignmentOfRegionRITWithOffline() throws IOException, + KeeperException, ServiceException, CoordinatedStateException, InterruptedException { + AssignmentManagerWithExtrasForTesting am = setUpMockedAssignmentManager( + this.server, this.serverManager); + ZKAssign.createNodeOffline(this.watcher, REGIONINFO, SERVERNAME_B); + am.gate.set(false); + // join the cluster - that's when the AM is really kicking in after a restart + am.joinCluster(); + while (!am.gate.get()) { + Thread.sleep(10); + } + assertTrue(am.getRegionStates().getRegionState(REGIONINFO).getState() + == RegionState.State.PENDING_OPEN); + am.shutdown(); + } /** * Test the scenario when the master is in failover and trying to process a * region which is in Opening state on a dead RS. Master will force offline the @@ -1341,6 +1358,16 @@ public class TestAssignmentManager { this.gate.set(true); } } + @Override + public void assign(RegionState state, + boolean setOfflineInZK, final boolean forceNewPlan) { + if (enabling) { + assignmentCount++; + } else { + super.assign(state, setOfflineInZK, forceNewPlan); + this.gate.set(true); + } + } @Override boolean assign(ServerName destination, List regions)