[core] Add limit to bundle interface (apache#4115)

herefree · Sep 3, 2024 · 6b8d72b · 6b8d72b
1 parent c574ae3
commit 6b8d72b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/paimon-core/src/main/java/org/apache/paimon/io/StatsCollectingSingleFileWriter.java b/paimon-core/src/main/java/org/apache/paimon/io/StatsCollectingSingleFileWriter.java
@@ -73,6 +73,15 @@ public void write(T record) throws IOException {
         }
     }
 
+    @Override
+    public void writeBundle(BundleRecords bundle) throws IOException {
+        Preconditions.checkState(
+                simpleStatsExtractor != null,
+                "Can't write bundle without simpleStatsExtractor, we may lose all the statistical information");
+
+        super.writeBundle(bundle);
+    }
+
     public SimpleColStats[] fieldStats() throws IOException {
         Preconditions.checkState(closed, "Cannot access metric unless the writer is closed.");
         if (simpleStatsExtractor != null) {