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

218
src/sql/engine/expr/ob_datum_cast.h
View File

@ -23,73 +23,142 @@
#include "share/object/ob_obj_cast.h"
#include "sql/engine/expr/ob_expr.h"
namespace oceanbase {
namespace sql {
namespace oceanbase
{
namespace sql
{
class ObPhysicalPlanCtx;
// extract accuracy info from %expr and call datum_accuracy_check() below.
int datum_accuracy_check(const ObExpr& expr, const uint64_t cast_mode, ObEvalCtx& ctx, const common::ObDatum& in_datum,
ObDatum& res_datum, int& warning);
int datum_accuracy_check(const ObExpr &expr,
const uint64_t cast_mode,
ObEvalCtx &ctx,
const common::ObDatum &in_datum,
ObDatum &res_datum,
int &warning);
// check if accuracy in %in_datum is ok. if ok, call res_datum.set_datum(in_datum).
// if not, will trunc data in %in_datum and put it in res_datum.
// this func makes sure data in %in_datum and %in_datum itself will not be changed.
int datum_accuracy_check(const ObExpr& expr, const uint64_t cast_mode, ObEvalCtx& ctx, const common::ObAccuracy& acc,
const common::ObDatum& in_datum, ObDatum& res_datum, int& warning);
// Check whether str is legal with check_cs_type as the character set
// In strict_mode, if the above check fails, an error code is returned
// Otherwise, return the maximum legal string with check_cs_type as the character set
int string_collation_check(const bool is_strict_mode, const common::ObCollationType check_cs_type,
const common::ObObjType str_type, common::ObString& str);
int datum_accuracy_check(const ObExpr &expr,
const uint64_t cast_mode,
ObEvalCtx &ctx,
const common::ObAccuracy &acc,
const common::ObDatum &in_datum,
ObDatum &res_datum,
int &warning);
int ob_datum_to_ob_time_with_date(const common::ObDatum& datum, const common::ObObjType type,
const common::ObTimeZoneInfo* tz_info, common::ObTime& ob_time, const int64_t cur_ts_value,
bool is_dayofmonth = false);
int ob_datum_to_ob_time_without_date(const common::ObDatum& datum, const common::ObObjType type,
const common::ObTimeZoneInfo* tz_info, common::ObTime& ob_time);
int common_datetime_string(const common::ObObjType in_type, const common::ObObjType out_type,
const common::ObScale in_scale, bool force_use_std_nls_format, const int64_t in_val, ObEvalCtx& ctx, char* buf,
int64_t buf_len, int64_t& out_len);
int padding_char_for_cast(int64_t padding_cnt, const common::ObCollationType& padding_cs_type,
common::ObIAllocator& alloc, common::ObString& padding_res);
// 检查str以check_cs_type作为字符集是否合法
// strict_mode下，如果上述检查失败，返回错误码
// 否则返回以check_cs_type作为字符集的最长合法字符串
int string_collation_check(const bool is_strict_mode,
const common::ObCollationType check_cs_type,
const common::ObObjType str_type,
common::ObString &str);
class ObDatumHexUtils {
// 将datum中的值转换为ob_time结构
int ob_datum_to_ob_time_with_date(const common::ObDatum &datum,
const common::ObObjType type,
const common::ObTimeZoneInfo* tz_info,
common::ObTime& ob_time,
const int64_t cur_ts_value,
const bool is_dayofmonth,
const ObDateSqlMode date_sql_mode);
int ob_datum_to_ob_time_without_date(const common::ObDatum &datum,
const common::ObObjType type,
const common::ObTimeZoneInfo *tz_info,
common::ObTime &ob_time);
// 进行datetime到string的转换，除了ob_datum_cast.cpp需要使用，有的表达式也需要将结果
// 从datetime转为string, 例如ObExprTimeStampAdd
int common_datetime_string(const common::ObObjType in_type,
const common::ObObjType out_type,
const common::ObScale in_scale,
bool force_use_std_nls_format,
const int64_t in_val, ObEvalCtx &ctx, char *buf,
int64_t buf_len, int64_t &out_len);
int padding_char_for_cast(int64_t padding_cnt,
const common::ObCollationType &padding_cs_type,
common::ObIAllocator &alloc,
common::ObString &padding_res);
// copied from ob_obj_cast.cpp，函数逻辑没有修改，只是将输入参数从ObObj变为ObDatum
class ObDatumHexUtils
{
public:
static int hextoraw_string(const ObExpr& expr, const common::ObString& in_str, ObEvalCtx& ctx, ObDatum& res_datum);
static int hextoraw(const ObExpr& expr, const common::ObDatum& in, const common::ObObjType& in_type,
const common::ObCollationType& in_cs_type, ObEvalCtx& ctx, ObDatum& res);
static int get_uint(const common::ObObjType& in_type, const common::ObDatum& in, common::ObIAllocator& alloc,
common::number::ObNumber& out);
static int uint_to_raw(
const common::number::ObNumber& uint_num, const ObExpr& expr, ObEvalCtx& ctx, ObDatum& res_datum);
static int unhex(const ObExpr& expr, const common::ObString& in_str, ObEvalCtx& ctx, ObDatum& res_datum);
static int rawtohex(const ObExpr& expr, const common::ObString& in_str, ObEvalCtx& ctx, ObDatum& res_datum);
static int hex(const ObExpr& expr, const common::ObString& in_str, ObEvalCtx& ctx, common::ObIAllocator& calc_alloc,
ObDatum& res_datum);
static int hextoraw_string(const ObExpr &expr,
const common::ObString &in_str,
ObEvalCtx &ctx,
ObDatum &res_datum);
static int hextoraw(const ObExpr &expr, const common::ObDatum &in,
const common::ObObjType &in_type,
const common::ObCollationType &in_cs_type,
ObEvalCtx &ctx, ObDatum &res);
static int get_uint(const common::ObObjType &in_type, const common::ObDatum &in,
common::ObIAllocator &alloc, common::number::ObNumber &out);
static int uint_to_raw(const common::number::ObNumber &uint_num, const ObExpr &expr,
ObEvalCtx &ctx, ObDatum &res_datum);
static int unhex(const ObExpr &expr,
const common::ObString &in_str,
ObEvalCtx &ctx,
ObDatum &res_datum);
static int rawtohex(const ObExpr &expr,
const common::ObString &in_str,
ObEvalCtx &ctx,
ObDatum &res_datum);
static int hex(const ObExpr &expr,
const common::ObString &in_str,
ObEvalCtx &ctx,
common::ObIAllocator &calc_alloc,
ObDatum &res_datum,
bool upper_case = true);
};
class ObDatumCast {
class ObDatumCast
{
public:
static int get_implicit_cast_function(const common::ObObjType in_type, const common::ObCollationType in_cs_type,
const common::ObObjType out_type, const common::ObCollationType out_cs_type, const int64_t cast_mode,
ObExpr::EvalFunc& eval_func);
static int choose_cast_function(const common::ObObjType in_type, const common::ObCollationType in_cs_type,
const common::ObObjType out_type, const common::ObCollationType out_cs_type, const int64_t cast_mode,
common::ObIAllocator& allocator, ObExpr& rt_expr);
static int get_enumset_cast_function(
const common::ObObjTypeClass in_tc, const common::ObObjType out_type, ObExpr::EvalEnumSetFunc& eval_func);
static int check_can_cast(const common::ObObjType in_type, const common::ObCollationType in_cs_type,
const common::ObObjType out_type, const common::ObCollationType out_cs_type);
static int is_trivial_cast(const common::ObObjType in_type, const common::ObCollationType in_cs_type,
const common::ObObjType out_type, const common::ObCollationType out_cs_type, const common::ObCastMode& cast_mode,
bool& just_eval_arg);
static int get_implicit_cast_function(const common::ObObjType in_type,
const common::ObCollationType in_cs_type,
const common::ObObjType out_type,
const common::ObCollationType out_cs_type,
const int64_t cast_mode,
ObExpr::EvalFunc &eval_func);
// 根据in_type/out_type等信息，获取cast func
static int choose_cast_function(const common::ObObjType in_type,
const common::ObCollationType in_cs_type,
const common::ObObjType out_type,
const common::ObCollationType out_cs_type,
const int64_t cast_mode,
common::ObIAllocator &allocator,
ObExpr &rt_expr);
static int get_enumset_cast_function(const common::ObObjTypeClass in_tc,
const common::ObObjType out_type,
ObExpr::EvalEnumSetFunc &eval_func);
// 检查转换是否合法，有些检查是没办法反映在转换矩阵里面
// 举例：string/text -> string/text时，如果是nonblob -> blob，
// 要求nonblob必须是char/varchar类型，这种检查在转换矩阵中是没法反映出来的
// 这些检查都会在该方法中进行
static int check_can_cast(const common::ObObjType in_type,
const common::ObCollationType in_cs_type,
const common::ObObjType out_type,
const common::ObCollationType out_cs_type);
// 有些cast是什么事情都不用做的，例如int->bit，直接调用cast_eval_arg()计算参数的值即可
// CG阶段会使用该方法判断是否可以不用给cast表达式分配结果空间，直接指向参数的结果即可
static int is_trivial_cast(const common::ObObjType in_type,
const common::ObCollationType in_cs_type,
const common::ObObjType out_type,
const common::ObCollationType out_cs_type,
const common::ObCastMode &cast_mode,
bool &just_eval_arg);
// The same as:
// 功能同:
// EXPR_DEFINE_CAST_CTX(expr_ctx, cast_mode)
// EXPR_CAST_OBJ_V2(obj_type, obj, res_obj)
static int cast_obj(ObEvalCtx& ctx, common::ObIAllocator& alloc, const common::ObObjType& dst_type,
const common::ObCollationType& dst_cs_type, const common::ObObj& src_obj, common::ObObj& dst_obj);
static int cast_obj(ObEvalCtx &ctx, common::ObIAllocator &alloc,
const common::ObObjType &dst_type,
const common::ObCollationType &dst_cs_type,
const common::ObObj &src_obj,
common::ObObj &dst_obj);
static bool is_implicit_cast(const ObExpr& expr)
static bool is_implicit_cast(const ObExpr &expr)
{
bool bret = false;
if (T_FUN_SYS_CAST == expr.type_ && CM_IS_IMPLICIT_CAST(expr.extra_)) {
@ -98,7 +167,7 @@ public:
return bret;
}
static bool is_explicit_cast(const ObExpr& expr)
static bool is_explicit_cast(const ObExpr &expr)
{
bool bret = false;
if (T_FUN_SYS_CAST == expr.type_ && CM_IS_EXPLICIT_CAST(expr.extra_)) {
@ -110,42 +179,53 @@ public:
class ObDatumCaster {
public:
ObDatumCaster() : inited_(false), eval_ctx_(NULL), cast_expr_(NULL), extra_cast_expr_(NULL)
{}
~ObDatumCaster()
{}
ObDatumCaster()
: inited_(false),
eval_ctx_(NULL),
cast_expr_(NULL),
extra_cast_expr_(NULL) {}
~ObDatumCaster() {}
// init eval_ctx_/cast_expr_/extra_cast_expr_/frame. all mem comes from ObExecContext.
// frame layout:
// ObDatum | ObDatum | ObDynReserveBuf | res_buf | ObDynReserveBuf | res_buf
// res_buf_len is 128
int init(ObExecContext& ctx);
int init(ObExecContext &ctx);
// same with ObObjCaster::to_type().
// input is ObExpr, and output is ObDatum, it's better if input is also ObDatum.
// we will do this later if necessary.
int to_type(const ObDatumMeta& dst_type, const ObExpr& src_expr, const common::ObCastMode& cm, common::ObDatum*& res);
int to_type(const ObDatumMeta &dst_type,
const ObExpr &src_expr,
const common::ObCastMode &cm,
common::ObDatum *&res,
int64_t batch_idx = 0);
// for xxx -> enumset.
int to_type(const ObDatumMeta& dst_type, const common::ObIArray<common::ObString>& str_values, const ObExpr& src_expr,
const common::ObCastMode& cm, common::ObDatum*& res);
int to_type(const ObDatumMeta &dst_type,
const common::ObIArray<common::ObString> &str_values,
const ObExpr &src_expr,
const common::ObCastMode &cm,
common::ObDatum *&res,
int64_t batch_idx = 0);
int destroy();
private:
DISALLOW_COPY_AND_ASSIGN(ObDatumCaster);
// setup following data member of ObExpr:
// datum_meta_, obj_meta_, obj_datum_map_, eval_func_,
// args_, arg_cnt_, parents_, parent_cnt_, basic_funcs_.
int setup_cast_expr(
const ObDatumMeta& dst_type, const ObExpr& src_expr, const common::ObCastMode cm, ObExpr& cast_expr);
int setup_cast_expr(const ObDatumMeta &dst_type,
const ObExpr &src_expr,
const common::ObCastMode cm,
ObExpr &cast_expr);
bool inited_;
ObEvalCtx* eval_ctx_;
ObExpr* cast_expr_;
ObExpr* extra_cast_expr_;
ObEvalCtx *eval_ctx_;
ObExpr *cast_expr_;
ObExpr *extra_cast_expr_;
};
} // namespace sql
} // namespace oceanbase
} // namespace sql
} // namespace oceanbase
#endif // _OB_EXPR_DATUM_CAST_
#endif // _OB_EXPR_DATUM_CAST_