From 175442477126a9dc75eead3becff7a92493e76c1 Mon Sep 17 00:00:00 2001
From: Kaijie Chen <ckj@apache.org>
Date: Mon, 5 Aug 2024 16:58:19 +0800
Subject: [PATCH] [fix](move-memtable) close stream when cancel load stream
 stub

---
 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 f322d67ceaf9c3..1e7eda71f1ee2e 100644
--- a/be/src/vec/sink/load_stream_stub.cpp
+++ b/be/src/vec/sink/load_stream_stub.cpp
@@ -339,6 +339,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;