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

132
src/sql/engine/expr/ob_expr_autoinc_nextval.h
View File

@ -13,41 +13,127 @@
#ifndef _OB_EXPR_NEXTVAL_H
#define _OB_EXPR_NEXTVAL_H
#include "sql/engine/expr/ob_expr_operator.h"
#include "sql/engine/expr/ob_i_expr_extra_info.h"
#include "share/ob_autoincrement_service.h"
namespace oceanbase {
namespace share {
class AutoincParam;
namespace oceanbase
{
namespace share
{
}
namespace sql {
namespace sql
{
class ObPhysicalPlanCtx;
class ObExprAutoincNextval : public ObFuncExprOperator {
class ObExprAutoincNextval : public ObFuncExprOperator
{
OB_UNIS_VERSION_V(1);
public:
explicit ObExprAutoincNextval(common::ObIAllocator& alloc);
explicit ObExprAutoincNextval(common::ObIAllocator &alloc);
virtual ~ObExprAutoincNextval();
virtual int calc_result_typeN(
ObExprResType& type, ObExprResType* types_array, int64_t param_num, common::ObExprTypeCtx& type_ctx) const;
virtual int calc_resultN(
common::ObObj& result, const common::ObObj* objs_array, int64_t param_num, common::ObExprCtx& expr_ctx) const;
virtual int cg_expr(ObExprCGCtx& op_cg_ctx, const ObRawExpr& raw_expr, ObExpr& rt_expr) const override;
static int eval_nextval(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& expr_datum);
static int get_uint_value(const ObExpr& input_expr, ObDatum* input_value, bool& is_zero, uint64_t& casted_value);
virtual int calc_result_typeN(ObExprResType &type,
ObExprResType *types_array,
int64_t param_num,
common::ObExprTypeCtx &type_ctx) const;
virtual int cg_expr(ObExprCGCtx &op_cg_ctx,
const ObRawExpr &raw_expr,
ObExpr &rt_expr) const override;
static int eval_nextval(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum);
static int get_uint_value(const ObExpr &input_expr, ObDatum *input_value,
bool &is_zero, uint64_t &casted_value);
private:
// check to generate auto-inc value or not and cast.
static int check_and_cast(common::ObObj& result, const common::ObObj* objs_array, int64_t param_num,
common::ObExprCtx& expr_ctx, share::AutoincParam* autoinc_param, bool& is_to_generate, uint64_t& casted_value);
static int generate_autoinc_value(uint64_t& new_val, share::ObAutoincrementService& auto_service,
share::AutoincParam* autoinc_param, ObPhysicalPlanCtx* plan_ctx);
//check to generate auto-inc value or not and cast.
static int check_and_cast(common::ObObj &result,
common::ObObjType result_type,
const common::ObObj *objs_array,
int64_t param_num,
common::ObExprCtx &expr_ctx,
share::AutoincParam *autoinc_param,
bool &is_to_generate,
uint64_t &casted_value);
static int generate_autoinc_value(const ObSQLSessionInfo &my_session,
uint64_t &new_val,
share::ObAutoincrementService &auto_service,
share::AutoincParam *autoinc_param,
ObPhysicalPlanCtx *plan_ctx);
// get the input value && check need generate
static int get_input_value(const ObExpr& input_expr, ObEvalCtx& ctx, ObDatum* input_value,
share::AutoincParam& autoinc_param, bool& is_to_generate, uint64_t& casted_value);
static int get_input_value(const ObExpr &expr,
ObEvalCtx &ctx,
ObDatum *input_value,
share::AutoincParam &autoinc_param,
bool &is_to_generate,
uint64_t &casted_value);
static int get_casted_value_by_result_type(common::ObCastCtx &cast_ctx,
common::ObObjType result_type,
const common::ObObj &param,
uint64_t &casted_value,
bool &try_sync);
// disallow copy
DISALLOW_COPY_AND_ASSIGN(ObExprAutoincNextval);
};
} // end namespace sql
} // end namespace oceanbase
// used in expr.extra_ to pass in table and column info
struct ObAutoincNextvalExtra
{
OB_UNIS_VERSION_V(1);
public:
ObAutoincNextvalExtra()
: autoinc_table_id_(common::OB_INVALID_ID),
autoinc_col_id_(common::OB_INVALID_ID),
autoinc_table_name_(),
autoinc_column_name_()
{}
virtual ~ObAutoincNextvalExtra() {}
static int init_autoinc_nextval_extra(common::ObIAllocator *allocator,
ObRawExpr *&expr,
const uint64_t autoinc_table_id,
const uint64_t autoinc_col_id,
const ObString autoinc_table_name,
const ObString autoinc_column_name);
TO_STRING_KV(K_(autoinc_table_id),
K_(autoinc_col_id),
K_(autoinc_table_name),
K_(autoinc_column_name));
uint64_t autoinc_table_id_;
uint64_t autoinc_col_id_;
ObString autoinc_table_name_;
ObString autoinc_column_name_;
};
// used in ObExprAutoincNextval to pass in table and column info
struct ObAutoincNextvalInfo : public ObIExprExtraInfo
{
OB_UNIS_VERSION(1);
public:
ObAutoincNextvalInfo(common::ObIAllocator &alloc, ObExprOperatorType type)
: ObIExprExtraInfo(alloc, type),
autoinc_table_id_(common::OB_INVALID_ID),
autoinc_col_id_(common::OB_INVALID_ID)
{}
virtual ~ObAutoincNextvalInfo() { }
static int init_autoinc_nextval_info(common::ObIAllocator *allocator,
const ObRawExpr &raw_expr,
ObExpr &expr,
const ObExprOperatorType type);
virtual int deep_copy(common::ObIAllocator &allocator,
const ObExprOperatorType type,
ObIExprExtraInfo *&copied_info) const override;
TO_STRING_KV(K_(autoinc_table_id), K_(autoinc_col_id));
uint64_t autoinc_table_id_;
uint64_t autoinc_col_id_;
};
}//end namespace sql
}//end namespace oceanbase
#endif