/**
 * 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_ENG
#include "common/sql_mode/ob_sql_mode_utils.h"
#include "sql/engine/dml/ob_table_update.h"
#include "sql/engine/ob_physical_plan_ctx.h"
#include "sql/engine/ob_physical_plan.h"
#include "sql/session/ob_sql_session_info.h"
#include "sql/engine/ob_exec_context.h"
#include "lib/profile/ob_perf_event.h"
namespace oceanbase
{
using namespace common;
using namespace storage;
using namespace share;
using namespace share::schema;
namespace sql
{

ObTableUpdate::ObTableUpdate(ObIAllocator &alloc)
    : ObTableModify(alloc),
      updated_column_ids_(alloc),
      updated_column_infos_(alloc),
      is_global_index_(false),
      new_spk_exprs_()
{
}

ObTableUpdate::~ObTableUpdate()
{
}

void ObTableUpdate::reset()
{
  updated_column_infos_.reset();
  updated_column_ids_.reset();
  is_global_index_ = false;
  ObTableModify::reset();
}

void ObTableUpdate::reuse()
{
  updated_column_infos_.reuse();
  updated_column_ids_.reuse();
  is_global_index_ = false;
  ObTableModify::reuse();
}

int ObTableUpdate::set_updated_column_info(int64_t array_index,
                                           uint64_t column_id,
                                           uint64_t project_index,
                                           bool auto_filled_timestamp,
                                           bool is_implicit,
                                           const ObString *column_name)
{
  int ret = OB_SUCCESS;
  ColumnContent column;
  column.projector_index_ = project_index;
  column.auto_filled_timestamp_ = auto_filled_timestamp;
  column.is_implicit_ = is_implicit;
  if (OB_NOT_NULL(column_name) && column.column_name_.empty()) {
    column.column_name_ = *column_name;
  }
  CK(array_index >= 0 && array_index < updated_column_ids_.count());
  CK(array_index >= 0 && array_index < updated_column_infos_.count());
  if (OB_SUCC(ret)) {
    updated_column_ids_.at(array_index) = column_id;
    updated_column_infos_.at(array_index) = column;
  }
  return ret;
}

int ObTableUpdate::inner_open(ObExecContext &ctx) const
{
  int ret = OB_SUCCESS;
  ObTableUpdateCtx *update_ctx = NULL;
  ObPhysicalPlanCtx *plan_ctx = NULL;
  ObSQLSessionInfo *my_session = NULL;
  int64_t schema_version = 0;
  int64_t binlog_row_image = ObBinlogRowImage::FULL;
  NG_TRACE(update_open);
  if (OB_FAIL(ObTableModify::inner_open(ctx))) {
    LOG_WARN("open child operator failed", K(ret));
  } else if (OB_ISNULL(update_ctx = GET_PHY_OPERATOR_CTX(ObTableUpdateCtx, ctx, get_id()))) {
    ret = OB_ERR_NULL_VALUE;
    LOG_WARN("get physical operator context failed", K_(id));
  } else if (OB_ISNULL(plan_ctx = ctx.get_physical_plan_ctx())) {
    ret = OB_ERR_NULL_VALUE;
    LOG_WARN("get physical plan context failed");
  } else if (OB_ISNULL(my_session = GET_MY_SESSION(ctx))) {
    ret = OB_ERR_UNEXPECTED;
    LOG_WARN("fail to get my session", K(ret));
  } else if (OB_ISNULL(my_phy_plan_)) {
    ret = OB_ERR_UNEXPECTED;
    LOG_ERROR("invalid physical plan", K(ret), K(my_phy_plan_));
  } else if (OB_FAIL(my_phy_plan_->get_base_table_version(index_tid_, schema_version))) {
    LOG_WARN("failed to get base table version", K(ret));
  } else if (OB_FAIL(my_session->get_binlog_row_image(binlog_row_image))) {
    LOG_WARN("fail to get binlog row image", K(ret));
  } else {
    update_ctx->dml_param_.timeout_ =  plan_ctx->get_ps_timeout_timestamp();
    update_ctx->dml_param_.schema_version_ = schema_version;
    update_ctx->dml_param_.is_total_quantity_log_ = (ObBinlogRowImage::FULL == binlog_row_image);
    update_ctx->dml_param_.tz_info_ = TZ_INFO(my_session);
    update_ctx->dml_param_.sql_mode_ = my_session->get_sql_mode();
    update_ctx->dml_param_.prelock_ = my_session->get_prelock();
    update_ctx->dml_param_.table_param_ = &table_param_;
    update_ctx->dml_param_.tenant_schema_version_ = plan_ctx->get_tenant_schema_version();
    update_ctx->dml_param_.is_ignore_ = is_ignore_;
    OZ(my_phy_plan_->get_encrypt_meta(index_tid_, update_ctx->dml_param_.encrypt_meta_legacy_,
                                      update_ctx->dml_param_.encrypt_meta_));
    if (OB_FAIL(ret)){
    } else if (gi_above_) {
      if (OB_FAIL(get_gi_task(ctx))) {
        LOG_WARN("get granule iterator task failed", K(ret));
      }
    }
    if (OB_SUCC(ret) && !from_multi_table_dml()) {
      OZ (TriggerHandle::do_handle_before_stmt(*this, *update_ctx,
                                               ObTriggerEvents::get_update_event()));
    }
    if (!has_instead_of_trigger_) {
      OZ (do_table_update(ctx), ret);
    } else {
      OZ (do_instead_of_trigger_update(ctx));
    }
  }
  NG_TRACE(update_end);
  return ret;
}

int ObTableUpdate::do_table_update(ObExecContext &ctx) const
{
  int ret = OB_SUCCESS;
  int64_t affected_rows = 0;
  ObTableUpdateCtx *update_ctx = NULL;
  ObPhysicalPlanCtx *plan_ctx = NULL;
  ObSQLSessionInfo *my_session = NULL;
  if (OB_ISNULL(update_ctx = GET_PHY_OPERATOR_CTX(ObTableUpdateCtx, ctx, get_id()))) {
    ret = OB_ERR_NULL_VALUE;
    LOG_WARN("get physical operator context failed", K(ret), K_(id));
  } else if (update_ctx->iter_end_) {
    LOG_DEBUG("can't get gi task, iter end", K(update_ctx->iter_end_), K(get_id()));
  } else if (OB_ISNULL(my_session = GET_MY_SESSION(ctx))) {
    ret = OB_ERR_UNEXPECTED;
    LOG_WARN("fail to get my session", K(ret));
  } else if (OB_FAIL(get_part_location(ctx, update_ctx->part_infos_))) {
    LOG_WARN("get part location failed", K(ret));
  } else if (is_returning()) {
    // do nothing, handle in ObTableUpdateReturning
  } else if (OB_ISNULL(plan_ctx = ctx.get_physical_plan_ctx())) {
    ret = OB_ERR_NULL_VALUE;
    LOG_WARN("get physical plan context failed", K(ret));
  } else if (OB_FAIL(update_rows(ctx, affected_rows))) {
    if (OB_TRY_LOCK_ROW_CONFLICT != ret && OB_TRANSACTION_SET_VIOLATION != ret) {
      LOG_WARN("update rows to partition storage failed",
               K(ret), K(column_ids_), K(updated_column_ids_));
    }
  } else if (!from_multi_table_dml()) {
    //dml meta info will be counted in multiple table dml operator, not here
    plan_ctx->add_row_matched_count(update_ctx->get_found_rows());
    plan_ctx->add_row_duplicated_count(update_ctx->get_changed_rows());
    plan_ctx->add_affected_rows(my_session->get_capability().cap_flags_.OB_CLIENT_FOUND_ROWS ?
            update_ctx->get_found_rows() : update_ctx->get_affected_rows());
  }
  SQL_ENG_LOG(DEBUG, "update rows end",
              K(ret), K(affected_rows),
              K(column_ids_), K(updated_column_ids_));
  return ret;
}

int ObTableUpdate::do_instead_of_trigger_update(ObExecContext &ctx) const
{
  int ret = OB_SUCCESS;
  ObTableUpdateCtx *update_ctx = GET_PHY_OPERATOR_CTX(ObTableUpdateCtx, ctx, get_id());
  ObPhysicalPlanCtx *plan_ctx = ctx.get_physical_plan_ctx();
  OV (OB_NOT_NULL(update_ctx), OB_ERR_NULL_VALUE);
  OV (OB_NOT_NULL(plan_ctx), OB_ERR_NULL_VALUE);
  OX (update_ctx->expr_ctx_.calc_buf_->reuse());
  OX (update_ctx->expr_ctx_.row_ctx_.reset());
  if (OB_SUCC(ret)) {
    const ObNewRow *full_row = NULL;
    int64_t affected_rows = 0;
    while (OB_SUCC(ret) && OB_SUCC(inner_get_next_row(ctx, full_row))) {
      CK (OB_NOT_NULL(full_row));
      OX (affected_rows += 1);
      if (OB_SUCC(ret)) {
        project_old_and_new_row(*update_ctx, *full_row);
        ObNewRow &old_row = update_ctx->old_row_;
        ObNewRow &new_row = update_ctx->new_row_;
        OZ (TriggerHandle::init_param_rows(*this, *update_ctx, old_row, new_row), old_row, new_row);
        OZ (TriggerHandle::do_handle_before_row(*this, *update_ctx, &new_row,
                                                ObTriggerEvents::get_update_event()),
            old_row, new_row);
      }
    }
    if (OB_ITER_END == ret) {
      plan_ctx->set_affected_rows(affected_rows);
      LOG_TRACE("update for instead of trigger success", K(plan_ctx->get_affected_rows()), K(ret));
      ret = OB_SUCCESS;
    } else {
      LOG_WARN("update for instead of trigger failed", K(ret));
    }
  }
  return ret;
}

int ObTableUpdate::inner_close(ObExecContext &ctx) const
{
  int ret = OB_SUCCESS;
  ObTableUpdateCtx *update_ctx = GET_PHY_OPERATOR_CTX(ObTableUpdateCtx, ctx, get_id());
  OV (OB_NOT_NULL(update_ctx), OB_ERR_NULL_VALUE);
  if (!from_multi_table_dml()) {
    OZ (TriggerHandle::do_handle_after_stmt(*this, *update_ctx,
                                            ObTriggerEvents::get_update_event()));
  }
  int close_ret = ObTableModify::inner_close(ctx);
  return (OB_SUCCESS == ret) ? close_ret : ret;
}

int ObTableUpdate::rescan(ObExecContext &ctx) const
{
  int ret = OB_SUCCESS;
  ObTableUpdateCtx *update_ctx = NULL;
  if (!gi_above_ || from_multi_table_dml()) {
    ret = OB_NOT_SUPPORTED;
    LOG_WARN("table update rescan not supported", K(ret));
    LOG_USER_ERROR(OB_NOT_SUPPORTED, "table update rescan");
  } else if (OB_ISNULL(update_ctx = GET_PHY_OPERATOR_CTX(ObTableUpdateCtx, ctx, get_id()))) {
    ret = OB_ERR_UNEXPECTED;
    LOG_WARN("table update context is null", K(ret), K(get_id()));
  } else if (OB_FAIL(ObTableModify::rescan(ctx))) {
    LOG_WARN("rescan child operator failed", K(ret));
  } else {
    update_ctx->found_rows_ = 0;
    update_ctx->changed_rows_ = 0;
    update_ctx->affected_rows_ = 0;
    update_ctx->has_got_old_row_ = false;
    update_ctx->part_infos_.reset();
    update_ctx->part_key_.reset();
    if (nullptr != update_ctx->rowkey_dist_ctx_) {
      update_ctx->rowkey_dist_ctx_->clear();
    }
  }
  if (OB_SUCC(ret)) {
    if (OB_FAIL(get_gi_task(ctx))) {
      LOG_WARN("get granule task failed", K(ret));
    } else if (OB_FAIL(do_table_update(ctx))) {
      LOG_WARN("do table update failed", K(ret));
    }
  }
  return ret;
}

// TRICK:
// ObTableUpdate按照设计是不对外吐数据的，如果ObResultSet调用get_next_row，岂不是会得到数据？
// FACT：
// ObTableUpdate::open()中会把自己封装成一个Iterator传递给存储层，存储层会将这里的数据消耗
// 干净，直到返回ITER_END。那么，当轮到ObResultSet调用get_next_row的时候，它只能得到ITER_END了。
// 一切，只是巧合而已。
int ObTableUpdate::get_next_row(ObExecContext &ctx, const ObNewRow *&row) const
{
  int ret = OB_SUCCESS;
  ObTableUpdateCtx *update_ctx = NULL;
  //update operator must has project operation
  if (OB_FAIL(try_check_status(ctx))) {
    LOG_WARN("check status failed", K(ret));
  } else if (OB_ISNULL(update_ctx = GET_PHY_OPERATOR_CTX(ObTableUpdateCtx, ctx, get_id()))) {
    ret = OB_ERR_NULL_VALUE;
    LOG_WARN("get physical operator context failed", K_(id));
  } else {
    update_ctx->expr_ctx_.calc_buf_->reuse();
    update_ctx->expr_ctx_.row_ctx_.reset();
  }
  if (OB_FAIL(ret)) {
    //do nothing
  } else if (!update_ctx->has_got_old_row_) {
    NG_TRACE_TIMES(2, update_start_next_row);
    const ObNewRow *full_row = NULL;
    bool is_updated = false;
    // 过滤未变化的行
    while(OB_SUCC(ret) && !is_updated && OB_SUCC(inner_get_next_row(ctx, full_row))) {
      CK(OB_NOT_NULL(full_row));
      if (OB_SUCC(ret)) {
        project_old_and_new_row(*update_ctx, *full_row);

        // check_rowkey_is_null and check_rowkey_whether_distinct only works for
        // mysql mode, mainly for the following cases
        // update t2 left outer join t1 on t1.c1 = t2.c1 set t1.c2 = t2.c2;
        if (OB_SUCC(ret) && !from_multi_table_dml()) {
          bool is_null = false;
          if (need_filter_null_row_) {
            if (OB_FAIL(check_rowkey_is_null(update_ctx->old_row_, primary_key_ids_.count(), is_null))) {
              LOG_WARN("check rowkey is null failed", K(ret), K(update_ctx->get_cur_row()), K(primary_key_ids_));
            } else if (is_null) {
              continue;
            }
          } else {
#if !defined(NDEBUG)
            if (need_check_pk_is_null()) {
              if (OB_FAIL(check_rowkey_is_null(update_ctx->old_row_,
                                               primary_key_ids_.count(), is_null))) {
                LOG_WARN("failed to check rowkey is null", K(ret));
              } else if (OB_UNLIKELY(is_null)) {
                ret = OB_ERR_UNEXPECTED;
                LOG_WARN("update row failed validity check", K(ret));
              }
            }
#endif
          }
        }
        if (OB_SUCC(ret) && !from_multi_table_dml()) {
          bool is_distinct = false;
          if (OB_FAIL(check_rowkey_whether_distinct(ctx,
                                                    update_ctx->old_row_,
                                                    primary_key_ids_.count(),
                                                    distinct_algo_,
                                                    update_ctx->rowkey_dist_ctx_,
                                                    is_distinct))) {
            LOG_WARN("check rowkey whether distinct failed", K(ret));
          } else if (!is_distinct) {
            continue;
          }
        }
        if (OB_SUCC(ret) && !from_multi_table_dml()) {
          ObNewRow &old_row = update_ctx->old_row_;
          ObNewRow &new_row = update_ctx->new_row_;
          OZ (TriggerHandle::init_param_rows(*this, *update_ctx, old_row, new_row), old_row, new_row);
          OZ (TriggerHandle::do_handle_before_row(*this, *update_ctx, &new_row,
                                                  ObTriggerEvents::get_update_event()),
              old_row, new_row);
          OZ (check_row_null(ctx, new_row, column_infos_, updated_column_infos_), new_row);
        }
      }
      if (OB_SUCC(ret)) {
        //check update row whether changed
        if (OB_LIKELY(!from_multi_table_dml() && !is_returning())) {
          //if update operator from multi table dml,
          //the row value will be check in multiple table dml operator
          //dml meta info will also be counted in multiple table dml operator
          OZ(check_updated_value(*update_ctx, *this, update_ctx->old_row_, update_ctx->new_row_, is_updated));
        } else if (OB_LIKELY(check_row_whether_changed(update_ctx->new_row_))) {
          is_updated = true;
        }
      }
      if (OB_SUCC(ret)) {
        ObNewRow &old_row = update_ctx->old_row_;
        ObNewRow &new_row = update_ctx->new_row_;
        if (is_updated) {
          if (!from_multi_table_dml()) {
            bool is_filtered = false;
            OZ (ForeignKeyHandle::do_handle(*update_ctx, fk_args_, old_row, new_row),
                old_row, new_row);
            OZ(ObPhyOperator::filter_row_for_view_check(update_ctx->expr_ctx_, new_row,
                                                        view_check_exprs_, is_filtered));
            OV(!is_filtered, OB_ERR_CHECK_OPTION_VIOLATED);
            OZ (ObPhyOperator::filter_row_for_check_cst(update_ctx->expr_ctx_, new_row,
                                                        check_constraint_exprs_, is_filtered));
            if (is_filtered && OB_SUCC(ret)) {
              ret = OB_ERR_CHECK_CONSTRAINT_VIOLATED;
              LOG_WARN("row is filtered by check filters, running is stopped", K(ret));
            }
          }
        } else {
          if (OB_FAIL(build_lock_row(*update_ctx, update_ctx->old_row_))) {
            LOG_WARN("build lock row failed", K(ret), K(update_ctx->old_row_));
          } else if (OB_FAIL(lock_row(ctx,
                                      update_ctx->lock_row_,
                                      update_ctx->dml_param_,
                                      update_ctx->part_key_))) {
            //没有发生更新，对当前行进行加锁
            if (OB_TRY_LOCK_ROW_CONFLICT != ret) {
              LOG_WARN("fail to lock row", K(ret), K(update_ctx->lock_row_), K(update_ctx->part_key_));
            }
          } else {
            LOG_DEBUG("lock row", K(ret), K(update_ctx->lock_row_), K(update_ctx->part_key_));
          }
        }
        if (!from_multi_table_dml()) {
          OZ (TriggerHandle::do_handle_after_row(*this, *update_ctx,
                                                 ObTriggerEvents::get_update_event()),
              old_row, new_row);
        }
      }
    } // while
    if (OB_SUCCESS != ret && OB_ITER_END != ret) {
      LOG_WARN("get next row from child operator failed", K(ret));
    }
    if (OB_SUCC(ret)) {
      // old row, keep old projector
      row = &(update_ctx->old_row_);
      LOG_DEBUG("get old row", K(update_ctx->old_row_));
      update_ctx->has_got_old_row_ = true;
    }
  } else {
    // new row
    const ObNewRow &new_row = update_ctx->new_row_;
    DLIST_FOREACH(cur_expr, new_spk_exprs_) {
      const ObColumnExpression *new_spk_expr = static_cast<const ObColumnExpression*>(cur_expr);
      int64_t result_idx = new_spk_expr->get_result_index();
      if (OB_UNLIKELY(result_idx < 0)
          || OB_UNLIKELY(result_idx >= new_row.count_)) {
        LOG_WARN("result index is invalid", K(ret), K(result_idx),
                 K(new_row.count_));
      } else if (OB_FAIL(new_spk_expr->calc(update_ctx->expr_ctx_,
                                            new_row,
                                            new_row.cells_[result_idx]))) {
        LOG_WARN("calc new spk expr failed", K(ret),
                 KPC(new_spk_expr), K(result_idx), K(new_row));
      }
    }
    if (OB_SUCC(ret)) {
      LOG_DEBUG("get new row", K(update_ctx->new_row_));
      row = &(update_ctx->new_row_);
      update_ctx->has_got_old_row_ = false;
    }
  }
  NG_TRACE_TIMES(2, update_end_next_row);
  return ret;
}

int ObTableUpdate::switch_iterator(ObExecContext &ctx) const
{
  int ret = OB_SUCCESS;
  ObTableUpdateCtx *update_ctx = NULL;
  if (OB_FAIL(ObTableModify::switch_iterator(ctx))) {
    if (OB_ITER_END != ret) {
      LOG_WARN("switch single child operator iterator failed", K(ret));
    }
  } else if (OB_ISNULL(update_ctx = GET_PHY_OPERATOR_CTX(ObTableUpdateCtx, ctx, get_id()))) {
    ret = OB_ERR_UNEXPECTED;
    LOG_WARN("update ctx is null", K(ret));
  } else {
    update_ctx->has_got_old_row_ = false;
  }
  return ret;
}

bool ObTableUpdate::check_row_whether_changed(const ObNewRow &new_row) const
{
  bool bret = false;
  if (updated_column_infos_.count() > 0 && new_row.is_valid()) {
    int64_t projector_index = updated_column_infos_.at(0).projector_index_;
    if (projector_index >= 0 && projector_index < new_row.get_count()) {
      const ObObj &updated_value = new_row.get_cell(projector_index);
      bret = !(updated_value.is_ext() && ObActionFlag::OP_LOCK_ROW == updated_value.get_ext());
    }
  }
  return bret;
}

int ObTableUpdate::inner_get_next_row(ObExecContext &ctx, const ObNewRow *&row) const
{
  int ret = OB_SUCCESS;
  ObTableUpdateCtx *update_ctx = NULL;
  if (OB_ISNULL(child_op_)) {
    ret = OB_NOT_INIT;
    LOG_WARN("child_op_ is null");
  } else if (OB_ISNULL(update_ctx = GET_PHY_OPERATOR_CTX(ObTableUpdateCtx, ctx, get_id()))) {
    ret = OB_ERR_NULL_VALUE;
    LOG_WARN("get physical operator context failed", K_(id));
  } else if (update_ctx->iter_end_) {
    LOG_DEBUG("can't get gi task, iter end", K(get_id()), K(update_ctx->iter_end_));
    ret = OB_ITER_END;
  } else {
    if (from_multi_table_dml()) {
      if (update_ctx->part_row_cnt_ <= 0) {
        if (update_ctx->cur_part_idx_ < update_ctx->part_infos_.count()) {
          update_ctx->part_row_cnt_ = update_ctx->part_infos_.at(update_ctx->cur_part_idx_).part_row_cnt_;
          update_ctx->part_key_ = update_ctx->part_infos_.at(update_ctx->cur_part_idx_).partition_key_;
          ++update_ctx->cur_part_idx_;
        }
      }
      if (OB_SUCC(ret)) {
        --update_ctx->part_row_cnt_;
      }
    }
    if (OB_SUCC(ret)) {
      ret = child_op_->get_next_row(ctx, row);
    }
  }

  if (OB_ITER_END == ret) {
    NG_TRACE(update_iter_end);
  }
  return ret;
}

void ObTableUpdate::project_old_and_new_row(ObTableUpdateCtx &ctx,
                                            const ObNewRow &full_row) const
{
  ctx.full_row_ = full_row;
  ObNewRow &new_row = ctx.new_row_;
  ObNewRow &old_row = ctx.old_row_;
  new_row.cells_ = full_row.cells_;
  new_row.count_ = full_row.count_;
  new_row.projector_ = ctx.new_row_projector_;
  new_row.projector_size_ = ctx.new_row_projector_size_;
  old_row = full_row;
  //old row and new row have the same projector size
  //but full row projector size is old row projector size + updated column count
  //so old row copy from full row and must reset projector size
  old_row.projector_size_ = ctx.new_row_projector_size_;
}

int ObTableUpdate::build_lock_row(ObTableUpdateCtx &update_ctx, const ObNewRow &old_row) const
{
  int ret = OB_SUCCESS;
  CK(OB_NOT_NULL(update_ctx.lock_row_.cells_));
  CK(update_ctx.lock_row_.count_ <= old_row.get_count());
  for (int64_t i = 0; OB_SUCC(ret) && i < update_ctx.lock_row_.count_; ++i) {
    update_ctx.lock_row_.cells_[i] = old_row.get_cell(i);
  }
  return ret;
}

int64_t ObTableUpdate::to_string_kv(char *buf, const int64_t buf_len) const
{
  int64_t pos = 0;
  J_KV(N_TID, table_id_,
       K_(index_tid),
       N_CID, column_ids_,
       N_UPDATED_CID, updated_column_ids_,
       K_(updated_column_infos),
       K_(is_global_index));
  return pos;
}

int ObTableUpdate::init_op_ctx(ObExecContext &ctx) const
{
  int ret = OB_SUCCESS;
  ObTableUpdateCtx *op_ctx = NULL;
  OZ(CREATE_PHY_OPERATOR_CTX(ObTableUpdateCtx, ctx, get_id(), get_type(), op_ctx));
  CK(OB_NOT_NULL(op_ctx));
  OZ(op_ctx->alloc_row_cells(projector_size_, op_ctx->lock_row_));
  if (OB_SUCC(ret)) {
    op_ctx->new_row_projector_ = projector_;
    op_ctx->new_row_projector_size_ = projector_size_;
  }
  OX (ctx.set_dml_event(ObDmlEventType::DE_UPDATING));
  OZ (op_ctx->init_update_columns(updated_column_infos_));
  if (tg_args_.count() > 0) {
    OZ(op_ctx->init_trigger_params(tg_event_, all_tm_points_,
                                   tg_columns_.get_projector(),
                                   tg_columns_.get_count(),
                                   tg_columns_.get_rowtype_count()));
  }
  return ret;
}

inline int ObTableUpdate::update_rows(ObExecContext &ctx, int64_t &affected_rows) const
{
  UNUSEDx(ctx, affected_rows);
  int ret = OB_NOT_SUPPORTED;
  LOG_USER_ERROR(OB_NOT_SUPPORTED, "table update rows");
  // ObTableUpdateCtx *update_ctx = NULL;
  // ObTaskExecutorCtx *executor_ctx = NULL;
  // ObSQLSessionInfo *my_session = ctx.get_my_session();
  // ObPartitionService *partition_service = NULL;
  // if (OB_ISNULL(my_session)) {
  //   ret = OB_ERR_UNEXPECTED;
  //   LOG_WARN("my_session is null");
  // } else if (OB_ISNULL(executor_ctx = GET_TASK_EXECUTOR_CTX(ctx))) {
  //   ret = OB_ERR_UNEXPECTED;
  //   LOG_WARN("fail to get task executor ctx", K(ret));
  // } else if (OB_ISNULL(partition_service = executor_ctx->get_partition_service())) {
  //   ret = OB_ERR_UNEXPECTED;
  //   LOG_WARN("fail to get partition service", K(ret));
  // } else if (OB_ISNULL(update_ctx = GET_PHY_OPERATOR_CTX(ObTableUpdateCtx, ctx, get_id()))) {
  //   ret = OB_ERR_NULL_VALUE;
  //   LOG_WARN("get physical operator context failed", K_(id));
  // } else if (OB_UNLIKELY(update_ctx->part_infos_.empty())) {
  //   ret = OB_ERR_UNEXPECTED;
  //   LOG_WARN("part infos is empty", K(ret));
  // } else if (OB_LIKELY(update_ctx->part_infos_.count() == 1)) {
  //   update_ctx->part_key_ = update_ctx->part_infos_.at(0).partition_key_;
  //   ObDMLRowIterator dml_row_iter(ctx, *this);
  //   if (OB_FAIL(dml_row_iter.init())) {
  //     LOG_WARN("init dml row iterator", K(ret));
  //   } else if (OB_FAIL(partition_service->update_rows(*my_session->get_tx_desc(),
  //                                                     update_ctx->dml_param_,
  //                                                     update_ctx->part_key_,
  //                                                     column_ids_,
  //                                                     updated_column_ids_,
  //                                                     &dml_row_iter,
  //                                                     affected_rows))) {
  //     if (OB_TRY_LOCK_ROW_CONFLICT != ret) {
  //       LOG_WARN("insert row to partition storage failed", K(ret));
  //     }
  //   }
  // } else {
  //   //多分区插入
  //   //给UPDATE语句建两个row的buffer用来去除row上的projector
  //   const ObNewRow *old_row = NULL;
  //   const ObNewRow *new_row = NULL;
  //   if (OB_FAIL(update_ctx->create_cur_rows(2, update_ctx->new_row_projector_size_, NULL, 0))) {
  //     ret = OB_ERR_UNEXPECTED;
  //     LOG_WARN("create current rows failed", K(ret), K_(projector_size));
  //   }
  //   while (OB_SUCC(ret) && OB_SUCC(get_next_row(ctx, old_row)) && OB_SUCC(get_next_row(ctx, new_row))) {
  //     if (OB_FAIL(copy_cur_row_by_projector(update_ctx->cur_rows_[0], old_row))) {
  //       LOG_WARN("copy old row failed", K(ret));
  //     } else if (OB_FAIL(copy_cur_row_by_projector(update_ctx->cur_rows_[1], new_row))) {
  //       LOG_WARN("copy new row failed", K(ret));
  //     } else if (OB_FAIL(partition_service->update_row(*my_session->get_tx_desc(),
  //                                                      update_ctx->dml_param_,
  //                                                      update_ctx->part_key_,
  //                                                      column_ids_,
  //                                                      updated_column_ids_,
  //                                                      *old_row,
  //                                                      *new_row))) {
  //       if (OB_TRY_LOCK_ROW_CONFLICT != ret) {
  //         LOG_WARN("update row to partition storage failed", K(ret));
  //       }
  //     } else {
  //       LOG_DEBUG("update multi part", K_(update_ctx->part_key), K(*old_row), K(*new_row));
  //     }
  //   }
  //   if (OB_ITER_END == ret) {
  //     ret = OB_SUCCESS;
  //   } else if (OB_FAIL(ret)) {
  //     LOG_WARN("process update row failed", K(ret));
  //   }
  // }
  return ret;
}

OB_DEF_SERIALIZE(ObTableUpdate)
{
  int ret = OB_SUCCESS;
  BASE_SER((ObTableUpdate, ObTableModify));
  LST_DO_CODE(OB_UNIS_ENCODE, updated_column_ids_, updated_column_infos_, is_global_index_);
  if (OB_SUCC(ret)) {
    if (OB_FAIL(serialize_dlist(new_spk_exprs_, buf, buf_len, pos))) {
      LOG_WARN("failed to serialize calc_exprs_", K(ret));
    }
  }
  return ret;
}

OB_DEF_DESERIALIZE(ObTableUpdate)
{
  int ret = OB_SUCCESS;
  BASE_DESER((ObTableUpdate, ObTableModify));
  LST_DO_CODE(OB_UNIS_DECODE, updated_column_ids_, updated_column_infos_, is_global_index_);
  OB_UNIS_DECODE_EXPR_DLIST(ObColumnExpression, new_spk_exprs_, my_phy_plan_);
  return ret;
}

OB_DEF_SERIALIZE_SIZE(ObTableUpdate)
{
  int64_t len = 0;
  BASE_ADD_LEN((ObTableUpdate, ObTableModify));
  LST_DO_CODE(OB_UNIS_ADD_LEN, updated_column_ids_, updated_column_infos_, is_global_index_);
  len += get_dlist_serialize_size(new_spk_exprs_);
  return len;
}
}  // namespace sql
}  // namespace oceanbase