[core] RollingFileWriter#abort shouldn't abort currentWriter twice

apache · Nov 19, 2024 · 672fc4d · 672fc4d
1 parent 220789d
commit 672fc4d
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/paimon-core/src/main/java/org/apache/paimon/io/RollingFileWriter.java b/paimon-core/src/main/java/org/apache/paimon/io/RollingFileWriter.java
@@ -78,7 +78,12 @@ public void write(T row) throws IOException {
                 openCurrentWriter();
             }
 
-            currentWriter.write(row);
+            try {
+                currentWriter.write(row);
+            } catch (Throwable e) {
+                currentWriter = null;
+                throw e;
+            }
             recordCount += 1;
 
             if (rollingFile()) {
@@ -102,7 +107,12 @@ public void writeBundle(BundleRecords bundle) throws IOException {
                 openCurrentWriter();
             }
 
-            currentWriter.writeBundle(bundle);
+            try {
+                currentWriter.writeBundle(bundle);
+            } catch (Throwable e) {
+                currentWriter = null;
+                throw e;
+            }
             recordCount += bundle.rowCount();
 
             if (rollingFile()) {