ISPN-14189 Compute the size after a query

Probably there is an issue on HSearch
infinispan · May 19, 2023 · 72fc8f0 · 72fc8f0
1 parent e1ae46b
commit 72fc8f0
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/query/src/main/java/org/infinispan/query/stats/impl/LocalIndexStatistics.java b/query/src/main/java/org/infinispan/query/stats/impl/LocalIndexStatistics.java
@@ -64,11 +64,13 @@ public CompletionStage<Map<String, IndexInfo>> computeIndexInfos() {
    private CompletionStage<IndexInfo> indexInfos(SearchIndexedEntity indexedEntity) {
       SearchSession session = searchMapping.getMappingSession();
       SearchScope<?> scope = session.scope(indexedEntity.javaClass(), indexedEntity.name());
+
       CompletionStage<Long> countStage = blockingManager.supplyBlocking(
             () -> session.search(scope).where(SearchPredicateFactory::matchAll).fetchTotalHitCount(), this);
-      CompletionStage<Long> sizeStage = indexedEntity.indexManager().unwrap(LuceneIndexManager.class)
-            .computeSizeInBytesAsync();
-      return countStage.thenCombine(sizeStage, IndexInfo::new);
+
+      return countStage
+            .thenCompose(count -> indexedEntity.indexManager().unwrap(LuceneIndexManager.class).computeSizeInBytesAsync()
+            .thenApply(size -> new IndexInfo(count, size)));
    }
 
    @Override