patch 4.0
This commit is contained in:
93
src/sql/engine/expr/ob_expr_dml_event.cpp
Normal file
93
src/sql/engine/expr/ob_expr_dml_event.cpp
Normal file
@ -0,0 +1,93 @@
|
||||
/**
|
||||
* 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_EXE
|
||||
#include "sql/engine/expr/ob_expr_dml_event.h"
|
||||
#include "sql/session/ob_sql_session_info.h"
|
||||
#include "sql/engine/ob_exec_context.h"
|
||||
#include "sql/engine/expr/ob_datum_cast.h"
|
||||
#include "sql/engine/dml/ob_table_modify_op.h"
|
||||
|
||||
namespace oceanbase
|
||||
{
|
||||
namespace sql
|
||||
{
|
||||
ObExprDmlEvent::ObExprDmlEvent(ObIAllocator& alloc)
|
||||
: ObFuncExprOperator(alloc,
|
||||
T_FUN_SYS_DML_EVENT,
|
||||
N_DML_EVENT,
|
||||
ONE_OR_TWO,
|
||||
NOT_ROW_DIMENSION,
|
||||
false,
|
||||
INTERNAL_IN_ORACLE_MODE)
|
||||
{
|
||||
}
|
||||
ObExprDmlEvent::~ObExprDmlEvent(){}
|
||||
|
||||
int ObExprDmlEvent::calc_result_typeN(ObExprResType &type,
|
||||
ObExprResType *dml,
|
||||
int64_t param_num,
|
||||
common::ObExprTypeCtx &type_ctx) const
|
||||
{
|
||||
UNUSEDx(dml, type_ctx, param_num);
|
||||
int ret = OB_SUCCESS;
|
||||
type.set_tinyint();
|
||||
type.set_precision(DEFAULT_PRECISION_FOR_BOOL);
|
||||
type.set_scale(DEFAULT_SCALE_FOR_INTEGER);
|
||||
type.set_result_flag(NOT_NULL_FLAG);
|
||||
return ret;
|
||||
}
|
||||
|
||||
int ObExprDmlEvent::cg_expr(ObExprCGCtx &op_cg_ctx,
|
||||
const ObRawExpr &raw_expr,
|
||||
ObExpr &rt_expr) const
|
||||
{
|
||||
int ret = OB_SUCCESS;
|
||||
UNUSEDx(op_cg_ctx, raw_expr);
|
||||
rt_expr.eval_func_ = &calc_dml_event;
|
||||
return ret;
|
||||
}
|
||||
|
||||
int ObExprDmlEvent::calc_dml_event(const ObExpr &expr,
|
||||
ObEvalCtx &ctx,
|
||||
ObDatum &expr_datum)
|
||||
{
|
||||
int ret = OB_SUCCESS;
|
||||
ObDatum *dml = NULL;
|
||||
ObDatum *update_col = NULL;
|
||||
OZ (expr.args_[0]->eval(ctx, dml));
|
||||
OV (OB_NOT_NULL(dml));
|
||||
if (2 == expr.arg_cnt_) {
|
||||
OZ (expr.args_[1]->eval(ctx, update_col));
|
||||
OV (OB_NOT_NULL(update_col) && !update_col->is_null(), OB_INVALID_ARGUMENT);
|
||||
}
|
||||
OX (expr_datum.set_bool(false));
|
||||
if (OB_SUCC(ret)) {
|
||||
if (dml->get_int() == static_cast<int64_t>(ctx.exec_ctx_.get_dml_event())) {
|
||||
if (static_cast<int64_t>(DE_UPDATING) == dml->get_int() && NULL != update_col) {
|
||||
bool find = false;
|
||||
const ObString &col = update_col->get_string();
|
||||
const ColContentFixedArray *update_cols = ctx.exec_ctx_.get_update_columns();
|
||||
for (int64_t i = 0; OB_SUCC(ret) && !find && i < update_cols->count(); i++) {
|
||||
find = (!update_cols->at(i).is_implicit_ && 0 == col.case_compare(update_cols->at(i).column_name_));
|
||||
}
|
||||
expr_datum.set_bool(find);
|
||||
} else {
|
||||
expr_datum.set_bool(true); // success
|
||||
}
|
||||
}
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
} // end namespace sql
|
||||
} // end namespace oceanbase
|
||||
Reference in New Issue
Block a user