fix

Author: JingsongLi

paimon-common/src/main/java/org/apache/paimon/utils/RetryUtils.java

@@ -21,13 +21,14 @@
 import org.apache.paimon.fs.FileIO;
 import org.apache.paimon.fs.Path;
 import org.apache.paimon.utils.JsonSerdeUtil;
-import org.apache.paimon.utils.RetryUtils;
 
 import org.apache.paimon.shade.jackson2.com.fasterxml.jackson.annotation.JsonCreator;
 import org.apache.paimon.shade.jackson2.com.fasterxml.jackson.annotation.JsonGetter;
 import org.apache.paimon.shade.jackson2.com.fasterxml.jackson.annotation.JsonIgnoreProperties;
 import org.apache.paimon.shade.jackson2.com.fasterxml.jackson.annotation.JsonProperty;
 
+import java.io.IOException;
+import java.io.UncheckedIOException;
 import java.util.Optional;
 
 /** Consumer which contains next snapshot. */
@@ -57,7 +58,33 @@ public static Consumer fromJson(String json) {
     }
 
     public static Optional<Consumer> fromPath(FileIO fileIO, Path path) {
-        return RetryUtils.retry(
-                () -> fileIO.readOverwrittenFileUtf8(path).orElse(null), Consumer::fromJson);
+        int retryNumber = 0;
+        Exception exception = null;
+        while (retryNumber++ < 10) {
+            Optional<String> content;
+            try {
+                content = fileIO.readOverwrittenFileUtf8(path);
+            } catch (IOException e) {
+                throw new UncheckedIOException(e);
+            }
+
+            if (!content.isPresent()) {
+                return Optional.empty();
+            }
+
+            try {
+                return content.map(Consumer::fromJson);
+            } catch (Exception e) {
+                // retry
+                exception = e;
+                try {
+                    Thread.sleep(200);
+                } catch (InterruptedException ie) {
+                    Thread.currentThread().interrupt();
+                    throw new RuntimeException(ie);
+                }
+            }
+        }
+        throw new RuntimeException("Retry fail after 10 times", exception);
     }
 }

paimon-core/src/main/java/org/apache/paimon/consumer/Consumer.java

@@ -770,7 +770,31 @@ public static Snapshot fromPath(FileIO fileIO, Path path) {
     }
 
     public static Snapshot tryFromPath(FileIO fileIO, Path path) throws FileNotFoundException {
-        return RetryUtils.retry(() -> fileIO.readFileUtf8(path), Snapshot::fromJson)
-                .orElseThrow(() -> new RuntimeException("Cannot read snapshot from " + path));
+        int retryNumber = 0;
+        Exception exception = null;
+        while (retryNumber++ < 10) {
+            String content;
+            try {
+                content = fileIO.readFileUtf8(path);
+            } catch (FileNotFoundException e) {
+                throw e;
+            } catch (IOException e) {
+                throw new RuntimeException("Fails to read snapshot from path " + path, e);
+            }
+
+            try {
+                return Snapshot.fromJson(content);
+            } catch (Exception e) {
+                // retry
+                exception = e;
+                try {
+                    Thread.sleep(200);
+                } catch (InterruptedException ie) {
+                    Thread.currentThread().interrupt();
+                    throw new RuntimeException(ie);
+                }
+            }
+        }
+        throw new RuntimeException("Retry fail after 10 times", exception);
     }
 }

paimon-core/src/main/java/org/apache/paimon/utils/SnapshotManager.java

@@ -33,6 +33,7 @@
 
 import javax.annotation.Nullable;
 
+import java.io.File;
 import java.io.IOException;
 import java.util.ArrayList;
 import java.util.HashSet;
@@ -46,6 +47,7 @@
 import static org.apache.paimon.SnapshotTest.newSnapshotManager;
 import static org.apache.paimon.catalog.Identifier.DEFAULT_MAIN_BRANCH;
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.Assertions.assertThatThrownBy;
 import static org.assertj.core.api.Assertions.fail;
 import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
 
@@ -54,6 +56,17 @@ public class SnapshotManagerTest {
 
     @TempDir java.nio.file.Path tempDir;
 
+    @Test
+    public void testRetry() throws IOException {
+        SnapshotManager snapshotManager =
+                newSnapshotManager(LocalFileIO.create(), new Path(tempDir.toString()));
+        File file = new File(tempDir.toFile(), "/snapshot/snapshot-1");
+        file.getParentFile().mkdir();
+        file.createNewFile();
+        assertThatThrownBy(() -> snapshotManager.snapshot(1))
+                .hasMessageContaining("Retry fail after 10 times");
+    }
+
     @Test
     public void testSnapshotPath() {
         SnapshotManager snapshotManager =