hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

init push

Browse Source
This commit is contained in:
oceanbase-admin
2021-05-31 22:56:52 +08:00
commit cea7de1475
7020 changed files with 5689869 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

724
src/sql/engine/expr/ob_expr_div.cpp Normal file
View File

@ -0,0 +1,724 @@
/**
* 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_div.h"
#include "lib/oblog/ob_log.h"
#include "sql/engine/expr/ob_expr_result_type_util.h"
#include "sql/session/ob_sql_session_info.h"
#include "sql/engine/ob_exec_context.h"
namespace oceanbase {
namespace sql {
using namespace common;
using namespace share;
using namespace common::number;
ObExprDiv::ObExprDiv(ObIAllocator& alloc, ObExprOperatorType type)
: ObArithExprOperator(alloc, type, N_DIV, 2, NOT_ROW_DIMENSION, ObExprResultTypeUtil::get_div_result_type,
ObExprResultTypeUtil::get_div_calc_type, div_funcs_)
{
param_lazy_eval_ = lib::is_oracle_mode();
}
#define ROUND_UP(scale) static_cast<ObScale>(((scale) + DIV_CALC_SCALE - 1) / DIV_CALC_SCALE * DIV_CALC_SCALE)
int ObExprDiv::calc_result_type2(
ObExprResType& type, ObExprResType& type1, ObExprResType& type2, ObExprTypeCtx& type_ctx) const
{
int ret = OB_SUCCESS;
int64_t div_precision_increment = 0;
CK(!OB_UNLIKELY(type_ctx.get_div_precision_increment() < 0));
div_precision_increment = type_ctx.get_div_precision_increment();
OC((ObArithExprOperator::calc_result_type2)(type, type1, type2, type_ctx));
if (OB_SUCC(ret)) {
const ObObjTypeClass result_tc = type.get_type_class();
if (ObNumberTC == result_tc || ObDoubleTC == result_tc || ObFloatTC == result_tc) {
if (is_oracle_mode()) {
type.set_scale(ORA_NUMBER_SCALE_UNKNOWN_YET);
type.set_precision(PRECISION_UNKNOWN_YET);
} else {
ObScale scale1 = static_cast<ObScale>(MAX(type1.get_scale(), 0));
ObScale scale2 = static_cast<ObScale>(MAX(type2.get_scale(), 0));
// res scale.
ObScale res_scale = static_cast<ObScale>(scale1 + div_precision_increment);
res_scale = static_cast<ObScale>(MIN(res_scale, OB_MAX_DECIMAL_SCALE));
type.set_scale(res_scale);
ObPrecision precision1 = static_cast<ObPrecision>(MAX(type1.get_precision(), 0));
if (OB_UNLIKELY(PRECISION_UNKNOWN_YET == type1.get_precision()) ||
OB_UNLIKELY(PRECISION_UNKNOWN_YET == type2.get_precision())) {
type.set_precision(PRECISION_UNKNOWN_YET);
} else {
type.set_precision(static_cast<ObPrecision>(precision1 + MAX(scale2, div_precision_increment)));
}
// calc scale.
if (OB_UNLIKELY(SCALE_UNKNOWN_YET == type1.get_scale()) ||
OB_UNLIKELY(SCALE_UNKNOWN_YET == type2.get_scale())) {
type.set_scale(SCALE_UNKNOWN_YET);
} else {
ObScale calc_scale = static_cast<ObScale>(
MAX(ROUND_UP(scale1) + ROUND_UP(scale2), ROUND_UP(scale1 + scale2 + div_precision_increment)));
type.set_calc_scale(calc_scale);
}
LOG_DEBUG("div calc_result_type2",
K(type.get_calc_scale()),
K(scale1),
K(scale2),
"new_scale1",
ROUND_UP(scale1),
"new_scale2",
ROUND_UP(scale2),
K(div_precision_increment));
}
} else if (ObIntervalTC == type.get_type_class()) {
type.set_scale(ObAccuracy::MAX_ACCURACY2[ORACLE_MODE][type.get_type()].get_scale());
type.set_precision(ObAccuracy::MAX_ACCURACY2[ORACLE_MODE][type.get_type()].get_precision());
}
type.unset_result_flag(OB_MYSQL_NOT_NULL_FLAG); // divided by zero
}
return ret;
}
int ObExprDiv::calc_result2(ObObj& result, const ObObj& left, const ObObj& right, ObExprCtx& expr_ctx) const
{
int ret = OB_SUCCESS;
ObScale calc_scale = result_type_.get_calc_scale();
LOG_DEBUG("div calc_result2", K(calc_scale), "scale", result_type_.get_scale());
ObObjType calc_type = result_type_.get_calc_type();
if (lib::is_oracle_mode()) {
if (OB_FAIL(param_eval(expr_ctx, left, 0))) {
LOG_WARN("parameter evaluate failed", K(ret));
} else if (left.is_null_oracle()) {
result.set_null();
} else if (OB_FAIL(param_eval(expr_ctx, right, 1))) {
LOG_WARN("parameter evaluate failed", K(ret));
} else if (ObIntervalTC == left.get_type_class() && ObNumberTC == right.get_type_class()) {
ret = div_interval(result, left, right, expr_ctx.calc_buf_, 0);
} else {
// When the div_xxx method is called later,
// these methods can obtain the compile-time result through result.get_type() type,
// in order to reasonably set the actual calculation result
result.set_type(get_result_type().get_type());
ret = ObArithExprOperator::calc_(
result, left, right, expr_ctx, calc_scale, calc_type, T_OP_AGG_DIV == type_ ? avg_div_funcs_ : div_funcs_);
}
} else {
result.set_type(get_result_type().get_type());
ret = ObArithExprOperator::calc_(
result, left, right, expr_ctx, calc_scale, calc_type, T_OP_AGG_DIV == type_ ? avg_div_funcs_ : div_funcs_);
}
return ret;
}
int ObExprDiv::calc(ObObj& res, const ObObj& left, const ObObj& right, ObIAllocator* allocator, ObScale calc_scale)
{
ObCalcTypeFunc calc_type_func = ObExprResultTypeUtil::get_div_result_type;
return ObArithExprOperator::calc(res, left, right, allocator, calc_scale, calc_type_func, div_funcs_);
}
int ObExprDiv::calc_for_avg(ObObj& res, const ObObj& left, const ObObj& right, ObExprCtx& expr_ctx, ObScale res_scale)
{
int ret = OB_SUCCESS;
ObCalcTypeFunc calc_type_func = ObExprResultTypeUtil::get_div_result_type;
ObScale calc_scale = static_cast<ObScale>(res_scale < 0 ? DIV_MAX_CALC_SCALE : (res_scale + 8) / 9 * 9);
if (OB_FAIL(ObArithExprOperator::calc(res, left, right, expr_ctx, calc_scale, calc_type_func, avg_div_funcs_))) {
LOG_WARN("calc failed", K(ret), K(left), K(right), K(calc_scale));
} else if (ob_is_number_tc(res.get_type()) && res_scale >= 0) {
number::ObNumber res_nmb = res.get_number();
if (OB_FAIL(res_nmb.round(res_scale))) {
LOG_WARN("round failed", K(ret), K(res_nmb), K(res_scale));
} else {
res.set_number(res_nmb);
}
}
return ret;
}
int ObExprDiv::calc_for_avg(
ObDatum& result, const ObDatum& sum, const int64_t count, ObEvalCtx& expr_ctx, const common::ObObjType type)
{
int ret = OB_SUCCESS;
UNUSED(result);
UNUSED(sum);
UNUSED(count);
UNUSED(expr_ctx);
UNUSED(type);
ret = OB_ERR_UNEXPECTED;
LOG_WARN("it should not arrive here", K(ret), K(type), K(lbt()));
return ret;
}
ObArithFunc ObExprDiv::div_funcs_[ObMaxTC] = {
NULL,
NULL,
NULL,
ObExprDiv::div_float,
ObExprDiv::div_double,
ObExprDiv::div_number,
NULL, // datetime
NULL, // date
NULL, // time
NULL, // year
NULL, // varchar
NULL, // extend
NULL, // unknown
NULL, // text
NULL, // bit
NULL, // enumset
NULL, // enumsetInner
NULL, // otimestamp
NULL, // raw
ObExprDiv::div_interval, // interval
};
ObArithFunc ObExprDiv::avg_div_funcs_[ObMaxTC] = {
NULL,
NULL,
NULL,
ObExprDiv::div_float,
ObExprDiv::div_double_no_overflow,
ObExprDiv::div_number,
NULL, // datetime
NULL, // date
NULL, // time
NULL, // year
NULL, // varchar
NULL, // extend
NULL, // unknown
NULL, // text
NULL, // bit
NULL, // enumset
NULL, // enumsetInner
NULL, // otimestamp
NULL, // raw
ObExprDiv::div_interval, // interval
};
int ObExprDiv::div_float(ObObj& res, const ObObj& left, const ObObj& right, ObIAllocator* allocator, ObScale scale)
{
int ret = OB_SUCCESS;
if (OB_UNLIKELY(!lib::is_oracle_mode())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("only oracle mode arrive here", K(ret), K(left), K(right));
} else if (OB_UNLIKELY(left.get_type_class() != right.get_type_class())) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("Invalid types", K(ret), K(left), K(right));
} else if (fabsf(right.get_float()) == 0.0) {
ret = OB_ERR_DIVISOR_IS_ZERO;
LOG_WARN("division by zero not allowed now", K(res), K(left), K(right));
} else {
res.set_float(left.get_float() / right.get_float());
if (OB_UNLIKELY(is_float_out_of_range(res.get_float()))) {
ret = OB_OPERATE_OVERFLOW;
char expr_str[OB_MAX_TWO_OPERATOR_EXPR_LENGTH];
int64_t pos = 0;
databuff_printf(
expr_str, OB_MAX_TWO_OPERATOR_EXPR_LENGTH, pos, "'(%e / %e)'", left.get_float(), right.get_float());
LOG_USER_ERROR(OB_OPERATE_OVERFLOW, "BINARY_FLOAT", expr_str);
LOG_WARN("double out of range", K(res), K(left), K(right), K(res));
}
LOG_DEBUG("succ to div float", K(res.get_float()), K(left.get_float()), K(right.get_float()));
}
UNUSED(allocator);
UNUSED(scale);
return ret;
}
int ObExprDiv::div_double(ObObj& res, const ObObj& left, const ObObj& right, ObIAllocator* allocator, ObScale scale)
{
int ret = OB_SUCCESS;
if (OB_UNLIKELY(left.get_type_class() != right.get_type_class())) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("Invalid types", K(ret), K(left), K(right));
} else if (fabs(right.get_double()) == 0.0) {
if (is_oracle_mode()) {
ret = OB_ERR_DIVISOR_IS_ZERO;
LOG_WARN("division by zero not allowed now", K(res), K(left), K(right));
} else {
res.set_null();
}
} else {
res.set_double(left.get_double() / right.get_double());
if (OB_UNLIKELY(is_double_out_of_range(res.get_double()))) {
ret = OB_OPERATE_OVERFLOW;
char expr_str[OB_MAX_TWO_OPERATOR_EXPR_LENGTH];
int64_t pos = 0;
databuff_printf(
expr_str, OB_MAX_TWO_OPERATOR_EXPR_LENGTH, pos, "'(%e / %e)'", left.get_double(), right.get_double());
LOG_USER_ERROR(OB_OPERATE_OVERFLOW, "DOUBLE", expr_str);
LOG_WARN("double out of range", K(res), K(left), K(right), K(res));
res.set_null();
}
LOG_DEBUG("succ to div double", K(res.get_double()), K(left.get_double()), K(right.get_double()));
}
UNUSED(allocator);
UNUSED(scale);
return ret;
}
int ObExprDiv::div_double_no_overflow(ObObj& res, const ObObj& left, const ObObj& right, ObIAllocator*, ObScale)
{
int ret = OB_SUCCESS;
if (OB_UNLIKELY(left.get_type_class() != right.get_type_class())) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("Invalid types", K(ret), K(left), K(right));
} else if (lib::is_mysql_mode() && (fabs(right.get_double()) == 0.0)) {
res.set_null();
} else {
res.set_double(left.get_double() / right.get_double());
LOG_DEBUG("succ to div double", K(res.get_double()), K(left.get_double()), K(right.get_double()));
}
return ret;
}
int ObExprDiv::div_number(
ObObj& res, const ObObj& left, const ObObj& right, ObIAllocator* allocator, ObScale calc_scale)
{
int ret = OB_SUCCESS;
number::ObNumber res_nmb;
if (OB_UNLIKELY(NULL == allocator)) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_ERROR("allocator is null", K(ret));
} else if (OB_UNLIKELY(right.get_number().is_zero())) {
if (is_oracle_mode()) {
ret = OB_ERR_DIVISOR_IS_ZERO;
LOG_WARN("divisor is equal to zero on oracle mode", K(ret), K(right));
} else {
res.set_null();
}
} else if (OB_FAIL(left.get_number().div_v3(right.get_number(), res_nmb, *allocator))) {
LOG_WARN("failed to div numbers", K(ret), K(left), K(right));
} else {
if (calc_scale >= 0) {
// calc_scale is calc_scale ,not res_scale.
// trunc with calc_scale and round with res_scale
if (OB_FAIL(res_nmb.trunc(calc_scale))) {
LOG_WARN("failed to trunc result number", K(ret), K(res_nmb), K(calc_scale));
}
}
if (OB_SUCC(ret)) {
if (ObUNumberType == res.get_type()) {
res.set_unumber(res_nmb);
} else {
res.set_number(res_nmb);
}
}
}
return ret;
}
int ObExprDiv::div_interval(
ObObj& res, const ObObj& left, const ObObj& right, ObIAllocator* allocator, ObScale calc_scale)
{
int ret = OB_SUCCESS;
number::ObNumber res_number;
number::ObNumber left_number;
if (OB_UNLIKELY(left.get_type_class() != ObIntervalTC) || OB_UNLIKELY(right.get_type_class() != ObNumberTC)) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("Invalid types", K(ret), K(left), K(right));
} else if (OB_UNLIKELY(NULL == allocator)) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_ERROR("allocator is null", K(ret));
} else if (OB_UNLIKELY(right.get_number().is_zero())) {
ret = OB_ERR_DIVISOR_IS_ZERO;
LOG_WARN("divisor is equal to zero on oracle mode", K(ret), K(right));
} else if (left.is_interval_ym()) {
int64_t result_nmonth = 0;
if (OB_FAIL(left_number.from(left.get_interval_ym().get_nmonth(), *allocator))) {
LOG_WARN("failed to convert to number", K(ret), K(left));
} else if (OB_FAIL(left_number.div_v3(right.get_number(), res_number, *allocator))) {
LOG_WARN("failed to do mul", K(ret), K(left), K(right));
} else if (OB_FAIL(res_number.trunc(0))) {
LOG_WARN("failed to do trunc", K(ret), K(res_number));
} else if (OB_UNLIKELY(!res_number.is_valid_int64(result_nmonth))) {
ret = OB_INVALID_NUMERIC;
LOG_WARN("failed to get int64_t from number", K(ret), K(res_number));
} else if (OB_FAIL(ObIntervalYMValue(result_nmonth).validate())) {
LOG_WARN("invalid interval ym result", K(ret), K(result_nmonth));
} else {
res.set_interval_ym(ObIntervalYMValue(result_nmonth));
}
} else {
int64_t result_nsecond = 0;
int64_t result_fs = 0;
number::ObNumber nsecond;
number::ObNumber fsecond;
number::ObNumber power10;
number::ObNumber midresult;
number::ObNumber nvalue;
ObIntervalDSValue interval_res;
static_assert(number::ObNumber::BASE == ObIntervalDSValue::MAX_FS_VALUE,
"the div caculation between interval day to second and number is base on this constrain");
if (OB_FAIL(nsecond.from(left.get_interval_ds().get_nsecond(), *allocator))) {
LOG_WARN("failed to convert interval to number", K(ret));
} else if (OB_FAIL(fsecond.from(static_cast<int64_t>(left.get_interval_ds().get_fs()), *allocator))) {
LOG_WARN("failed to convert interval to number", K(ret));
} else if (OB_FAIL(power10.from(static_cast<int64_t>(ObIntervalDSValue::MAX_FS_VALUE), *allocator))) {
LOG_WARN("failed to round number", K(ret));
} else if (OB_FAIL(nsecond.mul_v3(power10, nvalue, *allocator))) {
LOG_WARN("fail to div fs", K(ret));
} else if (OB_FAIL(nvalue.add_v3(fsecond, left_number, *allocator))) {
LOG_WARN("failed to add number", K(ret));
} else if (OB_FAIL(left_number.div_v3(right.get_number(), midresult, *allocator))) {
LOG_WARN("failed to do mul", K(ret));
} else if (OB_FAIL(midresult.div_v3(power10, res_number, *allocator))) {
LOG_WARN("failed to do mul", K(ret));
} else if (OB_FAIL(res_number.round(MAX_SCALE_FOR_ORACLE_TEMPORAL))) {
LOG_WARN("failed to round number", K(res_number));
} else if (OB_UNLIKELY(!res_number.is_int_parts_valid_int64(result_nsecond, result_fs))) {
ret = OB_INVALID_NUMERIC;
LOG_WARN("invalid date format", K(ret));
} else {
if (result_nsecond < 0) {
result_fs = -result_fs;
}
interval_res = ObIntervalDSValue(result_nsecond, static_cast<int32_t>(result_fs));
if (OB_FAIL(interval_res.validate())) {
LOG_WARN("invalid interval result", K(ret), K(interval_res));
} else {
res.set_interval_ds(interval_res);
}
}
}
res.set_scale(ObAccuracy::MAX_ACCURACY2[ORACLE_MODE][res.get_type()].get_scale());
UNUSED(calc_scale);
return ret;
return ret;
}
const ObScale ObExprDiv::DIV_CALC_SCALE = 9;
const ObScale ObExprDiv::DIV_MAX_CALC_SCALE = 100;
int ObExprDiv::div_float(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& datum)
{
int ret = OB_SUCCESS;
ObDatum* left = NULL;
ObDatum* right = NULL;
bool is_finish = false;
if (OB_UNLIKELY(!lib::is_oracle_mode())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("only oracle mode arrive here", K(ret));
} else if (OB_FAIL(get_arith_operand(expr, ctx, left, right, datum, is_finish))) {
LOG_WARN("get_arith_operand failed", K(ret));
} else if (is_finish) {
// do nothing
} else {
ret = div_float(datum, *left, *right, ctx);
}
return ret;
}
int ObExprDiv::div_double(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& datum)
{
int ret = OB_SUCCESS;
ObDatum* left = NULL;
ObDatum* right = NULL;
bool is_finish = false;
if (OB_FAIL(get_arith_operand(expr, ctx, left, right, datum, is_finish))) {
LOG_WARN("get_arith_operand failed", K(ret));
} else if (is_finish) {
// do nothing
} else {
const double left_d = left->get_double();
const double right_d = right->get_double();
if (is_mysql_mode() && (fabs(right_d) == 0.0)) {
datum.set_null();
} else {
const double result_d = left_d / right_d;
if (OB_UNLIKELY(is_double_out_of_range(result_d)) && T_OP_AGG_DIV != expr.type_) {
ret = OB_OPERATE_OVERFLOW;
char expr_str[OB_MAX_TWO_OPERATOR_EXPR_LENGTH];
int64_t pos = 0;
databuff_printf(expr_str, OB_MAX_TWO_OPERATOR_EXPR_LENGTH, pos, "'(%e / %e)'", left_d, right_d);
LOG_USER_ERROR(OB_OPERATE_OVERFLOW, is_oracle_mode() ? "BINARY_DOUBLE" : "DOUBLE", expr_str);
LOG_WARN("double out of range", K(ret), "left", left_d, "right", right_d);
} else {
datum.set_double(result_d);
}
LOG_DEBUG("succ to div double", K(left_d), K(right_d), K(result_d));
}
}
return ret;
}
int ObExprDiv::div_number(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& datum)
{
int ret = OB_SUCCESS;
ObDatum* left = NULL;
ObDatum* right = NULL;
bool is_finish = false;
if (OB_FAIL(get_arith_operand(expr, ctx, left, right, datum, is_finish))) {
LOG_WARN("get_arith_operand failed", K(ret));
} else if (is_finish) {
// do nothing
} else {
ret = div_number(datum, *left, *right, ctx, expr);
}
return ret;
}
int ObExprDiv::div_intervalym_number(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& datum)
{
int ret = OB_SUCCESS;
ObDatum* left = NULL;
ObDatum* right = NULL;
bool is_finish = false;
if (OB_ISNULL(ctx.exec_ctx_.get_my_session())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("my_session_ is null", K(ret));
} else if (OB_FAIL(get_arith_operand(expr, ctx, left, right, datum, is_finish))) {
LOG_WARN("get_arith_operand failed", K(ret));
} else if (is_finish) {
// do nothing
} else if (right->get_number().is_zero()) {
if (is_oracle_mode()) {
ret = OB_ERR_DIVISOR_IS_ZERO;
LOG_WARN("divisor is equal to zero on oracle mode", K(ret));
} else {
datum.set_null();
}
} else {
number::ObNumber lnum;
number::ObNumber rnum(right->get_number());
char local_buff[ObNumber::MAX_BYTE_LEN * 2];
ObDataBuffer local_alloc(local_buff, ObNumber::MAX_BYTE_LEN * 2);
ObNumber result_number;
int64_t result_nmonth = 0;
if (OB_FAIL(lnum.from(left->get_interval_nmonth(), local_alloc))) {
LOG_WARN("failed to convert to number", K(ret), K(left));
} else if (OB_FAIL(lnum.div_v3(rnum, result_number, local_alloc))) {
LOG_WARN("failed to do mul", K(ret), K(left), K(right));
} else if (OB_FAIL(result_number.trunc(0))) {
LOG_WARN("failed to do trunc", K(ret), K(result_number));
} else if (OB_UNLIKELY(!result_number.is_valid_int64(result_nmonth))) {
ret = OB_INVALID_NUMERIC;
LOG_WARN("failed to get int64_t from number", K(ret), K(result_number));
} else {
ObIntervalYMValue value = ObIntervalYMValue(result_nmonth);
if (OB_FAIL(value.validate())) {
LOG_WARN("invalid interval ym result", K(ret), K(value));
} else {
datum.set_interval_nmonth(result_nmonth);
}
}
}
return ret;
}
int ObExprDiv::div_intervalds_number(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& datum)
{
int ret = OB_SUCCESS;
ObDatum* left = NULL;
ObDatum* right = NULL;
bool is_finish = false;
if (OB_ISNULL(ctx.exec_ctx_.get_my_session())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("my_session_ is null", K(ret));
} else if (OB_FAIL(get_arith_operand(expr, ctx, left, right, datum, is_finish))) {
LOG_WARN("get_arith_operand failed", K(ret));
} else if (is_finish) {
// do nothing
} else if (right->get_number().is_zero()) {
if (is_oracle_mode()) {
ret = OB_ERR_DIVISOR_IS_ZERO;
LOG_WARN("divisor is equal to zero on oracle mode", K(ret));
} else {
datum.set_null();
}
} else {
char local_buff[ObNumber::MAX_BYTE_LEN * 7];
ObDataBuffer local_alloc(local_buff, ObNumber::MAX_BYTE_LEN * 7);
number::ObNumber lnum;
ObNumber result_number;
int64_t result_nsecond = 0;
int64_t result_fs = 0;
number::ObNumber nsecond;
number::ObNumber fsecond;
number::ObNumber power10;
number::ObNumber midresult;
number::ObNumber nvalue;
if (OB_FAIL(nsecond.from(left->get_interval_ds().get_nsecond(), local_alloc))) {
LOG_WARN("failed to convert interval to number", K(ret));
} else if (OB_FAIL(fsecond.from(static_cast<int64_t>(left->get_interval_ds().get_fs()), local_alloc))) {
LOG_WARN("failed to convert interval to number", K(ret));
} else if (OB_FAIL(power10.from(static_cast<int64_t>(ObIntervalDSValue::MAX_FS_VALUE), local_alloc))) {
LOG_WARN("failed to round number", K(ret));
} else if (OB_FAIL(nsecond.mul_v3(power10, nvalue, local_alloc))) {
LOG_WARN("fail to div fs", K(ret));
} else if (OB_FAIL(nvalue.add_v3(fsecond, lnum, local_alloc))) {
LOG_WARN("failed to add number", K(ret));
} else if (OB_FAIL(lnum.div_v3(right->get_number(), midresult, local_alloc))) {
LOG_WARN("failed to do mul", K(ret));
} else if (OB_FAIL(midresult.div_v3(power10, result_number, local_alloc))) {
LOG_WARN("failed to do mul", K(ret));
} else if (OB_FAIL(result_number.round(MAX_SCALE_FOR_ORACLE_TEMPORAL))) {
LOG_WARN("failed to round number", K(result_number));
} else if (OB_UNLIKELY(!result_number.is_int_parts_valid_int64(result_nsecond, result_fs))) {
ret = OB_INVALID_NUMERIC;
LOG_WARN("invalid date format", K(ret));
} else {
if (result_nsecond < 0) {
result_fs = -result_fs;
}
ObIntervalDSValue value = ObIntervalDSValue(result_nsecond, static_cast<int32_t>(result_fs));
if (OB_FAIL(value.validate())) {
LOG_WARN("invalid interval result", K(ret), K(value));
} else {
datum.set_interval_ds(value);
}
}
}
return ret;
}
int ObExprDiv::cg_expr(ObExprCGCtx& op_cg_ctx, const ObRawExpr& raw_expr, ObExpr& rt_expr) const
{
int ret = OB_SUCCESS;
UNUSED(raw_expr);
UNUSED(op_cg_ctx);
OB_ASSERT(2 == rt_expr.arg_cnt_);
OB_ASSERT(NULL != rt_expr.args_);
OB_ASSERT(NULL != rt_expr.args_[0]);
OB_ASSERT(NULL != rt_expr.args_[1]);
const common::ObObjType left = rt_expr.args_[0]->datum_meta_.type_;
const common::ObObjType right = rt_expr.args_[1]->datum_meta_.type_;
OB_ASSERT(left == input_types_[0].get_calc_type());
OB_ASSERT(right == input_types_[1].get_calc_type());
rt_expr.inner_functions_ = NULL;
LOG_DEBUG("arrive here cg_expr", K(ret), K(raw_expr), K(rt_expr));
rt_expr.div_calc_scale_ = raw_expr.get_result_type().get_calc_scale();
switch (rt_expr.datum_meta_.type_) {
case ObFloatType: {
rt_expr.eval_func_ = ObExprDiv::div_float;
break;
}
case ObDoubleType: {
rt_expr.eval_func_ = ObExprDiv::div_double;
break;
}
case ObUNumberType:
case ObNumberType: {
rt_expr.eval_func_ = ObExprDiv::div_number;
break;
}
case ObIntervalYMType: {
rt_expr.eval_func_ = ObExprDiv::div_intervalym_number;
break;
}
case ObIntervalDSType: {
rt_expr.eval_func_ = ObExprDiv::div_intervalds_number;
break;
}
default: {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("unexpected result type", K(ret), K(rt_expr.datum_meta_.type_));
}
}
return ret;
}
int ObExprDiv::div_float(ObDatum& result, const ObDatum& left, const ObDatum& right, ObEvalCtx& ctx)
{
int ret = OB_SUCCESS;
UNUSED(ctx);
const float left_f = left.get_float();
const float right_f = right.get_float();
if (fabsf(right_f) == 0.0) {
ret = OB_ERR_DIVISOR_IS_ZERO;
LOG_WARN("division by zero not allowed now", K(ret), K(right_f));
} else {
const float result_f = left_f / right_f;
if (OB_UNLIKELY(is_float_out_of_range(result_f))) {
ret = OB_OPERATE_OVERFLOW;
char expr_str[OB_MAX_TWO_OPERATOR_EXPR_LENGTH];
int64_t pos = 0;
databuff_printf(expr_str, OB_MAX_TWO_OPERATOR_EXPR_LENGTH, pos, "'(%e / %e)'", left_f, right_f);
LOG_USER_ERROR(OB_OPERATE_OVERFLOW, "BINARY_FLOAT", expr_str);
LOG_WARN("float out of range", K(ret), K(left_f), K(right_f));
} else {
result.set_float(result_f);
LOG_DEBUG("succ to div float", K(left_f), K(right_f), K(result_f));
}
}
return ret;
}
int ObExprDiv::div_number(
ObDatum& result, const ObDatum& left, const ObDatum& right, ObEvalCtx& ctx, const ObExpr& expr)
{
int ret = OB_SUCCESS;
if (right.get_number().is_zero()) {
if (is_oracle_mode()) {
ret = OB_ERR_DIVISOR_IS_ZERO;
LOG_WARN("divisor is equal to zero on oracle mode", K(ret));
} else {
result.set_null();
LOG_DEBUG("divisor is equal to zero", K(left), K(ret));
}
} else {
number::ObNumber lnum(left.get_number());
number::ObNumber rnum(right.get_number());
char local_buff[ObNumber::MAX_BYTE_LEN];
ObDataBuffer local_alloc(local_buff, ObNumber::MAX_BYTE_LEN);
ObNumber result_num;
if (OB_FAIL(lnum.div_v3(rnum, result_num, local_alloc))) {
LOG_WARN("add number failed", K(ret));
} else {
if (is_oracle_mode()) {
result.set_number(result_num);
} else {
int64_t div_pi = 0;
if (OB_FAIL(ctx.exec_ctx_.get_my_session()->get_div_precision_increment(div_pi))) {
LOG_WARN("get_div_precision_increment failed", K(ret));
} else {
// const int64_t scale1 = lnum.get_scale();
// const int64_t scale2 = rnum.get_scale();
// const int64_t new_scale1 = ROUND_UP(scale1);
// const int64_t new_scale2 = ROUND_UP(scale2);
// const int64_t calc_scale = ROUND_UP(new_scale1 + new_scale2 + div_pi);
const int64_t calc_scale = expr.div_calc_scale_;
if (calc_scale > 0 && OB_FAIL(result_num.trunc(calc_scale))) {
// calc_scale is calc_scale ,not res_scale.
// trunc with calc_scale and round with res_scale
LOG_WARN("failed to trunc result number", K(ret), K(result_num), K(calc_scale));
} else {
result.set_number(result_num);
}
LOG_DEBUG("finish div",
K(ret),
K(calc_scale),
/*K(scale1), K(scale2), K(new_scale1), K(new_scale2),*/
K(div_pi),
K(result_num),
K(lnum),
K(rnum));
}
}
}
}
return ret;
}
} // namespace sql
} // namespace oceanbase