From 31ff946e5b9f70f44a2c43e533bda0a6318c3e62 Mon Sep 17 00:00:00 2001
From: Sagar Upadhyaya <sagar.upadhyaya.121@gmail.com>
Date: Wed, 26 Aug 2026 13:03:31 -0700
Subject: [PATCH] Avoid cancelltion checks on every advance() call for posting
 enums

Signed-off-by: Sagar Upadhyaya <sagar.upadhyaya.121@gmail.com>
---
 .../org/opensearch/search/internal/ExitableDirectoryReader.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server/src/main/java/org/opensearch/search/internal/ExitableDirectoryReader.java b/server/src/main/java/org/opensearch/search/internal/ExitableDirectoryReader.java
index 9187c4c07cba0..744649766be15 100644
--- a/server/src/main/java/org/opensearch/search/internal/ExitableDirectoryReader.java
+++ b/server/src/main/java/org/opensearch/search/internal/ExitableDirectoryReader.java
@@ -255,7 +255,7 @@ public int nextDoc() throws IOException {
 
         @Override
         public int advance(int target) throws IOException {
-            queryCancellation.checkCancelled();
+            checkAndThrowWithSampling();
             return super.advance(target);
         }
     }