Fix wrong condition in block_wand_single_scorer and add debug_assert …

…to have an equality check on doc to break the loop.
quickwit-oss · Oct 31, 2021 · afbf2e3 · afbf2e3
1 parent 2aee4fb
commit afbf2e3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/query/boolean_query/block_wand.rs b/src/query/boolean_query/block_wand.rs
@@ -231,7 +231,7 @@ pub fn block_wand_single_scorer(
         // the threshold.
         while scorer.block_max_score() < threshold {
             let last_doc_in_block = scorer.last_doc_in_block();
-            if doc == TERMINATED {
+            if last_doc_in_block == TERMINATED {
                 return;
             }
             doc = last_doc_in_block + 1;
@@ -247,7 +247,7 @@ pub fn block_wand_single_scorer(
             if score > threshold {
                 threshold = callback(doc, score);
             }
-            if doc >= scorer.last_doc_in_block() {
+            if doc == scorer.last_doc_in_block() {
                 break;
             }
             doc = scorer.advance();