From 5c797c3d2fe4f0d5b16fc4fbacf01a2b41c01bbd Mon Sep 17 00:00:00 2001
From: Xin Zhang <desertsxin@gmail.com>
Date: Thu, 6 Nov 2025 11:55:24 +0000
Subject: [PATCH] fix(native): Fix NPE in HttpNativeExecutionTaskResultFetcher

---
 .../nativeprocess/HttpNativeExecutionTaskResultFetcher.java   | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/presto-spark-base/src/main/java/com/facebook/presto/spark/execution/nativeprocess/HttpNativeExecutionTaskResultFetcher.java b/presto-spark-base/src/main/java/com/facebook/presto/spark/execution/nativeprocess/HttpNativeExecutionTaskResultFetcher.java
index f97e7c9c3eae2..8b916dbe3edbb 100644
--- a/presto-spark-base/src/main/java/com/facebook/presto/spark/execution/nativeprocess/HttpNativeExecutionTaskResultFetcher.java
+++ b/presto-spark-base/src/main/java/com/facebook/presto/spark/execution/nativeprocess/HttpNativeExecutionTaskResultFetcher.java
@@ -173,7 +173,9 @@ private void onSuccess(PageBufferClient.PagesResponse pagesResponse)
         token = nextToken;
         if (pagesResponse.isClientComplete()) {
             workerClient.abortResultsAsync();
-            scheduledFuture.cancel(false);
+            if (scheduledFuture != null) {
+                scheduledFuture.cancel(false);
+            }
         }
         if (!pages.isEmpty()) {
             synchronized (taskHasResult) {