You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@hbase.apache.org by zh...@apache.org on 2019/03/30 12:23:41 UTC

[hbase] branch branch-2 updated: HBASE-22121 AsyncAdmin can not deal with non default meta replica

This is an automated email from the ASF dual-hosted git repository.

zhangduo pushed a commit to branch branch-2
in repository https://gitbox.apache.org/repos/asf/hbase.git


The following commit(s) were added to refs/heads/branch-2 by this push:
     new 4cf8e3b  HBASE-22121 AsyncAdmin can not deal with non default meta replica
4cf8e3b is described below

commit 4cf8e3b04c0cb48ddabf8b6cbfb22bdccfd26ca4
Author: zhangduo <zh...@apache.org>
AuthorDate: Sat Mar 30 18:40:41 2019 +0800

    HBASE-22121 AsyncAdmin can not deal with non default meta replica
    
    Signed-off-by: Guanghao Zhang <zg...@apache.org>
---
 .../hadoop/hbase/client/RawAsyncHBaseAdmin.java    |  26 ++++-
 .../hadoop/hbase/client/TestAsyncAdminBase.java    |  55 ++++++---
 .../hbase/client/TestAsyncRegionAdminApi.java      |   2 +-
 .../hbase/client/TestAsyncRegionAdminApi2.java     |   2 +-
 .../TestAsyncRegionAdminApiWithRegionReplicas.java | 128 +++++++++++++++++++++
 5 files changed, 189 insertions(+), 24 deletions(-)

