MetaExecutor interface definition and demo

Author: maochongxin

src/main/java/com/alipay/oceanbase/hbase/execute/AbstractObTableMetaExecutor.java

@@ -0,0 +1,31 @@
+package com.alipay.oceanbase.hbase.execute;
+
+import com.alipay.oceanbase.rpc.ObTableClient;
+import com.alipay.oceanbase.rpc.exception.ObTableException;
+import com.alipay.oceanbase.rpc.meta.ObTableMetaRequest;
+import com.alipay.oceanbase.rpc.meta.ObTableMetaResponse;
+import com.alipay.oceanbase.rpc.table.ObTable;
+
+import java.io.IOException;
+
+public abstract class AbstractObTableMetaExecutor<T> implements ObTableMetaExecutor<T> {
+
+    @Override
+    public T execute(ObTableClient client, ObTableMetaRequest request) throws IOException {
+        if (request.getMetaType() != getMetaType()) {
+            throw new IOException("Invalid meta type, expected " + getMetaType());
+        }
+        ObTable table = client.randomTable();
+        ObTableMetaResponse response;
+        try {
+            response = (ObTableMetaResponse) client.executeWithRetry(
+                    table,
+                    request,
+                    null /*tableName*/
+            );
+        } catch (Exception e) {
+            throw new IOException("Failed to execute request", e);
+        }
+        return parse(response);
+    }
+}

src/main/java/com/alipay/oceanbase/hbase/execute/ObTableMetaExecutor.java

@@ -0,0 +1,32 @@
+package com.alipay.oceanbase.hbase.execute;
+
+import com.alipay.oceanbase.rpc.ObTableClient;
+import com.alipay.oceanbase.rpc.meta.ObTableMetaRequest;
+import com.alipay.oceanbase.rpc.meta.ObTableMetaResponse;
+import com.alipay.oceanbase.rpc.meta.ObTableRpcMetaType;
+
+import java.io.IOException;
+
+public interface ObTableMetaExecutor<T> {
+    /**
+     * 执行元数据请求
+     * @param request 元数据请求
+     * @return 解析后的元数据对象
+     * @throws IOException 如果执行失败或解析失败
+     */
+    T execute(ObTableClient client, ObTableMetaRequest request) throws IOException;
+
+    /**
+     * 解析元数据响应, 用户需要重写
+     * @param response 元数据响应
+     * @return 解析后的元数据对象
+     * @throws IOException 如果解析失败
+     */
+    T parse(ObTableMetaResponse response) throws IOException;
+
+    /**
+     * 获取元信息类型, 用户需要重写
+     * @return 元信息类型
+     */
+    ObTableRpcMetaType getMetaType() throws IOException;
+}

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

@@ -19,6 +19,7 @@
 
 import com.alipay.oceanbase.hbase.OHTable;
 import com.alipay.oceanbase.hbase.exception.FeatureNotSupportedException;
+import com.alipay.oceanbase.rpc.ObTableClient;
 import org.apache.hadoop.classification.InterfaceAudience;
 import org.apache.hadoop.conf.Configuration;
 import org.apache.hadoop.hbase.*;
@@ -140,7 +141,9 @@ public BufferedMutator getBufferedMutator(BufferedMutatorParams params) throws I
 
     @Override
     public RegionLocator getRegionLocator(TableName tableName) throws IOException {
-        throw new FeatureNotSupportedException("not supported yet'");
+        ObTableClient obTableClient = ObTableClientManager.getOrCreateObTableClient(null/*args*/);
+        OHRegionLocatorExecutor executor = new OHRegionLocatorExecutor(obTableClient);
+        return executor.getRegionLocator(String.valueOf(tableName));
     }
 
     @Override

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

