[fix](move-memtable) close stream when cancel load stream stub (#38912)

Fix load stream leak when sink v2 meet error. Tested by `check_before_quit.groovy`.
apache · Aug 6, 2024 · e8d342d · e8d342d
1 parent 1f2f74d
commit e8d342d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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;