fix

dataroaring · Nov 4, 2023 · 8bfe833 · 8bfe833
1 parent ee18ee8
commit 8bfe833
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/be/src/vec/sink/load_stream_stub.cpp b/be/src/vec/sink/load_stream_stub.cpp
@@ -109,6 +109,7 @@ void LoadStreamStub::prepare() {
  return Status::InternalError("stream {} is already inited by {}", _stream_id, _load_id);
  }
  ++_use_cnt;
+ LOG(WARNING) << "prepare stream " << _stream_id << " load_id " << _load_id << " use_cnt " << _use_cnt;
 }
 
 // open_load_stream
@@ -210,11 +211,16 @@ Status LoadStreamStub::close_load(const std::vector<PTabletID>& tablets_to_commi
  if (--_use_cnt > 0) {
  return Status::OK();
  }
+ if (_use_cnt <= 0) {
+ return Status::InternalError("stream {} already closed", _stream_id);
+ }
+
+ DCHECK(_use_cnt == 0);
  PStreamHeader header;
  *header.mutable_load_id() = _load_id;
  header.set_src_id(_src_id);
  header.set_opcode(doris::PStreamHeader::CLOSE_LOAD);
- LOG(INFO) << "issue close load stream " << _stream_id;
+ LOG(INFO) << "issue close load stream " << _stream_id << " load_id " << _load_id;
  {
  std::lock_guard<std::mutex> lock(_tablets_to_commit_mutex);
  for (const auto& tablet : _tablets_to_commit) {