mutation,query_and_mutate set a isUserSpecifiedT based on T (#300)

Author: maochongxin

src/main/java/com/alipay/oceanbase/rpc/ObTableClient.java

@@ -3640,7 +3640,15 @@ ObTableQueryAndMutateRequest obTableQueryAndMutate(final ObTableOperation operat
         operations.addTableOperation(operation);
 
         ObTableQueryAndMutate queryAndMutate = buildObTableQueryAndMutate(obTableQuery, operations);
-
+        if (runningMode == RunningMode.HBASE) {
+            if (operation.getEntity() != null || operation.getEntity().getRowKeySize() != 3) {
+                throw new IllegalArgumentException("rowkey size is not 3");
+            }
+            long ts = (long)operation.getEntity().getRowKeyValue(2).getValue();
+            if (ts != -Long.MAX_VALUE) {
+                queryAndMutate.setIsUserSpecifiedT(true);
+            }
+        }
         ObTableQueryAndMutateRequest request = buildObTableQueryAndMutateRequest(queryAndMutate,
             tableName);
 
@@ -4541,4 +4549,17 @@ public byte[][] getHBaseTableEndKeys(String hbaseTableName) throws Exception {
 
         return endKeys;
     }
+    public static void setRowKeyValue(Mutation mutation, int index, Object value) {
+        if (mutation.getRowKeyValues() == null || (index < 0 || mutation.getRowKeyValues().size() <= index)) {
+            throw new IllegalArgumentException("rowkey is null or index is out of range");
+        }
+        ((ObObj) mutation.getRowKeyValues().get(index)).setValue(value);
+    }
+
+    public static Object getRowKeyValue(Mutation mutation, int index) {
+        if (mutation.getRowKeyValues() == null || (index < 0 || index >= mutation.getRowKeyValues().size())) {
+            throw new IllegalArgumentException("rowkey is null or index is out of range");
+        }
+        return ((ObObj) mutation.getRowKeyValues().get(index)).getValue();
+    }
 }

src/main/java/com/alipay/oceanbase/rpc/mutation/BatchOperation.java

@@ -362,8 +362,8 @@ private void negateHbaseTimestamp(Mutation mutation) {
         if (rowKey == null || rowKey.size() != 3) {
             throw new IllegalArgumentException("hbase rowkey length must be 3");
         } else {
-            long ts = ((long) ((ObObj) mutation.getRowKeyValues().get(2)).getValue());
-            ((ObObj) mutation.getRowKeyValues().get(2)).setValue(-ts);
+            long ts = (long)ObTableClient.getRowKeyValue(mutation, 2);
+            ObTableClient.setRowKeyValue(mutation, 2, -ts);
         }
     }
 }

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableSingleOp.java

@@ -131,6 +131,10 @@ public void setIsCheckNoExists(boolean isCheckNoExists) {
     public void setIsRollbackWhenCheckFailed(boolean isRollbackWhenCheckFailed) {
         singleOpFlag.setIsRollbackWhenCheckFailed(isRollbackWhenCheckFailed);
     }
+    
+    public void setIsUserSpecifiedT(boolean isUserSpecifiedT) {
+        singleOpFlag.setIsUserSpecifiedT(isUserSpecifiedT);
+    }
 
     public ObTableOperationType getSingleOpType() {
         return singleOpType;

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/mutate/ObTableQueryAndMutate.java

@@ -164,6 +164,10 @@ public void setIsCheckAndExecute(boolean isCheckAndExecute) {
     public void setIsCheckNoExists(boolean isCheckNoExists) {
         queryAndMutateFlag.setIsCheckNotExists(isCheckNoExists);
     }
+    
+    public void setIsUserSpecifiedT(boolean isUserSpecifiedT) {
+        queryAndMutateFlag.setIsUserSpecifiedT(isUserSpecifiedT);
+    }
 
     public void setIsRollbackWhenCheckFailed(boolean isRollbackWhenCheckFailed) {
         queryAndMutateFlag.setIsRollbackWhenCheckFailed(isRollbackWhenCheckFailed);

src/main/java/com/alipay/oceanbase/rpc/table/ObTableClientLSBatchOpsImpl.java

@@ -298,6 +298,12 @@ public void addOperation(Mutation mutation) throws Exception {
         ObTableSingleOp singleOp = new ObTableSingleOp();
         singleOp.setSingleOpType(type);
         singleOp.addEntity(entity);
+        if (ObTableClient.RunningMode.HBASE == obTableClient.getRunningMode()) {
+            long ts = (long) ObTableClient.getRowKeyValue(mutation, 2);
+            if (ts != -Long.MAX_VALUE) {
+                singleOp.setIsUserSpecifiedT(true);
+            }
+        }
         addOperation(singleOp);
     }