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

ExprCalcPartition performs defensive checks on the types of partition expressions and internal partition information.

Browse Source
This commit is contained in:
qingsuijiu 2024-12-17 11:15:19 +00:00 committed by ob-robot
parent dd30da929d
commit 54a4547147
1 changed files with 30 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/sql/engine/expr/ob_expr_calc_partition_id.cpp
View File

@ -244,8 +244,18 @@ int ObExprCalcPartitionBase::cg_expr(ObExprCGCtx &expr_cg_ctx,
if (GET_MIN_CLUSTER_VERSION() < CLUSTER_VERSION_4_3_5_0) {
fallback = true;
} else {
rt_expr.eval_vector_func_ =
ObPartition * const* part_array = table_schema->get_part_array();
if (OB_ISNULL(part_array) || OB_ISNULL(part_array[0]) ||
OB_ISNULL(part_array[0]->get_high_bound_val().get_obj_ptr())) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("partition_array is null", K(ret));
} else if (part_expr->obj_meta_.get_type() !=
part_array[0]->get_high_bound_val().get_obj_ptr()->get_meta().get_type()) {
fallback = true;
} else {
rt_expr.eval_vector_func_ =
ObExprCalcPartitionBase::fast_calc_partition_level_one_vector;
}
}
} else if (table_schema->is_list_part()) {
if (GET_MIN_CLUSTER_VERSION() < CLUSTER_VERSION_4_3_5_0) {
@ -256,8 +266,25 @@ int ObExprCalcPartitionBase::cg_expr(ObExprCGCtx &expr_cg_ctx,
if (ob_is_json(part_expr_type) || ob_is_urowid(part_expr_type)) {
fallback = true;
} else {
rt_expr.eval_vector_func_ =
ObExprCalcPartitionBase::fast_calc_partition_level_one_vector;
ObPartition * const* part_array = table_schema->get_part_array();
if (OB_ISNULL(part_array) || OB_ISNULL(part_array[0])) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("partition_array is null", K(ret));
} else {
const ObIArray<common::ObNewRow> &list_row_values =
part_array[0]->get_list_row_values();
ObObj *list_part_obj = list_row_values.at(0).cells_;
if (OB_ISNULL(list_part_obj)) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("list_part_obj is null", K(ret));
} else if (part_expr->obj_meta_.get_type() !=
list_part_obj->get_meta().get_type()) {
fallback = true;
} else {
rt_expr.eval_vector_func_ =
ObExprCalcPartitionBase::fast_calc_partition_level_one_vector;
}
}
}
}
} else {