hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
c5a94ea0e2718348fd0e7f3d9f18ed5fce5b69ee
oceanbase/src/sql/optimizer/ob_log_insert_all.cpp
yinyj17 703ed5c14e fix some replace expr bugs in optimizer
2023-06-14 07:12:27 +00:00

223 lines
9.0 KiB
C++
Raw Blame History

/**
* Copyright (c) 2021 OceanBase
* OceanBase CE is licensed under Mulan PubL v2.
* You can use this software according to the terms and conditions of the Mulan PubL v2.
* You may obtain a copy of Mulan PubL v2 at:
* http://license.coscl.org.cn/MulanPubL-2.0
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PubL v2 for more details.
*/
#define USING_LOG_PREFIX SQL_OPT
#include "sql/optimizer/ob_del_upd_log_plan.h"
#include "sql/resolver/dml/ob_insert_stmt.h"
#include "ob_log_insert_all.h"
#include "sql/ob_phy_table_location.h"
#include "sql/code_generator/ob_expr_generator_impl.h"
#include "sql/ob_sql_utils.h"
#include "sql/optimizer/ob_log_plan.h"
#include "sql/optimizer/ob_select_log_plan.h"
#include "sql/optimizer/ob_log_table_scan.h"
#include "sql/optimizer/ob_log_exchange.h"
#include "sql/engine/expr/ob_expr_column_conv.h"
using namespace oceanbase;
using namespace sql;
using namespace oceanbase::common;
using namespace share;
using namespace oceanbase::share::schema;
const char* ObLogInsertAll::get_name() const
{
const char *ret = "MULTI TABLE INSERT";
return ret;
}
int ObLogInsertAll::get_plan_item_info(PlanText &plan_text,
ObSqlPlanItem &plan_item)
{
int ret = OB_SUCCESS;
if (OB_FAIL(ObLogDelUpd::get_plan_item_info(plan_text, plan_item))) {
LOG_WARN("failed to get plan item info", K(ret));
} else {
BEGIN_BUF_PRINT;
if (OB_FAIL(print_table_infos(ObString::make_string("columns"),
buf,
buf_len,
pos,
type))) {
LOG_WARN("failed to print table infos", K(ret));
} else if (is_multi_conditions_insert()) {
if (is_multi_insert_first() &&
OB_FAIL(BUF_PRINTF(", multi_table_first(%s),\n ", "true"))) {
LOG_WARN("BUG_PRINTF fails", K(ret));
} else if (!is_multi_insert_first() && OB_FAIL(BUF_PRINTF(",\n "))) {
LOG_WARN("BUG_PRINTF fails", K(ret));
} else if (get_insert_all_table_info() == NULL) {
//do nothing
} else if (OB_ISNULL(get_stmt()) ||
OB_UNLIKELY(get_table_list().count() != get_insert_all_table_info()->count())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("get unexpected null", K(get_insert_all_table_info()->count()), K(get_stmt()),
K(ret));
} else {
int64_t pre_idx = -1;
for (int64_t i = 0; OB_SUCC(ret) && i < get_insert_all_table_info()->count(); ++i) {
ObInsertAllTableInfo* table_info = get_insert_all_table_info()->at(i);
const TableItem *table_item = nullptr;
if (OB_ISNULL(table_info) ||
OB_ISNULL(table_item = get_stmt()->get_table_item_by_id(table_info->table_id_))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("get unexpected null", K(table_item), K(table_info), K(ret));
} else if (table_info->when_cond_idx_ == pre_idx) {
BUF_PRINTF(",\"%.*s\"", table_item->table_name_.length(),table_item->table_name_.ptr());
} else if (table_info->when_cond_idx_ != -1) {//when
if (pre_idx != -1) {
BUF_PRINTF("}; ");
}
const ObIArray<ObRawExpr *> &when_exprs = table_info->when_cond_exprs_;
EXPLAIN_PRINT_EXPRS(when_exprs, type);
BUF_PRINTF(" then: table_list{");
BUF_PRINTF("\"%.*s\"", table_item->table_name_.length(), table_item->table_name_.ptr());
pre_idx = table_info->when_cond_idx_;
} else {//else
BUF_PRINTF("}; ELSE: table_list{");
BUF_PRINTF("\"%.*s\"", table_item->table_name_.length(), table_item->table_name_.ptr());
pre_idx = table_info->when_cond_idx_;
}
}
if (OB_SUCC(ret)) {
BUF_PRINTF("},\n ");
int64_t cnt = 0;
for (int64_t i = 0; OB_SUCC(ret) && i < get_index_dml_infos().count(); ++i) {
const IndexDMLInfo *dml_info = get_index_dml_infos().at(i);
if (OB_ISNULL(dml_info)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("dml info is null", K(ret), K(dml_info));
} else if (!dml_info->is_primary_index_) {
continue;
} else {
const ObIArray<ObRawExpr *> &conv_exprs = dml_info->column_convert_exprs_;
EXPLAIN_PRINT_EXPRS(conv_exprs, type);
if (cnt < get_table_list().count() - 1) {
BUF_PRINTF(",\n ");
}
++ cnt;
}
}
}
}
} else if (OB_FAIL(BUF_PRINTF(",\n "))) {
LOG_WARN("BUG_PRINTF fails", K(ret));
} else {
int64_t cnt = 0;
for (int64_t i = 0; OB_SUCC(ret) && i < get_index_dml_infos().count(); ++i) {
const IndexDMLInfo *dml_info = get_index_dml_infos().at(i);
if (OB_ISNULL(dml_info)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("dml info is null", K(ret), K(dml_info));
} else if (!dml_info->is_primary_index_) {
continue;
} else {
const ObIArray<ObRawExpr *> &conv_exprs = dml_info->column_convert_exprs_;
EXPLAIN_PRINT_EXPRS(conv_exprs, type);
if (cnt < get_table_list().count() - 1) {
BUF_PRINTF(",\n ");
}
++ cnt;
}
}
}
END_BUF_PRINT(plan_item.special_predicates_,
plan_item.special_predicates_len_);
}
return ret;
}
int ObLogInsertAll::get_op_exprs(ObIArray<ObRawExpr*> &all_exprs)
{
int ret = OB_SUCCESS;
if (OB_ISNULL(insert_all_table_info_)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("get unexpected null", K(insert_all_table_info_), K(ret));
} else if (OB_FAIL(ObLogInsert::get_op_exprs(all_exprs))) {
LOG_WARN("failed to allocate expr pre", K(ret));
} else {
for (int64_t i = 0; OB_SUCC(ret) && i < insert_all_table_info_->count(); ++i) {
if (OB_ISNULL(insert_all_table_info_->at(i))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("get null table info", K(ret), K(i));
} else if (OB_FAIL(append(all_exprs, insert_all_table_info_->at(i)->when_cond_exprs_))) {
LOG_WARN("failed to append exprs", K(ret));
} else { /*do nothing*/ }
}
}
return ret;
}
// int ObLogInsertAll::generate_rowid_expr_for_trigger()
// {
// int ret = OB_SUCCESS;
// // oracle mode does not have replace or insert on duplicate
// if (lib::is_oracle_mode() && !has_instead_of_trigger()) {
// for (int64_t i = 0; OB_SUCC(ret) && i < get_index_dml_infos().count(); ++i) {
// bool has_trg = false;
// IndexDMLInfo *dml_info = get_index_dml_infos().at(i);
// if (OB_ISNULL(dml_info)) {
// ret = OB_ERR_UNEXPECTED;
// LOG_WARN("dml info is null", K(ret), K(dml_info));
// } else if (!dml_info->is_primary_index_) {
// // do nothing
// } else if (OB_FAIL(check_has_trigger(dml_info->ref_table_id_, has_trg))) {
// LOG_WARN("failed to check has trigger", K(ret));
// } else if (!has_trg) {
// // do nothing
// } else if (OB_FAIL(generate_old_rowid_expr(*dml_info))) {
// LOG_WARN("failed to generate rowid expr", K(ret));
// } else if (OB_FAIL(generate_insert_new_rowid_expr(*dml_info))) {
// LOG_WARN("failed to generate new rowid expr", K(ret));
// } else { /*do nothing*/ }
// }
// }
// return ret;
// }
// int ObLogInsertAll::generate_multi_part_partition_id_expr()
// {
// int ret = OB_SUCCESS;
// for (int64_t i = 0; OB_SUCC(ret) && i < get_index_dml_infos().count(); ++i) {
// if (OB_ISNULL(get_index_dml_infos().at(i))) {
// ret = OB_ERR_UNEXPECTED;
// LOG_WARN("index dml info is null", K(ret));
// } else if (OB_FAIL(generate_old_calc_partid_expr(*get_index_dml_infos().at(i)))) {
// LOG_WARN("failed to generate calc partid expr", K(ret));
// } else if (OB_FAIL(generate_insert_new_calc_partid_expr(*get_index_dml_infos().at(i)))) {
// LOG_WARN("failed to generate new calc partid expr", K(ret));
// } else { /*do nothing*/ }
// }
// return ret;
// }
int ObLogInsertAll::inner_replace_op_exprs(ObRawExprReplacer &replacer)
{
int ret = OB_SUCCESS;
if (OB_ISNULL(insert_all_table_info_)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("unexpect null", K(ret));
} else if (OB_FAIL(ObLogInsert::inner_replace_op_exprs(replacer))) {
LOG_WARN("failed to replace op exprs", K(ret));
}
for (int64_t i = 0; OB_SUCC(ret) && i < insert_all_table_info_->count(); ++i) {
if (OB_ISNULL(insert_all_table_info_->at(i))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("get null table info", K(ret), K(i));
} else if (OB_FAIL(replace_exprs_action(replacer,
insert_all_table_info_->at(i)->when_cond_exprs_))) {
LOG_WARN("failed to replace expr", K(ret));
} else { /*do nothing*/ }
}
return ret;
}
Reference in New Issue View Git Blame Copy Permalink