hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 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

130
src/sql/optimizer/ob_insert_log_plan.h
View File

@ -12,48 +12,114 @@
#ifndef _OB_INSERT_LOG_PLAN_H
#define _OB_INSERT_LOG_PLAN_H
#include "ob_log_plan.h"
#include "sql/optimizer/ob_del_upd_log_plan.h"
namespace oceanbase {
namespace sql {
namespace oceanbase
{
namespace sql
{
class ObDMLStmt;
class ObLogInsert;
class ObInsertStmt;
struct ObDupKeyScanInfo;
typedef common::ObSEArray<common::ObSEArray<int64_t, 8, common::ModulePageAllocator, true>, 1,
common::ModulePageAllocator, true>
RowParamMap;
class ObInsertLogPlan : public ObLogPlan {
typedef common::ObSEArray<common::ObSEArray<int64_t, 8, common::ModulePageAllocator, true>, 1, common::ModulePageAllocator, true> RowParamMap;
class ObInsertLogPlan: public ObDelUpdLogPlan
{
public:
ObInsertLogPlan(ObOptimizerContext& ctx, const ObDMLStmt* insert_stmt) : ObLogPlan(ctx, insert_stmt), insert_op_(NULL)
{}
ObInsertLogPlan(ObOptimizerContext &ctx, const ObInsertStmt *insert_stmt)
: ObDelUpdLogPlan(ctx, insert_stmt)
{ }
virtual ~ObInsertLogPlan()
{}
int generate_raw_plan();
virtual int generate_plan();
inline const RowParamMap& get_row_params_map() const
{
return row_params_map_;
}
{ }
virtual int generate_raw_plan() override;
const ObInsertStmt *get_stmt() const override
{ return reinterpret_cast<const ObInsertStmt*>(stmt_); }
virtual int prepare_dml_infos() override;
common::ObIArray<IndexDMLInfo *> &get_replace_del_index_del_infos()
{ return replace_del_index_del_infos_; }
const common::ObIArray<IndexDMLInfo *> &get_replace_del_index_del_infos() const
{ return replace_del_index_del_infos_; }
common::ObIArray<IndexDMLInfo *> &get_insert_up_index_upd_infos()
{ return insert_up_index_upd_infos_; }
const common::ObIArray<IndexDMLInfo *> &get_insert_up_index_upd_infos() const
{ return insert_up_index_upd_infos_; }
protected:
int generate_values_op_as_child(ObLogicalOperator*& expr_values);
bool is_self_part_insert();
int map_value_param_index();
int generate_duplicate_key_checker(ObLogInsert& insert_op);
int generate_dupkey_scan_info(
const ObDupKeyScanInfo& dupkey_info, ObLogicalOperator*& scan_op, ObRawExpr*& calc_part_expr);
int allocate_insert_op(ObInsertStmt& insert_stmt, ObLogInsert*& insert_op);
int allocate_insert_all_op(ObInsertStmt& insert_stmt, ObLogInsert*& insert_op);
int allocate_pdml_insert_as_top(ObLogicalOperator*& top_op);
int allocate_insert_values_as_top(ObLogicalOperator *&top);
int candi_allocate_insert();
int build_lock_row_flag_expr(ObConstRawExpr *&lock_row_flag_expr);
int create_insert_plans(ObIArray<CandidatePlan> &candi_plans,
ObTablePartitionInfo *insert_table_part,
ObShardingInfo *insert_sharding,
ObConstRawExpr *lock_row_flag_expr,
const bool force_no_multi_part,
const bool force_multi_part,
ObIArray<CandidatePlan> &insert_plans);
int allocate_insert_as_top(ObLogicalOperator *&top,
ObRawExpr *lock_row_flag_expr,
ObTablePartitionInfo *table_partition_info,
ObShardingInfo *insert_sharding,
bool is_multi_part);
int candi_allocate_pdml_insert();
virtual int check_insert_need_multi_partition_dml(ObLogicalOperator &top,
ObTablePartitionInfo *insert_table_partition,
ObShardingInfo *insert_sharding,
bool &is_multi_part_dml);
virtual int check_insert_stmt_need_multi_partition_dml(bool &is_multi_part_dml,
bool &is_one_part_table);
virtual int check_insert_location_need_multi_partition_dml(ObLogicalOperator &top,
ObTablePartitionInfo *insert_table_partition,
ObShardingInfo *insert_sharding,
bool is_one_part_table,
bool &is_multi_part_dml);
int check_basic_sharding_for_insert_stmt(ObShardingInfo &target_sharding,
ObLogicalOperator &child,
bool &is_basic);
int check_if_match_partition_wise_insert(ObShardingInfo &target_sharding,
ObLogicalOperator &child,
bool &is_partition_wise);
virtual int prepare_table_dml_info_special(const ObDmlTableInfo& table_info,
IndexDMLInfo* table_dml_info,
ObIArray<IndexDMLInfo*> &index_dml_infos,
ObIArray<IndexDMLInfo*> &all_index_dml_infos) override;
int copy_index_dml_infos_for_replace(ObIArray<IndexDMLInfo*> &src_dml_infos,
ObIArray<IndexDMLInfo*> &dst_dml_infos);
int copy_index_dml_infos_for_insert_up(const ObInsertTableInfo& table_info,
IndexDMLInfo* table_dml_info,
ObIArray<IndexDMLInfo*> &index_dml_infos,
ObIArray<IndexDMLInfo*> &dst_dml_infos);
int prepare_unique_constraint_infos(const ObDmlTableInfo& table_info);
int prepare_unique_constraint_info(const ObTableSchema &index_schema,
const uint64_t table_id,
ObUniqueConstraintInfo &constraint_info);
int prepare_table_dml_info_for_ddl(const ObInsertTableInfo& table_info,
ObIArray<IndexDMLInfo*> &all_index_dml_infos);
int get_all_rowkey_columns_for_ddl(const ObInsertTableInfo& table_info,
const ObTableSchema* ddl_table_schema,
ObIArray<ObColumnRefRawExpr*> &column_exprs);
int get_all_columns_for_ddl(const ObInsertTableInfo& table_info,
const ObTableSchema* ddl_table_schema,
ObIArray<ObColumnRefRawExpr*> &column_exprs);
int get_all_part_columns_for_ddl(const ObInsertTableInfo& table_info,
const ObTableSchema* data_table_schema,
ObIArray<ObColumnRefRawExpr*> &column_exprs);
int build_column_conv_for_shadow_pk(const ObInsertTableInfo& table_info,
ObColumnRefRawExpr *column_expr,
ObRawExpr *&column_conv_expr);
private:
DISALLOW_COPY_AND_ASSIGN(ObInsertLogPlan);
private:
RowParamMap row_params_map_;
ObLogInsert* insert_op_;
common::ObSEArray<IndexDMLInfo *, 1, common::ModulePageAllocator, true> replace_del_index_del_infos_;
common::ObSEArray<IndexDMLInfo *, 1, common::ModulePageAllocator, true> insert_up_index_upd_infos_;
common::ObSEArray<ObUniqueConstraintInfo, 8, common::ModulePageAllocator, true> uk_constraint_infos_;
};
} // namespace sql
} // namespace oceanbase
#endif // _OB_INSERT_LOG_PLAN_H
}
}
#endif // _OB_INSERT_LOG_PLAN_H