Skip to content

Commit b274144

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 5d4786e commit b274144

32 files changed

Lines changed: 241 additions & 276 deletions

‎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;
@@ -188,6 +186,7 @@
188186
import org.apache.hadoop.hbase.replication.master.ReplicationLogCleaner;
189187
import org.apache.hadoop.hbase.replication.master.ReplicationPeerConfigUpgrader;
190188
import org.apache.hadoop.hbase.replication.regionserver.ReplicationStatus;
189+
import org.apache.hadoop.hbase.rsgroup.RSGroupBasedLoadBalancer;
191190
import org.apache.hadoop.hbase.rsgroup.RSGroupInfoManager;
192191
import org.apache.hadoop.hbase.security.AccessDeniedException;
193192
import org.apache.hadoop.hbase.security.SecurityConstants;
@@ -385,7 +384,7 @@ public void run() {
385384

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

388-
private LoadBalancer balancer;
387+
private RSGroupBasedLoadBalancer balancer;
389388
private RegionNormalizer normalizer;
390389
private BalancerChore balancerChore;
391390
private RegionNormalizerChore normalizerChore;
@@ -441,9 +440,6 @@ public void run() {
441440
private long splitPlanCount;
442441
private long mergePlanCount;
443442

444-
/* Handle favored nodes information */
445-
private FavoredNodesManager favoredNodesManager;
446-
447443
/** jetty server for master to redirect requests to regionserver infoServer */
448444
private Server masterJettyServer;
449445

@@ -768,7 +764,8 @@ public MetricsMaster getMasterMetrics() {
768764
@VisibleForTesting
769765
protected void initializeZKBasedSystemTrackers()
770766
throws IOException, InterruptedException, KeeperException, ReplicationException {
771-
this.balancer = LoadBalancerFactory.getLoadBalancer(conf);
767+
this.balancer = new RSGroupBasedLoadBalancer();
768+
this.balancer.setConf(conf);
772769
this.normalizer = RegionNormalizerFactory.getRegionNormalizer(conf);
773770
this.normalizer.setMasterServices(this);
774771
this.normalizer.setMasterRpcServices((MasterRpcServices)rpcServices);
@@ -1052,9 +1049,6 @@ private void finishActiveMasterInitialization(MonitoredTask status) throws IOExc
10521049
return temp;
10531050
});
10541051
}
1055-
if (this.balancer instanceof FavoredNodesPromoter) {
1056-
favoredNodesManager = new FavoredNodesManager(this);
1057-
}
10581052

10591053
// initialize load balancer
10601054
this.balancer.setMasterServices(this);
@@ -1104,11 +1098,11 @@ private void finishActiveMasterInitialization(MonitoredTask status) throws IOExc
11041098
// table states messing up master launch (namespace table, etc., are not assigned).
11051099
this.assignmentManager.processOfflineRegions();
11061100
// Initialize after meta is up as below scans meta
1107-
if (favoredNodesManager != null && !maintenanceMode) {
1101+
if (getFavoredNodesManager() != null && !maintenanceMode) {
11081102
SnapshotOfRegionAssignmentFromMeta snapshotOfRegionAssignment =
11091103
new SnapshotOfRegionAssignmentFromMeta(getConnection());
11101104
snapshotOfRegionAssignment.initialize();
1111-
favoredNodesManager.initialize(snapshotOfRegionAssignment);
1105+
getFavoredNodesManager().initialize(snapshotOfRegionAssignment);
11121106
}
11131107

11141108
// set cluster status again after user regions are assigned
@@ -2055,14 +2049,13 @@ public void move(final byte[] encodedRegionName, byte[] destServerName) throws I
20552049
LOG.debug("Unable to determine a plan to assign " + hri);
20562050
return;
20572051
}
2058-
// TODO: What is this? I don't get it.
2059-
if (dest.equals(serverName) && balancer instanceof BaseLoadBalancer
2060-
&& !((BaseLoadBalancer)balancer).shouldBeOnMaster(hri)) {
2052+
// TODO: deal with table on master for rs group.
2053+
if (dest.equals(serverName)) {
20612054
// To avoid unnecessary region moving later by balancer. Don't put user
20622055
// regions on master.
2063-
LOG.debug("Skipping move of region " + hri.getRegionNameAsString()
2064-
+ " to avoid unnecessary region moving later by load balancer,"
2065-
+ " because it should not be on master");
2056+
LOG.debug("Skipping move of region " + hri.getRegionNameAsString() +
2057+
" to avoid unnecessary region moving later by load balancer," +
2058+
" because it should not be on master");
20662059
return;
20672060
}
20682061
}
@@ -3514,12 +3507,14 @@ public boolean isSplitOrMergeEnabled(MasterSwitchType switchType) {
35143507

35153508
/**
35163509
* Fetch the configured {@link LoadBalancer} class name. If none is set, a default is returned.
3517-
*
3510+
* <p/>
3511+
* Notice that, the base load balancer will always be {@link RSGroupBasedLoadBalancer} now, so
3512+
* this method will return the balancer used inside each rs group.
35183513
* @return The name of the {@link LoadBalancer} in use.
35193514
*/
35203515
public String getLoadBalancerClassName() {
3521-
return conf.get(HConstants.HBASE_MASTER_LOADBALANCER_CLASS, LoadBalancerFactory
3522-
.getDefaultLoadBalancerClass().getName());
3516+
return conf.get(HConstants.HBASE_MASTER_LOADBALANCER_CLASS,
3517+
LoadBalancerFactory.getDefaultLoadBalancerClass().getName());
35233518
}
35243519

35253520
/**
@@ -3534,13 +3529,13 @@ public SplitOrMergeTracker getSplitOrMergeTracker() {
35343529
}
35353530

35363531
@Override
3537-
public LoadBalancer getLoadBalancer() {
3532+
public RSGroupBasedLoadBalancer getLoadBalancer() {
35383533
return balancer;
35393534
}
35403535

35413536
@Override
35423537
public FavoredNodesManager getFavoredNodesManager() {
3543-
return favoredNodesManager;
3538+
return balancer.getFavoredNodesManager();
35443539
}
35453540

35463541
private long executePeerProcedure(AbstractPeerProcedure<?> procedure) throws IOException {
@@ -3858,7 +3853,7 @@ public HbckChore getHbckChore() {
38583853
}
38593854

38603855
@Override
3861-
public RSGroupInfoManager getRSRSGroupInfoManager() {
3856+
public RSGroupInfoManager getRSGroupInfoManager() {
38623857
return rsGroupInfoManager;
38633858
}
38643859
}

‎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)