fix

apache · Aug 1, 2024 · 926c4ee · 926c4ee
1 parent 3bf7d5e
commit 926c4ee
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 10 deletions.
diff --git a/paimon-core/src/main/java/org/apache/paimon/utils/ObjectsCache.java b/paimon-core/src/main/java/org/apache/paimon/utils/ObjectsCache.java
@@ -35,7 +35,6 @@
 import java.io.IOException;
 import java.util.ArrayList;
 import java.util.List;
-import java.util.function.BiFunction;
 
 import static org.apache.paimon.utils.ObjectsFile.readFromIterator;
 
@@ -47,14 +46,14 @@ public class ObjectsCache<K, V> {
     private final ObjectSerializer<V> projectedSerializer;
     private final ThreadLocal<InternalRowSerializer> formatSerializer;
     private final FunctionWithIOException<K, Long> fileSizeFunction;
-    private final BiFunction<K, Long, CloseableIterator<InternalRow>> reader;
+    private final BiFunctionWithIOE<K, Long, CloseableIterator<InternalRow>> reader;
 
     public ObjectsCache(
             SegmentsCache<K> cache,
             ObjectSerializer<V> projectedSerializer,
             RowType formatSchema,
             FunctionWithIOException<K, Long> fileSizeFunction,
-            BiFunction<K, Long, CloseableIterator<InternalRow>> reader) {
+            BiFunctionWithIOE<K, Long, CloseableIterator<InternalRow>> reader) {
         this.cache = cache;
         this.projectedSerializer = projectedSerializer;
         this.formatSerializer =

diff --git a/paimon-core/src/main/java/org/apache/paimon/utils/ObjectsFile.java b/paimon-core/src/main/java/org/apache/paimon/utils/ObjectsFile.java
@@ -165,13 +165,10 @@ public String writeWithoutRolling(Iterator<T> records) {
         }
     }
 
-    private CloseableIterator<InternalRow> createIterator(Path file, @Nullable Long fileSize) {
-        try {
-            return FileUtils.createFormatReader(fileIO, readerFactory, file, fileSize)
-                    .toCloseableIterator();
-        } catch (IOException e) {
-            throw new RuntimeException(e);
-        }
+    private CloseableIterator<InternalRow> createIterator(Path file, @Nullable Long fileSize)
+            throws IOException {
+        return FileUtils.createFormatReader(fileIO, readerFactory, file, fileSize)
+                .toCloseableIterator();
     }
 
     private long fileSize(Path file) throws IOException {