203 lines
7.0 KiB
C++
203 lines
7.0 KiB
C++
/**
|
|
* 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_ENG
|
|
|
|
#include "sql/engine/expr/ob_expr_date_diff.h"
|
|
#include "lib/timezone/ob_time_convert.h"
|
|
#include "lib/ob_name_def.h"
|
|
|
|
namespace oceanbase {
|
|
using namespace common;
|
|
namespace sql {
|
|
|
|
ObExprDateDiff::ObExprDateDiff(ObIAllocator& alloc)
|
|
: ObFuncExprOperator(alloc, T_FUN_SYS_DATE_DIFF, N_DATE_DIFF, 2, NOT_ROW_DIMENSION)
|
|
{}
|
|
|
|
ObExprDateDiff::~ObExprDateDiff()
|
|
{}
|
|
|
|
int ObExprDateDiff::calc_result2(ObObj& result, const ObObj& left, const ObObj& right, ObExprCtx& expr_ctx) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
if (OB_UNLIKELY(left.is_null() || right.is_null())) {
|
|
result.set_null();
|
|
} else {
|
|
TYPE_CHECK(left, ObDateType);
|
|
TYPE_CHECK(right, ObDateType);
|
|
int64_t date_left = left.get_date();
|
|
int64_t date_right = right.get_date();
|
|
if (OB_UNLIKELY(ObTimeConverter::ZERO_DATE == date_left || ObTimeConverter::ZERO_DATE == date_right)) {
|
|
result.set_null();
|
|
} else {
|
|
result.set_int(date_left - date_right);
|
|
}
|
|
}
|
|
UNUSED(expr_ctx);
|
|
return ret;
|
|
}
|
|
|
|
ObExprMonthsBetween::ObExprMonthsBetween(ObIAllocator& alloc)
|
|
: ObFuncExprOperator(alloc, T_FUN_SYS_MONTHS_BETWEEN, N_MONTHS_BETWEEN, 2, NOT_ROW_DIMENSION)
|
|
{}
|
|
|
|
ObExprMonthsBetween::~ObExprMonthsBetween()
|
|
{}
|
|
|
|
int ObExprMonthsBetween::calc_result_type2(
|
|
ObExprResType& type, ObExprResType& type1, ObExprResType& type2, ObExprTypeCtx& type_ctx) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
UNUSED(type_ctx);
|
|
type.set_number();
|
|
type.set_scale(ObAccuracy::DDL_DEFAULT_ACCURACY2[ORACLE_MODE][ObNumberType].get_scale());
|
|
type.set_precision(ObAccuracy::DDL_DEFAULT_ACCURACY2[ORACLE_MODE][ObNumberType].get_precision());
|
|
type1.set_calc_type(ObDateTimeType);
|
|
type2.set_calc_type(ObDateTimeType);
|
|
return ret;
|
|
}
|
|
|
|
int ObExprMonthsBetween::calc_result2(ObObj& result, const ObObj& obj1, const ObObj& obj2, ObExprCtx& expr_ctx) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
const int64_t DAYS_PER_MONTH_BASE = 31;
|
|
int64_t date_value1 = 0;
|
|
int64_t date_value2 = 0;
|
|
int64_t rest_utc_diff = 0;
|
|
int64_t months_diff = 0;
|
|
number::ObNumber res_num;
|
|
number::ObNumber res_int;
|
|
number::ObNumber res_frac;
|
|
number::ObNumber res_numerator;
|
|
number::ObNumber res_denominator;
|
|
|
|
if (OB_ISNULL(expr_ctx.calc_buf_)) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
LOG_WARN("allocator is null", K(ret));
|
|
} else if (obj1.is_null_oracle() || obj2.is_null_oracle()) {
|
|
result.set_null();
|
|
} else if (OB_FAIL(obj1.get_datetime(date_value1))) {
|
|
LOG_WARN("fail to get datetime", K(ret));
|
|
} else if (OB_FAIL(obj2.get_datetime(date_value2))) {
|
|
LOG_WARN("fail to get datetime", K(ret));
|
|
} else if (OB_FAIL(ObTimeConverter::calc_days_and_months_between_dates(
|
|
date_value1, date_value2, months_diff, rest_utc_diff))) {
|
|
LOG_WARN("fail to calc diff of days", K(ret));
|
|
} else if (OB_FAIL(res_int.from(months_diff, *expr_ctx.calc_buf_))) {
|
|
LOG_WARN("fail to from integer", K(ret));
|
|
} else if (OB_FAIL(res_numerator.from(rest_utc_diff, *expr_ctx.calc_buf_))) {
|
|
LOG_WARN("fail to from integer", K(ret));
|
|
} else if (OB_FAIL(res_denominator.from(DAYS_PER_MONTH_BASE * USECS_PER_DAY, *expr_ctx.calc_buf_))) {
|
|
LOG_WARN("fail to from integer", K(ret));
|
|
} else if (OB_FAIL(res_numerator.div(res_denominator, res_frac, *expr_ctx.calc_buf_))) {
|
|
LOG_WARN("fail to div", K(ret));
|
|
} else if (OB_FAIL(res_int.add(res_frac, res_num, *expr_ctx.calc_buf_))) {
|
|
LOG_WARN("fail to add", K(ret));
|
|
} else {
|
|
result.set_number(res_num);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int ObExprDateDiff::cg_expr(ObExprCGCtx& op_cg_ctx, const ObRawExpr& raw_expr, ObExpr& rt_expr) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
UNUSED(op_cg_ctx);
|
|
UNUSED(raw_expr);
|
|
rt_expr.eval_func_ = ObExprDateDiff::eval_date_diff;
|
|
|
|
return ret;
|
|
}
|
|
|
|
int ObExprDateDiff::eval_date_diff(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& res_datum)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ObDatum* left = NULL;
|
|
ObDatum* right = NULL;
|
|
int64_t date_left = 0;
|
|
int64_t date_right = 0;
|
|
if (OB_FAIL(expr.args_[0]->eval(ctx, left)) || OB_FAIL(expr.args_[1]->eval(ctx, right))) {
|
|
LOG_WARN("fail to eval conv", K(ret), K(expr));
|
|
} else if (left->is_null() || right->is_null()) {
|
|
res_datum.set_null();
|
|
} else if (FALSE_IT(date_left = left->get_date())) {
|
|
// do nothing
|
|
} else if (FALSE_IT(date_right = right->get_date())) {
|
|
// do nothing
|
|
} else if (OB_UNLIKELY(ObTimeConverter::ZERO_DATE == date_left || ObTimeConverter::ZERO_DATE == date_right)) {
|
|
res_datum.set_null();
|
|
} else {
|
|
res_datum.set_int(date_left - date_right);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int ObExprMonthsBetween::cg_expr(ObExprCGCtx& op_cg_ctx, const ObRawExpr& raw_expr, ObExpr& rt_expr) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
UNUSED(op_cg_ctx);
|
|
UNUSED(raw_expr);
|
|
rt_expr.eval_func_ = ObExprMonthsBetween::eval_months_between;
|
|
|
|
return ret;
|
|
}
|
|
|
|
int ObExprMonthsBetween::eval_months_between(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& res_datum)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ObDatum* left = NULL;
|
|
ObDatum* right = NULL;
|
|
const int64_t DAYS_PER_MONTH_BASE = 31;
|
|
int64_t date_value1 = 0;
|
|
int64_t date_value2 = 0;
|
|
int64_t rest_utc_diff = 0;
|
|
int64_t months_diff = 0;
|
|
number::ObNumber res_num;
|
|
number::ObNumber res_int;
|
|
number::ObNumber res_frac;
|
|
number::ObNumber res_numerator;
|
|
number::ObNumber res_denominator;
|
|
common::ObArenaAllocator& tmp_allocator = ctx.get_reset_tmp_alloc();
|
|
if (OB_FAIL(expr.args_[0]->eval(ctx, left)) || OB_FAIL(expr.args_[1]->eval(ctx, right))) {
|
|
LOG_WARN("fail to eval conv", K(ret), K(expr));
|
|
} else if (left->is_null() || right->is_null()) {
|
|
res_datum.set_null();
|
|
} else if (FALSE_IT(date_value1 = left->get_datetime())) {
|
|
// do nothing
|
|
} else if (FALSE_IT(date_value2 = right->get_datetime())) {
|
|
// do nothing
|
|
} else if (OB_FAIL(ObTimeConverter::calc_days_and_months_between_dates(
|
|
date_value1, date_value2, months_diff, rest_utc_diff))) {
|
|
LOG_WARN("fail to calc diff of days", K(ret));
|
|
} else if (OB_FAIL(res_int.from(months_diff, tmp_allocator))) {
|
|
LOG_WARN("fail to from integer", K(ret));
|
|
} else if (OB_FAIL(res_numerator.from(rest_utc_diff, tmp_allocator))) {
|
|
LOG_WARN("fail to from integer", K(ret));
|
|
} else if (OB_FAIL(res_denominator.from(DAYS_PER_MONTH_BASE * USECS_PER_DAY, tmp_allocator))) {
|
|
LOG_WARN("fail to from integer", K(ret));
|
|
} else if (OB_FAIL(res_numerator.div(res_denominator, res_frac, tmp_allocator))) {
|
|
LOG_WARN("fail to div", K(ret));
|
|
} else if (OB_FAIL(res_int.add(res_frac, res_num, tmp_allocator))) {
|
|
LOG_WARN("fail to add", K(ret));
|
|
} else {
|
|
res_datum.set_number(res_num);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
} // namespace sql
|
|
} // namespace oceanbase
|