patch 4.0

src/sql/optimizer/ob_log_delete.cpp

@@ -14,37 +14,26 @@
 #include "ob_log_delete.h"
 #include "ob_log_plan.h"
 #include "sql/optimizer/ob_optimizer_context.h"
+#include "sql/optimizer/ob_opt_est_cost.h"
+#include "sql/optimizer/ob_del_upd_log_plan.h"
+#include "sql/optimizer/ob_join_order.h"
+#include "common/ob_smart_call.h"
 
 using namespace oceanbase;
 using namespace sql;
 using namespace oceanbase::common;
 
-int ObLogDelete::allocate_expr_pre(ObAllocExprContext& ctx)
+const char *ObLogDelete::get_name() const
 {
-  int ret = OB_SUCCESS;
-  LOG_TRACE("allocate expr pre for delete", K(is_pdml()));
-  if (OB_FAIL(add_table_columns_to_ctx(ctx))) {
-    LOG_WARN("failed to add table columns to ctx", K(ret));
-  } else if (OB_FAIL(ObLogicalOperator::allocate_expr_pre(ctx))) {
-    LOG_WARN("failed to allocate expr pre", K(ret));
-  } else if (is_pdml()) {
-    if (OB_FAIL(alloc_partition_id_expr(ctx))) {
-      LOG_WARN("failed alloc pseudo partition_id column for delete", K(ret));
-    }
-  } else { /*do nothing*/
-  }
-  return ret;
-}
-
-const char* ObLogDelete::get_name() const
-{
-  const char* name = NULL;
-  int ret = OB_SUCCESS;
+  const char *name = NULL;
   if (is_multi_part_dml()) {
-    name = "MULTI PARTITION DELETE";
+    name = "DISTRIBUTED DELETE";
   } else if (is_pdml() && is_index_maintenance()) {
+    // PDML 索引表的delete，使用 INDEX DELETE
     name = "INDEX DELETE";
   } else {
+    // 默认Delete的name：DELETE
+    // PDML 主表的delete，也默认使用name：DELETE
     name = ObLogDelUpd::get_name();
   }
   return name;
@@ -53,14 +42,115 @@ const char* ObLogDelete::get_name() const
 int ObLogDelete::est_cost()
 {
   int ret = OB_SUCCESS;
-  ObLogicalOperator* child = NULL;
+  ObLogicalOperator *child = NULL;
   if (OB_ISNULL(child = get_child(ObLogicalOperator::first_child))) {
     ret = OB_ERR_UNEXPECTED;
     LOG_WARN("child is null", K(ret), K(child));
   } else {
-    set_op_cost(child->get_card() * static_cast<double>(DELETE_ONE_ROW_COST));
-    set_cost(child->get_cost() + get_op_cost());
-    set_card(child->get_card());
+    double op_cost = 0.0;
+    if (OB_FAIL(inner_est_cost(child->get_card(), op_cost))) {
+      LOG_WARN("failed to get delete cost", K(ret));
+    } else {
+      set_op_cost(op_cost);
+      set_cost(child->get_cost() + get_op_cost());
+      set_card(child->get_card());
+    }
+  }
+  return ret;
+}
+
+int ObLogDelete::re_est_cost(EstimateCostInfo &param, double &card, double &cost)
+{
+  int ret = OB_SUCCESS;
+  ObLogicalOperator *child = NULL;
+  if (OB_ISNULL(child = get_child(ObLogicalOperator::first_child))) {
+    ret = OB_ERR_UNEXPECTED;
+    LOG_WARN("get unexpected null", K(child), K(ret));
+  } else {
+    double child_card = child->get_card();
+    double child_cost = child->get_cost();
+    double op_cost = 0.0;
+    if (OB_FAIL(SMART_CALL(child->re_est_cost(param, child_card, child_cost)))) {
+      LOG_WARN("failed to re est exchange cost", K(ret));
+    } else if (OB_FAIL(inner_est_cost(child_card, op_cost))) {
+      LOG_WARN("failed to get delete cost", K(ret));
+    } else {
+      cost = child_cost + op_cost;
+      card = child_card;
+      if (param.override_) {
+        set_op_cost(op_cost);
+        set_cost(cost);
+        set_card(card);
+      }
+    }
+  }
+  return ret;
+}
+
+int ObLogDelete::inner_est_cost(double child_card, double &op_cost)
+{
+  int ret = OB_SUCCESS;
+  ObDelUpCostInfo cost_info(0,0,0);
+  cost_info.affect_rows_ = child_card;
+  cost_info.index_count_ = get_index_dml_infos().count();
+  IndexDMLInfo* delete_dml_info = nullptr;
+  if (OB_ISNULL(get_plan()) || OB_UNLIKELY(cost_info.index_count_ <= 0) ||
+      OB_ISNULL(delete_dml_info = get_index_dml_infos().at(0))) {
+    ret = OB_ERR_UNEXPECTED;
+    LOG_WARN("get unexpected null", K(ret), K(get_plan()), K(delete_dml_info));
+  } else {
+    ObOptimizerContext &opt_ctx = get_plan()->get_optimizer_context();
+    cost_info.constraint_count_ = delete_dml_info->ck_cst_exprs_.count();
+    if (OB_FAIL(ObOptEstCost::cost_delete(cost_info, op_cost, opt_ctx.get_cost_model_type()))) {
+      LOG_WARN("failed to get delete cost", K(ret));
+    }
+  }
+  return ret;
+}
+
+int ObLogDelete::get_op_exprs(ObIArray<ObRawExpr*> &all_exprs)
+{
+  int ret = OB_SUCCESS;
+  if (OB_FAIL(ObLogDelUpd::inner_get_op_exprs(all_exprs, true))) {
+    LOG_WARN("failed to get op exprs", K(ret));
+  } else { /*do nothing*/ }
+  return ret;
+}
+
+int ObLogDelete::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 { /*do nothing*/ }
+  }
+  return ret;
+}
+
+int ObLogDelete::generate_rowid_expr_for_trigger()
+{
+  int ret = OB_SUCCESS;
+  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 { /*do nothing*/ }
+    }
   }
   return ret;
 }