From 0a3874f203f2f279dc1eda0348c2d9d8074b5a5b Mon Sep 17 00:00:00 2001
From: Kaijie Chen <ckj@apache.org>
Date: Wed, 7 Aug 2024 23:24:00 +0800
Subject: [PATCH] [fix](move-memtable) close stream when cancel load stream
 stub (#38912) (#39039)

backport #38912
---
 be/src/vec/sink/load_stream_stub.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/be/src/vec/sink/load_stream_stub.cpp b/be/src/vec/sink/load_stream_stub.cpp
index 93f3fd87a8571d..dc34b13e0ac155 100644
--- a/be/src/vec/sink/load_stream_stub.cpp
+++ b/be/src/vec/sink/load_stream_stub.cpp
@@ -338,6 +338,9 @@ Status LoadStreamStub::close_wait(RuntimeState* state, int64_t timeout_ms) {
 
 void LoadStreamStub::cancel(Status reason) {
     LOG(WARNING) << *this << " is cancelled because of " << reason;
+    if (_is_init.load()) {
+        brpc::StreamClose(_stream_id);
+    }
     {
         std::lock_guard<bthread::Mutex> lock(_cancel_mutex);
         _cancel_reason = reason;