diff --git a/hbase-client/src/main/java/org/apache/hadoop/hbase/client/RawAsyncHBaseAdmin.java b/hbase-client/src/main/java/org/apache/hadoop/hbase/client/RawAsyncHBaseAdmin.java
index 85662b1..3663218 100644
--- a/hbase-client/src/main/java/org/apache/hadoop/hbase/client/RawAsyncHBaseAdmin.java
+++ b/hbase-client/src/main/java/org/apache/hadoop/hbase/client/RawAsyncHBaseAdmin.java
@@ -2295,7 +2295,7 @@ class RawAsyncHBaseAdmin implements AsyncAdmin {
    * Get the region location for the passed region name. The region name may be a full region name
    * or encoded region name. If the region does not found, then it'll throw an
    * UnknownRegionException wrapped by a {@link CompletableFuture}
-   * @param regionNameOrEncodedRegionName
+   * @param regionNameOrEncodedRegionName region name or encoded region name
    * @return region location, wrapped by a {@link CompletableFuture}
    */
   @VisibleForTesting
@@ -2306,10 +2306,28 @@ class RawAsyncHBaseAdmin implements AsyncAdmin {
     try {
       CompletableFuture<Optional<HRegionLocation>> future;
       if (RegionInfo.isEncodedRegionName(regionNameOrEncodedRegionName)) {
-        future = AsyncMetaTableAccessor.getRegionLocationWithEncodedName(metaTable,
-          regionNameOrEncodedRegionName);
+        String encodedName = Bytes.toString(regionNameOrEncodedRegionName);
+        if (encodedName.length() < RegionInfo.MD5_HEX_LENGTH) {
+          // old format encodedName, should be meta region
+          future = connection.registry.getMetaRegionLocation()
+            .thenApply(locs -> Stream.of(locs.getRegionLocations())
+              .filter(loc -> loc.getRegion().getEncodedName().equals(encodedName)).findFirst());
+        } else {
+          future = AsyncMetaTableAccessor.getRegionLocationWithEncodedName(metaTable,
+            regionNameOrEncodedRegionName);
+        }
       } else {
-        future = AsyncMetaTableAccessor.getRegionLocation(metaTable, regionNameOrEncodedRegionName);
+        RegionInfo regionInfo =
+          MetaTableAccessor.parseRegionInfoFromRegionName(regionNameOrEncodedRegionName);
+        if (regionInfo.isMetaRegion()) {
+          future = connection.registry.getMetaRegionLocation()
+            .thenApply(locs -> Stream.of(locs.getRegionLocations())
+              .filter(loc -> loc.getRegion().getReplicaId() == regionInfo.getReplicaId())
+              .findFirst());
+        } else {
+          future =
+            AsyncMetaTableAccessor.getRegionLocation(metaTable, regionNameOrEncodedRegionName);
+        }
       }
 
       CompletableFuture<HRegionLocation> returnedFuture = new CompletableFuture<>();
diff --git a/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncAdminBase.java b/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncAdminBase.java
index da86418..1d19beb 100644
--- a/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncAdminBase.java
+++ b/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncAdminBase.java
@@ -19,6 +19,7 @@ package org.apache.hadoop.hbase.client;
 
 import static org.apache.hadoop.hbase.client.AsyncProcess.START_LOG_ERRORS_AFTER_COUNT_KEY;
 
+import java.io.IOException;
 import java.util.Arrays;
 import java.util.List;
 import java.util.concurrent.CompletableFuture;
@@ -83,7 +84,7 @@ public abstract class TestAsyncAdminBase {
     TEST_UTIL.getConfiguration().setInt(HConstants.HBASE_CLIENT_OPERATION_TIMEOUT, 120000);
     TEST_UTIL.getConfiguration().setInt(HConstants.HBASE_CLIENT_RETRIES_NUMBER, 2);
     TEST_UTIL.getConfiguration().setInt(START_LOG_ERRORS_AFTER_COUNT_KEY, 0);
-    TEST_UTIL.startMiniCluster(2);
+    TEST_UTIL.startMiniCluster(3);
     ASYNC_CONN = ConnectionFactory.createAsyncConnection(TEST_UTIL.getConfiguration()).get();
   }
 
@@ -103,36 +104,54 @@ public abstract class TestAsyncAdminBase {
   @After
   public void tearDown() throws Exception {
     admin.listTableNames(Pattern.compile(tableName.getNameAsString() + ".*"), false)
-        .whenCompleteAsync((tables, err) -> {
-          if (tables != null) {
-            tables.forEach(table -> {
-              try {
-                admin.disableTable(table).join();
-              } catch (Exception e) {
-                LOG.debug("Table: " + tableName + " already disabled, so just deleting it.");
-              }
-              admin.deleteTable(table).join();
-            });
-          }
-        }, ForkJoinPool.commonPool()).join();
+      .whenCompleteAsync((tables, err) -> {
+        if (tables != null) {
+          tables.forEach(table -> {
+            try {
+              admin.disableTable(table).join();
+            } catch (Exception e) {
+              LOG.debug("Table: " + tableName + " already disabled, so just deleting it.");
+            }
+            admin.deleteTable(table).join();
+          });
+        }
+      }, ForkJoinPool.commonPool()).join();
   }
 
-  protected void createTableWithDefaultConf(TableName tableName) {
+  protected void createTableWithDefaultConf(TableName tableName) throws IOException {
     createTableWithDefaultConf(tableName, null);
   }
 
-  protected void createTableWithDefaultConf(TableName tableName, byte[][] splitKeys) {
+  protected void createTableWithDefaultConf(TableName tableName, int regionReplication)
+      throws IOException {
+    createTableWithDefaultConf(tableName, regionReplication, null, FAMILY);
+  }
+
+  protected void createTableWithDefaultConf(TableName tableName, byte[][] splitKeys)
+      throws IOException {
     createTableWithDefaultConf(tableName, splitKeys, FAMILY);
   }
 
+  protected void createTableWithDefaultConf(TableName tableName, int regionReplication,
+      byte[][] splitKeys) throws IOException {
+    createTableWithDefaultConf(tableName, regionReplication, splitKeys, FAMILY);
+  }
+
   protected void createTableWithDefaultConf(TableName tableName, byte[][] splitKeys,
-      byte[]... families) {
-    TableDescriptorBuilder builder = TableDescriptorBuilder.newBuilder(tableName);
+      byte[]... families) throws IOException {
+    createTableWithDefaultConf(tableName, 1, splitKeys, families);
+  }
+
+  protected void createTableWithDefaultConf(TableName tableName, int regionReplication,
+      byte[][] splitKeys, byte[]... families) throws IOException {
+    TableDescriptorBuilder builder =
+      TableDescriptorBuilder.newBuilder(tableName).setRegionReplication(regionReplication);
     for (byte[] family : families) {
       builder.setColumnFamily(ColumnFamilyDescriptorBuilder.of(family));
     }
     CompletableFuture<Void> future = splitKeys == null ? admin.createTable(builder.build())
-        : admin.createTable(builder.build(), splitKeys);
+      : admin.createTable(builder.build(), splitKeys);
     future.join();
+    TEST_UTIL.waitUntilAllRegionsAssigned(tableName);
   }
 }
diff --git a/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncRegionAdminApi.java b/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncRegionAdminApi.java
index 3a8673c..b065a39 100644
--- a/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncRegionAdminApi.java
+++ b/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncRegionAdminApi.java
@@ -196,7 +196,7 @@ public class TestAsyncRegionAdminApi extends TestAsyncAdminBase {
             }
             regionServerCount.incrementAndGet();
           });
-    assertEquals(2, regionServerCount.get());
+    assertEquals(3, regionServerCount.get());
   }
 
   @Test
diff --git a/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncRegionAdminApi2.java b/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncRegionAdminApi2.java
index bb0b813..7d6d2e0 100644
--- a/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncRegionAdminApi2.java
+++ b/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncRegionAdminApi2.java
@@ -184,7 +184,7 @@ public class TestAsyncRegionAdminApi2 extends TestAsyncAdminBase {
   }
 
   @Test
-  public void testMergeRegionsInvalidRegionCount() throws InterruptedException {
+  public void testMergeRegionsInvalidRegionCount() throws Exception {
     byte[][] splitRows = new byte[][] { Bytes.toBytes("3"), Bytes.toBytes("6") };
     createTableWithDefaultConf(tableName, splitRows);
     List<RegionInfo> regions = admin.getRegions(tableName).join();
diff --git a/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncRegionAdminApiWithRegionReplicas.java b/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncRegionAdminApiWithRegionReplicas.java
new file mode 100644
index 0000000..13e5f45
--- /dev/null
+++ b/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncRegionAdminApiWithRegionReplicas.java
@@ -0,0 +1,128 @@
+/**
+ * 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.client;
+
+import static org.hamcrest.CoreMatchers.instanceOf;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotEquals;
+import static org.junit.Assert.assertThat;
+
+import java.io.IOException;
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.ExecutionException;
+import java.util.stream.Collectors;
+import org.apache.hadoop.hbase.HBaseClassTestRule;
+import org.apache.hadoop.hbase.HConstants;
+import org.apache.hadoop.hbase.HRegionLocation;
+import org.apache.hadoop.hbase.TableName;
+import org.apache.hadoop.hbase.testclassification.ClientTests;
+import org.apache.hadoop.hbase.testclassification.LargeTests;
+import org.apache.hadoop.hbase.util.Bytes;
+import org.junit.BeforeClass;
+import org.junit.ClassRule;
+import org.junit.Test;
+import org.junit.experimental.categories.Category;
+import org.junit.runner.RunWith;
+import org.junit.runners.Parameterized;
+
+@RunWith(Parameterized.class)
+@Category({ LargeTests.class, ClientTests.class })
+public class TestAsyncRegionAdminApiWithRegionReplicas extends TestAsyncAdminBase {
+
+  @ClassRule
+  public static final HBaseClassTestRule CLASS_RULE =
+    HBaseClassTestRule.forClass(TestAsyncRegionAdminApiWithRegionReplicas.class);
+
+  @BeforeClass
+  public static void setUpBeforeClass() throws Exception {
+    TEST_UTIL.getConfiguration().setInt(HConstants.META_REPLICAS_NUM, 3);
+    TestAsyncAdminBase.setUpBeforeClass();
+    try (AsyncRegistry registry = AsyncRegistryFactory.getRegistry(TEST_UTIL.getConfiguration())) {
+      RegionReplicaTestHelper.waitUntilAllMetaReplicasHavingRegionLocation(registry, 3);
+    }
+  }
+
+  private void testMoveNonDefaultReplica(TableName tableName)
+      throws InterruptedException, ExecutionException {
+    AsyncTableRegionLocator locator = ASYNC_CONN.getRegionLocator(tableName);
+    List<HRegionLocation> locs = locator.getAllRegionLocations().get();
+    // try region name
+    admin.move(locs.get(1).getRegion().getRegionName()).get();
+    assertNotEquals(locs.get(1).getServerName(),
+      locator.getRegionLocation(HConstants.EMPTY_START_ROW, 1, true).get());
+    // try encoded region name
+    admin.move(locs.get(2).getRegion().getEncodedNameAsBytes()).get();
+    assertNotEquals(locs.get(2).getServerName(),
+      locator.getRegionLocation(HConstants.EMPTY_START_ROW, 2, true).get());
+  }
+
+  @Test
+  public void testMoveNonDefaultReplica()
+      throws InterruptedException, ExecutionException, IOException {
+    createTableWithDefaultConf(tableName, 3);
+    testMoveNonDefaultReplica(tableName);
+    testMoveNonDefaultReplica(TableName.META_TABLE_NAME);
+  }
+
+  @Test
+  public void testSplitNonDefaultReplica()
+      throws InterruptedException, ExecutionException, IOException {
+    createTableWithDefaultConf(tableName, 3);
+    List<HRegionLocation> locs =
+      ASYNC_CONN.getRegionLocator(tableName).getAllRegionLocations().get();
+    try {
+      admin.splitRegion(locs.get(1).getRegion().getRegionName()).get();
+    } catch (ExecutionException e) {
+      assertThat(e.getCause(), instanceOf(IllegalArgumentException.class));
+    }
+    try {
+      admin.splitRegion(locs.get(2).getRegion().getEncodedNameAsBytes()).get();
+    } catch (ExecutionException e) {
+      assertThat(e.getCause(), instanceOf(IllegalArgumentException.class));
+    }
+  }
+
+  @Test
+  public void testMergeNonDefaultReplicas()
+      throws InterruptedException, ExecutionException, IOException {
+    byte[][] splitRows = new byte[][] { Bytes.toBytes(0) };
+    createTableWithDefaultConf(tableName, 3, splitRows);
+    List<HRegionLocation> locs =
+      ASYNC_CONN.getRegionLocator(tableName).getAllRegionLocations().get();
+    assertEquals(6, locs.size());
+    Map<Integer, List<RegionInfo>> replicaId2RegionInfo = locs.stream()
+      .map(HRegionLocation::getRegion).collect(Collectors.groupingBy(RegionInfo::getReplicaId));
+    List<RegionInfo> replicaOnes = replicaId2RegionInfo.get(1);
+    try {
+      admin
+        .mergeRegions(replicaOnes.get(0).getRegionName(), replicaOnes.get(1).getRegionName(), false)
+        .get();
+    } catch (ExecutionException e) {
+      assertThat(e.getCause(), instanceOf(IllegalArgumentException.class));
+    }
+    List<RegionInfo> replicaTwos = replicaId2RegionInfo.get(2);
+    try {
+      admin
+        .mergeRegions(replicaTwos.get(0).getRegionName(), replicaTwos.get(1).getRegionName(), false)
+        .get();
+    } catch (ExecutionException e) {
+      assertThat(e.getCause(), instanceOf(IllegalArgumentException.class));
+    }
+  }
+}