621 lines
24 KiB
C++
621 lines
24 KiB
C++
/**
|
|
* 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 "sql/engine/expr/ob_expr_obj_access.h"
|
|
#include "sql/engine/ob_physical_plan_ctx.h"
|
|
#include "sql/engine/ob_exec_context.h"
|
|
#include "sql/ob_spi.h"
|
|
#include "pl/ob_pl.h"
|
|
#include "pl/ob_pl_resolver.h"
|
|
#include "sql/engine/expr/ob_expr_lob_utils.h"
|
|
|
|
namespace oceanbase
|
|
{
|
|
using namespace common;
|
|
namespace sql
|
|
{
|
|
|
|
ObExprObjAccess::ExtraInfo::ExtraInfo(common::ObIAllocator &alloc, ObExprOperatorType type)
|
|
: ObIExprExtraInfo(alloc, type),
|
|
get_attr_func_(0),
|
|
param_idxs_(alloc),
|
|
access_idx_cnt_(0),
|
|
for_write_(false),
|
|
property_type_(pl::ObCollectionType::INVALID_PROPERTY),
|
|
coll_idx_(OB_INVALID_INDEX),
|
|
extend_size_(0),
|
|
access_idxs_(alloc)
|
|
{
|
|
}
|
|
|
|
OB_SERIALIZE_MEMBER(ObExprObjAccess::ExtraInfo,
|
|
get_attr_func_,
|
|
param_idxs_,
|
|
access_idx_cnt_,
|
|
for_write_,
|
|
property_type_,
|
|
coll_idx_,
|
|
extend_size_);
|
|
|
|
OB_SERIALIZE_MEMBER((ObExprObjAccess, ObExprOperator),
|
|
info_.get_attr_func_,
|
|
info_.param_idxs_,
|
|
info_.access_idx_cnt_,
|
|
info_.for_write_,
|
|
info_.property_type_,
|
|
info_.coll_idx_);
|
|
// extend_size_ is not needed here, we got extend size from result_type_
|
|
|
|
ObExprObjAccess::ObExprObjAccess(ObIAllocator &alloc)
|
|
: ObExprOperator(alloc, T_OBJ_ACCESS_REF, N_OBJ_ACCESS, PARAM_NUM_UNKNOWN, VALID_FOR_GENERATED_COL, NOT_ROW_DIMENSION,
|
|
INTERNAL_IN_MYSQL_MODE, INTERNAL_IN_ORACLE_MODE),
|
|
info_(alloc, T_OBJ_ACCESS_REF)
|
|
{
|
|
}
|
|
|
|
ObExprObjAccess::~ObExprObjAccess()
|
|
{
|
|
}
|
|
|
|
void ObExprObjAccess::ExtraInfo::reset()
|
|
{
|
|
get_attr_func_ = 0;
|
|
param_idxs_.reset();
|
|
access_idx_cnt_ = 0;
|
|
for_write_ = false;
|
|
property_type_ = pl::ObCollectionType::INVALID_PROPERTY;
|
|
coll_idx_ = OB_INVALID_INDEX;
|
|
extend_size_ = 0;
|
|
access_idxs_.reset();
|
|
}
|
|
|
|
void ObExprObjAccess::reset()
|
|
{
|
|
info_.reset();
|
|
ObExprOperator::reset();
|
|
}
|
|
|
|
int ObExprObjAccess::ExtraInfo::deep_copy(common::ObIAllocator &allocator,
|
|
const ObExprOperatorType type,
|
|
ObIExprExtraInfo *&copied_info) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
OZ(ObExprExtraInfoFactory::alloc(allocator, type, copied_info));
|
|
ExtraInfo &other = *static_cast<ExtraInfo *>(copied_info);
|
|
if (OB_SUCC(ret)) {
|
|
OZ(other.assign(*this));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int ObExprObjAccess::ExtraInfo::assign(const ObExprObjAccess::ExtraInfo &other)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
get_attr_func_ = other.get_attr_func_;
|
|
access_idx_cnt_ = other.access_idx_cnt_;
|
|
for_write_ = other.for_write_;
|
|
property_type_ = other.property_type_;
|
|
coll_idx_ = other.coll_idx_;
|
|
extend_size_ = other.extend_size_;
|
|
OZ(param_idxs_.assign(other.param_idxs_));
|
|
OZ(access_idxs_.assign(other.access_idxs_));
|
|
return ret;
|
|
}
|
|
|
|
int ObExprObjAccess::assign(const ObExprOperator &other)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
if (other.get_type() != T_OBJ_ACCESS_REF) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
LOG_WARN("expr operator is mismatch", K(other.get_type()));
|
|
} else if (OB_FAIL(ObExprOperator::assign(other))) {
|
|
LOG_WARN("assign parent expr failed", K(ret));
|
|
} else {
|
|
const ObExprObjAccess &other_expr = static_cast<const ObExprObjAccess &>(other);
|
|
OZ(info_.assign(other_expr.info_));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
#define GET_VALID_INT64_PARAM_FROM_NUMBER(obj) \
|
|
if (!obj.get_number().is_valid_int64(param_value)) { \
|
|
number::ObNumber number = obj.get_number(); \
|
|
if (OB_FAIL(number.round(0))) { \
|
|
LOG_WARN("failed to round number", K(ret), K(number)); \
|
|
} else if (!number.is_valid_int64(param_value)) { \
|
|
ret = OB_ARRAY_OUT_OF_RANGE; \
|
|
LOG_WARN("array index is out of range", K(ret), K(number)); \
|
|
} \
|
|
}
|
|
|
|
#define GET_VALID_INT64_PARAM(obj, skip_check_error) \
|
|
do { \
|
|
if (OB_SUCC(ret)) { \
|
|
int64_t param_value = 0; \
|
|
if (obj.is_integer_type() || obj.is_ext()) { \
|
|
param_value = obj.get_int(); \
|
|
} else if (obj.is_number()) { \
|
|
GET_VALID_INT64_PARAM_FROM_NUMBER(obj) \
|
|
} else if (obj.is_decimal_int()) { \
|
|
number::ObNumber numb; \
|
|
ObNumStackOnceAlloc alloc; \
|
|
ObObj tmp_obj; \
|
|
if (OB_FAIL(wide::to_number(obj.get_decimal_int(), obj.get_int_bytes(), obj.get_scale(), alloc, numb))) { \
|
|
LOG_WARN("fail to cast decimal int to number", K(ret)); \
|
|
} else { \
|
|
tmp_obj.set_number(numb); \
|
|
GET_VALID_INT64_PARAM_FROM_NUMBER(tmp_obj); \
|
|
} \
|
|
} else if (obj.is_null()) { \
|
|
if (!skip_check_error) { \
|
|
ret = OB_ERR_NUMERIC_OR_VALUE_ERROR; \
|
|
LOG_WARN("ORA-06502: PL/SQL: numeric or value error: NULL index table key value",\
|
|
K(ret), K(obj), K(i)); \
|
|
} else { \
|
|
param_value = OB_INVALID_INDEX; \
|
|
} \
|
|
} else { \
|
|
ret = OB_ERR_UNEXPECTED; \
|
|
LOG_WARN("obj param is invalid type", K(obj), K(i)); \
|
|
} \
|
|
if (OB_SUCC(ret) && OB_FAIL(param_array.push_back(param_value))) { \
|
|
LOG_WARN("store param array failed", K(ret), K(i)); \
|
|
} \
|
|
} \
|
|
} while (0)
|
|
|
|
int ObExprObjAccess::ExtraInfo::init_param_array(const ParamStore ¶m_store,
|
|
const ObObj *objs_stack,
|
|
int64_t param_num,
|
|
ParamArray ¶m_array) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
for (int64_t i = 0; OB_SUCC(ret) && i < param_idxs_.count(); ++i) {
|
|
CK (param_idxs_.at(i) >= 0 && param_idxs_.at(i) < param_store.count());
|
|
if (OB_SUCC(ret)) {
|
|
const ObObjParam &obj_param = param_store.at(param_idxs_.at(i));
|
|
GET_VALID_INT64_PARAM(obj_param, false);
|
|
}
|
|
}
|
|
for (int64_t i = 0; OB_SUCC(ret) && i < param_num; ++i) {
|
|
const ObObj &obj = objs_stack[i];
|
|
GET_VALID_INT64_PARAM(obj, (i == param_num - 1 && pl::ObCollectionType::NEXT_PROPERTY == property_type_));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int ObExprObjAccess::ExtraInfo::update_coll_first_last(
|
|
const ParamStore ¶m_store, const ObObj *objs_stack, int64_t param_num) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
bool found = false;
|
|
|
|
#define SEARCH_AND_UPDATE_COLL(get_elem, count) \
|
|
for (int64_t i = 0; OB_SUCC(ret) && !found && i < count; ++i) { \
|
|
const ObObj &obj = get_elem; \
|
|
if (obj.is_pl_extend() \
|
|
&& (pl::PL_NESTED_TABLE_TYPE == obj.get_meta().get_extend_type() \
|
|
|| pl::PL_VARRAY_TYPE == obj.get_meta().get_extend_type())) { \
|
|
found = true; \
|
|
pl::ObPLCollection *coll = reinterpret_cast<pl::ObPLCollection *>(obj.get_ext()); \
|
|
CK (OB_NOT_NULL(coll)); \
|
|
OX (coll->set_first(OB_INVALID_INDEX)); \
|
|
OX (coll->set_last(OB_INVALID_INDEX)); \
|
|
OX (found = true); \
|
|
} \
|
|
}
|
|
|
|
SEARCH_AND_UPDATE_COLL(objs_stack[i], param_num);
|
|
SEARCH_AND_UPDATE_COLL(param_store.at(i), param_store.count());
|
|
|
|
#undef SEARCH_AND_UPDATE_COLL
|
|
return ret;
|
|
}
|
|
|
|
int ObExprObjAccess::calc_result(ObObj &result,
|
|
ObIAllocator &alloc,
|
|
const ObObj *objs_stack,
|
|
int64_t param_num,
|
|
const ParamStore ¶m_store,
|
|
ObEvalCtx *ctx) const
|
|
{
|
|
|
|
return info_.calc(result,
|
|
alloc,
|
|
get_result_type(),
|
|
get_result_type().get_extend_size(),
|
|
param_store,
|
|
objs_stack,
|
|
param_num,
|
|
ctx);
|
|
}
|
|
|
|
int ObExprObjAccess::ExtraInfo::get_collection_attr(int64_t* params,
|
|
const pl::ObObjAccessIdx ¤t_access,
|
|
bool for_write,
|
|
void *¤t_value) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
pl::ObPLCollection *current_coll = reinterpret_cast<pl::ObPLCollection*>(current_value);
|
|
int64_t element_idx;
|
|
CK (OB_NOT_NULL(current_coll));
|
|
if (OB_SUCC(ret) && !current_coll->is_inited()) {
|
|
ret = OB_ERR_COLLECION_NULL;
|
|
LOG_WARN("Reference to uninitialized collection", K(ret), KPC(current_coll));
|
|
}
|
|
if (OB_SUCC(ret) && !current_access.is_property()) {
|
|
if (current_access.is_const()) {
|
|
element_idx = current_access.var_index_ - 1;
|
|
} else {
|
|
element_idx = params[current_access.var_index_] - 1;
|
|
}
|
|
if (element_idx < 0 || element_idx >= current_coll->get_count()) {
|
|
ret = OB_READ_NOTHING;
|
|
LOG_WARN("", K(ret), K(element_idx));
|
|
}
|
|
}
|
|
if (OB_SUCC(ret) && !current_access.is_property()) {
|
|
ObObj &element_obj = current_coll->get_data()[element_idx];
|
|
if (ObMaxType == element_obj.get_type()) {
|
|
if (!for_write) {
|
|
ret = OB_READ_NOTHING;
|
|
LOG_WARN("", K(ret), KPC(current_coll));
|
|
} else {
|
|
if (current_access.var_type_.is_composite_type()) {
|
|
element_obj.set_type(ObExtendType);
|
|
}
|
|
}
|
|
}
|
|
OX (current_value = &element_obj);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int ObExprObjAccess::ExtraInfo::get_record_attr(const pl::ObObjAccessIdx ¤t_access,
|
|
uint64_t udt_id,
|
|
bool for_write,
|
|
void *¤t_value,
|
|
ObEvalCtx &ctx) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
|
|
ObArenaAllocator alloc;
|
|
const pl::ObUserDefinedType *user_type = NULL;
|
|
const pl::ObRecordType *record_type = NULL;
|
|
pl::ObPLComposite *current_composite = reinterpret_cast<pl::ObPLComposite*>(current_value);
|
|
pl::ObPLRecord* current_record = static_cast<pl::ObPLRecord*>(current_composite);
|
|
ObObj* element_obj = NULL;
|
|
CK (OB_NOT_NULL(current_composite));
|
|
CK (current_composite->is_record());
|
|
CK (OB_NOT_NULL(current_record));
|
|
CK (OB_NOT_NULL(ctx.exec_ctx_.get_my_session()));
|
|
CK (OB_NOT_NULL(ctx.exec_ctx_.get_sql_ctx()));
|
|
CK (OB_NOT_NULL(ctx.exec_ctx_.get_sql_ctx()->schema_guard_));
|
|
CK (OB_NOT_NULL(ctx.exec_ctx_.get_sql_proxy()));
|
|
if (OB_FAIL(ret)) {
|
|
} else if (ctx.exec_ctx_.get_my_session()->get_pl_context()) {
|
|
pl::ObPLINS *ns = ctx.exec_ctx_.get_my_session()->get_pl_context()->get_current_ctx();
|
|
CK (OB_NOT_NULL(ns));
|
|
OZ (ns->get_user_type(udt_id, user_type));
|
|
} else {
|
|
pl::ObPLPackageGuard *package_guard = NULL;
|
|
OZ (ctx.exec_ctx_.get_package_guard(package_guard));
|
|
CK (OB_NOT_NULL(package_guard));
|
|
if (OB_SUCC(ret)) {
|
|
pl::ObPLResolveCtx resolve_ctx(alloc,
|
|
*ctx.exec_ctx_.get_my_session(),
|
|
*ctx.exec_ctx_.get_sql_ctx()->schema_guard_,
|
|
*package_guard,
|
|
*ctx.exec_ctx_.get_sql_proxy(),
|
|
false);
|
|
OZ (resolve_ctx.get_user_type(udt_id, user_type));
|
|
}
|
|
}
|
|
CK (OB_NOT_NULL(user_type));
|
|
CK (user_type->is_record_type());
|
|
CK (OB_NOT_NULL(record_type = static_cast<const pl::ObRecordType*>(user_type)));
|
|
CK (current_access.is_const());
|
|
if (OB_SUCC(ret) && user_type->is_object_type() && for_write_ && current_composite->is_null()) {
|
|
ret = OB_ERR_ACCESS_INTO_NULL;
|
|
LOG_WARN("", K(ret), KPC(current_composite));
|
|
}
|
|
OZ (current_record->get_element(current_access.var_index_, element_obj));
|
|
CK (OB_NOT_NULL(current_value = element_obj));
|
|
|
|
return ret;
|
|
}
|
|
|
|
int ObExprObjAccess::ExtraInfo::get_attr_func(int64_t param_cnt,
|
|
int64_t *params,
|
|
int64_t *element_val,
|
|
ObEvalCtx &ctx) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
void *current_value = NULL;
|
|
CK (OB_NOT_NULL(element_val));
|
|
CK (access_idxs_.count() > 0);
|
|
if (OB_SUCC(ret)) {
|
|
pl::ObPLComposite *composite_addr
|
|
= reinterpret_cast<pl::ObPLComposite*>(params[access_idxs_.at(0).var_index_]);
|
|
for (int64_t i = 1; OB_SUCC(ret) && i < access_idxs_.count(); ++i) {
|
|
const pl::ObPLDataType &parent_type = access_idxs_.at(i - 1).var_type_;
|
|
const pl::ObObjAccessIdx &parent_access = access_idxs_.at(i - 1);
|
|
const pl::ObObjAccessIdx ¤t_access = access_idxs_.at(i);
|
|
current_value = composite_addr;
|
|
if (parent_type.is_collection_type()) {
|
|
OZ (get_collection_attr(params,
|
|
current_access,
|
|
for_write_,
|
|
current_value));
|
|
} else {
|
|
OZ (get_record_attr(current_access,
|
|
parent_type.get_user_type_id(),
|
|
for_write_,
|
|
current_value,
|
|
ctx));;
|
|
}
|
|
if (OB_FAIL(ret)) {
|
|
} else if (current_access.var_type_.is_composite_type()) {
|
|
ObObj* value = reinterpret_cast<ObObj*>(current_value);
|
|
CK (OB_NOT_NULL(value));
|
|
CK (value->is_ext());
|
|
OX (composite_addr = reinterpret_cast<pl::ObPLComposite*>(value->get_ext()));
|
|
CK (OB_NOT_NULL(composite_addr));
|
|
}
|
|
}
|
|
if (OB_FAIL(ret)) {
|
|
} else if (pl::ObObjAccessIdx::get_final_type(access_idxs_).is_obj_type()) {
|
|
*element_val = reinterpret_cast<int64_t>(current_value);
|
|
} else {
|
|
*element_val = reinterpret_cast<int64_t>(composite_addr);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int ObExprObjAccess::ExtraInfo::calc(ObObj &result,
|
|
ObIAllocator &alloc,
|
|
const ObObjMeta &res_type,
|
|
const int32_t extend_size,
|
|
const ParamStore ¶m_store,
|
|
const common::ObObj *params,
|
|
int64_t param_num,
|
|
ObEvalCtx *ctx) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
typedef int32_t (*GetAttr)(int64_t, int64_t [], int64_t *);
|
|
GetAttr get_attr = reinterpret_cast<GetAttr>(get_attr_func_);
|
|
ParamArray param_array;
|
|
OZ (init_param_array(param_store, params, param_num, param_array));
|
|
|
|
if (OB_SUCC(ret)) {
|
|
int64_t *param_ptr = const_cast<int64_t *>(param_array.head());
|
|
int64_t attr_addr = 0;
|
|
if (OB_NOT_NULL(get_attr)) {
|
|
OZ (get_attr(param_array.count(), param_ptr, &attr_addr));
|
|
} else {
|
|
CK (OB_NOT_NULL(ctx));
|
|
OZ (get_attr_func(param_array.count(), param_ptr, &attr_addr, *ctx));
|
|
}
|
|
if (OB_FAIL(ret)) {
|
|
if (OB_ERR_COLLECION_NULL == ret && pl::ObCollectionType::EXISTS_PROPERTY == property_type_) {
|
|
ret = OB_SUCCESS;
|
|
result.set_tinyint(0);
|
|
}
|
|
} else if (OB_UNLIKELY(0 >= attr_addr)) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
LOG_WARN("get attribute failed", K(ret), K(attr_addr));
|
|
} else if (for_write_ || res_type.is_ext()) {
|
|
// 当对collection元素赋值的时候,强制设置collection的first和last为invalid
|
|
// 在获取first,last的时候,会根据这个标记进行更新。
|
|
// 主要是因为collection有删除操作的时候,而又赋值了,需要一个地方给出提示
|
|
// 这儿就是一个标记,需要明确的,这可能会导致first last暂时不准确。但get_first会自动更新
|
|
// 所有需要使用first,last的地方应该调用get_first这样的函数,而不能直接使用first的值
|
|
// assoc array 不需要处理,因为在accociative_index里面更新了first,last
|
|
if (for_write_ && OB_INVALID_INDEX != coll_idx_) {
|
|
if (0 != param_array.at(coll_idx_)) {
|
|
pl::ObPLCollection *coll = reinterpret_cast<pl::ObPLCollection*>(param_array.at(coll_idx_));
|
|
if (pl::PL_NESTED_TABLE_TYPE == coll->get_type()
|
|
|| pl::PL_VARRAY_TYPE == coll->get_type()) {
|
|
OX (coll->set_first(OB_INVALID_INDEX));
|
|
OX (coll->set_last(OB_INVALID_INDEX));
|
|
}
|
|
}
|
|
}
|
|
OX(result.set_extend(attr_addr, res_type.get_extend_type(), extend_size));
|
|
#ifdef OB_BUILD_ORACLE_PL
|
|
} else if (pl::ObCollectionType::INVALID_PROPERTY != property_type_) {
|
|
pl::ObPLCollection *coll = reinterpret_cast<pl::ObPLCollection*>(attr_addr);
|
|
pl::ObPLAssocArray *assoc =
|
|
(OB_NOT_NULL(coll) && coll->is_associative_array())
|
|
? static_cast<pl::ObPLAssocArray *>(coll) : NULL;
|
|
CK (OB_NOT_NULL(coll));
|
|
if (OB_SUCC(ret)) {
|
|
switch (property_type_) {
|
|
case pl::ObCollectionType::FIRST_PROPERTY: {
|
|
OZ (OB_NOT_NULL(assoc) ? assoc->first(result) : coll->first(result));
|
|
} break;
|
|
case pl::ObCollectionType::LAST_PROPERTY: {
|
|
OZ (OB_NOT_NULL(assoc) ? assoc->last(result) : coll->last(result));
|
|
} break;
|
|
case pl::ObCollectionType::PRIOR_PROPERTY: {
|
|
CK (param_array.count() > 1);
|
|
int64_t idx = param_array.count() - 1;
|
|
OZ (OB_NOT_NULL(assoc)
|
|
? assoc->prior(param_array.at(idx), result) : coll->prior(param_array.at(idx), result));
|
|
} break;
|
|
case pl::ObCollectionType::NEXT_PROPERTY: {
|
|
CK (param_array.count() > 1);
|
|
int64_t idx = param_array.count() - 1;
|
|
int64_t index_val = param_array.at(idx);
|
|
if (OB_SUCC(ret)) {
|
|
if(OB_INVALID_INDEX == param_array.at(idx)) {
|
|
if (OB_NOT_NULL(assoc)) {
|
|
OZ (assoc->next(index_val, result));
|
|
} else {
|
|
result.set_null();
|
|
}
|
|
} else {
|
|
OZ (OB_NOT_NULL(assoc) ? assoc->next(index_val, result) : coll->next(index_val, result));
|
|
}
|
|
}
|
|
} break;
|
|
case pl::ObCollectionType::EXISTS_PROPERTY: {
|
|
CK (param_array.count() > 1);
|
|
int64_t idx = param_array.count() - 1;
|
|
OZ (OB_NOT_NULL(assoc)
|
|
? assoc->exist(param_array.at(idx), result) : coll->exist(param_array.at(idx), result));
|
|
} break;
|
|
case pl::ObCollectionType::COUNT_PROPERTY: {
|
|
result.set_int(coll->get_actual_count());
|
|
} break;
|
|
case pl::ObCollectionType::LIMIT_PROPERTY: {
|
|
if (coll->is_varray()) {
|
|
pl::ObPLVArray *varr = static_cast<pl::ObPLVArray *>(coll);
|
|
CK (OB_NOT_NULL(varr));
|
|
OX (result.set_int(varr->get_capacity()));
|
|
} else {
|
|
result.set_null();
|
|
}
|
|
} break;
|
|
default: {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
LOG_WARN("invalid property type", K(ret), K(property_type_));
|
|
} break;
|
|
}
|
|
}
|
|
#endif
|
|
} else {
|
|
ObObj *datum = reinterpret_cast<ObObj*>(attr_addr);
|
|
if (ObMaxType == datum->get_type()) { //means has been deleted
|
|
ret = OB_READ_NOTHING;
|
|
LOG_WARN("accessing deleted element, no data found", K(ret), KPC(datum), K(result));
|
|
} else if (res_type.is_number() && datum->is_decimal_int()) {
|
|
ObCastCtx cast_ctx(&alloc, NULL, CM_NONE, res_type.get_collation_type(), NULL);
|
|
const ObObj *res_obj = nullptr;
|
|
if (OB_FAIL(ObObjCaster::to_type(ObNumberType, cast_ctx, *datum, result, res_obj))) {
|
|
LOG_WARN("failed to cast decimal int to number", K(ret));
|
|
}
|
|
} else if (OB_FAIL(result.apply(*datum))) {
|
|
LOG_WARN("apply failed", K(ret), KPC(datum), K(result), K(res_type));
|
|
}
|
|
if (OB_SUCC(ret)) {
|
|
if ((ObLongTextType == result.get_meta().get_type()
|
|
&& res_type.is_lob_locator())
|
|
|| (result.get_meta().is_lob_locator() && ObLongTextType == res_type.get_type())) {
|
|
} else if (!result.is_null()
|
|
&& result.get_meta().get_type() != res_type.get_type()) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
LOG_WARN("obj access result meta not equel to expr type",
|
|
K(ret), KPC(datum), K(result), K(res_type));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int ObExprObjAccess::ExtraInfo::from_raw_expr(const ObObjAccessRawExpr &raw_access)
|
|
{
|
|
int ret = 0;
|
|
if (OB_SUCC(ret)) {
|
|
extend_size_ = raw_access.get_result_type().get_extend_size();
|
|
get_attr_func_ = raw_access.get_get_attr_func_addr();
|
|
for_write_ = raw_access.for_write();
|
|
property_type_ = raw_access.get_property();
|
|
access_idx_cnt_ = raw_access.get_access_idxs().count();
|
|
coll_idx_ = OB_INVALID_INDEX;
|
|
if (raw_access.get_access_idxs().at(0).elem_type_.is_collection_type()) {
|
|
coll_idx_ = raw_access.get_access_idxs().at(0).var_index_;
|
|
}
|
|
OZ(param_idxs_.init(raw_access.get_var_indexs().count()));
|
|
OZ(param_idxs_.assign(raw_access.get_var_indexs()));
|
|
OZ(access_idxs_.init(raw_access.get_access_idxs().count()));
|
|
OZ(access_idxs_.assign(raw_access.get_access_idxs()));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int ObExprObjAccess::cg_expr(ObExprCGCtx &op_cg_ctx,
|
|
const ObRawExpr &raw_expr, ObExpr &rt_expr) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ObIAllocator &alloc = *op_cg_ctx.allocator_;
|
|
ExtraInfo *info = OB_NEWx(ExtraInfo, (&alloc), alloc, T_OBJ_ACCESS_REF);
|
|
if (NULL == info) {
|
|
ret = OB_ALLOCATE_MEMORY_FAILED;
|
|
LOG_WARN("allocate memory failed", K(ret));
|
|
} else {
|
|
const ObObjAccessRawExpr &raw_access = static_cast<const ObObjAccessRawExpr &>(raw_expr);
|
|
if (OB_SUCC(ret)) {
|
|
info->extend_size_ = raw_expr.get_result_type().get_extend_size();
|
|
info->get_attr_func_ = raw_access.get_get_attr_func_addr();
|
|
info->for_write_ = raw_access.for_write();
|
|
info->property_type_ = raw_access.get_property();
|
|
info->access_idx_cnt_ = raw_access.get_access_idxs().count();
|
|
int64_t coll_idx = OB_INVALID_INDEX;
|
|
if (raw_access.get_access_idxs().at(0).elem_type_.is_collection_type()) {
|
|
coll_idx = raw_access.get_access_idxs().at(0).var_index_;
|
|
}
|
|
info->coll_idx_ = coll_idx;
|
|
OZ(info->param_idxs_.init(raw_access.get_var_indexs().count()));
|
|
OZ(info->param_idxs_.assign(raw_access.get_var_indexs()));
|
|
OZ(info->access_idxs_.init(raw_access.get_access_idxs().count()));
|
|
OZ(info->access_idxs_.assign(raw_access.get_access_idxs()));
|
|
}
|
|
if (OB_SUCC(ret)) {
|
|
rt_expr.extra_info_ = info;
|
|
rt_expr.eval_func_ = eval_obj_access;
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int ObExprObjAccess::eval_obj_access(const ObExpr &expr,
|
|
ObEvalCtx &ctx,
|
|
ObDatum &expr_datum)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
const ExtraInfo *info = static_cast<const ExtraInfo *>(expr.extra_info_);
|
|
const ParamStore ¶m_store = ctx.exec_ctx_.get_physical_plan_ctx()->get_param_store();
|
|
OZ(expr.eval_param_value(ctx));
|
|
ObObj params[expr.arg_cnt_];
|
|
for (int64_t i = 0; OB_SUCC(ret) && i < expr.arg_cnt_; i++) {
|
|
const ObExpr *e = expr.args_[i];
|
|
const ObDatum &d = e->locate_expr_datum(ctx);
|
|
OZ(d.to_obj(params[i], e->obj_meta_, e->obj_datum_map_));
|
|
}
|
|
|
|
ObObj result;
|
|
ObEvalCtx::TempAllocGuard alloc_guard(ctx);
|
|
OZ(info->calc(result,
|
|
alloc_guard.get_allocator(),
|
|
expr.obj_meta_,
|
|
info->extend_size_,
|
|
param_store,
|
|
params,
|
|
expr.arg_cnt_,
|
|
&ctx));
|
|
|
|
OZ(expr_datum.from_obj(result, expr.obj_datum_map_));
|
|
if (is_lob_storage(result.get_type())) {
|
|
OZ (ob_adjust_lob_datum(result, expr.obj_meta_, ctx.exec_ctx_.get_allocator(), expr_datum));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
|
|
|
|
} // namespace sql
|
|
} // namespace oceanbase
|