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

[FEAT MERGE] GIS

Browse Source
This commit is contained in:
obdev
2022-12-23 12:08:27 +00:00
committed by ob-robot
parent 38846e73d6
commit 40d215fc5a
519 changed files with 158600 additions and 8396 deletions
Download Patch File Download Diff File Expand all files Collapse all files

195
src/sql/engine/expr/ob_expr_st_asewkt.cpp Normal file
View File

@ -0,0 +1,195 @@
/**
* 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.
* This file contains implementation for st_astext.
*/
#define USING_LOG_PREFIX SQL_ENG
#include "sql/engine/expr/ob_expr_st_asewkt.h"
#include "observer/omt/ob_tenant_srs_mgr.h"
#include "sql/engine/expr/ob_geo_expr_utils.h"
#include "lib/geo/ob_geo_utils.h"
#include "lib/geo/ob_geo_reverse_coordinate_visitor.h"
#include "lib/geo/ob_geo_to_wkt_visitor.h"
#include "lib/geo/ob_geo_func_common.h"
using namespace oceanbase::common;
using namespace oceanbase::sql;
using namespace oceanbase::omt;
namespace oceanbase
{
namespace sql
{
ObExprPrivSTAsEwkt::ObExprPrivSTAsEwkt(ObIAllocator &alloc)
: ObFuncExprOperator(alloc, T_FUN_SYS_PRIV_ST_ASEWKT, N_PRIV_ST_ASEWKT, MORE_THAN_ZERO, NOT_ROW_DIMENSION)
{
}
ObExprPrivSTAsEwkt::~ObExprPrivSTAsEwkt()
{
}
int ObExprPrivSTAsEwkt::calc_result_typeN(ObExprResType& type,
ObExprResType* types_stack,
int64_t param_num,
ObExprTypeCtx& type_ctx) const
{
UNUSED(type_ctx);
int ret = OB_SUCCESS;
if (OB_UNLIKELY(param_num > 2)) {
ObString fun_name(N_PRIV_ST_ASEWKT);
ret = OB_ERR_PARAM_SIZE;
LOG_USER_ERROR(OB_ERR_PARAM_SIZE, fun_name.length(), fun_name.ptr());
} else {
if (ob_is_geometry(types_stack[0].get_type())
|| ob_is_null(types_stack[0].get_type())) {
// do nothing
} else if (ob_is_string_type(types_stack[0].get_type())) {
types_stack[0].set_calc_type(ObGeometryType);
types_stack[0].set_calc_collation_type(CS_TYPE_BINARY);
types_stack[0].set_calc_collation_level(CS_LEVEL_IMPLICIT);
} else {
ret = OB_ERR_INVALID_TYPE_FOR_ARGUMENT;
LOG_WARN("invalid type", K(ret), K(types_stack[0].get_type()));
}
if (OB_SUCC(ret) && param_num > 1) {
if (ob_is_integer_type(types_stack[1].get_type())
|| ob_is_null(types_stack[1].get_type())) {
// do nothing
} else {
ret = OB_ERR_INVALID_TYPE_FOR_ARGUMENT;
LOG_WARN("invalid type", K(ret), K(types_stack[1].get_type()));
}
}
if (OB_SUCC(ret)) {
type.set_type(ObLongTextType);
type.set_collation_type(CS_TYPE_UTF8MB4_GENERAL_CI);
type.set_collation_level(CS_LEVEL_IMPLICIT);
type.set_accuracy(ObAccuracy::DDL_DEFAULT_ACCURACY[ObLongTextType]);
}
}
return ret;
}
/*
* Input is either ob geometry or swkb.
* axis-order issue https://aone.alibaba-inc.com/v2/project/81079/bug/41824316
* may still need srs for validation input swkb in cast(hexstring->obgeometry)
*/
int ObExprPrivSTAsEwkt::eval_priv_st_asewkt(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &res)
{
int ret = OB_SUCCESS;
ObArenaAllocator tmp_allocator;
int num_args = expr.arg_cnt_;
bool is_null_result = false;
ObString res_wkt;
ObDatum *gis_datum = NULL;
int64_t maxdecimaldigits = DEFAULT_DIGITS_IN_DOUBLE;
// get geo
if (OB_FAIL(expr.args_[0]->eval(ctx, gis_datum))) {
LOG_WARN("eval geo args failed", K(ret));
} else if (gis_datum->is_null()) {
is_null_result = true;
} else if (num_args > 1) { // get maxdecimaldigits
ObDatum *precsion_data = NULL;
if (OB_FAIL(expr.args_[1]->eval(ctx, precsion_data))) {
LOG_WARN("eval maxdecimaldigits args failed", K(ret));
} else if (precsion_data->is_null()){
is_null_result = true;
} else {
maxdecimaldigits = precsion_data->get_int();
}
} else { /* do nothing */ }
if (OB_SUCC(ret)) {
if ((!is_null_result) && OB_FAIL(ObGeoTypeUtil::geo_to_ewkt(gis_datum->get_string(),
res_wkt,
tmp_allocator,
maxdecimaldigits))) {
LOG_WARN("eval geo to ewkt failed", K(ret), K(gis_datum->get_string()), K(maxdecimaldigits));
} else if (is_null_result) {
res.set_null();
} else {
char *buf = NULL;
if (OB_ISNULL(buf = expr.get_str_res_mem(ctx, res_wkt.length()))) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_WARN("allocate memory failed", K(ret));
} else {
MEMCPY(buf, res_wkt.ptr(), res_wkt.length());
res.set_string(buf, res_wkt.length());
}
}
}
return ret;
}
int ObExprPrivSTAsEwkt::calc_resultN(common::ObObj &result,
const common::ObObj *objs,
int64_t param_num,
common::ObExprCtx &expr_ctx) const
{
int ret = OB_SUCCESS;
ObArenaAllocator tmp_allocator;
bool is_null_result = false;
ObString res_wkt;
int64_t maxdecimaldigits = DEFAULT_DIGITS_IN_DOUBLE;
// get geo
if (objs[0].is_null()) {
is_null_result = true;
} else if (param_num > 1) { // get maxdecimaldigits
if (objs[1].is_null()){
is_null_result = true;
} else {
maxdecimaldigits = objs[1].get_int();
}
} else { /* do nothing */ }
if (OB_SUCC(ret)) {
if ((!is_null_result) && OB_FAIL(ObGeoTypeUtil::geo_to_ewkt(objs[0].get_string(),
res_wkt,
tmp_allocator,
maxdecimaldigits))) {
LOG_WARN("eval geo to ewkt failed", K(ret), K(objs[0].get_string()), K(maxdecimaldigits));
} else if (is_null_result) {
result.set_null();
} else {
ObIAllocator *allocator = expr_ctx.calc_buf_;
char *buf = static_cast<char*>(allocator->alloc(res_wkt.length()));
if (OB_ISNULL(buf)) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_WARN("allocate memory failed", K(ret));
} else {
MEMCPY(buf, res_wkt.ptr(), res_wkt.length());
result.set_string(ObLongTextType, buf, res_wkt.length());
result.set_collation_type(result_type_.get_collation_type());
}
}
}
return ret;
}
int ObExprPrivSTAsEwkt::cg_expr(ObExprCGCtx &expr_cg_ctx,
const ObRawExpr &raw_expr,
ObExpr &rt_expr) const
{
UNUSED(expr_cg_ctx);
UNUSED(raw_expr);
rt_expr.eval_func_ = eval_priv_st_asewkt;
return OB_SUCCESS;
}
} // namespace sql
} // namespace oceanbase