236 lines
7.8 KiB
C++
236 lines
7.8 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 <string.h>
|
|
#include "sql/parser/ob_item_type.h"
|
|
#include "sql/engine/expr/ob_expr_utl_i18n.h"
|
|
#include "sql/engine/expr/ob_expr_hex.h"
|
|
#include "sql/engine/expr/ob_expr_util.h"
|
|
#include "sql/engine/expr/ob_expr_operator.h"
|
|
#include "share/object/ob_obj_cast.h"
|
|
#include "lib/oblog/ob_log.h"
|
|
#include "sql/session/ob_sql_session_info.h"
|
|
using namespace oceanbase::common;
|
|
|
|
namespace oceanbase {
|
|
namespace sql {
|
|
|
|
// UTL_I18N.STRING_TO_RAW begin
|
|
ObExprUtlI18nStringToRaw::ObExprUtlI18nStringToRaw(ObIAllocator& alloc)
|
|
: ObStringExprOperator(alloc, T_FUN_SYS_UTL_I18N_STRING_TO_RAW, N_UTL_I18N_STRING_TO_RAW, 2)
|
|
{}
|
|
|
|
ObExprUtlI18nStringToRaw::~ObExprUtlI18nStringToRaw()
|
|
{}
|
|
|
|
int ObExprUtlI18nStringToRaw::calc_result_type2(
|
|
ObExprResType& type, ObExprResType& type1, ObExprResType& type2, common::ObExprTypeCtx& type_ctx) const
|
|
{
|
|
UNUSED(type_ctx);
|
|
int ret = OB_SUCCESS;
|
|
|
|
if (ob_is_null(type1.get_type())) {
|
|
type.set_null();
|
|
} else {
|
|
type1.set_calc_type(common::ObVarcharType);
|
|
type.set_raw();
|
|
type.set_collation_level(common::CS_LEVEL_NUMERIC);
|
|
if (ob_is_null(type2.get_type())) {
|
|
} else {
|
|
type2.set_calc_type(common::ObVarcharType);
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int ObExprUtlI18nStringToRaw::calc(common::ObObj& result, const common::ObObj& obj1, const common::ObObj& obj2,
|
|
common::ObCastCtx& cast_ctx, const ObSQLSessionInfo* session_info)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
|
|
if (obj1.is_null()) {
|
|
result.set_null();
|
|
} else if (OB_UNLIKELY(!obj1.is_varchar())) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
LOG_WARN("input is not varchar", K(ret), K(obj1), K(obj2));
|
|
} else {
|
|
if (OB_ISNULL(cast_ctx.allocator_v2_)) {
|
|
ret = OB_INVALID_ARGUMENT;
|
|
LOG_WARN("allocator in cast ctx is NULL", K(ret));
|
|
} else if (CS_TYPE_INVALID == obj1.get_collation_type()) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
LOG_WARN("collation type of input is invalid", K(ret), K(obj1));
|
|
} else {
|
|
ObCollationType dest_collation = CS_TYPE_INVALID;
|
|
ObString dest_str;
|
|
if (obj2.is_null()) {
|
|
dest_collation = session_info->get_nls_collation();
|
|
} else { // obj2.is_varchar()
|
|
dest_collation =
|
|
ObCharset::get_default_collation_oracle(ObCharset::charset_type_by_name_oracle(obj2.get_string()));
|
|
}
|
|
if (OB_FAIL(ret)) {
|
|
} else if (CS_TYPE_INVALID != dest_collation) {
|
|
if (OB_FAIL(ObExprUtil::convert_string_collation(
|
|
obj1.get_string(), obj1.get_collation_type(), dest_str, dest_collation, *cast_ctx.allocator_v2_))) {
|
|
LOG_WARN("fail to convert string collation", K(ret), K(obj1), K(dest_str), K(dest_collation));
|
|
} else {
|
|
result.set_raw(dest_str.ptr(), dest_str.length());
|
|
}
|
|
} else {
|
|
result.set_null();
|
|
}
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int ObExprUtlI18nStringToRaw::calc_result2(
|
|
common::ObObj& result, const common::ObObj& obj1, const common::ObObj& obj2, common::ObExprCtx& expr_ctx) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
const ObSQLSessionInfo* session_info = NULL;
|
|
|
|
if (OB_ISNULL(session_info = expr_ctx.my_session_)) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
LOG_WARN("session info is null", K(ret));
|
|
} else if (OB_ISNULL(expr_ctx.calc_buf_)) {
|
|
ret = OB_NOT_INIT;
|
|
LOG_WARN("varchar buffer not init", K(ret));
|
|
} else {
|
|
EXPR_DEFINE_CAST_CTX(expr_ctx, CM_NONE);
|
|
if (OB_FAIL(calc(result, obj1, obj2, cast_ctx, session_info))) {
|
|
LOG_WARN("fail to calc", K(ret), K(obj1), K(obj2));
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
// UTL_I18N.STRING_TO_RAW end
|
|
|
|
// UTL_I18N.RAW_TO_CHAR begin
|
|
ObExprUtlI18nRawToChar::ObExprUtlI18nRawToChar(ObIAllocator& alloc)
|
|
: ObStringExprOperator(alloc, T_FUN_SYS_UTL_I18N_RAW_TO_CHAR, N_UTL_I18N_RAW_TO_CHAR, 2)
|
|
{}
|
|
|
|
ObExprUtlI18nRawToChar::~ObExprUtlI18nRawToChar()
|
|
{}
|
|
|
|
int ObExprUtlI18nRawToChar::calc_result_type2(
|
|
ObExprResType& type, ObExprResType& type1, ObExprResType& type2, common::ObExprTypeCtx& type_ctx) const
|
|
{
|
|
UNUSED(type_ctx);
|
|
int ret = OB_SUCCESS;
|
|
const ObBasicSessionInfo* session_info = NULL;
|
|
if (OB_ISNULL(session_info = type_ctx.get_session())) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
LOG_WARN("session info is null", K(ret));
|
|
} else if (ob_is_null(type1.get_type())) {
|
|
type.set_null();
|
|
} else if (ob_is_string_type(type1.get_type()) || ob_is_raw(type1.get_type())) {
|
|
type.set_varchar();
|
|
type.set_collation_level(common::CS_LEVEL_COERCIBLE);
|
|
type.set_collation_type(session_info->get_nls_collation());
|
|
if (ob_is_null(type2.get_type())) {
|
|
} else {
|
|
type2.set_calc_type(common::ObVarcharType);
|
|
}
|
|
} else {
|
|
ret = OB_ERR_WRONG_FUNC_ARGUMENTS_TYPE;
|
|
LOG_WARN("wrong type of arguments in function", K(ret), K(type1));
|
|
ObString func_name("RAW_TO_CHAR");
|
|
LOG_USER_ERROR(OB_ERR_WRONG_FUNC_ARGUMENTS_TYPE, func_name.length(), func_name.ptr());
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int ObExprUtlI18nRawToChar::calc(common::ObObj& result, const common::ObObj& obj1, const common::ObObj& obj2,
|
|
common::ObCastCtx& cast_ctx, const ObSQLSessionInfo* session_info)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
|
|
if (obj1.is_null()) {
|
|
result.set_null();
|
|
} else if (OB_UNLIKELY(!obj1.is_raw() && !obj1.is_string_type())) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
LOG_WARN("wrong number or types of arguments in function", K(ret), K(obj1));
|
|
} else if (OB_ISNULL(cast_ctx.allocator_v2_)) {
|
|
ret = OB_INVALID_ARGUMENT;
|
|
LOG_WARN("allocator in cast ctx is NULL", K(ret));
|
|
} else {
|
|
ObObj tmp_obj;
|
|
ObString src_str;
|
|
ObString dest_str;
|
|
ObCollationType src_collation = CS_TYPE_INVALID;
|
|
if (obj1.is_raw()) {
|
|
src_str = obj1.get_string();
|
|
} else {
|
|
if (OB_FAIL(ObHexUtils::hextoraw(obj1, cast_ctx, tmp_obj))) {
|
|
LOG_WARN("fail to hextoraw", K(ret), K(obj1));
|
|
} else {
|
|
src_str = tmp_obj.get_string();
|
|
}
|
|
}
|
|
if (OB_FAIL(ret)) {
|
|
} else if (obj2.is_null()) {
|
|
src_collation = session_info->get_nls_collation();
|
|
} else { // obj2.is_varchar()
|
|
src_collation =
|
|
ObCharset::get_default_collation_oracle(ObCharset::charset_type_by_name_oracle(obj2.get_string()));
|
|
}
|
|
if (OB_FAIL(ret)) {
|
|
} else if (CS_TYPE_INVALID != src_collation) {
|
|
if (OB_FAIL(ObExprUtil::convert_string_collation(
|
|
src_str, src_collation, dest_str, session_info->get_nls_collation(), *cast_ctx.allocator_v2_))) {
|
|
LOG_WARN("fail to convert string collation", K(ret), K(obj1), K(src_collation));
|
|
} else {
|
|
result.set_varchar(dest_str.ptr(), dest_str.length());
|
|
result.set_collation_type(session_info->get_nls_collation());
|
|
}
|
|
} else {
|
|
result.set_null();
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int ObExprUtlI18nRawToChar::calc_result2(
|
|
common::ObObj& result, const common::ObObj& obj1, const common::ObObj& obj2, common::ObExprCtx& expr_ctx) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
const ObSQLSessionInfo* session_info = NULL;
|
|
|
|
if (OB_ISNULL(session_info = expr_ctx.my_session_)) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
LOG_WARN("session info is null", K(ret));
|
|
} else if (OB_ISNULL(expr_ctx.calc_buf_)) {
|
|
ret = OB_NOT_INIT;
|
|
LOG_WARN("varchar buffer not init", K(ret));
|
|
} else {
|
|
EXPR_DEFINE_CAST_CTX(expr_ctx, CM_NONE);
|
|
if (OB_FAIL(calc(result, obj1, obj2, cast_ctx, session_info))) {
|
|
LOG_WARN("fail to calc", K(ret), K(obj1), K(obj2));
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
// UTL_I18N.RAW_TO_CHAR end
|
|
|
|
} // namespace sql
|
|
} // namespace oceanbase
|