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

init push

Browse Source
This commit is contained in:
oceanbase-admin
2021-05-31 22:56:52 +08:00
commit cea7de1475
7020 changed files with 5689869 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

113
src/sql/engine/cmd/ob_partition_executor_utils.h Normal file
View File

@ -0,0 +1,113 @@
/**
* 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.
*/
#ifndef OCEANBASE_SQL_OB_PARTITION_EXECUTOR_UTILS_
#define OCEANBASE_SQL_OB_PARTITION_EXECUTOR_UTILS_
#include "common/object/ob_object.h"
#include "lib/container/ob_se_array.h"
#include "common/sql_mode/ob_sql_mode.h"
#include "sql/resolver/ob_stmt_type.h"
#include "sql/engine/expr/ob_expr_res_type.h"
#include "sql/resolver/ddl/ob_table_stmt.h"
namespace oceanbase {
namespace share {
namespace schema {
class ObPartition;
class ObSubPartition;
} // namespace schema
} // namespace share
namespace sql {
class ObExecContext;
class ObRawExpr;
class ObCreateTableStmt;
class ObCreateTablegroupStmt;
class ObCreateIndexStmt;
class ObTableStmt;
class ObTablegroupStmt;
class ObPartitionExecutorUtils {
public:
const static int OB_DEFAULT_ARRAY_SIZE = 16;
static int calc_values_exprs(ObExecContext& ctx, ObCreateTableStmt& stmt);
static int calc_values_exprs(ObExecContext& ctx, ObCreateIndexStmt& stmt);
static int calc_values_exprs_for_alter_table(
ObExecContext& ctx, share::schema::ObTableSchema& table_schema, ObPartitionedStmt& stmt);
static int cast_list_expr_to_obj(ObExecContext& ctx, const stmt::StmtType stmt_type, const bool is_subpart,
const int64_t real_part_num, share::schema::ObPartition** partition_array,
share::schema::ObSubPartition** subpartition_array, common::ObIArray<ObRawExpr*>& list_fun_expr,
common::ObIArray<ObRawExpr*>& list_values_exprs);
static int cast_expr_to_obj(ObExecContext& ctx, const stmt::StmtType stmt_type, bool is_list_part,
common::ObIArray<ObRawExpr*>& partition_fun_expr, common::ObIArray<ObRawExpr*>& partition_value_exprs,
common::ObIArray<common::ObObj>& partition_value_objs);
static int set_range_part_high_bound(ObExecContext& ctx, const stmt::StmtType stmt_type,
share::schema::ObTableSchema& table_schema, ObPartitionedStmt& stmt, bool is_subpart);
/*--------------tablegroup related start------------------*/
static int calc_values_exprs(ObExecContext& ctx, ObCreateTablegroupStmt& stmt);
static int calc_values_exprs(ObExecContext& ctx, ObCreateTablegroupStmt& stmt, bool is_subpart);
static int cast_range_expr_to_obj(ObExecContext& ctx, ObCreateTablegroupStmt& stmt, bool is_subpart,
common::ObIArray<common::ObObj>& range_partition_obj);
static int cast_range_expr_to_obj(ObExecContext& ctx, common::ObIArray<ObRawExpr*>& range_values_exprs,
const int64_t fun_expr_num, const stmt::StmtType stmt_type, const bool is_subpart, const int64_t real_part_num,
share::schema::ObPartition** partition_array, share::schema::ObSubPartition** subpartition_array,
common::ObIArray<common::ObObj>& range_partition_obj);
static int cast_list_expr_to_obj(ObExecContext& ctx, ObCreateTablegroupStmt& stmt, bool is_subpar);
static int cast_list_expr_to_obj(ObExecContext& ctx, ObTablegroupStmt& stmt, const bool is_subpart,
share::schema::ObPartition** partition_array, share::schema::ObSubPartition** subpartition_array);
static int cast_expr_to_obj(ObExecContext& ctx, int64_t fun_expr_num,
common::ObIArray<ObRawExpr*>& range_values_exprs, common::ObIArray<common::ObObj>& range_partition_obj);
/*--------------tablegroup related end------------------*/
static int calc_range_values_exprs(ObExecContext& ctx, ObCreateIndexStmt& stmt);
template <typename T>
static int check_increasing_range_value(T** array, int64_t part_num, const stmt::StmtType stmt_type);
static int expr_cal_and_cast(const sql::stmt::StmtType& stmt_type, bool is_list_part, ObExecContext& ctx,
const common::ObObjType fun_expr_type, const common::ObCollationType fun_collation_type, ObRawExpr* expr,
common::ObObj& value_obj);
static int expr_cal_and_cast_with_check_varchar_len(const stmt::StmtType& stmt_type, bool is_list_part,
ObExecContext& ctx, const sql::ObExprResType& dst_res_type, ObRawExpr* expr, common::ObObj& value_obj);
static int set_list_part_rows(ObExecContext& ctx, const stmt::StmtType stmt_type,
share::schema::ObTableSchema& table_schema, ObIArray<ObRawExpr*>& list_fun_exprs,
ObIArray<ObRawExpr*>& list_values_exprs, bool is_subpart);
static int set_individual_range_part_high_bound(ObExecContext& ctx, const stmt::StmtType stmt_type,
share::schema::ObTableSchema& table_schema, ObPartitionedStmt& stmt);
static int set_individual_list_part_rows(ObExecContext& ctx, const stmt::StmtType stmt_type,
share::schema::ObTableSchema& table_schema, ObIArray<ObRawExpr*>& list_fun_exprs,
ObDDLStmt::array_array_t& list_values_exprs_array);
static int row_expr_to_array(ObRawExpr* row_expr, ObIArray<ObRawExpr*>& list_values_expr_array);
private:
static int calc_values_exprs(ObExecContext& ctx, const stmt::StmtType stmt_type,
share::schema::ObTableSchema& table_schema, ObPartitionedStmt& stmt, bool is_subpart);
static int sort_list_paritition_if_need(share::schema::ObTableSchema& table_schema);
};
} // end namespace sql
} // end namespace oceanbase
#endif // OCEANBASE_SQL_OB_PARTITION_EXECUTOR_UTILS_