init push
This commit is contained in:
oceanbase-admin
170
src/sql/engine/expr/ob_expr_maketime.cpp
Normal file
170
src/sql/engine/expr/ob_expr_maketime.cpp
Normal file
@ -0,0 +1,170 @@
|
||||
/**
|
||||
* 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 <math.h>
|
||||
#include "lib/timezone/ob_time_convert.h"
|
||||
#include "share/object/ob_obj_cast.h"
|
||||
#include "sql/engine/expr/ob_expr_maketime.h"
|
||||
#include "sql/engine/expr/ob_expr_util.h"
|
||||
#include "sql/engine/ob_exec_context.h"
|
||||
#include "sql/session/ob_sql_session_info.h"
|
||||
|
||||
using namespace oceanbase::common;
|
||||
using namespace oceanbase::sql;
|
||||
|
||||
namespace oceanbase {
|
||||
namespace sql {
|
||||
|
||||
ObExprMakeTime::ObExprMakeTime(ObIAllocator& alloc)
|
||||
: ObFuncExprOperator(alloc, T_FUN_SYS_MAKETIME, N_MAKETIME, 3, NOT_ROW_DIMENSION)
|
||||
{}
|
||||
|
||||
ObExprMakeTime::~ObExprMakeTime()
|
||||
{}
|
||||
|
||||
int ObExprMakeTime::calc_result3(ObObj& result, const ObObj& input_hour, const ObObj& input_minute,
|
||||
const ObObj& input_second, ObExprCtx& expr_ctx) const
|
||||
{
|
||||
int ret = OB_SUCCESS;
|
||||
// Null value handling refers to MySQL implemention Item_func_maketime::get_time
|
||||
// sql/item_timefunc.cc of version 5.6.
|
||||
if (OB_UNLIKELY(input_hour.is_null()) || OB_UNLIKELY(input_minute.is_null()) || OB_UNLIKELY(input_second.is_null())) {
|
||||
result.set_null();
|
||||
} else {
|
||||
ObTime ob_time(DT_TYPE_TIME);
|
||||
int32_t hour = input_hour.get_int32();
|
||||
int32_t minute = input_minute.get_int32();
|
||||
number::ObNumber sec(input_second.get_number());
|
||||
int64_t second_quotient = 0;
|
||||
int64_t second_remainder = 0;
|
||||
if (OB_FAIL(fetch_second_quotient(second_quotient, sec))) {
|
||||
LOG_WARN("failed to fetch data from ObNumber obj sec", K(ret));
|
||||
} else if (OB_UNLIKELY(minute < 0) || OB_UNLIKELY(minute > 59) || OB_UNLIKELY(second_quotient < 0) ||
|
||||
OB_UNLIKELY(second_quotient > 59) || OB_UNLIKELY(sec.is_negative())) {
|
||||
// This part of null handling also refers to MySQL Item_func_maketime::get_time()
|
||||
result.set_null();
|
||||
} else if (OB_FAIL(fetch_second_remainder(second_quotient, second_remainder, *(expr_ctx.calc_buf_), sec))) {
|
||||
LOG_WARN("failed to fetch data from ObNumber obj sec", K(ret));
|
||||
} else {
|
||||
if (hour < (-TIME_MAX_HOUR)) {
|
||||
hour = -(TIME_MAX_HOUR + 1);
|
||||
}
|
||||
ob_time.parts_[DT_HOUR] = hour < 0 ? -hour : hour;
|
||||
ob_time.parts_[DT_MIN] = minute;
|
||||
ob_time.parts_[DT_SEC] = (int32_t)second_quotient; // value < 60, no impact
|
||||
ob_time.parts_[DT_USEC] = (int32_t)second_remainder; // value < 999999, no impact
|
||||
if (OB_FAIL(ObTimeConverter::validate_time(ob_time))) {
|
||||
LOG_WARN("time value is invalid or out of range", K(ret));
|
||||
} else {
|
||||
int64_t value = ObTimeConverter::ob_time_to_time(ob_time);
|
||||
int overflow = ObTimeConverter::time_overflow_trunc(value);
|
||||
OX(result.set_time(hour < 0 ? -(value) : value));
|
||||
}
|
||||
}
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
int ObExprMakeTime::fetch_second_quotient(int64_t& quotient, number::ObNumber& sec)
|
||||
{
|
||||
int ret = OB_SUCCESS;
|
||||
if (OB_FAIL(sec.extract_valid_int64_with_trunc(quotient))) {
|
||||
LOG_WARN("fail to get second quotient", K(ret));
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
int ObExprMakeTime::fetch_second_remainder(
|
||||
const int64_t quotient, int64_t& remainder, ObIAllocator& alloc, number::ObNumber& sec)
|
||||
{
|
||||
int ret = OB_SUCCESS;
|
||||
number::ObNumber second_num_quotient;
|
||||
number::ObNumber second_num_remainder;
|
||||
number::ObNumber second_num_remainder_transform;
|
||||
number::ObNumber const_1M;
|
||||
OZ(second_num_quotient.from(quotient, alloc));
|
||||
OZ(sec.sub(second_num_quotient, second_num_remainder, alloc));
|
||||
// According to Mysql User Manual(from 5.x to 8.0), the second scale is 6(range 0 - 999999).
|
||||
// https://dev.mysql.com/doc/refman/8.0/en/time.html
|
||||
OZ(const_1M.from((int64_t)1000000, alloc));
|
||||
// Transformation example: 0.1234564 * 1000,000 => 123456.4
|
||||
OZ(second_num_remainder.mul(const_1M, second_num_remainder_transform, alloc));
|
||||
OZ(second_num_remainder_transform.round(common::DEFAULT_SCALE_FOR_INTEGER));
|
||||
OZ(second_num_remainder_transform.extract_valid_int64_with_trunc(remainder));
|
||||
return ret;
|
||||
}
|
||||
|
||||
int ObExprMakeTime::eval_maketime(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& result)
|
||||
{
|
||||
int ret = OB_SUCCESS;
|
||||
ObDatum* hour = NULL;
|
||||
ObDatum* minute = NULL;
|
||||
ObDatum* second = NULL;
|
||||
// Fetch input argument, hour, minute, second
|
||||
if (OB_FAIL(expr.eval_param_value(ctx, hour, minute, second))) {
|
||||
LOG_WARN("unexpected input value ", K(ret));
|
||||
} else if (OB_UNLIKELY(hour->is_null()) || OB_UNLIKELY(minute->is_null()) || OB_UNLIKELY(second->is_null())) {
|
||||
// Function maketime returns null when it matched above conditions.
|
||||
// Above logic refers to MySQL implemention Item_func_maketime::get_time
|
||||
// sql/item_timefunc.cc of version 5.6.
|
||||
result.set_null();
|
||||
} else {
|
||||
ObTime ob_time(DT_TYPE_TIME);
|
||||
int32_t h = hour->get_int32();
|
||||
int32_t min = minute->get_int32();
|
||||
number::ObNumber sec(second->get_number());
|
||||
ObArenaAllocator& alloc = ctx.get_reset_tmp_alloc();
|
||||
int64_t second_quotient = 0;
|
||||
int64_t second_remainder = 0;
|
||||
if (OB_FAIL(fetch_second_quotient(second_quotient, sec))) {
|
||||
LOG_WARN("failed to fetch data from ObNumber obj sec", K(ret));
|
||||
} else if (OB_UNLIKELY(min < 0) || OB_UNLIKELY(min > 59) || OB_UNLIKELY(second_quotient < 0) ||
|
||||
OB_UNLIKELY(second_quotient > 59) || OB_UNLIKELY(sec.is_negative())) {
|
||||
result.set_null();
|
||||
} else if (OB_FAIL(fetch_second_remainder(second_quotient, second_remainder, alloc, sec))) {
|
||||
LOG_WARN("failed to fetch data from ObNumber obj sec", K(ret));
|
||||
} else {
|
||||
if (h < (-TIME_MAX_HOUR)) {
|
||||
h = -(TIME_MAX_HOUR + 1);
|
||||
}
|
||||
ob_time.parts_[DT_HOUR] = h < 0 ? -h : h;
|
||||
ob_time.parts_[DT_MIN] = min;
|
||||
ob_time.parts_[DT_SEC] = (int32_t)second_quotient; // value < 60, no impact
|
||||
ob_time.parts_[DT_USEC] = (int32_t)second_remainder; // value < 999999, no impact
|
||||
if (OB_FAIL(ObTimeConverter::validate_time(ob_time))) {
|
||||
LOG_WARN("time value is invalid or out of range", K(ret));
|
||||
} else {
|
||||
int64_t value = ObTimeConverter::ob_time_to_time(ob_time);
|
||||
int overflow = ObTimeConverter::time_overflow_trunc(value);
|
||||
OX(result.set_time(h < 0 ? -(value) : value));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int ObExprMakeTime::cg_expr(ObExprCGCtx& expr_cg_ctx, const ObRawExpr& raw_expr, ObExpr& rt_expr) const
|
||||
{
|
||||
int ret = OB_SUCCESS;
|
||||
UNUSED(expr_cg_ctx);
|
||||
UNUSED(raw_expr);
|
||||
CK(3 == rt_expr.arg_cnt_);
|
||||
if (OB_SUCC(ret)) {
|
||||
rt_expr.eval_func_ = eval_maketime;
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
} // namespace sql
|
||||
} // namespace oceanbase
|
||||
Reference in New Issue
Block a user