diff --git a/be/src/olap/delta_writer.cpp b/be/src/olap/delta_writer.cpp index efd9d06212a60e..5a05e9ade3a57c 100644 --- a/be/src/olap/delta_writer.cpp +++ b/be/src/olap/delta_writer.cpp @@ -311,6 +311,7 @@ Status DeltaWriter::close() { } Status DeltaWriter::close_wait(google::protobuf::RepeatedPtrField* tablet_vec, + google::protobuf::RepeatedPtrField* tablet_errors, bool is_broken) { std::lock_guard l(_lock); DCHECK(_is_init) @@ -321,7 +322,13 @@ Status DeltaWriter::close_wait(google::protobuf::RepeatedPtrField* } // return error if previous flush failed - RETURN_NOT_OK(_flush_token->wait()); + Status s = _flush_token->wait(); + if (!s.ok()) { + PTabletError* tablet_error = tablet_errors->Add(); + tablet_error->set_tablet_id(_tablet->tablet_id()); + tablet_error->set_msg(s.get_error_msg()); + return s; + } // use rowset meta manager to save meta _cur_rowset = _rowset_writer->build(); diff --git a/be/src/olap/delta_writer.h b/be/src/olap/delta_writer.h index e095fff13216c8..091d8ef213adf2 100644 --- a/be/src/olap/delta_writer.h +++ b/be/src/olap/delta_writer.h @@ -67,7 +67,9 @@ class DeltaWriter { Status close(); // wait for all memtables to be flushed. // mem_consumption() should be 0 after this function returns. - Status close_wait(google::protobuf::RepeatedPtrField* tablet_vec, bool is_broken); + Status close_wait(google::protobuf::RepeatedPtrField* tablet_vec, + google::protobuf::RepeatedPtrField* tablet_errors, + bool is_broken); // abandon current memtable and wait for all pending-flushing memtables to be destructed. // mem_consumption() should be 0 after this function returns. diff --git a/be/src/runtime/load_channel.h b/be/src/runtime/load_channel.h index fa88ec20e214df..1d0c3f04e1becc 100644 --- a/be/src/runtime/load_channel.h +++ b/be/src/runtime/load_channel.h @@ -81,7 +81,8 @@ class LoadChannel { bool finished = false; auto index_id = request.index_id(); RETURN_IF_ERROR(channel->close(request.sender_id(), request.backend_id(), &finished, - request.partition_ids(), response->mutable_tablet_vec())); + request.partition_ids(), response->mutable_tablet_vec(), + response->mutable_tablet_errors())); if (finished) { std::lock_guard l(_lock); _tablets_channels.erase(index_id); diff --git a/be/src/runtime/tablets_channel.cpp b/be/src/runtime/tablets_channel.cpp index d74b03342af8cd..3f807f32433fe2 100644 --- a/be/src/runtime/tablets_channel.cpp +++ b/be/src/runtime/tablets_channel.cpp @@ -80,7 +80,8 @@ Status TabletsChannel::open(const PTabletWriterOpenRequest& request) { Status TabletsChannel::close(int sender_id, int64_t backend_id, bool* finished, const google::protobuf::RepeatedField& partition_ids, - google::protobuf::RepeatedPtrField* tablet_vec) { + google::protobuf::RepeatedPtrField* tablet_vec, + google::protobuf::RepeatedPtrField* tablet_errors) { std::lock_guard l(_lock); if (_state == kFinished) { return _close_status; @@ -128,8 +129,9 @@ Status TabletsChannel::close(int sender_id, int64_t backend_id, bool* finished, for (auto writer : need_wait_writers) { // close may return failed, but no need to handle it here. // tablet_vec will only contains success tablet, and then let FE judge it. - writer->close_wait(tablet_vec, (_broken_tablets.find(writer->tablet_id()) != - _broken_tablets.end())); + writer->close_wait( + tablet_vec, tablet_errors, + (_broken_tablets.find(writer->tablet_id()) != _broken_tablets.end())); } } return Status::OK(); @@ -259,4 +261,4 @@ std::ostream& operator<<(std::ostream& os, const TabletsChannelKey& key) { return os; } -} // namespace doris \ No newline at end of file +} // namespace doris diff --git a/be/src/runtime/tablets_channel.h b/be/src/runtime/tablets_channel.h index 725fe440832b57..eb39956cba68a1 100644 --- a/be/src/runtime/tablets_channel.h +++ b/be/src/runtime/tablets_channel.h @@ -77,7 +77,8 @@ class TabletsChannel { // no-op when this channel has been closed or cancelled Status close(int sender_id, int64_t backend_id, bool* finished, const google::protobuf::RepeatedField& partition_ids, - google::protobuf::RepeatedPtrField* tablet_vec); + google::protobuf::RepeatedPtrField* tablet_vec, + google::protobuf::RepeatedPtrField* tablet_error); // no-op when this channel has been closed or cancelled Status cancel(); @@ -235,4 +236,4 @@ Status TabletsChannel::add_batch(const TabletWriterAddRequest& request, } return Status::OK(); } -} // namespace doris \ No newline at end of file +} // namespace doris