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

217
src/sql/engine/expr/ob_expr_lrpad.h
View File

@ -15,134 +15,193 @@
#include "sql/engine/expr/ob_expr_operator.h"
namespace oceanbase {
namespace sql {
class ObExprBaseLRpad : public ObStringExprOperator {
namespace oceanbase
{
namespace sql
{
class ObExprBaseLRpad : public ObStringExprOperator
{
public:
enum LRpadType { LPAD_TYPE = 0, RPAD_TYPE = 1 };
explicit ObExprBaseLRpad(common::ObIAllocator& alloc, ObExprOperatorType type, const char* name, int32_t param_num);
enum LRpadType { LPAD_TYPE = 0, RPAD_TYPE = 1};
explicit ObExprBaseLRpad(common::ObIAllocator &alloc,
ObExprOperatorType type,
const char *name,
int32_t param_num);
virtual ~ObExprBaseLRpad();
static int calc_type(ObExprResType& type, ObExprResType& text, ObExprResType& len, ObExprResType* pad_text,
common::ObExprTypeCtx& type_ctx);
static int calc_type(ObExprResType &type,
ObExprResType &text,
ObExprResType &len,
ObExprResType *pad_text,
common::ObExprTypeCtx &type_ctx);
static int padding(LRpadType type, const common::ObCollationType coll_type, const char* text,
const int64_t& text_size, const char* pad, const int64_t& pad_size, const int64_t& prefix_size,
const int64_t& repeat_count,
const bool& pad_space, // for oracle
common::ObIAllocator* allocator, char*& result, int64_t& size);
static int padding(LRpadType type,
const common::ObCollationType coll_type,
const char *text,
const int64_t &text_size,
const char *pad,
const int64_t &pad_size,
const int64_t &prefix_size,
const int64_t &repeat_count,
const bool &pad_space, // for oracle
common::ObIAllocator *allocator,
char* &result,
int64_t &size);
int calc(const LRpadType type, const common::ObObj& text, const common::ObObj& len, const common::ObObj& pad_text,
common::ObExprCtx& expr_ctx, common::ObObj& result) const;
int calc(const LRpadType type,
const common::ObObj &text,
const common::ObObj &len,
const common::ObObj &pad_text,
common::ObExprCtx &expr_ctx,
common::ObObj &result) const;
// @param [in/out] use result.collation as input param
// for ObExprPad
static int calc_mysql(const LRpadType type, const ObExprResType result_type, const common::ObObj& text,
const common::ObObj& len, const common::ObObj& pad_text, const ObSQLSessionInfo* session,
common::ObIAllocator* allocator, common::ObObj& result);
static int calc_mysql(const LRpadType type,
const ObExprResType result_type,
const common::ObObj &text,
const common::ObObj &len,
const common::ObObj &pad_text,
const ObSQLSessionInfo *session,
common::ObIAllocator *allocator,
common::ObObj &result);
static int get_padding_info_mysql(const common::ObCollationType& cs, const common::ObString& str_text,
const int64_t& len, const common::ObString& str_padtext, const int64_t max_result_size, int64_t& repeat_count,
int64_t& prefix_size, int64_t& size);
static int get_padding_info_mysql(const common::ObCollationType &cs,
const common::ObString &str_text,
const int64_t &len,
const common::ObString &str_padtext,
const int64_t max_result_size,
int64_t &repeat_count,
int64_t &prefix_size,
int64_t &size);
static int calc_oracle(const LRpadType type, const ObExprResType result_type, const common::ObObj& text,
const common::ObObj& len, const common::ObObj& pad_text, common::ObExprCtx& expr_ctx, common::ObObj& result);
static int calc_type_length_mysql(const ObExprResType result_type, const common::ObObj& text,
const common::ObObj& pad_text, const common::ObObj& len, const ObBasicSessionInfo* session, int64_t& result_size);
static int calc_type_length_oracle(const ObExprResType& result_type, const common::ObObj& text,
const common::ObObj& pad_text, const common::ObObj& len, int64_t& result_size);
static int calc_oracle(const LRpadType type,
const ObExprResType result_type,
const common::ObObj &text,
const common::ObObj &len,
const common::ObObj &pad_text,
common::ObExprCtx &expr_ctx,
common::ObObj &result);
static int calc_type_length_mysql(const ObExprResType result_type,
const common::ObObj &text,
const common::ObObj &pad_text,
const common::ObObj &len,
const ObBasicSessionInfo *session,
int64_t &result_size);
static int calc_type_length_oracle(const ObExprResType &result_type,
const common::ObObj &text,
const common::ObObj &pad_text,
const common::ObObj &len,
int64_t &result_size);
static int get_padding_info_oracle(const common::ObCollationType cs, const common::ObString& str_text,
const int64_t& width, const common::ObString& str_padtext, const int64_t max_result_size, int64_t& repeat_count,
int64_t& prefix_size, bool& pad_space);
static int get_padding_info_oracle(const common::ObCollationType cs,
const common::ObString &str_text,
const int64_t &width,
const common::ObString &str_padtext,
const int64_t max_result_size,
int64_t &repeat_count,
int64_t &prefix_size,
bool &pad_space);
// for engine 3.0
static int calc_mysql_pad_expr(const ObExpr& expr, ObEvalCtx& ctx, LRpadType pad_type, ObDatum& res);
static int calc_mysql(const LRpadType pad_type, const ObExpr& expr, const common::ObDatum& text,
const common::ObDatum& len, const common::ObDatum& pad_text, const ObSQLSessionInfo& session,
common::ObIAllocator& res_alloc, ObDatum& res);
static int calc_oracle_pad_expr(const ObExpr& expr, ObEvalCtx& ctx, LRpadType pad_type, ObDatum& res);
static int calc_oracle(LRpadType pad_type, const ObExpr& expr, const common::ObDatum& text,
const common::ObDatum& len, const common::ObDatum& pad_text, common::ObIAllocator& res_alloc, ObDatum& res);
static int calc_mysql_pad_expr(const ObExpr &expr, ObEvalCtx &ctx, LRpadType pad_type,
ObDatum &res);
static int calc_mysql(const LRpadType pad_type, const ObExpr &expr, const common::ObDatum &text,
const common::ObDatum &len, const common::ObDatum &pad_text,
const ObSQLSessionInfo &session, common::ObIAllocator &res_alloc,
ObDatum &res);
static int calc_oracle_pad_expr(const ObExpr &expr, ObEvalCtx &ctx, LRpadType pad_type,
ObDatum &res);
static int calc_oracle(LRpadType pad_type, const ObExpr &expr, const common::ObDatum &text,
const common::ObDatum &len, const common::ObDatum &pad_text,
common::ObIAllocator &res_alloc, ObDatum &res);
private:
DISALLOW_COPY_AND_ASSIGN(ObExprBaseLRpad);
};
class ObExprLpad : public ObExprBaseLRpad {
class ObExprLpad : public ObExprBaseLRpad
{
public:
explicit ObExprLpad(common::ObIAllocator& alloc);
explicit ObExprLpad(common::ObIAllocator &alloc);
virtual ~ObExprLpad();
virtual int calc_result_type3(ObExprResType& type, ObExprResType& text, ObExprResType& len, ObExprResType& pad_text,
common::ObExprTypeCtx& type_ctx) const override;
virtual int calc_result_type3(ObExprResType &type,
ObExprResType &text,
ObExprResType &len,
ObExprResType &pad_text,
common::ObExprTypeCtx &type_ctx) const override;
virtual int calc_result3(common::ObObj& result, const common::ObObj& text, const common::ObObj& len,
const common::ObObj& pad_text, common::ObExprCtx& expr_ctx) const override;
// for engine 3.0
virtual int cg_expr(ObExprCGCtx& expr_cg_ctx, const ObRawExpr& raw_expr, ObExpr& rt_expr) const override;
static int calc_mysql_lpad_expr(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& res);
virtual int cg_expr(ObExprCGCtx &expr_cg_ctx, const ObRawExpr &raw_expr,
ObExpr &rt_expr) const override;
static int calc_mysql_lpad_expr(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &res);
private:
DISALLOW_COPY_AND_ASSIGN(ObExprLpad);
};
class ObExprRpad : public ObExprBaseLRpad {
class ObExprRpad : public ObExprBaseLRpad
{
public:
explicit ObExprRpad(common::ObIAllocator& alloc);
explicit ObExprRpad(common::ObIAllocator& alloc, ObExprOperatorType type, const char* name);
explicit ObExprRpad(common::ObIAllocator &alloc);
explicit ObExprRpad(common::ObIAllocator &alloc,
ObExprOperatorType type,
const char *name);
virtual ~ObExprRpad();
virtual int calc_result_type3(ObExprResType& type, ObExprResType& text, ObExprResType& len, ObExprResType& pad_text,
common::ObExprTypeCtx& type_ctx) const override;
virtual int calc_result_type3(ObExprResType &type,
ObExprResType &text,
ObExprResType &len,
ObExprResType &pad_text,
common::ObExprTypeCtx &type_ctx) const override;
virtual int calc_result3(common::ObObj& result, const common::ObObj& text, const common::ObObj& len,
const common::ObObj& pad_text, common::ObExprCtx& expr_ctx) const override;
// for engine 3.0
virtual int cg_expr(ObExprCGCtx& expr_cg_ctx, const ObRawExpr& raw_expr, ObExpr& rt_expr) const override;
static int calc_mysql_rpad_expr(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& res);
virtual int cg_expr(ObExprCGCtx &expr_cg_ctx, const ObRawExpr &raw_expr,
ObExpr &rt_expr) const;
static int calc_mysql_rpad_expr(const ObExpr &expr, ObEvalCtx &ctx,
ObDatum &res);
private:
DISALLOW_COPY_AND_ASSIGN(ObExprRpad);
};
class ObExprOracleLpad : public ObExprBaseLRpad {
class ObExprOracleLpad : public ObExprBaseLRpad
{
public:
explicit ObExprOracleLpad(common::ObIAllocator& alloc);
explicit ObExprOracleLpad(common::ObIAllocator &alloc);
virtual ~ObExprOracleLpad();
virtual int calc_result_typeN(ObExprResType& type, ObExprResType* types_array, int64_t param_num,
common::ObExprTypeCtx& type_ctx) const override;
virtual int calc_resultN(common::ObObj& result, const common::ObObj* objs_array, int64_t param_num,
common::ObExprCtx& expr_ctx) const override;
// for engine 3.0
virtual int cg_expr(ObExprCGCtx& expr_cg_ctx, const ObRawExpr& raw_expr, ObExpr& rt_expr) const override;
static int calc_oracle_lpad_expr(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& res);
virtual int calc_result_typeN(ObExprResType &type,
ObExprResType *types_array,
int64_t param_num,
common::ObExprTypeCtx &type_ctx) const override;
virtual int cg_expr(ObExprCGCtx &expr_cg_ctx, const ObRawExpr &raw_expr,
ObExpr &rt_expr) const;
static int calc_oracle_lpad_expr(const ObExpr &expr, ObEvalCtx &ctx,
ObDatum &res);
private:
DISALLOW_COPY_AND_ASSIGN(ObExprOracleLpad);
};
class ObExprOracleRpad : public ObExprBaseLRpad {
class ObExprOracleRpad : public ObExprBaseLRpad
{
public:
explicit ObExprOracleRpad(common::ObIAllocator& alloc);
explicit ObExprOracleRpad(common::ObIAllocator &alloc);
virtual ~ObExprOracleRpad();
virtual int calc_result_typeN(ObExprResType& type, ObExprResType* types_array, int64_t param_num,
common::ObExprTypeCtx& type_ctx) const override;
virtual int calc_resultN(common::ObObj& result, const common::ObObj* objs_array, int64_t param_num,
common::ObExprCtx& expr_ctx) const override;
// for engine 3.0
virtual int cg_expr(ObExprCGCtx& expr_cg_ctx, const ObRawExpr& raw_expr, ObExpr& rt_expr) const override;
static int calc_oracle_rpad_expr(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& res);
virtual int calc_result_typeN(ObExprResType &type,
ObExprResType *types_array,
int64_t param_num,
common::ObExprTypeCtx &type_ctx) const override;
virtual int cg_expr(ObExprCGCtx &expr_cg_ctx, const ObRawExpr &raw_expr,
ObExpr &rt_expr) const;
static int calc_oracle_rpad_expr(const ObExpr &expr, ObEvalCtx &ctx,
ObDatum &res);
private:
DISALLOW_COPY_AND_ASSIGN(ObExprOracleRpad);
};
} // namespace sql
} // namespace oceanbase
#endif // OCEANBASE_SQL_ENGINE_EXPR_OB_SQL_EXPR_RPAD_
} // namespace sql
} // namespace oceanbase
#endif // OCEANBASE_SQL_ENGINE_EXPR_OB_SQL_EXPR_RPAD_