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

[CP] adjust merge into partition prune constraint

Browse Source
This commit is contained in:
chimyue
2023-07-13 14:42:17 +00:00
committed by ob-robot
parent 59a7f93a3f
commit a61043f93b
9 changed files with 150 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/sql/resolver/dml/ob_stmt_expr_visitor.cpp
View File

@ -115,8 +115,7 @@ int ObStmtExprCopier::do_visit(ObRawExpr *&expr)
return copier_.copy_on_replace(expr, expr);
}
int ObSharedExprChecker::init(ObDMLStmt &stmt,
bool ignore_column /* default false */ )
int ObSharedExprChecker::init(ObDMLStmt &stmt)
{
int ret = OB_SUCCESS;
hash::ObHashSet<uint64_t> stmt_expr_set;
@ -126,7 +125,6 @@ int ObSharedExprChecker::init(ObDMLStmt &stmt,
LOG_WARN("failed to create expr set", K(ret));
} else {
stmt_expr_set_ = &stmt_expr_set;
ignore_column_ = ignore_column;
set_relation_scope();
if (OB_FAIL(stmt.iterate_stmt_expr(*this))) {
LOG_WARN("failed to iterate stmt expr", K(ret));
@ -154,7 +152,12 @@ int ObSharedExprChecker::is_shared_expr(const ObRawExpr *expr, bool &is_shared)
{
int ret = OB_SUCCESS;
is_shared = false;
if (shared_expr_set_.created()) {
if (OB_ISNULL(expr)) {
/* do nothing */
} else if (T_FUN_COLUMN_CONV == expr->get_expr_type()
|| expr->is_column_ref_expr()) {
/* do nothing */
} else if (shared_expr_set_.created()) {
uint64_t key = reinterpret_cast<uint64_t>(expr);
int tmp_ret = shared_expr_set_.exist_refactored(key);
if (OB_HASH_EXIST == tmp_ret) {
@ -175,14 +178,13 @@ int ObSharedExprChecker::do_visit(ObRawExpr *&expr)
if (OB_ISNULL(expr) || OB_ISNULL(stmt_expr_set_)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("unexpected null", K(ret), K(expr), K(stmt_expr_set_));
} else if (expr->is_column_ref_expr() && ignore_column_) {
/* do nothing */
} else if (expr->has_flag(CNT_COLUMN) ||
expr->has_flag(CNT_AGG) ||
expr->has_flag(CNT_SET_OP) ||
expr->has_flag(CNT_WINDOW_FUNC) ||
expr->has_flag(CNT_DYNAMIC_PARAM) ||
expr->has_flag(CNT_SUB_QUERY)) {
bool need_check_in_share = T_FUN_COLUMN_CONV == expr->get_expr_type();
if (OB_FAIL(stmt_expr_set_->set_refactored(key, 0))) {
if (OB_HASH_EXIST == ret) {
is_shared = true;
@ -193,7 +195,7 @@ int ObSharedExprChecker::do_visit(ObRawExpr *&expr)
LOG_WARN("failed to add expr into set", K(ret));
}
}
if (OB_SUCC(ret) && !is_shared) {
if (OB_SUCC(ret) && (!is_shared || need_check_in_share)) {
for (int64_t i = 0; OB_SUCC(ret) && i < expr->get_param_count(); ++i) {
if (OB_FAIL(SMART_CALL(do_visit(expr->get_param_expr(i))))) {
LOG_WARN("failed to visit first", K(ret));

6
src/sql/resolver/dml/ob_stmt_expr_visitor.h
View File

@ -207,18 +207,16 @@ class ObSharedExprChecker : public ObStmtExprVisitor
{
public:
ObSharedExprChecker() : stmt_expr_set_(NULL),
shared_expr_set_(),
ignore_column_(false) {}
shared_expr_set_() {}
virtual int do_visit(ObRawExpr *&expr) override;
int init(ObDMLStmt &stmt, bool ignore_column = false);
int init(ObDMLStmt &stmt);
int destroy();
int is_shared_expr(const ObRawExpr *expr, bool &is_shared) const;
private:
hash::ObHashSet<uint64_t> *stmt_expr_set_;
hash::ObHashSet<uint64_t> shared_expr_set_;
bool ignore_column_;
};
class ObStmtExecParamFormatter : public ObStmtExprVisitor