add hash-like partition case for region locator

Author: maochongxin

src/main/java/com/alipay/oceanbase/hbase/util/OHTableDescriptorExecutor.java

@@ -10,7 +10,6 @@
 import org.apache.hadoop.hbase.HColumnDescriptor;
 import org.apache.hadoop.hbase.HTableDescriptor;
 import org.apache.hadoop.hbase.TableName;
-import sun.font.SunFontManager;
 
 import java.io.IOException;
 import java.util.HashMap;

src/test/java/com/alipay/oceanbase/hbase/OHConnectionTest.java

@@ -22,6 +22,7 @@
 import org.apache.hadoop.conf.Configuration;
 import org.apache.hadoop.hbase.Cell;
 import org.apache.hadoop.hbase.CellUtil;
+import org.apache.hadoop.hbase.HConstants;
 import org.apache.hadoop.hbase.TableName;
 import org.apache.hadoop.hbase.client.*;
 import org.apache.hadoop.hbase.util.Bytes;
@@ -30,6 +31,7 @@
 
 import java.io.IOException;
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
 import java.util.Optional;
 import java.util.concurrent.*;
@@ -990,6 +992,21 @@ public void testRangePartitionWithRegionLocator() throws Exception {
             });
         }
     }
+    @Test
+    public void testKeyPartitionWithRegionLocator() throws IOException {
+        final String tableNameStr = "test_multi_cf";
+        final TableName tableName = TableName.valueOf(tableNameStr);
+        final Configuration conf = ObHTableTestUtil.newConfiguration();
+        connection = ConnectionFactory.createConnection(conf);
+        hTable = connection.getTable(tableName);
+        RegionLocator locator = connection.getRegionLocator(tableName);
+        byte[][] startKeys = locator.getStartKeys();
+        byte[][] endKeys = locator.getEndKeys();
+        Assert.assertEquals("Should have 1 region", 1, startKeys.length);
+        Assert.assertEquals("Should have 1 region", 1, endKeys.length);
+        Assert.assertEquals(startKeys[0], endKeys[0]);
+        Assert.assertEquals(startKeys[0], HConstants.EMPTY_BYTE_ARRAY);
+    }
 
     @Test
     public void testBufferedMutatorPeriodicFlush() throws Exception {

src/test/java/com/alipay/oceanbase/hbase/OHTableAdminInterfaceTest.java

@@ -268,6 +268,121 @@ public void testGetStartEndKeysOHTablePoolLoadNon() throws Exception {
         Assert.assertEquals(0, startEndKeys.getSecond()[0].length);
     }
 
+    @Test
+    public void testAdminEnDisableTable() throws Exception {
+        java.sql.Connection conn = ObHTableTestUtil.getConnection();
+        Statement st = conn.createStatement();
+        st.execute("CREATE TABLEGROUP IF NOT EXISTS test_multi_cf SHARDING = 'ADAPTIVE';\n" +
+                "\n" +
+                "CREATE TABLE IF NOT EXISTS `test_multi_cf$family_with_group1` (\n" +
+                "    `K` varbinary(1024) NOT NULL,\n" +
+                "    `Q` varbinary(256) NOT NULL,\n" +
+                "    `T` bigint(20) NOT NULL,\n" +
+                "    `V` varbinary(1024) DEFAULT NULL,\n" +
+                "    PRIMARY KEY (`K`, `Q`, `T`)\n" +
+                ") TABLEGROUP = test_multi_cf PARTITION BY KEY(`K`) PARTITIONS 3;\n" +
+                "\n" +
+                "CREATE TABLE IF NOT EXISTS `test_multi_cf$family_with_group2` (\n" +
+                "    `K` varbinary(1024) NOT NULL,\n" +
+                "    `Q` varbinary(256) NOT NULL,\n" +
+                "    `T` bigint(20) NOT NULL,\n" +
+                "    `V` varbinary(1024) DEFAULT NULL,\n" +
+                "    PRIMARY KEY (`K`, `Q`, `T`)\n" +
+                ") TABLEGROUP = test_multi_cf PARTITION BY KEY(`K`) PARTITIONS 3;\n" +
+                "\n" +
+                "CREATE TABLE IF NOT EXISTS `test_multi_cf$family_with_group3` (\n" +
+                "    `K` varbinary(1024) NOT NULL,\n" +
+                "    `Q` varbinary(256) NOT NULL,\n" +
+                "    `T` bigint(20) NOT NULL,\n" +
+                "    `V` varbinary(1024) DEFAULT NULL,\n" +
+                "    PRIMARY KEY (`K`, `Q`, `T`)\n" +
+                ") TABLEGROUP = test_multi_cf PARTITION BY KEY(`K`) PARTITIONS 3;\n" +
+                "\n" +
+                "CREATE DATABASE IF NOT EXISTS `n1`;\n" +
+                "use `n1`;\n" +
+                "CREATE TABLEGROUP IF NOT EXISTS `n1:test` SHARDING = 'ADAPTIVE';\n" +
+                "CREATE TABLE IF NOT EXISTS `n1:test$family_group` (\n" +
+                "      `K` varbinary(1024) NOT NULL,\n" +
+                "      `Q` varbinary(256) NOT NULL,\n" +
+                "      `T` bigint(20) NOT NULL,\n" +
+                "      `V` varbinary(1024) DEFAULT NULL,\n" +
+                "      PRIMARY KEY (`K`, `Q`, `T`)\n" +
+                ") TABLEGROUP = `n1:test`;" +
+                "\n" +
+                "CREATE TABLE IF NOT EXISTS `n1:test$family1` (\n" +
+                "      `K` varbinary(1024) NOT NULL,\n" +
+                "      `Q` varbinary(256) NOT NULL,\n" +
+                "      `T` bigint(20) NOT NULL,\n" +
+                "      `V` varbinary(1024) DEFAULT NULL,\n" +
+                "      PRIMARY KEY (`K`, `Q`, `T`)\n" +
+                ") TABLEGROUP = `n1:test`;");
+        Configuration conf = ObHTableTestUtil.newConfiguration();
+        Connection connection = ConnectionFactory.createConnection(conf);
+        Admin admin = connection.getAdmin();
+        admin.disableTable(TableName.valueOf("test_multi_cf"));
+        assertTrue(admin.tableExists(TableName.valueOf("n1", "test")));
+        assertTrue(admin.tableExists(TableName.valueOf("test_multi_cf")));
+        // disable a non-existed table
+        IOException thrown = assertThrows(IOException.class,
+                () -> {
+                    admin.disableTable(TableName.valueOf("tablegroup_not_exists"));
+                });
+        assertTrue(thrown.getCause() instanceof ObTableException);
+        Assert.assertEquals(ResultCodes.OB_TABLEGROUP_NOT_EXIST.errorCode, ((ObTableException) thrown.getCause()).getErrorCode());
+
+        // write an enabled table, should succeed
+        batchInsert(10, "test_multi_ch");
+        // disable a disabled table
+        thrown = assertThrows(IOException.class,
+                () -> {
+                    admin.disableTable(TableName.valueOf("test_multi_cf"));
+                });
+        assertTrue(thrown.getCause() instanceof ObTableException);
+        Assert.assertEquals(ResultCodes.OB_KV_TABLE_NOT_DISABLED.errorCode, ((ObTableException) thrown.getCause()).getErrorCode());
+
+        // write an enabled table, should fail
+        batchInsert(10, "test_multi_ch");
+        enDisableRead(10, "test_multi_ch");
+
+        // enable a disabled table
+        admin.enableTable(TableName.valueOf("test_multi_cf"));
+
+        // write an enabled table, should succeed
+        batchInsert(10, "test_multi_ch");
+        enDisableRead(10, "test_multi_ch");
+
+        // enable an enabled table
+        thrown = assertThrows(IOException.class,
+                () -> {
+                    admin.disableTable(TableName.valueOf("n1", "test");
+                });
+        assertTrue(thrown.getCause() instanceof ObTableException);
+        Assert.assertEquals(ResultCodes.OB_KV_TABLE_NOT_ENABLED.errorCode, ((ObTableException) thrown.getCause()).getErrorCode());
+
+        admin.deleteTable(TableName.valueOf("n1", "test"));
+        admin.deleteTable(TableName.valueOf("test_multi_cf"));
+        assertFalse(admin.tableExists(TableName.valueOf("n1", "test")));
+        assertFalse(admin.tableExists(TableName.valueOf("test_multi_cf")));
+    }
+
+    private void enDisableRead(int rows, String tablegroup) throws Exception {
+        Configuration conf = ObHTableTestUtil.newConfiguration();
+        Connection connection = ConnectionFactory.createConnection(conf);
+        Table table = connection.getTable(TableName.valueOf(tablegroup));
+        List<Row> batchLsit = new LinkedList<>();
+        for (int i = 0; i < rows; ++i) {
+            Get get = new Get(toBytes("Key" + i));
+            batchLsit.add(get);
+            if (i % 100 == 0) { // 100 rows one batch to avoid OB_TIMEOUT
+                Object[] results = new Object[batchLsit.size()];
+                table.batch(batchLsit, results);
+                batchLsit.clear();
+            }
+        }
+        Object[] results = new Object[batchLsit.size()];
+        table.batch(batchLsit, results);
+    }
+
     @Test
     public void testAdminGetRegionMetrics() throws Exception {
         java.sql.Connection conn = ObHTableTestUtil.getConnection();