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

patch 4.0

Browse Source
This commit is contained in:
wangzelin.wzl
2022-10-24 10:34:53 +08:00
parent 4ad6e00ec3
commit 93a1074b0c
10533 changed files with 2588271 additions and 2299373 deletions
Download Patch File Download Diff File Expand all files Collapse all files

114
src/sql/engine/expr/ob_expr_calc_urowid.cpp
View File

@ -13,74 +13,57 @@
#define USING_LOG_PREFIX SQL_ENG
#include "ob_expr_calc_urowid.h"
#include "sql/resolver/expr/ob_raw_expr.h"
#include "lib/container/ob_array_helper.h"
#include "common/data_buffer.h"
using namespace oceanbase::common;
using namespace oceanbase::sql;
namespace oceanbase {
namespace sql {
ObExprCalcURowID::ObExprCalcURowID(ObIAllocator& alloc)
: ObExprOperator(alloc, T_FUN_SYS_CALC_UROWID, N_CALC_UROWID, MORE_THAN_ZERO, NOT_ROW_DIMENSION)
{}
namespace oceanbase
{
namespace sql
{
ObExprCalcURowID::ObExprCalcURowID(ObIAllocator &alloc)
: ObExprOperator(alloc, T_FUN_SYS_CALC_UROWID, N_CALC_UROWID, MORE_THAN_ZERO, NOT_ROW_DIMENSION,
false, INTERNAL_IN_ORACLE_MODE)
{
}
ObExprCalcURowID::~ObExprCalcURowID()
{}
ObExprCalcURowID::~ObExprCalcURowID() {}
int ObExprCalcURowID::calc_result_typeN(
ObExprResType& type, ObExprResType* types, int64_t param_num, ObExprTypeCtx& type_ctx) const
int ObExprCalcURowID::calc_result_typeN(ObExprResType &type,
ObExprResType *types,
int64_t param_num,
ObExprTypeCtx &type_ctx) const
{
int ret = OB_SUCCESS;
UNUSED(type_ctx);
if (OB_UNLIKELY(param_num < 1) || OB_ISNULL(types)) {
if (OB_UNLIKELY(param_num < 1)
|| OB_ISNULL(types)) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid argument for calc_result_typeN", K(ret), K(param_num), K(types), K(share::is_mysql_mode()));
} else if (share::is_mysql_mode()) {
LOG_WARN("invalid argument for calc_result_typeN",
K(ret), K(param_num), K(types), K(lib::is_mysql_mode()));
} else if (lib::is_mysql_mode()) {
ret = OB_NOT_SUPPORTED;
LOG_WARN("calc_urowid is not supported in mysql mode", K(ret));
LOG_USER_ERROR(OB_NOT_SUPPORTED, "urowid in mysql mode");
} else {
type.set_type(ObURowIDType);
type.set_length(OB_MAX_USER_ROW_KEY_LENGTH);
}
return ret;
}
int ObExprCalcURowID::calc_resultN(ObObj& result, const ObObj* obj_stack, int64_t param_num, ObExprCtx& expr_ctx) const
{
int ret = OB_SUCCESS;
int64_t version = -1;
if (OB_ISNULL(expr_ctx.calc_buf_) || OB_ISNULL(obj_stack) || OB_UNLIKELY(param_num < 2)) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid argument", K(ret), K(expr_ctx.calc_buf_), K(obj_stack), K(param_num));
} else if (OB_UNLIKELY(share::is_mysql_mode())) {
ret = OB_NOT_SUPPORTED;
LOG_WARN("calc_urowid is not supported in mysql mode", K(ret));
} else if (OB_UNLIKELY(!obj_stack[0].is_number()) ||
OB_UNLIKELY(!obj_stack[0].get_number().is_valid_int64(version)) ||
OB_UNLIKELY(!ObURowIDData::is_valid_version(version))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("first arg must be version info(number type)", K(ret), K(obj_stack[0]));
} else {
ObArrayHelper<ObObj> pk_vals(param_num, const_cast<ObObj*>(obj_stack) + 1, param_num - 1);
LOG_TRACE("calc urowid", K(pk_vals), K(param_num));
ObURowIDData urowid_data;
if (OB_FAIL(urowid_data.set_rowid_content(pk_vals, version, *expr_ctx.calc_buf_))) {
LOG_WARN("failed to set rowid content", K(ret), K(pk_vals));
} else {
result.set_urowid(urowid_data);
}
}
return ret;
}
int ObExprCalcURowID::cg_expr(ObExprCGCtx& cg_ctx, const ObRawExpr& raw_expr, ObExpr& rt_expr) const
int ObExprCalcURowID::cg_expr(ObExprCGCtx &cg_ctx,
const ObRawExpr &raw_expr,
ObExpr &rt_expr) const
{
int ret = OB_SUCCESS;
UNUSED(raw_expr);
UNUSED(cg_ctx);
if (OB_UNLIKELY(rt_expr.arg_cnt_ < 1) || OB_ISNULL(rt_expr.args_) ||
OB_UNLIKELY(ObURowIDType != rt_expr.datum_meta_.type_)) {
if (OB_UNLIKELY(rt_expr.arg_cnt_ < 1) || OB_ISNULL(rt_expr.args_)
|| OB_UNLIKELY(ObURowIDType != rt_expr.datum_meta_.type_)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("unexpected params", K(rt_expr.arg_cnt_), K(rt_expr.args_), K(rt_expr.type_));
} else {
@ -89,23 +72,26 @@ int ObExprCalcURowID::cg_expr(ObExprCGCtx& cg_ctx, const ObRawExpr& raw_expr, Ob
return ret;
}
int ObExprCalcURowID::calc_urowid(const ObExpr& rt_expr, ObEvalCtx& eval_ctx, ObDatum& expr_datum)
int ObExprCalcURowID::calc_urowid(const ObExpr &rt_expr,
ObEvalCtx &eval_ctx,
ObDatum &expr_datum)
{
int ret = OB_SUCCESS;
if (OB_UNLIKELY(share::is_mysql_mode())) {
if (OB_UNLIKELY(lib::is_mysql_mode())) {
ret = OB_NOT_SUPPORTED;
LOG_WARN("calc_urowid is not supported in mysql mode", K(ret));
LOG_USER_ERROR(OB_NOT_SUPPORTED, "urowid in mysql mode");
} else if (OB_UNLIKELY(2 > rt_expr.arg_cnt_)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("unexpected arg cnt", K(ret), K(rt_expr.arg_cnt_));
} else {
int64_t version = -1;
ObDatum* param = NULL;
ObDatum *param = NULL;
if (OB_FAIL(rt_expr.args_[0]->eval(eval_ctx, param))) {
LOG_WARN("eval version arg failed", K(ret));
} else {
number::ObNumber ver_nmb(param->get_number());
if (OB_UNLIKELY(!ver_nmb.is_valid_int64(version)) || OB_UNLIKELY(!ObURowIDData::is_valid_version(version))) {
if (OB_UNLIKELY(!ver_nmb.is_valid_int64(version))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("first arg must be version info(number type)", K(ret), K(ver_nmb));
}
@ -125,17 +111,20 @@ int ObExprCalcURowID::calc_urowid(const ObExpr& rt_expr, ObEvalCtx& eval_ctx, Ob
} else {
// do nothing
}
} // end for
} // end for
ObDataBuffer data_alloc;
int64_t data_size = ObURowIDData::needed_content_buf_size(pk_vals);
char* data_buf = NULL;
int64_t data_size = ObURowIDData::needed_content_buf_size(pk_vals, version);
char *data_buf = NULL;
if (OB_SUCC(ret)) {
if (OB_ISNULL(data_buf = rt_expr.get_str_res_mem(eval_ctx, data_size))) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_WARN("failed to allocate memory", K(ret));
} else {
data_alloc.set_data(data_buf, data_size);
if (OB_FAIL(urowid_data.set_rowid_content(pk_vals, version, data_alloc))) {
if (all_pk_is_null(version, pk_vals)) {
// trigger before-row or outer join add extra null for rowid column will occur pks all are null.
expr_datum.set_null();
} else if (OB_FAIL(urowid_data.set_rowid_content(pk_vals, version, data_alloc))) {
LOG_WARN("failed to set rowid content", K(ret));
} else {
expr_datum.set_urowid(urowid_data);
@ -146,5 +135,22 @@ int ObExprCalcURowID::calc_urowid(const ObExpr& rt_expr, ObEvalCtx& eval_ctx, Ob
}
return ret;
}
} // end namespace sql
} // end namespace oceanbase
bool ObExprCalcURowID::all_pk_is_null(int64_t version, ObIArray<ObObj> &pk_vals)
{
bool is_all = !pk_vals.empty();
if (ObURowIDData::HEAP_TABLE_ROWID_VERSION == version ||
ObURowIDData::EXT_HEAP_TABLE_ROWID_VERSION == version) {
is_all = pk_vals.at(0).is_null();
} else {
for (int64_t i = 0; is_all && i < pk_vals.count(); ++i) {
is_all &= pk_vals.at(i).is_null();
}
}
return is_all;
}
} // end namespace sql
} // end namespace oceanbase