revert batch atomic change

Author: JackShi148

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

@@ -42,7 +42,7 @@
 public class ObTableBatchOperationRequest extends ObTableAbstractOperationRequest {
 
     private ObTableBatchOperation batchOperation;
-    private boolean               batchOperationAsAtomic = true;
+    private boolean               batchOperationAsAtomic = false;
 
     /*
      * Get pcode.
@@ -155,6 +155,13 @@ public boolean isBatchOperationAsAtomic() {
         return batchOperationAsAtomic;
     }
 
+    /*
+     * Set BatchOperation As Atomic.
+     */
+    public void setBatchOperationAsAtomic(boolean batchOperationAsAtomic) {
+        this.batchOperationAsAtomic = batchOperationAsAtomic;
+    }
+
     public void setBatchOpReturnOneResult(boolean returnOneResult) {
         if (returnOneResult == true) {
             this.option_flag.setReturnOneResult(true);

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

@@ -139,6 +139,7 @@ public void addObTableOperation(ObTableOperationType type, Object[] rowkeys, Str
      */
     public List<Object> execute() throws RemotingException, InterruptedException {
 
+        request.setBatchOperationAsAtomic(isAtomicOperation());
         request.setBatchOpReturnOneResult(isReturnOneResult());
         Object result = obTable.execute(request);
         checkObTableOperationResult(result);
@@ -172,6 +173,7 @@ public List<Object> execute() throws RemotingException, InterruptedException {
      */
     public List<Object> executeWithResult() throws Exception {
 
+        request.setBatchOperationAsAtomic(isAtomicOperation());
         request.setBatchOpReturnOneResult(isReturnOneResult());
         Object result = obTable.execute(request);
         checkObTableOperationResult(result);

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

@@ -222,7 +222,6 @@ public Map<Long, ObPair<ObTableParam, List<ObPair<Integer, ObTableOperation>>>>
         List<ObTableOperation> operations = batchOperation.getTableOperations();
         Map<Long, ObPair<ObTableParam, List<ObPair<Integer, ObTableOperation>>>> partitionOperationsMap = new HashMap<Long, ObPair<ObTableParam, List<ObPair<Integer, ObTableOperation>>>>();
 
-        Long firstPartId = null;
         for (int i = 0; i < operations.size(); i++) {
             ObTableOperation operation = operations.get(i);
             ObRowKey rowKeyObject = operation.getEntity().getRowKey();
@@ -238,14 +237,6 @@ public Map<Long, ObPair<ObTableParam, List<ObPair<Integer, ObTableOperation>>>>
             } else {
                 tableObPair = obTableClient.getODPTableWithRowKeyValue(tableName, rowKey, false);
             }
-            if (firstPartId == null) {
-                firstPartId = tableObPair.getLeft();
-            } else if (atomicOperation) {
-                if (!firstPartId.equals(tableObPair.getLeft())) {
-                    throw new IllegalStateException(
-                        "The operations should be atomic but query from different partitions");
-                }
-            }
             ObPair<ObTableParam, List<ObPair<Integer, ObTableOperation>>> obTableOperations = partitionOperationsMap
                 .get(tableObPair.getLeft());
             if (obTableOperations == null) {
@@ -293,6 +284,7 @@ public void partitionExecute(ObTableOperationResult[] results,
             subRequest.setConsistencyLevel(obTableClient.getReadConsistency()
                 .toObTableConsistencyLevel());
         }
+        subRequest.setBatchOperationAsAtomic(isAtomicOperation());
         subRequest.setBatchOpReturnOneResult(isReturnOneResult());
         ObTableBatchOperationResult subObTableBatchOperationResult;
 

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

@@ -316,7 +316,6 @@ public Map<Long, Map<Long, ObPair<ObTableParam, List<ObPair<Integer, ObTableSing
         Map<Long, Map<Long, ObPair<ObTableParam, List<ObPair<Integer, ObTableSingleOp>>>>> lsOperationsMap =
                 new HashMap();
 
-        Long firstPartId = null;
         for (int i = 0; i < operations.size(); i++) {
             ObTableSingleOp operation = operations.get(i);
             List<ObObj> rowkeyObjs = operation.getRowkeyObjs();
@@ -332,14 +331,6 @@ public Map<Long, Map<Long, ObPair<ObTableParam, List<ObPair<Integer, ObTableSing
             } else {
                 tableObPair = obTableClient.getODPTableWithRowKeyValue(tableName, rowKey, false);
             }
-            if(firstPartId == null) {
-                firstPartId = tableObPair.getLeft();
-            }
-            else if(atomicOperation) {
-                if(!firstPartId.equals(tableObPair.getLeft())) {
-                    throw new IllegalStateException("The operations should be atomic but query from different partitions");
-                }
-            }
             long lsId = tableObPair.getRight().getLsId();
 
             Map<Long, ObPair<ObTableParam, List<ObPair<Integer, ObTableSingleOp>>>> tabletOperations