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

fix core of query pushdown into set stmt

Browse Source
This commit is contained in:
yinyj17
2024-01-05 02:17:10 +00:00
committed by ob-robot
parent 0608c45482
commit b3343fdd74
1 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/sql/rewrite/ob_transform_query_push_down.cpp
View File

@ -462,6 +462,7 @@ int ObTransformQueryPushDown::check_set_op_expr_reference(ObSelectStmt *select_s
} else { } else {
ObSEArray<ObRawExpr*, 4> relation_exprs; ObSEArray<ObRawExpr*, 4> relation_exprs;
ObSEArray<ObRawExpr*, 4> set_op_exprs; ObSEArray<ObRawExpr*, 4> set_op_exprs;
ObSEArray<ObRawExpr*, 4> column_exprs;
ObBitSet<> selected_set_op_idx; ObBitSet<> selected_set_op_idx;
ObStmtExprGetter visitor; ObStmtExprGetter visitor;
visitor.set_relation_scope(); visitor.set_relation_scope();
@ -470,7 +471,13 @@ int ObTransformQueryPushDown::check_set_op_expr_reference(ObSelectStmt *select_s
LOG_WARN("failed to get relation exprs", K(ret)); LOG_WARN("failed to get relation exprs", K(ret));
} else if (OB_FAIL(ObRawExprUtils::extract_set_op_exprs(relation_exprs, } else if (OB_FAIL(ObRawExprUtils::extract_set_op_exprs(relation_exprs,
set_op_exprs))) { set_op_exprs))) {
LOG_WARN("failed to extract column ids", K(ret)); LOG_WARN("failed to extract set op exprs", K(ret));
} else if (FALSE_IT(relation_exprs.reuse())) {
} else if (OB_FAIL(select_stmt->get_relation_exprs(relation_exprs, visitor))) {
LOG_WARN("failed to get relation exprs", K(ret));
} else if (OB_FAIL(ObRawExprUtils::extract_column_exprs(relation_exprs,
column_exprs))) {
LOG_WARN("failed to extract column exprs", K(ret));
} }
for (int64_t i = 0; OB_SUCC(ret) && i < select_offset.count(); ++i) { for (int64_t i = 0; OB_SUCC(ret) && i < select_offset.count(); ++i) {
if (select_offset.at(i) != -1) { if (select_offset.at(i) != -1) {
@ -486,6 +493,15 @@ int ObTransformQueryPushDown::check_set_op_expr_reference(ObSelectStmt *select_s
is_valid = false; is_valid = false;
} }
} }
for (int64_t i = 0; OB_SUCC(ret) && is_valid && i < column_exprs.count(); ++i) {
ObColumnRefRawExpr *expr = static_cast<ObColumnRefRawExpr*>(column_exprs.at(i));
if (OB_ISNULL(expr)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("unexpected null expr", K(ret), KPC(column_exprs.at(i)));
} else if (!selected_set_op_idx.has_member(expr->get_column_id() - OB_APP_MIN_COLUMN_ID)) {
is_valid = false;
}
}
} }
return ret; return ret;
} }