HBASE-26089 Support RegionCoprocessor on CompactionServer (#3580)

Signed-off-by: Duo Zhang <zhangduo@apache.org>

Author: nyl3532016

hbase-server/src/main/java/org/apache/hadoop/hbase/AbstractServer.java

@@ -17,6 +17,10 @@
  */
 package org.apache.hadoop.hbase;
 
+import static org.apache.hadoop.hbase.compactionserver.HCompactionServer.COMPACTIONSERVER;
+import static org.apache.hadoop.hbase.master.HMaster.MASTER;
+import static org.apache.hadoop.hbase.regionserver.HRegionServer.REGIONSERVER;
+
 import java.io.IOException;
 import java.lang.reflect.Constructor;
 import java.lang.reflect.InvocationTargetException;
@@ -431,4 +435,18 @@ public FileSystem getFileSystem() {
   protected abstract AbstractRpcServices getRpcService();
 
   protected abstract String getProcessName();
+
+  public ServerType getServerType() {
+    String processName = getProcessName();
+    if (processName.equals(MASTER)) {
+      return ServerType.Master;
+    }
+    if (processName.equals(REGIONSERVER)) {
+      return ServerType.RegionServer;
+    }
+    if (processName.equals(COMPACTIONSERVER)) {
+      return ServerType.CompactionServer;
+    }
+    return ServerType.ReplicationServer;
+  }
 }

hbase-server/src/main/java/org/apache/hadoop/hbase/ServerType.java

@@ -0,0 +1,29 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.hadoop.hbase;
+
+import org.apache.yetus.audience.InterfaceAudience;
+
+/**
+ * Enum describing Server Type
+ */
+@InterfaceAudience.LimitedPrivate(HBaseInterfaceAudience.COPROC)
+public enum ServerType {
+  Master, RegionServer, CompactionServer, ReplicationServer
+}

hbase-server/src/main/java/org/apache/hadoop/hbase/compactionserver/CompactionThreadManager.java

@@ -19,7 +19,6 @@
 
 import java.io.IOException;
 import java.util.ArrayList;
-import java.util.Collection;
 import java.util.Collections;
 import java.util.Comparator;
 import java.util.HashSet;
@@ -240,7 +239,7 @@ Optional<CompactionContext> selectCompaction(HStore store, RegionInfo regionInfo
     excludeFiles.addAll(compactedFiles);
     // Convert files names to store files
     status.setStatus("Convert current compacting and compacted files to store files");
-    List<HStoreFile> excludeStoreFiles = getExcludedStoreFiles(store, excludeFiles);
+    List<HStoreFile> excludeStoreFiles = store.getStoreFilesBaseOnFileNames(excludeFiles);
     LOG.info(
       "Start select store: {}, excludeFileNames: {}, excluded: {}, compacting: {}, compacted: {}",
       logStr, excludeFiles.size(), excludeStoreFiles.size(), compactingFiles.size(),
@@ -354,23 +353,11 @@ private void reportCompactionCompleted(CompactionTask task, List<String> newFile
     }
   }
 
-  private List<HStoreFile> getExcludedStoreFiles(HStore store, Set<String> excludeFileNames) {
-    Collection<HStoreFile> storefiles = store.getStorefiles();
-    List<HStoreFile> storeFiles = new ArrayList<>();
-    for (HStoreFile storefile : storefiles) {
-      String name = storefile.getPath().getName();
-      if (excludeFileNames.contains(name)) {
-        storeFiles.add(storefile);
-      }
-    }
-    return storeFiles;
-  }
-
   private HStore getStore(final Configuration conf, final FileSystem fs, final Path rootDir,
       final TableDescriptor htd, final RegionInfo hri, final String familyName) throws IOException {
     HRegionFileSystem regionFs = new HRegionFileSystem(conf, fs,
         CommonFSUtils.getTableDir(rootDir, htd.getTableName()), hri);
-    HRegion region = new HRegion(regionFs, null, conf, htd, null);
+    HRegion region = new HRegion(regionFs, conf, htd, server);
     ColumnFamilyDescriptor columnFamilyDescriptor = htd.getColumnFamily(Bytes.toBytes(familyName));
     HStore store;
     if (columnFamilyDescriptor.isMobEnabled()) {

hbase-server/src/main/java/org/apache/hadoop/hbase/compactionserver/HCompactionServer.java

@@ -19,6 +19,7 @@
 
 import java.io.IOException;
 import java.util.Collection;
+import java.util.List;
 import java.util.concurrent.atomic.LongAdder;
 
 import org.apache.hadoop.conf.Configuration;
@@ -28,9 +29,12 @@
 import org.apache.hadoop.hbase.HBaseConfiguration;
 import org.apache.hadoop.hbase.HConstants;
 import org.apache.hadoop.hbase.ServerName;
+import org.apache.hadoop.hbase.TableName;
 import org.apache.hadoop.hbase.YouAreDeadException;
+import org.apache.hadoop.hbase.coprocessor.RegionCoprocessorService;
 import org.apache.hadoop.hbase.fs.HFileSystem;
 import org.apache.hadoop.hbase.log.HBaseMarkers;
+import org.apache.hadoop.hbase.regionserver.HRegion;
 import org.apache.hadoop.hbase.regionserver.compactions.CompactionProgress;
 import org.apache.hadoop.hbase.security.SecurityConstants;
 import org.apache.hadoop.hbase.security.Superusers;
@@ -54,7 +58,7 @@
 import org.apache.hadoop.hbase.shaded.protobuf.generated.CompactionServerStatusProtos.CompactionServerStatusService;
 
 @InterfaceAudience.Private
-public class HCompactionServer extends AbstractServer {
+public class HCompactionServer extends AbstractServer implements RegionCoprocessorService {
 
   /** compaction server process name */
   public static final String COMPACTIONSERVER = "compactionserver";
@@ -312,4 +316,21 @@ static HCompactionServer constructCompactionServer(
     }
   }
 
+  @Override
+  public HRegion getRegion(final String encodedRegionName) {
+    throw new UnsupportedOperationException(
+        "Method getRegion is not supported in HCompactionServer");
+  }
+
+  @Override
+  public List<HRegion> getRegions(TableName tableName) {
+    throw new UnsupportedOperationException(
+        "Method getRegions is not supported in HCompactionServer");
+  }
+
+  @Override
+  public List<HRegion> getRegions() {
+    throw new UnsupportedOperationException(
+        "Method getRegions is not supported in HCompactionServer");
+  }
 }