adapt for disable/enable table test (#228)

* adapt for disable/enable table

Author: GroundWu

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

@@ -4,6 +4,7 @@
 import com.alipay.oceanbase.rpc.bolt.transport.TransportCodes;
 import com.alipay.oceanbase.hbase.exception.FeatureNotSupportedException;
 import com.alipay.oceanbase.rpc.exception.ObTableTransportException;
+import com.alipay.oceanbase.rpc.meta.ObTableRpcMetaType;
 import org.apache.hadoop.conf.Configuration;
 import org.apache.hadoop.hbase.*;
 import org.apache.hadoop.hbase.client.*;
@@ -243,7 +244,19 @@ public Future<Void> truncateTableAsync(TableName tableName, boolean b) throws IO
 
     @Override
     public void enableTable(TableName tableName) throws IOException {
-        throw new FeatureNotSupportedException("does not support yet");
+        OHConnectionConfiguration connectionConf = new OHConnectionConfiguration(conf);
+        ObTableClient tableClient = ObTableClientManager.getOrCreateObTableClientByTableName(tableName, connectionConf);
+        OHTableAccessControlExecutor executor = new OHTableAccessControlExecutor(tableClient, ObTableRpcMetaType.HTABLE_ENABLE_TABLE);
+        try {
+            executor.enableTable(tableName.getNameAsString());
+        } catch (IOException e) {
+            if (e.getCause() instanceof ObTableTransportException
+                    && ((ObTableTransportException) e.getCause()).getErrorCode() == TransportCodes.BOLT_TIMEOUT) {
+                throw new TimeoutIOException(e.getCause());
+            } else {
+                throw e;
+            }
+        }
     }
 
     @Override
@@ -268,7 +281,19 @@ public Future<Void> disableTableAsync(TableName tableName) throws IOException {
 
     @Override
     public void disableTable(TableName tableName) throws IOException {
-        throw new FeatureNotSupportedException("does not support yet");
+        OHConnectionConfiguration connectionConf = new OHConnectionConfiguration(conf);
+        ObTableClient tableClient = ObTableClientManager.getOrCreateObTableClientByTableName(tableName, connectionConf);
+        OHTableAccessControlExecutor executor = new OHTableAccessControlExecutor(tableClient, ObTableRpcMetaType.HTABLE_DISABLE_TABLE);
+        try {
+            executor.disableTable(tableName.getNameAsString());
+        } catch (IOException e) {
+            if (e.getCause() instanceof ObTableTransportException
+                    && ((ObTableTransportException) e.getCause()).getErrorCode() == TransportCodes.BOLT_TIMEOUT) {
+                throw new TimeoutIOException(e.getCause());
+            } else {
+                throw e;
+            }
+        }
     }
 
     @Override
@@ -283,12 +308,19 @@ public HTableDescriptor[] disableTables(Pattern pattern) throws IOException {
 
     @Override
     public boolean isTableEnabled(TableName tableName) throws IOException {
-        throw new FeatureNotSupportedException("does not support yet");
+        return isDisabled(tableName) == false;
     }
 
     @Override
     public boolean isTableDisabled(TableName tableName) throws IOException {
-        throw new FeatureNotSupportedException("does not support yet");
+        return isDisabled(tableName) == true;
+    }
+
+    private boolean isDisabled(TableName tableName) throws IOException {
+        OHConnectionConfiguration connectionConf = new OHConnectionConfiguration(conf);
+        ObTableClient tableClient = ObTableClientManager.getOrCreateObTableClientByTableName(tableName, connectionConf);
+        OHTableDescriptorExecutor tableDescriptor = new OHTableDescriptorExecutor(tableName.getNameAsString(), tableClient);
+        return tableDescriptor.isDisable();
     }
 
     @Override

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

@@ -1,7 +1,6 @@
 package com.alipay.oceanbase.hbase.util;
 
 import com.alibaba.fastjson.JSON;
-import com.alibaba.fastjson.JSONObject;
 import com.alipay.oceanbase.hbase.execute.AbstractObTableMetaExecutor;
 import com.alipay.oceanbase.rpc.ObTableClient;
 import com.alipay.oceanbase.rpc.meta.ObTableMetaRequest;
@@ -38,7 +37,7 @@ public void enableTable(String tableName) throws IOException, TableNotFoundExcep
         ObTableMetaRequest request = new ObTableMetaRequest();
         request.setMetaType(getMetaType());
         Map<String, Object> requestData = new HashMap<>();
-        requestData.put("name", tableName);
+        requestData.put("table_name", tableName);
         String jsonData = JSON.toJSONString(requestData);
         request.setData(jsonData);
         execute(tableClient, request);
@@ -48,7 +47,7 @@ public void disableTable(String tableName) throws IOException, TableNotFoundExce
         ObTableMetaRequest request = new ObTableMetaRequest();
         request.setMetaType(getMetaType());
         Map<String, Object> requestData = new HashMap<>();
-        requestData.put("name", tableName);
+        requestData.put("table_name", tableName);
         String jsonData = JSON.toJSONString(requestData);
         request.setData(jsonData);
         execute(tableClient, request);

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

@@ -7,6 +7,7 @@
 import com.alipay.oceanbase.rpc.meta.ObTableMetaRequest;
 import com.alipay.oceanbase.rpc.meta.ObTableMetaResponse;
 import com.alipay.oceanbase.rpc.meta.ObTableRpcMetaType;
+import com.alipay.oceanbase.rpc.table.ObTable;
 import org.apache.hadoop.hbase.HColumnDescriptor;
 import org.apache.hadoop.hbase.HTableDescriptor;
 import org.apache.hadoop.hbase.TableName;
@@ -82,4 +83,51 @@ public HTableDescriptor getTableDescriptor() throws IOException {
 
         return execute(client, request);
     }
+
+    public boolean isDisable() throws IOException {
+        boolean isDisable = false;
+        final ObTableMetaRequest request = new ObTableMetaRequest();
+        request.setMetaType(getMetaType());
+        final Map<String, String> requestData = new HashMap<>();
+        requestData.put("table_name", tableName);
+
+        final String jsonData = JSON.toJSONString(requestData);
+        request.setData(jsonData);
+        try {
+            ObTableMetaResponse response = innerExecute(client, request);
+            final String responseData = response.getData();
+            final JSONObject jsonMap = Optional.<JSONObject>ofNullable(JSON.parseObject(responseData))
+                    .orElseThrow(() -> new IOException("jsonMap is null"));
+            JSONObject tbDesc = jsonMap.getJSONObject("tableDesc");
+            if (tbDesc != null) {
+                String state = tbDesc.getString("state");
+                if (state.compareToIgnoreCase("disable") == 0) {
+                    isDisable = true;
+                } else {
+                    isDisable = false;
+                }
+            }
+        } catch (IOException e) {
+            throw e;
+        }
+        return isDisable;
+    }
+
+    private ObTableMetaResponse innerExecute(ObTableClient client, ObTableMetaRequest request) throws IOException {
+        if (request.getMetaType() != getMetaType()) {
+            throw new IOException("Invalid meta type, expected " + getMetaType());
+        }
+        ObTable table = client.getRandomTable();
+        ObTableMetaResponse response;
+        try {
+            response = (ObTableMetaResponse) client.executeWithRetry(
+                    table,
+                    request,
+                    null /*tableName*/
+            );
+        } catch (Exception e) {
+            throw new IOException("Failed to execute request", e);
+        }
+        return response;
+    }
 }