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

add DEGREES() function that converts radian to degree.

Browse Source
This commit is contained in:
Monk-Liu
2021-10-13 20:00:57 +08:00
committed by wangzelin.wzl
parent 614d2970e8
commit 75c14d5c88
9 changed files with 180 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
src/sql/engine/expr/ob_expr_degrees.cpp Normal file
View File

@ -0,0 +1,86 @@
// Copyright (c) 2014-2021 Alibaba Inc. All Right Reserved.
// Author:
// liuqifan.lqf@antgroup.com
#define USING_LOG_PREFIX SQL_ENG
#include "sql/engine/expr/ob_expr_degrees.h"
#include "lib/number/ob_number_v2.h"
#include "sql/engine/expr/ob_expr_util.h"
#include "sql/parser/ob_item_type.h"
#include "sql/session/ob_sql_session_info.h"
#include <cmath>
using namespace oceanbase::common;
using namespace oceanbase::sql;
namespace oceanbase {
namespace sql {
const double ObExprDegrees::degrees_ratio_ = 180.0 / std::acos(-1);
ObExprDegrees::ObExprDegrees(ObIAllocator &alloc)
: ObFuncExprOperator(alloc, T_FUN_SYS_DEGREES, N_DEGREES, 1, NOT_ROW_DIMENSION)
{}
ObExprDegrees::~ObExprDegrees()
{}
int ObExprDegrees::calc_result_type1(ObExprResType &type, ObExprResType &radian, ObExprTypeCtx &type_ctx) const
{
UNUSED(type_ctx);
int ret = OB_SUCCESS;
if (NOT_ROW_DIMENSION != row_dimension_ || ObMaxType == radian.get_type()) {
ret = OB_ERR_INVALID_TYPE_FOR_OP;
} else {
type.set_double();
radian.set_calc_type(ObDoubleType);
ObExprOperator::calc_result_flag1(type, radian);
}
return ret;
}
int ObExprDegrees::calc_result1(ObObj &result, const ObObj &radian_obj, ObExprCtx &expr_ctx) const
{
int ret = OB_SUCCESS;
double val = 0.0;
if (OB_ISNULL(expr_ctx.calc_buf_)) {
ret = OB_NOT_INIT;
LOG_WARN("expr_ctx.calc_buf_ is NULL", K(ret));
} else if (OB_FAIL(radian_obj.get_double(val))) {
LOG_WARN("get double from obj failed in degrees", K(ret), K(radian_obj), K(val));
} else {
result.set_double(val * degrees_ratio_);
}
return ret;
}
int ObExprDegrees::calc_degrees_expr(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &res_datum)
{
int ret = OB_SUCCESS;
ObDatum *radian = NULL;
if (OB_FAIL(expr.args_[0]->eval(ctx, radian))) {
LOG_WARN("eval radian arg failed", K(ret), K(expr));
} else if (radian->is_null()) {
res_datum.set_null();
} else {
const double val = radian->get_double();
// cal result;
res_datum.set_double(val * degrees_ratio_);
}
return ret;
}
int ObExprDegrees::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);
if (OB_UNLIKELY(1 != rt_expr.arg_cnt_) || (ObDoubleType != rt_expr.args_[0]->datum_meta_.type_)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("invalid arg_cnt_ or res type is invalid", K(ret), K(rt_expr));
} else {
rt_expr.eval_func_ = calc_degrees_expr;
}
return ret;
}
} // namespace sql
} // namespace oceanbase