@@ -0,0 +1,84 @@
+package com.alipay.oceanbase.hbase.util;
+
+import com.alipay.oceanbase.rpc.ObTableClient;
+import org.apache.hadoop.hbase.HRegionLocation;
+import org.apache.hadoop.hbase.TableName;
+import org.apache.hadoop.hbase.client.RegionLocator;
+import org.apache.hadoop.hbase.util.Pair;
+
+import java.io.IOException;
+import java.util.Collections;
+import java.util.List;
+
+public class OHRegionLocator implements RegionLocator {
+    public OHRegionLocator(byte[][] startKeys, byte[][] endKeys) {
+        
+    }
+
+    @Override
+    public HRegionLocation getRegionLocation(byte[] bytes) throws IOException {
+        return null;
+    }
+
+    @Override
+    public HRegionLocation getRegionLocation(byte[] bytes, boolean b) throws IOException {
+        return null;
+    }
+
+    @Override
+    public List<HRegionLocation> getAllRegionLocations() throws IOException {
+        return Collections.emptyList();
+    }
+
+    /**
+     * Gets the starting row key for every region in the currently open table.
+     * <p>
+     * This is mainly useful for the MapReduce integration.
+     *
+     * @return Array of region starting row keys
+     * @throws IOException if a remote or network exception occurs
+     */
+    @Override
+    public byte[][] getStartKeys() throws IOException {
+        return null;
+    }
+
+    /**
+     * Gets the ending row key for every region in the currently open table.
+     * <p>
+     * This is mainly useful for the MapReduce integration.
+     *
+     * @return Array of region ending row keys
+     * @throws IOException if a remote or network exception occurs
+     */
+    @Override
+    public byte[][] getEndKeys() throws IOException {
+        return null;
+    }
+
+    /**
+     * Gets the starting and ending row keys for every region in the currently
+     * open table.
+     * <p>
+     * This is mainly useful for the MapReduce integration.
+     *
+     * @return Pair of arrays of region starting and ending row keys
+     * @throws IOException if a remote or network exception occurs
+     */
+    @Override
+    public Pair<byte[][], byte[][]> getStartEndKeys() throws IOException {
+        return null;
+    }
+
+    @Override
+    public TableName getName() {
+        return null;
+    }
+
+    private ObTableClient tableClient;
+
+    @Override
+    public void close() throws IOException {
+
+    }
+}

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

@@ -0,0 +1,49 @@
+package com.alipay.oceanbase.hbase.util;
+
+import com.alibaba.fastjson.JSON;
+import com.alipay.oceanbase.hbase.execute.AbstractObTableMetaExecutor;
+import com.alipay.oceanbase.rpc.ObTableClient;
+import com.alipay.oceanbase.rpc.exception.ObTableException;
+import com.alipay.oceanbase.rpc.location.model.TableEntry;
+import com.alipay.oceanbase.rpc.meta.ObTableMetaRequest;
+import com.alipay.oceanbase.rpc.meta.ObTableMetaResponse;
+import com.alipay.oceanbase.rpc.meta.ObTableRpcMetaType;
+
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.Map;
+
+public class OHRegionLocatorExecutor extends AbstractObTableMetaExecutor<OHRegionLocator> {
+    OHRegionLocatorExecutor(ObTableClient client) {
+        this.client = client;
+    }
+    
+    @Override
+    public ObTableRpcMetaType getMetaType() {
+        return ObTableRpcMetaType.HTABLE_REGION_LOCATOR;
+    }
+
+    @Override
+    public OHRegionLocator parse(ObTableMetaResponse response) throws IOException {
+        try {
+            String jsonData = response.getData();
+            // process json
+            return new OHRegionLocator(null, null);
+        } catch (IllegalArgumentException e) {
+            throw new IOException("msg", e);
+        }
+    }
+
+    public OHRegionLocator getRegionLocator(String tableName) throws IOException {
+        ObTableMetaRequest request = new ObTableMetaRequest();
+        request.setMetaType(getMetaType());
+        Map<String, Object> requestData = new HashMap<>();
+        requestData.put("table_name", tableName);
+        String jsonData = JSON.toJSONString(requestData);
+        request.setData(jsonData);
+
+        return execute(client, request);
+    }
+    
+    private final ObTableClient client;
+}