switch on the idempotent flag for retries

ydb-platform · Nov 19, 2024 · 62ed6d6 · 62ed6d6
1 parent 18882c5
commit 62ed6d6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/java/tech/ydb/spark/connector/impl/YdbConnector.java b/src/main/java/tech/ydb/spark/connector/impl/YdbConnector.java
@@ -61,7 +61,10 @@ public YdbConnector(String catalogName, Map<String, String> props) {
                     .sessionPoolSize(1, poolSize)
                     .build();
             this.schemeClient = SchemeClient.newClient(gt).build();
-            this.retryCtx = SessionRetryContext.create(tableClient).build();
+            this.retryCtx = SessionRetryContext.create(tableClient)
+                    .idempotent(true)
+                    .maxRetries(20)
+                    .build();
             this.transport = gt;
             gt = null; // to avoid closing below
         } finally {