From e88bd3ca4ce2048056440f3d78da84cb52329b89 Mon Sep 17 00:00:00 2001
From: yiguolei <676222867@qq.com>
Date: Sat, 13 Apr 2024 17:50:22 +0800
Subject: [PATCH] [bugfix](feoom) add timeout to send be request to avoid too
 long to occupy memory (#33614)

Co-authored-by: yiguolei <yiguolei@gmail.com>
---
 .../main/java/org/apache/doris/rpc/BackendServiceClient.java   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/fe/fe-core/src/main/java/org/apache/doris/rpc/BackendServiceClient.java b/fe/fe-core/src/main/java/org/apache/doris/rpc/BackendServiceClient.java
index 3f4bb84676..d7fb04c526 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/rpc/BackendServiceClient.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/rpc/BackendServiceClient.java
@@ -85,7 +85,8 @@ public class BackendServiceClient {
 
     public ListenableFuture<InternalService.PCancelPlanFragmentResult> cancelPlanFragmentAsync(
             InternalService.PCancelPlanFragmentRequest request) {
-        return stub.cancelPlanFragment(request);
+        return stub.withDeadlineAfter(execPlanTimeout, TimeUnit.MILLISECONDS)
+                .cancelPlanFragment(request);
     }
 
     public Future<InternalService.PFetchDataResult> fetchDataAsync(InternalService.PFetchDataRequest request) {