Skip to content

Commit 15c28d0

Browse files
Apache9Apache9
committed
HBASE-22971 Deprecated RSGroupAdminEndpoint and make RSGroup feature always enabled (#595)
Signed-off-by: Guanghao Zhang <zghao@apache.org>
1 parent 01485b9 commit 15c28d0

32 files changed

+241
-276
lines changed

‎hbase-server/src/main/java/org/apache/hadoop/hbase/favored/FavoredNodeLoadBalancer.java‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -321,6 +321,7 @@ private void addRegionToMap(Map<ServerName, List<RegionInfo>> assignmentMapForFa
321321
regionsOnServer.add(region);
322322
}
323323

324+
@Override
324325
public synchronized List<ServerName> getFavoredNodes(RegionInfo regionInfo) {
325326
return this.fnm.getFavoredNodes(regionInfo);
326327
}

‎hbase-server/src/main/java/org/apache/hadoop/hbase/favored/FavoredNodesPromoter.java‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,4 +35,6 @@ void generateFavoredNodesForDaughter(List<ServerName> servers,
3535

3636
void generateFavoredNodesForMergedRegion(RegionInfo merged, RegionInfo [] mergeParents)
3737
throws IOException;
38+
39+
List<ServerName> getFavoredNodes(RegionInfo regionInfo);
3840
}

‎hbase-server/src/main/java/org/apache/hadoop/hbase/master/HMaster.java‎

Lines changed: 19 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,6 @@
9191
import org.apache.hadoop.hbase.exceptions.DeserializationException;
9292
import org.apache.hadoop.hbase.executor.ExecutorType;
9393
import org.apache.hadoop.hbase.favored.FavoredNodesManager;
94-
import org.apache.hadoop.hbase.favored.FavoredNodesPromoter;
9594
import org.apache.hadoop.hbase.http.InfoServer;
9695
import org.apache.hadoop.hbase.ipc.CoprocessorRpcUtils;
9796
import org.apache.hadoop.hbase.ipc.RpcServer;
@@ -107,7 +106,6 @@
107106
import org.apache.hadoop.hbase.master.assignment.TransitRegionStateProcedure;
108107
import org.apache.hadoop.hbase.master.assignment.UnassignProcedure;
109108
import org.apache.hadoop.hbase.master.balancer.BalancerChore;
110-
import org.apache.hadoop.hbase.master.balancer.BaseLoadBalancer;
111109
import org.apache.hadoop.hbase.master.balancer.ClusterStatusChore;
112110
import org.apache.hadoop.hbase.master.balancer.LoadBalancerFactory;
113111
import org.apache.hadoop.hbase.master.cleaner.DirScanPool;
@@ -187,6 +185,7 @@
187185
import org.apache.hadoop.hbase.replication.master.ReplicationLogCleaner;
188186
import org.apache.hadoop.hbase.replication.master.ReplicationPeerConfigUpgrader;
189187
import org.apache.hadoop.hbase.replication.regionserver.ReplicationStatus;
188+
import org.apache.hadoop.hbase.rsgroup.RSGroupBasedLoadBalancer;
190189
import org.apache.hadoop.hbase.rsgroup.RSGroupInfoManager;
191190
import org.apache.hadoop.hbase.security.AccessDeniedException;
192191
import org.apache.hadoop.hbase.security.SecurityConstants;
@@ -384,7 +383,7 @@ public void run() {
384383

385384
private final LockManager lockManager = new LockManager(this);
386385

387-
private LoadBalancer balancer;
386+
private RSGroupBasedLoadBalancer balancer;
388387
private RegionNormalizer normalizer;
389388
private BalancerChore balancerChore;
390389
private RegionNormalizerChore normalizerChore;
@@ -439,9 +438,6 @@ public void run() {
439438
private long splitPlanCount;
440439
private long mergePlanCount;
441440

442-
/* Handle favored nodes information */
443-
private FavoredNodesManager favoredNodesManager;
444-
445441
/** jetty server for master to redirect requests to regionserver infoServer */
446442
private Server masterJettyServer;
447443

@@ -766,7 +762,8 @@ public MetricsMaster getMasterMetrics() {
766762
@VisibleForTesting
767763
protected void initializeZKBasedSystemTrackers()
768764
throws IOException, InterruptedException, KeeperException, ReplicationException {
769-
this.balancer = LoadBalancerFactory.getLoadBalancer(conf);
765+
this.balancer = new RSGroupBasedLoadBalancer();
766+
this.balancer.setConf(conf);
770767
this.normalizer = RegionNormalizerFactory.getRegionNormalizer(conf);
771768
this.normalizer.setMasterServices(this);
772769
this.normalizer.setMasterRpcServices((MasterRpcServices)rpcServices);
@@ -1050,9 +1047,6 @@ private void finishActiveMasterInitialization(MonitoredTask status) throws IOExc
10501047
return temp;
10511048
});
10521049
}
1053-
if (this.balancer instanceof FavoredNodesPromoter) {
1054-
favoredNodesManager = new FavoredNodesManager(this);
1055-
}
10561050

10571051
// initialize load balancer
10581052
this.balancer.setMasterServices(this);
@@ -1102,11 +1096,11 @@ private void finishActiveMasterInitialization(MonitoredTask status) throws IOExc
11021096
// table states messing up master launch (namespace table, etc., are not assigned).
11031097
this.assignmentManager.processOfflineRegions();
11041098
// Initialize after meta is up as below scans meta
1105-
if (favoredNodesManager != null && !maintenanceMode) {
1099+
if (getFavoredNodesManager() != null && !maintenanceMode) {
11061100
SnapshotOfRegionAssignmentFromMeta snapshotOfRegionAssignment =
11071101
new SnapshotOfRegionAssignmentFromMeta(getConnection());
11081102
snapshotOfRegionAssignment.initialize();
1109-
favoredNodesManager.initialize(snapshotOfRegionAssignment);
1103+
getFavoredNodesManager().initialize(snapshotOfRegionAssignment);
11101104
}
11111105

11121106
// set cluster status again after user regions are assigned
@@ -2038,14 +2032,13 @@ public void move(final byte[] encodedRegionName, byte[] destServerName) throws I
20382032
LOG.debug("Unable to determine a plan to assign " + hri);
20392033
return;
20402034
}
2041-
// TODO: What is this? I don't get it.
2042-
if (dest.equals(serverName) && balancer instanceof BaseLoadBalancer
2043-
&& !((BaseLoadBalancer)balancer).shouldBeOnMaster(hri)) {
2035+
// TODO: deal with table on master for rs group.
2036+
if (dest.equals(serverName)) {
20442037
// To avoid unnecessary region moving later by balancer. Don't put user
20452038
// regions on master.
2046-
LOG.debug("Skipping move of region " + hri.getRegionNameAsString()
2047-
+ " to avoid unnecessary region moving later by load balancer,"
2048-
+ " because it should not be on master");
2039+
LOG.debug("Skipping move of region " + hri.getRegionNameAsString() +
2040+
" to avoid unnecessary region moving later by load balancer," +
2041+
" because it should not be on master");
20492042
return;
20502043
}
20512044
}
@@ -3497,12 +3490,14 @@ public boolean isSplitOrMergeEnabled(MasterSwitchType switchType) {
34973490

34983491
/**
34993492
* Fetch the configured {@link LoadBalancer} class name. If none is set, a default is returned.
3500-
*
3493+
* <p/>
3494+
* Notice that, the base load balancer will always be {@link RSGroupBasedLoadBalancer} now, so
3495+
* this method will return the balancer used inside each rs group.
35013496
* @return The name of the {@link LoadBalancer} in use.
35023497
*/
35033498
public String getLoadBalancerClassName() {
3504-
return conf.get(HConstants.HBASE_MASTER_LOADBALANCER_CLASS, LoadBalancerFactory
3505-
.getDefaultLoadBalancerClass().getName());
3499+
return conf.get(HConstants.HBASE_MASTER_LOADBALANCER_CLASS,
3500+
LoadBalancerFactory.getDefaultLoadBalancerClass().getName());
35063501
}
35073502

35083503
/**
@@ -3517,13 +3512,13 @@ public SplitOrMergeTracker getSplitOrMergeTracker() {
35173512
}
35183513

35193514
@Override
3520-
public LoadBalancer getLoadBalancer() {
3515+
public RSGroupBasedLoadBalancer getLoadBalancer() {
35213516
return balancer;
35223517
}
35233518

35243519
@Override
35253520
public FavoredNodesManager getFavoredNodesManager() {
3526-
return favoredNodesManager;
3521+
return balancer.getFavoredNodesManager();
35273522
}
35283523

35293524
private long executePeerProcedure(AbstractPeerProcedure<?> procedure) throws IOException {
@@ -3809,7 +3804,7 @@ public HbckChore getHbckChore() {
38093804
}
38103805

38113806
@Override
3812-
public RSGroupInfoManager getRSRSGroupInfoManager() {
3807+
public RSGroupInfoManager getRSGroupInfoManager() {
38133808
return rsGroupInfoManager;
38143809
}
38153810
}

‎hbase-server/src/main/java/org/apache/hadoop/hbase/master/LoadBalancer.java‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,15 @@ public interface LoadBalancer extends Configurable, Stoppable, ConfigurationObse
6464
// We deliberately use 'localhost' so the operation will fail fast
6565
ServerName BOGUS_SERVER_NAME = ServerName.valueOf("localhost,1,1");
6666

67+
/**
68+
* Config for pluggable load balancers.
69+
* @deprecated since 3.0.0, will be removed in 4.0.0. In the new implementation, as the base load
70+
* balancer will always be the rs group based one, you should just use
71+
* {@link org.apache.hadoop.hbase.HConstants#HBASE_MASTER_LOADBALANCER_CLASS} to
72+
* config the per group load balancer.
73+
*/
74+
@Deprecated
75+
String HBASE_RSGROUP_LOADBALANCER_CLASS = "hbase.rsgroup.grouploadbalancer.class";
6776
/**
6877
* Set the current cluster status. This allows a LoadBalancer to map host name to a server
6978
*/

‎hbase-server/src/main/java/org/apache/hadoop/hbase/master/MasterRpcServices.java‎

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@
7474
import org.apache.hadoop.hbase.master.procedure.MasterProcedureUtil.NonceProcedureRunnable;
7575
import org.apache.hadoop.hbase.master.procedure.ServerCrashProcedure;
7676
import org.apache.hadoop.hbase.mob.MobUtils;
77+
import org.apache.hadoop.hbase.net.Address;
7778
import org.apache.hadoop.hbase.procedure.MasterProcedureManager;
7879
import org.apache.hadoop.hbase.procedure2.LockType;
7980
import org.apache.hadoop.hbase.procedure2.LockedResource;
@@ -219,10 +220,8 @@
219220
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.IsNormalizerEnabledResponse;
220221
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.IsProcedureDoneRequest;
221222
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.IsProcedureDoneResponse;
222-
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos
223-
.IsSnapshotCleanupEnabledRequest;
224-
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos
225-
.IsSnapshotCleanupEnabledResponse;
223+
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.IsSnapshotCleanupEnabledRequest;
224+
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.IsSnapshotCleanupEnabledResponse;
226225
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.IsSnapshotDoneRequest;
227226
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.IsSnapshotDoneResponse;
228227
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.IsSplitOrMergeEnabledRequest;
@@ -275,10 +274,8 @@
275274
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.SetNormalizerRunningResponse;
276275
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.SetQuotaRequest;
277276
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.SetQuotaResponse;
278-
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos
279-
.SetSnapshotCleanupRequest;
280-
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos
281-
.SetSnapshotCleanupResponse;
277+
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.SetSnapshotCleanupRequest;
278+
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.SetSnapshotCleanupResponse;
282279
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.SetSplitOrMergeEnabledRequest;
283280
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.SetSplitOrMergeEnabledResponse;
284281
import org.apache.hadoop.hbase.shaded.protobuf.generated.MasterProtos.SetTableStateInMetaRequest;
@@ -2380,12 +2377,18 @@ public ClearDeadServersResponse clearDeadServers(RpcController controller,
23802377
LOG.debug("Some dead server is still under processing, won't clear the dead server list");
23812378
response.addAllServerName(request.getServerNameList());
23822379
} else {
2380+
DeadServer deadServer = master.getServerManager().getDeadServers();
2381+
Set<Address> clearedServers = new HashSet<>();
23832382
for (HBaseProtos.ServerName pbServer : request.getServerNameList()) {
2384-
if (!master.getServerManager().getDeadServers()
2385-
.removeDeadServer(ProtobufUtil.toServerName(pbServer))) {
2383+
ServerName server = ProtobufUtil.toServerName(pbServer);
2384+
if (!deadServer.removeDeadServer(server)) {
23862385
response.addServerName(pbServer);
2386+
} else {
2387+
clearedServers.add(server.getAddress());
23872388
}
23882389
}
2390+
master.getRSGroupInfoManager().removeServers(clearedServers);
2391+
LOG.info("Remove decommissioned servers {} from RSGroup done", clearedServers);
23892392
}
23902393

23912394
if (master.cpHost != null) {

‎hbase-server/src/main/java/org/apache/hadoop/hbase/master/MasterServices.java‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -539,5 +539,5 @@ default SplitWALManager getSplitWALManager(){
539539
/**
540540
* @return the {@link RSGroupInfoManager}
541541
*/
542-
RSGroupInfoManager getRSRSGroupInfoManager();
542+
RSGroupInfoManager getRSGroupInfoManager();
543543
}

‎hbase-server/src/main/java/org/apache/hadoop/hbase/master/assignment/AssignmentManager.java‎

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,7 @@
6868
import org.apache.hadoop.hbase.procedure2.ProcedureInMemoryChore;
6969
import org.apache.hadoop.hbase.procedure2.util.StringUtils;
7070
import org.apache.hadoop.hbase.regionserver.SequenceId;
71+
import org.apache.hadoop.hbase.rsgroup.RSGroupBasedLoadBalancer;
7172
import org.apache.hadoop.hbase.util.Bytes;
7273
import org.apache.hadoop.hbase.util.EnvironmentEdgeManager;
7374
import org.apache.hadoop.hbase.util.HasThread;
@@ -325,6 +326,11 @@ private LoadBalancer getBalancer() {
325326
return master.getLoadBalancer();
326327
}
327328

329+
private FavoredNodesPromoter getFavoredNodePromoter() {
330+
return (FavoredNodesPromoter) ((RSGroupBasedLoadBalancer) master.getLoadBalancer())
331+
.getInternalBalancer();
332+
}
333+
328334
private MasterProcedureEnv getProcedureEnvironment() {
329335
return master.getMasterProcedureExecutor().getEnvironment();
330336
}
@@ -368,7 +374,7 @@ public RegionStateStore getRegionStateStore() {
368374

369375
public List<ServerName> getFavoredNodes(final RegionInfo regionInfo) {
370376
return this.shouldAssignRegionsWithFavoredNodes
371-
? ((FavoredStochasticBalancer) getBalancer()).getFavoredNodes(regionInfo)
377+
? getFavoredNodePromoter().getFavoredNodes(regionInfo)
372378
: ServerName.EMPTY_SERVER_LIST;
373379
}
374380

@@ -1774,8 +1780,8 @@ public void markRegionAsSplit(final RegionInfo parent, final ServerName serverNa
17741780
regionStateStore.splitRegion(parent, daughterA, daughterB, serverName);
17751781
if (shouldAssignFavoredNodes(parent)) {
17761782
List<ServerName> onlineServers = this.master.getServerManager().getOnlineServersList();
1777-
((FavoredNodesPromoter)getBalancer()).
1778-
generateFavoredNodesForDaughter(onlineServers, parent, daughterA, daughterB);
1783+
getFavoredNodePromoter().generateFavoredNodesForDaughter(onlineServers, parent, daughterA,
1784+
daughterB);
17791785
}
17801786
}
17811787

@@ -1800,8 +1806,7 @@ public void markRegionAsMerged(final RegionInfo child, final ServerName serverNa
18001806
}
18011807
regionStateStore.mergeRegions(child, mergeParents, serverName);
18021808
if (shouldAssignFavoredNodes(child)) {
1803-
((FavoredNodesPromoter)getBalancer()).
1804-
generateFavoredNodesForMergedRegion(child, mergeParents);
1809+
getFavoredNodePromoter().generateFavoredNodesForMergedRegion(child, mergeParents);
18051810
}
18061811
}
18071812

‎hbase-server/src/main/java/org/apache/hadoop/hbase/master/balancer/FavoredStochasticBalancer.java‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -473,6 +473,7 @@ private List<ServerName> getOnlineFavoredNodes(List<ServerName> onlineServers,
473473
}
474474
}
475475

476+
@Override
476477
public synchronized List<ServerName> getFavoredNodes(RegionInfo regionInfo) {
477478
return this.fnm.getFavoredNodes(regionInfo);
478479
}

‎hbase-server/src/main/java/org/apache/hadoop/hbase/master/balancer/LoadBalancerFactory.java‎

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,11 @@
1717
*/
1818
package org.apache.hadoop.hbase.master.balancer;
1919

20-
import org.apache.yetus.audience.InterfaceAudience;
2120
import org.apache.hadoop.conf.Configuration;
2221
import org.apache.hadoop.hbase.HConstants;
2322
import org.apache.hadoop.hbase.master.LoadBalancer;
24-
import org.apache.hadoop.util.ReflectionUtils;
23+
import org.apache.hadoop.hbase.util.ReflectionUtils;
24+
import org.apache.yetus.audience.InterfaceAudience;
2525

2626
/**
2727
* The class that creates a load balancer from a conf.
@@ -30,8 +30,7 @@
3030
public class LoadBalancerFactory {
3131

3232
/**
33-
* The default {@link LoadBalancer} class.
34-
*
33+
* The default {@link LoadBalancer} class.
3534
* @return The Class for the default {@link LoadBalancer}.
3635
*/
3736
public static Class<? extends LoadBalancer> getDefaultLoadBalancerClass() {
@@ -40,16 +39,15 @@ public static Class<? extends LoadBalancer> getDefaultLoadBalancerClass() {
4039

4140
/**
4241
* Create a loadbalancer from the given conf.
43-
* @param conf
4442
* @return A {@link LoadBalancer}
4543
*/
4644
public static LoadBalancer getLoadBalancer(Configuration conf) {
47-
4845
// Create the balancer
4946
Class<? extends LoadBalancer> balancerKlass =
50-
conf.getClass(HConstants.HBASE_MASTER_LOADBALANCER_CLASS, getDefaultLoadBalancerClass(),
51-
LoadBalancer.class);
52-
return ReflectionUtils.newInstance(balancerKlass, conf);
53-
47+
conf.getClass(HConstants.HBASE_MASTER_LOADBALANCER_CLASS, getDefaultLoadBalancerClass(),
48+
LoadBalancer.class);
49+
LoadBalancer balancer = ReflectionUtils.newInstance(balancerKlass);
50+
balancer.setConf(conf);
51+
return balancer;
5452
}
5553
}

‎hbase-server/src/main/java/org/apache/hadoop/hbase/master/procedure/AbstractStateMachineNamespaceProcedure.java‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
package org.apache.hadoop.hbase.master.procedure;
2020

2121
import java.io.IOException;
22+
import java.util.function.Supplier;
2223
import org.apache.hadoop.hbase.NamespaceDescriptor;
2324
import org.apache.hadoop.hbase.TableName;
2425
import org.apache.hadoop.hbase.master.MasterFileSystem;
@@ -28,6 +29,7 @@
2829
import org.apache.hbase.thirdparty.com.google.common.annotations.VisibleForTesting;
2930

3031
import org.apache.hadoop.hbase.procedure2.StateMachineProcedure;
32+
import org.apache.hadoop.hbase.rsgroup.RSGroupInfo;
3133
import org.apache.hadoop.hbase.util.FSUtils;
3234

3335
/**
@@ -122,4 +124,13 @@ public static void createDirectory(MasterFileSystem mfs, NamespaceDescriptor nsD
122124
protected void releaseSyncLatch() {
123125
ProcedurePrepareLatch.releaseLatch(syncLatch, this);
124126
}
127+
128+
protected final void checkNamespaceRSGroup(MasterProcedureEnv env, NamespaceDescriptor nd)
129+
throws IOException {
130+
Supplier<String> forWhom = () -> "namespace " + nd.getName();
131+
RSGroupInfo rsGroupInfo = MasterProcedureUtil.checkGroupExists(
132+
env.getMasterServices().getRSGroupInfoManager()::getRSGroup,
133+
MasterProcedureUtil.getNamespaceGroup(nd), forWhom);
134+
MasterProcedureUtil.checkGroupNotEmpty(rsGroupInfo, forWhom);
135+
}
125136
}

0 commit comments

Comments
 (0)