hbase-server/src/test/java/org/apache/hadoop/hbase/MockRegionServerServices.java

   1 /*
   2  * Licensed to the Apache Software Foundation (ASF) under one
   3  * or more contributor license agreements.  See the NOTICE file
   4  * distributed with this work for additional information
   5  * regarding copyright ownership.  The ASF licenses this file
   6  * to you under the Apache License, Version 2.0 (the
   7  * "License"); you may not use this file except in compliance
   8  * with the License.  You may obtain a copy of the License at
   9  *
  10  *     http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  */
  18 package org.apache.hadoop.hbase;
  19
  20 import com.google.protobuf.Service;
  21 import java.io.IOException;
  22 import java.net.InetSocketAddress;
  23 import java.util.Collection;
  24 import java.util.Collections;
  25 import java.util.HashMap;
  26 import java.util.List;
  27 import java.util.Map;
  28 import java.util.Map.Entry;
  29 import java.util.Optional;
  30 import java.util.concurrent.ConcurrentSkipListMap;
  31 import java.util.concurrent.atomic.AtomicBoolean;
  32 import org.apache.hadoop.conf.Configuration;
  33 import org.apache.hadoop.fs.FileSystem;
  34 import org.apache.hadoop.hbase.client.AsyncClusterConnection;
  35 import org.apache.hadoop.hbase.client.Connection;
  36 import org.apache.hadoop.hbase.client.RegionInfo;
  37 import org.apache.hadoop.hbase.client.locking.EntityLock;
  38 import org.apache.hadoop.hbase.executor.ExecutorService;
  39 import org.apache.hadoop.hbase.fs.HFileSystem;
  40 import org.apache.hadoop.hbase.io.hfile.BlockCache;
  41 import org.apache.hadoop.hbase.ipc.RpcServerInterface;
  42 import org.apache.hadoop.hbase.mob.MobFileCache;
  43 import org.apache.hadoop.hbase.quotas.RegionServerRpcQuotaManager;
  44 import org.apache.hadoop.hbase.quotas.RegionServerSpaceQuotaManager;
  45 import org.apache.hadoop.hbase.quotas.RegionSizeStore;
  46 import org.apache.hadoop.hbase.regionserver.FlushRequester;
  47 import org.apache.hadoop.hbase.regionserver.HRegion;
  48 import org.apache.hadoop.hbase.regionserver.HeapMemoryManager;
  49 import org.apache.hadoop.hbase.regionserver.LeaseManager;
  50 import org.apache.hadoop.hbase.regionserver.MetricsRegionServer;
  51 import org.apache.hadoop.hbase.regionserver.Region;
  52 import org.apache.hadoop.hbase.regionserver.RegionServerAccounting;
  53 import org.apache.hadoop.hbase.regionserver.RegionServerServices;
  54 import org.apache.hadoop.hbase.regionserver.ReplicationSourceService;
  55 import org.apache.hadoop.hbase.regionserver.SecureBulkLoadManager;
  56 import org.apache.hadoop.hbase.regionserver.ServerNonceManager;
  57 import org.apache.hadoop.hbase.regionserver.compactions.CompactionRequester;
  58 import org.apache.hadoop.hbase.regionserver.throttle.ThroughputController;
  59 import org.apache.hadoop.hbase.security.access.AccessChecker;
  60 import org.apache.hadoop.hbase.security.access.ZKPermissionWatcher;
  61 import org.apache.hadoop.hbase.util.Bytes;
  62 import org.apache.hadoop.hbase.wal.WAL;
  63 import org.apache.hadoop.hbase.zookeeper.ZKWatcher;
  64 import org.slf4j.Logger;
  65 import org.slf4j.LoggerFactory;
  66
  67 import org.apache.hadoop.hbase.shaded.protobuf.generated.HBaseProtos;
  68
  69 /**
  70  * Basic mock region server services.  Should only be instantiated by HBaseTestingUtility.b
  71  */
  72 public class MockRegionServerServices implements RegionServerServices {
  73   protected static final Logger LOG = LoggerFactory.getLogger(MockRegionServerServices.class);
  74   private final Map<String, Region> regions = new HashMap<>();
  75   private final ConcurrentSkipListMap<byte[], Boolean> rit =
  76     new ConcurrentSkipListMap<>(Bytes.BYTES_COMPARATOR);
  77   private HFileSystem hfs = null;
  78   private final Configuration conf;
  79   private ZKWatcher zkw = null;
  80   private ServerName serverName = null;
  81   private RpcServerInterface rpcServer = null;
  82   private volatile boolean abortRequested;
  83   private volatile boolean stopping = false;
  84   private final AtomicBoolean running = new AtomicBoolean(true);
  85
  86   MockRegionServerServices(ZKWatcher zkw) {
  87     this(zkw, null);
  88   }
  89
  90   MockRegionServerServices(ZKWatcher zkw, ServerName serverName) {
  91     this.zkw = zkw;
  92     this.serverName = serverName;
  93     this.conf = (zkw == null ? new Configuration() : zkw.getConfiguration());
  94   }
  95
  96   MockRegionServerServices(){
  97     this(null, null);
  98   }
  99
 100   public MockRegionServerServices(Configuration conf) {
 101     this.conf = conf;
 102   }
 103
 104   @Override
 105   public boolean removeRegion(HRegion r, ServerName destination) {
 106     return this.regions.remove(r.getRegionInfo().getEncodedName()) != null;
 107   }
 108
 109   @Override
 110   public Region getRegion(String encodedRegionName) {
 111     return this.regions.get(encodedRegionName);
 112   }
 113
 114   @Override
 115   public List<Region> getRegions(TableName tableName) throws IOException {
 116     return null;
 117   }
 118
 119   @Override
 120   public List<Region> getRegions() {
 121     return null;
 122   }
 123
 124   @Override
 125   public void addRegion(HRegion r) {
 126     this.regions.put(r.getRegionInfo().getEncodedName(), r);
 127   }
 128
 129   @Override
 130   public void postOpenDeployTasks(PostOpenDeployContext context) throws IOException {
 131     addRegion(context.getRegion());
 132   }
 133
 134   @Override
 135   public boolean isStopping() {
 136     return this.stopping;
 137   }
 138
 139   @Override
 140   public RpcServerInterface getRpcServer() {
 141     return rpcServer;
 142   }
 143
 144   public void setRpcServer(RpcServerInterface rpc) {
 145     this.rpcServer = rpc;
 146   }
 147
 148   @Override
 149   public ConcurrentSkipListMap<byte[], Boolean> getRegionsInTransitionInRS() {
 150     return rit;
 151   }
 152
 153   @Override
 154   public FlushRequester getFlushRequester() {
 155     return null;
 156   }
 157
 158   @Override
 159   public CompactionRequester getCompactionRequestor() {
 160     return null;
 161   }
 162
 163   @Override
 164   public Connection getConnection() {
 165     return null;
 166   }
 167
 168   @Override
 169   public ZKWatcher getZooKeeper() {
 170     return zkw;
 171   }
 172
 173   @Override
 174   public CoordinatedStateManager getCoordinatedStateManager() {
 175     return null;
 176   }
 177
 178   @Override
 179   public RegionServerAccounting getRegionServerAccounting() {
 180     return null;
 181   }
 182
 183   @Override
 184   public RegionServerRpcQuotaManager getRegionServerRpcQuotaManager() {
 185     return null;
 186   }
 187
 188   @Override
 189   public ServerName getServerName() {
 190     return this.serverName;
 191   }
 192
 193   @Override
 194   public Configuration getConfiguration() {
 195     return conf;
 196   }
 197
 198   @Override
 199   public void abort(String why, Throwable e) {
 200     this.abortRequested = true;
 201     stop(why);
 202   }
 203
 204   @Override
 205   public void stop(String why) {
 206     this.stopping = true;
 207     if (running.compareAndSet(true, false)) {
 208       LOG.info("Shutting down due to request '" + why + "'");
 209     }
 210   }
 211
 212   @Override
 213   public boolean isStopped() {
 214     return !(running.get());
 215   }
 216
 217   @Override
 218   public boolean isAborted() {
 219     return this.abortRequested;
 220   }
 221
 222   @Override
 223   public HFileSystem getFileSystem() {
 224     return this.hfs;
 225   }
 226
 227   public void setFileSystem(FileSystem hfs) {
 228     this.hfs = (HFileSystem)hfs;
 229   }
 230
 231   @Override
 232   public LeaseManager getLeaseManager() {
 233     return null;
 234   }
 235
 236   @Override
 237   public List<WAL> getWALs() throws IOException {
 238     return Collections.emptyList();
 239   }
 240
 241   @Override
 242   public WAL getWAL(RegionInfo regionInfo) throws IOException {
 243     return null;
 244   }
 245
 246   @Override
 247   public ExecutorService getExecutorService() {
 248     return null;
 249   }
 250
 251   @Override
 252   public ChoreService getChoreService() {
 253     return null;
 254   }
 255
 256   @Override
 257   public void updateRegionFavoredNodesMapping(String encodedRegionName,
 258       List<HBaseProtos.ServerName> favoredNodes) {
 259   }
 260
 261   @Override
 262   public InetSocketAddress[] getFavoredNodesForRegion(String encodedRegionName) {
 263     return null;
 264   }
 265
 266   @Override
 267   public ServerNonceManager getNonceManager() {
 268     return null;
 269   }
 270
 271   @Override
 272   public boolean reportRegionStateTransition(RegionStateTransitionContext context) {
 273     return false;
 274   }
 275
 276   @Override
 277   public boolean registerService(Service service) {
 278     return false;
 279   }
 280
 281   @Override
 282   public HeapMemoryManager getHeapMemoryManager() {
 283     return null;
 284   }
 285
 286   @Override
 287   public double getCompactionPressure() {
 288     return 0;
 289   }
 290
 291   @Override
 292   public ThroughputController getFlushThroughputController() {
 293     return null;
 294   }
 295
 296   @Override
 297   public double getFlushPressure() {
 298     return 0;
 299   }
 300
 301   @Override
 302   public MetricsRegionServer getMetrics() {
 303     return null;
 304   }
 305
 306   @Override
 307   public EntityLock regionLock(List<RegionInfo> regionInfos, String description, Abortable abort)
 308       throws IOException {
 309     return null;
 310   }
 311
 312   @Override
 313   public SecureBulkLoadManager getSecureBulkLoadManager() {
 314     return null;
 315   }
 316
 317   @Override
 318   public void unassign(byte[] regionName) throws IOException {
 319   }
 320
 321   @Override
 322   public RegionServerSpaceQuotaManager getRegionServerSpaceQuotaManager() {
 323     return null;
 324   }
 325
 326   @Override
 327   public Connection createConnection(Configuration conf) throws IOException {
 328     return null;
 329   }
 330
 331   @Override
 332   public boolean reportRegionSizesForQuotas(RegionSizeStore sizeStore) {
 333     return true;
 334   }
 335
 336   @Override
 337   public boolean reportFileArchivalForQuotas(
 338       TableName tableName, Collection<Entry<String,Long>> archivedFiles) {
 339     return true;
 340   }
 341
 342   @Override
 343   public boolean isClusterUp() {
 344     return true;
 345   }
 346
 347   @Override
 348   public ReplicationSourceService getReplicationSourceService() {
 349     return null;
 350   }
 351
 352   @Override
 353   public TableDescriptors getTableDescriptors() {
 354     return null;
 355   }
 356
 357   @Override
 358   public Optional<BlockCache> getBlockCache() {
 359     return Optional.empty();
 360   }
 361
 362   @Override
 363   public Optional<MobFileCache> getMobFileCache() {
 364     return Optional.empty();
 365   }
 366
 367   @Override
 368   public AccessChecker getAccessChecker() {
 369     return null;
 370   }
 371
 372   @Override
 373   public ZKPermissionWatcher getZKPermissionWatcher() {
 374     return null;
 375   }
 376
 377   @Override
 378   public AsyncClusterConnection getAsyncClusterConnection() {
 379     return null;
 380   }
 381 }