make sure increment before schedule

Signed-off-by: Kaituo Li <[email protected]>
opensearch-project · Oct 24, 2024 · 2b497b9 · 2b497b9
1 parent 64da191
commit 2b497b9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/java/org/opensearch/timeseries/transport/ResultProcessor.java b/src/main/java/org/opensearch/timeseries/transport/ResultProcessor.java
@@ -229,16 +229,16 @@ class PageListener implements ActionListener<CompositeRetriever.Page> {
 
         @Override
         public void onResponse(CompositeRetriever.Page entityFeatures) {
+            // Increment pagesInFlight to track the processing of this page
+            pagesInFlight.incrementAndGet();
+
             // start processing next page after sending out features for previous page
             if (pageIterator.hasNext()) {
                 pageIterator.next(this);
             } else if (config.getImputationOption() != null) {
                 scheduleImputeHCTask();
             }
 
-            // Increment pagesInFlight to track the processing of this page
-            pagesInFlight.incrementAndGet();
-
             if (entityFeatures != null && false == entityFeatures.isEmpty()) {
                 LOG
                     .info(