[Bug] Fix the IndexOutOfBoundsException when lookup empty sortedRun (a…

…pache#2530)
tongcheng-elong · Dec 20, 2023 · 998e5b1 · 998e5b1
1 parent 1308150
commit 998e5b1
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/paimon-core/src/main/java/org/apache/paimon/mergetree/LookupUtils.java b/paimon-core/src/main/java/org/apache/paimon/mergetree/LookupUtils.java
@@ -59,6 +59,9 @@ public static <T> T lookup(
             SortedRun level,
             BiFunctionWithIOE<InternalRow, DataFileMeta, T> lookup)
             throws IOException {
+        if (!level.nonEmpty()) {
+            return null;
+        }
         List<DataFileMeta> files = level.files();
         int left = 0;
         int right = files.size() - 1;

diff --git a/paimon-core/src/test/java/org/apache/paimon/mergetree/LookupLevelsTest.java b/paimon-core/src/test/java/org/apache/paimon/mergetree/LookupLevelsTest.java
@@ -198,6 +198,22 @@ public void testMaxDiskSize() throws IOException {
         assertThat(lookupLevels.lookupFiles().estimatedSize()).isEqualTo(0);
     }
 
+    @Test
+    public void testLookupEmptyLevel() throws IOException {
+        Levels levels =
+                new Levels(
+                        comparator,
+                        Arrays.asList(
+                                newFile(1, kv(1, 11), kv(3, 33), kv(5, 5)),
+                                // empty level 2
+                                newFile(3, kv(2, 22), kv(5, 55))),
+                        3);
+        LookupLevels lookupLevels = createLookupLevels(levels, MemorySize.ofMebiBytes(10));
+
+        KeyValue kv = lookupLevels.lookup(row(2), 1);
+        assertThat(kv).isNotNull();
+    }
+
     private LookupLevels createLookupLevels(Levels levels, MemorySize maxDiskSize) {
         return new LookupLevels(
                 levels,