hbase-server/src/test/java/org/apache/hadoop/hbase/client/AbstractTestCITimeout.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.client;
  19
  20 import java.io.IOException;
  21 import java.util.List;
  22 import java.util.Optional;
  23 import java.util.concurrent.atomic.AtomicLong;
  24 import org.apache.hadoop.conf.Configuration;
  25 import org.apache.hadoop.hbase.Cell;
  26 import org.apache.hadoop.hbase.HBaseTestingUtility;
  27 import org.apache.hadoop.hbase.HConstants;
  28 import org.apache.hadoop.hbase.coprocessor.ObserverContext;
  29 import org.apache.hadoop.hbase.coprocessor.RegionCoprocessor;
  30 import org.apache.hadoop.hbase.coprocessor.RegionCoprocessorEnvironment;
  31 import org.apache.hadoop.hbase.coprocessor.RegionObserver;
  32 import org.apache.hadoop.hbase.util.Bytes;
  33 import org.apache.hadoop.hbase.util.Threads;
  34 import org.apache.hadoop.hbase.wal.WALEdit;
  35 import org.junit.AfterClass;
  36 import org.junit.BeforeClass;
  37 import org.junit.Rule;
  38 import org.junit.rules.TestName;
  39
  40 /**
  41  * Based class for testing timeout logic for {@link ConnectionImplementation}.
  42  */
  43 public abstract class AbstractTestCITimeout {
  44
  45   protected static final HBaseTestingUtility TEST_UTIL = new HBaseTestingUtility();
  46
  47   protected static final byte[] FAM_NAM = Bytes.toBytes("f");
  48
  49   @Rule
  50   public final TestName name = new TestName();
  51
  52   /**
  53    * This copro sleeps 20 second. The first call it fails. The second time, it works.
  54    */
  55   public static class SleepAndFailFirstTime implements RegionCoprocessor, RegionObserver {
  56     static final AtomicLong ct = new AtomicLong(0);
  57     static final String SLEEP_TIME_CONF_KEY = "hbase.coprocessor.SleepAndFailFirstTime.sleepTime";
  58     static final long DEFAULT_SLEEP_TIME = 20000;
  59     static final AtomicLong sleepTime = new AtomicLong(DEFAULT_SLEEP_TIME);
  60
  61     public SleepAndFailFirstTime() {
  62     }
  63
  64     @Override
  65     public Optional<RegionObserver> getRegionObserver() {
  66       return Optional.of(this);
  67     }
  68
  69     @Override
  70     public void postOpen(ObserverContext<RegionCoprocessorEnvironment> c) {
  71       RegionCoprocessorEnvironment env = c.getEnvironment();
  72       Configuration conf = env.getConfiguration();
  73       sleepTime.set(conf.getLong(SLEEP_TIME_CONF_KEY, DEFAULT_SLEEP_TIME));
  74     }
  75
  76     @Override
  77     public void preGetOp(final ObserverContext<RegionCoprocessorEnvironment> e, final Get get,
  78         final List<Cell> results) throws IOException {
  79       Threads.sleep(sleepTime.get());
  80       if (ct.incrementAndGet() == 1) {
  81         throw new IOException("first call I fail");
  82       }
  83     }
  84
  85     @Override
  86     public void prePut(final ObserverContext<RegionCoprocessorEnvironment> e, final Put put,
  87         final WALEdit edit, final Durability durability) throws IOException {
  88       Threads.sleep(sleepTime.get());
  89       if (ct.incrementAndGet() == 1) {
  90         throw new IOException("first call I fail");
  91       }
  92     }
  93
  94     @Override
  95     public void preDelete(final ObserverContext<RegionCoprocessorEnvironment> e,
  96         final Delete delete, final WALEdit edit, final Durability durability) throws IOException {
  97       Threads.sleep(sleepTime.get());
  98       if (ct.incrementAndGet() == 1) {
  99         throw new IOException("first call I fail");
 100       }
 101     }
 102
 103     @Override
 104     public Result preIncrement(final ObserverContext<RegionCoprocessorEnvironment> e,
 105         final Increment increment) throws IOException {
 106       Threads.sleep(sleepTime.get());
 107       if (ct.incrementAndGet() == 1) {
 108         throw new IOException("first call I fail");
 109       }
 110       return null;
 111     }
 112
 113   }
 114
 115   public static class SleepCoprocessor implements RegionCoprocessor, RegionObserver {
 116     public static final int SLEEP_TIME = 5000;
 117
 118     @Override
 119     public Optional<RegionObserver> getRegionObserver() {
 120       return Optional.of(this);
 121     }
 122
 123     @Override
 124     public void preGetOp(final ObserverContext<RegionCoprocessorEnvironment> e, final Get get,
 125         final List<Cell> results) throws IOException {
 126       Threads.sleep(SLEEP_TIME);
 127     }
 128
 129     @Override
 130     public void prePut(final ObserverContext<RegionCoprocessorEnvironment> e, final Put put,
 131         final WALEdit edit, final Durability durability) throws IOException {
 132       Threads.sleep(SLEEP_TIME);
 133     }
 134
 135     @Override
 136     public Result preIncrement(final ObserverContext<RegionCoprocessorEnvironment> e,
 137         final Increment increment) throws IOException {
 138       Threads.sleep(SLEEP_TIME);
 139       return null;
 140     }
 141
 142     @Override
 143     public void preDelete(final ObserverContext<RegionCoprocessorEnvironment> e,
 144         final Delete delete, final WALEdit edit, final Durability durability) throws IOException {
 145       Threads.sleep(SLEEP_TIME);
 146     }
 147   }
 148
 149   @BeforeClass
 150   public static void setUpBeforeClass() throws Exception {
 151     TEST_UTIL.getConfiguration().setBoolean(HConstants.STATUS_PUBLISHED, true);
 152     // Up the handlers; this test needs more than usual.
 153     TEST_UTIL.getConfiguration().setInt(HConstants.REGION_SERVER_HIGH_PRIORITY_HANDLER_COUNT, 10);
 154     TEST_UTIL.getConfiguration().setInt(HConstants.HBASE_CLIENT_RETRIES_NUMBER, 5);
 155     TEST_UTIL.getConfiguration().setInt(HConstants.REGION_SERVER_HANDLER_COUNT, 3);
 156     TEST_UTIL.startMiniCluster(2);
 157
 158   }
 159
 160   @AfterClass
 161   public static void tearDownAfterClass() throws Exception {
 162     TEST_UTIL.shutdownMiniCluster();
 163   }
 164 }