Fix test failure in TestHistogramCollectorManager

apache · Feb 21, 2025 · e779e55 · e779e55
1 parent 5491c94
commit e779e55
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/...sandbox/src/java/org/apache/lucene/sandbox/facet/plain/histograms/HistogramCollector.java b/...sandbox/src/java/org/apache/lucene/sandbox/facet/plain/histograms/HistogramCollector.java
@@ -262,7 +262,7 @@ public void finish() throws IOException {
     }
   }
 
-  private static void checkMaxBuckets(int size, int maxBuckets) {
+  static void checkMaxBuckets(int size, int maxBuckets) {
     if (size > maxBuckets) {
       throw new IllegalStateException(
           "Collected "

diff --git a/.../src/java/org/apache/lucene/sandbox/facet/plain/histograms/HistogramCollectorManager.java b/.../src/java/org/apache/lucene/sandbox/facet/plain/histograms/HistogramCollectorManager.java
@@ -89,6 +89,7 @@ public LongIntHashMap reduce(Collection<HistogramCollector> collectors) throws I
     for (HistogramCollector collector : collectors) {
       for (LongIntCursor cursor : collector.getCounts()) {
         reduced.addTo(cursor.key, cursor.value);
+        HistogramCollector.checkMaxBuckets(reduced.size(), maxBuckets);
       }
     }
     return reduced;