database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[branch-2.1](arrow-flight-sql) Open regression-test/pipeline/p0/arrow_flight_sql (#37727)

Browse Source 
pick #36854
This commit is contained in:
Xinyi Zou
2024-07-16 16:23:43 +08:00
committed by GitHub
parent 9fd4795080
commit 1f779ba9de
6 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
be/src/runtime/buffer_control_block.cpp
View File

@ -232,7 +232,7 @@ Status BufferControlBlock::get_arrow_batch(std::shared_ptr<arrow::RecordBatch>*
if (_is_close) {
return Status::OK();
}
return Status::InternalError("Abnormal Ending");
return Status::InternalError("Get Arrow Batch Abnormal Ending");
}
Status BufferControlBlock::close(Status exec_status) {

2
be/src/service/arrow_flight/arrow_flight_batch_reader.cpp
View File

@ -43,7 +43,7 @@ arrow::Result<std::shared_ptr<ArrowFlightBatchReader>> ArrowFlightBatchReader::C
auto schema = ExecEnv::GetInstance()->result_mgr()->find_arrow_schema(statement_->query_id);
if (schema == nullptr) {
ARROW_RETURN_NOT_OK(arrow::Status::Invalid(fmt::format(
"not found arrow flight schema, maybe query has been canceled, queryid: {}",
"Client not found arrow flight schema, maybe query has been canceled, queryid: {}",
print_id(statement_->query_id))));
}
std::shared_ptr<ArrowFlightBatchReader> result(new ArrowFlightBatchReader(statement_, schema));

4
be/src/service/internal_service.cpp
View File

@ -851,9 +851,9 @@ void PInternalServiceImpl::fetch_arrow_flight_schema(google::protobuf::RpcContro
ExecEnv::GetInstance()->result_mgr()->find_arrow_schema(
UniqueId(request->finst_id()).to_thrift());
if (schema == nullptr) {
LOG(INFO) << "not found arrow flight schema, maybe query has been canceled";
LOG(INFO) << "FE not found arrow flight schema, maybe query has been canceled";
auto st = Status::NotFound(
"not found arrow flight schema, maybe query has been canceled");
"FE not found arrow flight schema, maybe query has been canceled");
st.to_protobuf(result->mutable_status());
return;
}