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

101
src/sql/engine/expr/ob_expr_convert_tz.cpp
View File

@ -18,18 +18,21 @@
#include "sql/session/ob_sql_session_info.h"
#include "sql/engine/ob_exec_context.h"
namespace oceanbase {
using namespace common;
namespace sql {
ObExprConvertTZ::ObExprConvertTZ(common::ObIAllocator &alloc)
: ObFuncExprOperator(alloc, T_FUN_SYS_CONVERT_TZ, "convert_TZ", 3, NOT_ROW_DIMENSION)
{}
int ObExprConvertTZ::calc_result_type3(ObExprResType &type, ObExprResType &input1, ObExprResType &input2,
ObExprResType &input3, common::ObExprTypeCtx &type_ctx) const
namespace oceanbase
{
using namespace common;
namespace sql
{
ObExprConvertTZ::ObExprConvertTZ(common::ObIAllocator &alloc):
ObFuncExprOperator(alloc, T_FUN_SYS_CONVERT_TZ, "convert_TZ", 3, NOT_ROW_DIMENSION){}
int ObExprConvertTZ::calc_result_type3(ObExprResType &type,
ObExprResType &input1,
ObExprResType &input2,
ObExprResType &input3,
common::ObExprTypeCtx &type_ctx) const
{
UNUSED(type_ctx);
int ret = OB_SUCCESS;
const ObSQLSessionInfo *session = NULL;
if (OB_ISNULL(session = type_ctx.get_session())) {
@ -49,34 +52,19 @@ int ObExprConvertTZ::calc_result_type3(ObExprResType &type, ObExprResType &input
return ret;
}
int ObExprConvertTZ::calc_result3(common::ObObj &result, const common::ObObj &input1, const common::ObObj &input2,
const common::ObObj &input3, common::ObExprCtx &expr_ctx) const
{
int ret = OB_SUCCESS;
if (OB_UNLIKELY(input1.is_null() || input2.is_null() || input3.is_null())) {
result.set_null();
} else {
int64_t timestamp_data = input1.get_datetime();
if (OB_FAIL(
calc_convert_tz(timestamp_data, input2.get_string(), input3.get_string(), expr_ctx.my_session_, result))) {
LOG_WARN("convert time zone failed", K(ret));
}
}
return ret;
}
template <typename T>
int ObExprConvertTZ::calc_convert_tz(int64_t timestamp_data,
const ObString &tz_str_s, // source time zone (input2)
const ObString &tz_str_d, // destination time zone (input3)
ObSQLSessionInfo *session, T &result)
const ObString &tz_str_s,//source time zone (input2)
const ObString &tz_str_d,//destination time zone (input3)
ObSQLSessionInfo *session,
T &result)
{
int ret = OB_SUCCESS;
int32_t offset_s = 0;
int32_t offset_d = 0;
if (OB_FAIL(ObExprConvertTZ::parse_string(timestamp_data, tz_str_s, session, false))) {
if(OB_FAIL(ObExprConvertTZ::parse_string(timestamp_data, tz_str_s,session, false))){
LOG_WARN("source time zone parse failed", K(ret), K(tz_str_s));
} else if (OB_FAIL(ObExprConvertTZ::parse_string(timestamp_data, tz_str_d, session, true))) {
} else if(OB_FAIL(ObExprConvertTZ::parse_string(timestamp_data, tz_str_d,session, true))){
LOG_WARN("source time zone parse failed", K(ret), K(tz_str_d));
}
if (OB_FAIL(ret)) {
@ -93,27 +81,28 @@ int ObExprConvertTZ::calc_convert_tz(int64_t timestamp_data,
return ret;
}
int ObExprConvertTZ::parse_string(
int64_t &timestamp_data, const ObString &tz_str, ObSQLSessionInfo *session, const bool input_utc_time)
int ObExprConvertTZ::parse_string(int64_t &timestamp_data, const ObString &tz_str,
ObSQLSessionInfo *session, const bool input_utc_time)
{
int ret = OB_SUCCESS;
int ret_more = 0;
int32_t offset = 0;
if (OB_FAIL(ObTimeConverter::str_to_offset(
tz_str, offset, ret_more, false /* oracle_mode */, true /* need_check_valid */))) {
if (OB_FAIL(ObTimeConverter::str_to_offset(tz_str, offset, ret_more,
false /* oracle_mode */, true /* need_check_valid */)))
{
LOG_WARN("direct str_to_offset failed");
if (OB_LIKELY(OB_ERR_UNKNOWN_TIME_ZONE == ret)) {
if (OB_LIKELY(OB_ERR_UNKNOWN_TIME_ZONE == ret)){
const ObTimeZoneInfo *tz_info = NULL;
ObTimeZoneInfoPos *target_tz_pos = NULL;
ObTimeZoneInfoPos target_tz_pos;
if (OB_ISNULL(tz_info = TZ_INFO(session))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("tz info is null", K(ret), K(session));
} else if (OB_FAIL(find_time_zone_pos(tz_str, *tz_info, target_tz_pos))) {
} else if (OB_FAIL(find_time_zone_pos(tz_str, *tz_info, target_tz_pos))){
LOG_WARN("find time zone position failed", K(ret), K(ret_more));
if (OB_ERR_UNKNOWN_TIME_ZONE == ret && OB_SUCCESS != ret_more) {
ret = ret_more;
}
} else if (OB_FAIL(calc(timestamp_data, *target_tz_pos, input_utc_time))) {
} else if (OB_FAIL(calc(timestamp_data, target_tz_pos, input_utc_time))) {
LOG_WARN("calc failed", K(ret), K(timestamp_data));
}
} else {
@ -126,26 +115,25 @@ int ObExprConvertTZ::parse_string(
return ret;
}
int ObExprConvertTZ::find_time_zone_pos(
const ObString &tz_name, const ObTimeZoneInfo &tz_info, ObTimeZoneInfoPos *&tz_info_pos)
int ObExprConvertTZ::find_time_zone_pos(const ObString &tz_name,
const ObTimeZoneInfo &tz_info,
ObTimeZoneInfoPos &tz_info_pos)
{
int ret = OB_SUCCESS;
tz_info_pos = NULL;
ObTZInfoMap *tz_info_map = NULL;
if (OB_ISNULL(tz_info_map = const_cast<ObTZInfoMap *>(tz_info.get_tz_info_map()))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("tz_info_map is NULL", K(ret));
} else if (OB_FAIL(tz_info_map->get_tz_info_by_name(tz_name, tz_info_pos))) {
LOG_WARN("fail to get_tz_info_by_name", K(tz_name), K(ret));
tz_info_map->id_map_.revert(tz_info_pos);
tz_info_pos = NULL;
} else {
tz_info_pos->set_error_on_overlap_time(tz_info.is_error_on_overlap_time());
tz_info_pos.set_error_on_overlap_time(tz_info.is_error_on_overlap_time());
}
return ret;
}
int ObExprConvertTZ::calc(int64_t &timestamp_data, const ObTimeZoneInfoPos &tz_info_pos, const bool input_utc_time)
int ObExprConvertTZ::calc(int64_t &timestamp_data, const ObTimeZoneInfoPos &tz_info_pos,
const bool input_utc_time)
{
int ret = OB_SUCCESS;
const int64_t input_value = timestamp_data;
@ -162,13 +150,15 @@ int ObExprConvertTZ::calc(int64_t &timestamp_data, const ObTimeZoneInfoPos &tz_i
return ret;
}
int ObExprConvertTZ::cg_expr(ObExprCGCtx &expr_cg_ctx, const ObRawExpr &raw_expr, ObExpr &expr) const
int ObExprConvertTZ::cg_expr(ObExprCGCtx &expr_cg_ctx, const ObRawExpr &raw_expr,
ObExpr &expr) const
{
int ret = OB_SUCCESS;
UNUSED(expr_cg_ctx);
UNUSED(raw_expr);
CK(3 == expr.arg_cnt_);
CK(OB_NOT_NULL(expr.args_) && OB_NOT_NULL(expr.args_[0]) && OB_NOT_NULL(expr.args_[1]) && OB_NOT_NULL(expr.args_[2]));
CK(3 == expr.arg_cnt_ );
CK(OB_NOT_NULL(expr.args_) && OB_NOT_NULL(expr.args_[0])
&& OB_NOT_NULL(expr.args_[1]) && OB_NOT_NULL(expr.args_[2]));
CK(ObDateTimeType == expr.args_[0]->datum_meta_.type_);
CK(ObDateTimeType == expr.datum_meta_.type_);
CK(ObVarcharType == expr.args_[1]->datum_meta_.type_);
@ -189,16 +179,13 @@ int ObExprConvertTZ::eval_convert_tz(const ObExpr &expr, ObEvalCtx &ctx, ObDatum
res.set_null();
} else {
int64_t timestamp_data = timestamp->get_datetime();
if (OB_FAIL(calc_convert_tz(timestamp_data,
time_zone_s->get_string(),
time_zone_d->get_string(),
ctx.exec_ctx_.get_my_session(),
res))) {
if (OB_FAIL(calc_convert_tz(timestamp_data, time_zone_s->get_string(), time_zone_d->get_string(),
ctx.exec_ctx_.get_my_session(), res))) {
LOG_WARN("calc convert tz zone failed", K(ret));
}
}
return ret;
}
} // namespace sql
} // namespace oceanbase
}
}