hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
674e87b495c286e7520619b537f3b656c5135706
oceanbase/src/sql/engine/expr/ob_expr_day_of_func.cpp
obdev 3081f5616f fix not set expr_datum when warn_on_fail
2023-02-24 14:13:19 +00:00

661 lines
25 KiB
C++
Raw Blame History

/**
* 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.
*/
#define USING_LOG_PREFIX SQL_EXE
#include "ob_expr_day_of_func.h"
#include "lib/timezone/ob_time_convert.h"
#include "share/object/ob_obj_cast.h"
#include "sql/session/ob_sql_session_info.h"
#include "sql/engine/ob_exec_context.h"
#include "share/ob_time_utility2.h"
#include "sql/engine/expr/ob_datum_cast.h"
using namespace oceanbase::share;
using namespace oceanbase::sql;
namespace oceanbase
{
namespace sql
{
ObExprDayOfMonth::ObExprDayOfMonth(ObIAllocator &alloc)
: ObExprTimeBase(alloc, DT_MDAY, T_FUN_SYS_DAY_OF_MONTH, N_DAY_OF_MONTH) {};
ObExprDayOfMonth::~ObExprDayOfMonth() {}
int ObExprDayOfMonth::calc_dayofmonth(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum)
{
return ObExprTimeBase::calc(expr, ctx, expr_datum, DT_MDAY, true, true);
}
ObExprDay::ObExprDay(ObIAllocator &alloc)
: ObExprTimeBase(alloc, DT_MDAY, T_FUN_SYS_DAY, N_DAY) {};
ObExprDay::~ObExprDay() {}
ObExprDayOfWeek::ObExprDayOfWeek(ObIAllocator &alloc)
: ObExprTimeBase(alloc, DT_WDAY, T_FUN_SYS_DAY_OF_WEEK, N_DAY_OF_WEEK) {};
ObExprDayOfWeek::~ObExprDayOfWeek() {}
int ObExprDayOfWeek::calc_dayofweek(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum)
{
int ret = OB_SUCCESS;
if (OB_FAIL(ObExprTimeBase::calc(expr, ctx, expr_datum, DT_WDAY, true))) {
LOG_WARN("calc day of week failed", K(ret));
} else if (!expr_datum.is_null()) {
expr_datum.set_int32(expr_datum.get_int32() % 7 + 1);
}
return ret;
}
ObExprDayOfYear::ObExprDayOfYear(ObIAllocator &alloc)
: ObExprTimeBase(alloc, DT_YDAY, T_FUN_SYS_DAY_OF_YEAR, N_DAY_OF_YEAR) {};
ObExprDayOfYear::~ObExprDayOfYear() { }
int ObExprDayOfYear::calc_dayofyear(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum)
{
return ObExprTimeBase::calc(expr, ctx, expr_datum, DT_YDAY, true);
}
ObExprToSeconds::ObExprToSeconds(ObIAllocator &alloc)
: ObFuncExprOperator(alloc, T_FUN_SYS_TO_SECONDS, N_TO_SECONDS, 1, NOT_ROW_DIMENSION) {};
ObExprToSeconds::~ObExprToSeconds() {}
int ObExprToSeconds::cg_expr(ObExprCGCtx &op_cg_ctx,
const ObRawExpr &raw_expr,
ObExpr &rt_expr) const
{
UNUSED(op_cg_ctx);
UNUSED(raw_expr);
int ret = OB_SUCCESS;
if (rt_expr.arg_cnt_ != 1) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("to_seconds expr should have one param", K(ret), K(rt_expr.arg_cnt_));
} else if (OB_ISNULL(rt_expr.args_) || OB_ISNULL(rt_expr.args_[0])) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("children of to_seconds expr is null", K(ret), K(rt_expr.args_));
} else {
rt_expr.eval_func_ = ObExprToSeconds::calc_toseconds;
}
return ret;
}
int ObExprToSeconds::calc_toseconds(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum)
{
int ret = OB_SUCCESS;
ObDatum *param_datum = NULL;
const ObSQLSessionInfo *session = NULL;
if (OB_ISNULL(session = ctx.exec_ctx_.get_my_session())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("session is null", K(ret));
} else if (OB_FAIL(expr.args_[0]->eval(ctx, param_datum))) {
LOG_WARN("eval param value failed");
} else if (OB_UNLIKELY(param_datum->is_null())) {
expr_datum.set_null();
} else {
ObTime ot;
ObDateSqlMode date_sql_mode;
date_sql_mode.init(session->get_sql_mode());
if (OB_FAIL(ob_datum_to_ob_time_with_date(*param_datum, expr.args_[0]->datum_meta_.type_,
get_timezone_info(session), ot,
get_cur_time(ctx.exec_ctx_.get_physical_plan_ctx()),
false, date_sql_mode,
expr.args_[0]->obj_meta_.has_lob_header()))) {
LOG_WARN("cast to ob time failed", K(ret));
uint64_t cast_mode = 0;
ObSQLUtils::get_default_cast_mode(session->get_stmt_type(), session, cast_mode);
if (CM_IS_WARN_ON_FAIL(cast_mode)) {
ret = OB_SUCCESS;
expr_datum.set_null();
} else { /* do nothing */ }
} else {
int64_t days = ObTimeConverter::ob_time_to_date(ot) + DAYS_FROM_ZERO_TO_BASE;
int64_t seconds = days * 60 * 60 * 24L + ot.parts_[DT_HOUR] * 60 * 60L
+ ot.parts_[DT_MIN] * 60L + ot.parts_[DT_SEC];
if (seconds < 0) {
expr_datum.set_null();
} else {
expr_datum.set_int(seconds);
}
}
}
return ret;
}
ObExprSecToTime::ObExprSecToTime(ObIAllocator &alloc)
: ObFuncExprOperator(alloc, T_FUN_SYS_SEC_TO_TIME, N_SEC_TO_TIME, 1, NOT_ROW_DIMENSION) {};
ObExprSecToTime::~ObExprSecToTime() {}
int ObExprSecToTime::cg_expr(ObExprCGCtx &op_cg_ctx,
const ObRawExpr &raw_expr,
ObExpr &rt_expr) const
{
UNUSED(op_cg_ctx);
UNUSED(raw_expr);
int ret = OB_SUCCESS;
if (rt_expr.arg_cnt_ != 1) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("sectotime expr should have one param", K(ret), K(rt_expr.arg_cnt_));
} else if (OB_ISNULL(rt_expr.args_) || OB_ISNULL(rt_expr.args_[0])) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("children of sectotime expr is null", K(ret), K(rt_expr.args_));
} else {
CK(ObNumberType == rt_expr.args_[0]->datum_meta_.type_);
rt_expr.eval_func_ = ObExprSecToTime::calc_sectotime;
}
return ret;
}
int ObExprSecToTime::calc_sectotime(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum)
{
int ret = OB_SUCCESS;
ObDatum *param_datum = NULL;
const ObSQLSessionInfo *session = NULL;
if (OB_ISNULL(session = ctx.exec_ctx_.get_my_session())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("session is null", K(ret));
} else if (OB_FAIL(expr.args_[0]->eval(ctx, param_datum))) {
LOG_WARN("eval param value failed");
} else if (OB_UNLIKELY(param_datum->is_null())) {
expr_datum.set_null();
} else {
number::ObNumber num_usec;
number::ObNumber num_sec;
number::ObNumber million;
int64_t int_usec = 0;
char local_buff[number::ObNumber::MAX_BYTE_LEN * 3];
ObDataBuffer local_alloc(local_buff, number::ObNumber::MAX_BYTE_LEN * 3);
if (OB_FAIL(num_sec.from(param_datum->get_number(), local_alloc))) {
LOG_WARN("failed to create number", K(ret), K(number::ObNumber(param_datum->get_number())));
} else if (OB_FAIL(million.from(static_cast<int64_t>(1000000), local_alloc))) {
LOG_WARN("failed to create number million", K(ret));
} else if (OB_FAIL(num_sec.mul(million, num_usec, local_alloc))) {
LOG_WARN("failed to mul number", K(ret), K(num_sec), K(million));
} else if (OB_FAIL(num_usec.round(0))) {
LOG_WARN("failed to round number", K(ret), K(num_usec));
} else if (!num_usec.is_valid_int64(int_usec)) {
int_usec = num_usec.is_negative() ? -INT64_MAX : INT64_MAX;
}
if (OB_SUCC(ret)) {
if (num_usec.is_negative()) {
int_usec = -int_usec;
}
if (OB_FAIL(ObTimeConverter::time_overflow_trunc(int_usec))) {
uint64_t cast_mode = 0;
ObSQLUtils::get_default_cast_mode(session->get_stmt_type(), session, cast_mode);
if (CM_IS_WARN_ON_FAIL(cast_mode)) {
ret = OB_SUCCESS;
expr_datum.set_null();
} else {
LOG_WARN("time value is out of range", K(ret), K(int_usec));
}
}
}
if (OB_SUCC(ret)) {
if (num_usec.is_negative()) {
int_usec = -int_usec;
}
expr_datum.set_time(int_usec);
}
}
return ret;
}
ObExprTimeToSec::ObExprTimeToSec(ObIAllocator &alloc)
: ObFuncExprOperator(alloc, T_FUN_SYS_TIME_TO_SEC, N_TIME_TO_SEC, 1, NOT_ROW_DIMENSION) {};
ObExprTimeToSec::~ObExprTimeToSec() {}
int ObExprTimeToSec::cg_expr(ObExprCGCtx &op_cg_ctx,
const ObRawExpr &raw_expr,
ObExpr &rt_expr) const
{
UNUSED(op_cg_ctx);
UNUSED(raw_expr);
int ret = OB_SUCCESS;
if (rt_expr.arg_cnt_ != 1) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("timetosec expr should have one param", K(ret), K(rt_expr.arg_cnt_));
} else if (OB_ISNULL(rt_expr.args_) || OB_ISNULL(rt_expr.args_[0])) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("children of timetosec expr is null", K(ret), K(rt_expr.args_));
} else {
CK(ObTimeType == rt_expr.args_[0]->datum_meta_.type_);
rt_expr.eval_func_ = ObExprTimeToSec::calc_timetosec;
}
return ret;
}
int ObExprTimeToSec::calc_timetosec(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum)
{
int ret = OB_SUCCESS;
ObDatum *param_datum = NULL;
const ObSQLSessionInfo *session = NULL;
if (OB_ISNULL(session = ctx.exec_ctx_.get_my_session())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("session is null", K(ret));
} else if (OB_FAIL(expr.args_[0]->eval(ctx, param_datum))) {
LOG_WARN("eval param value failed");
} else if (OB_UNLIKELY(param_datum->is_null())) {
expr_datum.set_null();
} else {
int64_t t_val = param_datum->get_time();
expr_datum.set_int(USEC_TO_SEC(t_val));
}
return ret;
}
ObExprSubAddtime::ObExprSubAddtime(ObIAllocator &alloc, ObExprOperatorType type, const char *name, int32_t param_num, int32_t dimension)
: ObFuncExprOperator(alloc, type, name, 2, NOT_ROW_DIMENSION)
{}
ObExprSubtime::ObExprSubtime(ObIAllocator &alloc)
: ObExprSubAddtime(alloc, T_FUN_SYS_SUBTIME, N_SUB_TIME, 2, NOT_ROW_DIMENSION)
{}
ObExprAddtime::ObExprAddtime(ObIAllocator &alloc)
: ObExprSubAddtime(alloc, T_FUN_SYS_ADDTIME, N_ADD_TIME, 2, NOT_ROW_DIMENSION)
{}
int ObExprSubAddtime::calc_result_type2(ObExprResType &type,
ObExprResType &date_arg,
ObExprResType &time_arg,
ObExprTypeCtx &type_ctx) const
{
int ret = OB_SUCCESS;
ObScale scale1 = date_arg.get_scale();
ObScale scale2 = time_arg.get_scale();
if (OB_UNLIKELY(ObNullTC == date_arg.get_type_class())) {
type.set_datetime();
type.set_scale(MAX_SCALE_FOR_TEMPORAL);
type.set_precision(static_cast<ObPrecision>(DATETIME_MIN_LENGTH + MAX_SCALE_FOR_TEMPORAL));
} else if (ObTimeTC == date_arg.get_type_class()) {
type.set_time();
type.set_scale(MAX(scale1, scale2));
type.set_precision(static_cast<ObPrecision>(TIME_MIN_LENGTH + type.get_scale()));
} else if (ObDateTimeType == date_arg.get_type() || ObTimestampType == date_arg.get_type()) {
type.set_datetime();
type.set_scale(MAX(scale1, scale2));
type.set_precision(static_cast<ObPrecision>(DATETIME_MIN_LENGTH + type.get_scale()));
} else {
type.set_varchar();
type.set_length(DATETIME_MAX_LENGTH);
type.set_scale(-1);
ret = aggregate_charsets_for_string_result(type, &date_arg, 1, type_ctx.get_coll_type());
}
if (OB_SUCC(ret)) {
// date_arg无法设置calc_type的原因是，date_arg类型是varchar时，设置calc_type为time和datetime都不合适
// 如果设置为time，那么当字符串中包含日期时，转换后日期信息直接丢失了
// 如果设置为datetime，虽然保留了日期信息，但是当字符串不包含日期时，mysql会将字符串最右边视为秒，
// 即'123'的含义是1分23秒，但是OB cast成datetime类型会把字符串右边视为日，即‘123’的含义是1月23日
if (ob_is_enumset_tc(date_arg.get_type())) {
date_arg.set_calc_type(ObVarcharType);
}
if (ob_is_enumset_tc(time_arg.get_type())) {
time_arg.set_calc_type(ObVarcharType);
} else {
type_ctx.set_cast_mode(type_ctx.get_cast_mode() | CM_NULL_ON_WARN);
time_arg.set_calc_type(ObTimeType);
}
}
return ret;
}
int ObExprSubAddtime::calc_result2(common::ObObj &result,
const common::ObObj &date_arg,
const common::ObObj &time_arg,
common::ObExprCtx &expr_ctx) const
{
int ret = OB_SUCCESS;
int64_t t_val1 = 0;
int64_t t_val2 = 0;
ObTime ot1(DT_TYPE_TIME);
bool param_with_date = true;
ObObjType result_type = get_result_type().get_type();
const ObSQLSessionInfo *session = NULL;
EXPR_DEFINE_CAST_CTX(expr_ctx, CM_NONE);
if (OB_UNLIKELY(date_arg.is_null() || time_arg.is_null())) {
result.set_null();
} else if (OB_ISNULL(expr_ctx.calc_buf_) || OB_ISNULL(session = expr_ctx.my_session_)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("allocator or session is null", K(ret), K(expr_ctx.calc_buf_));
} else {
if (ObVarcharType == result_type) {
if (OB_FAIL(ob_obj_to_ob_time_without_date(date_arg,
get_timezone_info(session), ot1))) {
LOG_WARN("obj to ob time without date failed", K(ret), K(date_arg));
} else {
if (0 == ot1.parts_[DT_YEAR] && 0 == ot1.parts_[DT_MON] && 0 == ot1.parts_[DT_MDAY]) {
param_with_date = false;
}
}
}
if (OB_SUCC(ret)) {
if (ObTimeType == time_arg.get_type()) {
t_val2 = time_arg.get_time();
} else {
ObTime ot2(DT_TYPE_TIME);
if (OB_FAIL(ob_obj_to_ob_time_without_date(time_arg, get_timezone_info(session), ot2))) {
LOG_WARN("cast the second param failed", K(ret));
} else {
t_val2 = ObTimeConverter::ob_time_to_time(ot2);
}
}
if (OB_FAIL(ret)) {
} else if (ObTimeType == result_type || !param_with_date) { // a#, subaddtime(时间1, 时间2)
if (ObVarcharType == result_type) {
t_val1 = ObTimeConverter::ob_time_to_time(ot1);
if (IS_NEG_TIME(ot1.mode_)) {
t_val1 = -t_val1;
}
} else {
TYPE_CHECK(date_arg, ObTimeType);
t_val1 = date_arg.get_time();
}
if (OB_SUCC(ret)) {
int64_t int_usec = (get_type() == T_FUN_SYS_SUBTIME) ? t_val1 - t_val2 : t_val1 + t_val2;
if (OB_FAIL(ObTimeConverter::time_overflow_trunc(int_usec))) {
if (CM_IS_WARN_ON_FAIL(cast_ctx.cast_mode_)) {
ret = OB_SUCCESS;
result.set_time(int_usec);
} else {
LOG_WARN("time value is out of range", K(ret), K(int_usec));
}
} else {
result.set_time(int_usec);
}
}
} else { // b#, subaddtime(日期时间1, 时间2)
const ObTimeZoneInfo *tz_info = NULL;
if (OB_ISNULL(tz_info = get_timezone_info(expr_ctx.my_session_))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("tz info is null", K(ret));
} else if (ObVarcharType == result_type) {
ObTimeConvertCtx cvrt_ctx(tz_info, false);
if (OB_FAIL(ObTimeConverter::ob_time_to_datetime(ot1, cvrt_ctx, t_val1))) {
LOG_WARN("ob_time_to_datetime failed", K(ret));
}
} else {
if (ObDateTimeType != date_arg.get_type() && ObTimestampType != date_arg.get_type()) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid date arg type", K(ret));
} else {
int64_t offset = ObTimestampType == date_arg.get_type() ? tz_info->get_offset() : 0;
t_val1 = date_arg.get_datetime() + offset * USECS_PER_SEC;
}
}
if (OB_SUCC(ret)) {
int64_t int_usec = (get_type() == T_FUN_SYS_SUBTIME) ? t_val1 - t_val2 : t_val1 + t_val2;
if (ObTimeConverter::is_valid_datetime(int_usec)) {
result.set_datetime(int_usec);
} else {
result.set_null();
}
}
}
if (OB_SUCC(ret) && ObVarcharType == result_type) {
if (OB_FAIL(ObObjCaster::to_type(ObVarcharType, cast_ctx, result, result))) {
LOG_WARN("failed to cast object to ObVarcharType ", K(result), K(ret));
}
}
}
if (OB_FAIL(ret)) {
uint64_t cast_mode = 0;
ObSQLUtils::get_default_cast_mode(session->get_stmt_type(), session, cast_mode);
if (CM_IS_WARN_ON_FAIL(cast_mode) && OB_ALLOCATE_MEMORY_FAILED != ret) {
ret = OB_SUCCESS;
result.set_null();
}
}
}
return ret;
}
int ObExprSubAddtime::cg_expr(ObExprCGCtx &op_cg_ctx,
const ObRawExpr &raw_expr,
ObExpr &rt_expr) const
{
UNUSED(op_cg_ctx);
UNUSED(raw_expr);
int ret = OB_SUCCESS;
if (rt_expr.arg_cnt_ != 2) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("subaddtime expr should have two params", K(ret), K(rt_expr.arg_cnt_));
} else if (OB_ISNULL(rt_expr.args_) || OB_ISNULL(rt_expr.args_[0])
|| OB_ISNULL(rt_expr.args_[1])) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("children of subaddtime expr is null", K(ret), K(rt_expr.args_));
} else {
const ObObjType result_type = rt_expr.datum_meta_.type_;
const ObObjType param1_type = rt_expr.args_[0]->datum_meta_.type_;
const ObObjType param2_type = rt_expr.args_[1]->datum_meta_.type_;
if (ObTimeType != param2_type && ObVarcharType != param2_type) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("second param of subaddtime expr is not time type", K(ret), K(param2_type));
} else {
switch (result_type) {
case ObDateTimeType :
if (ObNullType != param1_type && ObDateTimeType != param1_type
&& ObTimestampType != param1_type) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid type of first argument", K(ret), K(result_type), K(param1_type));
} else {
rt_expr.eval_func_ = ObExprSubAddtime::subaddtime_datetime;
}
break;
case ObTimeType :
if (ObTimeType != param1_type) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid type of first argument", K(ret), K(result_type), K(param1_type));
} else {
//这里虽然参数类型、结果类型是TimeType，和上面不同，但是get/set_time/datetime的含义是相同的，
//两个计算函数的本质都是取第一个参数都int64_t值，与第二个参数相减，然后把结果放入expr_datum中
rt_expr.eval_func_ = ObExprSubAddtime::subaddtime_datetime;
}
break;
case ObVarcharType :
rt_expr.eval_func_ = ObExprSubAddtime::subaddtime_varchar;
break;
default :
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid expr type", K(rt_expr.datum_meta_.type_));
}
}
}
return ret;
}
int ObExprSubAddtime::subaddtime_common(const ObExpr &expr,
ObEvalCtx &ctx,
ObDatum &expr_datum,
bool &null_res,
ObDatum *&date_arg,
ObDatum *&time_arg,
int64_t &time_val)
{
int ret = OB_SUCCESS;
null_res = false;
const ObSQLSessionInfo *session = NULL;
if (OB_ISNULL(session = ctx.exec_ctx_.get_my_session())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("session is null", K(ret));
} else if (OB_FAIL(expr.args_[0]->eval(ctx, date_arg))) {
LOG_WARN("eval the first param value failed");
} else if (OB_FAIL(expr.args_[1]->eval(ctx, time_arg))) {
LOG_WARN("eval the second param value failed");
} else if (OB_UNLIKELY(date_arg->is_null() || time_arg->is_null())) {
expr_datum.set_null();
null_res = true;
} else {
ObTime ot2(DT_TYPE_TIME);
if (ObTimeType == expr.args_[1]->datum_meta_.type_) {
time_val = time_arg->get_time();
} else if (OB_FAIL(ob_datum_to_ob_time_without_date(*time_arg, expr.args_[1]->datum_meta_.type_,
get_timezone_info(ctx.exec_ctx_.get_my_session()), ot2,
expr.args_[1]->obj_meta_.has_lob_header()))) {
LOG_WARN("cast the second param failed", K(ret));
expr_datum.set_null();
null_res = true;
uint64_t cast_mode = 0;
ObSQLUtils::get_default_cast_mode(session->get_stmt_type(), session, cast_mode);
if (CM_IS_WARN_ON_FAIL(cast_mode)) {
ret = OB_SUCCESS;
}
} else {
time_val = ObTimeConverter::ob_time_to_time(ot2);
}
}
return ret;
}
int ObExprSubAddtime::subaddtime_datetime(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum)
{
int ret = OB_SUCCESS;
ObDatum *date_arg = NULL;
ObDatum *time_arg = NULL;
int64_t t_val1 = 0;
int64_t t_val2 = 0;
bool null_res = false;
const ObSQLSessionInfo *session = ctx.exec_ctx_.get_my_session();
const ObTimeZoneInfo *tz_info = NULL;
if (OB_FAIL(subaddtime_common(expr, ctx, expr_datum, null_res, date_arg, time_arg, t_val2))) {
LOG_WARN("calc subaddtime failed", K(ret));
} else if (OB_ISNULL(tz_info = get_timezone_info(session))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("timezone info of session is null", K(ret));
} else if (!null_res) {
int64_t offset = ObTimestampType == expr.args_[0]->datum_meta_.type_
? tz_info->get_offset() : 0;
t_val1 = date_arg->get_datetime() + offset * USECS_PER_SEC;
int64_t int_usec = (expr.type_ == T_FUN_SYS_SUBTIME) ? t_val1 - t_val2 : t_val1 + t_val2;
if (ObTimeConverter::is_valid_datetime(int_usec)) {
expr_datum.set_datetime(int_usec);
} else {
expr_datum.set_null();
}
}
return ret;
}
int ObExprSubAddtime::subaddtime_varchar(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum)
{
int ret = OB_SUCCESS;
ObDatum *date_arg = NULL;
ObDatum *time_arg = NULL;
int64_t t_val1 = 0;
int64_t t_val2 = 0;
bool null_res = false;
const ObSQLSessionInfo *session = ctx.exec_ctx_.get_my_session();
const ObTimeZoneInfo *tz_info = NULL;
if (OB_FAIL(subaddtime_common(expr, ctx, expr_datum, null_res, date_arg, time_arg, t_val2))) {
LOG_WARN("calc subaddtime failed", K(ret));
} else if (OB_ISNULL(tz_info = get_timezone_info(session))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("timezone info of session is null", K(ret));
} else if (!null_res) {
ObTime ot1(DT_TYPE_TIME);
if (OB_FAIL(ob_datum_to_ob_time_without_date(*date_arg, expr.args_[0]->datum_meta_.type_,
tz_info, ot1, expr.args_[0]->obj_meta_.has_lob_header()))) {
LOG_WARN("cast the first param failed", K(ret));
expr_datum.set_null();
} else {
bool param_with_date = true;
if (0 == ot1.parts_[DT_YEAR] && 0 == ot1.parts_[DT_MON] && 0 == ot1.parts_[DT_MDAY]) {
param_with_date = false;
}
if (param_with_date) {
ObTimeConvertCtx cvrt_ctx(tz_info, false);
if (OB_FAIL(ObTimeConverter::ob_time_to_datetime(ot1, cvrt_ctx, t_val1))) {
LOG_WARN("ob_time_to_datetime failed", K(ret));
}
} else {
t_val1 = ObTimeConverter::ob_time_to_time(ot1);
if (IS_NEG_TIME(ot1.mode_)) {
t_val1 = -t_val1;
}
}
if (OB_SUCC(ret)) {
int64_t int_usec = (expr.type_ == T_FUN_SYS_SUBTIME) ? t_val1 - t_val2 : t_val1 + t_val2;
const int64_t datetime_buf_len = DATETIME_MAX_LENGTH + 1;
char *buf = expr.get_str_res_mem(ctx, datetime_buf_len);
int64_t pos = 0;
//兼容mysql行为，有毫秒则显示6位小数，否则不显示
ObString format;
if (OB_ISNULL(buf)) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_WARN("allocate memory failed", K(ret));
} else if (param_with_date && OB_FAIL(ObTimeConverter::datetime_to_str(int_usec, NULL,
format, -1, buf,
datetime_buf_len, pos, true))) {
LOG_WARN("datetime to str failed", K(ret));
} else if (!param_with_date && OB_FAIL(ObTimeConverter::time_to_str(int_usec, -1, buf,
datetime_buf_len, pos, true))) {
LOG_WARN("time to str failed", K(ret));
} else {
expr_datum.ptr_ = buf;
expr_datum.pack_ = static_cast<uint32_t>(pos);
}
}
}
if (OB_FAIL(ret) && OB_ALLOCATE_MEMORY_FAILED != ret) {
uint64_t cast_mode = 0;
ObSQLUtils::get_default_cast_mode(session->get_stmt_type(), session, cast_mode);
if (CM_IS_WARN_ON_FAIL(cast_mode) ) {
ret = OB_SUCCESS;
expr_datum.set_null();
}
}
}
return ret;
}
ObExprDayName::ObExprDayName(ObIAllocator &alloc)
: ObExprTimeBase(alloc, DT_WDAY, T_FUN_SYS_DAY_NAME, N_DAY_NAME) {};
ObExprDayName::~ObExprDayName() {}
int ObExprDayName::calc_dayname(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum)
{
int ret = OB_SUCCESS;
if (OB_FAIL(ObExprTimeBase::calc(expr, ctx, expr_datum, DT_WDAY, true))) {
LOG_WARN("dayname calc day of dayweek failed", K(ret));
}
return ret;
}
int ObExprDayName::calc_result_type1(ObExprResType &type,
ObExprResType &type1,
common::ObExprTypeCtx &type_ctx) const
{
ObCollationType cs_type = type_ctx.get_coll_type();
type.set_varchar();
type.set_full_length(DAYNAME_MAX_LENGTH, type1.get_length_semantics());
type.set_collation_type(cs_type);
type.set_collation_level(CS_LEVEL_IMPLICIT);
common::ObObjTypeClass tc1 = ob_obj_type_class(type1.get_type());
if (ob_is_enumset_tc(type1.get_type())) {
type1.set_calc_type(common::ObVarcharType);
type1.set_collation_type(cs_type);
type1.set_collation_level(CS_LEVEL_IMPLICIT);
} else if ((common::ObFloatTC == tc1) || (common::ObDoubleTC == tc1)) {
type1.set_calc_type(common::ObIntType);
}
return OB_SUCCESS;
}
} //namespace sql
} //namespace oceanbase
Reference in New Issue View Git Blame Copy Permalink