[Enhancement] Reuse the label for retry to avoid duplicate load (#284)

Signed-off-by: PengFei Li <[email protected]>
StarRocks · Sep 13, 2023 · 21a970d · 21a970d
1 parent 36e2d25
commit 21a970d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...ream-load-sdk/src/main/java/com/starrocks/data/load/stream/v2/TransactionTableRegion.java b/...ream-load-sdk/src/main/java/com/starrocks/data/load/stream/v2/TransactionTableRegion.java
@@ -125,6 +125,10 @@ public LabelGenerator getLabelGenerator() {
 
     @Override
     public void setLabel(String label) {
+        // Reuse the same label to avoid duplicate load if retry happens
+        if (numRetries > 0 && label != null) {
+            return;
+        }
         this.label = label;
     }