hcq/oceanbase
hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[CP] fix temp table access rescan bug

Browse Source
This commit is contained in:
zzg19950727 2023-11-14 04:15:00 +00:00 committed by ob-robot
parent d04bc629df
commit c4557981e4
1 changed files with 32 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
src/sql/engine/basic/ob_temp_table_access_op.cpp
View File

@ -170,9 +170,39 @@ OB_SERIALIZE_MEMBER_INHERIT(ObTempTableAccessOpSpec, ObOpSpec,
int ObTempTableAccessOp::inner_rescan()
{
int ret = OB_SUCCESS;
is_started_ = false;
int64_t result_id = 0;
cur_idx_ = 0;
return ObOperator::inner_rescan();
is_started_ = false;
can_rescan_ = true;
if (!MY_SPEC.is_distributed_) {
interm_result_ids_.reuse();
if (OB_FAIL(get_local_interm_result_id(result_id))) {
LOG_WARN("failed to get local result id", K(ret));
} else if (OB_FAIL(interm_result_ids_.push_back(result_id))) {
LOG_WARN("failed to push back result id", K(ret));
}
} else {
int64_t index = 0;
bool is_end = false;
while (!is_end && OB_SUCC(ret)) {
if (OB_FAIL(check_status())) {
LOG_WARN("check status failed", K(ret));
} else
if (OB_FAIL(MY_INPUT.check_finish(is_end, index))) {
LOG_WARN("failed to check finish.", K(ret));
} else if (!is_end) {
result_id = MY_INPUT.interm_result_ids_.at(index);
if (OB_FAIL(interm_result_ids_.push_back(result_id))) {
LOG_WARN("failed to push back result id", K(ret));
}
}
}
}
if (OB_SUCC(ret) &&
OB_FAIL(ObOperator::inner_rescan())) {
LOG_WARN("failed to rescan", K(ret));
}
return ret;
}
int ObTempTableAccessOp::inner_open()
@ -230,7 +260,6 @@ int ObTempTableAccessOp::inner_get_next_row()
}
if (OB_SUCC(ret) && is_end) {
ret = OB_ITER_END;
can_rescan_ = true;
} else if (OB_FAIL(ret)) {
} else if (OB_ISNULL(tmp_sr)) {
ret = OB_ERR_UNEXPECTED;
@ -298,7 +327,6 @@ int ObTempTableAccessOp::inner_get_next_batch(const int64_t max_row_cnt)
if (is_end) {
brs_.size_ = 0;
brs_.end_ = true;
can_rescan_ = true;
} else {
brs_.size_ = read_rows;
}