reschedule all operations if meet shoud retry exceptions (#408)

Author: JackShi148

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

@@ -414,11 +414,20 @@ private Row calculateRowKey(ObPair<Integer, ObTableSingleOp> operationPair) {
         return rowKey;
     }
 
-    private BatchIdxOperationPairList extractOperations(TabletOperationsMap tabletOperationsMap) {
+    private BatchIdxOperationPairList extractOperations(Map.Entry<Long, TabletOperationsMap> currentEntry,
+                                                        Iterator<Map.Entry<Long, TabletOperationsMap>> iterator) {
+        // only reschedule operations from the current entry to avoid repeatedly executing operations
         BatchIdxOperationPairList operationsWithIndex = new BatchIdxOperationPairList();
-        for (ObPair<ObTableParam, BatchIdxOperationPairList> pair : tabletOperationsMap.values()) {
+        TabletOperationsMap currentTabletMap = currentEntry.getValue();
+        for (ObPair<ObTableParam, BatchIdxOperationPairList> pair : currentTabletMap.values()) {
             operationsWithIndex.addAll(pair.getRight());
         }
+        while (iterator.hasNext()) {
+            currentTabletMap = iterator.next().getValue();
+            for (ObPair<ObTableParam, BatchIdxOperationPairList> pair : currentTabletMap.values()) {
+                operationsWithIndex.addAll(pair.getRight());
+            }
+        }
         return operationsWithIndex;
     }
 
@@ -847,16 +856,17 @@ private void executeWithRetries(ObTableSingleOpResult[] results,
             }
             boolean allPartitionsSuccess = true;
 
-            for (Map.Entry<Long, TabletOperationsMap> currentEntry : currentPartitions.entrySet()) {
+            Iterator<Map.Entry<Long, TabletOperationsMap>> iterator = currentPartitions.entrySet().iterator();
+            while (iterator.hasNext()) {
+                Map.Entry<Long, TabletOperationsMap> currentEntry = iterator.next();
                 try {
                     partitionExecute(results, currentEntry);
                 } catch (Exception e) {
                     if (shouldRetry(e)) {
                         retryCount++;
                         errCode = ((ObTableNeedFetchMetaException) e).getErrorCode();
                         errMsg = e.getMessage();
-                        BatchIdxOperationPairList failedOperations = extractOperations(currentEntry
-                            .getValue());
+                        BatchIdxOperationPairList failedOperations = extractOperations(currentEntry, iterator); // reschedule failed and sequent operations
                         currentPartitions = prepareOperations(failedOperations);
                         allPartitionsSuccess = false;
                         break;