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

[Shein-Feature] support auto_increment attribute for part column

Browse Source
This commit is contained in:
raywill
2021-07-30 11:00:12 +08:00
committed by wangzelin.wzl
parent 6ee95017b0
commit b75f46946e
9 changed files with 64 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/sql/optimizer/ob_log_insert.cpp
View File

@ -686,6 +686,7 @@ int ObLogInsert::need_multi_table_dml(AllocExchContext& ctx, ObShardingInfo& sha
ObSEArray<ObShardingInfo*, 2> input_sharding;
ObShardingInfo output_sharding;
ObLogicalOperator* child = NULL;
bool has_auto_inc_part_key = false;
bool has_rand_part_key = false;
bool has_subquery_part_key = false;
bool trigger_exist = false;
@ -701,7 +702,12 @@ int ObLogInsert::need_multi_table_dml(AllocExchContext& ctx, ObShardingInfo& sha
} else if (OB_ISNULL(tbl_schema)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("table schema is null", K(ret), K(insert_stmt));
} else if (modify_multi_tables()) {
is_needed = true;
} else if (is_table_update_part_key() && tbl_schema->get_all_part_num() > 1) {
is_needed = true;
}
if (OB_SUCC(ret) && OB_FAIL(generate_sharding_info(target_sharding_info))) {
if (ret == OB_NO_PARTITION_FOR_GIVEN_VALUE && trigger_exist) {
ret = OB_SUCCESS;
@ -709,18 +715,18 @@ int ObLogInsert::need_multi_table_dml(AllocExchContext& ctx, ObShardingInfo& sha
LOG_WARN("failed to generate sharding info", K(ret));
}
}
if (OB_FAIL(ret) || is_needed) {
} else if (modify_multi_tables()) {
is_needed = true;
} else if (is_table_update_part_key() && tbl_schema->get_all_part_num() > 1) {
is_needed = true;
// nop
} else if (is_table_insert_sequence_part_key()) {
is_needed = true;
} else if (OB_FAIL(insert_stmt->part_key_has_auto_inc(has_auto_inc_part_key))) {
LOG_WARN("check to check whether part key containts auto inc column", K(ret));
} else if (OB_FAIL(insert_stmt->part_key_has_rand_value(has_rand_part_key))) {
LOG_WARN("failed to check whether part key containts random value", K(ret));
} else if (OB_FAIL(insert_stmt->part_key_has_subquery(has_subquery_part_key))) {
LOG_WARN("check to check whether part key containts subquery", K(ret));
} else if (has_rand_part_key || has_subquery_part_key) {
} else if (has_auto_inc_part_key || has_rand_part_key || has_subquery_part_key) {
is_needed = true;
} else if (get_part_hint() != NULL && insert_stmt->value_from_select()) {
is_needed = true;
@ -735,11 +741,6 @@ int ObLogInsert::need_multi_table_dml(AllocExchContext& ctx, ObShardingInfo& sha
} else if (is_basic) {
is_needed = false;
sharding_info.copy_with_part_keys(target_sharding_info);
} else if (target_sharding_info.is_distributed() && child->get_sharding_info().is_match_all() &&
insert_stmt->with_explicit_autoinc_column() && !is_insert_select()) {
is_needed = false;
ctx.plan_type_ = AllocExchContext::DistrStat::DISTRIBUTED;
sharding_info.copy_with_part_keys(target_sharding_info);
} else if (OB_FAIL(check_if_match_partition_wise_insert(
ctx, target_sharding_info, child->get_sharding_info(), is_match))) {
LOG_WARN("failed to check partition wise", K(ret));