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

patch 4.0

Browse Source
This commit is contained in:
wangzelin.wzl
2022-10-24 10:34:53 +08:00
parent 4ad6e00ec3
commit 93a1074b0c
10533 changed files with 2588271 additions and 2299373 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
src/sql/engine/expr/ob_expr_space.cpp
View File

@ -20,14 +20,18 @@
#include "sql/session/ob_sql_session_info.h"
#include "sql/engine/ob_exec_context.h"
namespace oceanbase {
namespace sql {
namespace oceanbase
{
namespace sql
{
using namespace common;
ObExprSpace::ObExprSpace(ObIAllocator& alloc)
: ObFuncExprOperator(alloc, T_FUN_SYS_SPACE, N_SPACE, 1, NOT_ROW_DIMENSION)
{}
ObExprSpace::ObExprSpace(ObIAllocator &alloc)
: ObFuncExprOperator(alloc, T_FUN_SYS_SPACE, N_SPACE, 1, NOT_ROW_DIMENSION) {}
inline int ObExprSpace::calc_result_type1(ObExprResType& type, ObExprResType& type1, ObExprTypeCtx& type_ctx) const
inline int ObExprSpace::calc_result_type1(
ObExprResType &type,
ObExprResType &type1,
ObExprTypeCtx &type_ctx) const
{
int ret = OB_SUCCESS;
// space is mysql only expr
@ -43,51 +47,7 @@ inline int ObExprSpace::calc_result_type1(ObExprResType& type, ObExprResType& ty
return ret;
}
int ObExprSpace::calc_result1(ObObj& result, const ObObj& count, ObExprCtx& expr_ctx) const
{
int ret = OB_SUCCESS;
int64_t max_allow_packet_size = 0;
EXPR_DEFINE_CAST_CTX(expr_ctx, CM_NONE);
if (count.is_null()) {
result.set_null();
} else {
if (OB_ISNULL(expr_ctx.my_session_)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("Unexpected null session in space function", K(ret));
} else if (OB_FAIL(expr_ctx.my_session_->get_max_allowed_packet(max_allow_packet_size))) {
if (OB_ENTRY_NOT_EXIST == ret) { // for compatibility with server before 1470
ret = OB_SUCCESS;
max_allow_packet_size = OB_MAX_VARCHAR_LENGTH;
} else {
LOG_WARN("Failed to get max allow packet size", K(ret));
}
}
if (OB_SUCC(ret)) {
TYPE_CHECK(count, ObIntType);
ObString local_text(1, " ");
int64_t local_count = count.get_int();
if ((local_count > max_allow_packet_size)) {
LOG_WARN(
"Result of space was larger than max_allow_packet_size", K(ret), K(local_count), K(max_allow_packet_size));
LOG_USER_WARN(OB_ERR_FUNC_RESULT_TOO_LARGE, "space", static_cast<int>(max_allow_packet_size));
result.set_null();
} else if (OB_FAIL(ObExprRepeat::calc(
result, ObVarcharType, local_text, local_count, expr_ctx.calc_buf_, max_allow_packet_size))) {
LOG_WARN("Failed to cacl result");
}
}
}
if (OB_FAIL(ret)) {
LOG_WARN("Calc result error", K(ret));
} else if (OB_LIKELY(!result.is_null())) {
result.set_collation(result_type_);
} else {
// do nothing
}
return ret;
}
int ObExprSpace::cg_expr(ObExprCGCtx&, const ObRawExpr&, ObExpr& rt_expr) const
int ObExprSpace::cg_expr(ObExprCGCtx &, const ObRawExpr &, ObExpr &rt_expr) const
{
int ret = OB_SUCCESS;
CK(1 == rt_expr.arg_cnt_);
@ -95,10 +55,10 @@ int ObExprSpace::cg_expr(ObExprCGCtx&, const ObRawExpr&, ObExpr& rt_expr) const
return ret;
}
int ObExprSpace::eval_space(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& expr_datum)
int ObExprSpace::eval_space(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum)
{
int ret = OB_SUCCESS;
ObDatum* count = NULL;
ObDatum *count = NULL;
int64_t max_size = 0;
bool is_null = false;
ObString output;
@ -111,10 +71,12 @@ int ObExprSpace::eval_space(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& expr_da
} else if (OB_FAIL(ctx.exec_ctx_.get_my_session()->get_max_allowed_packet(max_size))) {
LOG_WARN("get max length failed", K(ret));
} else if (count->get_int() > max_size) {
LOG_WARN("Result of space was larger than max_allow_packet_size", K(count->get_int()), K(max_size));
LOG_WARN("Result of space was larger than max_allow_packet_size",
K(count->get_int()), K(max_size));
LOG_USER_WARN(OB_ERR_FUNC_RESULT_TOO_LARGE, "space", static_cast<int>(max_size));
expr_datum.set_null();
} else if (OB_FAIL(ObExprRepeat::repeat(output, is_null, space, count->get_int(), expr_res_alloc, max_size))) {
} else if (OB_FAIL(ObExprRepeat::repeat(output, is_null,
space, count->get_int(), expr_res_alloc, max_size))) {
LOG_WARN("do repeat failed", K(ret));
} else {
if (is_null) {
@ -127,5 +89,5 @@ int ObExprSpace::eval_space(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& expr_da
return ret;
}
} // namespace sql
} // namespace oceanbase
} // sql
} // oceanbase