add NeedTabletId flag

Author: JackShi148

src/main/java/com/alipay/oceanbase/hbase/OHTable.java

@@ -2373,6 +2373,7 @@ private BatchOperation buildBatchOperation(String tableName, List<? extends Row>
         }
         batch.setEntityType(ObTableEntityType.HKV);
         batch.setServerCanRetry(OHBaseFuncUtils.serverCanRetry(obTableClient));
+        batch.setNeedTabletId(OHBaseFuncUtils.needTabletId(obTableClient));
         return batch;
     }
 
@@ -2478,6 +2479,7 @@ private ObTableQueryRequest buildObTableQueryRequest(ObTableQuery obTableQuery,
         request.setTableQuery(obTableQuery);
         request.setTableName(targetTableName);
         request.setServerCanRetry(OHBaseFuncUtils.serverCanRetry(obTableClient));
+        request.setNeedTabletId(OHBaseFuncUtils.needTabletId(obTableClient));
         return request;
     }
 
@@ -2492,6 +2494,7 @@ private ObTableQueryAsyncRequest buildObTableQueryAsyncRequest(ObTableQuery obTa
         asyncRequest.setTableName(targetTableName);
         asyncRequest.setObTableQueryRequest(request);
         asyncRequest.setServerCanRetry(OHBaseFuncUtils.serverCanRetry(obTableClient));
+        asyncRequest.setNeedTabletId(OHBaseFuncUtils.needTabletId(obTableClient));
         return asyncRequest;
     }
 
@@ -2507,6 +2510,7 @@ private ObTableQueryAndMutateRequest buildObTableQueryAndMutateRequest(ObTableQu
         request.setEntityType(ObTableEntityType.HKV);
         request.setReturningAffectedEntity(true);
         request.setServerCanRetry(OHBaseFuncUtils.serverCanRetry(obTableClient));
+        request.setNeedTabletId(OHBaseFuncUtils.needTabletId(obTableClient));
         return request;
     }
 

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

@@ -111,4 +111,15 @@ public static boolean serverCanRetry(ObTableClient tableClient) {
             return true;
         }
     }
+
+    public static boolean needTabletId(ObTableClient tableClient) {
+        if (tableClient.isOdpMode()) {
+            return ObGlobal.isDistributeNeedTabletIdSupport()
+                    && ObGlobal.OB_PROXY_VERSION >= ObGlobal.OB_PROXY_VERSION_4_3_6_0
+                    && tableClient.getServerCapacity().isSupportDistributedExecute();
+        } else {
+            return ObGlobal.isDistributeNeedTabletIdSupport()
+                    && tableClient.getServerCapacity().isSupportDistributedExecute();
+        }
+    }
 }