Update

Signed-off-by: Peng Huo <penghuo@gmail.com>

Author: penghuo

opensearch/src/main/java/org/opensearch/sql/opensearch/storage/scan/BackgroundSearchScanner.java

@@ -172,8 +172,11 @@ public SearchBatchResult fetchNextBatch(OpenSearchRequest request) {
 
       // Pre-fetch next batch if needed
       if (!stopIteration && isAsync()) {
+        ProfileContext ctx = QueryProfiling.current();
         nextBatchFuture =
-            CompletableFuture.supplyAsync(() -> client.search(request), backgroundExecutor);
+            CompletableFuture.supplyAsync(
+                () -> QueryProfiling.withCurrentContext(ctx, () -> client.search(request)),
+                backgroundExecutor);
       }
     } else {
       iterator = Collections.emptyIterator();