Files
oceanbase/src/pl/ob_pl_code_generator.cpp

8800 lines
405 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 PL
#include "ob_pl_code_generator.h"
#include "ob_pl_type.h"
#include "ob_pl_compile.h"
#include "ob_pl_package.h"
#include "lib/string/ob_sql_string.h"
#include "common/ob_smart_call.h"
#include "sql/resolver/expr/ob_raw_expr_util.h"
#include "sql/resolver/expr/ob_raw_expr.h"
#include "sql/code_generator/ob_expr_generator_impl.h"
#include "lib/hash/ob_hashmap.h"
#include "parser/parse_stmt_item_type.h"
namespace oceanbase
{
using namespace common;
using namespace jit;
using namespace sql;
namespace pl
{
int ObPLCodeGenerateVisitor::generate(const ObPLStmt &s)
{
int ret = OB_SUCCESS;
OZ (generator_.restart_cg_when_goto_dest(s));
OZ (s.accept(*this));
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLStmtBlock &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {
ObLLVMBasicBlock exit;
if (s.has_label()) {
if (OB_FAIL(generator_.get_helper().create_block(ObString("exit_block"), generator_.get_func(), exit))) {
LOG_WARN("failed to create block", K(s.get_stmts()), K(ret));
} else {
ObLLVMBasicBlock null_start;
generator_.set_label(s.get_label(), s.get_level(), null_start, exit);
}
}
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_stmts().count(); ++i) {
ObPLStmt *stmt = s.get_stmts().at(i);
if (OB_ISNULL(stmt)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("stmt in block is NULL", K(i), K(s.get_stmts()), K(ret));
} else if (OB_FAIL(SMART_CALL(generate(*stmt)))) {
LOG_WARN("failed to generate", K(i), K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
if (s.has_eh()) { //如果有eh,跳到eh的exit分支
if (NULL != generator_.get_current_exception()->exit_.get_v()) {
if (OB_FAIL(generator_.finish_current(generator_.get_current_exception()->exit_))) {
LOG_WARN("failed to finish current", K(s.get_stmts()), K(ret));
} else if (NULL != exit.get_v()) {
if (generator_.get_current_exception()->exit_.get_v() == generator_.get_exit().get_v()) {
//exit是终点,不再跳转
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current_exception()->exit_))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(exit))) {
LOG_WARN("failed to create br", K(ret));
} else { /*do nothing*/ }
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
if (OB_FAIL(generator_.set_current(NULL == exit.get_v() ? generator_.get_current_exception()->exit_ : exit))) {
LOG_WARN("failed to set current", K(ret));
}
}
}
if (OB_SUCC(ret)) {
if (OB_FAIL(generator_.reset_exception())) {
LOG_WARN("failed to reset exception", K(ret));
}
}
} else if (NULL != exit.get_v()) { //如果没有eh,调到BLOCK自己的exit分支
if (OB_FAIL(generator_.get_helper().create_br(exit))) {
LOG_WARN("failed to create br", K(ret));
} else if (OB_FAIL(generator_.set_current(exit))) {
LOG_WARN("failed to set current", K(ret));
} else { /*do nothing*/ }
} else { /*do nothing*/ }
if (OB_SUCC(ret) && NULL != exit.get_v()) {
if (OB_FAIL(generator_.reset_label())) {
LOG_WARN("failed to reset label", K(ret));
}
}
}
//释放内存
if (OB_SUCC(ret) && NULL != generator_.get_current().get_v()) {
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_namespace().get_symbols().count(); ++i) {
const ObPLVar *var = s.get_variable(s.get_namespace().get_symbols().at(i));
if (OB_ISNULL(var)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("var is NULL", K(i), K(s.get_namespace().get_symbols().at(i)), K(var), K(ret));
} else if (var->get_type().is_collection_type() && var->get_name().length() != 0) {
ObSEArray<ObLLVMValue, 2> args;
ObLLVMValue int_value;
if (s.get_namespace().get_symbols().at(i) < generator_.get_param_size()) {
// 不要释放入参的内存
} else if (OB_FAIL(args.push_back(generator_.get_vars().at(generator_.CTX_IDX)))) { //PL的执行环境
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.get_helper().get_int64(s.get_namespace().get_symbols().at(i), int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(args.push_back(int_value))) {
LOG_WARN("push_back error", K(ret));
} else {
ObLLVMValue ret_err;
if (OB_FAIL(generator_.get_helper().create_call(ObString("spi_destruct_collection"), generator_.get_spi_service().spi_destruct_collection_, args, ret_err))) {
LOG_WARN("failed to create call", K(ret));
} else if (OB_FAIL(generator_.check_success(ret_err, s.get_stmt_id(), s.in_notfound(), s.in_warning()))) {
LOG_WARN("failed to check success", K(ret));
}
}
}
}
// close cursor
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_namespace().get_cursors().count(); ++i) {
const ObPLCursor *cursor = s.get_cursor(s.get_namespace().get_cursors().at(i));
OZ (generator_.generate_handle_ref_cursor(cursor, s, s.in_notfound(), s.in_warning()));
}
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLDeclareUserTypeStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else {
const ObUserDefinedType *user_defined_type = s.get_user_type();
if (OB_ISNULL(user_defined_type)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("user defined type is null");
} else if (OB_FAIL(generator_.generate_user_type(*user_defined_type))) {
LOG_WARN("failed to generate user type", K(user_defined_type->get_type()));
} else { /*do nothing*/ }
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLDeclareVarStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else {
OZ (generator_.get_helper().set_insert_point(generator_.get_current()));
OZ (generator_.set_debug_location(s));
}
if (OB_SUCC(ret) && OB_NOT_NULL(generator_.get_current().get_v())) {
ObLLVMType ir_type;
ObLLVMValue value;
bool is_complex_type_var = false;
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_index().count(); ++i) {
const ObPLVar *var = s.get_var(i);
CK (OB_NOT_NULL(var));
if (OB_SUCC(ret)) {
if (var->get_type().is_obj_type()) {
OZ (var->get_type().generate_construct(generator_, *s.get_namespace(), value, &s));
OX (generator_.get_vars().at(s.get_index(i) + generator_.USER_ARG_OFFSET) = value);
} else { // Record和Collection的内存不在栈上申请, 统一在Allocator中申请, 执行结束后统一释放
ObSEArray<ObLLVMValue, 3> args;
ObLLVMValue var_idx, init_value, var_value, extend_value;
ObLLVMValue ret_err;
ObLLVMValue var_type, type_id;
ObLLVMValue null_int;
int64_t init_size = 0;
is_complex_type_var = true;
// Step 1: 初始化内存
CK (OB_NOT_NULL(s.get_namespace()));
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.get_helper().get_int8(var->get_type().get_type(), var_type));
OZ (args.push_back(var_type));
OZ (generator_.get_helper().get_int64(var->get_type().get_user_type_id(), type_id));
OZ (args.push_back(type_id));
OZ (generator_.get_helper().get_int64(s.get_index(i), var_idx));
OZ (args.push_back(var_idx));
OZ (s.get_namespace()->get_size(PL_TYPE_INIT_SIZE, var->get_type(), init_size));
OZ (generator_.get_helper().get_int32(init_size, init_value));
OZ (args.push_back(init_value));
OZ (generator_.generate_null_pointer(ObIntType, null_int));
OZ (args.push_back(null_int));
OZ (generator_.get_helper().create_call(ObString("spi_alloc_complex_var"),
generator_.get_spi_service().spi_alloc_complex_var_,
args,
ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
// Step 2: 初始化类型内容, 如Collection的rowsize,element type等
if (OB_SUCC(ret) && !var->get_type().is_opaque_type()) {
OZ (generator_.extract_objparam_from_context(
generator_.get_vars().at(generator_.CTX_IDX),
s.get_index(i),
var_value));
OZ (generator_.extract_extend_from_objparam(var_value, var->get_type(), extend_value));
OZ (var->get_type().generate_construct(generator_, *s.get_namespace(),
extend_value, &s));
#ifndef NDEBUG
{
ObLLVMType int_type;
ObLLVMValue tmp;
OZ (generator_.get_helper().get_llvm_type(ObIntType, int_type));
OZ (generator_.get_helper().create_ptr_to_int(ObString("cast_pointer_to_int"), extend_value, int_type, tmp));
OZ (generator_.generate_debug(ObString("print new construct"), tmp));
}
#endif
}
// Step 3: 默认值是构造函数,调用构造函数的初始化
if (OB_SUCC(ret)
&& var->get_type().is_collection_type()
&& PL_CONSTRUCT_COLLECTION == s.get_default()) {
ObLLVMValue package_id;
args.reset();
OZ (generator_.get_helper().get_int64(OB_INVALID_ID, package_id));
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (args.push_back(package_id));
OZ (args.push_back(var_value));
OZ (generator_.get_helper().create_call(ObString("spi_construct_collection"), generator_.get_spi_service().spi_construct_collection_, args, ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
}
}
}
}
// 处理变量默认值
if (OB_SUCC(ret)
&& OB_INVALID_INDEX != s.get_default()
&& PL_CONSTRUCT_COLLECTION != s.get_default()) {
ObLLVMValue p_result_obj;
OZ (generator_.generate_expr(s.get_default(),
s,
is_complex_type_var ? OB_INVALID_ID : s.get_index(0),
p_result_obj));
CK (OB_NOT_NULL(s.get_default_expr()));
// 检查notnull约束
CK (OB_NOT_NULL(s.get_var(0)));
OZ (generator_.generate_check_not_null(s, s.get_var(0)->is_not_null(), p_result_obj));
if (OB_SUCC(ret)) {
ObLLVMValue result;
if (OB_FAIL(generator_.extract_datum_from_objparam(p_result_obj, s.get_default_expr()->get_data_type(), result))) {
LOG_WARN("failed to extract_datum_from_objparam", K(ret));
} else {
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_index().count(); ++i) {
const ObPLVar *var = s.get_var(i);
CK (OB_NOT_NULL(var));
// 将默认值存储到paramstore, 第0个变量已经在generate_expr处理过了
if (OB_SUCC(ret) && (is_complex_type_var || 0 != i)) {
ObLLVMValue into_meta_p, ori_meta_p, ori_meta;
ObLLVMValue into_accuracy_p, ori_accuracy_p, ori_accuracy;
ObLLVMValue into_obj, dest_datum;
OZ (generator_.extract_objparam_from_context(generator_.get_vars().at(generator_.CTX_IDX), s.get_index(i), into_obj));
if (var->get_type().is_obj_type()) {
OZ (generator_.extract_datum_ptr_from_objparam(into_obj, s.get_var(i)->get_type().get_obj_type(), dest_datum));
OZ (generator_.get_helper().create_store(result, generator_.get_vars().at(s.get_index(i) + generator_.USER_ARG_OFFSET)));
OZ (generator_.get_helper().create_store(result, dest_datum));
} else {
ObLLVMValue allocator, src_datum;
const ObConstRawExpr *const_expr = static_cast<const ObConstRawExpr*>(s.get_default_expr());
OZ (generator_.generate_null(ObIntType, allocator));
CK (NULL != s.get_symbol_table());
if (OB_FAIL(ret)) {
} else if (T_NULL == s.get_default_expr()->get_expr_type() ||
(T_QUESTIONMARK == const_expr->get_expr_type() &&
0 == s.get_symbol_table()->get_symbol(const_expr->get_value().get_unknown())->get_name().case_compare(ObPLResolver::ANONYMOUS_ARG) &&
NULL != s.get_symbol_table()->get_symbol(const_expr->get_value().get_unknown())->get_pl_data_type().get_data_type() &&
ObNullType == s.get_symbol_table()->get_symbol(const_expr->get_value().get_unknown())->get_pl_data_type().get_data_type()->get_obj_type())) {
/*
* allocator可以是null,因为这儿只是修改原复杂类型的属性信息
* 对于record来说把内部的元素初始化成一个null,
* 对于collection来说把count改为-1标记为未初始化状态
*/
OZ (generator_.extract_extend_from_objparam(into_obj,
var->get_type(),
dest_datum));
OZ (var->get_type().generate_assign_with_null(generator_,
*s.get_namespace(),
allocator,
dest_datum));
} else {
OZ (generator_.extract_obobj_ptr_from_objparam(p_result_obj, src_datum));
OZ (generator_.extract_obobj_ptr_from_objparam(into_obj, dest_datum));
OZ (var->get_type().generate_copy(generator_, *s.get_namespace(),
allocator, src_datum, dest_datum,
s.get_block()->in_notfound(),
s.get_block()->in_warning(),
OB_INVALID_ID));
OZ (generator_.extract_meta_ptr_from_objparam(p_result_obj, ori_meta_p));
OZ (generator_.extract_meta_ptr_from_objparam(into_obj, into_meta_p));
OZ (generator_.get_helper().create_load(ObString("load_meta"),
ori_meta_p, ori_meta));
OZ (generator_.get_helper().create_store(ori_meta, into_meta_p));
OZ (generator_.extract_accuracy_ptr_from_objparam(p_result_obj, ori_accuracy_p));
OZ (generator_.extract_accuracy_ptr_from_objparam(into_obj, into_accuracy_p));
OZ (generator_.get_helper().create_load(ObString("load_accuracy"),
ori_accuracy_p, ori_accuracy));
OZ (generator_.get_helper().create_store(ori_accuracy, into_accuracy_p));
}
}
}
}
}
}
}
if (lib::is_mysql_mode()) {
ObLLVMValue ret_err;
ObSEArray<ObLLVMValue, 1> args;
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.get_helper().create_call(ObString("spi_clear_diagnostic_area"),
generator_.get_spi_service().spi_clear_diagnostic_area_,
args,
ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
}
}
return ret;
}
/***************************************************************************************/
/* 注意:以下是内存排列有关的代码,修改这里一定要十分理解各种数据类型在LLVM端和SQL端的内存排列和生命周期。
* 如有问题请联系如颠ryan.ly
***************************************************************************************/
int ObPLCodeGenerateVisitor::visit(const ObPLAssignStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (s.get_into().count() != s.get_value().count()) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("into exprs not equal to value exprs", K(ret), K(s.get_into().count()), K(s.get_value().count()));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {
ObLLVMValue stack;
OZ (generator_.get_helper().stack_save(stack));
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_into().count(); ++i) {
const ObRawExpr *into_expr = s.get_into_expr(i);
const ObRawExpr *value_expr = NULL;
if (OB_ISNULL(into_expr)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("a assign stmt must has expr", K(s), K(into_expr), K(ret));
} else {
int64_t result_idx = OB_INVALID_INDEX;
if (into_expr->is_const_raw_expr()) {
const ObConstRawExpr* const_expr = static_cast<const ObConstRawExpr*>(into_expr);
if (const_expr->get_value().is_unknown()) {
const ObPLVar *var = NULL;
int64_t idx = const_expr->get_value().get_unknown();
CK (OB_NOT_NULL(var = s.get_variable(idx)));
if (OB_SUCC(ret)) {
const ObPLDataType &into_type = var->get_type();
if (!into_type.is_collection_type() && !into_type.is_record_type()) {
result_idx = idx;
}
}
} else {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("Unexpected const expr", K(const_expr->get_value()), K(ret));
}
}
if (OB_SUCC(ret)) {
ObLLVMValue p_result_obj;
ObObjAccessIdx::AccessType alloc_scop = ObObjAccessIdx::IS_INVALID;
uint64_t package_id = OB_INVALID_ID;
uint64_t var_idx = OB_INVALID_ID;
ObLLVMValue allocator;
jit::ObLLVMValue src_datum;
jit::ObLLVMValue dest_datum;
const ObPLBlockNS *ns = s.get_namespace();
if (OB_ISNULL(ns)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("Assign stmt must have a valid namespace", K(ret));
} else if (PL_CONSTRUCT_COLLECTION == s.get_value_index(i)) {
ObPLDataType final_type;
CK (into_expr->is_obj_access_expr());
OZ (static_cast<const ObObjAccessRawExpr*>(into_expr)->get_final_type(final_type));
CK (final_type.is_collection_type());
} else {
value_expr = s.get_value_expr(i);
if (OB_FAIL(generator_.generate_expr(s.get_value_index(i), s, result_idx,
p_result_obj))) {
LOG_WARN("failed to generate calc_expr func", K(ret));
}
if (lib::is_mysql_mode()) {
ObLLVMValue ret_err;
ObSEArray<ObLLVMValue, 1> args;
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.get_helper().create_call(ObString("spi_clear_diagnostic_area"),
generator_.get_spi_service().spi_clear_diagnostic_area_,
args,
ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
}
}
if (OB_FAIL(ret)) {
} else if (OB_FAIL(ObObjAccessIdx::datum_need_copy(into_expr, value_expr, alloc_scop))) {
LOG_WARN("failed to check if datum_need_copy", K(*into_expr), K(*value_expr), K(ret));
} else if (ObObjAccessIdx::IS_LOCAL == alloc_scop
|| (ObObjAccessIdx::IS_PKG != alloc_scop)) {
alloc_scop = ObObjAccessIdx::IS_LOCAL;
}
if (OB_SUCC(ret) && ObObjAccessIdx::IS_PKG == alloc_scop) { // 如果是Package变量, 记录下PackageId
OZ (ObObjAccessIdx::get_package_id(into_expr, package_id, &var_idx));
}
if (OB_SUCC(ret)) {
//如果是LOCAL,使用运行时语句级allocator
if (ObObjAccessIdx::IS_LOCAL == alloc_scop) {
/*
* 如果不是Collection,原则上应该进一步向上找到母体Collection的allocator,这样太复杂了,我们放弃这么做,直接用生命周期更长的运行时语句级allocator
* 这样的后果是:Collection里的string和number类型数据有的是在NestedTable自己的allocator里,有的不是。
* 这其实没有问题,基础数据类型不会占用过多内存,而且语句级内存最终会释放,不会泄漏。
*/
if (OB_FAIL(generator_.extract_allocator_from_context(generator_.get_vars().at(generator_.CTX_IDX), allocator))) {
LOG_WARN("Failed to extract_allocator_from_nestedtable", K(*into_expr), K(ret));
}
} else {
//如果不是LOCAL,那么一定是PKG或者是INVALID,传入NULL allocator;
//对于PKG, 由SPI根据package_id从Session上获取对应PackageState上的Allocator;
if (OB_FAIL(generator_.generate_null(ObIntType, allocator))) { //初始化一个空的allocator
LOG_WARN("Failed to extract_allocator_from_nestedtable", K(*into_expr), K(ret));
}
}
}
if (OB_SUCC(ret)) {
/**
* 分三种情况讨论:
* 1、Local变量
* a、原生基本数据类型(不在母结构中的基本数据类型,通过Question Mark访问):直接找到该变量调用create store,同时需要修改param_store里的值(对于string和number也是如此,指针指向同一份内存)
* b、结构中的基本数据类型(在Record或Collection里中的基本数据类型,通过ObjAccess访问):直接找到该变量的地址调用create store或者调用COPY,不必修改param_store里的值(对于string和number也是如此,指针指向同一份内存)
* 注意: 为什么不全部直接调用COPY函数?————其实全部直接调COPY函数代码更简洁,考虑到大多数情况下不需要内存深拷贝,直接create store性能更好。
* 这样会造成一种可能性:NestedTable里的string和number类型数据有的是在Collection自己的allocator里,有的不是。————这其实不一定有问题,只要目的端数据的内存生命周期比源端长就可以。
* c、集合数据类型(通过ObjAccess访问):调用该类型的COPY函数,其COPY函数递归进行子域的COPY,不必修改param_store里的值
* 注意: NestedTable的数据域的结构内存是自身的allocator_分配的,同时String/Number指针指向的内存也是。
* copy之后的Collection同样也是这样。如果有母子关系,Collection的allocator_也相互独立,没有任何所属关系。
* 2、PKG变量
* PKG变量只能通过通过ObjAccess访问访问,同1.b和1.c
* 3、USER/SYS变量
* 通过SPI直接赋值。
* */
if (into_expr->is_const_raw_expr()) { //Local Basic Variables
const ObConstRawExpr *const_expr = static_cast<const ObConstRawExpr*>(into_expr);
const ObPLVar *var = NULL;
CK (OB_NOT_NULL(const_expr));
CK (OB_NOT_NULL(var = s.get_variable(const_expr->get_value().get_unknown())));
OZ (generator_.generate_check_not_null(s, var->is_not_null(), p_result_obj));
if (OB_SUCC(ret)) {
int64_t idx = const_expr->get_value().get_unknown();
const ObPLDataType &into_type = var->get_type();
ObLLVMValue datum;
ObLLVMValue &var_addr = generator_.get_vars().at(idx + generator_.USER_ARG_OFFSET);
if (!into_type.is_collection_type() && !into_type.is_record_type()
&& !into_type.is_ref_cursor_type()) {
OZ (generator_.extract_datum_from_objparam(
p_result_obj, into_expr->get_result_type().get_type(), datum));
OZ (generator_.get_helper().create_store(datum, var_addr));
}
if (OB_SUCC(ret)
&& (ObObjAccessIdx::IS_INVALID != alloc_scop
|| into_type.is_collection_type()
|| into_type.is_record_type()
|| into_type.is_ref_cursor_type())) {
ObLLVMValue into_obj;
OZ (generator_.extract_objparam_from_context(generator_.get_vars().at(generator_.CTX_IDX), idx, into_obj));
if (into_type.is_obj_type()) {
OZ (generator_.extract_datum_ptr_from_objparam(
into_obj, into_expr->get_result_type().get_type(), dest_datum));
OZ (generator_.extract_datum_ptr_from_objparam(
p_result_obj, into_expr->get_result_type().get_type(), src_datum));
OZ (into_type.generate_copy(generator_,
*ns,
allocator,
src_datum,
dest_datum,
s.get_block()->in_notfound(),
s.get_block()->in_warning(),
package_id));
} else if (into_type.is_ref_cursor_type()) {
OZ (generator_.extract_datum_ptr_from_objparam(
p_result_obj, into_expr->get_result_type().get_type(), src_datum));
#ifndef NDEBUG
OZ (generator_.generate_debug(ObString("ref cursor data"), src_datum));
#endif
// ref cursor 主要调用ref cursortype的generate copy函数,处理cursor的引用计数
OZ (into_type.generate_copy(generator_,
*ns,
allocator,
src_datum,
var_addr,
s.get_block()->in_notfound(),
s.get_block()->in_warning(),
package_id));
} else {
OZ (generator_.extract_obobj_ptr_from_objparam(p_result_obj, src_datum));
OZ (generator_.extract_obobj_ptr_from_objparam(into_obj, dest_datum));
OZ (into_type.generate_copy(generator_,
*ns,
allocator,
src_datum,
dest_datum,
s.get_block()->in_notfound(),
s.get_block()->in_warning(),
package_id));
}
}
// 处理Nocopy参数
if (OB_SUCC(ret) && generator_.get_param_size() > 0) {
ObSEArray<ObLLVMValue, 2> args;
ObLLVMValue ret_err;
ObLLVMValue idx_value;
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.get_helper().get_int64(idx, idx_value));
OZ (args.push_back(idx_value));
OZ (generator_.get_helper().create_call(
ObString("spi_process_nocopy_params"),
generator_.get_spi_service().spi_process_nocopy_params_,
args, ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
}
}
//Sys Var/User Var or PKG Basic Variables or Subprogram Basic Variables
} else if (into_expr->is_sys_func_expr()) {
CK (OB_NOT_NULL(value_expr));
OZ (generator_.generate_set_variable(s.get_into_index(i),
p_result_obj,
T_DEFAULT == value_expr->get_expr_type(),
s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
} else if (into_expr->is_obj_access_expr()) { //ADT
ObLLVMValue into_address;
ObPLDataType final_type;
OZ (generator_.generate_expr(s.get_into_index(i), s, OB_INVALID_INDEX, into_address));
OZ (static_cast<const ObObjAccessRawExpr*>(into_expr)->get_final_type(final_type));
if (s.get_value_index(i) != PL_CONSTRUCT_COLLECTION) {
OZ (generator_.generate_check_not_null(s, final_type.get_not_null(), p_result_obj));
}
if (OB_FAIL(ret)) {
} else if (final_type.is_obj_type()) {
OZ (generator_.extract_extend_from_objparam(into_address, final_type, dest_datum));
if (ObObjAccessIdx::IS_INVALID != alloc_scop) {
OZ (generator_.extract_datum_ptr_from_objparam(
p_result_obj,
into_expr->get_result_type().get_type(),
src_datum));
OZ (final_type.generate_copy(generator_,
*ns,
allocator,
src_datum,
dest_datum,
s.get_block()->in_notfound(),
s.get_block()->in_warning(),
package_id));
} else {
ObLLVMValue ori_meta_p, into_meta_p, ori_meta;
ObLLVMValue ori_accuracy_p, into_accuracy_p, ori_accuracy;
OZ (generator_.extract_datum_from_objparam(
p_result_obj, into_expr->get_result_type().get_type(), src_datum));
OZ (generator_.get_helper().create_store(src_datum, dest_datum));
OZ (generator_.extract_meta_ptr_from_objparam(p_result_obj, ori_meta_p));
OZ (generator_.extract_meta_ptr_from_objparam(into_address, into_meta_p));
OZ (generator_.get_helper().create_load(ObString("load_meta"), ori_meta_p, ori_meta));
OZ (generator_.get_helper().create_store(ori_meta, into_meta_p));
OZ (generator_.extract_accuracy_ptr_from_objparam(p_result_obj, ori_accuracy_p));
OZ (generator_.extract_accuracy_ptr_from_objparam(into_address, into_accuracy_p));
OZ (generator_.get_helper().create_load(ObString("load_accuracy"), ori_accuracy_p, ori_accuracy));
OZ (generator_.get_helper().create_store(ori_accuracy, into_accuracy_p));
}
} else {
// 设置复杂类型的ID
ObLLVMValue dest;
ObLLVMValue src;
OZ (generator_.extract_accuracy_ptr_from_objparam(into_address, dest));
OZ (generator_.get_helper().get_int64(final_type.get_user_type_id(), src));
OZ (generator_.get_helper().create_store(src, dest));
if (OB_FAIL(ret)) {
} else if (PL_CONSTRUCT_COLLECTION == s.get_value_index(i)){//右值是个数组的构造函数走SPI初始化函数
ObSEArray<ObLLVMValue, 2> args;
ObLLVMValue ret_err;
ObLLVMValue v_package_id;
CK (final_type.is_collection_type());
OZ (generator_.get_helper().get_int64(package_id, v_package_id));
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (args.push_back(v_package_id));
OZ (args.push_back(into_address));
OZ (generator_.get_helper().create_call(
ObString("spi_construct_collection"),
generator_.get_spi_service().spi_construct_collection_,
args, ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
} else {
OZ (generator_.extract_extend_from_objparam(into_address,
final_type,
dest_datum));
if (OB_SUCC(ret) && final_type.is_collection_type()) { //如果是表,用表自带的allocator
OZ (generator_.extract_allocator_from_collection(dest_datum,allocator),
*into_expr);
}
CK (OB_NOT_NULL(value_expr));
// 这儿的dest_datum不一定是一个复杂类型,它可能是一个null
// 如 rec := null这样的赋值,这个时候如果extract_extend,结果的类型其实是错误的。
// 所以这儿需要区分一下。
if (OB_FAIL(ret)) {
} else if (T_NULL == value_expr->get_expr_type()) {
OZ (final_type.generate_assign_with_null(generator_,
*ns,
allocator,
dest_datum));
} else {
OZ (generator_.extract_obobj_ptr_from_objparam(p_result_obj, src_datum));
OZ (generator_.extract_obobj_ptr_from_objparam(into_address, dest_datum));
OZ (final_type.generate_copy(generator_,
*ns,
allocator,
src_datum,
dest_datum,
s.get_block()->in_notfound(),
s.get_block()->in_warning(),
package_id));
OZ (generator_.generate_debug("print dest obj2", dest_datum));
OZ (generator_.generate_debug("print dest objparam2", into_address));
}
}
}
if (OB_SUCC(ret) && package_id != OB_INVALID_ID && var_idx != OB_INVALID_ID) {
OZ (generator_.generate_update_package_changed_info(s, package_id, var_idx));
}
} else {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("Invalid into expr type", K(*into_expr), K(ret));
}
}
}
}
}
OZ (generator_.get_helper().stack_restore(stack));
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLIfStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("faile to create goto label", K(ret));
} else {
ObLLVMBasicBlock continue_branch;
ObLLVMBasicBlock then_branch;
ObLLVMBasicBlock else_branch;
ObLLVMBasicBlock current = generator_.get_current();
if (OB_FAIL(generator_.get_helper().create_block(ObString("continue"), generator_.get_func(), continue_branch))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("then"), generator_.get_func(), then_branch))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(generator_.set_current(then_branch))) {
LOG_WARN("failed to set current", K(ret));
} else {
if (OB_ISNULL(s.get_then())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("a if must have then body", K(s.get_then()), K(ret));
} else if (OB_FAIL(visit(*s.get_then()))) {
LOG_WARN("failed to visit then clause", K(ret));
} else {
if (OB_FAIL(generator_.finish_current(continue_branch))) {
LOG_WARN("failed to finish current", K(ret));
} else {
if (NULL == s.get_else()) {
else_branch = continue_branch;
} else {
if (OB_FAIL(generator_.get_helper().create_block(ObString("else"),
generator_.get_func(),
else_branch))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(generator_.set_current(else_branch))) {
LOG_WARN("failed to set current", K(ret));
} else if (OB_FAIL(visit(*s.get_else()))) {
LOG_WARN("failed to visit else clause", K(ret));
} else if (OB_FAIL(generator_.finish_current(continue_branch))) {
LOG_WARN("failed to finish current", K(ret));
} else { /*do nothing*/ }
}
}
}
if (OB_SUCC(ret)) {
if (OB_FAIL(generator_.set_current(current))) {
LOG_WARN("failed to set current", K(ret));
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else {
const sql::ObRawExpr *expr = s.get_cond_expr();
if (OB_ISNULL(expr)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("a assign stmt must has expr", K(expr), K(ret));
} else {
ObLLVMValue p_result_obj;
ObLLVMValue result;
ObLLVMValue is_false;
if (OB_FAIL(generator_.generate_expr(s.get_cond(), s, OB_INVALID_INDEX,
p_result_obj))) {
LOG_WARN("failed to generate calc_expr func", K(ret));
} else if (OB_FAIL(generator_.extract_value_from_objparam(p_result_obj,
expr->get_data_type(),
result))) {
LOG_WARN("failed to extract_value_from_objparam", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_icmp_eq(result, FALSE, is_false))) {
LOG_WARN("failed to create_icmp_eq", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_cond_br(is_false, else_branch, then_branch))) {
LOG_WARN("failed to create_cond_br", K(ret));
} else if (OB_FAIL(generator_.set_current(continue_branch))) {
LOG_WARN("failed to set current", K(ret));
} else { /*do nothing*/ }
}
}
}
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLLeaveStmt &s)
{
return generator_.generate_loop_control(s);
}
int ObPLCodeGenerateVisitor::visit(const ObPLIterateStmt &s)
{
return generator_.generate_loop_control(s);
}
int ObPLCodeGenerateVisitor::visit(const ObPLWhileStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("faile to generate goto lab", K(ret));
} else {
ObLLVMBasicBlock while_begin;
ObLLVMBasicBlock continue_begin;
ObLLVMBasicBlock do_body;
ObLLVMBasicBlock alter_while;
if (OB_FAIL(generator_.get_helper().create_block(ObString("while_begin"), generator_.get_func(), while_begin))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("continue_begin"),
generator_.get_func(), continue_begin))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("do_body"), generator_.get_func(), do_body))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("after_while"), generator_.get_func(), alter_while))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (s.has_label() && OB_FAIL(generator_.set_label(s.get_label(), s.get_level(), while_begin, alter_while))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(while_begin))) {
LOG_WARN("failed to create_br", K(ret));
} else if (OB_FAIL(generator_.set_current(while_begin))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else {
ObLLVMValue p_result_obj;
ObLLVMValue result;
ObLLVMValue is_false;
ObLLVMValue count_value;
ObLLVMValue stack;
if (OB_ISNULL(s.get_body()) || OB_ISNULL(s.get_cond_expr())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("a if must have then body", K(s), K(s.get_body()), K(s.get_cond()), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_ialloca(ObString("count_value"), ObIntType, 0, count_value))) {
LOG_WARN("failed to create_ialloca", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(continue_begin))) {
LOG_WARN("failed to create_br", K(ret));
} else if (OB_FAIL(generator_.set_current(continue_begin))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().stack_save(stack))) {
LOG_WARN("failed to stack_save", K(ret));
} else if (OB_FAIL(generator_.generate_expr(s.get_cond(), s, OB_INVALID_INDEX,
p_result_obj))) {
LOG_WARN("failed to generate calc_expr func", K(ret));
} else if (OB_FAIL(generator_.extract_value_from_objparam(p_result_obj, s.get_cond_expr()->get_data_type(), result))) {
LOG_WARN("failed to extract_value_from_objparam", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_icmp_eq(result, FALSE, is_false))) {
LOG_WARN("failed to create_icmp_eq", K(ret));
} else if (OB_FAIL(generator_.get_helper().stack_restore(stack))) {
LOG_WARN("failed to stack_restore", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_cond_br(is_false, alter_while, do_body))) {
LOG_WARN("failed to create_cond_br", K(ret));
} else if (OB_FAIL(generator_.set_current(do_body))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else {
if (OB_FAIL(generator_.get_helper().stack_save(stack))) {
LOG_WARN("failed to stack_save", K(ret));
} else if (OB_FAIL(generator_.set_loop(stack, s.get_level(), continue_begin, alter_while))) {
LOG_WARN("failed to set loop stack", K(ret));
} else if (OB_FAIL(SMART_CALL(generate(*s.get_body())))) {
LOG_WARN("failed to generate exception body", K(ret));
} else if (OB_FAIL(generator_.reset_loop())) {
LOG_WARN("failed to reset loop stack", K(ret));
} else if (NULL != generator_.get_current().get_v()) {
if (OB_FAIL(generator_.generate_early_exit(count_value, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()))) {
LOG_WARN("failed to generate calc_expr func", K(ret));
} else if (OB_FAIL(generator_.generate_expr(s.get_cond(), s, OB_INVALID_INDEX,
p_result_obj))) {
LOG_WARN("failed to generate calc_expr func", K(ret));
} else if (OB_FAIL(generator_.extract_value_from_objparam(p_result_obj, s.get_cond_expr()->get_data_type(), result))) {
LOG_WARN("failed to extract_value_from_objparam", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_icmp_eq(result, FALSE, is_false))) {
LOG_WARN("failed to create_icmp_eq", K(ret));
} else if (OB_FAIL(generator_.get_helper().stack_restore(stack))) {
LOG_WARN("failed to stack_restore", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_cond_br(is_false, alter_while, do_body))) {
LOG_WARN("failed to create_cond_br", K(ret));
} else { /*do nothing*/ }
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
if (OB_FAIL(generator_.set_current(alter_while))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else if (s.has_label() && OB_FAIL(generator_.reset_label())) {
LOG_WARN("failed to reset_label", K(s), K(ret));
} else { /*do nothing*/ }
}
}
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLForLoopStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto labe", K(ret));
} else {
ObLLVMBasicBlock forloop_begin;
ObLLVMBasicBlock forloop_body;
ObLLVMBasicBlock forloop_continue;
ObLLVMBasicBlock forloop_end;
OZ (generator_.get_helper().create_block(ObString("forloop_begin"), generator_.get_func(), forloop_begin));
OZ (generator_.get_helper().create_block(ObString("forloop_body"), generator_.get_func(), forloop_body));
OZ (generator_.get_helper().create_block(ObString("forloop_continue"), generator_.get_func(), forloop_continue));
OZ (generator_.get_helper().create_block(ObString("forloop_end"), generator_.get_func(), forloop_end));
if (s.has_label()) {
OZ (generator_.set_label(s.get_label(), s.get_level(), forloop_continue, forloop_end));
}
OZ (generator_.get_helper().create_br(forloop_begin));
OZ (generator_.set_current(forloop_begin));
if (OB_SUCC(ret)) {
ObLLVMValue count_value;
ObLLVMValue p_lower_obj, p_upper_obj, p_index_obj, p_index_value;
ObLLVMValue lower_obj, upper_obj, index_obj;
ObLLVMValue lower_value, upper_value, index_value;
ObLLVMValue is_true, stack;
ObLLVMValue allocator, into_obj, src_datum, dest_datum;
ObLLVMType ir_type;
const ObPLVar *var = s.get_index_var();
// 生成INDEX指针
CK (OB_NOT_NULL(var));
OZ (generator_.get_helper().create_ialloca(ObString("count_value"), ObIntType, 0, count_value));
OZ (generator_.get_llvm_type(var->get_type(), ir_type));
OZ (generator_.get_helper().create_alloca(var->get_name(), ir_type, p_index_obj));
OZ (generator_.extract_allocator_from_context(generator_.get_vars().at(generator_.CTX_IDX), allocator));
OZ (generator_.extract_objparam_from_context(generator_.get_vars().at(generator_.CTX_IDX), s.get_ident(), into_obj));
OZ (generator_.extract_datum_ptr_from_objparam(into_obj, var->get_type().get_obj_type(), dest_datum));
OX (generator_.get_vars().at(s.get_ident() + generator_.USER_ARG_OFFSET) = p_index_obj);
// 计算上界与下界
if (OB_SUCC(ret)) {
OZ (generator_.generate_bound_and_check(s, s.get_is_forall(), lower_value, upper_value, lower_obj, upper_obj, forloop_end));
}
// 给INDEX附初始值,并存储进param_stroe
if (OB_SUCC(ret)) {
if (!s.get_reverse()) {
// 下标从lower bound开始
OZ (generator_.get_helper().create_store(lower_obj, p_index_obj));
OZ (generator_.get_helper().create_store(lower_obj, dest_datum));
} else {
// 下标从upper bound开始
OZ (generator_.get_helper().create_store(upper_obj, p_index_obj));
OZ (generator_.get_helper().create_store(upper_obj, dest_datum));
}
}
if (s.is_values_bound()) {
ObLLVMValue p_value;
OZ (generator_.generate_expr(s.get_value(), s, s.get_ident(), p_value));
}
// 跳转到loop body
if (OB_SUCC(ret)) {
OZ (generator_.get_helper().create_br(forloop_body));
OZ (generator_.set_current(forloop_body));
}
// 生成循环体
if (OB_SUCC(ret)) {
OZ (generator_.get_helper().stack_save(stack));
OZ (generator_.set_loop(stack, s.get_level(), forloop_continue, forloop_end));
OZ (generator_.generate_early_exit(count_value, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
OZ (SMART_CALL(generate(*s.get_body())));
OZ (generator_.reset_loop());
if (OB_SUCC(ret)
&& NULL != generator_.get_current().get_v()
&& generator_.get_current().get_v() != generator_.get_exit().get_v()) {
OZ (generator_.set_current(generator_.get_current()));
OZ (generator_.get_helper().create_br(forloop_continue));
}
}
OZ (generator_.set_current(forloop_continue));
OZ (generator_.generate_next_and_check(s,
p_index_obj,
p_index_value,
index_obj,
index_value,
dest_datum,
lower_value,
upper_value,
is_true));
// 根据条件跳转
if (OB_SUCC(ret)) {
OZ (generator_.get_helper().stack_restore(stack));
OZ (generator_.get_helper().create_cond_br(is_true, forloop_body, forloop_end));
}
// forloop_end
if (OB_SUCC(ret)) {
OZ (generator_.set_current(forloop_end));
if (s.has_label()) {
OZ (generator_.reset_label());
}
}
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLCursorForLoopStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto labe", K(ret));
} else {
ObLLVMBasicBlock cursor_forloop_begin;
ObLLVMBasicBlock cursor_forloop_body;
ObLLVMBasicBlock cursor_forloop_end;
ObLLVMBasicBlock cursor_forloop_fetch;
ObLLVMBasicBlock cursor_forloop_check_success;
ObLLVMValue ret_err;
ObLLVMValue count_value;
if (OB_FAIL(generator_.get_helper().create_block(ObString("cursor_forloop_begin"), generator_.get_func(), cursor_forloop_begin))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("cursor_forloop_body"), generator_.get_func(), cursor_forloop_body))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("cursor_forloop_end"), generator_.get_func(), cursor_forloop_end))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("cursor_forloop_fetch"), generator_.get_func(), cursor_forloop_fetch))) {
LOG_WARN("failed to craete block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("cursor_forloop_check_success"), generator_.get_func(), cursor_forloop_check_success))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (s.has_label() && OB_FAIL(generator_.set_label(s.get_label(), s.get_level(), cursor_forloop_begin, cursor_forloop_end))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(cursor_forloop_begin))) {
LOG_WARN("failed to create_br", K(ret));
}
if (OB_SUCC(ret)) {
const ObUserDefinedType *user_defined_type = s.get_user_type();
const ObPLVar *var = s.get_index_var();
if (OB_FAIL(generator_.set_current(cursor_forloop_begin))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_ialloca(ObString("count_value"), ObIntType, 0, count_value))) {
LOG_WARN("failed to create_ialloca", K(ret));
} else if (OB_ISNULL(user_defined_type)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("user defined type is null");
} else if (OB_FAIL(generator_.generate_user_type(*user_defined_type))) {
LOG_WARN("failed to generate user type", K(user_defined_type->get_type()));
} else if (OB_ISNULL(var)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("user var is null", K(s), K(ret));
} else {
if (var->get_type().is_composite_type()) {
ObSEArray<ObLLVMValue, 3> args;
ObLLVMValue var_idx, init_value, var_value, extend_value;
ObLLVMValue ret_err;
ObLLVMValue var_type, type_id;
ObLLVMValue null_int;
int64_t init_size = 0;
CK (OB_NOT_NULL(s.get_namespace()));
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.get_helper().get_int8(var->get_type().get_type(), var_type));
OZ (args.push_back(var_type));
OZ (generator_.get_helper().get_int64(var->get_type().get_user_type_id(), type_id));
OZ (args.push_back(type_id));
OZ (generator_.get_helper().get_int64(s.get_index_index(), var_idx));
OZ (args.push_back(var_idx));
OZ (s.get_namespace()->get_size(PL_TYPE_INIT_SIZE, var->get_type(), init_size));
OZ (generator_.get_helper().get_int32(init_size, init_value));
OZ (args.push_back(init_value));
OZ (generator_.generate_null_pointer(ObIntType, null_int));
OZ (args.push_back(null_int));
OZ (generator_.get_helper().create_call(ObString("spi_alloc_complex_var"),
generator_.get_spi_service().spi_alloc_complex_var_,
args,
ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
OZ (generator_.extract_objparam_from_context(
generator_.get_vars().at(generator_.CTX_IDX),
s.get_index_index(),
var_value));
OZ (generator_.extract_extend_from_objparam(var_value, var->get_type(), extend_value));
OZ (var->get_type().generate_construct(generator_, *s.get_namespace(),
extend_value, &s));
OZ (generator_.set_var_addr_to_param_store(s.get_index_index(), extend_value, init_value));
generator_.get_vars().at(s.get_index_index() + generator_.USER_ARG_OFFSET) = extend_value;
} else {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("cursor index is not a record", K(*var), K(ret));
}
if (OB_SUCC(ret)) {
if (OB_ISNULL(s.get_cursor())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("cursor is null", K(s.get_cursor_index()), K(s.get_cursor()), K(ret));
} else if (s.get_need_declare()
&& generator_.generate_declare_cursor(static_cast<const ObPLStmt&>(s),
s.get_cursor_index())) {
LOG_WARN("failed to generate declare cursor", K(ret));
} else if (OB_FAIL(generator_.generate_open(static_cast<const ObPLStmt&>(s),
s.get_cursor()->get_value(),
s.get_cursor()->get_package_id(),
s.get_cursor()->get_routine_id(),
s.get_index()))) {
LOG_WARN("failed to generate_open", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(cursor_forloop_fetch))) {
LOG_WARN("failed to create_br cursor_forloop_fetch", K(ret));
}
}
}
}
if (OB_SUCC(ret)) {
ObLLVMValue is_not_found;
ObLLVMValue stack;
CK (OB_NOT_NULL(s.get_cursor()));
OZ (generator_.set_current(cursor_forloop_fetch));
OZ (generator_.get_helper().stack_save(stack));
OZ (generator_.generate_fetch(static_cast<const ObPLStmt&>(s),
static_cast<const ObPLInto&>(s),
s.get_cursor()->get_package_id(),
s.get_cursor()->get_routine_id(),
s.get_index(),
static_cast<int64_t>(INT64_MAX),
s.get_user_type(),
s.get_expand_user_types(),
ret_err));
OZ (generator_.get_helper().create_icmp_eq(ret_err, OB_READ_NOTHING, is_not_found));
OZ (generator_.get_helper().stack_restore(stack));
OZ (generator_.get_helper().create_cond_br(is_not_found, cursor_forloop_end, cursor_forloop_check_success));
}
if (OB_SUCC(ret)) {
ObLLVMValue stack;
OZ (generator_.set_current(cursor_forloop_check_success));
OZ (generator_.get_helper().stack_save(stack));
OZ (generator_.check_success(ret_err,
s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
OZ (generator_.get_helper().stack_restore(stack));
OZ (generator_.get_helper().create_br(cursor_forloop_body));
}
// 生成循环体
if (OB_SUCC(ret)) {
ObLLVMValue stack;
if (OB_FAIL(generator_.set_current(cursor_forloop_body))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().stack_save(stack))) {
LOG_WARN("failed to stack_save", K(ret));
} else if (OB_FAIL(generator_.set_loop(stack,
s.get_level(),
cursor_forloop_fetch,
cursor_forloop_end,
&s))) {
LOG_WARN("failed to set loop stack", K(ret));
} else if (OB_FAIL(SMART_CALL(generate(*s.get_body())))) {
LOG_WARN("failed to generate exception body", K(ret));
} else if (OB_FAIL(generator_.reset_loop())) {
LOG_WARN("failed to reset loop stack", K(ret));
} else { /*do nothing*/ }
// 重置堆栈,并且跳转到循环头
if (OB_SUCC(ret)
&& NULL != generator_.get_current().get_v()
&& generator_.get_current().get_v() != generator_.get_exit().get_v()) {
if (OB_FAIL(generator_.set_current(generator_.get_current()))) {
LOG_WARN("failed to set current", K(ret));
} else if (OB_FAIL(generator_.generate_early_exit(count_value, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()))) {
LOG_WARN("failed to generate calc_expr func", K(ret));
} else if (OB_FAIL(generator_.get_helper().stack_restore(stack))) {
LOG_WARN("failed to stack_restore", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(cursor_forloop_fetch))) {
LOG_WARN("failed to create_br", K(ret));
}
}
}
// 循环结束, 关闭游标
if (OB_SUCC(ret)) {
OZ (generator_.set_current(cursor_forloop_end));
CK (OB_NOT_NULL(s.get_cursor()));
OZ (generator_.generate_close(static_cast<const ObPLStmt&>(s),
s.get_cursor()->get_package_id(),
s.get_cursor()->get_routine_id(),
s.get_index()));
if (s.has_label()) {
OZ (generator_.reset_label());
}
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLForAllStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
// 控制流已断, 后面的语句不再处理
} else {
OZ (generator_.get_helper().set_insert_point(generator_.get_current()));
OZ (generator_.set_debug_location(s));
OZ (generator_.generate_goto_label(s));
}
// 设置隐式游标在forall语句中
if (OB_SUCC(ret) && OB_NOT_NULL(generator_.get_current().get_v())) {
ObSEArray<ObLLVMValue, 2> args;
ObLLVMValue ret_err;
ObLLVMValue is_save_exception;
OZ (generator_.get_helper().get_int8(static_cast<int64_t>(s.get_save_exception()), is_save_exception));
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (args.push_back(is_save_exception));
OZ (generator_.get_helper().create_call(ObString("set_implicit_cursor_in_forall"),
generator_.get_set_implicit_cursor_in_forall_func(),
args,
ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
}
// 执行forall,forall的循环体中只会有一句DML,不会出现跳转,因此这里不判断控制流是否断开
if (OB_SUCC(ret) && OB_NOT_NULL(generator_.get_current().get_v())) {
const ObPLForLoopStmt& for_loop = static_cast<const ObPLForLoopStmt&>(s);
if (OB_NOT_NULL(s.get_sql_stmt()) && !s.get_save_exception()) {
ObLLVMValue lower, upper, lower_obj, upper_obj, p_index_obj;
ObLLVMValue ret_err, is_need_rollback;
ObLLVMType ir_type;
ObLLVMBasicBlock illegal_block, after_block, rollback_block, not_rollback_block;
ObPLSqlStmt *sql_stmt = const_cast<ObPLSqlStmt*>(s.get_sql_stmt());
const ObPLVar *var = s.get_index_var();
sql_stmt->set_forall_sql(true);
CK (OB_NOT_NULL(var));
OZ (generator_.get_helper().create_block(ObString("illegal_block"), generator_.get_func(), illegal_block));
OZ (generator_.get_helper().create_block(ObString("after_block"), generator_.get_func(), after_block));
OZ (generator_.get_helper().create_block(ObString("rollback_block"), generator_.get_func(), rollback_block));
OZ (generator_.get_helper().create_block(ObString("not_rollback_block"), generator_.get_func(), not_rollback_block));
OZ (generator_.get_llvm_type(var->get_type(), ir_type));
OZ (generator_.get_helper().create_alloca(var->get_name(), ir_type, p_index_obj));
OX (generator_.get_vars().at(s.get_ident() + generator_.USER_ARG_OFFSET) = p_index_obj);
OZ (generator_.generate_bound_and_check(s, true, lower, upper, lower_obj, upper_obj, illegal_block));
OZ (generator_.generate_sql(*(s.get_sql_stmt()), ret_err));
OZ (generator_.get_helper().create_icmp_eq(ret_err, OB_BATCHED_MULTI_STMT_ROLLBACK, is_need_rollback));
OZ (generator_.get_helper().create_cond_br(is_need_rollback, rollback_block, not_rollback_block));
OZ (generator_.set_current(not_rollback_block));
OZ (generator_.generate_after_sql(*(s.get_sql_stmt()), ret_err));
OZ (generator_.get_helper().create_br(after_block));
OZ (generator_.set_current(rollback_block));
sql_stmt->set_forall_sql(false);
OZ (visit(for_loop));
OZ (generator_.get_helper().create_br(after_block));
OZ (generator_.get_helper().set_insert_point(illegal_block));
OZ (generator_.get_helper().create_br(after_block));
OZ (generator_.set_current(after_block));
} else {
OZ (visit(for_loop));
}
}
// 设置隐式游标不在forall环境中
if (OB_SUCC(ret) && OB_NOT_NULL(generator_.get_current().get_v())) {
ObSEArray<ObLLVMValue, 2> args;
ObLLVMValue ret_err;
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.get_helper().create_call(ObString("unset_implicit_cursor_in_forall"),
generator_.get_unset_implicit_cursor_in_forall_func(),
args,
ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLRepeatStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {
ObLLVMBasicBlock repeat;
ObLLVMBasicBlock alter_repeat;
ObLLVMValue count_value;
if (OB_FAIL(generator_.get_helper().create_block(ObString("repeat"), generator_.get_func(), repeat))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("after_repeat"), generator_.get_func(), alter_repeat))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (s.has_label() && OB_FAIL(generator_.set_label(s.get_label(), s.get_level(), repeat, alter_repeat))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_ialloca(ObString("count_value"), ObIntType, 0, count_value))) {
LOG_WARN("failed to create_ialloca", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(repeat))) {
LOG_WARN("failed to create_br", K(ret));
} else if (OB_FAIL(generator_.set_current(repeat))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else {
ObLLVMValue stack;
ObLLVMValue p_result_obj;
ObLLVMValue result;
ObLLVMValue is_false;
if (OB_FAIL(generator_.get_helper().stack_save(stack))) {
LOG_WARN("failed to stack_save", K(ret));
} else if (OB_ISNULL(s.get_body()) || OB_ISNULL(s.get_cond_expr())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("a if must have then body", K(s), K(s.get_body()), K(s.get_cond()), K(ret));
} else if (OB_FAIL(generator_.set_loop(stack, s.get_level(), repeat, alter_repeat))) {
LOG_WARN("failed to set loop stack", K(ret));
} else if (OB_FAIL(SMART_CALL(generate(*s.get_body())))) {
LOG_WARN("failed to generate exception body", K(ret));
} else if (OB_FAIL(generator_.reset_loop())) {
LOG_WARN("failed to reset loop stack", K(ret));
} else if (NULL != generator_.get_current().get_v()) {
if (OB_FAIL(generator_.generate_early_exit(count_value, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()))) {
LOG_WARN("failed to generate calc_expr func", K(ret));
} else if (OB_FAIL(generator_.generate_expr(s.get_cond(), s, OB_INVALID_INDEX,
p_result_obj))) {
LOG_WARN("failed to generate calc_expr func", K(ret));
} else if (OB_FAIL(generator_.extract_value_from_objparam(p_result_obj, s.get_cond_expr()->get_data_type(), result))) {
LOG_WARN("failed to extract_value_from_objparam", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_icmp_eq(result, FALSE, is_false))) {
LOG_WARN("failed to create_icmp_eq", K(ret));
} else if (OB_FAIL(generator_.get_helper().stack_restore(stack))) {
LOG_WARN("failed to stack_restore", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_cond_br(is_false, repeat, alter_repeat))) {
LOG_WARN("failed to create_cond_br", K(ret));
}
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
if (OB_FAIL(generator_.set_current(alter_repeat))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else if (s.has_label() && OB_FAIL(generator_.reset_label())) {
LOG_WARN("failed to reset_label", K(s), K(ret));
} else { /*do nothing*/ }
}
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLLoopStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {
ObLLVMBasicBlock loop;
ObLLVMBasicBlock alter_loop;
ObLLVMValue count_value;
if (OB_FAIL(generator_.get_helper().create_block(ObString("loop"), generator_.get_func(), loop))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("after_loop"), generator_.get_func(), alter_loop))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (s.has_label() && OB_FAIL(generator_.set_label(s.get_label(), s.get_level(), loop, alter_loop))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_ialloca(ObString("count_value"), ObIntType, 0, count_value))) {
LOG_WARN("failed to create_ialloca", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(loop))) {
LOG_WARN("failed to create_br", K(ret));
} else if (OB_FAIL(generator_.set_current(loop))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else {
ObLLVMValue stack;
if (OB_FAIL(generator_.get_helper().stack_save(stack))) {
LOG_WARN("failed to stack_save", K(ret));
} else if (OB_FAIL(generator_.set_loop(stack, s.get_level(), loop, alter_loop))) {
LOG_WARN("failed to set loop stack", K(ret));
} else if (OB_ISNULL(s.get_body())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("a if must have valid body", K(s), K(s.get_body()), K(ret));
} else if (OB_FAIL(SMART_CALL(generate(*s.get_body())))) {
LOG_WARN("failed to generate exception body", K(ret));
} else if (OB_FAIL(generator_.reset_loop())) {
LOG_WARN("failed to reset loop stack", K(ret));
} else if (NULL != generator_.get_current().get_v()) {
if (OB_FAIL(generator_.generate_early_exit(count_value, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()))) {
LOG_WARN("failed to generate calc_expr func", K(ret));
} else if (OB_FAIL(generator_.get_helper().stack_restore(stack))) {
LOG_WARN("failed to stack_restore", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(loop))) {
LOG_WARN("failed to create_cond_br", K(ret));
} else { /*do nothing*/ }
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
if (OB_FAIL(generator_.set_current(alter_loop))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else if (s.has_label() && OB_FAIL(generator_.reset_label())) {
LOG_WARN("failed to reset_label", K(s), K(ret));
} else { /*do nothing*/ }
}
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLReturnStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {
ObLLVMBasicBlock normal_return;
if (OB_FAIL(generator_.get_helper().create_block(ObString("return"), generator_.get_func(), normal_return))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(normal_return))) {
LOG_WARN("failed to create_br", K(ret));
} else if (OB_FAIL(generator_.set_current(normal_return))) {
LOG_WARN("failed to set current", K(s), K(ret));
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (s.get_ret() != OB_INVALID_INDEX) {
ObLLVMValue p_result_obj;
ObLLVMValue result;
ObLLVMValue p_result;
OZ (generator_.generate_expr(s.get_ret(), s, OB_INVALID_INDEX, p_result_obj));
if (OB_SUCC(ret) && generator_.get_ast().get_ret_type().is_composite_type()) {
//return NULL as UDT means returning a uninitialized UDT object
ObLLVMBasicBlock null_branch;
ObLLVMBasicBlock final_branch;
ObLLVMValue p_type_value;
ObLLVMValue type_value;
ObLLVMValue is_null;
ObLLVMValue p_obj_value;
ObLLVMValue obj_value;
const ObUserDefinedType *user_type = NULL;
OZ (generator_.get_helper().create_block(ObString("null_branch"),
generator_.get_func(),
null_branch));
OZ (generator_.get_helper().create_block(ObString("final_branch"),
generator_.get_func(),
final_branch));
OZ (generator_.extract_type_ptr_from_objparam(p_result_obj, p_type_value));
OZ (generator_.get_helper().create_load(ObString("load_type"), p_type_value, type_value));
OZ (generator_.get_helper().create_icmp_eq(type_value, ObNullType, is_null));
OZ (generator_.get_helper().create_cond_br(is_null, null_branch, final_branch));
//null branch
OZ (generator_.set_current(null_branch));
OZ (s.get_namespace()->get_user_type(generator_.get_ast().get_ret_type().get_user_type_id(),
user_type));
ObSEArray<ObLLVMValue, 3> args;
ObLLVMType int_type;
jit::ObLLVMType ir_type, ir_ptr_type;
ObLLVMValue var_idx, init_value, extend_ptr, extend_value, composite_ptr, p_obj;
ObLLVMValue ret_err;
ObLLVMValue var_type, type_id;
int64_t init_size = 0;
// Step 1: 初始化内存
CK (OB_NOT_NULL(s.get_namespace()));
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.get_helper().get_int8(user_type->get_type(), var_type));
OZ (args.push_back(var_type));
OZ (generator_.get_helper().get_int64(
generator_.get_ast().get_ret_type().get_user_type_id(),
type_id));
OZ (args.push_back(type_id));
OZ (generator_.get_helper().get_int64(OB_INVALID_INDEX, var_idx));
OZ (args.push_back(var_idx));
OZ (user_type->get_size(*s.get_namespace(), PL_TYPE_INIT_SIZE, init_size));
OZ (generator_.get_helper().get_int32(init_size, init_value));
OZ (args.push_back(init_value));
OZ (generator_.get_helper().get_llvm_type(ObIntType, int_type));
OZ (generator_.get_helper().create_alloca(ObString("exception_type"), int_type, extend_ptr));
OZ (args.push_back(extend_ptr));
OZ (generator_.get_helper().create_call(ObString("spi_alloc_complex_var"),
generator_.get_spi_service().spi_alloc_complex_var_,
args,
ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
// Step 2: 初始化类型内容, 如Collection的rowsize,element type等
OZ (generator_.get_helper().create_load("load_extend", extend_ptr, extend_value));
OZ (generator_.get_llvm_type(*user_type, ir_type));
OZ (ir_type.get_pointer_to(ir_ptr_type));
OZ (generator_.get_helper().create_int_to_ptr(ObString("cast_extend_to_ptr"), extend_value, ir_ptr_type, composite_ptr));
OZ (user_type->ObPLDataType::generate_construct( //must call ObPLDataType's
generator_, *s.get_namespace(), composite_ptr, &s));
OZ (generator_.get_helper().create_gep(ObString("extract_obj_pointer"), p_result_obj, 0, p_obj));
OZ (generator_.generate_set_extend(p_obj, var_type, init_value, extend_value));
OZ (generator_.finish_current(final_branch));
OZ (generator_.set_current(final_branch));
}
OZ (generator_.extract_obobj_from_objparam(p_result_obj, result));
OZ (generator_.extract_result_from_context(
generator_.get_vars().at(generator_.CTX_IDX), p_result));
CK (OB_NOT_NULL(s.get_ret_expr()));
if (OB_FAIL(ret)) {
} else if (s.get_ret_expr()->is_obj_access_expr()) {
const ObObjAccessRawExpr *obj_access
= static_cast<const ObObjAccessRawExpr*>(s.get_ret_expr());
const common::ObIArray<pl::ObObjAccessIdx> *access_idxs = NULL;
ObPLDataType pl_type;
int64_t local_idx = OB_INVALID_INDEX;
CK (OB_NOT_NULL(obj_access));
CK (OB_NOT_NULL(access_idxs = &(obj_access->get_access_idxs())));
OZ (obj_access->get_final_type(pl_type));
OX (local_idx = ObObjAccessIdx::get_local_variable_idx(*access_idxs));
if (OB_FAIL(ret)) {
} else if (ObObjAccessIdx::is_local_variable(*access_idxs)
&& local_idx < access_idxs->count()
&& access_idxs->at(local_idx).elem_type_.is_collection_type()
&& pl_type.is_obj_type()) {
ObLLVMValue allocator;
ObLLVMValue src_datum;
ObLLVMValue dest_datum;
OZ (generator_.extract_allocator_from_context(
generator_.get_vars().at(generator_.CTX_IDX), allocator));
OZ (generator_.extract_datum_ptr_from_objparam(
p_result_obj, pl_type.get_obj_type(), src_datum));
OZ (generator_.extract_datum_ptr_from_objparam(
p_result, pl_type.get_obj_type(), dest_datum));
OZ (pl_type.generate_copy(generator_,
*(s.get_namespace()),
allocator,
src_datum,
dest_datum,
s.get_block()->in_notfound(),
s.get_block()->in_warning(),
OB_INVALID_ID));
} else {
OZ (generator_.get_helper().create_store(result, p_result));
}
} else {
OZ (generator_.get_helper().create_store(result, p_result));
}
if (OB_SUCC(ret) && s.is_return_ref_cursor_type()) {
ObSEArray<ObLLVMValue, 16> args;
ObLLVMValue addend;
// check ref cursor
OZ (args.push_back(generator_.get_vars()[generator_.CTX_IDX]));
OZ (args.push_back(p_result));
OZ (generator_.get_helper().get_int64(1, addend));
OZ (args.push_back(addend));
jit::ObLLVMValue ret_err;
// 这儿为啥需要对ref count +1, 因为一个被return的ref cursor,在函数block结束的时候,会被dec ref
// 那么这个时候可能会减到0, 从而导致这个被return的ref cursor被close了。
// 这个+1,在ob_expr_udf那儿会对这个ref cursor -1进行平衡操作。
OZ (generator_.get_helper().create_call(ObString("spi_add_ref_cursor_refcount"),
generator_.get_spi_service().spi_add_ref_cursor_refcount_,
args,
ret_err));
OZ (generator_.check_success(ret_err,
s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
}
}
if (OB_SUCC(ret)) {
LOG_DEBUG("generate return stmt, close cursor",
K(generator_.get_ast().get_cursor_table().get_count()),
K(s.get_stmt_id()),
K(generator_.get_ast().get_name()));
// close cursor
for (int64_t i = 0; OB_SUCC(ret) && i < generator_.get_ast().get_cursor_table().get_count(); ++i) {
const ObPLCursor *cursor = generator_.get_ast().get_cursor_table().get_cursor(i);
OZ (generator_.generate_handle_ref_cursor(cursor, s, false, false));
}
ObLLVMValue ret_value;
ObLLVMBasicBlock null_block;
if (OB_FAIL(generator_.get_helper().create_load(ObString("load_ret"), generator_.get_vars().at(generator_.RET_IDX), ret_value))) {
LOG_WARN("failed to create_load", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_ret(ret_value))) {
LOG_WARN("failed to create_ret", K(ret));
} else if (OB_FAIL(generator_.set_current(null_block))) { // Return语句会结束函数, 切断控制流, 后面的语句不再Codegen
LOG_WARN("failed to set current", K(s), K(ret));
}
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLSqlStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
// 控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {
ObLLVMValue ret_err;
OZ (generator_.generate_sql(s, ret_err));
OZ (generator_.generate_after_sql(s, ret_err));
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLExecuteStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else {
ObLLVMType int_type;
ObLLVMValue null_pointer;
ObLLVMValue int_value;
ObSEArray<ObLLVMValue, 16> args;
ObLLVMValue sql_addr;
ObLLVMValue params;
ObLLVMType params_type;
ObLLVMValue into_array_value;
ObLLVMValue into_count_value;
ObLLVMValue type_array_value;
ObLLVMValue type_count_value;
ObLLVMValue exprs_not_null_array_value;
ObLLVMValue pl_integer_range_array_value;
ObLLVMValue is_bulk;
ObLLVMValue is_returning;
ObLLVMValue ret_err;
OZ (generator_.get_helper().set_insert_point(generator_.get_current()));
OZ (generator_.set_debug_location(s));
OZ (generator_.generate_goto_label(s));
OZ (generator_.get_helper().get_llvm_type(ObIntType, int_type));
OZ (generator_.generate_null_pointer(ObIntType, null_pointer));
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX))); //PL的执行环境
/*
* 为什么sql和using同样都是表达式,sql直接传rawexpr,而using调用generate_expr传入obobjparam???
* 这是因为直接传rawexpr,由spi_execute_immediate进行计算会省掉一次spi交互。
* 而using不仅是传入参数,还可以做传出参数,所以必须用obobjparam向外传递结果。
*/
OZ (generator_.generate_pointer(generator_.get_expr(s.get_sql()), sql_addr));
OZ (args.push_back(sql_addr));
// param exprs & param count
ObLLVMType param_mode_arr_type; //param mode array elem type
ObLLVMValue param_mode_arr_value; // param mode array
ObLLVMType param_mode_arr_pointer;
if (OB_FAIL(ret)) {
} else if (s.get_using().empty()) {
OZ (generator_.get_helper().create_ptr_to_int(
ObString("cast_pointer_to_int"), null_pointer, int_type, int_value));
OZ (args.push_back(int_value));
// param mode
OZ (args.push_back(null_pointer));
} else {
OZ (generator_.get_adt_service().get_argv(s.get_using().count(), params_type));
OZ (generator_.get_helper().create_alloca(
ObString("execute_immediate_params"), params_type, params));
OZ (int_type.get_pointer_to(param_mode_arr_pointer));
OZ (ObLLVMHelper::get_array_type(int_type,
s.get_using().count(),
param_mode_arr_type));
OZ (generator_.get_helper().create_alloca(ObString("param_mode_array"),
param_mode_arr_type,
param_mode_arr_value));
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_using().count(); ++i) {
ObLLVMValue p_result_obj;
ObLLVMValue p_arg;
ObLLVMValue pp_arg;
ObLLVMValue param_mode_value;
ObLLVMValue param_mode_arr_elem;
OZ (generator_.get_helper().get_int64(s.get_using().at(i).mode_, param_mode_value));
OZ (generator_.get_helper().create_gep(ObString("extract_param_mode"),
param_mode_arr_value, i, param_mode_arr_elem));
OZ (generator_.get_helper().create_store(param_mode_value, param_mode_arr_elem));
int64_t udt_id =
generator_.get_ast().get_expr(s.get_using_index(i))->get_result_type().get_udt_id();
if (s.is_pure_out(i)) {
OZ (generator_.generate_new_objparam(p_result_obj, udt_id));
} else {
OZ (generator_.generate_expr(s.get_using_index(i), s, OB_INVALID_INDEX, p_result_obj));
}
OZ (generator_.get_helper().create_ptr_to_int(
ObString("cast_arg_to_pointer"), p_result_obj, int_type, p_arg));
OZ (generator_.extract_arg_from_argv(params, i, pp_arg));
OZ (generator_.get_helper().create_store(p_arg, pp_arg));
}
OZ (generator_.get_helper().create_ptr_to_int(
ObString("cast_pointer_to_int"), params, int_type, int_value));
OZ (args.push_back(int_value));
OZ (generator_.get_helper().create_bit_cast(ObString("cast_param_mode_arr_pointer"),
param_mode_arr_value, param_mode_arr_pointer, param_mode_arr_value));
OZ (args.push_back(param_mode_arr_value));
}
OZ (generator_.get_helper().get_int64(s.get_using().count(), int_value));
OZ (args.push_back(int_value));
//result_exprs & result_count
OZ (generator_.generate_into(s, into_array_value, into_count_value,
type_array_value, type_count_value,
exprs_not_null_array_value,
pl_integer_range_array_value,
is_bulk));
OZ (args.push_back(into_array_value));
OZ (args.push_back(into_count_value));
OZ (args.push_back(type_array_value));
OZ (args.push_back(type_count_value));
OZ (args.push_back(exprs_not_null_array_value));
OZ (args.push_back(pl_integer_range_array_value));
OZ (args.push_back(is_bulk));
OZ (generator_.get_helper().get_int8(s.get_is_returning(), is_returning));
OZ (args.push_back(is_returning));
// execution
OZ (generator_.get_helper().create_call(
ObString("spi_execute_immediate"),
generator_.get_spi_service().spi_execute_immediate_, args, ret_err));
OZ (generator_.check_success(
ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
// into result
OZ (generator_.generate_into_restore(s.get_into(), s.get_exprs(), s.get_symbol_table()));
// out result
OZ (generator_.generate_out_params(s, s.get_using(), params));
}
return ret;
}
#define COLLECTION_STMT_COMM(expr) \
const ObObjAccessRawExpr *access_expr = NULL; \
ObPLDataType type; \
const ObUserDefinedType *user_type = NULL; \
const ObPLBlockNS *ns = NULL; \
uint64_t type_id = OB_INVALID_ID; \
int64_t row_size = 0; \
int64_t filed_cnt = 0; \
uint64_t package_id = OB_INVALID_ID; \
uint64_t var_idx = OB_INVALID_ID; \
OZ (generator_.get_helper().set_insert_point(generator_.get_current())); \
CK (OB_NOT_NULL(expr)); \
CK (expr->is_obj_access_expr()); \
CK (OB_NOT_NULL(access_expr = static_cast<const ObObjAccessRawExpr*>(expr))); \
OZ (access_expr->get_final_type(type)); \
CK (OB_INVALID_ID != (type_id = type.get_user_type_id())); \
CK (OB_NOT_NULL(ns = s.get_namespace())); \
CK (OB_NOT_NULL(s.get_namespace()->get_type_table())); \
if (OB_SUCC(ret) \
&& OB_ISNULL(user_type = s.get_namespace()->get_type_table()->get_type(type_id))) { \
CK (OB_NOT_NULL(user_type \
= s.get_namespace()->get_type_table()->get_external_type(type_id))); \
} \
OZ (user_type->get_size(*s.get_namespace(), PL_TYPE_ROW_SIZE, row_size)); \
OZ (user_type->get_field_count(*s.get_namespace(), filed_cnt)); \
if (OB_SUCC(ret) \
&& ObObjAccessIdx::is_package_variable(access_expr->get_access_idxs())) { \
OZ (ObObjAccessIdx::get_package_id(access_expr, package_id, &var_idx)); \
}
int ObPLCodeGenerateVisitor::visit(const ObPLExtendStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {
COLLECTION_STMT_COMM(s.get_extend_expr());
if (OB_SUCC(ret)) {
ObLLVMValue int_value;
ObSEArray<ObLLVMValue, 16> args;
if (OB_FAIL(args.push_back(generator_.get_vars().at(generator_.CTX_IDX)))) { //PL的执行环境
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.generate_pointer(generator_.get_expr(s.get_extend()), int_value))) {
LOG_WARN("failed to generate a pointer", K(ret));
} else if (OB_FAIL(args.push_back(int_value))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.get_helper().get_int64(filed_cnt, int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(args.push_back(int_value))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.generate_pointer(generator_.get_expr(s.get_n()), int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(args.push_back(int_value))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.generate_pointer(generator_.get_expr(s.get_i()), int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(args.push_back(int_value))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.get_helper().get_int64(package_id, int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(args.push_back(int_value))) {
LOG_WARN("push_back error", K(ret));
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
ObLLVMValue ret_err;
if (OB_FAIL(generator_.get_helper().create_call(ObString("spi_extend_collection"), generator_.get_spi_service().spi_extend_collection_, args, ret_err))) {
LOG_WARN("failed to create call", K(ret));
} else if (OB_FAIL(generator_.check_success(ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()))) {
LOG_WARN("failed to check success", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret) && package_id != OB_INVALID_ID && var_idx != OB_INVALID_ID) {
OZ (generator_.generate_update_package_changed_info(s, package_id, var_idx));
}
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLTrimStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else {
COLLECTION_STMT_COMM(s.get_trim_expr());
if (OB_SUCC(ret)) {
ObLLVMValue int_value;
ObSEArray<ObLLVMValue, 4> args;
ObLLVMValue ret_err;
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.generate_pointer(generator_.get_expr(s.get_trim()), int_value));
OZ (args.push_back(int_value));
OZ (generator_.get_helper().get_int64(row_size, int_value));
OZ (args.push_back(int_value));
OZ (generator_.generate_pointer(generator_.get_expr(s.get_n()), int_value));
OZ (args.push_back(int_value));
OZ (generator_.get_helper().create_call(ObString("spi_trim_collection"),
generator_.get_spi_service().spi_trim_collection_, args, ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(),
s.get_block()->in_notfound(), s.get_block()->in_warning()));
}
if (OB_SUCC(ret) && package_id != OB_INVALID_ID && var_idx != OB_INVALID_ID) {
OZ (generator_.generate_update_package_changed_info(s, package_id, var_idx));
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLDeleteStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else {
COLLECTION_STMT_COMM(s.get_delete_expr());
if (OB_SUCC(ret)) {
ObLLVMValue int_value;
ObSEArray<ObLLVMValue, 5> args;
ObLLVMValue ret_err;
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.generate_pointer(generator_.get_expr(s.get_delete()), int_value));
OZ (args.push_back(int_value));
OZ (generator_.get_helper().get_int64(row_size, int_value));
OZ (args.push_back(int_value));
OZ (generator_.generate_pointer(generator_.get_expr(s.get_m()), int_value));
OZ (args.push_back(int_value));
OZ (generator_.generate_pointer(generator_.get_expr(s.get_n()), int_value));
OZ (args.push_back(int_value));
OZ (generator_.get_helper().create_call(ObString("spi_delete_collection"),
generator_.get_spi_service().spi_delete_collection_, args, ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(),
s.get_block()->in_notfound(), s.get_block()->in_warning()));
}
if (OB_SUCC(ret) && package_id != OB_INVALID_ID && var_idx != OB_INVALID_ID) {
OZ (generator_.generate_update_package_changed_info(s, package_id, var_idx));
}
}
return ret;
}
#undef COLLECTION_STMT_COMM
int ObPLCodeGenerateVisitor::visit(const ObPLDeclareCondStmt &s)
{
int ret = OB_SUCCESS;
UNUSED(s);
//do nothing
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLDeclareHandlerStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {
ObLLVMBasicBlock exception;
ObLLVMBasicBlock body;
ObLLVMLandingPad catch_result;
ObLLVMType landingpad_type;
ObLLVMValue int_value;
if (OB_FAIL(generator_.get_helper().create_block(ObString("exception"), generator_.get_func(), exception))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("body"), generator_.get_func(), body))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(body))) {
LOG_WARN("failed to create_br", K(ret));
} else if (OB_FAIL(generator_.get_helper().set_insert_point(exception))) {
LOG_WARN("failed to set_insert_point", K(ret));
} else if (OB_FAIL(generator_.get_adt_service().get_landingpad_result(landingpad_type))) {
LOG_WARN("failed to get_landingpad_result", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_landingpad(ObString("landingPad"), landingpad_type, catch_result))) {
LOG_WARN("failed to create_landingpad", K(ret));
} else if (OB_FAIL(catch_result.set_cleanup())) {
LOG_WARN("failed to set_cleanup", K(ret));
#ifndef NDEBUG
} else if (OB_FAIL(generator_.get_helper().get_int64(3333, int_value))) {
LOG_WARN("failed to get_int64", K(ret));
} else if (OB_FAIL(generator_.generate_debug(ObString("debug"), int_value))) {
LOG_WARN("failed to create_call", K(ret));
#endif
} else {
common::ObArray<ObLLVMGlobalVariable> condition_clauses;
common::ObArray<std::pair<ObPLConditionType, int64_t>> precedences;
ObSEArray<ObLLVMValue, 8> condition_elements;
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_handlers().count(); ++i) {
if (OB_ISNULL(s.get_handler(i).get_desc())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("handler is NULL", K(i), K(s.get_handler(i)), K(ret));
} else {
ObSqlString const_name;
for (int64_t j = 0; OB_SUCC(ret) && j < s.get_handler(i).get_desc()->get_conditions().count(); ++j) {
const_name.reset();
condition_elements.reset();
if (OB_FAIL(generator_.get_helper().get_int64(s.get_handler(i).get_desc()->get_condition(j).type_, int_value))) {
LOG_WARN("failed to get_int64", K(ret));
} else if (OB_FAIL(condition_elements.push_back(int_value))) {
LOG_WARN("push_back error", K(ret));
} else if (s.get_handler(i).get_desc()->get_condition(j).type_ >= MAX_TYPE) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("type value error", K(ret));
} else if (OB_FAIL(const_name.append(ConditionType[s.get_handler(i).get_desc()->get_condition(j).type_]))) {
LOG_WARN("append error", K(ret));
} else if (OB_FAIL(generator_.get_helper().get_int64(s.get_handler(i).get_desc()->get_condition(j).error_code_, int_value))) {
LOG_WARN("failed to get_int64", K(ret));
} else if (OB_FAIL(condition_elements.push_back(int_value))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(const_name.append_fmt("%ld%ld", s.get_handler(i).get_level(), s.get_handler(i).get_desc()->get_condition(j).error_code_))) {
LOG_WARN("append_fmt error", K(ret));
} else {
ObLLVMValue elem_value;
if (NULL == s.get_handler(i).get_desc()->get_condition(j).sql_state_ || 0 == s.get_handler(i).get_desc()->get_condition(j).str_len_) {
ObLLVMValue str;
ObLLVMValue len;
if (OB_FAIL(generator_.generate_empty_string(str, len))) {
LOG_WARN("failed to generate_empty_string", K(ret));
} else if (OB_FAIL(condition_elements.push_back(str))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(condition_elements.push_back(len))) {
LOG_WARN("push_back error", K(ret));
} else { /*do nothing*/ }
} else {
if (OB_FAIL(generator_.get_helper().create_global_string(ObString(s.get_handler(i).get_desc()->get_condition(j).str_len_, s.get_handler(i).get_desc()->get_condition(j).sql_state_), elem_value))) {
LOG_WARN("failed to create_global_string", K(ret));
} else if (OB_FAIL(condition_elements.push_back(elem_value))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.get_helper().get_int64(s.get_handler(i).get_desc()->get_condition(j).str_len_, int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(condition_elements.push_back(int_value))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(const_name.append(s.get_handler(i).get_desc()->get_condition(j).sql_state_, s.get_handler(i).get_desc()->get_condition(j).str_len_))) {
LOG_WARN("append error", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
ObLLVMType llvm_type;
ObLLVMConstant const_value;
ObLLVMGlobalVariable const_condition;
ObLLVMType condition_value_type;
if (OB_FAIL(generator_.get_helper().get_int64(s.get_handler(i).get_level(), elem_value))) { //level(stmt_id)
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(condition_elements.push_back(elem_value))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.generate_null(ObTinyIntType, elem_value))) { //signal
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(condition_elements.push_back(elem_value))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.get_adt_service().get_pl_condition_value(condition_value_type))) {
LOG_WARN("failed to get_pl_condition_value", K(ret));
} else if (OB_FAIL(generator_.get_helper().get_or_insert_global(ObString(const_name.length(), const_name.ptr()), condition_value_type, const_condition))) {
LOG_WARN("failed to get_or_insert_global", K(ret));
} else if (OB_FAIL(ObLLVMHelper::get_const_struct(condition_value_type, condition_elements, const_value))) {
LOG_WARN("failed to get_const_struct", K(ret));
} else if (OB_FAIL(const_condition.set_initializer(const_value))) {
LOG_WARN("failed to set_initializer", K(ret));
} else if (OB_FAIL(const_condition.set_constant())) {
LOG_WARN("failed to set_constant", K(ret));
} else if (OB_FAIL(condition_clauses.push_back(const_condition))) {
LOG_WARN("failed to add_clause", K(ret));
} else if (OB_FAIL(precedences.push_back(std::make_pair(s.get_handler(i).get_desc()->get_condition(j).type_, s.get_handler(i).get_level())))) {
LOG_WARN("push back error", K(ret));
} else { /*do nothing*/ }
}
}
}
}
}
common::ObArray<int64_t> position_map;
int64_t pos = OB_INVALID_INDEX;
for (int64_t i = 0; OB_SUCC(ret) && i < precedences.count(); ++i) {
if (OB_FAIL(find_next_procedence_condition(precedences, position_map, pos))) {
LOG_WARN("failed to find next condition", K(ret));
} else if (OB_INVALID_INDEX == pos) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("pos is invalid", K(ret), K(pos));
} else if (OB_FAIL(catch_result.add_clause(condition_clauses.at(pos)))) {
LOG_WARN("failed to add clause", K(ret));
} else if (OB_FAIL(position_map.push_back(pos))) {
LOG_WARN("push back error", K(ret));
}
}
if (OB_SUCC(ret)) {
common::ObArray<int64_t> tmp_array;
for (int64_t i = 0; OB_SUCC(ret) && i < position_map.count(); ++i) {
bool find = false;
for (int64_t j = 0; OB_SUCC(ret) && !find && j < position_map.count(); ++j) {
if (i == position_map.at(j)) {
if (OB_FAIL(tmp_array.push_back(j))) {
LOG_WARN("push back error", K(ret));
} else {
find = true;
}
}
}
}
if (FAILEDx(position_map.assign(tmp_array))) {
LOG_WARN("assign error", K(ret));
}
}
if (OB_SUCC(ret)) {
ObLLVMValue unwindException;
ObLLVMValue retTypeInfoIndex;
ObLLVMValue unwindException_header;
ObLLVMType condition_type;
ObLLVMType condition_pointer_type;
ObLLVMValue condition;
if (OB_FAIL(generator_.get_helper().create_extract_value(ObString("extract_unwind_exception"), catch_result, 0, unwindException))) {
LOG_WARN("failed to create_extract_value", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_extract_value(ObString("extract_ret_type_info_index"), catch_result, 1, retTypeInfoIndex))) {
LOG_WARN("failed to create_extract_value", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_const_gep1_64(ObString("extract_unwind_exception_header"), unwindException, generator_.get_eh_service().pl_exception_base_offset_, unwindException_header))) {
LOG_WARN("failed to create_const_gep1_64", K(ret));
} else if (OB_FAIL(generator_.get_adt_service().get_pl_condition_value(condition_type))) {
LOG_WARN("failed to get_pl_condition_value", K(ret));
} else if (OB_FAIL(condition_type.get_pointer_to(condition_pointer_type))) {
LOG_WARN("failed to get_pointer_to", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_pointer_cast(ObString("cast_header"), unwindException_header, condition_pointer_type, condition))) {
LOG_WARN("failed to create_pointer_cast", K(ret));
} else {
ObLLVMValue type;
ObLLVMValue code;
ObLLVMValue name;
ObLLVMValue stmt_id;
ObLLVMValue signal;
if (OB_FAIL(generator_.extract_type_from_condition_value(condition, type))) {
LOG_WARN("failed to extract_type_from_condition_value", K(ret));
} else if (OB_FAIL(generator_.extract_code_from_condition_value(condition, code))) {
LOG_WARN("failed to extract_code_from_condition_value", K(ret));
} else if (OB_FAIL(generator_.extract_name_from_condition_value(condition, name))) {
LOG_WARN("failed to extract_name_from_condition_value", K(ret));
} else if (OB_FAIL(generator_.extract_stmt_from_condition_value(condition, stmt_id))) {
LOG_WARN("failed to extract_stmt_from_condition_value", K(ret));
} else if (OB_FAIL(generator_.extract_signal_from_condition_value(condition, signal))) {
LOG_WARN("failed to extract_signal_from_condition_value", K(ret));
#ifndef NDEBUG
} else if (OB_FAIL(generator_.generate_debug(ObString("debug"), unwindException))) {
LOG_WARN("failed to create_call", K(ret));
} else if (OB_FAIL(generator_.generate_debug(ObString("debug"), retTypeInfoIndex))) {
LOG_WARN("failed to create_call", K(ret));
} else if (OB_FAIL(generator_.generate_debug(ObString("debug"), type))) {
LOG_WARN("failed to create_call", K(ret));
} else if (OB_FAIL(generator_.generate_debug(ObString("debug"), code))) {
LOG_WARN("failed to create_call", K(ret));
} else if (OB_FAIL(generator_.generate_debug(ObString("debug"), name))) {
LOG_WARN("failed to create_call", K(ret));
} else if (OB_FAIL(generator_.generate_debug(ObString("debug"), stmt_id))) {
LOG_WARN("failed to create_call", K(ret));
#endif
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
//定义resume出口和exit出口
ObLLVMBasicBlock resume_handler;
ObLLVMBasicBlock exit_handler;
if (OB_FAIL(generator_.get_helper().create_block(ObString("resume_handler"), generator_.get_func(), resume_handler))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("exit_handler"), generator_.get_func(), exit_handler))) {
LOG_WARN("failed to create block", K(ret));
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
ObLLVMSwitch switch_inst;
if (OB_FAIL(generator_.get_helper().create_switch(retTypeInfoIndex, resume_handler, switch_inst))) {
LOG_WARN("failed to create switch", K(ret));
} else {
int64_t index = 0;
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_handlers().count(); ++i) {
ObLLVMBasicBlock case_branch;
if (OB_FAIL(generator_.get_helper().create_block(ObString("case"), generator_.get_func(), case_branch))) {
LOG_WARN("failed to create block", K(ret));
} else {
for (int64_t j = 0; OB_SUCC(ret) && j < s.get_handler(i).get_desc()->get_conditions().count(); ++j) {
if (OB_FAIL(generator_.get_helper().get_int32(position_map.at(index++) + 1, int_value))) {
LOG_WARN("failed to get int32", K(ret));
} else if (OB_FAIL(switch_inst.add_case(int_value, case_branch))) {
LOG_WARN("failed to add_case", K(ret));
} else { /*do nothing*/ }
}
/*
* 如果捕捉到了该exception,处理完body后,视handler的ACTION和level决定下一个目的地:
* EXIT:跳出当前block
* CONTINUE:跳到抛出exception的语句的下一句
* 我们对Handler做了改写,所以这里的逻辑是:
* 原生EXIT:跳出当前block
* 下降的CONTINUE:跳出当前block
* 下降的EXIT:继续向上抛
*/
if (OB_SUCC(ret)) {
if (NULL == s.get_handler(i).get_desc()->get_body() || (s.get_handler(i).get_desc()->is_exit() && !s.get_handler(i).is_original())) { //下降的EXIT,不执行body
if (OB_FAIL(generator_.get_helper().set_insert_point(case_branch))) {
LOG_WARN("failed to set_insert_point", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(s.get_handler(i).get_desc()->is_exit() && !s.get_handler(i).is_original() ? resume_handler : exit_handler))) {
LOG_WARN("failed to create_br", K(ret));
} else { /*do nothing*/ }
} else {
ObLLVMValue p_status, status;
ObLLVMBasicBlock current = generator_.get_current();
ObLLVMValue old_exception, old_ob_error;
ObSEArray<ObLLVMValue, 2> args;
ObLLVMValue result;
ObLLVMValue p_old_sqlcode, is_need_pop_warning_buf;
ObLLVMType int_type;
OZ (generator_.set_current(case_branch));
#ifndef NDEBUG
OZ (generator_.get_helper().get_int64(1111+i, int_value));
OZ (generator_.generate_debug(ObString("debug"), int_value));
#endif
OZ (generator_.get_helper().get_int32(OB_SUCCESS, int_value));
OZ (generator_.get_helper().create_store(int_value, generator_.get_vars().at(generator_.RET_IDX)));
OZ (generator_.extract_status_from_context(generator_.get_vars().at(generator_.CTX_IDX), p_status));
OZ (generator_.get_helper().create_load(ObString("load status"), p_status, status));
OZ (generator_.get_helper().create_store(int_value, p_status));
// 记录下当前捕获到的ObError, 用于设置再次抛出该异常时设置status
OX (old_ob_error = generator_.get_saved_ob_error());
OX (generator_.get_saved_ob_error() = status);
OX (args.reset());
OZ (generator_.get_helper().get_llvm_type(ObIntType, int_type));
OZ (generator_.get_helper().create_alloca(ObString("old_sqlcode"), int_type, p_old_sqlcode));
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (args.push_back(p_old_sqlcode));
OZ (generator_.get_helper().create_call(ObString("spi_get_pl_exception_code"), generator_.get_spi_service().spi_get_pl_exception_code_, args, result));
OZ (generator_.check_success(result, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
// 记录下当前捕获到的Exception, 用于SIGNAL语句抛出当前异常
OX (old_exception = generator_.get_saved_exception());
OX (generator_.get_saved_exception() = unwindException);
// 设置当前ExceptionCode到SQLCODE
OX (args.reset());
OZ (generator_.get_helper().get_int8(false, is_need_pop_warning_buf));
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (args.push_back(code));
OZ (args.push_back(is_need_pop_warning_buf));
OZ (generator_.get_helper().create_call(ObString("spi_set_pl_exception_code"), generator_.get_spi_service().spi_set_pl_exception_code_, args, result));
OZ (generator_.check_success(result, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
/*
// Check当前是否是嵌套事务的ExceptionHandler, 失效掉嵌套事务内部的Exception Handler
OX (args.reset());
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (args.push_back(code));
OZ (generator_.get_helper().create_call(
ObString("spi_check_exception_handler_legal"),
generator_.get_spi_service().spi_check_exception_handler_legal_,
args,
result));
OZ (generator_.check_success(
result,
s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
*/
// Codegen当前Handler的Body
OZ (SMART_CALL(generate(*s.get_handler(i).get_desc()->get_body())));
// 恢复原来的Exception
OX (generator_.get_saved_exception() = old_exception);
OX (generator_.get_saved_ob_error() = old_ob_error);
// 恢复原来的SQLCODE
if (OB_SUCC(ret)
&& OB_NOT_NULL(generator_.get_current().get_v())) {
ObLLVMValue old_code;
if (OB_ISNULL(old_exception.get_v())) {
OZ (generator_.get_helper().create_load(ObString("load_old_sqlcode"), p_old_sqlcode, old_code));
OZ (generator_.generate_debug(ObString("debug"), int_value));
//OZ (generator_.get_helper().get_int64(OB_SUCCESS, old_code));
} else {
ObLLVMValue old_unwindException_header;
ObLLVMType old_condition_type;
ObLLVMType old_condition_pointer_type;
ObLLVMValue old_condition;
OZ (generator_.get_helper().create_const_gep1_64(ObString("extract_unwind_exception_header"), unwindException, generator_.get_eh_service().pl_exception_base_offset_, old_unwindException_header));
OZ (generator_.get_adt_service().get_pl_condition_value(old_condition_type));
OZ (condition_type.get_pointer_to(old_condition_pointer_type));
OZ (generator_.get_helper().create_pointer_cast(ObString("cast_header"), old_unwindException_header, old_condition_pointer_type, old_condition));
OZ (generator_.extract_code_from_condition_value(old_condition, old_code));
}
OX (args.reset());
OZ (generator_.get_helper().get_int8(true, is_need_pop_warning_buf));
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (args.push_back(old_code));
OZ (args.push_back(is_need_pop_warning_buf));
OZ (generator_.get_helper().create_call(ObString("spi_set_pl_exception_code"), generator_.get_spi_service().spi_set_pl_exception_code_, args, result));
OZ (generator_.check_success(result, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
}
OZ (generator_.finish_current(s.get_handler(i).get_desc()->is_exit() && !s.get_handler(i).is_original() ? resume_handler : exit_handler)); //下降的EXIT才抛出
OZ (generator_.set_current(current));
}
}
}
}
if (OB_SUCC(ret)) {
ObLLVMType unwind_exception_type;
ObLLVMType unwind_exception_pointer_type;
if (OB_FAIL(generator_.set_exception(exception,
exit_handler,
s.get_level()))) { //这里才可以压栈exception
LOG_WARN("failed to set_exception", K(ret));
} else if (OB_FAIL(generator_.get_helper().set_insert_point(resume_handler))) {
LOG_WARN("failed to set_insert_point", K(ret));
} else if (NULL != generator_.get_parent_exception()) {
#ifndef NDEBUG
if (OB_FAIL(generator_.get_helper().get_int64(2222, int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(generator_.generate_debug(ObString("debug"), int_value))) {
LOG_WARN("failed to create_call", K(ret));
} else
#endif
if (OB_FAIL(generator_.get_adt_service().get_unwind_exception(unwind_exception_type))) {
LOG_WARN("failed to get_unwind_exception", K(ret));
} else if (OB_FAIL(unwind_exception_type.get_pointer_to(unwind_exception_pointer_type))) {
LOG_WARN("failed to get_pointer_to", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_pointer_cast(ObString("cast_unwind_exception"), unwindException, unwind_exception_pointer_type, unwindException))) {
LOG_WARN("failed to create_pointer_cast", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_invoke(ObString("call_resume"), generator_.get_eh_service().eh_resume_, unwindException, body, generator_.get_parent_exception()->exception_))) {
LOG_WARN("failed to create block", K(ret));
} else { /*do nothing*/ }
} else {
if (OB_FAIL(generator_.get_helper().create_resume(catch_result))) {
LOG_WARN("failed to create_resume", K(ret));
}
}
}
if (OB_SUCC(ret)) {
if (OB_FAIL(generator_.set_current(body))) {
LOG_WARN("failed to set_current", K(ret));
}
}
}
}
}
}
}
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLSignalStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else {
OZ (generator_.get_helper().set_insert_point(generator_.get_current()));
OZ (generator_.set_debug_location(s));
OZ (generator_.generate_goto_label(s));
ObLLVMBasicBlock normal;
ObLLVMType unwind_exception_type, unwind_exception_pointer_type;
ObLLVMType condition_type;
ObLLVMType condition_pointer_type;
ObLLVMValue unwindException = generator_.get_saved_exception();
ObLLVMValue ob_error_code = generator_.get_saved_ob_error();
ObLLVMValue unwind_exception_header;
ObLLVMValue condition;
ObLLVMValue sql_state;
ObLLVMValue error_code;
OZ (generator_.get_adt_service().get_unwind_exception(unwind_exception_type));
OZ (unwind_exception_type.get_pointer_to(unwind_exception_pointer_type));
OZ (generator_.get_adt_service().get_pl_condition_value(condition_type));
OZ (condition_type.get_pointer_to(condition_pointer_type));
if (OB_FAIL(ret)) {
} else if (s.is_signal_null()) {
ObLLVMValue status;
CK (OB_NOT_NULL(generator_.get_saved_exception().get_v()));
CK (OB_NOT_NULL(generator_.get_saved_ob_error().get_v()));
OZ (generator_.extract_status_from_context(generator_.get_vars().at(generator_.CTX_IDX), status));
OZ (generator_.get_helper().create_store(ob_error_code, status));
OZ (generator_.get_helper().create_const_gep1_64(ObString("extract_unwind_exception_header"),
unwindException,
generator_.get_eh_service().pl_exception_base_offset_,
unwind_exception_header));
OZ (generator_.get_helper().create_pointer_cast(ObString("cast_header"),
unwind_exception_header,
condition_pointer_type,
condition));
OZ (generator_.extract_name_from_condition_value(condition, sql_state));
OZ (generator_.extract_code_from_condition_value(condition, error_code));
OZ (generator_.get_helper().create_pointer_cast(
ObString("cast_unwind_exception"), unwindException,
unwind_exception_pointer_type, unwindException));
OZ (generator_.get_helper().create_block(ObString("normal"), generator_.get_func(), normal));
OZ (generator_.raise_exception(unwindException, error_code, sql_state, normal,
s.get_block()->in_notfound(), s.get_block()->in_warning(), true));
OZ (generator_.set_current(normal));
} else {
ObLLVMValue type, ob_err_code, err_code, sql_state, str_len, is_signal, stmt_id, loc, err_code_ptr;
if (lib::is_mysql_mode() && (s.is_resignal_stmt() || s.get_cond_type() != ERROR_CODE)) {
ObLLVMValue int_value;
ObLLVMType int_type, int32_type, int32_type_ptr;
ObSEArray<ObLLVMValue, 5> args;
int64_t *err_idx = const_cast<int64_t *>(s.get_expr_idx(
static_cast<int64_t>(SignalCondInfoItem::DIAG_MYSQL_ERRNO)));
int64_t *msg_idx = const_cast<int64_t *>(s.get_expr_idx(
static_cast<int64_t>(SignalCondInfoItem::DIAG_MESSAGE_TEXT)));
OZ (generator_.get_helper().get_llvm_type(ObIntType, int_type));
OZ (generator_.get_helper().get_llvm_type(ObInt32Type, int32_type));
OZ (int32_type.get_pointer_to(int32_type_ptr));
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.generate_pointer(NULL != err_idx ? generator_.get_expr(*err_idx) : NULL, int_value));
OZ (args.push_back(int_value));
OZ (generator_.generate_pointer(NULL != msg_idx ? generator_.get_expr(*msg_idx) : NULL, int_value));
OZ (args.push_back(int_value));
OZ (generator_.generate_string(ObString(s.get_str_len(), s.get_sql_state()), sql_state, str_len));
OZ (args.push_back(sql_state));
OZ (generator_.get_helper().create_alloca(ObString("error_code"), int_type, err_code_ptr));
if (s.is_resignal_stmt()) {
// ObLLVMValue code_ptr;
CK (OB_NOT_NULL(generator_.get_saved_exception().get_v()));
CK (OB_NOT_NULL(generator_.get_saved_ob_error().get_v()));
OZ (generator_.get_helper().create_const_gep1_64(ObString("extract_unwind_exception_header"),
unwindException,
generator_.get_eh_service().pl_exception_base_offset_,
unwind_exception_header));
OZ (generator_.get_helper().create_pointer_cast(ObString("cast_header"),
unwind_exception_header,
condition_pointer_type,
condition));
OZ (generator_.extract_name_from_condition_value(condition, sql_state));
OZ (generator_.extract_code_from_condition_value(condition, error_code));
OZ (generator_.get_helper().create_store(error_code, err_code_ptr));
} else {
OZ (generator_.get_helper().get_int64(OB_SUCCESS, err_code));
OZ (generator_.get_helper().create_store(err_code, err_code_ptr));
}
OZ (args.push_back(err_code_ptr));
OZ (args.push_back(sql_state));
OZ (generator_.get_helper().get_int8(!s.is_resignal_stmt(), is_signal));
OZ (args.push_back(is_signal));
OZ (generator_.get_helper().create_call(ObString("spi_process_resignal"),
generator_.get_spi_service().spi_process_resignal_error_,
args,
ob_err_code));
OZ (generator_.check_success(ob_err_code, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
OZ (generator_.get_helper().create_load(ObString("load_error_code"), err_code_ptr, err_code));
OZ (generator_.get_helper().create_bit_cast(ObString("cast_int64_to_int32"), err_code_ptr, int32_type_ptr, err_code_ptr));
OZ (generator_.get_helper().create_load(ObString("load_error_code"), err_code_ptr, ob_err_code));
} else {
OZ (generator_.get_helper().get_int32(s.get_ob_error_code(), ob_err_code));
OZ (generator_.get_helper().get_int64(s.get_error_code(), err_code));
}
OZ (generator_.get_helper().get_int64(s.get_cond_type(), type));
OZ (generator_.get_helper().create_block(ObString("normal"), generator_.get_func(), normal));
OZ (generator_.generate_string(ObString(s.get_str_len(), s.get_sql_state()), sql_state, str_len));
OZ (generator_.get_helper().get_int64(s.get_stmt_id(), stmt_id));
// 暂时先用stmtid, 这个id就是col和line的组合
OZ (generator_.get_helper().get_int64(s.get_stmt_id(), loc));
OZ (generator_.generate_exception(type, ob_err_code, err_code, sql_state, str_len, stmt_id,
normal, loc, s.get_block()->in_notfound(),
s.get_block()->in_warning(), true/*is signal*/));
OZ (generator_.set_current(normal));
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLRaiseAppErrorStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
// 控制流已断, 后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {
ObLLVMValue int_value;
ObSEArray<ObLLVMValue, 3> args;
ObLLVMValue ret_err;
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.generate_pointer(generator_.get_expr(s.get_params().at(0)), int_value));
OZ (args.push_back(int_value));
OZ (generator_.generate_pointer(generator_.get_expr(s.get_params().at(1)), int_value));
OZ (args.push_back(int_value));
OZ (generator_.get_helper().create_call(ObString("spi_raise_application_error"), generator_.get_spi_service().spi_raise_application_error_, args, ret_err));
OZ (generator_.check_success(ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLCallStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {
ObLLVMValue params;
ObLLVMType argv_type;
ObLLVMType int_type;
if (OB_FAIL(generator_.get_helper().get_llvm_type(ObIntType, int_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(generator_.get_adt_service().get_argv(s.get_params().count(), argv_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_alloca(ObString("inner_pl_argv"), argv_type, params))) {
LOG_WARN("failed to create_alloca", K(ret));
} else {
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_params().count(); ++i) {
// 传递入参
ObLLVMValue p_result_obj;
if (s.is_pure_out(i)) {
const ObPLVar *var =
OB_INVALID_INDEX == s.get_out_index(i) ? NULL : s.get_variable(s.get_out_index(i));
const ObPLDataType *pl_type = OB_ISNULL(var) ? NULL : &(var->get_type());
ObPLDataType final_type;
if (lib::is_oracle_mode() && NULL != var && var->get_type().is_cursor_type()) {
OZ (generator_.extract_objparam_from_context(generator_.get_vars().at(
generator_.CTX_IDX),
s.get_out_index(i),
p_result_obj));
} else {
OZ (generator_.generate_new_objparam(p_result_obj), K(i), KPC(var));
}
if (OB_SUCC(ret)
&& OB_ISNULL(pl_type)
&& OB_NOT_NULL(s.get_param_expr(i))
&& s.get_param_expr(i)->is_obj_access_expr()) {
const ObObjAccessRawExpr *obj_access
= static_cast<const ObObjAccessRawExpr *>(s.get_param_expr(i));
CK (OB_NOT_NULL(obj_access));
OZ (obj_access->get_final_type(final_type));
OX (pl_type = &final_type);
}
if (OB_SUCC(ret)
&& OB_NOT_NULL(pl_type)
&& pl_type->is_composite_type()
&& !pl_type->is_opaque_type()) { // 普通类型构造空的ObjParam即可,复杂类型需要构造对应的指针
ObLLVMType ir_type;
ObLLVMValue value;
ObLLVMValue const_value;
if (OB_FAIL(generator_.get_llvm_type(*pl_type, ir_type))) {
LOG_WARN("failed to get llvm type", K(ret));
} else if (OB_FAIL(
generator_.get_helper().create_alloca(ObString("PureOut"), ir_type, value))) {
LOG_WARN("failed to create alloca", K(ret));
} else if (OB_FAIL(pl_type->generate_construct(generator_,
*s.get_namespace(),
value,
&s))) {
LOG_WARN("failed to generate_construct", K(ret));
} else if (pl_type->is_collection_type()) {
const ObUserDefinedType *user_type = NULL;
if (OB_ISNULL(s.get_namespace())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("ns is NULL", K(ret));
} else if (OB_FAIL(s.get_namespace()
->get_pl_data_type_by_id(pl_type->get_user_type_id(), user_type))) {
LOG_WARN("failed to get pl data type by id", K(ret));
} else if (OB_ISNULL(user_type)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("user type is NULL", K(pl_type->get_user_type_id()), K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
int64_t init_size = OB_INVALID_SIZE;
ObLLVMValue p_dest_value;
ObLLVMType int_type;
ObLLVMValue var_addr;
ObLLVMValue var_type;
ObLLVMValue init_value;
ObLLVMValue extend_value;
OZ (generator_.get_helper().get_int8(pl_type->get_type(), var_type));
OZ (pl_type->get_size(*s.get_namespace(), PL_TYPE_INIT_SIZE, init_size));
OZ (generator_.get_helper().get_int32(init_size, init_value));
OZ (generator_.get_helper().get_llvm_type(ObIntType, int_type));
OZ (generator_.get_helper().create_ptr_to_int(ObString("cast_ptr_to_int64"),
value,
int_type,
var_addr));
OZ (generator_.extract_obobj_ptr_from_objparam(p_result_obj, p_dest_value));
OZ (generator_.generate_set_extend(p_dest_value, var_type, init_value, var_addr));
}
}
} else {
const ObPLVar *var = OB_INVALID_INDEX == s.get_out_index(i)
? NULL : s.get_variable(s.get_out_index(i));
if (lib::is_oracle_mode() && NULL != var && var->get_type().is_cursor_type()) {
OZ (generator_.extract_objparam_from_context(generator_.get_vars().at(
generator_.CTX_IDX),
s.get_out_index(i),
p_result_obj));
} else if (!s.get_param_expr(i)->is_obj_access_expr()
|| !(static_cast<const ObObjAccessRawExpr *>(s.get_param_expr(i)))->for_write()) {
OZ (generator_.generate_expr(s.get_param(i), s, OB_INVALID_INDEX, p_result_obj));
} else {
ObLLVMValue address;
ObPLDataType final_type;
const ObObjAccessRawExpr *obj_access
= static_cast<const ObObjAccessRawExpr *>(s.get_param_expr(i));
CK (OB_NOT_NULL(obj_access));
OZ (obj_access->get_final_type(final_type));
OZ (generator_.generate_expr(s.get_param(i),
s,
OB_INVALID_INDEX,
final_type.is_obj_type() ? address : p_result_obj));
if (OB_SUCC(ret) && final_type.is_obj_type()) {
ObLLVMType obj_type;
ObLLVMType obj_type_ptr;
ObLLVMValue p_obj;
ObLLVMValue src_obj;
ObLLVMValue p_dest_obj;
OZ (generator_.generate_new_objparam(p_result_obj));
OZ (generator_.extract_extend_from_objparam(address, final_type, p_obj));
OZ (generator_.get_adt_service().get_obj(obj_type));
OZ (obj_type.get_pointer_to(obj_type_ptr));
OZ (generator_.get_helper().create_bit_cast(
ObString("cast_addr_to_obj_ptr"), p_obj, obj_type_ptr, p_obj));
OZ (generator_.get_helper().create_load(ObString("load obj value"), p_obj, src_obj));
OZ (generator_.extract_datum_ptr_from_objparam(p_result_obj, ObNullType, p_dest_obj));
OZ (generator_.get_helper().create_store(src_obj, p_dest_obj));
}
}
}
if (OB_SUCC(ret)) {
ObLLVMValue p_arg;
ObLLVMValue pp_arg;
if (OB_FAIL(generator_.get_helper().create_ptr_to_int(ObString("cast_arg_to_pointer"), p_result_obj, int_type, p_arg))) {
LOG_WARN("failed to create_PtrToInt", K(ret));
} else if (OB_FAIL(generator_.extract_arg_from_argv(params, i, pp_arg))) {
LOG_WARN("failed to extract_arg_from_argv", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_store(p_arg, pp_arg))) {
LOG_WARN("failed to create_store", K(ret));
} else { /*do nothing*/ }
}
}
if (OB_SUCC(ret)) {
ObSEArray<ObLLVMValue, 5> args;
ObLLVMValue argv;
ObLLVMValue int_value;
ObLLVMValue array_value;
ObLLVMValue nocopy_array_value;
uint64_t package_id = s.get_package_id();
package_id = 1 == s.get_is_object_udf()
? share::schema::ObUDTObjectType::mask_object_id(package_id) : package_id;
if (OB_FAIL(args.push_back(generator_.get_vars().at(generator_.CTX_IDX)))) { //PL的执行环境
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.get_helper().get_int64(package_id, int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(args.push_back(int_value))) { //PL的package id
LOG_WARN("push_back error", K(ret));
}else if (OB_FAIL(generator_.get_helper().get_int64(s.get_proc_id(), int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(args.push_back(int_value))) { //PL的proc id
LOG_WARN("push_back error", K(ret));
}else if (OB_FAIL(generator_.generate_int64_array(s.get_subprogram_path(), array_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(args.push_back(array_value))) { //PL的subprogram path
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.get_helper().get_int64(s.get_subprogram_path().count(),
int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(args.push_back(int_value))) { //subprogram path长度
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.get_helper().get_int64(static_cast<uint64_t>(s.get_stmt_id()), int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(args.push_back(int_value))) { // line number;
LOG_WARN("push back line number error", K(ret));
} else if (OB_FAIL(generator_.get_helper().get_int64(s.get_params().count(), int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(args.push_back(int_value))) { //argc
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_ptr_to_int(ObString("cast_argv_to_pointer"), params, int_type, argv))) {
LOG_WARN("failed to create_PtrToInt", K(ret));
} else if (OB_FAIL(args.push_back(argv))) {//argv
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generator_.generate_int64_array(s.get_nocopy_params(),
nocopy_array_value))) {
LOG_WARN("failed to get int64_t array", K(ret));
} else if (OB_FAIL(args.push_back(nocopy_array_value))) {
LOG_WARN("failed to push back", K(ret));
} else {
ObLLVMValue result;
if (NULL == generator_.get_current_exception()) {
if (OB_FAIL(generator_.get_helper().create_call(ObString("inner_pl_execute"), generator_.get_pl_execute(), args, result))) {
LOG_WARN("failed to create_call", K(ret));
}
OZ (generator_.generate_debug(ObString("debug inner pl execute"), result));
} else {
ObLLVMBasicBlock alter_inner_call;
if (OB_FAIL(generator_.get_helper().create_block(ObString("alter_inner_call"), generator_.get_func(), alter_inner_call))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_invoke(ObString("inner_pl_execute"), generator_.get_pl_execute(), args, alter_inner_call, generator_.get_current_exception()->exception_, result))) {
LOG_WARN("failed to create_call", K(ret));
} else if (OB_FAIL(generator_.set_current(alter_inner_call))) {
LOG_WARN("failed to set_current", K(ret));
} else { /*do nothing*/ }
}
OZ (generator_.generate_debug(ObString("debug inner pl execute result"), result));
OZ (generator_.check_success(
result, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
OZ (generator_.generate_out_params(s, s.get_params(), params));
}
}
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLDeclareCursorStmt &s)
{
return generator_.generate_declare_cursor(s, s.get_cursor_index());
}
int ObPLCodeGenerateVisitor::visit(const ObPLOpenStmt &s)
{
int ret = OB_SUCCESS;
OZ (generator_.generate_goto_label(s));
CK (OB_NOT_NULL(s.get_cursor()));
OZ (generator_.generate_open(static_cast<const ObPLStmt&>(s),
s.get_cursor()->get_value(),
s.get_cursor()->get_package_id(),
s.get_cursor()->get_routine_id(),
s.get_index()));
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLOpenForStmt &s)
{
int ret = OB_SUCCESS;
OZ (generator_.generate_goto_label(s));
OZ (generator_.generate_open_for(s));
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLFetchStmt &s)
{
int ret = OB_SUCCESS;
ObLLVMValue ret_err;
OZ (generator_.generate_goto_label(s));
if (OB_FAIL(generator_.generate_fetch(static_cast<const ObPLStmt&>(s),
static_cast<const ObPLInto&>(s),
s.get_package_id(),
s.get_routine_id(),
s.get_index(),
s.get_limit(),
s.get_user_type(),
s.get_expand_user_types(),
ret_err))) {
LOG_WARN("failed to generate fetch", K(ret));
} else if (lib::is_mysql_mode()) { //Mysql模式直接检查抛出异常
OZ (generator_.check_success(
ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning(), true));
} else { //Oracle模式如果是OB_READ_NOTHING错误,吞掉异常不抛出
ObLLVMValue is_not_found;
ObLLVMBasicBlock fetch_end;
ObLLVMBasicBlock fetch_check_success;
if (OB_FAIL(generator_.get_helper().create_block(ObString("fetch_end"), generator_.get_func(), fetch_end))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_block(ObString("fetch_check_success"), generator_.get_func(), fetch_check_success))) {
LOG_WARN("failed to create block", K(s), K(ret));
} else if (OB_FAIL(generator_.get_helper().create_icmp_eq(ret_err, OB_READ_NOTHING, is_not_found))) {
LOG_WARN("failed to create_icmp_eq", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_cond_br(is_not_found, fetch_end, fetch_check_success))) {
LOG_WARN("failed to create_cond_br", K(ret));
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
if (OB_FAIL(generator_.set_current(fetch_check_success))) {
LOG_WARN("failed to set current", K(ret));
} else if (OB_FAIL(generator_.check_success(ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()))) {
LOG_WARN("failed to check success", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(fetch_end))) {
LOG_WARN("failed to create_br", K(ret));
} else if (OB_FAIL(generator_.set_current(fetch_end))) {
LOG_WARN("failed to set current", K(ret));
} else { /*do nothing*/ }
}
}
if (OB_SUCC(ret)) {
if (OB_FAIL(generator_.generate_into_restore(s.get_into(), s.get_exprs(), s.get_symbol_table()))) {
LOG_WARN("Failed to generate_into", K(ret));
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLCloseStmt &s)
{
int ret = OB_SUCCESS;
OZ (generator_.generate_goto_label(s));
if (OB_FAIL(generator_.generate_close(static_cast<const ObPLStmt&>(s),
s.get_package_id(),
s.get_routine_id(),
s.get_index()))) {
LOG_WARN("failed to generate close", K(ret));
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLNullStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLPipeRowStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else {
OZ (generator_.get_helper().set_insert_point(generator_.get_current()));
OZ (generator_.set_debug_location(s));
OZ (generator_.generate_goto_label(s));
ObLLVMBasicBlock pipe_row_block;
ObLLVMBasicBlock init_result_block;
ObLLVMBasicBlock pipe_result_block;
OZ (generator_.get_helper().create_block(
ObString("pipe_row_block"), generator_.get_func(), pipe_row_block));
OZ (generator_.get_helper().create_block(
ObString("init_result_block"), generator_.get_func(), init_result_block));
OZ (generator_.get_helper().create_block(
ObString("pipe_result_block"), generator_.get_func(), pipe_result_block));
OZ (generator_.get_helper().create_br(pipe_row_block));
OZ (generator_.set_current(pipe_row_block));
ObLLVMValue p_result;
ObLLVMValue p_type_value;
ObLLVMValue type_value;
ObLLVMValue is_invalid;
ObLLVMType obj_param_type;
ObLLVMType obj_param_type_pointer;
// 检查result数组是否已经初始化
OZ (generator_.extract_result_from_context(
generator_.get_vars().at(generator_.CTX_IDX), p_result));
OZ (generator_.get_adt_service().get_objparam(obj_param_type));
OZ (obj_param_type.get_pointer_to(obj_param_type_pointer));
OZ (generator_.get_helper().create_bit_cast(ObString("obobj_to_obobjparam"),
p_result, obj_param_type_pointer, p_result));
OZ (generator_.extract_type_ptr_from_objparam(p_result, p_type_value));
OZ (generator_.get_helper().create_load(
ObString("load_result_type"), p_type_value, type_value));
OZ (generator_.get_helper().create_icmp_eq(type_value, ObMaxType, is_invalid));
OZ (generator_.get_helper().create_cond_br(is_invalid, init_result_block, pipe_result_block));
// 初始化result数组
ObSEArray<ObLLVMValue, 3> args;
ObLLVMValue var_idx, init_value, var_value, extend_value;
ObLLVMValue ret_err;
ObLLVMValue var_type, type_id;
int64_t init_size = 0;
OZ (generator_.set_current(init_result_block));
CK (OB_NOT_NULL(s.get_namespace()));
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.get_helper().get_int8(s.get_type().get_type(), var_type));
OZ (args.push_back(var_type));
OZ (generator_.get_helper().get_int64(s.get_type().get_user_type_id(), type_id));
OZ (args.push_back(type_id));
OZ (generator_.get_helper().get_int64(OB_INVALID_INDEX, var_idx));
OZ (args.push_back(var_idx));
OZ (s.get_namespace()->get_size(PL_TYPE_INIT_SIZE, s.get_type(), init_size));
OZ (generator_.get_helper().get_int32(init_size, init_value));
OZ (args.push_back(init_value));
OZ (generator_.generate_null_pointer(ObIntType, init_value));
OZ (args.push_back(init_value));
OZ (generator_.get_helper().create_call(ObString("spi_alloc_complex_var"),
generator_.get_spi_service().spi_alloc_complex_var_,
args,
ret_err));
OZ (generator_.check_success(
ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
OZ (generator_.extract_result_from_context(
generator_.get_vars().at(generator_.CTX_IDX), var_value));
OZ (generator_.get_helper().create_bit_cast(
ObString("obobj_to_obobjparam"), var_value, obj_param_type_pointer, var_value));
OZ (generator_.extract_extend_from_objparam(var_value, s.get_type(), extend_value));
OZ (s.get_type().generate_construct(
generator_, *s.get_namespace(), extend_value, &s));
OZ (generator_.get_helper().create_br(pipe_result_block));
// 将当前行插入Result数组中
ObLLVMValue p_result_obj;
OZ (generator_.set_current(pipe_result_block));
OZ (generator_.generate_expr(s.get_row(), s, OB_INVALID_INDEX, p_result_obj));
OX (args.reset());
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX))); //PL的执行环境
OZ (args.push_back(p_result_obj));
OZ (generator_.get_helper().create_call(ObString("spi_pipe_row_to_result"),
generator_.get_spi_service().spi_pipe_row_to_result_,
args, ret_err));
OZ (generator_.check_success(ret_err,
s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLGotoStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {
ObLLVMBasicBlock entry_blk;
if (!s.get_dst_label().empty()) {
if (OB_FAIL(generator_.get_helper().create_block(ObString("goto_blk"),
generator_.get_func(), entry_blk))) {
LOG_WARN("faile to create basic block.", K(ret));
} else if (OB_FAIL(generator_.get_helper().create_br(entry_blk))) {
LOG_WARN("failed to create br instr", K(ret));
} else if (OB_FAIL(generator_.set_current(entry_blk))) {
LOG_WARN("failed to set current block", K(ret));
} else if (OB_FAIL(generator_.get_helper().set_insert_point(entry_blk))) {
LOG_WARN("failed to set insert point", K(ret));
} else {
#define GEN_BR_WITH_COLSE_CURSOR(goto_dst) \
do { \
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_cursor_stmt_count(); ++i) { \
const ObPLCursorForLoopStmt *cs = \
static_cast<const ObPLCursorForLoopStmt *>(s.get_cursor_stmt(i)); \
if (OB_NOT_NULL(cs)) { \
CK (OB_NOT_NULL(cs->get_cursor())); \
OZ (generator_.generate_close(static_cast<const ObPLStmt&>(*cs), \
cs->get_cursor()->get_package_id(), \
cs->get_cursor()->get_routine_id(), \
cs->get_index()));\
} else {\
ret = OB_ERR_UNEXPECTED;\
LOG_WARN("null goto cursor for loop stmt", K(ret));\
}\
}\
if (OB_SUCC(ret)) { \
if (OB_SUCC(ret) && OB_FAIL(generator_.get_helper().create_br(goto_dst))) { \
LOG_WARN("failed to create br instr", K(ret)); \
}\
}\
} while(0)
//跳转目的地和跳转语句之间的所有循环的栈都需要重置
#define RESTORE_LOOP_STACK(src_level, dst_level) \
do { \
for (int64_t i = generator_.get_loop_count() - 1; OB_SUCC(ret) && i >= 0; --i) { \
ObPLCodeGenerator::LoopStack::LoopInfo &loop_info = generator_.get_loops()[i]; \
LOG_DEBUG("loop info", K(i), K(loop_info.level_), K(src_level), K(dst_level)); \
if (loop_info.level_ <= src_level && loop_info.level_ >= dst_level) { \
if (OB_FAIL(generator_.get_helper().stack_restore(loop_info.loop_))) { \
LOG_WARN("failed to stack_restore", K(ret), K(src_level), K(dst_level)); \
} else { \
LOG_DEBUG("success stack restore", \
K(i), K(loop_info.level_), K(src_level), K(dst_level)); \
} \
} \
} \
} while(0)
hash::HashMapPair<ObPLCodeGenerator::goto_label_flag, ObLLVMBasicBlock> pair;
int tmp_ret = generator_.get_goto_label_map().get_refactored(
s.get_dst_stmt()->get_stmt_id(), pair);
if (OB_SUCCESS == tmp_ret) {
RESTORE_LOOP_STACK(s.get_level(), s.get_dst_stmt()->get_level());
GEN_BR_WITH_COLSE_CURSOR(pair.second);
} else if (OB_HASH_NOT_EXIST == tmp_ret) {
ObLLVMBasicBlock dst_blk;
if (OB_FAIL(generator_.get_helper().create_block(s.get_dst_label(),
generator_.get_func(), dst_blk))) {
LOG_WARN("faile to create dst block", K(ret));
} else if (OB_FAIL(pair.init(ObPLCodeGenerator::goto_label_flag::GOTO_LABEL_EXIST,
dst_blk))) {
LOG_WARN("failed to init pair", K(ret));
} else if (OB_FAIL(generator_.get_goto_label_map().set_refactored(
s.get_dst_stmt()->get_stmt_id(), pair))) {
LOG_WARN("fill hash map failed", K(ret));
} else {
GEN_BR_WITH_COLSE_CURSOR(dst_blk);
}
} else {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("failed to create goto label", K(ret));
}
#undef RESTORE_LOOP_STACK
}
if (OB_SUCC(ret)) {
ObLLVMBasicBlock new_blk;
if (OB_FAIL(generator_.get_helper().create_block(ObString("after_goto"),
generator_.get_func(), new_blk))) {
LOG_WARN("failed to create basic block", K(ret));
} else if (OB_FAIL(generator_.set_current(new_blk))) {
LOG_WARN("failed to set current block", K(ret));
}
}
} else {
ret = OB_ERR_ZERO_LENGTH_IDENTIFIER;
LOG_WARN("goto label is empty", K(ret));
}
}
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLInterfaceStmt &s)
{
int ret = OB_SUCCESS;
ObSEArray<ObLLVMValue, 2> args;
ObLLVMValue entry;
ObLLVMValue ret_err;
OZ (args.push_back(generator_.get_vars().at(generator_.CTX_IDX)));
OZ (generator_.get_helper().get_int64(s.get_entry(), entry));
OZ (args.push_back(entry));
OZ (generator_.get_helper().create_call(ObString("spi_interface_impl"),
generator_.get_spi_service().spi_interface_impl_,
args,
ret_err));
OZ (generator_.check_success(ret_err,
s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
return ret;
}
int ObPLCodeGenerateVisitor::visit(const ObPLDoStmt &s)
{
int ret = OB_SUCCESS;
if (NULL == generator_.get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(generator_.get_helper().set_insert_point(generator_.get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(generator_.set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(generator_.generate_goto_label(s))) {
LOG_WARN("failed to generate goto label", K(ret));
} else {
if (OB_SUCC(ret)) {
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_value().count(); ++i) {
const ObRawExpr *value_expr = s.get_value_expr(i);
int64_t result_idx = OB_INVALID_INDEX;
ObLLVMValue p_result_obj;
CK(OB_NOT_NULL(value_expr));
OZ(generator_.generate_expr(s.get_value_index(i), s, result_idx,p_result_obj));
}
}
}
return ret;
}
int ObPLCodeGenerateVisitor::find_next_procedence_condition(common::ObIArray<std::pair<ObPLConditionType, int64_t>> &conditions,
common::ObIArray<int64_t> &position_map, int64_t &idx)
{
int ret = OB_SUCCESS;
idx = OB_INVALID_INDEX;
ObPLConditionType type = INVALID_TYPE;
int64_t level = OB_INVALID_INDEX;
for (int64_t i = 0; i < conditions.count(); ++i) {
bool need_ignore = false;
for (int64_t j = 0; j < position_map.count(); ++j) {
if (i == position_map.at(j)) {
need_ignore = true;
break;
}
}
if (need_ignore) {
continue;
} else if (INVALID_TYPE == type) {
type = conditions.at(i).first;
level = conditions.at(i).second;
idx = i;
} else {
int compare = ObPLDeclareHandlerStmt::DeclareHandler::compare_condition(conditions.at(i).first, conditions.at(i).second, type, level);
if (compare <= 0) {
//do nothing, just continue
} else {
type = conditions.at(i).first;
level = conditions.at(i).second;
idx = i;
}
}
}
return ret;
}
int ObPLCodeGenerator::generate_user_type(const ObUserDefinedType &type)
{
int ret = OB_SUCCESS;
ObLLVMStructType udt_ir_type;
ObSEArray<ObLLVMType, 16> elem_type_array;
bool is_cursor_type = false;
switch (type.get_type()) {
case PL_RECORD_TYPE: {
const ObRecordType &record_type = static_cast<const ObRecordType&>(type);
OZ (build_record_type(record_type, elem_type_array));
}
break;
case PL_CURSOR_TYPE:
case PL_REF_CURSOR_TYPE: {
is_cursor_type = true;
}
break;
default: {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("user defined type is invalid", K(type.get_type()), K(ret));
}
break;
}
if (OB_SUCC(ret) && !is_cursor_type) {
if (!type.is_subtype()) {
OZ (helper_.create_struct_type(type.get_name(), elem_type_array, udt_ir_type), type);
if (OB_SUCC(ret) && OB_ISNULL(user_type_map_.get(type.get_user_type_id()))) {
OZ (user_type_map_.set_refactored(type.get_user_type_id(), udt_ir_type), type);
}
} else {
CK (1 == elem_type_array.count());
if (OB_SUCC(ret) && OB_ISNULL(user_type_map_.get(type.get_user_type_id()))) {
OZ (user_type_map_.set_refactored(type.get_user_type_id(), elem_type_array.at(0)), type);
}
}
}
LOG_DEBUG("generator user type",
K(ret), K(type.get_user_type_id()), K(ast_.get_id()), K(ast_.get_name()), K(lbt()));
return ret;
}
int ObPLCodeGenerator::build_record_type(const ObRecordType &record_type,
ObIArray<jit::ObLLVMType> &elem_type_array)
{
int ret = OB_SUCCESS;
//int64_t count_;
ObLLVMType count_type;
ObLLVMType is_null_type;
ObLLVMType null_array_type;
ObLLVMType meta_type;
ObLLVMType meta_array_type;
ObLLVMType member_type;
ObLLVMType member_array_type;
OZ (build_composite(elem_type_array));
OZ (helper_.get_llvm_type(ObInt32Type, count_type));
OZ (elem_type_array.push_back(count_type));
OZ (helper_.get_llvm_type(ObTinyIntType, is_null_type));
for (int64_t i = 0; OB_SUCC(ret) && i < record_type.get_record_member_count(); ++i) {
OZ (elem_type_array.push_back(is_null_type));
}
OZ (adt_service_.get_data_type(meta_type));
for (int64_t i = 0; OB_SUCC(ret) && i < record_type.get_record_member_count(); ++i) {
OZ (elem_type_array.push_back(meta_type));
}
for (int64_t i = 0; OB_SUCC(ret) && i < record_type.get_record_member_count(); ++i) {
CK (OB_NOT_NULL(record_type.get_record_member_type(i)));
OZ (get_datum_type(*record_type.get_record_member_type(i), member_type));
OZ (elem_type_array.push_back(member_type));
}
return ret;
}
int ObPLCodeGenerator::build_composite(ObIArray<jit::ObLLVMType> &elem_type_array)
{
int ret = OB_SUCCESS;
//ObPLType type_;
//uint64_t id_;
//bool is_null_;
ObLLVMType type;
ObLLVMType id;
ObLLVMType is_null;
OZ (helper_.get_llvm_type(ObInt32Type, type));
OZ (elem_type_array.push_back(type));
OZ (helper_.get_llvm_type(ObIntType, id));
OZ (elem_type_array.push_back(id));
OZ (helper_.get_llvm_type(ObTinyIntType, is_null));
OZ (elem_type_array.push_back(is_null));
return ret;
}
int ObPLCodeGenerator::init()
{
int ret = OB_SUCCESS;
if (debug_mode_ && OB_FAIL(di_helper_.init(helper_.get_jc()))) {
LOG_WARN("failed to init di helper", K(ret));
} else if (OB_FAIL(init_spi_service())) {
LOG_WARN("failed to init spi service", K(ret));
} else if (OB_FAIL(init_adt_service())) {
LOG_WARN("failed to init adt service", K(ret));
} else if (OB_FAIL(init_eh_service())) {
LOG_WARN("failed to init eh service", K(ret));
} else if (OB_FAIL(init_di_adt_service())) {
LOG_WARN("failed to init di service", K(ret));
} else {
ObSEArray<ObLLVMType, 8> arg_types;
ObLLVMFunctionType ft;
ObLLVMType pl_exec_context_type;
ObLLVMType pl_exec_context_pointer_type;
ObLLVMType int64_type;
ObLLVMType int64_pointer_type;
ObLLVMType int32_type;
ObLLVMType bool_type;
if (OB_FAIL(adt_service_.get_pl_exec_context(pl_exec_context_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(pl_exec_context_type.get_pointer_to(pl_exec_context_pointer_type))) {
LOG_WARN("failed to get_pointer_to", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObIntType, int64_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(int64_type.get_pointer_to(int64_pointer_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObInt32Type, int32_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObTinyIntType, bool_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //uint64_t package id
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //uint64_t proc id
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_pointer_type))) { //int64_t* subprogram path
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //int64_t path length
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //int64_t line number
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //int64_t ArgC
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //int64_t[] ArgV
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_pointer_type))) { //int64_t* nocopy params
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("pl_execute"), ft, pl_execute_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
//declare user type var addr
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //int64_t var_index
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //int64_t var_addr
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int32_type))) { //int32_t init_value
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("set_user_type_var"), ft, set_user_type_var_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
// declare set_implicit_in_forall
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(bool_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("set_implicit_cursor_in_forall"), ft, set_implicit_cursor_in_forall_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
// declare unset_implicit_in_forall
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("unset_implicit_cursor_in_forall"), ft, unset_implicit_cursor_in_forall_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
}
return ret;
}
int ObPLCodeGenerator::init_spi_service()
{
int ret = OB_SUCCESS;
ObSEArray<ObLLVMType, 9> arg_types;
ObLLVMFunctionType ft;
ObLLVMType pl_exec_context_type;
ObLLVMType pl_exec_context_pointer_type;
ObLLVMType obj_param_type;
ObLLVMType obj_param_pointer_type;
ObLLVMType obj_type;
ObLLVMType obj_pointer_type;
ObLLVMType data_type;
ObLLVMType data_type_pointer_type;
ObLLVMType int64_type;
ObLLVMType int32_type;
ObLLVMType bool_type;
ObLLVMType char_type;
ObLLVMType int_pointer_type;
ObLLVMType bool_type_pointer_type;
if (OB_FAIL(adt_service_.get_pl_exec_context(pl_exec_context_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(pl_exec_context_type.get_pointer_to(pl_exec_context_pointer_type))) {
LOG_WARN("failed to get_pointer_to", K(ret));
} else if (OB_FAIL(adt_service_.get_objparam(obj_param_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(obj_param_type.get_pointer_to(obj_param_pointer_type))) {
LOG_WARN("failed to get_pointer_to", K(ret));
} else if (OB_FAIL(adt_service_.get_obj(obj_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(obj_type.get_pointer_to(obj_pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(adt_service_.get_data_type(data_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(data_type.get_pointer_to(data_type_pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObIntType, int64_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObInt32Type, int32_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObTinyIntType, bool_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObCharType, char_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(int64_type.get_pointer_to(int_pointer_type))) {
LOG_WARN("failed to get_pointer_to", K(ret));
} else if (OB_FAIL(bool_type.get_pointer_to(bool_type_pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(obj_param_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_calc_expr"), ft, spi_service_.spi_calc_expr_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(obj_param_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_calc_package_expr"), ft, spi_service_.spi_calc_package_expr_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (arg_types.push_back(obj_param_pointer_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(obj_param_pointer_type));
OZ (arg_types.push_back(bool_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_convert_objparam"), ft, spi_service_.spi_convert_objparam_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(obj_param_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(bool_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_set_variable"), ft, spi_service_.spi_set_variable_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(char_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(data_type_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(bool_type_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(bool_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_query"), ft, spi_service_.spi_query_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (arg_types.push_back(char_type));
OZ (arg_types.push_back(bool_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_end_trans"), ft, spi_service_.spi_end_trans_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(char_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(data_type_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(bool_type_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(bool_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(bool_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_execute"), ft, spi_service_.spi_execute_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int_pointer_type))) { // param_mode
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(data_type_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(bool_type_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(bool_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(bool_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_execute_immediate"), ft, spi_service_.spi_execute_immediate_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (arg_types.push_back(bool_type)); //int8 type
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(int32_type));
OZ (arg_types.push_back(int_pointer_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_alloc_complex_var"), ft, spi_service_.spi_alloc_complex_var_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(obj_param_pointer_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_construct_collection"), ft, spi_service_.spi_construct_collection_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_clear_diagnostic_area"), ft, spi_service_.spi_clear_diagnostic_area_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { // package id
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_extend_collection"), ft, spi_service_.spi_extend_collection_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_delete_collection"), ft, spi_service_.spi_delete_collection_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_trim_collection"),
ft, spi_service_.spi_trim_collection_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_cursor_init"), ft, spi_service_.spi_cursor_init_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type)); //函数第一个参数必须是基础环境信息隐藏参数
OZ (arg_types.push_back(char_type)); //sql
OZ (arg_types.push_back(char_type));//id
OZ (arg_types.push_back(int64_type));//type
OZ (arg_types.push_back(bool_type));//for_update
OZ (arg_types.push_back(bool_type));//hidden_rowid
OZ (arg_types.push_back(int_pointer_type));//sql_param_exprs
OZ (arg_types.push_back(int64_type));//sql_param_count
OZ (arg_types.push_back(int64_type));//package_id
OZ (arg_types.push_back(int64_type));//routine_id
OZ (arg_types.push_back(int64_type));//cursor_index
OZ (arg_types.push_back(int_pointer_type));//formal_param_idxs
OZ (arg_types.push_back(int_pointer_type));//actual_param_exprs
OZ (arg_types.push_back(int64_type));//cursor_param_count
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_cursor_open"), ft, spi_service_.spi_cursor_open_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type)); //函数第一个参数必须是基础环境信息隐藏参数
OZ (arg_types.push_back(int64_type)); //sql_expr
OZ (arg_types.push_back(int_pointer_type));//sql_param_exprs
OZ (arg_types.push_back(int64_type));//sql_param_count
OZ (arg_types.push_back(int64_type));//package_id
OZ (arg_types.push_back(int64_type));//routine_id
OZ (arg_types.push_back(int64_type));//cursor_index
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_dynamic_open"), ft, spi_service_.spi_dynamic_open_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type)); //函数第一个参数必须是基础环境信息隐藏参数
OZ (arg_types.push_back(int64_type));//package_id
OZ (arg_types.push_back(int64_type));//routine_id
OZ (arg_types.push_back(int64_type));//cursor_index
OZ (arg_types.push_back(int_pointer_type));//into_exprs
OZ (arg_types.push_back(int64_type));//into_count
OZ (arg_types.push_back(data_type_pointer_type));//column_types
OZ (arg_types.push_back(int64_type));//type_count
OZ (arg_types.push_back(bool_type_pointer_type));//exprs_not_null_flag
OZ (arg_types.push_back(int_pointer_type));//pl_integer_ranges
OZ (arg_types.push_back(bool_type));//is_bulk
OZ (arg_types.push_back(int64_type));//limit
OZ (arg_types.push_back(data_type_pointer_type));//return_type
OZ (arg_types.push_back(int64_type));//return_type_count
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_cursor_fetch"), ft, spi_service_.spi_cursor_fetch_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type)); //函数第一个参数必须是基础环境信息隐藏参数
OZ (arg_types.push_back(int64_type));//package_id
OZ (arg_types.push_back(int64_type));//routine_id
OZ (arg_types.push_back(int64_type));//cursor_index
OZ (arg_types.push_back(bool_type));//ignore
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_cursor_close"), ft, spi_service_.spi_cursor_close_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_destruct_collection"), ft, spi_service_.spi_destruct_collection_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //src
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //dest
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //row_size
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { // package id
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_init_collection"), ft, spi_service_.spi_init_collection_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(int64_type))) { //src
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_reset_collection"), ft, spi_service_.spi_reset_collection_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { // 函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //src index
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //dest index
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int32_type))) { //subarray lower pos
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int32_type))) { //subarray upper pos
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_sub_nestedtable"), ft, spi_service_.spi_sub_nestedtable_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type)); //函数第一个参数必须是基础环境信息隐藏参数
OZ (arg_types.push_back(int64_type)); //allocator
OZ (arg_types.push_back(obj_pointer_type)); //src
OZ (arg_types.push_back(obj_pointer_type)); //dest
OZ (arg_types.push_back(data_type_pointer_type)); //dest type
OZ (arg_types.push_back(int64_type)); // package id
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_copy_datum"), ft, spi_service_.spi_copy_datum_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { // error code
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(bool_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_set_pl_exception_code"), ft, spi_service_.spi_set_pl_exception_code_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("spi_get_pl_exception_code"), ft, spi_service_.spi_get_pl_exception_code_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(int64_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_raise_application_error"), ft, spi_service_.spi_raise_application_error_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_check_early_exit"), ft, spi_service_.spi_check_early_exit_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (arg_types.push_back(obj_param_pointer_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(
ObString("spi_pipe_row_to_result"), ft, spi_service_.spi_pipe_row_to_result_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (arg_types.push_back(int64_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_check_exception_handler_legal"),
ft, spi_service_.spi_check_exception_handler_legal_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (arg_types.push_back(int64_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_interface_impl"),
ft, spi_service_.spi_interface_impl_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (arg_types.push_back(int64_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_process_nocopy_params"),
ft, spi_service_.spi_process_nocopy_params_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type)); //函数第一个参数必须是基础环境信息隐藏参数
OZ (arg_types.push_back(int64_type)); //allocator
OZ (arg_types.push_back(obj_pointer_type)); //src
OZ (arg_types.push_back(obj_pointer_type)); //dest
OZ (arg_types.push_back(data_type_pointer_type)); //dest type
OZ (arg_types.push_back(int64_type)); // package id
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_copy_ref_cursor"), ft, spi_service_.spi_copy_ref_cursor_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type)); //函数第一个参数必须是基础环境信息隐藏参数
OZ (arg_types.push_back(obj_pointer_type)); // ref cursor
OZ (arg_types.push_back(int64_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_add_ref_cursor_refcount"), ft, spi_service_.spi_add_ref_cursor_refcount_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type)); //函数第一个参数必须是基础环境信息隐藏参数
OZ (arg_types.push_back(int64_type)); //package id
OZ (arg_types.push_back(int64_type)); //routine id
OZ (arg_types.push_back(int64_type)); //cursor index
OZ (arg_types.push_back(int64_type)); //addend
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_handle_ref_cursor_refcount"), ft, spi_service_.spi_handle_ref_cursor_refcount_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(int64_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_update_package_change_info"),
ft, spi_service_.spi_update_package_change_info_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(obj_param_pointer_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_check_composite_not_null"),
ft, spi_service_.spi_check_composite_not_null_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (arg_types.push_back(int64_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(
ObString("spi_update_location"), ft, spi_service_.spi_update_location_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(pl_exec_context_pointer_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(char_type));
OZ (arg_types.push_back(int_pointer_type));
OZ (arg_types.push_back(char_type));
OZ (arg_types.push_back(bool_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(ObString("spi_process_resignal"), ft, spi_service_.spi_process_resignal_error_));
}
return ret;
}
int ObPLCodeGenerator::init_adt_service()
{
int ret = OB_SUCCESS;
if (OB_FAIL(adt_service_.init(get_param_size(), ast_.get_symbol_table().get_count()))) {
LOG_WARN("failed to init adt service", K(ret));
}
return ret;
}
int ObPLCodeGenerator::init_eh_service()
{
int ret = OB_SUCCESS;
eh_service_.pl_exception_class_ = ObPLEHService::get_exception_class();
eh_service_.pl_exception_base_offset_ = ObPLEHService::get_exception_base_offset();
ObSEArray<ObLLVMType, 8> arg_types;
ObLLVMFunctionType ft;
ObLLVMType unwind_exception_type;
ObLLVMType unwind_exception_pointer_type;
ObLLVMType condition_type;
ObLLVMType condition_pointer_type;
ObLLVMType obj_type;
ObLLVMType obj_pointer_type;
ObLLVMType objparam_type;
ObLLVMType objparam_pointer_type;
ObLLVMType int_pointer_type;
ObLLVMType int32_pointer_type;
ObLLVMType tinyint_pointer_type;
ObLLVMType char_pointer_type;
ObLLVMType int64_type;
ObLLVMType int32_type;
ObLLVMType int8_type;
ObLLVMType char_type;
ObLLVMType void_type;
if (OB_FAIL(adt_service_.get_unwind_exception(unwind_exception_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(unwind_exception_type.get_pointer_to(unwind_exception_pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(adt_service_.get_pl_condition_value(condition_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(condition_type.get_pointer_to(condition_pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(adt_service_.get_obj(obj_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(obj_type.get_pointer_to(obj_pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(adt_service_.get_objparam(objparam_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(objparam_type.get_pointer_to(objparam_pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObIntType, int64_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObInt32Type, int32_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObTinyIntType, int8_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObCharType, char_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.get_void_type(void_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(int64_type.get_pointer_to(int_pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(int32_type.get_pointer_to(int32_pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(int8_type.get_pointer_to(tinyint_pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(char_type.get_pointer_to(char_pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
if (OB_FAIL(arg_types.push_back(int64_type))) { // obplexeccontext
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { // obplfunction
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { // line number
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(condition_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(unwind_exception_pointer_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("eh_create_exception"), ft, eh_service_.eh_create_exception_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(unwind_exception_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("_Unwind_RaiseException"), ft, eh_service_.eh_raise_exception_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(unwind_exception_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(void_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("_Unwind_Resume"), ft, eh_service_.eh_resume_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(int32_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int32_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int8_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int8_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("eh_personality"), ft, eh_service_.eh_personality_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(int8_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int32_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(char_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(int_pointer_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int32_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("eh_convert_exception"), ft, eh_service_.eh_convert_exception_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
arg_types.reset();
if (OB_FAIL(arg_types.push_back(char_type))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(int64_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(ObString("eh_classify_exception"), ft, eh_service_.eh_classify_exception))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
//for debug
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(char_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(int64_type));
OZ (ObLLVMFunctionType::get(void_type, arg_types, ft));
OZ (helper_.create_function(ObString("eh_debug_int64"), ft, eh_service_.eh_debug_int64_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(char_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(int_pointer_type));
OZ (ObLLVMFunctionType::get(void_type, arg_types, ft));
OZ (helper_.create_function(ObString("eh_debug_int64ptr"), ft, eh_service_.eh_debug_int64ptr_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(char_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(int32_type));
OZ (ObLLVMFunctionType::get(void_type, arg_types, ft));
OZ (helper_.create_function(ObString("eh_debug_int32"), ft, eh_service_.eh_debug_int32_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(char_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(int32_pointer_type));
OZ (ObLLVMFunctionType::get(void_type, arg_types, ft));
OZ (helper_.create_function(ObString("eh_debug_int32ptr"), ft, eh_service_.eh_debug_int32ptr_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(char_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(int8_type));
OZ (ObLLVMFunctionType::get(void_type, arg_types, ft));
OZ (helper_.create_function(ObString("eh_debug_int8"), ft, eh_service_.eh_debug_int8_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(char_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(tinyint_pointer_type));
OZ (ObLLVMFunctionType::get(void_type, arg_types, ft));
OZ (helper_.create_function(ObString("eh_debug_int8ptr"), ft, eh_service_.eh_debug_int8ptr_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(char_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(obj_pointer_type));
OZ (ObLLVMFunctionType::get(void_type, arg_types, ft));
OZ (helper_.create_function(ObString("eh_debug_obj"), ft, eh_service_.eh_debug_obj_));
}
if (OB_SUCC(ret)) {
arg_types.reset();
OZ (arg_types.push_back(char_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(objparam_pointer_type));
OZ (ObLLVMFunctionType::get(void_type, arg_types, ft));
OZ (helper_.create_function(ObString("eh_debug_objparam"), ft, eh_service_.eh_debug_objparam_));
}
return ret;
}
int ObPLCodeGenerator::set_var_addr_to_param_store(
int64_t var_index, jit::ObLLVMValue &var, jit::ObLLVMValue &init_value)
{
int ret = OB_SUCCESS;
ObLLVMValue ir_index;
ObLLVMType int_type;
ObLLVMValue var_addr;
if (OB_FAIL(helper_.get_int64(var_index, ir_index))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObIntType, int_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.create_ptr_to_int(ObString("cast_ptr_to_int64"), var, int_type, var_addr))) {
LOG_WARN("failed to create ptr to int", K(ret));
#ifndef NDEBUG
} else if (OB_FAIL(generate_debug(ObString("debug"), var_addr))) {
LOG_WARN("failed to create call", K(ret));
#endif
} else {
ObSEArray<ObLLVMValue, 4> args;
if (OB_FAIL(args.push_back(vars_.at(CTX_IDX)))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(args.push_back(ir_index))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(args.push_back(var_addr))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(args.push_back(init_value))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(helper_.create_call(ObString("set_var_addr"), get_user_type_var_func(), args))) {
LOG_WARN("failed to create call", K(ret));
} else { /*do nothing*/ }
}
return ret;
}
int ObPLCodeGenerator::generate_get_collection_attr(ObLLVMValue &param_array,
const ObObjAccessIdx &current_access,
int64_t access_i,
bool for_write,
ObLLVMValue &current_value,
ObLLVMValue &ret_value_ptr,
ObLLVMBasicBlock& exit)
{
int ret = OB_SUCCESS;
ObLLVMValue ret_value;
// 对于数组类型的访问, 需要确保其已经通过构造函数进行了初始化
ObLLVMValue is_inited, not_init_value, not_init;
ObLLVMBasicBlock after_init_block, not_init_block;
OZ (helper_.create_block(ObString("after_init_block"), func_, after_init_block));
OZ (helper_.create_block(ObString("not_init_block"), func_, not_init_block));
OZ (generate_debug("generate_get_collection_attr", current_value));
OZ (extract_count_from_collection(current_value, is_inited));
OZ (helper_.get_int64(-1, not_init_value));
OZ (helper_.create_icmp(is_inited, not_init_value, ObLLVMHelper::ICMP_EQ, not_init));
OZ (helper_.create_cond_br(not_init, not_init_block, after_init_block));
OZ (helper_.set_insert_point(not_init_block));
OZ (helper_.get_int32(OB_NOT_INIT, ret_value));
OZ (helper_.create_store(ret_value, ret_value_ptr));
OZ (helper_.create_br(exit));
OZ (helper_.set_insert_point(after_init_block));
ObLLVMValue data_value;
if (!current_access.is_property()) {
// 取出集合中元素的值ObCollection->data_
// 当时获取Collection的固有属性时, 不需要这么做, 固有属性都存储在Collection本身上
OZ (extract_data_from_collection(current_value, data_value));
}
// 获取下标或者固有属性在Collection中的位置信息
if (OB_SUCC(ret)) {
ObLLVMValue element_idx;
// 常量, 直接Codegen掉
if (current_access.is_property()) {
//do nothing
} else if (current_access.is_const()) {
OZ (helper_.get_int64(current_access.var_index_ - 1, element_idx));
} else {
// 非常量,那么首先用obj access中的var_index获取到变量的值,然后再获取element
ObLLVMValue element_idx_ptr;
OZ (helper_.create_gep(ObString("param_value"),
param_array,
current_access.var_index_,
element_idx_ptr));
OZ (helper_.create_load(ObString("element_idx"), element_idx_ptr, element_idx));
OZ (helper_.create_dec(element_idx, element_idx));
}
if (OB_SUCC(ret)) {
// 如果不是固有属性, 需要check下标的合法性, 避免访问越界
if (!current_access.is_property()) {
ObLLVMValue low, high_ptr, high, is_true;
ObLLVMBasicBlock check_block, after_block, error_block;
ObLLVMBasicBlock delete_block, after_delete_block;
char check_block_name[OB_MAX_OBJECT_NAME_LENGTH];
char check_after_name[OB_MAX_OBJECT_NAME_LENGTH];
char check_error_name[OB_MAX_OBJECT_NAME_LENGTH];
char delete_block_name[OB_MAX_OBJECT_NAME_LENGTH];
char after_delete_name[OB_MAX_OBJECT_NAME_LENGTH];
int64_t check_pos = 0, after_pos = 0, error_pos = 0;
int64_t delete_pos = 0, after_delete_pos = 0;
OZ (databuff_printf(check_block_name,
OB_MAX_OBJECT_NAME_LENGTH,
check_pos,
"check_block_%ld",
access_i));
OZ (databuff_printf(check_after_name,
OB_MAX_OBJECT_NAME_LENGTH,
after_pos,
"after_block_%ld",
access_i));
OZ (databuff_printf(check_error_name,
OB_MAX_OBJECT_NAME_LENGTH,
error_pos,
"check_error_%ld",
access_i));
OZ (databuff_printf(delete_block_name,
OB_MAX_OBJECT_NAME_LENGTH,
delete_pos,
"delete_pos_%ld",
access_i));
OZ (databuff_printf(after_delete_name,
OB_MAX_OBJECT_NAME_LENGTH,
after_delete_pos,
"after_delete_pos_%ld",
access_i));
OZ (helper_.create_block(ObString(check_pos, check_block_name), func_, check_block));
OZ (helper_.create_block(ObString(after_pos, check_after_name), func_, after_block));
OZ (helper_.create_block(ObString(error_pos, check_error_name), func_, error_block));
OZ (helper_.get_int64(0, low));
OZ (helper_.create_icmp(element_idx, low, ObLLVMHelper::ICMP_SGE, is_true));
OZ (helper_.create_cond_br(is_true, check_block, error_block));
OZ (helper_.set_insert_point(check_block));
OZ (helper_.create_gep(ObString("rowcount"),
current_value,
IDX_COLLECTION_COUNT,
high_ptr));
OZ (helper_.create_load(ObString("load_rowcount"), high_ptr, high));
OZ (helper_.create_icmp(high, element_idx, ObLLVMHelper::ICMP_SGT, is_true));
OZ (helper_.create_cond_br(is_true, after_block, error_block));
OZ (helper_.set_insert_point(error_block));
OZ (helper_.get_int32(OB_ARRAY_OUT_OF_RANGE, ret_value));
OZ (helper_.create_store(ret_value, ret_value_ptr));
OZ (helper_.create_br(exit));
OZ (helper_.set_insert_point(after_block));
OZ (helper_.get_int32(OB_SUCCESS, ret_value));
OZ (helper_.create_store(ret_value, ret_value_ptr));
OZ (helper_.create_gep(ObString("table_element_array"),
data_value,
element_idx,
current_value));
if (OB_SUCC(ret)) {
// check deleted value
ObLLVMValue p_type_value;
ObLLVMValue type_value;
ObLLVMValue is_deleted;
OZ (helper_.create_block(ObString(delete_pos, delete_block_name), func_, delete_block));
OZ (helper_.create_block(ObString(after_delete_pos, after_delete_name), func_, after_delete_block));
OZ (extract_type_ptr_from_obj(current_value, p_type_value));
OZ (helper_.create_load(ObString("load_type"), p_type_value, type_value));
OZ (helper_.create_icmp_eq(type_value, ObMaxType, is_deleted));
OZ (helper_.create_cond_br(is_deleted, delete_block, after_delete_block));
OZ (helper_.set_insert_point(delete_block));
if (!for_write) {
OZ (helper_.get_int32(OB_READ_NOTHING, ret_value));
OZ (helper_.create_store(ret_value, ret_value_ptr));
OZ (helper_.create_br(exit));
} else {
if (current_access.var_type_.is_composite_type()) {
OZ (helper_.get_int8(ObExtendType, type_value));
OZ (helper_.create_store(type_value, p_type_value));
}
OZ (helper_.create_br(after_delete_block));
}
OZ (helper_.set_insert_point(after_delete_block));
}
} else {
//do nothing
}
}
}
return ret;
}
int ObPLCodeGenerator::generate_get_record_attr(const ObObjAccessIdx &current_access,
uint64_t udt_id,
bool for_write,
ObLLVMValue &current_value,
ObLLVMValue &ret_value_ptr,
ObLLVMBasicBlock& exit)
{
int ret = OB_SUCCESS;
ObLLVMValue ret_value;
ObLLVMValue data_value;
const ObUserDefinedType *user_type = NULL;
const ObRecordType *record_type = NULL;
int64_t field_cnt = OB_INVALID_COUNT;
int64_t element_idx = RECORD_META_OFFSET; //type_ is 0 and count_ is 1
if (NULL == (user_type = ast_.get_user_type_table().get_type(udt_id))
&& NULL == (user_type = ast_.get_user_type_table().get_external_type(udt_id))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("get user type failed", K(ret));
}
CK (OB_NOT_NULL(user_type));
CK (user_type->is_record_type());
OX (record_type = static_cast<const ObRecordType*>(user_type));
OX (element_idx += record_type->get_record_member_count()); //null map
OX (element_idx += record_type->get_record_member_count()); //record meta list
CK (current_access.is_const());
OX (element_idx += current_access.var_index_); //访问的域对应的下标
if (OB_SUCC(ret) && user_type->is_object_type() && for_write) {
ObLLVMBasicBlock null_block, not_null_block;
ObLLVMValue is_null, err_value, is_null_object;
OZ (extract_isnull_from_record(current_value, is_null_object));
#ifndef NDEBUG
OZ (generate_debug(ObString("object instance is null"), is_null_object));
#endif
OZ (get_helper().create_block(ObString("null_block"), get_func(), null_block));
OZ (get_helper().create_block(ObString("not_null_block"), get_func(), not_null_block));
OZ (get_helper().create_icmp_eq(is_null_object, TRUE, is_null));
OZ (get_helper().create_cond_br(is_null, null_block, not_null_block));
OZ (set_current(null_block));
OZ (get_helper().get_int32(OB_ERR_ACCESS_INTO_NULL, err_value));
OZ (helper_.create_store(err_value, ret_value_ptr));
OZ (helper_.create_br(exit));
OZ (set_current(not_null_block));
}
OZ (helper_.create_gep(ObString("extract record data"),
current_value,
element_idx,
current_value));
return ret;
}
int ObPLCodeGenerator::generate_get_attr(ObLLVMValue &param_array,
const ObIArray<ObObjAccessIdx> &obj_access,
bool for_write,
ObLLVMValue &value_ptr, // 读取的值
ObLLVMValue &ret_value_ptr, // 标识是否越界
ObLLVMBasicBlock& exit) // 越界后需要跳转的BLOCK
{
int ret = OB_SUCCESS;
ObLLVMType user_type;
ObLLVMValue composite_addr;
ObLLVMValue ret_value;
ObLLVMValue value;
ObLLVMType int64_type;
CK (!obj_access.empty());
CK (obj_access.at(0).var_type_.is_composite_type()
|| obj_access.at(0).var_type_.is_cursor_type());
// 获取数据类型
OZ (helper_.get_llvm_type(ObIntType, int64_type));
// 获取变量首地址
OZ (helper_.get_int32(OB_SUCCESS, ret_value));
OZ (helper_.create_store(ret_value, ret_value_ptr));
OZ (helper_.create_gep(ObString("param_value"),
param_array,
obj_access.at(0).var_index_,
composite_addr));
OZ (helper_.create_load(ObString("element_idx"), composite_addr, composite_addr));
// 逐级解析复杂数据类型
for (int64_t i = 1; OB_SUCC(ret) && i < obj_access.count(); ++i) {
const ObPLDataType &parent_type = obj_access.at(i - 1).var_type_;
OZ (user_type_map_.get_refactored(parent_type.get_user_type_id(), user_type),
parent_type.get_user_type_id());
OZ (user_type.get_pointer_to(user_type));
OZ (helper_.create_int_to_ptr(ObString("var_value"), composite_addr, user_type, value));
CK (!obj_access.at(i).is_invalid());
if (OB_SUCC(ret)) {
if (parent_type.is_collection_type()) { //数组类型
OZ (generate_get_collection_attr(param_array,
obj_access.at(i),
i,
for_write,
value,
ret_value_ptr,
exit));
} else if (parent_type.is_record_type()) { //record类型
OZ (generate_get_record_attr(obj_access.at(i),
parent_type.get_user_type_id(),
for_write,
value,
ret_value_ptr,
exit));
} else {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("unexpected user type" , K(obj_access.at(i - 1).var_type_), K(ret));
}
}
if (obj_access.at(i).var_type_.is_composite_type()) {
OZ (extract_value_from_obj(value, ObExtendType, composite_addr));
}
}
if (ObObjAccessIdx::get_final_type(obj_access).is_obj_type()) {
OZ (helper_.create_ptr_to_int(ObString("element_value_addr"), value, int64_type, value));
} else {
value = composite_addr;
}
OZ (helper_.create_store(value, value_ptr));
return ret;
}
int ObPLCodeGenerator::generate_declare_cursor(const ObPLStmt &s, const int64_t &cursor_index)
{
int ret = OB_SUCCESS;
if (NULL == get_current().get_v()) {
// 控制流已断,后面的语句不再处理
} else {
const ObPLCursor *cursor = s.get_cursor(cursor_index);
CK (OB_NOT_NULL(cursor));
CK (ObPLCursor::INVALID != cursor->get_state());
if (OB_SUCC(ret) && ObPLCursor::DUP_DECL != cursor->get_state()) { //如果是无效的cursor跳过即可
ObLLVMValue cursor_index_value;
ObLLVMValue cursor_value;
ObLLVMValue ret_err;
ObSEArray<ObLLVMValue, 2> args;
OZ (get_helper().set_insert_point(get_current()));
OZ (set_debug_location(s));
OZ (get_helper().get_int64(cursor->get_index(), cursor_index_value));
OZ (args.push_back(get_vars().at(CTX_IDX)));
OZ (args.push_back(cursor_index_value));
OZ (get_helper().create_call(ObString("spi_cursor_init"), get_spi_service().spi_cursor_init_, args, ret_err));
OZ (check_success(ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
OZ (extract_objparam_from_context(get_vars().at(CTX_IDX), cursor->get_index(), cursor_value));
OZ (helper_.create_gep(ObString("obj"), cursor_value, 0, cursor_value));
OX (get_vars().at(cursor->get_index() + USER_ARG_OFFSET) = cursor_value);
}
}
return ret;
}
int ObPLCodeGenerator::generate_open(
const ObPLStmt &s, const ObPLSql &cursor_sql,
const uint64_t package_id, const uint64_t routine_id, const int64_t cursor_index)
{
int ret = OB_SUCCESS;
if (NULL == get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(get_helper().set_insert_point(get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else {
ObSEArray<ObLLVMValue, 8> args;
//sql & ps_id & stmt_type & params & param count
ObLLVMValue str;
ObLLVMValue len;
ObLLVMValue ps_sql;
ObLLVMValue type;
ObLLVMValue for_update;
ObLLVMValue hidden_rowid;
ObLLVMValue sql_params;
ObLLVMValue sql_param_count;
ObLLVMValue package_id_value;
ObLLVMValue routine_id_value;
ObLLVMValue cursor_index_value;
ObLLVMValue formal_params;
ObLLVMValue actual_params;
ObLLVMValue cursor_param_count;
ObLLVMValue ret_err;
OZ (args.push_back(get_vars().at(CTX_IDX)));
OZ (generate_sql(cursor_sql, str, len, ps_sql, type, for_update, hidden_rowid, sql_params,
sql_param_count));
OZ (args.push_back(str));
OZ (args.push_back(ps_sql));
OZ (args.push_back(type));
OZ (args.push_back(for_update));
OZ (args.push_back(hidden_rowid));
OZ (args.push_back(sql_params));
OZ (args.push_back(sql_param_count));
OZ (get_helper().get_int64(package_id, package_id_value));
OZ (args.push_back(package_id_value));
OZ (get_helper().get_int64(routine_id, routine_id_value));
OZ (args.push_back(routine_id_value));
OZ (get_helper().get_int64(cursor_index, cursor_index_value));
OZ (args.push_back(cursor_index_value));
if (PL_OPEN == s.get_type() || PL_OPEN_FOR == s.get_type()) {
const ObPLOpenStmt &open_stmt = static_cast<const ObPLOpenStmt &>(s);
const ObPLCursor *cursor = open_stmt.get_cursor();
OZ (generate_int64_array(cursor->get_formal_params(), formal_params));
OZ (generate_expression_array(static_cast<const ObPLOpenStmt&>(s).get_params(), actual_params, cursor_param_count));
} else if (PL_CURSOR_FOR_LOOP == s.get_type()) {
const ObPLCursorForLoopStmt& for_stmt = static_cast<const ObPLCursorForLoopStmt&>(s);
const ObPLCursor *cursor = for_stmt.get_cursor();
OZ (generate_int64_array(cursor->get_formal_params(), formal_params));
OZ (generate_expression_array(static_cast<const ObPLCursorForLoopStmt&>(s).get_params(), actual_params, cursor_param_count));
} else { //must be OPEN FOR
OZ (generate_null_pointer(ObIntType, formal_params));
OZ (generate_null_pointer(ObIntType, actual_params));
OZ (helper_.get_int64(0, cursor_param_count));
}
OZ (args.push_back(formal_params));
OZ (args.push_back(actual_params));
OZ (args.push_back(cursor_param_count));
OZ (get_helper().create_call(ObString("spi_cursor_open"), get_spi_service().spi_cursor_open_, args, ret_err));
OZ (check_success(ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
}
return ret;
}
int ObPLCodeGenerator::generate_open_for(const ObPLOpenForStmt &s)
{
int ret = OB_SUCCESS;
CK (OB_NOT_NULL(s.get_cursor()));
// 这个不能,因为如果是ref cursor,这个ref count可能会有多个,这个调用会清零
// 同时在ob_spi.cpp的 open_cursor_check函数中,reopen之前做了close操作
OZ (generate_close(s,
s.get_package_id(),
s.get_cursor()->get_routine_id(),
s.get_index(),
true/*ignoe if not opened*/));
if (OB_SUCC(ret)) {
if (s.is_dynamic()) {
if (NULL == get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(get_helper().set_insert_point(get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else {
ObArray<int64_t> using_exprs;
ObSEArray<ObLLVMValue, 8> args;
ObLLVMValue sql_addr;
ObLLVMValue sql_params;
ObLLVMValue sql_param_count;
ObLLVMValue package_id_value;
ObLLVMValue routine_id_value;
ObLLVMValue cursor_index_value;
ObLLVMValue ret_err;
OZ (args.push_back(get_vars().at(CTX_IDX)));
OZ (generate_pointer(get_expr(s.get_dynamic_sql()), sql_addr));
OZ (args.push_back(sql_addr));
for (int64_t i = 0; OB_SUCC(ret) && i < s.get_using().count(); ++i) {
OZ (using_exprs.push_back(s.get_using_index(i)));
}
OZ (generate_expression_array(using_exprs, sql_params, sql_param_count));
OZ (args.push_back(sql_params));
OZ (args.push_back(sql_param_count));
OZ (get_helper().get_int64(s.get_cursor()->get_package_id(), package_id_value));
OZ (args.push_back(package_id_value));
OZ (get_helper().get_int64(s.get_cursor()->get_routine_id(), routine_id_value));
OZ (args.push_back(routine_id_value));
OZ (get_helper().get_int64(s.get_index(), cursor_index_value));
OZ (args.push_back(cursor_index_value));
OZ (get_helper().create_call(ObString("spi_dynamic_open"),
get_spi_service().spi_dynamic_open_, args, ret_err));
OZ (check_success(ret_err, s.get_stmt_id(), s.get_block()->in_notfound(),
s.get_block()->in_warning()));
}
} else {
OZ(generate_open(s,
s.get_static_sql(),
s.get_cursor()->get_package_id(),
s.get_cursor()->get_routine_id(),
s.get_index()));
}
}
return ret;
}
int ObPLCodeGenerator::generate_fetch(const ObPLStmt &s,
const ObPLInto &into,
const uint64_t &package_id,
const uint64_t &routine_id,
const int64_t &cursor_index,
const int64_t &limit,
const ObUserDefinedType *user_defined_type,
const ObIArray<ObDataType> &user_types,
jit::ObLLVMValue &ret_err)
{
int ret = OB_SUCCESS;
if (NULL == get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(get_helper().set_insert_point(get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else {
ObSEArray<ObLLVMValue, 8> args;
ObLLVMValue package_id_value;
ObLLVMValue routine_id_value;
ObLLVMValue cursor_index_value;
OZ (args.push_back(get_vars().at(CTX_IDX)));
OZ (get_helper().get_int64(package_id, package_id_value));
OZ (args.push_back(package_id_value));
OZ (get_helper().get_int64(routine_id, routine_id_value));
OZ (args.push_back(routine_id_value));
OZ (get_helper().get_int64(cursor_index, cursor_index_value));
OZ (args.push_back(cursor_index_value));
ObLLVMValue into_array_value;
ObLLVMValue into_count_value;
ObLLVMValue type_array_value;
ObLLVMValue type_count_value;
ObLLVMValue exprs_not_null_array_value;
ObLLVMValue pl_integer_array_value;
ObLLVMValue is_bulk;
OZ (generate_into(into, into_array_value, into_count_value,
type_array_value, type_count_value,
exprs_not_null_array_value,
pl_integer_array_value,
is_bulk));
OZ (args.push_back(into_array_value));
OZ (args.push_back(into_count_value));
OZ (args.push_back(type_array_value));
OZ (args.push_back(type_count_value));
OZ (args.push_back(exprs_not_null_array_value));
OZ (args.push_back(pl_integer_array_value));
OZ (args.push_back(is_bulk));
//limit
if (OB_SUCC(ret)) {
ObLLVMValue limit_value;
if (limit != INT64_MAX) {
ObLLVMBasicBlock null_block, not_null_block;
ObLLVMValue p_limit_value, p_type_value, type_value, is_null;
ObLLVMValue result;
OZ (get_helper().create_block(ObString("null block"), get_func(), null_block));
OZ (get_helper().create_block(ObString("not null block"), get_func(), not_null_block));
OZ (generate_expr(limit, s, OB_INVALID_INDEX, p_limit_value));
OZ (extract_type_ptr_from_objparam(p_limit_value, p_type_value));
OZ (get_helper().create_load(ObString("load_type"), p_type_value, type_value));
OZ (get_helper().create_icmp_eq(type_value, ObNullType, is_null));
OZ (get_helper().create_cond_br(is_null, null_block, not_null_block));
OZ (set_current(null_block));
OZ (get_helper().get_int32(OB_ERR_NULL_VALUE, result));
OZ (check_success(result, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
OZ (get_helper().create_br(not_null_block));
OZ (set_current(not_null_block));
CK (OB_NOT_NULL(s.get_expr(limit)));
OZ (extract_value_from_objparam(p_limit_value, s.get_expr(limit)->get_data_type(), limit_value));
} else {
OZ (get_helper().get_int64(limit, limit_value));
}
OZ (args.push_back(limit_value));
}
ObLLVMValue return_type_array_value;
ObLLVMValue return_type_count_value;
ObLLVMValue type_value;
ObLLVMType data_type;
ObLLVMType data_type_pointer;
ObLLVMType array_type;
if (OB_ISNULL(user_defined_type)) {
OZ (adt_service_.get_data_type(data_type));
OZ (data_type.get_pointer_to(data_type_pointer));
OZ (ObLLVMHelper::get_null_const(data_type_pointer, return_type_array_value));
OZ (helper_.get_int64(0, return_type_count_value));
} else {
OZ (adt_service_.get_data_type(data_type));
OZ (data_type.get_pointer_to(data_type_pointer));
OZ (ObLLVMHelper::get_array_type(data_type,
user_types.count(),
array_type));
OZ (helper_.create_alloca(ObString("datatype_array"),
array_type,
return_type_array_value));
for (int64_t i = 0; OB_SUCC(ret) && i < user_types.count(); ++i) {
type_value.reset();
OZ (helper_.create_gep(ObString("extract_datatype"),
return_type_array_value,
i, type_value));
OZ (store_data_type(user_types.at(i), type_value));
}
OZ (helper_.create_bit_cast(ObString("datatype_array_to_pointer"),
return_type_array_value, data_type_pointer, return_type_array_value));
OZ (helper_.get_int64(static_cast<int64_t>(user_types.count()),
return_type_count_value));
}
OZ (args.push_back(return_type_array_value));
OZ (args.push_back(return_type_count_value));
OZ (get_helper().create_call(ObString("spi_cursor_fetch"),
get_spi_service().spi_cursor_fetch_,
args, ret_err));
}
return ret;
}
int ObPLCodeGenerator::generate_close(const ObPLStmt &s,
const uint64_t &package_id,
const uint64_t &routine_id,
const int64_t &cursor_index,
bool ignore,
bool exception)
{
int ret = OB_SUCCESS;
if (NULL == get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(get_helper().set_insert_point(get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(set_debug_location(s))) {
LOG_WARN("failed to set debug location", K(ret));
} else {
ObLLVMValue ret_err;
ObSEArray<ObLLVMValue, 2> args;
ObLLVMValue cursor_index_value, ignore_value;
ObLLVMValue package_id_value;
ObLLVMValue routine_id_value;
OZ (args.push_back(get_vars().at(CTX_IDX)));
OZ (get_helper().get_int64(package_id, package_id_value));
OZ (args.push_back(package_id_value));
OZ (get_helper().get_int64(routine_id, routine_id_value));
OZ (args.push_back(routine_id_value));
OZ (get_helper().get_int64(cursor_index, cursor_index_value));
OZ (args.push_back(cursor_index_value));
OZ (get_helper().get_int8(ignore, ignore_value));
OZ (args.push_back(ignore_value));
OZ (get_helper().create_call(ObString("spi_cursor_close"), get_spi_service().spi_cursor_close_, args, ret_err));
if (OB_SUCC(ret) && exception) {
OZ (check_success(ret_err,
s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
}
}
return ret;
}
int ObPLCodeGenerator::generate_check_not_null(const ObPLStmt &s,
bool is_not_null,
ObLLVMValue &p_result_obj)
{
int ret = OB_SUCCESS;
if (is_not_null) {
ObLLVMBasicBlock illegal_block, legal_block;
ObLLVMBasicBlock check_composite, do_check_composite;
ObLLVMValue p_type_value, type_value, is_null, is_extend;
ObLLVMValue result, ret_err;
ObSEArray<ObLLVMValue, 1> args;
OZ (get_helper().create_block(ObString("illegal_block"), get_func(), illegal_block));
OZ (get_helper().create_block(ObString("legal_block"), get_func(), legal_block));
OZ (get_helper().create_block(ObString("check_composite"), get_func(), check_composite));
OZ (get_helper().create_block(ObString("do_check_composite"), get_func(), do_check_composite));
OZ (extract_type_ptr_from_objparam(p_result_obj, p_type_value));
OZ (get_helper().create_load(ObString("load_type"), p_type_value, type_value));
OZ (get_helper().create_icmp_eq(type_value, ObNullType, is_null));
OZ (get_helper().create_cond_br(is_null, illegal_block, check_composite));
OZ (set_current(illegal_block));
OZ (get_helper().get_int32(OB_ERR_NUMERIC_OR_VALUE_ERROR, result));
OZ (check_success(result, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
OZ (get_helper().create_br(legal_block));
OZ (set_current(check_composite));
OZ (get_helper().create_icmp_eq(type_value, ObExtendType, is_extend));
OZ (get_helper().create_cond_br(is_extend, do_check_composite, legal_block));
OZ (set_current(do_check_composite));
OZ (args.push_back(p_result_obj));
OZ (get_helper().create_call(ObString("spi_check_composite_not_null"),
get_spi_service().spi_check_composite_not_null_,
args,
ret_err));
OZ (check_success(ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
OZ (get_helper().create_br(legal_block));
OZ (set_current(legal_block));
}
return ret;
}
int ObPLCodeGenerator::generate_normal_next_and_check(const ObPLForLoopStmt &s,
ObLLVMValue &p_index_obj,
ObLLVMValue &p_index_value,
ObLLVMValue &index_obj,
ObLLVMValue &index_value,
ObLLVMValue &dest_datum,
ObLLVMValue &lower_value,
ObLLVMValue &upper_value,
ObLLVMValue &is_true)
{
int ret = OB_SUCCESS;
CK (OB_NOT_NULL(get_current().get_v()));
// 对INDEX +1/-1 并将结果存储进param_store
if (!s.get_reverse()) {
OZ (extract_value_ptr_from_obj(
p_index_obj, s.get_lower_expr()->get_data_type(), p_index_value));
OZ (get_helper().create_load(ObString("load_index_value"), p_index_value, index_value));
OZ (get_helper().create_inc(index_value, index_value));
OZ (get_helper().create_store(index_value, p_index_value));
OZ (cast_to_int64(p_index_value));
OZ (get_helper().create_load(ObString("load_index_obj"), p_index_obj, index_obj));
OZ (get_helper().create_store(index_obj, dest_datum));
OZ (get_helper().create_icmp(index_value, upper_value, ObLLVMHelper::ICMP_SLE, is_true));
} else {
OZ (extract_value_ptr_from_obj(
p_index_obj, s.get_upper_expr()->get_data_type(), p_index_value));
OZ (get_helper().create_load(ObString("load_index_value"), p_index_value, index_value));
OZ (get_helper().create_dec(index_value, index_value));
OZ (get_helper().create_store(index_value, p_index_value));
OZ (cast_to_int64(p_index_value));
OZ (get_helper().create_load(ObString("load_index_obj"), p_index_obj, index_obj));
OZ (get_helper().create_store(index_obj, dest_datum));
OZ (get_helper().create_icmp(index_value, lower_value, ObLLVMHelper::ICMP_SGE, is_true));
}
return ret;
}
int ObPLCodeGenerator::generate_expr_next_and_check(const ObPLForLoopStmt &s,
ObLLVMValue &p_index_obj,
ObLLVMValue &p_index_value,
ObLLVMValue &index_obj,
ObLLVMValue &index_value,
ObLLVMValue &dest_datum,
ObLLVMValue &upper_value,
ObLLVMValue &is_true)
{
int ret = OB_SUCCESS;
ObLLVMValue p_type_value, type_value, bool_value, p_is_continue_value;
ObLLVMBasicBlock normal_block, not_null_block, continue_block, not_continue_block;
CK (OB_NOT_NULL(s.get_next_expr()));
OZ (get_helper().create_block(ObString("normal_block"), get_func(), normal_block));
OZ (get_helper().create_block(ObString("not_null_block"), get_func(), not_null_block));
OZ (get_helper().create_block(ObString("continue_block"), get_func(), continue_block));
OZ (get_helper().create_block(ObString("not_continue_block"), get_func(), not_continue_block));
// 初始化一个变量用于存储比较的结果
OZ (get_helper().create_ialloca(
ObString("p_is_continue_value"), ObTinyIntType, TRUE, p_is_continue_value));
if (s.is_values_bound()) {
OX (p_index_obj = get_vars().at(s.get_ident() + USER_ARG_OFFSET));
OZ (get_helper().create_load(ObString("load_index_obj"), p_index_obj, index_obj));
OZ (get_helper().create_store(index_obj, dest_datum));
}
// 计算NextValue
OZ (generate_expr(s.get_next(), s, s.get_ident(), p_index_obj));
if (s.is_values_bound()) {
ObLLVMValue p_tmp_obj;
OX (p_tmp_obj = get_vars().at(s.get_ident() + USER_ARG_OFFSET));
OZ (extract_obobj_from_objparam(p_index_obj, index_obj));
OZ (get_helper().create_store(index_obj, p_tmp_obj));
}
// 判断NextValue是否为Null, Null说明已经迭代到结尾
OZ (extract_type_ptr_from_objparam(p_index_obj, p_type_value));
OZ (get_helper().create_load(ObString("load_type"), p_type_value, type_value));
OZ (get_helper().create_icmp_eq(type_value, ObNullType, bool_value));
OZ (get_helper().create_cond_br(bool_value, not_continue_block, not_null_block));
// 不为Null还需要跟UpperValue做比较, 如果已经大于UpperValue, 也说明迭代到结尾了
OZ (set_current(not_null_block));
OZ (extract_value_ptr_from_objparam(
p_index_obj, s.get_next_expr()->get_data_type(), p_index_value));
OZ (get_helper().create_load(ObString("load_index_value"), p_index_value, index_value));
OZ (get_helper().create_icmp(index_value, upper_value, ObLLVMHelper::ICMP_SLE, bool_value));
OZ (get_helper().create_cond_br(bool_value, continue_block, not_continue_block));
// 设置is_true的值, 因为is_true不是指针, 因此在这个函数中仅能且必须被设置一次
OZ (set_current(continue_block));
OZ (get_helper().create_istore(TRUE, p_is_continue_value));
OZ (get_helper().create_br(normal_block));
OZ (set_current(not_continue_block));
OZ (get_helper().create_istore(FALSE, p_is_continue_value));
OZ (get_helper().create_br(normal_block));
OZ (set_current(normal_block));
OZ (get_helper().create_load(ObString("load_null_value"), p_is_continue_value, bool_value));
OZ (get_helper().create_icmp_eq(bool_value, TRUE, is_true));
// 如果是values of子句, 需要将paramstore中index的位置赋值为数组的实际值而不是下标
if (s.is_values_bound()) {
ObLLVMBasicBlock do_nothing_block, value_block;
ObLLVMValue p_value;
OZ (get_helper().create_block(
ObString("do_nothing_block"), get_func(), do_nothing_block));
OZ (get_helper().create_block(
ObString("value_block"), get_func(), value_block));
OZ (get_helper().create_cond_br(is_true, value_block, do_nothing_block));
OZ (set_current(value_block));
OZ (generate_expr(s.get_value(), s, s.get_ident(), p_value));
OZ (get_helper().create_br(do_nothing_block));
OZ (set_current(do_nothing_block));
}
return ret;
}
int ObPLCodeGenerator::generate_next_and_check(const ObPLForLoopStmt &s,
ObLLVMValue &p_index_obj,
ObLLVMValue &p_index_value,
ObLLVMValue &index_obj,
ObLLVMValue &index_value,
ObLLVMValue &dest_datum,
ObLLVMValue &lower_value,
ObLLVMValue &upper_value,
ObLLVMValue &is_true)
{
int ret = OB_SUCCESS;
if (s.is_normal_bound()) {
OZ (generate_normal_next_and_check(s,
p_index_obj,
p_index_value,
index_obj,
index_value,
dest_datum,
lower_value,
upper_value,
is_true));
} else {
OZ (generate_expr_next_and_check(s,
p_index_obj,
p_index_value,
index_obj,
index_value,
dest_datum,
upper_value,
is_true));
}
return ret;
}
int ObPLCodeGenerator::generate_indices_with_between_bound(const ObPLForLoopStmt &s,
ObLLVMValue &p_lower_obj)
{
int ret = OB_SUCCESS;
ObLLVMValue p_exist_obj, exist_obj, exist_value, is_exist;
ObLLVMBasicBlock exist_block, not_exist_block;
CK (get_current().get_v());
OZ (get_helper().create_block(ObString("exist_block"),
get_func(),
exist_block));
OZ (get_helper().create_block(ObString("not_exist_block"),
get_func(),
not_exist_block));
OZ (generate_expr(s.get_exists(), s, OB_INVALID_INDEX, p_exist_obj));
OZ (extract_value_from_objparam(
p_exist_obj, s.get_exists_expr()->get_data_type(), exist_value));
OZ (get_helper().create_icmp_eq(exist_value, TRUE, is_exist));
OZ (get_helper().create_cond_br(is_exist, exist_block, not_exist_block));
OZ (set_current(not_exist_block));
OZ (generate_expr(s.get_next(), s, s.get_ident(), p_exist_obj));
OZ (get_helper().create_load(ObString("load_exist_lower_obj"), p_exist_obj, exist_obj));
OZ (get_helper().create_store(exist_obj, p_lower_obj));
OZ (get_helper().create_br(exist_block));
OZ (set_current(exist_block));
return ret;
}
int ObPLCodeGenerator::generate_bound_and_check(const ObPLForLoopStmt &s,
bool is_forall,
ObLLVMValue &lower_value, // lower value int64_t
ObLLVMValue &upper_value, // upper value int64_t
ObLLVMValue &lower_obj, // lower obj ObObject
ObLLVMValue &upper_obj, // upper obj ObObject
ObLLVMBasicBlock &illegal_range_block) // ret_err int32_t
{
int ret = OB_SUCCESS;
if (get_current().get_v()) {
// 控制流已断, 后面的语句不再处理
} else {
OZ (get_helper().set_insert_point(get_current()));
OZ (set_debug_location(s));
}
if (OB_SUCC(ret)) {
ObLLVMBasicBlock check_null_block, illegal_null_block, body_block, check_up_low_block;
ObLLVMValue p_lower_obj, p_upper_obj;
ObSEArray<ObLLVMValue, 5> args;
ObLLVMValue result;
CK (OB_NOT_NULL(s.get_body()));
CK (OB_NOT_NULL(s.get_lower_expr()));
CK (OB_NOT_NULL(s.get_upper_expr()));
OZ (get_helper().create_block(ObString("check_null_block"),
get_func(),
check_null_block));
OZ (get_helper().create_block(ObString("body_block"),
get_func(),
body_block));
OZ (get_helper().create_block(ObString("check_up_low_block"),
get_func(),
check_up_low_block));
OZ (get_helper().create_block(ObString("illegal_null_block"),
get_func(),
illegal_null_block));
// 计算lower, upper
OZ (generate_expr(s.get_lower(), s, s.get_ident(), p_lower_obj));
OZ (generate_expr(s.get_upper(), s, OB_INVALID_INDEX, p_upper_obj));
if (s.is_indices_with_between_bound()) {
OZ (generate_indices_with_between_bound(s, p_lower_obj));
}
if (OB_SUCC(ret)) {
// 在FORALL中上界或者下界为NULL, 不报错
// 在FORLOOP中上界或者下界为NULL, 报错
// 如果上界下界均不为NULL, 不报错, 范围不符合条件什么都不做
ObLLVMValue p_type_value, type_value, is_null, is_legal;
OZ (get_helper().get_int32(OB_SUCCESS, result));
OZ (extract_type_ptr_from_objparam(p_lower_obj, p_type_value));
OZ (get_helper().create_load(ObString("load_type"), p_type_value, type_value));
OZ (get_helper().create_icmp_eq(type_value, ObNullType, is_null));
OZ (get_helper().create_cond_br(is_null, illegal_null_block, check_null_block));
OZ (get_helper().set_insert_point(check_null_block));
OZ (extract_type_ptr_from_objparam(p_upper_obj, p_type_value));
OZ (get_helper().create_load(ObString("load_type"), p_type_value, type_value));
OZ (get_helper().create_icmp_eq(type_value, ObNullType, is_null));
OZ (get_helper().create_cond_br(is_null, illegal_null_block, check_up_low_block));
OZ (set_current(illegal_null_block));
OZ (get_helper().get_int32(OB_ERR_NULL_VALUE, result));
if (!is_forall) {
OZ (check_success(result, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
}
OZ (get_helper().create_br(illegal_range_block));
OZ (get_helper().set_insert_point(check_up_low_block));
OZ (extract_value_from_objparam(p_lower_obj, s.get_lower_expr()->get_data_type(), lower_value));
OZ (extract_datum_from_objparam(p_lower_obj, s.get_lower_expr()->get_data_type(), lower_obj));
OZ (extract_value_from_objparam(p_upper_obj, s.get_upper_expr()->get_data_type(), upper_value));
OZ (extract_datum_from_objparam(p_upper_obj, s.get_upper_expr()->get_data_type(), upper_obj));
/*!
* Without REVERSE, the value of index starts at lower_bound and increases by one with
* each iteration of the loop until it reaches upper_bound. If lower_bound is greater than
* upper_bound, then the statements never run.
* With REVERSE, the value of index starts at upper_bound and decreases by one with each
* iteration of the loop until it reaches lower_bound. If upper_bound is less than lower_
* bound, then the statements never run.
*/
OZ (get_helper().create_icmp(lower_value, upper_value, ObLLVMHelper::ICMP_SLE, is_legal));
OZ (get_helper().create_cond_br(is_legal, body_block, illegal_range_block));
OZ (set_current(body_block));
}
if (OB_SUCC(ret) && is_forall) {
const ObPLForAllStmt *forall_stmt = static_cast<const ObPLForAllStmt*>(&s);
CK (OB_NOT_NULL(forall_stmt));
if (OB_SUCC(ret)) {
ObLLVMValue src_idx_value, dst_idx_value, ret_err;
ObSEArray<ObLLVMValue, 5> args;
const hash::ObHashMap<int64_t, int64_t> &sub_map = forall_stmt->get_tab_to_subtab_map();
for (hash::ObHashMap<int64_t, int64_t>::const_iterator it = sub_map.begin();
OB_SUCC(ret) && it != sub_map.end(); ++it) {
args.reset();
int64_t src_idx = (*it).first;
int64_t dst_idx = (*it).second;
OZ (get_helper().get_int64(src_idx, src_idx_value));
OZ (get_helper().get_int64(dst_idx, dst_idx_value));
OZ (args.push_back(get_vars().at(CTX_IDX)));
OZ (args.push_back(src_idx_value));
OZ (args.push_back(dst_idx_value));
OZ (args.push_back(lower_value));
OZ (args.push_back(upper_value));
OZ (get_helper().create_call(ObString("spi_sub_nestedtable"),
get_spi_service().spi_sub_nestedtable_,
args,
ret_err));
OZ (check_success(ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
}
}
}
}
return ret;
}
int ObPLCodeGenerator::generate_sql(const ObPLSqlStmt &s, ObLLVMValue &ret_err)
{
int ret = OB_SUCCESS;
if (OB_ISNULL(get_current().get_v())) {
// 控制流已断,后面的语句不再处理
} else {
OZ (get_helper().set_insert_point(get_current()));
OZ (set_debug_location(s));
}
if (OB_FAIL(ret)) {
} else if (stmt::T_END_TRANS == s.get_stmt_type()) {
bool is_rollback = false;
ObSEArray<ObLLVMValue, 2> args;
ObLLVMValue is_rollback_value, sql, length;
// only allow stmt here: COMMIT; COMMIT COMMENT 'X'; ROLLBACK;
// COMMIT WORK; ROLLBACK WORK;
CK (((s.get_sql().length() >= 5 && 0 == STRNCASECMP(s.get_sql().ptr(), "commit", 5))
|| (s.get_sql().length() >= 8 && 0 == STRNCASECMP(s.get_sql().ptr(), "rollback", 8))));
OX (is_rollback
= (s.get_sql().length() >= 8 && 0 == STRNCASECMP(s.get_sql().ptr(), "rollback", 8)));
OZ (generate_string(s.get_sql(), sql, length));
OZ (args.push_back(get_vars().at(CTX_IDX)));
OZ (args.push_back(sql));
OZ (get_helper().get_int8(is_rollback, is_rollback_value));
OZ (args.push_back(is_rollback_value));
OZ (get_helper().create_call(ObString("spi_end_trans"), get_spi_service().spi_end_trans_, args, ret_err));
LOG_DEBUG("explicit end trans in pl", K(ret), K(s.get_sql()));
} else {
ObSEArray<ObLLVMValue, 16> args;
ObLLVMValue str;
ObLLVMValue len;
ObLLVMValue ps_sql;
ObLLVMValue type;
ObLLVMValue for_update;
ObLLVMValue hidden_rowid;
ObLLVMValue params;
ObLLVMValue count;
OZ (args.push_back(get_vars().at(CTX_IDX)));
OZ (generate_sql(s, str, len, ps_sql, type, for_update, hidden_rowid, params, count));
if (OB_SUCC(ret)) {
if (s.get_params().empty()) {
OZ (args.push_back(str));
OZ (args.push_back(type));
} else {
OZ (args.push_back(ps_sql));
OZ (args.push_back(type));
OZ (args.push_back(params));
OZ (args.push_back(count));
}
}
if (OB_SUCC(ret)) {
ObLLVMValue into_array_value;
ObLLVMValue into_count_value;
ObLLVMValue type_array_value;
ObLLVMValue type_count_value;
ObLLVMValue exprs_not_null_array_value;
ObLLVMValue pl_integer_range_array_value;
ObLLVMValue is_bulk;
OZ (generate_into(s, into_array_value, into_count_value,
type_array_value, type_count_value,
exprs_not_null_array_value,
pl_integer_range_array_value,
is_bulk));
OZ (args.push_back(into_array_value));
OZ (args.push_back(into_count_value));
OZ (args.push_back(type_array_value));
OZ (args.push_back(type_count_value));
OZ (args.push_back(exprs_not_null_array_value));
OZ (args.push_back(pl_integer_range_array_value));
OZ (args.push_back(is_bulk));
}
if (OB_SUCC(ret)) {
if (s.get_params().empty()) {
OZ (get_helper().create_call(ObString("spi_query"), get_spi_service().spi_query_, args, ret_err));
} else { //有外部变量,走prepare/execute接口
ObLLVMValue is_forall;
OZ (get_helper().get_int8(static_cast<int64_t>(s.is_forall_sql()), is_forall));
OZ (args.push_back(is_forall));
OZ (get_helper().create_call(ObString("spi_execute"), get_spi_service().spi_execute_, args, ret_err));
}
}
}
return ret;
}
int ObPLCodeGenerator::generate_after_sql(const ObPLSqlStmt &s, ObLLVMValue &ret_err)
{
int ret = OB_SUCCESS;
if (OB_ISNULL(get_current().get_v())) {
// 控制流已断, 后面的语句不再处理
} else {
OZ (check_success(ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
OZ (generate_into_restore(s.get_into(), s.get_exprs(), s.get_symbol_table()));
if (OB_SUCC(ret) && lib::is_mysql_mode()) {
ObLLVMValue is_not_found;
ObLLVMBasicBlock normal_end;
ObLLVMBasicBlock reset_ret;
ObLLVMValue ret_success;
OZ (get_helper().create_block(ObString("sql_end"), get_func(), normal_end));
OZ (get_helper().create_block(ObString("sql_check_success"), get_func(), reset_ret));
OZ (get_helper().create_icmp_eq(ret_err, OB_READ_NOTHING, is_not_found));
OZ (get_helper().create_cond_br(is_not_found, reset_ret, normal_end));
OZ (get_helper().set_insert_point(reset_ret));
OZ (get_helper().get_int32(OB_SUCCESS, ret_success));
OZ (get_helper().create_store(ret_success, get_vars().at(RET_IDX)));
OZ (get_helper().create_br(normal_end));
OZ (set_current(normal_end));
}
}
return ret;
}
int ObPLCodeGenerator::get_llvm_type(const ObPLDataType &pl_type, ObLLVMType &ir_type)
{
int ret = OB_SUCCESS;
ir_type = NULL;
if (pl_type.is_obj_type()) {
if (OB_FAIL(ObPLDataType::get_llvm_type(pl_type.get_obj_type(), get_helper(), get_adt_service(), ir_type))) {
LOG_WARN("failed to get datum type", K(ret));
}
} else if (pl_type.is_user_type()) {
uint64_t user_type_id = pl_type.get_user_type_id();
if (OB_FAIL(get_user_type_map().get_refactored(user_type_id, ir_type))) {
LOG_WARN("get user type map failed", K(ret), K(user_type_id));
}
} else {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("pl type is invalid", K(pl_type.get_type()));
}
return ret;
}
int ObPLCodeGenerator::get_datum_type(const ObPLDataType &pl_type, ObLLVMType &ir_type)
{
int ret = OB_SUCCESS;
ir_type = NULL;
// if (pl_type.is_obj_type()) {
if (OB_FAIL(ObPLDataType::get_datum_type(pl_type.get_obj_type(), helper_, adt_service_, ir_type))) { //集合类型里的基础数据类型统一用int64存储
LOG_WARN("failed to get datum type", K(ret));
}
// } else if (pl_type.is_user_type()) {
// uint64_t user_type_id = pl_type.get_user_type_id();
// if (OB_FAIL(user_type_map_.get_refactored(user_type_id, ir_type))) {
// LOG_WARN("get user type map failed", K(ret), K(user_type_id));
// }
// } else {
// ret = OB_ERR_UNEXPECTED;
// LOG_WARN("pl type is invalid", K(pl_type.get_type()));
// }
return ret;
}
int ObPLCodeGenerator::generate_string(const ObString &string, ObLLVMValue &str, ObLLVMValue &len)
{
int ret = OB_SUCCESS;
if (string.empty()) {
if (OB_FAIL(generate_empty_string(str, len))) {
LOG_WARN("failed to generate_empty_string", K(ret));
}
} else {
ObLLVMValue llvm_string;
ObLLVMType llvm_string_type;
ObLLVMValue p_llvm_string;
ObLLVMType llvm_type;
OZ (helper_.get_llvm_type(ObCharType, llvm_type));
OZ (helper_.get_string(string, llvm_string));
OZ (llvm_string.get_type(llvm_string_type));
OZ (helper_.create_alloca(ObString("string"), llvm_string_type, p_llvm_string));
OZ (helper_.create_store(llvm_string, p_llvm_string));
OZ (helper_.create_pointer_cast(ObString("string_to_char"), p_llvm_string, llvm_type, str));
OZ (helper_.get_int64(string.length(), len));
}
return ret;
}
int ObPLCodeGenerator::generate_global_string(const ObString &string, ObLLVMValue &str, ObLLVMValue &len)
{
int ret = OB_SUCCESS;
if (string.empty()) {
OZ (generate_empty_string(str, len));
} else {
ObLLVMType llvm_type;
ObLLVMValue llvm_string;
OZ (helper_.get_llvm_type(ObCharType, llvm_type));
OZ (helper_.get_string(string, llvm_string));
OZ (helper_.get_global_string(llvm_string, llvm_string));
OZ (helper_.create_pointer_cast(ObString("string_to_char"), llvm_string, llvm_type, str));
OZ (helper_.get_int64(string.length(), len));
}
return ret;
}
int ObPLCodeGenerator::generate_empty_string(ObLLVMValue &str, ObLLVMValue &len)
{
int ret = OB_SUCCESS;
ObLLVMType llvm_type;
if (OB_FAIL(helper_.get_llvm_type(ObCharType, llvm_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(ObLLVMHelper::get_null_const(llvm_type, str))) {
LOG_WARN("failed to get_null_const", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObIntType, llvm_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(ObLLVMHelper::get_null_const(llvm_type, len))) {
LOG_WARN("failed to get_null_const", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::generate_null(ObObjType type, ObLLVMValue &value)
{
int ret = OB_SUCCESS;
ObLLVMType llvm_type;
if (OB_FAIL(helper_.get_llvm_type(type, llvm_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(ObLLVMHelper::get_null_const(llvm_type, value))) {
LOG_WARN("failed to get_null_const", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::generate_null_pointer(ObObjType type, ObLLVMValue &value)
{
int ret = OB_SUCCESS;
ObLLVMType llvm_type;
ObLLVMType llvm_pointer_type;
if (OB_FAIL(helper_.get_llvm_type(type, llvm_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(llvm_type.get_pointer_to(llvm_pointer_type))) {
LOG_WARN("failed to get_pointer_to", K(ret));
} else if (OB_FAIL(ObLLVMHelper::get_null_const(llvm_pointer_type, value))) {
LOG_WARN("failed to get_null_const", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::generate_int64_array(const ObIArray<int64_t> &array, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObArray<uint64_t> uint64_array;
for (int64_t i = 0; OB_SUCC(ret) && i < array.count(); ++i) {
uint64_t v = array.at(i);
OZ (uint64_array.push_back(v));
}
OZ (generate_uint64_array(uint64_array, result));
return ret;
}
int ObPLCodeGenerator::generate_uint64_array(const ObIArray<uint64_t> &array, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
if (array.empty()) {
OZ (generate_null_pointer(ObIntType, result));
} else {
ObLLVMValue llvm_array;
if (OB_FAIL(helper_.get_uint64_array(array, llvm_array))) {
LOG_WARN("failed to get_uint64_array", K(ret));
} else if (OB_FAIL(generate_array(llvm_array, result))) {
LOG_WARN("failed to generate_array", K(ret));
} else { /*do nothing*/ }
}
return ret;
}
int ObPLCodeGenerator::generate_array(const ObLLVMValue array, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObSEArray<int64_t, 2> indices;
ObLLVMType array_type;
ObLLVMValue p_array;
if (OB_FAIL(indices.push_back(0)) || OB_FAIL(indices.push_back(0))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(array.get_type(array_type))) {
LOG_WARN("failed to get_type", K(ret));
} else if (OB_FAIL(helper_.create_alloca(ObString("array"), array_type, p_array))) {
LOG_WARN("failed to create_alloca", K(ret));
} else if (OB_FAIL(helper_.create_store(array, p_array))) {
LOG_WARN("failed to create_store", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_first_addr_from_array"), p_array, indices, result))) {
LOG_WARN("failed to create_gep", K(ret));
} else { /*do nothing*/ }
return ret;
}
//just for debug
int ObPLCodeGenerator::generate_debug(const ObString &name, int64_t value)
{
int ret = OB_SUCCESS;
ObLLVMValue int_value;
OZ (helper_.get_int64(value, int_value));
OZ (generate_debug(name, int_value));
return ret;
}
int ObPLCodeGenerator::generate_debug(const ObString &name, ObLLVMValue &value)
{
int ret = OB_SUCCESS;
ObSEArray<ObLLVMValue, 4> args;
ObLLVMValue str;
ObLLVMValue len;
OZ (generate_string(name, str, len));
OZ (args.push_back(str));
OZ (args.push_back(len));
OZ (args.push_back(value));
if (12 == value.get_type_id()) { //12 is IntegerTyID
switch (value.get_type().get_width()) {
case 8: {
OZ (helper_.create_call(ObString("debug"), get_eh_service().eh_debug_int8_, args));
}
break;
case 32: {
OZ (helper_.create_call(ObString("debug"), get_eh_service().eh_debug_int32_, args));
}
break;
case 64: {
OZ (helper_.create_call(ObString("debug"), get_eh_service().eh_debug_int64_, args));
}
break;
default: {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("Unexpected integer to debug", K(value.get_type().get_width()), K(ret));
}
break;
}
} else if (14 == value.get_type_id() && 12 == value.get_type().get_child(0).get_id()) { //14 is PointerTyID
switch (value.get_type().get_child(0).get_width()) {
case 8: {
OZ (helper_.create_call(ObString("debug"), get_eh_service().eh_debug_int8ptr_, args));
}
break;
case 32: {
OZ (helper_.create_call(ObString("debug"), get_eh_service().eh_debug_int32ptr_, args));
}
break;
case 64: {
OZ (helper_.create_call(ObString("debug"), get_eh_service().eh_debug_int64ptr_, args));
}
break;
default: {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("Unexpected integer pointer to debug", K(value.get_type().get_width()), K(ret));
}
break;
}
} else if (14 == value.get_type_id() && 15 == value.get_type().get_child(0).get_id()) { //14 is PointerTyID, 15 is StructTyID
ObLLVMType obj_type;
ObLLVMType objparam_type;
OZ (adt_service_.get_obj(obj_type));
OZ (adt_service_.get_objparam(objparam_type));
if (OB_SUCC(ret)) {
bool is_same = false;
OZ (value.get_type().get_child(0).same_as(obj_type, is_same));
if (OB_SUCC(ret)) {
if (is_same) {
OZ (helper_.create_call(ObString("debug"), get_eh_service().eh_debug_obj_, args));
} else {
OZ (value.get_type().get_child(0).same_as(objparam_type, is_same));
if (OB_SUCC(ret)) {
if (is_same) {
OZ (helper_.create_call(ObString("debug"), get_eh_service().eh_debug_objparam_, args));
} else {
ObLLVMType int_type;
ObLLVMValue int_value;
OZ (helper_.get_llvm_type(ObIntType, int_type));
OZ (helper_.create_ptr_to_int(ObString("object_to_int64"),
value,
int_type,
int_value));
OX (args.at(2) = int_value);
OZ (helper_.create_call(ObString("debug"), get_eh_service().eh_debug_int64_, args));
}
}
}
}
}
} else {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("Unexpected value to debug", K(value.get_type_id())/*, K(value.get_type().get_child(0).get_id()), K(ret)*/);
}
return ret;
}
#define STORE_META(first, second, name, get_func) \
do { \
if (OB_SUCC(ret)) { \
indices.reset(); \
if (OB_FAIL(indices.push_back(0)) || OB_FAIL(indices.push_back(first)) || OB_FAIL(indices.push_back(second))) { \
LOG_WARN("push_back error", K(ret)); \
} else if (OB_FAIL(helper_.create_gep(ObString(name), p_obj, indices, dest))) { \
LOG_WARN("failed to create gep", K(ret)); \
} else if (OB_FAIL(helper_.get_int8(meta.get_func(), src))) { \
LOG_WARN("failed to get_int64", K(meta.get_type()), K(meta), K(ret)); \
} else if (OB_FAIL(helper_.create_store(src, dest))) { \
LOG_WARN("failed to create store", K(ret)); \
} else { /*do nothing*/ } \
} \
} while (0)
#define STORE_ELEMENT(idx, name, get_func, length) \
do { \
if (OB_SUCC(ret)) { \
if (OB_FAIL(helper_.create_gep(ObString(name), p_obj, idx, dest))) { \
LOG_WARN("failed to create gep", K(ret)); \
} else if (OB_FAIL(helper_.get_int##length(object.get_func(), src))) { \
LOG_WARN("failed to get_int64", K(object), K(ret)); \
} else if (OB_FAIL(helper_.create_store(src, dest))) { \
LOG_WARN("failed to create store", K(ret)); \
} else { /*do nothing*/ } \
} \
} while (0)
int ObPLCodeGenerator::generate_obj(const ObObj &object, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMType obj_type;
if (OB_FAIL(adt_service_.get_obj(obj_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.create_alloca(ObString("ObObj"), obj_type, result))) {
LOG_WARN("failed to create_alloca", K(ret));
} else if (OB_FAIL(store_obj(object, result))) {
LOG_WARN("failed to store obj", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::store_obj(const ObObj &object, ObLLVMValue &p_obj)
{
int ret = OB_SUCCESS;
const ObObjMeta &meta = object.get_meta();
ObSEArray<int64_t, 3> indices;
ObLLVMValue src;
ObLLVMValue dest;
//obj type
STORE_META(0, 0, "extract_obj_type", get_type);
//collation level
STORE_META(0, 1, "extract_obj_collation_level", get_collation_level);
//collation type
STORE_META(0, 2, "extract_obj_collation_type", get_collation_type);
//scale
STORE_META(0, 3, "extract_obj_scale", get_scale);
//length
STORE_ELEMENT(1, "extract_obj_length", get_val_len, 32);
//value
STORE_ELEMENT(2, "extract_obj_value", get_int, 64);
return ret;
}
int ObPLCodeGenerator::store_objparam(const ObObjParam &object, ObLLVMValue &p_objparam)
{
int ret = OB_SUCCESS;
ObLLVMValue p_obj;
if (OB_FAIL(helper_.create_gep(ObString("extract_obj_pointer"), p_objparam, 0, p_obj))) {
LOG_WARN("failed to create gep", K(ret));
} else if (OB_FAIL(store_obj(object, p_obj))) {
LOG_WARN("failed to store obj", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::store_data_type(const ObDataType &object, jit::ObLLVMValue &p_obj)
{
int ret = OB_SUCCESS;
const ObObjMeta &meta = object.get_meta_type();
ObSEArray<int64_t, 2> indices;
ObLLVMValue src;
ObLLVMValue dest;
//obj type
STORE_META(0, 0, "extract_obj_type", get_type);
//collation level
STORE_META(0, 1, "extract_obj_collation_level", get_collation_level);
//collation type
STORE_META(0, 2, "extract_obj_collation_type", get_collation_type);
//scale
STORE_META(0, 3, "extract_obj_scale", get_scale);
//accuracy
STORE_ELEMENT(1, "extract_accuracy", get_accuracy_value, 64);
//charset
STORE_ELEMENT(2, "extract_charset", get_charset_type, 32);
//is_binary_collation
STORE_ELEMENT(3, "extract_binary_collation", is_binary_collation, 8);
//is_zero_fill
STORE_ELEMENT(4, "extract_zero_fill", is_zero_fill, 8);
return ret;
}
int ObPLCodeGenerator::store_elem_desc(const ObElemDesc &object, jit::ObLLVMValue &p_obj)
{
int ret = OB_SUCCESS;
ObLLVMValue src;
ObLLVMValue dest;
OZ (store_data_type(object, p_obj));
//pl type
STORE_ELEMENT(5, "extract_pl_type", get_pl_type, 32);
//not_null
STORE_ELEMENT(6, "extract_not_null", is_not_null, 8);
//field_count
STORE_ELEMENT(7, "extract_field_count", get_field_count, 32);
return ret;
}
#define GENERATE_OBJECT(name, class) \
int ObPLCodeGenerator::generate_##name(const class &object, jit::ObLLVMValue &result) \
{ \
int ret = OB_SUCCESS; \
ObLLVMType object_type; \
if (OB_FAIL(adt_service_.get_##name(object_type))) { \
LOG_WARN("failed to get_llvm_type", K(ret)); \
} else if (OB_FAIL(helper_.create_alloca(ObString("NewObject"), object_type, result))) { \
LOG_WARN("failed to create_alloca", K(ret)); \
} else if (OB_FAIL(store_##name(object, result))) { \
LOG_WARN("failed to store", K(ret)); \
} else { /*do nothing*/ } \
return ret; \
}
GENERATE_OBJECT(data_type, ObDataType)
GENERATE_OBJECT(elem_desc, ObElemDesc)
#define INIT_OBJPARAM_ELEMENT(name, length, value) \
do { \
if (OB_SUCC(ret)) { \
if (OB_FAIL(extract_##name##_ptr_from_objparam(result, dest))) { \
LOG_WARN("failed to extract ptr from objparam", K(ret)); \
} else if (OB_FAIL(helper_.get_int##length(value, src))) { \
LOG_WARN("failed to get_int8", K(ret)); \
} else if (OB_FAIL(helper_.create_store(src, dest))) { \
LOG_WARN("failed to create_store", K(ret)); \
} else { /*do nothing*/ } \
} \
} while (0)
int ObPLCodeGenerator::generate_new_objparam(ObLLVMValue &result, int64_t udt_id)
{
int ret = OB_SUCCESS;
ObLLVMType objparam_type;
ObLLVMValue const_value;
if (OB_FAIL(adt_service_.get_objparam(objparam_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.create_alloca(ObString("ObObjParam"), objparam_type, result))) {
LOG_WARN("failed to create_alloca", K(ret));
} else if (OB_FAIL(ObLLVMHelper::get_null_const(objparam_type, const_value))) {
LOG_WARN("failed to get_null_const", K(ret));
} else if (OB_FAIL(helper_.create_store(const_value, result))) {
LOG_WARN("failed to create_store", K(ret));
} else {
ObLLVMValue src;
ObLLVMValue dest;
//init cs level
INIT_OBJPARAM_ELEMENT(cslevel, 8, ObCollationLevel::CS_LEVEL_INVALID);
//init scale
INIT_OBJPARAM_ELEMENT(scale, 8, -1);
//init accuracy
INIT_OBJPARAM_ELEMENT(accuracy, 64, udt_id);
//init flag
INIT_OBJPARAM_ELEMENT(flag, 8, 1);
//init raw_text_pos
INIT_OBJPARAM_ELEMENT(raw_text_pos, 32, -1);
//init raw_text_len
INIT_OBJPARAM_ELEMENT(raw_text_len, 32, -1);
//init param_meta
INIT_OBJPARAM_ELEMENT(type, 8, 0);
INIT_OBJPARAM_ELEMENT(cslevel, 8, 7);
INIT_OBJPARAM_ELEMENT(cstype, 8, 0);
INIT_OBJPARAM_ELEMENT(scale, 8, -1);
}
return ret;
}
int ObPLCodeGenerator::generate_set_extend(ObLLVMValue &p_obj,
ObPLType type,
int32_t size,
int64_t ptr)
{
int ret = OB_SUCCESS;
ObLLVMValue var_type;
ObLLVMValue init_value;
ObLLVMValue extend_value;
OZ (get_helper().get_int8(type, var_type));
OZ (get_helper().get_int32(size, init_value));
OZ (get_helper().get_int32(ptr, extend_value));
OZ (generate_set_extend(p_obj, var_type, init_value, extend_value));
return ret;
}
int ObPLCodeGenerator::generate_set_extend(ObLLVMValue &p_obj,
ObLLVMValue &type,
ObLLVMValue &size,
ObLLVMValue &ptr)
{
int ret = OB_SUCCESS;
ObSEArray<int64_t, 3> indices;
ObLLVMValue dest;
//init extend
ObObj extend_obj(ObExtendType);
extend_obj.set_int_value(0);
OZ (store_obj(extend_obj, p_obj));
//set extend type
OZ (indices.push_back(0));
OZ (indices.push_back(0));
OZ (indices.push_back(3));
OZ (helper_.create_gep("extend_type", p_obj, indices, dest));
OZ (helper_.create_store(type, dest));
//set extend size
OZ (helper_.create_gep("extend_type", p_obj, 1, dest));
OZ (helper_.create_store(size, dest));
//set ptr
OZ (helper_.create_gep("extend_type", p_obj, 2, dest));
OZ (helper_.create_store(ptr, dest));
return ret;
}
int ObPLCodeGenerator::generate_spi_calc(int64_t expr_idx,
int64_t stmt_id,
bool in_notfound,
bool in_warning,
int64_t result_idx,
ObLLVMValue &p_result_obj)
{
int ret = OB_SUCCESS;
const ObSqlExpression *expr = get_expr(expr_idx);
if (OB_ISNULL(expr)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("expr is NULL", K(ret));
} else {
ObLLVMValue expr_addr;
if (OB_FAIL(generate_pointer(expr, expr_addr))) {
LOG_WARN("failed to generate a pointer", K(ret));
} else {
ObSEArray<ObLLVMValue, 4> args;
ObLLVMValue result;
ObLLVMValue int_value;
ObLLVMValue package_id;
int64_t udt_id = ast_.get_expr(expr_idx)->get_result_type().get_udt_id();
if (OB_FAIL(args.push_back(vars_.at(CTX_IDX)))) { //PL的执行环境
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(args.push_back(expr_addr))) { //表达式的绝对地址
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(helper_.get_int64(result_idx, int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(args.push_back(int_value))) { //结果在ObArray里的位置
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generate_new_objparam(p_result_obj, udt_id))) {
LOG_WARN("failed to generate_new_objparam", K(ret));
} else if (OB_FAIL(args.push_back(p_result_obj))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(helper_.create_call(ObString("calc_expr"), get_spi_service().spi_calc_expr_, args, result))) {
LOG_WARN("failed to create call", K(ret));
} else if (OB_FAIL(check_success(result, stmt_id, in_notfound, in_warning))) {
LOG_WARN("failed to check success", K(ret));
} else { /*do nothing*/ }
}
}
return ret;
}
int ObPLCodeGenerator::generate_spi_package_calc(uint64_t package_id,
int64_t expr_idx,
const ObPLStmt &s,
ObLLVMValue &p_result_obj)
{
int ret = OB_SUCCESS;
ObSEArray<ObLLVMValue, 4> args;
ObLLVMValue result;
ObLLVMValue v_package_id;
ObLLVMValue v_expr_idx;
CK (OB_NOT_NULL(s.get_block()));
OZ (args.push_back(vars_.at(CTX_IDX)));
OZ (helper_.get_int64(package_id, v_package_id));
OZ (helper_.get_int64(expr_idx, v_expr_idx));
OZ (args.push_back(v_package_id));
OZ (args.push_back(v_expr_idx));
OZ (generate_new_objparam(p_result_obj));
OZ (args.push_back(p_result_obj));
OZ (helper_.create_call(ObString("calc_package_expr"), get_spi_service().spi_calc_package_expr_, args, result));
OZ (check_success(result, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
return ret;
}
int ObPLCodeGenerator::generate_llvm_calc(int64_t expr_idx,
int64_t stmt_id,
bool in_notfound,
bool in_warning,
int64_t result_idx,
ObLLVMValue &p_result_obj)
{
/*
* 为防止计算溢出,用int64进行计算
*/
#define GET_LLVM_VALUE(expr, value) \
do { \
if (OB_SUCC(ret)) { \
int64_t int_value = static_cast<const ObConstRawExpr*>(expr)->get_value().get_unknown(); \
if (T_QUESTIONMARK == expr->get_expr_type()) { \
if (OB_FAIL(extract_value_from_context(vars_.at(CTX_IDX), int_value, ObIntType, value))) { \
LOG_WARN("failed to extract_value_from_context", K(ret)); \
} \
} else if (expr->is_const_raw_expr()) { \
if (OB_FAIL(helper_.get_int64(int_value, value))) { \
LOG_WARN("failed to get int64", K(ret)); \
} \
} else { \
ret = OB_INVALID_ARGUMENT; \
LOG_WARN("invalid expr type", K(*expr), K(ret)); \
} \
} \
} while (0)
int ret = OB_SUCCESS;
const ObRawExpr *expr = ast_.get_expr(expr_idx);
int64_t udt_id = expr->get_result_type().get_udt_id();
if (OB_ISNULL(expr)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("expr is NULL", K(ret));
} else if (OB_FAIL(generate_new_objparam(p_result_obj, udt_id))) {
LOG_WARN("failed to generate_new_objparam", K(ret));
} else if (expr->is_const_raw_expr()) {
const ObObj &const_value = static_cast<const ObConstRawExpr*>(expr)->get_value();
if (!const_value.is_int32()) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("Invalid simple integer", K(const_value), K(ret));
} else {
if (OB_FAIL(generate_const_calc(const_value.get_int32(), p_result_obj))) {
LOG_WARN("failed to generate const calc", K(const_value), K(ret));
}
#ifndef NDEBUG
OZ (generate_debug(ObString("generate_llvm_calc result"), p_result_obj));
#endif
}
} else {
if (OB_ISNULL(expr->get_param_expr(0)) || OB_ISNULL(expr->get_param_expr(1))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("child expr is NULL", K(expr->get_param_expr(0)), K(expr->get_param_expr(1)), K(ret));
} else if (!expr->get_param_expr(0)->get_result_type().is_int32() || !expr->get_param_expr(1)->get_result_type().is_int32()) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("Unexpected integer type", K(*expr->get_param_expr(0)), K(*expr->get_param_expr(1)), K(ret));
} else {
const ObRawExpr *left_expr = expr->get_param_expr(0);
const ObRawExpr *right_expr = expr->get_param_expr(1);
ObLLVMValue left;
ObLLVMValue right;
GET_LLVM_VALUE(left_expr, left);
GET_LLVM_VALUE(right_expr, right);
if (OB_SUCC(ret)) {
if (IS_COMMON_COMPARISON_OP(expr->get_expr_type())) {
if (OB_FAIL(generate_compare_calc(left, right, expr->get_expr_type(), p_result_obj))) {
LOG_WARN("failed to generate_new_objparam", K(ret));
}
} else {
if (OB_FAIL(generate_arith_calc(left, right, expr->get_expr_type(), expr->get_result_type(), stmt_id, in_notfound, in_warning, p_result_obj))) {
LOG_WARN("failed to generate_new_objparam", K(ret));
}
}
}
}
}
//存储进全局符号表和paramstore
if (OB_SUCC(ret) && OB_INVALID_INDEX != result_idx) {
ObLLVMValue p_objparam;
ObLLVMValue obj;
ObLLVMValue objparam;
if (OB_FAIL(extract_datum_from_objparam(p_result_obj, ObInt32Type, obj))) {
LOG_WARN("failed to extract_datum_from_objparam", K(ret));
} else if (OB_FAIL(helper_.create_store(obj, vars_.at(result_idx + USER_ARG_OFFSET)))) {
LOG_WARN("failed to create_store", K(ret));
} else if (OB_FAIL(helper_.create_load("load_out_arg", p_result_obj, objparam))) {
LOG_WARN("failed to create_load", K(ret));
} else if (OB_FAIL(extract_objparam_from_context(vars_.at(CTX_IDX), result_idx, p_objparam))) {
LOG_WARN("failed to extract_datum_from_objparam", K(ret));
} else if (OB_FAIL(helper_.create_store(objparam, p_objparam))) {
LOG_WARN("failed to create_store", K(ret));
} else { /*do nothing*/ }
}
return ret;
}
int ObPLCodeGenerator::generate_const_calc(int32_t value, ObLLVMValue &p_result_obj)
{
int ret = OB_SUCCESS;
ObLLVMValue p_value;
ObLLVMValue calc_result;
ObObjParam objparam;
objparam.set_type(ObInt32Type);
objparam.set_param_meta();
if (OB_FAIL(store_objparam(objparam, p_result_obj))) {
LOG_WARN("Not supported yet", K(ret));
} else if (OB_FAIL(extract_value_ptr_from_objparam(p_result_obj, ObInt32Type, p_value))) {
LOG_WARN("failed to extract_value_ptr_from_objparam", K(ret));
} else if (OB_FAIL(helper_.get_int32(value, calc_result))) {
LOG_WARN("failed to extract_value_ptr_from_objparam", K(ret));
} else if (OB_FAIL(helper_.create_store(calc_result, p_value))) {
LOG_WARN("failed to create store", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::generate_compare_calc(ObLLVMValue &left,
ObLLVMValue &right,
ObItemType type,
ObLLVMValue &p_result_obj)
{
int ret = OB_SUCCESS;
ObLLVMHelper::CMPTYPE compare_type = ObLLVMHelper::ICMP_EQ;
switch (type) {
case T_OP_EQ: {
compare_type = ObLLVMHelper::ICMP_EQ;
}
break;
case T_OP_NSEQ: {
ret = OB_NOT_SUPPORTED;
LOG_WARN("T_OP_NSEQ is not supported", K(type), K(ret));
LOG_USER_ERROR(OB_NOT_SUPPORTED, "null safe equals operator");
}
break;
case T_OP_LE: {
compare_type = ObLLVMHelper::ICMP_SLE;
}
break;
case T_OP_LT: {
compare_type = ObLLVMHelper::ICMP_SLT;
}
break;
case T_OP_GE: {
compare_type = ObLLVMHelper::ICMP_SGE;
}
break;
case T_OP_GT: {
compare_type = ObLLVMHelper::ICMP_SGT;
}
break;
case T_OP_NE: {
compare_type = ObLLVMHelper::ICMP_NE;
}
break;
default: {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("Unexpected expr type", K(type), K(ret));
}
break;
}
if (OB_SUCC(ret)) {
ObLLVMValue is_true;
ObLLVMBasicBlock true_branch;
ObLLVMBasicBlock false_branch;
ObLLVMBasicBlock end_branch;
#ifndef NDEBUG
OZ (generate_debug(ObString("debug"), left));
OZ (generate_debug(ObString("debug"), right));
#endif
if (OB_FAIL(helper_.create_block(ObString("true_branch"), get_func(), true_branch))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(helper_.create_block(ObString("false_branch"), get_func(), false_branch))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(helper_.create_block(ObString("end_branch"), get_func(), end_branch))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(helper_.create_icmp(left, right, compare_type, is_true))) { //这里的is_true是int1(1 bit),需要转成int8(8 bit)
LOG_WARN("failed to create_icmp_eq", K(ret));
} else if (OB_FAIL(helper_.create_cond_br(is_true, true_branch, false_branch))) {
LOG_WARN("failed to create_cond_br", K(ret));
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
ObObjParam objparam;
objparam.set_tinyint(1);
objparam.set_param_meta();
if (OB_FAIL(helper_.set_insert_point(true_branch))) { \
LOG_WARN("failed to set insert point", K(ret)); \
} else if (OB_FAIL(store_objparam(objparam, p_result_obj))) {
LOG_WARN("failed to store objparam", K(ret));
} else if (OB_FAIL(helper_.create_br(end_branch))) {
LOG_WARN("failed to create_br", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
ObObjParam objparam;
objparam.set_tinyint(0);
objparam.set_param_meta();
if (OB_FAIL(helper_.set_insert_point(false_branch))) { \
LOG_WARN("failed to set insert point", K(ret)); \
} else if (OB_FAIL(store_objparam(objparam, p_result_obj))) {
LOG_WARN("failed to store objparam", K(ret));
} else if (OB_FAIL(helper_.create_br(end_branch))) {
LOG_WARN("failed to create_br", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret) && OB_FAIL(set_current(end_branch))) {
LOG_WARN("failed to set_current", K(ret));
}
}
return ret;
}
int ObPLCodeGenerator::generate_arith_calc(ObLLVMValue &left,
ObLLVMValue &right,
ObItemType type,
const ObExprResType &result_type,
int64_t stmt_id,
bool in_notfound,
bool in_warning,
ObLLVMValue &p_result_obj)
{
int ret = OB_SUCCESS;
ObLLVMValue p_value;
ObLLVMValue calc_result;
if (T_OP_ADD == type) {
if (OB_FAIL(helper_.create_add(left, right, calc_result))) {
LOG_WARN("failed to create_add", K(ret));
}
} else if (T_OP_MINUS == type) {
if (OB_FAIL(helper_.create_sub(left, right, calc_result))) {
LOG_WARN("failed to create_add", K(ret));
}
} else {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("Not supported yet", K(type), K(ret));
}
if (OB_SUCC(ret)) { //检查值域
ObLLVMBasicBlock succ_block;
ObLLVMBasicBlock error_block;
ObLLVMBasicBlock check_block;
ObLLVMBasicBlock end_block;
ObLLVMValue low;
ObLLVMValue high;
ObLLVMValue is_true;
ObLLVMValue ret_value;
ObLLVMValue ret_value_ptr;
OZ (helper_.create_ialloca(ObString("ret_value"), ObInt32Type, OB_SUCCESS, ret_value_ptr));
OZ (helper_.create_block(ObString("succ_block"), func_, succ_block));
OZ (helper_.create_block(ObString("error_block"), func_, error_block));
OZ (helper_.create_block(ObString("check_block"), func_, check_block));
OZ (helper_.create_block(ObString("end_block"), func_, end_block));
OZ (helper_.get_int64(INT32_MIN, low));
OZ (helper_.create_icmp(calc_result, low, ObLLVMHelper::ICMP_SGE, is_true));
OZ (helper_.create_cond_br(is_true, check_block, error_block));
OZ (helper_.set_insert_point(check_block));
OZ (helper_.get_int64(INT32_MAX, high));
OZ (helper_.create_icmp(high, calc_result, ObLLVMHelper::ICMP_SGE, is_true));
OZ (helper_.create_cond_br(is_true, succ_block, error_block));
OZ (helper_.set_insert_point(error_block));
OZ (helper_.get_int32(OB_NUMERIC_OVERFLOW, ret_value));
OZ (helper_.create_store(ret_value, ret_value_ptr));
OZ (helper_.create_br(end_block));
OZ (helper_.set_insert_point(succ_block));
OZ (helper_.get_int32(OB_SUCCESS, ret_value));
OZ (helper_.create_store(ret_value, ret_value_ptr));
OZ (helper_.create_br(end_block));
OZ (set_current(end_block));
OZ (helper_.create_load(ObString("load_ret"), ret_value_ptr, ret_value));
OZ (check_success(ret_value, stmt_id, in_notfound, in_warning));
}
if (OB_SUCC(ret)) {
ObObjParam objparam;
objparam.set_type(result_type.get_type()); //计算之后的结果放在int32里
objparam.set_param_meta();
if (OB_FAIL(store_objparam(objparam, p_result_obj))) {
LOG_WARN("Not supported yet", K(type), K(ret));
} else if (OB_FAIL(extract_value_ptr_from_objparam(p_result_obj, ObIntType, p_value))) {
LOG_WARN("failed to extract_value_ptr_from_objparam", K(ret));
} else if (OB_FAIL(helper_.create_store(calc_result, p_value))) {
LOG_WARN("failed to create store", K(ret));
} else { /*do nothing*/ }
}
return ret;
}
int ObPLCodeGenerator::generate_expr(int64_t expr_idx,
const ObPLStmt &s,
int64_t result_idx,
ObLLVMValue &p_result_obj)
{
int ret = OB_SUCCESS;
ObLLVMValue expr_addr;
if (OB_INVALID_INDEX == expr_idx || OB_ISNULL(s.get_block())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("Unexpected argument", K(expr_idx), K(s.get_block()), K(ret));
} else if (lib::is_oracle_mode() && ast_.is_simple_calc(expr_idx)) {
if (OB_FAIL(generate_llvm_calc(expr_idx,
s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning(),
result_idx, p_result_obj))) {
LOG_WARN("failed to create_store", K(ret));
}
} else {
if (OB_FAIL(generate_spi_calc(expr_idx,
s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning(),
result_idx,
p_result_obj))) {
LOG_WARN("failed to create_store", K(ret));
}
}
return ret;
}
int ObPLCodeGenerator::generate_early_exit(ObLLVMValue &count,
int64_t stmt_id,
bool in_notfound,
bool in_warning)
{
int ret = OB_SUCCESS;
ObLLVMBasicBlock early_exit;
ObLLVMBasicBlock after_check;
ObLLVMValue is_true, need_check, first_check;
ObSEArray<ObLLVMValue, 4> args;
ObLLVMValue result;
ObLLVMValue count_value;
OZ (helper_.create_block(ObString("early_exit"), get_func(), early_exit));
OZ (helper_.create_block(ObString("after_check"), get_func(), after_check));
OZ (helper_.create_load(ObString("load_count"), count, count_value));
OZ (helper_.create_inc(count_value, count_value));
OZ (helper_.create_store(count_value, count));
#ifndef NDEBUG
OZ (generate_debug(ObString("print count value"), count_value));
#endif
/*!
* need_check和first_check不等说明需要check, 相等则不需要check
* 因为need_check和firstcheck不可能同时为true; 当同时为false时说明两个条件都没满足; 只有一个true时则说明满足一个条件;
* 引入first_check是为了避免类似下面的匿名块形成的死循环无法退出;
* BEGIN
* <<outer_loop>>
* FOR idx IN 1..10 LOOP
* CONTINUE outer_loop;
* END LOOP;
* END;
*/
OZ (helper_.create_icmp(count_value, EARLY_EXIT_CHECK_CNT, ObLLVMHelper::ICMP_SGE, need_check)); // 到达check次数
OZ (helper_.create_icmp(count_value, 1, ObLLVMHelper::ICMP_EQ, first_check)); // 该值为1说明是循环的第一次check
OZ (helper_.create_icmp(need_check, first_check, ObLLVMHelper::ICMP_NE, is_true));
OZ (helper_.create_cond_br(is_true, early_exit, after_check));
//处理check early exit
OZ (set_current(early_exit));
OZ (helper_.create_istore(1, count));
OZ (args.push_back(get_vars().at(CTX_IDX)));
OZ (helper_.create_call(ObString("check_early_exit"), get_spi_service().spi_check_early_exit_, args, result));
OZ (check_success(result, stmt_id, in_notfound, in_warning));
OZ (helper_.create_br(after_check));
//处理正常流程
OZ (set_current(after_check));
return ret;
}
int ObPLCodeGenerator::generate_pointer(const void *ptr, ObLLVMValue &value)
{
int ret = OB_SUCCESS;
int64_t addr = reinterpret_cast<int64_t>(ptr);
if (OB_FAIL(helper_.get_int64(addr, value))) {
LOG_WARN("failed to get int64", K(ret));
}
return ret;
}
int ObPLCodeGenerator::generate_expression_array(const ObIArray<int64_t> &exprs,
jit::ObLLVMValue &value,
jit::ObLLVMValue &count)
{
int ret = OB_SUCCESS;
ObSEArray<uint64_t, 8> expr_addrs;
for (int64_t i = 0; OB_SUCC(ret) && i < exprs.count(); ++i) {
if (OB_FAIL(expr_addrs.push_back(reinterpret_cast<uint64_t>(get_expr(exprs.at(i)))))) {
LOG_WARN("store expr addr failed", K(ret));
}
}
if (OB_SUCC(ret)) {
ObLLVMValue first_addr;
if (expr_addrs.empty()) {
if (OB_FAIL(generate_null_pointer(ObIntType, value))) {
LOG_WARN("failed to generate_null_pointer", K(ret));
}
} else {
if (OB_FAIL(generate_uint64_array(expr_addrs, value))) {
LOG_WARN("failed to get_uint64_array", K(ret));
}
}
if (OB_SUCC(ret)) {
ObLLVMValue int_value;
if (OB_FAIL(helper_.get_int64(exprs.count(), count))) {
LOG_WARN("failed to get int64", K(ret));
}
}
}
return ret;
}
#define CHECK_COND_CONTROL \
do { \
if (OB_SUCC(ret)) { \
ObLLVMBasicBlock do_control; \
ObLLVMBasicBlock after_control; \
if (OB_INVALID_INDEX != control.get_cond()) { \
ObLLVMValue p_result_obj; \
ObLLVMValue result; \
ObLLVMValue is_false; \
if (OB_FAIL(helper_.create_block(ObString("do_control"), get_func(), do_control))) { \
LOG_WARN("failed to create block", K(ret)); \
} else if (OB_FAIL(helper_.create_block(ObString("after_control"), \
get_func(), after_control))) { \
LOG_WARN("failed to create block", K(ret)); \
} else if (OB_FAIL(generate_expr(control.get_cond(), control, \
OB_INVALID_INDEX, p_result_obj))) { \
LOG_WARN("failed to generate calc_expr func", K(ret)); \
} else if (OB_FAIL(extract_value_from_objparam(p_result_obj, \
control.get_cond_expr()->get_data_type(), result))) { \
LOG_WARN("failed to extract_value_from_objparam", K(ret)); \
} else if (OB_FAIL(helper_.create_icmp_eq(result, FALSE, is_false))) { \
LOG_WARN("failed to create_icmp_eq", K(ret)); \
} else if (OB_FAIL(helper_.create_cond_br(is_false, after_control, do_control))) { \
LOG_WARN("failed to create_cond_br", K(ret)); \
} else if (OB_FAIL(set_current(after_control))) { \
LOG_WARN("failed to set insert point", K(ret)); \
} else if (OB_FAIL(helper_.set_insert_point(do_control))) { \
LOG_WARN("failed to set insert point", K(ret)); \
} else { } \
} else { \
if (OB_FAIL(helper_.create_block(ObString("do_control"), get_func(), do_control))) { \
LOG_WARN("failed to create block", K(ret)); \
} else if (OB_FAIL(helper_.create_block(ObString("after_control"), \
get_func(), after_control))) { \
LOG_WARN("failed to create block", K(ret)); \
} else if (OB_FAIL(helper_.create_br(do_control))) { \
LOG_WARN("failed to create_cond_br", K(ret)); \
} else if (OB_FAIL(set_current(after_control))) { \
LOG_WARN("failed to set insert point", K(ret)); \
} else if (OB_FAIL(helper_.set_insert_point(do_control))) { \
LOG_WARN("failed to set insert point", K(ret)); \
} else { } \
} \
} \
} while(0)
//跳转目的地和跳转语句之间的所有循环的栈都需要重置
#define RESTORE_LOOP_STACK \
do { \
if (OB_SUCC(ret)) { \
const ObPLCodeGenerator::LabelStack::LabelInfo *label_info = get_label(control.get_next_label()); \
if (OB_ISNULL(label_info)) { \
ret = OB_ERR_LABEL_ILLEGAL; \
LOG_WARN("label info is NULL", K(control.get_next_label()), K(ret)); \
LOG_USER_ERROR(OB_ERR_LABEL_ILLEGAL, control.get_next_label().length(), \
control.get_next_label().ptr()); \
} else { \
LOG_DEBUG("label info is not null", K(control.get_next_label()), K(label_info->name_), K(label_info->level_)); \
for (int64_t i = get_loop_count() - 1; OB_SUCC(ret) && i >= 0; --i) { \
ObPLCodeGenerator::LoopStack::LoopInfo &loop_info = get_loops()[i]; \
LOG_DEBUG("loop info", K(i), K(loop_info.level_), K(control.get_level())); \
if (loop_info.level_ <= control.get_level() && loop_info.level_ >= label_info->level_) { \
if (OB_FAIL(helper_.stack_restore(loop_info.loop_))) { \
LOG_WARN("failed to stack_restore", K(ret)); \
} else { \
LOG_DEBUG("success stack restore", K(i), K(loop_info.level_)); \
} \
} \
} \
} \
} \
} while(0)
int ObPLCodeGenerator::generate_loop_control(const ObPLLoopControl &control)
{
int ret = OB_SUCCESS;
if (NULL == get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else if (OB_FAIL(helper_.set_insert_point(get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(set_debug_location(control))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(generate_goto_label(control))) {
LOG_WARN("failed to generate goto label", K(ret));
} else if (!control.get_next_label().empty()) {
CHECK_COND_CONTROL;
RESTORE_LOOP_STACK;
if (OB_SUCC(ret)) {
ObLLVMBasicBlock next = PL_LEAVE == control.get_type() ? get_label(control.get_next_label())->exit_ : get_label(control.get_next_label())->start_;
if (OB_ISNULL(next.get_v())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("a loop must have valid body", K(next), K(ret));
} else if (OB_FAIL(helper_.create_br(next))) {
LOG_WARN("failed to create br", K(ret));
} else {
if (OB_FAIL(set_current(get_current()))) { //设置CURRENT, 调整INSERT POINT点
LOG_WARN("failed to set current", K(ret));
}
}
}
} else {
CHECK_COND_CONTROL;
if (OB_SUCC(ret)) {
ObPLCodeGenerator::LoopStack::LoopInfo *loop_info = get_current_loop();
if (OB_ISNULL(loop_info)) {
ret = OB_ERR_EXIT_CONTINUE_ILLEGAL;
LOG_WARN("illegal EXIT/CONTINUE statement; it must appear inside a loop", K(ret));
} else if (OB_FAIL(helper_.stack_restore(loop_info->loop_))) {
LOG_WARN("failed to stack_restore", K(ret));
} else {
ObLLVMBasicBlock next = PL_LEAVE == control.get_type() ? loop_info->exit_ : loop_info->start_;
if (OB_ISNULL(next.get_v())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("a loop must have valid body", K(ret));
} else if (OB_FAIL(helper_.create_br(next))) {
LOG_WARN("failed to create br", K(ret));
} else {
if (OB_FAIL(set_current(get_current()))) { //设置CURRENT, 调整INSERT POINT点
LOG_WARN("failed to set current", K(ret));
}
}
}
}
}
return ret;
}
int ObPLCodeGenerator::generate_sql(const ObPLSql &sql,
jit::ObLLVMValue &str,
jit::ObLLVMValue &length,
jit::ObLLVMValue &ps_sql,
jit::ObLLVMValue &type,
jit::ObLLVMValue &for_update,
jit::ObLLVMValue &hidden_rowid,
jit::ObLLVMValue &params,
jit::ObLLVMValue &count)
{
int ret = OB_SUCCESS;
ObLLVMValue int_value;
if (sql.get_params().empty()) {
if (OB_FAIL(generate_global_string(sql.get_sql(), str, length))) {
LOG_WARN("failed to get_string", K(ret));
}
} else {
if (OB_FAIL(generate_null(ObCharType, str))) {
LOG_WARN("failed to get_null_const", K(ret));
}
}
if (OB_SUCC(ret)) {
if (OB_FAIL(generate_global_string(sql.get_ps_sql(), ps_sql, length))) {
LOG_WARN("failed to get_string", K(ret));
} else if (OB_FAIL(helper_.get_int64(sql.get_stmt_type(), type))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(helper_.get_int8(sql.is_for_update(), for_update))) {
LOG_WARN("failed to get int8", K(ret));
} else if (OB_FAIL(helper_.get_int8(sql.has_hidden_rowid(), hidden_rowid))) {
LOG_WARN("failed to get int8", K(ret));
} else if (OB_FAIL(generate_expression_array(
sql.is_forall_sql() ? sql.get_array_binding_params() : sql.get_params(), params, count))) {
LOG_WARN("get precalc expr array ir value failed", K(ret));
} else { /*do nothing*/ }
}
return ret;
}
int ObPLCodeGenerator::generate_into(const ObPLInto &into,
ObLLVMValue &into_array_value,
ObLLVMValue &into_count_value,
ObLLVMValue &type_array_value,
ObLLVMValue &type_count_value,
ObLLVMValue &exprs_not_null_array_value,
ObLLVMValue &pl_integer_range_array_value,
ObLLVMValue &is_bulk)
{
int ret = OB_SUCCESS;
if (OB_FAIL(generate_expression_array(into.get_into(), into_array_value, into_count_value))) {
LOG_WARN("Failed to generate_into", K(ret));
} else if (OB_FAIL(helper_.get_int8(static_cast<int64_t>(into.is_bulk()), is_bulk))) {
LOG_WARN("failed to get int8", K(ret));
} else if (into.get_data_type().count() != into.get_not_null_flags().count()
|| into.get_data_type().count() != into.get_pl_integer_ranges().count()) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("not null flags count is inconsistent with data type count.",
K(ret),
K(into.get_data_type()),
K(into.get_not_null_flags()),
K(into.get_pl_integer_ranges()));
} else {
ObLLVMType data_type;
ObLLVMType data_type_pointer;
ObLLVMType array_type;
ObLLVMType bool_type;
ObLLVMType not_null_pointer;
ObLLVMType nn_array_type; //not null array
ObLLVMType int64_type;
ObLLVMType range_range_pointer;
ObLLVMType range_range_type;
OZ (adt_service_.get_data_type(data_type));
OZ (data_type.get_pointer_to(data_type_pointer));
OZ (ObLLVMHelper::get_array_type(data_type,
into.get_data_type().count(),
array_type));
OZ (helper_.create_alloca(ObString("datatype_array"),
array_type,
type_array_value));
OZ (helper_.get_llvm_type(ObTinyIntType, bool_type));
OZ (bool_type.get_pointer_to(not_null_pointer));
OZ (ObLLVMHelper::get_array_type(bool_type,
into.get_data_type().count(),
nn_array_type));
OZ (helper_.create_alloca(ObString("not_null_array"),
nn_array_type,
exprs_not_null_array_value));
OZ (helper_.get_llvm_type(ObIntType, int64_type));
OZ (int64_type.get_pointer_to(range_range_pointer));
OZ (ObLLVMHelper::get_array_type(int64_type,
into.get_data_type().count(),
range_range_type));
OZ (helper_.create_alloca(ObString("range_array"),
range_range_type,
pl_integer_range_array_value));
if (OB_SUCC(ret)) {
ObLLVMValue type_value;
ObLLVMValue nnv_dest; // nnv is short for not null value
ObLLVMValue nnv_src;
ObLLVMValue src_not_null_value;
ObLLVMValue range_dest;
ObLLVMValue range_src;
for (int64_t i = 0; OB_SUCC(ret) && i < into.get_data_type().count(); ++i) {
type_value.reset();
nnv_src.reset();
OZ (helper_.create_gep(ObString("extract_datatype"), type_array_value, i, type_value));
OZ (store_data_type(into.get_data_type(i), type_value));
OZ (helper_.create_gep(ObString("extract_not_null"),
exprs_not_null_array_value, i, nnv_dest));
OZ (helper_.get_int8(static_cast<int64_t>(into.get_not_null_flag(i)), nnv_src));
OZ (helper_.create_store(nnv_src, nnv_dest));
OZ (helper_.create_gep(ObString("extract_range"),
pl_integer_range_array_value, i, range_dest));
OZ (helper_.get_int64(into.get_pl_integer_range(i), range_src));
OZ (helper_.create_store(range_src, range_dest));
}
OZ (helper_.create_bit_cast(ObString("datatype_array_to_pointer"),
type_array_value, data_type_pointer, type_array_value));
OZ (helper_.create_bit_cast(ObString("not_null_array_to_pointer"),
exprs_not_null_array_value, not_null_pointer, exprs_not_null_array_value));
OZ (helper_.create_bit_cast(ObString("range_array_to_pointer"),
pl_integer_range_array_value, range_range_pointer, pl_integer_range_array_value));
OZ (helper_.get_int64(static_cast<int64_t>(into.get_data_type().count()), type_count_value));
}
}
return ret;
}
int ObPLCodeGenerator::generate_into_restore(const ObIArray<int64_t> &into,
const common::ObIArray<ObRawExpr*> *exprs,
const ObPLSymbolTable *symbol_table)
{
int ret = OB_SUCCESS;
if (OB_ISNULL(exprs) || OB_ISNULL(symbol_table)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("symbol_table is NULL", K(exprs), K(symbol_table), K(ret));
} else {
for (int64_t i = 0; OB_SUCC(ret) && i < into.count(); ++i) {
const ObRawExpr *into_expr = NULL;
if (into.at(i) < 0 || into.at(i) >= exprs->count() || OB_ISNULL(into_expr = exprs->at(into.at(i)))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("into expr is NULL", K(i), K(into.at(i)), K(exprs->count()), K(into_expr), K(ret));
} else if (into_expr->is_const_raw_expr() && !into_expr->get_result_type().is_ext()) {
//只有值传递的变量(Local 基本变量)需要回存进全局符号表
if (!static_cast<const ObConstRawExpr*>(into_expr)->get_value().is_unknown()) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("Invalid into expr", K(static_cast<const ObConstRawExpr*>(into_expr)->get_value()), K(ret));
} else {
int64_t idx = static_cast<const ObConstRawExpr*>(into_expr)->get_value().get_unknown();
const ObPLVar* var = symbol_table->get_symbol(idx);
if (OB_ISNULL(var)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("var is NULL", K(i), K(idx), K(var), K(ret));
} else {
ObLLVMValue result;
if (OB_FAIL(extract_datum_from_context(vars_.at(CTX_IDX), idx, var->get_type().get_obj_type(), result))) {
LOG_WARN("failed to extract_value_from_context", K(ret));
} else if (var->get_type().is_composite_type()) {
ObLLVMType ir_type;
ObLLVMType ir_popinter_type;
if (OB_FAIL(get_llvm_type(var->get_type(), ir_type))) {
LOG_WARN("get pl ir type failed", K(ret));
} else if (OB_FAIL(ir_type.get_pointer_to(ir_popinter_type))) {
LOG_WARN("failed to get pointer", K(ret));
} else if (OB_FAIL(helper_.create_int_to_ptr(ObString("cast_int_to_ptr"), result, ir_popinter_type, result))) {
LOG_WARN("failed to create_int_to_ptr", K(ret));
} else if (OB_FAIL(helper_.create_load(ObString("load_var"), result, result))) {
LOG_WARN("failed to create_load", K(ret));
} else { /*do nothing*/ }
} else { /*do nothing*/ }
if (OB_SUCC(ret) && OB_FAIL(helper_.create_store(result, vars_.at(idx + USER_ARG_OFFSET)))) {
LOG_WARN("failed to create_store", K(ret));
}
}
}
} else { /*do nothing*/ }
}
}
return ret;
}
int ObPLCodeGenerator::generate_set_variable(int64_t expr,
ObLLVMValue &value,
bool is_default,
int64_t stmt_id,
bool in_notfound,
bool in_warning)
{
int ret = OB_SUCCESS;
ObSEArray<ObLLVMValue, 5> args;
ObLLVMValue expr_addr;
ObLLVMValue is_default_value;
ObLLVMValue result;
if (OB_FAIL(args.push_back(get_vars().at(CTX_IDX)))) { //PL的执行环境
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(generate_pointer(get_expr(expr), expr_addr))) {
LOG_WARN("failed to generate a pointer", K(ret));
} else if (OB_FAIL(args.push_back(expr_addr))) { //expr
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(args.push_back(value))) { //value
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(helper_.get_int8(is_default, is_default_value))) {
LOG_WARN("failed tio get int8", K(is_default), K(ret));
} else if (OB_FAIL(args.push_back(is_default_value))) { //is_default
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(get_helper().create_call(ObString("spi_set_variable"), get_spi_service().spi_set_variable_, args, result))) {
LOG_WARN("failed to create call", K(ret));
} else if (OB_FAIL(check_success(result, stmt_id, in_notfound, in_warning))) {
LOG_WARN("failed to check success", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::generate_update_package_changed_info(const ObPLStmt &s,
uint64_t package_id,
uint64_t var_idx)
{
int ret = OB_SUCCESS;
ObSEArray<ObLLVMValue, 3> args;
ObLLVMValue package_id_value;
ObLLVMValue var_idx_value;
ObLLVMValue result;
OZ (args.push_back(get_vars().at(CTX_IDX)));
OZ (helper_.get_int64(package_id, package_id_value));
OZ (args.push_back(package_id_value));
OZ (helper_.get_int64(var_idx, var_idx_value));
OZ (args.push_back(var_idx_value));
OZ (get_helper().create_call(ObString("spi_update_package_change_info"),
get_spi_service().spi_update_package_change_info_,
args, result));
OZ (check_success(
result, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
return ret;
}
int ObPLCodeGenerator::generate_exception(ObLLVMValue &type,
ObLLVMValue &ob_error_code,
ObLLVMValue &error_code,
ObLLVMValue &sql_state,
ObLLVMValue &str_len,
ObLLVMValue &stmt_id,
ObLLVMBasicBlock &normal,
ObLLVMValue &line_number,
bool in_notfound,
bool in_warning,
bool signal)
{
int ret = OB_SUCCESS;
if (OB_FAIL(helper_.set_insert_point(get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else {
ObSEArray<ObLLVMValue, 2> args;
ObLLVMValue arg;
ObLLVMValue status;
if (OB_FAIL(extract_status_from_context(get_vars().at(CTX_IDX), status))) {
LOG_WARN("failed to extract_status_from_context", K(ret));
} else if (OB_FAIL(helper_.create_store(ob_error_code, status))) {
LOG_WARN("failed to create_store", K(ret));
} else if (OB_FAIL(extract_pl_ctx_from_context(get_vars().at(CTX_IDX), arg))) { //obplcontext
LOG_WARN("failed to set pl context", K(ret));
} else if (OB_FAIL(args.push_back(arg))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(extract_pl_function_from_context(get_vars().at(CTX_IDX), arg))) { //obplfunction
LOG_WARN("failed to set pl function", K(ret));
} else if (OB_FAIL(args.push_back(arg))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(args.push_back(line_number))) { // line number
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(extract_allocator_from_context(get_vars().at(CTX_IDX), arg))) { //allocator
LOG_WARN("failed to set extract_allocator_from_context", K(ret));
} else if (OB_FAIL(args.push_back(arg))) {
LOG_WARN("push_back error", K(ret));
} else {
ObLLVMType condition_type;
ObLLVMValue condition;
ObLLVMValue type_pointer;
ObLLVMValue error_code_pointer;
ObLLVMValue sql_state_pointer;
ObLLVMValue str_len_pointer;
ObLLVMValue stmt_id_pointer;
ObLLVMValue signal_pointer;
ObLLVMValue int_value;
if (OB_FAIL(adt_service_.get_pl_condition_value(condition_type))) {
LOG_WARN("failed to get_pl_condition_value", K(ret));
} else if (OB_FAIL(helper_.create_alloca(ObString("handler_condition"), condition_type, condition))) {
LOG_WARN("failed to create_alloca", K(ret));
} else if (OB_FAIL(extract_type_ptr_from_condition_value(condition, type_pointer))) {
LOG_WARN("failed to extract_type_ptr_from_condition_value", K(ret));
} else if (OB_FAIL(helper_.create_store(type, type_pointer))) {
LOG_WARN("failed to create_store", K(ret));
} else if (OB_FAIL(extract_code_ptr_from_condition_value(condition, error_code_pointer))) {
LOG_WARN("failed to extract_code_ptr_from_condition_value", K(ret));
} else if (OB_FAIL(helper_.create_store(error_code, error_code_pointer))) {
LOG_WARN("failed to create_store", K(ret));
} else if (OB_FAIL(extract_name_ptr_from_condition_value(condition, sql_state_pointer))) {
LOG_WARN("failed to extract_name_ptr_from_condition_value", K(ret));
} else if (OB_FAIL(helper_.create_store(sql_state, sql_state_pointer))) {
LOG_WARN("failed to create_store", K(ret));
} else if (OB_FAIL(extract_len_ptr_from_condition_value(condition, str_len_pointer))) {
LOG_WARN("failed to extract_len_ptr_from_condition_value", K(ret));
} else if (OB_FAIL(helper_.create_store(str_len, str_len_pointer))) {
LOG_WARN("failed to create_store", K(ret));
} else if (OB_FAIL(extract_stmt_ptr_from_condition_value(condition, stmt_id_pointer))) {
LOG_WARN("failed to extract_stmt_ptr_from_condition_value", K(ret));
} else if (OB_FAIL(helper_.create_store(stmt_id, stmt_id_pointer))) {
LOG_WARN("failed to create_store", K(ret));
} else if (OB_FAIL(extract_signal_ptr_from_condition_value(condition, signal_pointer))) {
LOG_WARN("failed to extract_signal_ptr_from_condition_value", K(ret));
} else if (OB_FAIL(helper_.get_int8(signal, int_value))) {
LOG_WARN("failed to get int8", K(ret));
} else if (OB_FAIL(helper_.create_store(int_value, signal_pointer))) {
LOG_WARN("failed to create_store", K(ret));
} else if (OB_FAIL(args.push_back(condition))) {
LOG_WARN("push_back error", K(ret));
#ifndef NDEBUG
} else if (OB_FAIL(helper_.get_int64(4444, int_value))) {
LOG_WARN("failed to get int64", K(ret));
} else if (OB_FAIL(generate_debug(ObString("debug"), int_value))) {
LOG_WARN("failed to create_call", K(ret));
#endif
} else {
ObLLVMValue exception;
if (OB_FAIL(helper_.create_call(ObString("create_exception"), get_eh_service().eh_create_exception_, args, exception))) {
LOG_WARN("failed to create_call", K(ret));
} else {
OZ (raise_exception(exception, error_code, sql_state, normal, in_notfound, in_warning, signal));
}
}
}
}
return ret;
}
int ObPLCodeGenerator::raise_exception(ObLLVMValue &exception,
ObLLVMValue &error_code,
ObLLVMValue &sql_state,
ObLLVMBasicBlock &normal,
bool in_notfound,
bool in_warning,
bool signal)
{
int ret = OB_SUCCESS;
/*
* MySQL模式下:
* 如果到了顶层handler,都没有捕捉到了该exception,视exception的类型决定下一个目的地:
* SQLEXCEPTION:继续抛出异常
* SQLWARNING:跳到抛出exception的语句的下一句
* NOT FOUND:如果显式(raised by SIGNAL or RESIGNAL)抛出,继续抛出异常;如果是隐式抛出(raised normally),跳到抛出exception的语句的下一句
*
* Oracle模式下:
* 遇到错误都抛出
*/
ObLLVMBasicBlock current = get_current();
ObLLVMBasicBlock raise_exception;
OZ (helper_.create_block(ObString("raise_exception"), get_func(), raise_exception));
OZ (set_current(raise_exception));
if (OB_SUCC(ret)) {
ObLLVMValue ret_value;
ObLLVMValue exception_result;
if (OB_ISNULL(get_current_exception())) {
OZ (helper_.create_call(ObString("raise_exception"),
get_eh_service().eh_raise_exception_,
exception,
exception_result));
#if defined(__aarch64__)
// On ARM, _Unwind_RaiseException may failed.
OZ (generate_debug(ObString("CALL: failed to raise exception!"), exception_result));
OZ (helper_.create_load(ObString("load_ret"), get_vars().at(RET_IDX), ret_value));
OZ (helper_.create_ret(ret_value));
#else
OZ (helper_.create_unreachable());
#endif
} else {
/*
* 关闭从当前位置开始到目的exception位置所有For Loop Cursor
* 因为此时已经在exception过程中,只尝试关闭,不再check_success
*/
if (NULL != get_current_loop()
&& get_current_loop()->level_ >= get_current_exception()->level_) {
for (int64_t i = get_loop_count(); OB_SUCC(ret) && i > 0; --i) {
if (get_loops()[i - 1].level_ >= get_current_exception()->level_
&& NULL != get_loops()[i - 1].cursor_) {
LOG_INFO("close ForLoop Cursor while raising exception",
K(*get_loops()[i - 1].cursor_->get_cursor()),
K(ret));
OZ (generate_close(*get_loops()[i - 1].cursor_,
get_loops()[i - 1].cursor_->get_cursor()->get_package_id(),
get_loops()[i - 1].cursor_->get_cursor()->get_routine_id(),
get_loops()[i - 1].cursor_->get_index(),
false,/*cannot ignoe as must have been opened*/
false/*此时已经在exception里,如果出错不能再抛exception了*/));
}
}
}
OZ (helper_.create_invoke(ObString("raise_exception"),
get_eh_service().eh_raise_exception_,
exception,
get_exit(),
get_current_exception()->exception_,
exception_result));
}
}
if (OB_SUCC(ret)) {
OZ (set_current(current));
if (OB_FAIL(ret)) {
} else if (lib::is_oracle_mode()) {
OZ (helper_.create_br(raise_exception));
} else if (lib::is_mysql_mode()) {
ObLLVMBasicBlock normal_raise_block;
ObLLVMValue exception_class;
ObLLVMSwitch switch_inst1;
ObLLVMSwitch switch_inst2;
ObLLVMValue int_value;
OZ (helper_.create_block(ObString("normal_raise_block"), get_func(), normal_raise_block));
OZ (helper_.create_switch(error_code, normal_raise_block, switch_inst1));
OZ (helper_.get_int64(ER_WARN_TOO_MANY_RECORDS, int_value));
OZ (switch_inst1.add_case(int_value, raise_exception));
OZ (helper_.get_int64(WARN_DATA_TRUNCATED, int_value));
OZ (switch_inst1.add_case(int_value, raise_exception));
OZ (helper_.get_int64(ER_SIGNAL_WARN, int_value));
OZ (switch_inst1.add_case(int_value, raise_exception));
OZ (set_current(normal_raise_block));
OZ (helper_.create_call(ObString("get_exception_class"), get_eh_service().eh_classify_exception, sql_state, exception_class));
OZ (helper_.create_switch(exception_class, raise_exception, switch_inst2));
OZ (helper_.get_int64(SQL_WARNING, int_value));
OZ (switch_inst2.add_case(int_value, in_warning ? raise_exception : normal));
OZ (helper_.get_int64(NOT_FOUND, int_value));
OZ (switch_inst2.add_case(int_value, signal || in_notfound ? raise_exception : normal));
}
}
return ret;
}
int ObPLCodeGenerator::check_success(jit::ObLLVMValue &ret_err, int64_t stmt_id,
bool in_notfound, bool in_warning, bool signal)
{
int ret = OB_SUCCESS;
if (OB_FAIL(helper_.set_insert_point(get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(helper_.create_store(ret_err, vars_.at(RET_IDX)))) {
LOG_WARN("failed to create_store", K(ret));
} else {
ObLLVMBasicBlock success_branch;
ObLLVMBasicBlock fail_branch;
ObLLVMValue is_true;
if (OB_FAIL(helper_.create_block(ObString("ob_success"), get_func(), success_branch))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(helper_.create_block(ObString("ob_fail"), get_func(), fail_branch))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(helper_.create_icmp_eq(ret_err, OB_SUCCESS, is_true))) {
LOG_WARN("failed to create_icmp_eq", K(ret));
} else if (OB_FAIL(helper_.create_cond_br(is_true, success_branch, fail_branch))) {
LOG_WARN("failed to create_cond_br", K(ret));
} else if (OB_FAIL(set_current(fail_branch))) {
LOG_WARN("failed to set_current", K(ret));
} else {
ObLLVMType int_type;
ObLLVMType char_type;
ObLLVMValue type_ptr;
ObLLVMValue error_code_ptr;
ObLLVMValue sql_state_ptr;
ObLLVMValue str_len_ptr;
if (OB_FAIL(helper_.get_llvm_type(ObIntType, int_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObCharType, char_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.create_alloca(ObString("exception_type"), int_type, type_ptr))) {
LOG_WARN("failed to create_alloca", K(ret));
} else if (OB_FAIL(helper_.create_alloca(ObString("error_code"), int_type, error_code_ptr))) {
LOG_WARN("failed to create_alloca", K(ret));
} else if (OB_FAIL(helper_.create_alloca(ObString("sql_state"), char_type, sql_state_ptr))) {
LOG_WARN("failed to create_alloca", K(ret));
} else if (OB_FAIL(helper_.create_alloca(ObString("str_len"), int_type, str_len_ptr))) {
LOG_WARN("failed to create_alloca", K(ret));
} else {
ObSEArray<ObLLVMValue, 2> args;
ObLLVMValue oracle_mode;
if (OB_FAIL(helper_.get_int8(oracle_mode_, oracle_mode))) {
LOG_WARN("helper get int8 failed", K(ret));
} else if (OB_FAIL(args.push_back(oracle_mode))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(args.push_back(ret_err))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(args.push_back(type_ptr))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(args.push_back(error_code_ptr))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(args.push_back(sql_state_ptr))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(args.push_back(str_len_ptr))) {
LOG_WARN("push_back error", K(ret));
} else {
ObLLVMValue result;
ObLLVMValue type;
ObLLVMValue error_code;
ObLLVMValue sql_state;
ObLLVMValue str_len;
ObLLVMValue stmt_id_value;
ObLLVMValue line_number_value;
// stmt id目前是col和line的组合,暂时先用这个
int64_t line_number = stmt_id;
if (OB_FAIL(helper_.get_int64(line_number, line_number_value))) {
LOG_WARN("failed to get_line_number", K(ret));
} else if (OB_FAIL(helper_.get_int64(stmt_id, stmt_id_value))) {
LOG_WARN("failed to get_int64", K(ret));
} else if (OB_FAIL(helper_.create_call(ObString("convert_exception"), get_eh_service().eh_convert_exception_, args, result))) {
LOG_WARN("failed to create_call", K(ret));
} else if (OB_FAIL(helper_.create_load(ObString("load_type"), type_ptr, type))) {
LOG_WARN("failed to create_load", K(ret));
} else if (OB_FAIL(helper_.create_load(ObString("load_error_code"), error_code_ptr, error_code))) {
LOG_WARN("failed to create_load", K(ret));
} else if (OB_FAIL(helper_.create_load(ObString("load_sql_state"), sql_state_ptr, sql_state))) {
LOG_WARN("failed to create_load", K(ret));
} else if (OB_FAIL(helper_.create_load(ObString("load_str_len"), str_len_ptr, str_len))) {
LOG_WARN("failed to create_load", K(ret));
} else if (OB_FAIL(generate_exception(type, ret_err, error_code, sql_state, str_len, stmt_id_value, success_branch, line_number_value, in_notfound, in_warning, signal))) {
LOG_WARN("failed to generate exception", K(ret));
} else if (OB_FAIL(set_current(success_branch))) {
LOG_WARN("failed to set_current", K(ret));
} else { /*do nothing*/ }
}
}
}
}
return ret;
}
int ObPLCodeGenerator::finish_current(const ObLLVMBasicBlock &next)
{
int ret = OB_SUCCESS;
if (NULL == get_current().get_v()) {
//如果current为NULL,说明控制流已经被显式切走了(例如Iterate、Leave语句)
} else if (get_current().get_v() == get_exit().get_v()) {
//如果current是exit,说明控制流已经显式结束(如return)
} else if (OB_FAIL(helper_.set_insert_point(get_current()))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(helper_.create_br(next))) {
LOG_WARN("failed to create br", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::generate_prototype()
{
int ret = OB_SUCCESS;
ObSEArray<ObLLVMType, 8> arg_types;
ObLLVMFunctionType ft;
ObLLVMType pl_exec_context_type;
ObLLVMType pl_exec_context_pointer_type;
ObLLVMType argv_type;
ObLLVMType argv_pointer_type;
if (OB_FAIL(adt_service_.get_pl_exec_context(pl_exec_context_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(pl_exec_context_type.get_pointer_to(pl_exec_context_pointer_type))) {
LOG_WARN("failed to get_pointer_to", K(ret));
} else if (OB_FAIL(adt_service_.get_argv(argv_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(argv_type.get_pointer_to(argv_pointer_type))) {
LOG_WARN("failed to get_pointer_to", K(ret));
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
ObLLVMType int64_type;
if (OB_FAIL(arg_types.push_back(pl_exec_context_pointer_type))) { //函数第一个参数必须是基础环境信息隐藏参数ObPLExecCtx*
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(helper_.get_llvm_type(ObIntType, int64_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(arg_types.push_back(int64_type))) { //第二个参数是int64_t ArgC
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(arg_types.push_back(argv_pointer_type))) { //第三个参数是int64_t[] ArgV
LOG_WARN("push_back error", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
if (get_ast().get_ret_type().is_obj_type()
&& (OB_ISNULL(get_ast().get_ret_type().get_data_type())
|| get_ast().get_ret_type().get_data_type()->get_meta_type().is_invalid())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("return type is invalid", K(func_), K(ret));
} else {
ObLLVMType ret_type;
ObLLVMFunctionType ft;
if (OB_FAIL(helper_.get_llvm_type(ObInt32Type, ret_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(ObLLVMFunctionType::get(ret_type, arg_types, ft))) {
LOG_WARN("failed to get function type", K(ret));
} else if (OB_FAIL(helper_.create_function(get_ast().get_name(), ft, func_))) {
LOG_WARN("failed to create function", K(ret));
} else { /*do nothing*/ }
}
}
//生成参数
if (OB_SUCC(ret)) {
//获取数据类型属性
int64_t size = 0;
if (OB_FAIL(func_.get_argument_size(size))) {
LOG_WARN("failed to get argument size", K(ret));
} else {
ObLLVMValue arg;
int64_t i = CTX_IDX;
for (int64_t j = 0; OB_SUCC(ret) && j < size; ++j) {
if (OB_FAIL(func_.get_argument(j, arg))) {
LOG_WARN("failed to get argument", K(j), K(ret));
} else if (OB_FAIL(arg.set_name(ArgName[i]))) {
LOG_WARN("failed to set name", K(j), K(ret));
} else {
vars_.at(i++) = arg;
}
}
}
}
return ret;
}
int ObPLCodeGenerator::unset_debug_location()
{
int ret = OB_SUCCESS;
if (debug_mode_) {
jit::ObLLVMDIScope scope;
if (OB_FAIL(di_helper_.get_current_scope(scope))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(helper_.unset_debug_location(&scope))) {
LOG_WARN("failed to unset debug location", K(ret));
}
}
return ret;
}
int ObPLCodeGenerator::set_debug_location(const ObPLStmt &s)
{
int ret = OB_SUCCESS;
if (debug_mode_) {
uint32_t line = s.get_line() + 1;
uint32_t col = s.get_col();
jit::ObLLVMDIScope scope;
if (OB_FAIL(di_helper_.get_current_scope(scope))) {
LOG_WARN("failed to get current scope", K(line), K(col), K(ret));
} else if (OB_FAIL(helper_.set_debug_location(line, col, &scope))) {
LOG_WARN("failed to set debug location", K(line), K(col), K(ret));
}
}
return ret;
}
int ObPLCodeGenerator::get_di_llvm_type(const ObPLDataType &pl_type, ObLLVMDIType &di_type)
{
int ret = OB_SUCCESS;
if (debug_mode_) {
if (pl_type.is_obj_type()) {
if (OB_FAIL(di_adt_service_.get_di_type(pl_type.get_obj_type(), di_type))) {
LOG_WARN("failed to get di type", K(ret));
}
} else if (pl_type.is_user_type()) {
uint64_t user_type_id = pl_type.get_user_type_id();
if (OB_FAIL(di_user_type_map_.get_refactored(user_type_id, di_type))) {
LOG_WARN("failed to get di user type map", K(user_type_id), K(ret));
}
} else {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("pl type is invalid", K(pl_type.get_type()));
}
}
return ret;
}
int ObPLCodeGenerator::get_di_datum_type(const ObPLDataType &pl_type, ObLLVMDIType &di_type)
{
int ret = OB_SUCCESS;
if (debug_mode_) {
if (pl_type.is_obj_type()) {
// why we call get_datum_type(), not get_di_type().
// see ObPLCodeGenerate::get_datum_type().
if (OB_FAIL(di_adt_service_.get_datum_type(pl_type.get_obj_type(), di_type))) {
LOG_WARN("failed to get datum type", K(ret));
}
} else if (pl_type.is_user_type()) {
uint64_t user_type_id = pl_type.get_user_type_id();
if (OB_FAIL(di_user_type_map_.get_refactored(user_type_id, di_type))) {
LOG_WARN("failed to get di user type map", K(user_type_id), K(ret));
}
} else {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("pl type is invalid", K(pl_type.get_type()));
}
}
return ret;
}
int ObPLCodeGenerator::generate_di_user_type(const ObUserDefinedType &type, uint32_t line)
{
int ret = OB_SUCCESS;
if (debug_mode_ && false) {
if (type.is_record_type()) {
const ObRecordType &record_type = static_cast<const ObRecordType&>(type);
if (OB_FAIL(generate_di_record_type(record_type, line))) {
LOG_WARN("failed to generate di record type", K(ret));
}
} else {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("user defined type is invalid", K(type.get_type()));
}
}
return ret;
}
int ObPLCodeGenerator::generate_di_record_type(const ObRecordType &record_type, uint32_t line)
{
int ret = OB_SUCCESS;
if (debug_mode_) {
ObSEArray<ObString, 8> cell_names;
ObSEArray<ObLLVMDIType, 8> cell_types;
ObLLVMDIType di_record_type;
for (int64_t i = 0; OB_SUCC(ret) && i < record_type.get_record_member_count(); ++i) {
const ObString *cell_name = record_type.get_record_member_name(i);
const ObPLDataType *data_type = record_type.get_record_member_type(i);
ObLLVMDIType cell_type;
if (OB_ISNULL(cell_name) || OB_ISNULL(data_type)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("cell name or data type is null");
} else if (OB_FAIL(get_di_datum_type(*data_type, cell_type))) {
LOG_WARN("failed to get di datum type", K(i), K(*data_type), K(ret));
} else if (OB_FAIL(cell_names.push_back(*cell_name))) {
LOG_WARN("failed to push back cell name", K(ret));
} else if (OB_FAIL(cell_types.push_back(cell_type))) {
LOG_WARN("failed to push back cell type", K(ret));
}
}
if (OB_SUCC(ret)) {
const ObString &type_name = record_type.get_name();
uint64_t type_id = record_type.get_user_type_id();
if (OB_FAIL(di_adt_service_.get_record_type(type_name, line, cell_names, cell_types, di_record_type))) {
LOG_WARN("failed to get record type", K(record_type), K(ret));
} else if (OB_FAIL(di_user_type_map_.set_refactored(type_id, di_record_type))) {
LOG_WARN("failed to set di record type to user type map", K(record_type), K(ret));
}
}
}
return ret;
}
int ObPLCodeGenerator::generate_di_argument()
{
int ret = OB_SUCCESS;
if (debug_mode_) {
ObLLVMDIType di_ret_type;
ObLLVMDIType di_ctx_type;
ObLLVMDIType di_argc_type;
ObLLVMDIType di_argv_type;
ObLLVMDIType di_argv_ptr_type;
if (OB_FAIL(di_adt_service_.get_obj_type(ObInt32Type, di_ret_type))) {
LOG_WARN("failed to get element di type", K(ret));
} else if (OB_FAIL(di_adt_service_.get_exec_ctx(di_ctx_type))) {
LOG_WARN("failed to get element di type", K(ret));
} else if (OB_FAIL(di_adt_service_.get_obj_type(ObIntType, di_argc_type))) {
LOG_WARN("failed to get element di type", K(ret));
} else if (OB_FAIL(di_adt_service_.get_argv(get_ast().get_arg_count(), di_argv_type))) {
LOG_WARN("failed to get element di type", K(ret));
} else if (OB_FAIL(di_helper_.create_pointer_type(di_argv_type, di_argv_ptr_type))) {
LOG_WARN("failed to create di poiner type", K(ret));
}
}
return ret;
}
int ObPLCodeGenerator::generate_di_local_variable(const ObPLVar &var,
uint32_t arg_no, uint32_t line, ObLLVMValue &value)
{
int ret = OB_SUCCESS;
if (debug_mode_) {
ObLLVMDIType type;
if (OB_FAIL(get_di_llvm_type(var.get_type(), type))) {
LOG_WARN("failed to get llvm di type", K(ret));
} else if (OB_FAIL(generate_di_local_variable(var.get_name(), type, arg_no, line, value))) {
LOG_WARN("failed to generate di local variable", K(ret));
}
}
return ret;
}
int ObPLCodeGenerator::generate_di_local_variable(const ObString &name, ObLLVMDIType &type,
uint32_t arg_no, uint32_t line, ObLLVMValue &value)
{
int ret = OB_SUCCESS;
if (debug_mode_) {
ObLLVMBasicBlock block;
ObLLVMDILocalVariable variable;
if (OB_FAIL(helper_.get_insert_block(block))) {
LOG_WARN("failed to get insert block", K(name), K(ret));
} else if (OB_FAIL(di_helper_.create_local_variable(name, arg_no, line, type, variable))) {
LOG_WARN("failed to create di local variable", K(name), K(ret));
} else if (OB_FAIL(di_helper_.insert_declare(value, variable, block))) {
LOG_WARN("failed to insert declare", K(name), K(ret));
}
}
return ret;
}
int ObPLCodeGenerator::init_di_adt_service()
{
int ret = OB_SUCCESS;
if (debug_mode_) {
ObSqlString sql;
if (get_ast().get_package_name().empty()) {
CK (!get_ast().get_db_name().empty());
OZ (sql.append_fmt("%.*s.%.*s",
get_ast().get_db_name().length(), get_ast().get_db_name().ptr(),
get_ast().get_name().length(), get_ast().get_name().ptr()));
} else {
CK (!get_ast().get_db_name().empty());
OZ (sql.append_fmt("%.*s.%.*s.%.*s",
get_ast().get_db_name().length(), get_ast().get_db_name().ptr(),
get_ast().get_package_name().length(), get_ast().get_package_name().ptr(),
get_ast().get_name().length(), get_ast().get_name().ptr()));
}
if (OB_FAIL(ret)) {
} else if (OB_FAIL(di_adt_service_.init(sql.ptr()))) {
LOG_WARN("failed to init di service", K(ret));
}
}
return ret;
}
int ObPLCodeGenerator::generate_di_prototype()
{
int ret = OB_SUCCESS;
if (debug_mode_) {
ObSqlString sql;
int64_t argc = 0;
ObLLVMDIType ret_type;
ObLLVMDIType pl_exec_ctx_type;
ObLLVMDIType pl_exec_ctx_ptr_type;
ObLLVMDIType argc_type;
ObLLVMDIType argv_type;
ObLLVMDIType argv_ptr_type;
ObSEArray<ObLLVMDIType, 8> member_types;
ObLLVMDISubroutineType sr_type;
ObLLVMDISubprogram sp;
if (get_ast().get_package_name().empty()) {
CK (!get_ast().get_db_name().empty());
OZ (sql.append_fmt("%.*s.%.*s",
get_ast().get_db_name().length(), get_ast().get_db_name().ptr(),
get_ast().get_name().length(), get_ast().get_name().ptr()));
} else {
CK (!get_ast().get_db_name().empty());
OZ (sql.append_fmt("%.*s.%.*s.%.*s",
get_ast().get_db_name().length(), get_ast().get_db_name().ptr(),
get_ast().get_package_name().length(), get_ast().get_package_name().ptr(),
get_ast().get_name().length(), get_ast().get_name().ptr()));
}
if (OB_FAIL(ret)) {
} else if (OB_FAIL(func_.get_argument_size(argc))) {
LOG_WARN("failed to get argument size", K(ret));
} else if (OB_FAIL(di_adt_service_.get_obj_type(ObInt32Type, ret_type))) {
LOG_WARN("failed to get di ret type", K(ret));
} else if (OB_FAIL(di_adt_service_.get_exec_ctx(pl_exec_ctx_type))) {
LOG_WARN("failed to get di pl exec ctx type", K(ret));
} else if (OB_FAIL(di_helper_.create_pointer_type(pl_exec_ctx_type, pl_exec_ctx_ptr_type))) {
LOG_WARN("failed to create di poiner type", K(ret));
} else if (OB_FAIL(di_adt_service_.get_obj_type(ObIntType, argc_type))) {
LOG_WARN("failed to get di argc type", K(ret));
} else if (OB_FAIL(di_adt_service_.get_argv(argc, argv_type))) {
LOG_WARN("failed to get di argv type", K(ret));
} else if (OB_FAIL(di_helper_.create_pointer_type(argv_type, argv_ptr_type))) {
LOG_WARN("failed to create di poiner type", K(ret));
} else if (OB_FAIL(member_types.push_back(ret_type))) {
LOG_WARN("failed to push back di ret type", K(ret));
} else if (OB_FAIL(member_types.push_back(pl_exec_ctx_ptr_type))) {
LOG_WARN("failed to push back di pl exec ctx type", K(ret));
} else if (OB_FAIL(member_types.push_back(argc_type))) {
LOG_WARN("failed to push back di argc type", K(ret));
} else if (OB_FAIL(member_types.push_back(argv_ptr_type))) {
LOG_WARN("failed to push back di argv type", K(ret));
} else if (OB_FAIL(di_helper_.create_subroutine_type(member_types, sr_type))) {
LOG_WARN("failed to create di subroutine type", K(ret));
} else if (OB_ISNULL(sr_type.get_v())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("failed to create di subroutine type", K(ret));
} else if (OB_FAIL(di_helper_.create_function(sql.ptr(), sr_type, sp))) {
LOG_WARN("failed to create di subprogram", K(ret));
} else {
func_.set_subprogram(&sp);
}
}
return ret;
}
int ObPLCodeGenerator::init_argument()
{
int ret = OB_SUCCESS;
ObLLVMValue int32_value;
if (OB_FAIL(helper_.create_ialloca(ObString(ArgName[RET_IDX]), ObInt32Type, OB_SUCCESS, vars_.at(RET_IDX)))) {
LOG_WARN("failed to create_alloca", K(ret));
} else if (OB_FAIL(helper_.get_int32(OB_SUCCESS, int32_value))) {
LOG_WARN("failed to get_int32", K(ret));
} else if (OB_FAIL(helper_.create_store(int32_value, vars_.at(RET_IDX)))) {
LOG_WARN("failed to create_store", K(ret));
} else if (OB_FAIL(unset_debug_location())) {
LOG_WARN("failed to unset debug location", K(ret));
} else {
for (int64_t i = 0; i < get_ast().get_arg_count(); ++i) {
const ObPLVar *var = get_ast().get_symbol_table().get_symbol(i);
if (OB_ISNULL(var)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("var is NULL", K(i), K(var), K(ret));
} else {
ObLLVMValue arg;
ObLLVMValue addr;
ObLLVMValue init_value;
if (var->get_type().is_composite_type() || var->get_type().is_cursor_type()) {
if (OB_FAIL(extract_objparam_from_argv(vars_.at(ARGV_IDX), i, arg))) {
LOG_WARN("failed to extract_objparam_from_argv", K(i), K(ret));
} else if (OB_FAIL(extract_value_from_objparam(arg, ObExtendType, addr))) {
LOG_WARN("failed to extract_value_from_objparam", K(ret));
} else if (OB_FAIL(helper_.get_int32(0, init_value))) {
LOG_WARN("failed to get_int32", K(ret));
} else if (OB_FAIL(set_var_addr_to_param_store(i, addr, init_value))) {
LOG_WARN("set var addr to param store failed", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("obj"), arg, 0, arg))) {
LOG_WARN("failed to create_gep", K(ret));
} else {
get_vars().at(i + USER_ARG_OFFSET) = arg;
}
} else if (OB_ISNULL(var->get_type().get_data_type())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("basic pl type got null data type is unexpected", K(ret), K(var->get_type()));
} else {
const ObDataType &data_type = *(var->get_type().get_data_type());
ObLLVMType ir_type;
if (OB_FAIL(extract_datum_from_argv(vars_.at(ARGV_IDX), i, data_type.get_obj_type(), arg))) {
LOG_WARN("failed to extract_datum_from_argv", K(ret));
} else if (OB_FAIL(ObPLDataType::get_llvm_type(data_type.get_obj_type(), helper_, adt_service_, ir_type))) {
LOG_WARN("failed to get_ir_type", K(data_type.get_obj_type()), K(ret));
} else if (OB_FAIL(helper_.create_alloca(var->get_name(), ir_type, vars_.at(i + USER_ARG_OFFSET)))) {
LOG_WARN("failed to create_alloca", K(ret));
} else if (OB_FAIL(helper_.create_store(arg, vars_.at(i + USER_ARG_OFFSET)))) {
LOG_WARN("failed to create_store", K(ret));
}
}
}
}
}
return ret;
}
int ObPLCodeGenerator::prepare_local_user_type()
{
int ret = OB_SUCCESS;
int64_t count = ast_.get_user_type_table().get_count();
for (int64_t i = 0; OB_SUCC(ret) && i < count; ++i) {
const ObUserDefinedType *user_type = ast_.get_user_type_table().get_type(i);
CK (OB_NOT_NULL(user_type));
OZ (generate_user_type(*user_type), K(i), KPC(user_type));
}
return ret;
}
int ObPLCodeGenerator::prepare_external()
{
int ret = OB_SUCCESS;
int64_t count = ast_.get_user_type_table().get_external_types().count();
for (int64_t i = 0; OB_SUCC(ret) && i < count; ++i) {
const ObUserDefinedType *user_type = ast_.get_user_type_table().get_external_types().at(i);
CK (OB_NOT_NULL(user_type));
OZ (generate_user_type(*user_type), K(i), K(count), KPC(user_type));
}
LOG_DEBUG("pl/sql code generator prepare_external types",
K(ret),
K(ast_.get_user_type_table().get_external_types().count()),
K(ast_.get_user_type_table().get_external_types()),
K(&(ast_.get_user_type_table().get_external_types())),
K(&(ast_)),
K(ast_.get_db_name()),
K(ast_.get_name()),
K(ast_.get_id()));
return ret;
}
int ObPLCodeGenerator::prepare_expression(ObPLCompileUnit &pl_func)
{
int ret = OB_SUCCESS;
ObArray<ObSqlExpression*> array;
for (int64_t i = 0; OB_SUCC(ret) && i < ast_.get_exprs().count(); ++i) {
ObSqlExpression *expr = NULL;
if (OB_FAIL(pl_func.get_sql_expression_factory().alloc(expr))) {
LOG_WARN("failed to alloc expr", K(ret));
} else if (OB_ISNULL(expr)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("failed to create expr", K(ret));
} else if (OB_FAIL(array.push_back(expr))) {
LOG_WARN("push back error", K(ret));
} else { /*do nothing*/ }
}
if (OB_SUCC(ret)) {
if (OB_FAIL(pl_func.set_expressions(array))) {
LOG_WARN("failed to set expressions", K(ret));
}
}
return ret;
}
int ObPLCodeGenerator::prepare_subprogram(ObPLFunction &pl_func)
{
int ret = OB_SUCCESS;
OZ (ObPLCompiler::compile_subprogram_table(allocator_,
session_info_,
pl_func.get_exec_env(),
ast_.get_routine_table(),
pl_func,
schema_guard_));
return ret;
}
int ObPLCodeGenerator::generate_obj_access_expr()
{
int ret = OB_SUCCESS;
for (int64_t i = 0; OB_SUCC(ret) && i < ast_.get_obj_access_exprs().count(); ++i) {
const ObRawExpr *expr = ast_.get_obj_access_expr(i);
if (OB_ISNULL(expr)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("expr is null", K(i), K(ast_.get_obj_access_exprs()), K(ret));
} else if (!expr->is_obj_access_expr()) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("expr is not obj access", K(i), K(*expr), K(ret));
} else {
const ObObjAccessRawExpr *obj_access_expr = static_cast<const ObObjAccessRawExpr*>(expr);
if (OB_FAIL(generate_get_attr_func(
obj_access_expr->get_access_idxs(),
obj_access_expr->get_var_indexs().count() + obj_access_expr->get_param_count(),
obj_access_expr->get_func_name(),
obj_access_expr->for_write()))) {
LOG_WARN("generate get attr function failed",
K(ret),
K(obj_access_expr->get_access_idxs()),
K(obj_access_expr->get_func_name()));
}
}
}
return ret;
}
/*
* 把ObObjAccessRawExpr翻译成一个函数,其中的每个ObObjAccessIdx可能是:
* 1、const,即常量,只能出现在表的下标位置,如a(1)里的1;
* 2、property,即固有属性,只能出现在表的属性位置,如a.count里的count;
* 3、local,即PL内部变量,可能出现在
* 1)、初始内存位置,如a(1)里的a;
* 2)、表的下标位置,如a(i)里的i;
* 3)、record的属性位置,如a.b里的b;
* 4、external(包括IS_PKG、IS_USER、IS_SESSION、IS_GLOBAL),可能出现在
* 1)、初始内存位置,如a(1)里的a;
* 2)、表的下标位置,如a(i)里的i;
* 3)、record的属性位置,如a.b里的b;
* 5、ns不可能出现,在resolver阶段已经被剥掉了。
* 对于以上几种情况:
* 1、直接用作解析复杂变量;
* 2、直接用作解析复杂变量;
* 3、因为把变量在param store里的下标放在var_indexs_里,所以根据var_indexs_从param store里取出值使用;
* 4、因为把计算的结果作为ObObjAccessRawExpr的child传给了表达式计算,所以直接取obj_stack的值使用。
* 函数签名:
* int32_t get_func(int64_t param_cnt, int64_t* params, int64_t* element_val);
* */
int ObPLCodeGenerator::generate_get_attr_func(const ObIArray<ObObjAccessIdx> &idents,
int64_t param_count, const ObString &func_name,
bool for_write)
{
int ret = OB_SUCCESS;
ObSEArray<ObLLVMType, 4> arg_types;
ObLLVMFunctionType ft;
ObLLVMType bool_type;
ObLLVMType array_type;
ObLLVMType array_pointer_type;
ObLLVMType int64_type;
ObLLVMType int32_type;
ObLLVMType int64_pointer_type;
OZ (helper_.get_llvm_type(ObIntType, int64_type));
OZ (helper_.get_llvm_type(ObInt32Type, int32_type));
OZ (helper_.get_array_type(int64_type, param_count, array_type));
OZ (array_type.get_pointer_to(array_pointer_type));
OZ (int64_type.get_pointer_to(int64_pointer_type));
OZ (arg_types.push_back(int64_type));
OZ (arg_types.push_back(array_pointer_type));
OZ (arg_types.push_back(int64_pointer_type));
OZ (ObLLVMFunctionType::get(int32_type, arg_types, ft));
OZ (helper_.create_function(func_name, ft, func_));
OZ (helper_.create_block(ObString("entry"), func_, entry_));
OZ (helper_.create_block(ObString("exit"), func_, exit_));
OZ (set_current(entry_));
if (OB_SUCC(ret)) {
//获取数据类型属性
ObLLVMValue param_cnt;
ObLLVMValue params_ptr;
ObLLVMValue element_value;
ObLLVMValue result_value_ptr;
ObLLVMValue ret_value_ptr;
ObLLVMValue ret_value;
OZ (func_.get_argument(0,param_cnt));
OZ (param_cnt.set_name(ObString("param_cnt")));
OZ (func_.get_argument(1, params_ptr));
OZ (params_ptr.set_name(ObString("param_array")));
OZ (func_.get_argument(2, result_value_ptr));
OZ (result_value_ptr.set_name(ObString("result_value_ptr")));
OZ (helper_.create_alloca(ObString("ret_value"), int32_type, ret_value_ptr));
OZ (generate_get_attr(params_ptr, idents, for_write,
result_value_ptr, ret_value_ptr, exit_), idents);
#ifndef NDEBUG
OZ (generate_debug(ObString("function name"), helper_.get_function_address(func_name)));
#endif
OZ (helper_.create_br(exit_));
OZ (helper_.set_insert_point(exit_));
OZ (helper_.create_load(ObString("load_ret_value"), ret_value_ptr, ret_value));
OZ (helper_.create_ret(ret_value));
}
return ret;
}
int ObPLCodeGenerator::final_expression(ObPLCompileUnit &pl_func)
{
int ret = OB_SUCCESS;
for (int64_t i = 0; OB_SUCC(ret) && i < ast_.get_obj_access_exprs().count(); ++i) {
ObRawExpr *expr = ast_.get_obj_access_expr(i);
if (OB_ISNULL(expr)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("obj_access_expr is null");
} else if (!expr->is_obj_access_expr()) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("not a obj access", K(*expr), K(ret));
} else {
ObObjAccessRawExpr* obj_access_expr = static_cast<ObObjAccessRawExpr*>(expr);
obj_access_expr->set_get_attr_func_addr(helper_.get_function_address(obj_access_expr->get_func_name()));
}
}
if (OB_SUCC(ret)) {
{
// generate static engine expressions
sql::ObRawExprUniqueSet raw_exprs(false);
for (int64_t i = 0; OB_SUCC(ret) && i < ast_.get_exprs().count(); i++) {
OZ(raw_exprs.append(ast_.get_expr(i)));
}
sql::ObStaticEngineExprCG se_cg(pl_func.get_allocator(),
&session_info_,
&schema_guard_,
0 /* original param cnt */,
0/* param count*/);
se_cg.set_rt_question_mark_eval(true);
OZ(se_cg.generate(raw_exprs, pl_func.get_frame_info()));
uint32_t expr_op_size = 0;
RowDesc row_desc;
ObExprGeneratorImpl expr_generator(pl_func.get_expr_operator_factory(), 0, 0,
&expr_op_size, row_desc);
for (int64_t i = 0; OB_SUCC(ret) && i < ast_.get_exprs().count(); ++i) {
ObRawExpr *raw_expr = ast_.get_expr(i);
ObSqlExpression *expression = static_cast<ObSqlExpression*>(get_expr(i));
if (OB_ISNULL(raw_expr) || OB_ISNULL(expression)) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("Invalid arguments", K(i), K(raw_expr), K(expression), K(ret));
} else {
// TODO bin.lb: No need to generate expression if static engine enabled
// https://aone.alibaba-inc.com/project/81079/task/34963274
if (OB_FAIL(expr_generator.generate(*raw_expr, *expression))) {
SQL_LOG(WARN, "Generate post_expr error", K(ret), KPC(raw_expr));
} else {
expression->set_expr(raw_expr->rt_expr_);
}
}
}
if (OB_SUCC(ret)) {
pl_func.set_expr_op_size(std::max(pl_func.get_frame_info().need_ctx_cnt_,
static_cast<int64_t>(expr_op_size)));
}
}
}
return ret;
}
int ObPLCodeGenerator::generate_goto_label(const ObPLStmt &stmt)
{
int ret = OB_SUCCESS;
if (stmt.get_is_goto_dst()) {
if (NULL == get_current().get_v()) {
//控制流已断,后面的语句不再处理
} else {
// 去看一下对应的goto是否已经cg了,没有的话就记录一下这个label地址。
hash::HashMapPair<ObPLCodeGenerator::goto_label_flag, ObLLVMBasicBlock> pair;
int tmp_ret = get_goto_label_map().get_refactored(stmt.get_stmt_id(), pair);
if (OB_HASH_NOT_EXIST == tmp_ret) {
ObLLVMBasicBlock label_block;
const ObString *lab = stmt.get_label();
if (OB_FAIL(get_helper().create_block(NULL == lab ? ObString("") : *lab, get_func(),
label_block))) {
LOG_WARN("create goto label failed", K(ret));
} else if (OB_FAIL(get_helper().create_br(label_block))) {
LOG_WARN("failed to create_br", K(ret));
} else if (OB_FAIL(get_helper().set_insert_point(label_block))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(set_current(label_block))) {
LOG_WARN("failed to set current block", K(ret));
} else if (OB_FAIL(pair.init(ObPLCodeGenerator::goto_label_flag::GOTO_LABEL_NONEXIST,
label_block))) {
LOG_WARN("init label block pair failed.", K(ret));
} else if (OB_FAIL(get_goto_label_map().set_refactored(stmt.get_stmt_id(), pair))) {
LOG_WARN("set label block failed", K(ret));
} else {}
} else if (OB_SUCCESS == tmp_ret) {
// 这个时候,goto已经cg了,拿到对应的block地址,做为新的block
if (ObPLCodeGenerator::goto_label_flag::GOTO_LABEL_EXIST == pair.first) {
ObLLVMBasicBlock &goto_block = pair.second;
if (OB_FAIL(get_helper().create_br(goto_block))) {
LOG_WARN("failed to create_br", K(ret));
} else if (OB_FAIL(get_helper().set_insert_point(goto_block))) {
LOG_WARN("failed to set insert point", K(ret));
} else if (OB_FAIL(set_current(goto_block))) {
LOG_WARN("failed to set current block", K(ret));
} else {}
} else {
// do nothing
}
} else {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("failed to label block", K(ret));
}
}
} else {
// do nothing
}
return ret;
}
int ObPLCodeGenerator::generate_out_param(
const ObPLStmt &s, const ObIArray<InOutParam> &param_desc, ObLLVMValue &params, int64_t i)
{
int ret = OB_SUCCESS;
ObLLVMType obj_param_type;
ObLLVMType obj_param_type_pointer;
ObLLVMValue pp_arg;
ObLLVMValue p_arg;
OZ (get_adt_service().get_objparam(obj_param_type));
OZ (obj_param_type.get_pointer_to(obj_param_type_pointer));
// 获取输出结果
OZ (extract_arg_from_argv(params, i, pp_arg));
OZ (get_helper().create_load("load_out_arg_pointer", pp_arg, p_arg));
OZ (get_helper().create_int_to_ptr(
ObString("cast_pointer_to_arg"), p_arg, obj_param_type_pointer, p_arg));
if (OB_FAIL(ret)) {
} else if (OB_INVALID_INDEX != param_desc.at(i).out_idx_) { // 处理本地变量
//存储进全局符号表和param store
ObLLVMValue result;
ObLLVMValue p_param;
ObPLDataType pl_type = s.get_variable(param_desc.at(i).out_idx_)->get_type();
if (pl_type.is_composite_type() || pl_type.is_cursor_type()) {
if (param_desc.at(i).is_pure_out()) {
// 对于INOUT参数, 复杂类型传递的是指针, 因此什么都不需要做;
// 对于OUT参数, 复杂类型构造了新的ObjParam, 这里进行COPY;
ObLLVMValue into_address;
ObLLVMValue allocator;
ObLLVMValue src_datum;
ObLLVMValue dest_datum;
OZ (extract_objparam_from_context(
get_vars().at(CTX_IDX), param_desc.at(i).out_idx_, into_address));
if (pl_type.is_collection_type()) {
ObLLVMValue dest_collection;
OZ (extract_extend_from_objparam(into_address, pl_type, dest_collection));
OZ (extract_allocator_from_collection(dest_collection, allocator));
} else {
OZ (generate_null(ObIntType, allocator));
}
OZ (extract_obobj_ptr_from_objparam(into_address, dest_datum));
OZ (extract_obobj_ptr_from_objparam(p_arg, src_datum));
OZ (pl_type.generate_copy(*this,
*(s.get_namespace()),
allocator,
src_datum,
dest_datum,
s.get_block()->in_notfound(),
s.get_block()->in_warning(),
OB_INVALID_ID));
}
} else { //处理基础类型的出参
ObSEArray<ObLLVMValue, 4> args;
ObLLVMValue result_idx;
ObLLVMValue ret_err;
ObLLVMValue p_result_obj;
ObLLVMValue need_set;
OZ (get_helper().get_int64(param_desc.at(i).out_idx_, result_idx));
OZ (generate_new_objparam(p_result_obj));
OZ (get_helper().get_int8(true, need_set));
OZ (args.push_back(get_vars().at(CTX_IDX)));
OZ (args.push_back(p_arg));
OZ (args.push_back(result_idx));
OZ (args.push_back(p_result_obj));
OZ (args.push_back(need_set));
OZ (get_helper().create_call(ObString("spi_convert_objparam"),
get_spi_service().spi_convert_objparam_,
args,
ret_err));
OZ (check_success(
ret_err, s.get_stmt_id(), s.get_block()->in_notfound(), s.get_block()->in_warning()));
OZ (extract_datum_from_objparam(p_result_obj, pl_type.get_obj_type(), result));
OZ (get_helper().create_store(
result, get_vars().at(param_desc.at(i).out_idx_ + USER_ARG_OFFSET)),
param_desc.at(i).out_idx_);
}
} else { // 处理外部变量(Sys Var/User Var or PKG Basic Variables or Subprogram Basic Variables)
const ObRawExpr *expr = NULL;
CK (OB_NOT_NULL(expr = s.get_expr(param_desc.at(i).param_)));
if (OB_FAIL(ret)) {
} else if (expr->is_sys_func_expr()) {
OZ (generate_set_variable(param_desc.at(i).param_,
p_arg,
T_DEFAULT == expr->get_expr_type(),
s.get_stmt_id(),
s.get_block()->in_notfound(),
s.get_block()->in_warning()));
} else if (expr->is_obj_access_expr()) {
ObLLVMValue address;
ObLLVMValue src_datum;
ObLLVMValue dest_datum;
ObLLVMValue allocator;
ObPLDataType final_type;
const ObObjAccessRawExpr *obj_access = NULL;
uint64_t package_id = OB_INVALID_ID;
uint64_t var_idx = OB_INVALID_ID;
CK (OB_NOT_NULL(obj_access = static_cast<const ObObjAccessRawExpr *>(expr)));
if (OB_SUCC(ret)
&& ObObjAccessIdx::is_package_variable(obj_access->get_access_idxs())) {
OZ (ObObjAccessIdx::get_package_id(obj_access, package_id, &var_idx));
}
OZ (generate_null(ObIntType, allocator));
CK (OB_NOT_NULL(obj_access));
OZ (generate_expr(param_desc.at(i).param_, s, OB_INVALID_INDEX, address));
OZ (obj_access->get_final_type(final_type));
OZ (generate_check_not_null(s, final_type.get_not_null(), p_arg));
if (final_type.is_obj_type()) {
OZ (extract_datum_ptr_from_objparam(
p_arg, obj_access->get_result_type().get_type(), src_datum));
OZ (extract_extend_from_objparam(address, final_type, dest_datum));
OZ (final_type.generate_copy(*this,
*(s.get_namespace()),
allocator,
src_datum,
dest_datum,
s.get_block()->in_notfound(),
s.get_block()->in_warning(),
package_id));
} else {
OZ (extract_obobj_ptr_from_objparam(p_arg, src_datum));
OZ (extract_obobj_ptr_from_objparam(address, dest_datum));
OZ (final_type.generate_copy(*this,
*(s.get_namespace()),
allocator,
src_datum,
dest_datum,
s.get_block()->in_notfound(),
s.get_block()->in_warning(),
package_id));
}
if (OB_SUCC(ret) && package_id != OB_INVALID_ID && var_idx != OB_INVALID_ID) {
OZ (generate_update_package_changed_info(s, package_id, var_idx));
}
} else {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("Invalid expr", K(i), K(*expr), K(ret));
}
}
return ret;
}
int ObPLCodeGenerator::generate_out_params(
const ObPLStmt &s, const ObIArray<InOutParam> &param_desc, ObLLVMValue &params)
{
int ret = OB_SUCCESS;
ObSEArray<int64_t, 8> nocopy_params;
if (PL_CALL == s.get_type()) {
const ObPLCallStmt *call_stmt = static_cast<const ObPLCallStmt*>(&s);
OZ (nocopy_params.assign(call_stmt->get_nocopy_params()));
CK (nocopy_params.count() == 0 || nocopy_params.count() == param_desc.count());
}
// 先处理NoCopy的参数
for (int64_t i = 0; OB_SUCC(ret) && i < nocopy_params.count(); ++i) {
if (nocopy_params.at(i) != OB_INVALID_INDEX && param_desc.at(i).is_out()) {
OZ (generate_out_param(s, param_desc, params, i));
}
}
for (int64_t i = 0; OB_SUCC(ret) && i < param_desc.count(); ++i) {
// 处理输出参数的情况, NoCopy参数已经处理过了, 处理非NoCopy的参数
if (nocopy_params.count() > 0 && nocopy_params.at(i) != OB_INVALID_INDEX) {
// do nothing...
} else if (param_desc.at(i).is_out()) {
OZ (generate_out_param(s, param_desc, params, i));
}
}
return ret;
}
int ObPLCodeGenerator::generate(ObPLPackage &pl_package)
{
int ret = OB_SUCCESS;
OZ (prepare_external());
OZ (prepare_local_user_type());
OZ (prepare_expression(pl_package));
OZ (generate_obj_access_expr());
if (OB_SUCC(ret)) {
#ifndef NDEBUG
LOG_INFO("================Original LLVM Module================", K(debug_mode_));
helper_.dump_module();
#endif
OZ (helper_.verify_module(), pl_package);
OX (helper_.compile_module(!debug_mode_));
}
OZ (final_expression(pl_package));
return ret;
}
int ObPLCodeGenerator::generate(ObPLFunction &pl_func)
{
int ret = OB_SUCCESS;
ObPLFunctionAST &ast = static_cast<ObPLFunctionAST&>(ast_);
if (debug_mode_
|| !ast.get_is_all_sql_stmt()
|| !ast_.get_obj_access_exprs().empty()) {
OZ (generate_normal(pl_func));
} else {
OZ (generate_simple(pl_func));
}
LOG_TRACE("generate pl function",
K(debug_mode_), K(ast.get_is_all_sql_stmt()), K(ast.get_obj_access_exprs().empty()));
return ret;
}
int ObPLCodeGenerator::generate_simple(ObPLFunction &pl_func)
{
int ret = OB_SUCCESS;
ObPLFunctionAST &ast = static_cast<ObPLFunctionAST&>(ast_);
common::ObFixedArray<ObPLSqlInfo, common::ObIAllocator> &sql_infos = pl_func.get_sql_infos();
CK (!debug_mode_);
CK (ast.get_is_all_sql_stmt());
OZ (prepare_expression(pl_func));
OZ (final_expression(pl_func));
OZ (pl_func.set_variables(get_ast().get_symbol_table()));
OZ (pl_func.get_dependency_table().assign(get_ast().get_dependency_table()));
OX (pl_func.add_members(get_ast().get_flag()));
OX (pl_func.set_pipelined(get_ast().get_pipelined()));
OX (pl_func.set_action((uint64_t)(&ObPL::simple_execute)));
OX (pl_func.set_can_cached(get_ast().get_can_cached()));
OX (pl_func.set_is_all_sql_stmt(get_ast().get_is_all_sql_stmt()));
OX (sql_infos.set_capacity(static_cast<uint32_t>(ast.get_sql_stmts().count())));
for (int64_t i = 0; OB_SUCC(ret) && i < ast.get_sql_stmts().count(); ++i) {
const ObPLSqlStmt *sql_stmt = ast.get_sql_stmts().at(i);
ObPLSqlInfo sql_info(pl_func.get_allocator());
CK (OB_NOT_NULL(sql_stmt));
OZ (sql_info.generate(*sql_stmt, pl_func.get_expressions()));
OZ (sql_infos.push_back(sql_info));
}
return ret;
}
int ObPLCodeGenerator::generate_normal(ObPLFunction &pl_func)
{
int ret = OB_SUCCESS;
// 初始化符号表
for (int64_t i = 0;
OB_SUCC(ret) && i < get_ast().get_symbol_table().get_count() + USER_ARG_OFFSET + 1;
++i) {
ObLLVMValue dummy_value;
if (OB_FAIL(vars_.push_back(dummy_value))) {
LOG_WARN("failed to push back dummy value", K(ret), K(i));
}
}
if (OB_FAIL(ret)) {
} else if (OB_FAIL(generate_prototype())) {
LOG_WARN("failed to generate a pointer", K(ret));
} else if (OB_FAIL(func_.set_personality(get_eh_service().eh_personality_))) {
LOG_WARN("failed to set_personality", K(ret));
} else if (OB_FAIL(helper_.create_block(ObString("entry"), func_, entry_))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(helper_.create_block(ObString("exit"), func_, exit_))) {
LOG_WARN("failed to create block", K(ret));
} else if (OB_FAIL(set_current(entry_))) {
LOG_WARN("failed to set current", K(ret));
} else if (OB_FAIL(generate_di_prototype())) {
LOG_WARN("failed to generate di prototype", K(ret));
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
ObPLCodeGenerateVisitor visitor(*this);
if (OB_ISNULL(get_ast().get_body())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("pl body is NULL", K(ret));
} else if (OB_FAIL(helper_.set_insert_point(entry_))) {
LOG_WARN("failed set_insert_point", K(ret));
} else if (OB_FAIL(set_debug_location(*get_ast().get_body()))) {
LOG_WARN("failed to set debug location", K(ret));
} else if (OB_FAIL(init_argument())) {
LOG_WARN("failed to init augument", K(ret));
} else if (OB_FAIL(prepare_external())) {
LOG_WARN("failed to prepare external", K(ret));
} else if (lib::is_oracle_mode() && OB_FAIL(prepare_local_user_type())) {
LOG_WARN("failed to prepare local user type", K(ret));
} else if (OB_FAIL(prepare_expression(pl_func))) {
LOG_WARN("failed to prepare expression", K(ret));
} else if (OB_FAIL(prepare_subprogram(pl_func))) {
LOG_WARN("failed to prepare subprogram", K(ret));
} else if (OB_FAIL(SMART_CALL(visitor.generate(*get_ast().get_body())))) {
LOG_WARN("failed to generate a pl body", K(ret));
}
}
if (OB_SUCC(ret)) {
if (get_ast().get_ret_type().is_obj_type()
&& (OB_ISNULL(get_ast().get_ret_type().get_data_type())
|| get_ast().get_ret_type().get_data_type()->get_meta_type().is_invalid())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("return type is invalid", K(func_), K(ret));
} else if (!current_.is_terminated()) { //如果当前block没有终止符,强制跳转
if (OB_FAIL(finish_current(exit_))) {
LOG_WARN("failed to finish_current", K(ret));
}
} else { /*do nothing*/ }
if (OB_SUCC(ret)) {
ObLLVMValue ret_value;
if (OB_FAIL(helper_.set_insert_point(exit_))) {
LOG_WARN("failed to set_insert_point", K(ret));
} else if (OB_FAIL(helper_.create_load(ObString("load_ret"), vars_.at(RET_IDX), ret_value))) {
LOG_WARN("failed to create_load", K(ret));
} else if (OB_FAIL(helper_.create_ret(ret_value))) {
LOG_WARN("failed to create_ret", K(ret));
} else { /*do nothing*/ }
}
}
if (OB_SUCC(ret)) {
if (OB_FAIL(generate_obj_access_expr())) {
LOG_WARN("generate obj access expr failed", K(ret));
} else if (debug_mode_ && OB_FAIL(di_helper_.finalize())) {
LOG_WARN("failed to finalize", K(ret));
}
}
if (OB_SUCC(ret)) {
#ifndef NDEBUG
LOG_INFO("================Original================", K(pl_func), K(debug_mode_));
helper_.dump_module();
#endif
OZ (helper_.verify_module(), pl_func);
OX (helper_.compile_module(!debug_mode_));
}
if (OB_SUCC(ret)) {
if (OB_FAIL(final_expression(pl_func))) {
LOG_WARN("generate obj access expr failed", K(ret));
} else if (OB_FAIL(pl_func.set_variables(get_ast().get_symbol_table()))) {
LOG_WARN("failed to set variables", K(get_ast().get_symbol_table()), K(ret));
} else if (OB_FAIL(pl_func.get_dependency_table().assign(get_ast().get_dependency_table()))) {
LOG_WARN("failed to set ref objects", K(get_ast().get_dependency_table()), K(ret));
} else if (OB_FAIL(pl_func.set_types(get_ast().get_user_type_table()))) {
LOG_WARN("failed to set types", K(ret));
} else {
pl_func.add_members(get_ast().get_flag());
pl_func.set_pipelined(get_ast().get_pipelined());
pl_func.set_action(helper_.get_function_address(get_ast().get_name()));
pl_func.set_can_cached(get_ast().get_can_cached());
pl_func.set_is_all_sql_stmt(get_ast().get_is_all_sql_stmt());
}
}
if (debug_mode_) {
OX (helper_.dump_debuginfo());
OZ (pl_func.set_variables_debuginfo(get_ast().get_symbol_debuginfo_table()));
OZ (pl_func.set_name_debuginfo(get_ast()));
}
OX (helper_.final());
debug_mode_ = false;
return ret;
}
int ObPLCodeGenerator::extract_meta_ptr_from_obj(ObLLVMValue &p_obj, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObSEArray<int64_t, 2> indices;
if (OB_FAIL(indices.push_back(0)) || OB_FAIL(indices.push_back(0))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_meta_pointer"), p_obj,
indices, result))) {
LOG_WARN("failed to create gep", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_type_ptr_from_obj(ObLLVMValue &p_obj, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_meta;
OZ (extract_meta_ptr_from_obj(p_obj, p_meta));
OZ (helper_.create_gep(ObString("extract_type_pointer"), p_meta, 0, result));
return ret;
}
int ObPLCodeGenerator::extract_meta_ptr_from_objparam(ObLLVMValue &p_objparam, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObSEArray<int64_t, 3> indices;
if (OB_FAIL(indices.push_back(0)) || OB_FAIL(indices.push_back(0)) || OB_FAIL(indices.push_back(0))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_meta_pointer"), p_objparam, indices, result))) {
LOG_WARN("failed to create gep", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_accuracy_ptr_from_objparam(ObLLVMValue &p_objparam, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObSEArray<int64_t, 2> indices;
if (OB_FAIL(indices.push_back(0)) || OB_FAIL(indices.push_back(1))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_meta_pointer"), p_objparam, indices, result))) {
LOG_WARN("failed to create gep", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_param_flag_ptr_from_objparam(ObLLVMValue &p_objparam, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObSEArray<int64_t, 2> indices;
if (OB_FAIL(indices.push_back(0)) || OB_FAIL(indices.push_back(3))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_meta_pointer"),
p_objparam, indices, result))) {
LOG_WARN("failed to create gep", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_raw_text_pos_ptr_from_objparam(ObLLVMValue &p_objparam,
ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObSEArray<int64_t, 2> indices;
if (OB_FAIL(indices.push_back(0)) || OB_FAIL(indices.push_back(4))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_meta_pointer"),
p_objparam, indices, result))) {
LOG_WARN("failed to create gep", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_raw_text_len_ptr_from_objparam(ObLLVMValue &p_objparam,
ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObSEArray<int64_t, 2> indices;
if (OB_FAIL(indices.push_back(0)) || OB_FAIL(indices.push_back(5))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_meta_pointer"), p_objparam, indices, result))) {
LOG_WARN("failed to create gep", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_type_ptr_from_objparam(ObLLVMValue &p_objparam, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_meta;
OZ (extract_meta_ptr_from_objparam(p_objparam, p_meta));
OZ (helper_.create_gep(ObString("extract_type_pointer"), p_meta, 0, result));
return ret;
}
int ObPLCodeGenerator::extract_cslevel_ptr_from_objparam(ObLLVMValue &p_objparam, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_meta;
if (OB_FAIL(extract_meta_ptr_from_objparam(p_objparam, p_meta))) {
LOG_WARN("faled to extract_meta_ptr_from_objparam", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_scale_pointer"), p_meta, 1, result))) {
LOG_WARN("failed to create gep", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_cstype_ptr_from_objparam(ObLLVMValue &p_objparam, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_meta;
if (OB_FAIL(extract_meta_ptr_from_objparam(p_objparam, p_meta))) {
LOG_WARN("faled to extract_meta_ptr_from_objparam", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_scale_pointer"), p_meta, 2, result))) {
LOG_WARN("failed to create gep", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_scale_ptr_from_objparam(ObLLVMValue &p_objparam, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_meta;
if (OB_FAIL(extract_meta_ptr_from_objparam(p_objparam, p_meta))) {
LOG_WARN("faled to extract_meta_ptr_from_objparam", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_scale_pointer"), p_meta, 3, result))) {
LOG_WARN("failed to create gep", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_flag_ptr_from_objparam(ObLLVMValue &p_objparam, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_meta;
if (OB_FAIL(extract_param_flag_ptr_from_objparam(p_objparam, p_meta))) {
LOG_WARN("faled to extract_meta_ptr_from_objparam", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_scale_pointer"), p_meta, 1, result))) {
LOG_WARN("failed to create gep", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_obobj_ptr_from_objparam(ObLLVMValue &p_objparam, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
OZ (helper_.create_gep(ObString("extract_obj_pointer"), p_objparam, 0, result));
return ret;
}
int ObPLCodeGenerator::extract_obobj_from_objparam(ObLLVMValue &p_objparam, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_obj;
if (OB_FAIL(extract_obobj_ptr_from_objparam(p_objparam, p_obj))) {
LOG_WARN("failed to create gep", K(ret));
} else if (OB_FAIL(helper_.create_load(ObString("load_value"), p_obj, result))) {
LOG_WARN("failed to create load", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_datum_ptr_from_objparam(ObLLVMValue &p_objparam, ObObjType type, ObLLVMValue &result)
{
UNUSED(type);
int ret = OB_SUCCESS;
ObLLVMValue datum_addr;
ObLLVMType datum_type;
ObLLVMType datum_pointer_type;
/* if (ob_is_string_tc(type)
|| ob_is_number_tc(type)
|| ob_is_text_tc(type)
|| ob_is_otimestampe_tc(type)
|| ob_is_raw_tc(type)) {*/
ObSEArray<int64_t, 3> indices;
if (OB_FAIL(indices.push_back(0)) || OB_FAIL(indices.push_back(0))) {
LOG_WARN("push_back error", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_int64_pointer"), p_objparam, indices, datum_addr))) {
LOG_WARN("failed to create gep", K(ret));
} else if (OB_FAIL(adt_service_.get_obj(datum_type))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(datum_type.get_pointer_to(datum_pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.create_bit_cast(ObString("cast_addr_to_datum"), datum_addr, datum_pointer_type, result))) {
LOG_WARN("failed to create bit cast", K(ret));
} else { /*do nothing*/ }
/* } else {
if (OB_FAIL(extract_value_ptr_from_objparam(p_objparam, type, result))) {
LOG_WARN("failed to extract value from objparam", K(ret));
}
}*/
return ret;
}
int ObPLCodeGenerator::cast_to_int64(ObLLVMValue &p_value)
{
int ret = OB_SUCCESS;
ObLLVMValue p_value_int64, value_int64, value;
ObLLVMType llvm_type, pointer_type;
OZ (helper_.get_llvm_type(ObIntType, llvm_type));
OZ (llvm_type.get_pointer_to(pointer_type));
OZ (helper_.create_load(ObString("load value"), p_value, value));
OZ (helper_.create_sext(ObString("sext to int64"), value, llvm_type, value_int64));
// #ifndef NDEBUG
// OZ (generate_debug(ObString("after sext value to int64"), value_int64));
// #endif
OZ (helper_.create_bit_cast(ObString("bitcast"), p_value, pointer_type, p_value_int64));
OZ (helper_.create_store(value_int64, p_value_int64));
return ret;
}
int ObPLCodeGenerator::extract_value_ptr_from_obj(ObLLVMValue &p_obj, ObObjType type, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_obj_int64;
ObLLVMType llvm_type;
ObLLVMType pointer_type;
if (OB_FAIL(helper_.create_gep(ObString("extract_int64_pointer"), p_obj, 2, p_obj_int64))) {
LOG_WARN("failed to create gep", K(ret));
} else {
switch (type) {
case ObNullType: {
ObLLVMValue false_value;
OZ (helper_.get_llvm_type(ObTinyIntType, llvm_type));
OZ (llvm_type.get_pointer_to(pointer_type));
OZ (helper_.create_bit_cast(ObString("cast_int64_to_int8"), p_obj_int64, pointer_type, result));
OZ (helper_.get_int8(false, false_value));
OZ (helper_.create_store(false_value, result));
}
break;
case ObTinyIntType:
case ObUTinyIntType: {
if (OB_FAIL(helper_.get_llvm_type(ObTinyIntType, llvm_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(llvm_type.get_pointer_to(pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.create_bit_cast(ObString("cast_int64_to_int8"), p_obj_int64, pointer_type, result))) {
LOG_WARN("failed to create bit cast", K(ret));
} else { /*do nothing*/ }
}
break;
case ObSmallIntType:
case ObUSmallIntType: {
if (OB_FAIL(helper_.get_llvm_type(ObSmallIntType, llvm_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(llvm_type.get_pointer_to(pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.create_bit_cast(ObString("cast_int64_to_int16"), p_obj_int64, pointer_type, result))) {
LOG_WARN("failed to create bit cast", K(ret));
} else { /*do nothing*/ }
}
break;
case ObMediumIntType:
case ObInt32Type:
case ObUMediumIntType:
case ObUInt32Type: {
if (OB_FAIL(helper_.get_llvm_type(ObInt32Type, llvm_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(llvm_type.get_pointer_to(pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.create_bit_cast(ObString("cast_int64_to_int32"), p_obj_int64, pointer_type, result))) {
LOG_WARN("failed to create bit cast", K(ret));
} else { /*do nothing*/ }
}
break;
case ObIntType:
case ObUInt64Type:
case ObDateTimeType:
case ObTimestampType:
case ObDateType:
case ObTimeType:
case ObYearType:
case ObBitType:
case ObEnumType:
case ObSetType: {
result = p_obj_int64;
}
break;
case ObFloatType:
case ObUFloatType: {
if (OB_FAIL(helper_.get_llvm_type(ObFloatType, llvm_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(llvm_type.get_pointer_to(pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.create_bit_cast(ObString("cast_int64_to_float"), p_obj_int64, pointer_type, result))) {
LOG_WARN("failed to create bit cast", K(ret));
} else { /*do nothing*/ }
}
break;
case ObDoubleType:
case ObUDoubleType: {
if (OB_FAIL(helper_.get_llvm_type(ObDoubleType, llvm_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(llvm_type.get_pointer_to(pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.create_bit_cast(ObString("cast_int64_to_double"), p_obj_int64, pointer_type, result))) {
LOG_WARN("failed to create bit cast", K(ret));
} else { /*do nothing*/ }
}
break;
case ObNumberType:
case ObUNumberType:
case ObNumberFloatType: {
if (OB_FAIL(helper_.get_llvm_type(ObNumberType, llvm_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(llvm_type.get_pointer_to(pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.create_bit_cast(ObString("cast_int64_to_number"), p_obj_int64, pointer_type, result))) {
LOG_WARN("failed to create_bit_cast", K(ret));
} else { /*do nothing*/ }
}
break;
case ObNVarchar2Type:
case ObNCharType:
case ObVarcharType:
case ObCharType:
case ObHexStringType:
case ObURowIDType: {
if (OB_FAIL(helper_.get_llvm_type(ObCharType, llvm_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(llvm_type.get_pointer_to(pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.create_bit_cast(ObString("cast_int64_to_char"), p_obj_int64, pointer_type, result))) {
LOG_WARN("failed to create_bit_cast", K(ret));
} else { /*do nothing*/ }
}
break;
case ObExtendType: {
if (OB_FAIL(helper_.get_llvm_type(ObExtendType, llvm_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(llvm_type.get_pointer_to(pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.create_bit_cast(ObString("cast_int64_to_extend"), p_obj_int64, pointer_type, result))) {
LOG_WARN("failed to create_bit_cast", K(ret));
} else { /*do nothing*/ }
}
break;
case ObUnknownType: {
if (OB_FAIL(helper_.get_llvm_type(ObUnknownType, llvm_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(llvm_type.get_pointer_to(pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.create_bit_cast(ObString("cast_int64_to_unknown"), p_obj_int64, pointer_type, result))) {
LOG_WARN("failed to create_bit_cast", K(ret));
} else { /*do nothing*/ }
}
break;
case ObTimestampTZType:
case ObTimestampLTZType:
case ObTimestampNanoType://TODO: not support now, @yanhua
case ObRawType: //TODO: not support now, @xiaofeng
case ObIntervalYMType: //TODO: not support now, @jim.wjh
case ObIntervalDSType: //TODO: not support now, @jim.wjh
default: {
if (OB_FAIL(generate_null_pointer(ObIntType, result))) {
LOG_WARN("failed to get pointer to", K(ret));
}
}
break;
}
}
return ret;
}
int ObPLCodeGenerator::extract_value_from_obj(jit::ObLLVMValue &p_obj,
ObObjType type,
jit::ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue result_ptr;
OZ (extract_value_ptr_from_obj(p_obj, type, result_ptr));
OZ (helper_.create_load(ObString("load_value"), result_ptr, result));
return ret;
}
int ObPLCodeGenerator::extract_value_ptr_from_objparam(ObLLVMValue &p_objparam, ObObjType type, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_obj;
if (OB_FAIL(helper_.create_gep(ObString("extract_obj_pointer"), p_objparam, 0, p_obj))) {
LOG_WARN("failed to create gep", K(ret));
} else if (OB_FAIL(extract_value_ptr_from_obj(p_obj, type, result))) {
LOG_WARN("failed to extract_value_ptr_from_obj", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_datum_from_objparam(ObLLVMValue &p_objparam, ObObjType type, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue result_ptr;
if (OB_FAIL(extract_datum_ptr_from_objparam(p_objparam, type, result_ptr))) {
LOG_WARN("failed to create gep", K(ret));
} else if (OB_FAIL(helper_.create_load(ObString("load_datum"), result_ptr, result))) {
LOG_WARN("failed to create load", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_value_from_objparam(ObLLVMValue &p_objparam, ObObjType type, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue result_ptr;
if (OB_FAIL(extract_value_ptr_from_objparam(p_objparam, type, result_ptr))) {
LOG_WARN("failed to create gep", K(ret));
} else if (OB_FAIL(helper_.create_load(ObString("load_value"), result_ptr, result))) {
LOG_WARN("failed to create load", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_extend_from_objparam(ObLLVMValue &p_objparam, const ObPLDataType &type, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue extend;
ObLLVMType llvm_type;
ObLLVMType addr_type;
if (OB_FAIL(extract_value_from_objparam(p_objparam, ObExtendType, extend))) {
LOG_WARN("failed to extract_value_from_objparam", K(ret));
} else if (OB_FAIL(get_llvm_type(type, llvm_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(llvm_type.get_pointer_to(addr_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.create_int_to_ptr(ObString("cast_extend_to_ptr"), extend, addr_type, result))) {
LOG_WARN("failed to create_bit_cast", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_extend_from_obj(ObLLVMValue &p_obj,
const ObPLDataType &type,
ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue extend;
ObLLVMType llvm_type;
ObLLVMType addr_type;
if (OB_FAIL(extract_value_from_obj(p_obj, ObExtendType, extend))) {
LOG_WARN("failed to extract_value_from_objparam", K(ret));
} else if (OB_FAIL(get_llvm_type(type, llvm_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(llvm_type.get_pointer_to(addr_type))) {
LOG_WARN("failed to get_llvm_type", K(ret));
} else if (OB_FAIL(helper_.create_int_to_ptr(ObString("cast_extend_to_ptr"),
extend,
addr_type,
result))) {
LOG_WARN("failed to create_bit_cast", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_obj_ptr_from_result(jit::ObLLVMValue &p_objparam,
jit::ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue extend;
ObLLVMType obj_type;
ObLLVMType p_obj_type;
OZ (extract_value_from_obj(p_objparam, ObExtendType, extend));
OZ (adt_service_.get_obj(obj_type));
OZ (obj_type.get_pointer_to(p_obj_type));
OZ (helper_.create_int_to_ptr(ObString("cast_extend_to_obj_ptr"), extend, p_obj_type, result));
return ret;
}
int ObPLCodeGenerator::extract_objparam_from_store(ObLLVMValue &p_param_store, const int64_t idx, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
const static int64_t BlocksIDX = 3;
const static int64_t SEArrayDataIDX = 1;
ObLLVMValue param_store;
ObLLVMValue blocks;
ObLLVMValue p_blocks_carray;
ObLLVMValue p_obj_block;
ObLLVMValue obj_block;
ObLLVMValue p_obj_param;
ObLLVMValue result_tmp_p;
ObLLVMType obj_param_type;
ObLLVMType p_obj_param_type;
ObArray<int64_t> extract_obj_param_idxes;
ObArray<int64_t> extract_block_addr_idxes;
if (OB_FAIL(helper_.create_load(ObString("load_param_store"),
p_param_store,
param_store))) {
LOG_WARN("failed to create load", K(ret));
} else if (OB_FAIL(helper_.create_extract_value(ObString("extract_blocks"),
param_store, BlocksIDX,
blocks))) {
LOG_WARN("failed to create extract value", K(ret));
} else if (OB_FAIL(helper_.create_extract_value(ObString("extract_blocks_pointer"),
blocks, SEArrayDataIDX,
p_blocks_carray))) {
LOG_WARN("failed to create extract value", K(ret));
} else {
const int64_t block_obj_num = ParamStore::BLOCK_CAPACITY;
const int64_t block_idx = idx / block_obj_num;
const int64_t obj_in_block_idx = idx % block_obj_num;
if (OB_FAIL(extract_block_addr_idxes.push_back(block_idx))
|| OB_FAIL(extract_block_addr_idxes.push_back(0))) {
LOG_WARN("failed to push back element", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_block_pointer"),
p_blocks_carray,
extract_block_addr_idxes, p_obj_block))) {
LOG_WARN("failed to create gep", K(ret));
} else if (OB_FAIL(helper_.create_load(ObString("load_obj_block"),
p_obj_block, obj_block))) {
LOG_WARN("failed to create_load", K(ret));
} else if (OB_FAIL(adt_service_.get_objparam(obj_param_type))) {
LOG_WARN("failed to get_objparam", K(ret));
} else if (OB_FAIL(obj_param_type.get_pointer_to(p_obj_param_type))) {
LOG_WARN("failed to get_pointer_to", K(ret));
} else if (OB_FAIL(helper_.create_int_to_ptr(ObString("cast_block_to_objparam_p"),
obj_block,
p_obj_param_type, p_obj_param))) {
LOG_WARN("failed to create_int_to_ptr cast", K(ret));
} else if (OB_FAIL(extract_obj_param_idxes.push_back(obj_in_block_idx))
|| OB_FAIL(extract_obj_param_idxes.push_back(0))) {
LOG_WARN("failed to push back element", K(ret));
} else if (OB_FAIL(helper_.create_gep(ObString("extract_objparam"),
p_obj_param,
extract_obj_param_idxes,
result_tmp_p))) {
LOG_WARN("failed to create gep", K(ret));
} else if (OB_FAIL(helper_.create_bit_cast(ObString("bitcast"),
result_tmp_p, p_obj_param_type, result))) {
LOG_WARN("failed to cast obj pointer to objparam pointer", K(ret));
}
}
return ret;
}
#define DEFINE_EXTRACT_CONTEXT_ELEM(item, idx) \
int ObPLCodeGenerator::extract_##item##_from_context(jit::ObLLVMValue &p_pl_exex_ctx, jit::ObLLVMValue &result) \
{ \
int ret = OB_SUCCESS; \
ObLLVMValue pl_exex_ctx; \
OZ (helper_.create_load(ObString("load_pl_exex_ctx"), p_pl_exex_ctx, pl_exex_ctx)); \
OZ (helper_.create_extract_value(ObString("extract_"#item), pl_exex_ctx, idx, result)); \
return ret; \
}
DEFINE_EXTRACT_CONTEXT_ELEM(allocator, IDX_PLEXECCTX_ALLOCATOR)
DEFINE_EXTRACT_CONTEXT_ELEM(param_store, IDX_PLEXECCTX_PARAMS)
DEFINE_EXTRACT_CONTEXT_ELEM(result, IDX_PLEXECCTX_RESULT)
DEFINE_EXTRACT_CONTEXT_ELEM(status, IDX_PLEXECCTX_STATUS)
DEFINE_EXTRACT_CONTEXT_ELEM(pl_ctx, IDX_PLEXECCTX_PL_CTX)
DEFINE_EXTRACT_CONTEXT_ELEM(pl_function, IDX_PLEXECCTX_FUNC)
int ObPLCodeGenerator::extract_objparam_from_context(ObLLVMValue &p_pl_exex_ctx, int64_t idx, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_param_store;
if (OB_FAIL(extract_param_store_from_context(p_pl_exex_ctx, p_param_store))) {
LOG_WARN("failed to extract_param_store_from_context", K(ret));
} else if (OB_FAIL(extract_objparam_from_store(p_param_store, idx, result))) {
LOG_WARN("failed to extract_objparam_from_store", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_value_from_context(ObLLVMValue &p_pl_exex_ctx, int64_t idx, ObObjType type, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_objparam;
if (OB_FAIL(extract_objparam_from_context(p_pl_exex_ctx, idx, p_objparam))) {
LOG_WARN("failed to extract_param_store_from_context", K(ret));
} else if (OB_FAIL(extract_value_from_objparam(p_objparam, type, result))) {
LOG_WARN("failed to extract_objparam_from_store", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_datum_from_context(ObLLVMValue &p_pl_exex_ctx, int64_t idx, ObObjType type, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_objparam;
if (OB_FAIL(extract_objparam_from_context(p_pl_exex_ctx, idx, p_objparam))) {
LOG_WARN("failed to extract_param_store_from_context", K(ret));
} else if (OB_FAIL(extract_datum_from_objparam(p_objparam, type, result))) {
LOG_WARN("failed to extract_objparam_from_store", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_arg_from_argv(ObLLVMValue &p_argv, int64_t idx, ObLLVMValue &result)
{
return helper_.create_gep(ObString("extract_arg"), p_argv, idx, result);
}
int ObPLCodeGenerator::extract_objparam_from_argv(jit::ObLLVMValue &p_argv,
const int64_t idx,
jit::ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue pp_arg;
ObLLVMValue p_arg;
ObLLVMType objparam;
ObLLVMType pointer_type;
if (OB_FAIL(extract_arg_from_argv(p_argv, idx, pp_arg))) {
LOG_WARN("failed to create load", K(ret));
} else if (OB_FAIL(helper_.create_load(ObString("load_arg"), pp_arg, p_arg))) {
LOG_WARN("failed to create load", K(ret));
} else if (OB_FAIL(adt_service_.get_objparam(objparam))) {
LOG_WARN("failed to get argv type", K(ret));
} else if (OB_FAIL(objparam.get_pointer_to(pointer_type))) {
LOG_WARN("failed to get pointer to", K(ret));
} else if (OB_FAIL(helper_.create_int_to_ptr(ObString("cast_arg_to_pointer"), p_arg,
pointer_type, result))) {
LOG_WARN("failed to create bit cast", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_datum_from_argv(ObLLVMValue &p_argv,
int64_t idx,
ObObjType type,
ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_objparam;
if (OB_FAIL(extract_objparam_from_argv(p_argv, idx, p_objparam))) {
LOG_WARN("failed to create load", K(ret));
} else if (OB_FAIL(extract_datum_from_objparam(p_objparam, type, result))) {
LOG_WARN("push_back error", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_value_from_argv(ObLLVMValue &p_argv, const int64_t idx, ObObjType type, ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue pp_arg;
ObLLVMValue p_arg;
ObLLVMValue p_objparam;
ObLLVMType objparam;
ObLLVMType pointer_type;
if (OB_FAIL(extract_objparam_from_argv(p_argv, idx, p_objparam))) {
LOG_WARN("failed to create load", K(ret));
} else if (OB_FAIL(extract_value_from_objparam(p_objparam, type, result))) {
LOG_WARN("push_back error", K(ret));
} else { /*do nothing*/ }
return ret;
}
int ObPLCodeGenerator::extract_notnull_from_record(jit::ObLLVMValue &p_record, int64_t idx,
jit::ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_result;
OZ (extract_notnull_ptr_from_record(p_record, idx, p_result));
OZ (helper_.create_load(ObString("load_record_elem"), p_result, result));
return ret;
}
int ObPLCodeGenerator::extract_notnull_ptr_from_record(jit::ObLLVMValue &p_record,
int64_t idx,
jit::ObLLVMValue &result)
{
int ret = OB_SUCCESS;
return helper_.create_gep(ObString("extract_record_elem"),
p_record,
RECORD_META_OFFSET + idx,
result);
}
int ObPLCodeGenerator::extract_meta_from_record(jit::ObLLVMValue &p_record,
int64_t member_cnt,
int64_t idx,
jit::ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_result;
OZ (extract_meta_ptr_from_record(p_record, member_cnt, idx, p_result));
OZ (helper_.create_load(ObString("load_record_elem"), p_result, result));
return ret;
}
int ObPLCodeGenerator::extract_meta_ptr_from_record(jit::ObLLVMValue &p_record,
int64_t member_cnt,
int64_t idx,
jit::ObLLVMValue &result)
{
int ret = OB_SUCCESS;
return helper_.create_gep(ObString("extract_record_elem"),
p_record,
RECORD_META_OFFSET + member_cnt + idx,
result);
}
int ObPLCodeGenerator::extract_element_from_record(jit::ObLLVMValue &p_record,
int64_t member_cnt,
int64_t idx,
jit::ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_result;
OZ (extract_element_ptr_from_record(p_record, member_cnt, idx, p_result));
OZ (helper_.create_load(ObString("load_record_elem"), p_result, result));
return ret;
}
int ObPLCodeGenerator::extract_element_ptr_from_record(jit::ObLLVMValue &p_record,
int64_t member_cnt,
int64_t idx,
jit::ObLLVMValue &result)
{
int ret = OB_SUCCESS;
return helper_.create_gep(ObString("extract_record_elem"),
p_record,
RECORD_META_OFFSET + member_cnt + member_cnt + idx,
result);
}
#define DEFINE_EXTRACT_PTR_FROM_STRUCT(item, s, idx) \
int ObPLCodeGenerator::extract_##item##_ptr_from_##s(jit::ObLLVMValue &p_struct, jit::ObLLVMValue &result) \
{ \
return helper_.create_gep(ObString("extract_"#item), p_struct, idx, result); \
}
#define DEFINE_EXTRACT_VALUE_FROM_STRUCT(item, s) \
int ObPLCodeGenerator::extract_##item##_from_##s(jit::ObLLVMValue &p_struct, jit::ObLLVMValue &result) \
{ \
int ret = OB_SUCCESS; \
ObLLVMValue p_result; \
OZ (extract_##item##_ptr_from_##s(p_struct, p_result)); \
OZ (helper_.create_load(ObString("load_"#item), p_result, result)); \
return ret; \
}
DEFINE_EXTRACT_PTR_FROM_STRUCT(type, condition_value, IDX_CONDITION_TYPE)
DEFINE_EXTRACT_PTR_FROM_STRUCT(code, condition_value, IDX_CONDITION_CODE)
DEFINE_EXTRACT_PTR_FROM_STRUCT(name, condition_value, IDX_CONDITION_STATE)
DEFINE_EXTRACT_PTR_FROM_STRUCT(len, condition_value, IDX_CONDITION_LEN)
DEFINE_EXTRACT_PTR_FROM_STRUCT(stmt, condition_value, IDX_CONDITION_STMT)
DEFINE_EXTRACT_PTR_FROM_STRUCT(signal, condition_value, IDX_CONDITION_SIGNAL)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(type, condition_value)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(code, condition_value)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(name, condition_value)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(len, condition_value)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(stmt, condition_value)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(signal, condition_value)
DEFINE_EXTRACT_PTR_FROM_STRUCT(type, collection, IDX_COLLECTION_TYPE)
DEFINE_EXTRACT_PTR_FROM_STRUCT(id, collection, IDX_COLLECTION_ID)
DEFINE_EXTRACT_PTR_FROM_STRUCT(isnull, collection, IDX_COLLECTION_ISNULL)
DEFINE_EXTRACT_PTR_FROM_STRUCT(allocator, collection, IDX_COLLECTION_ALLOCATOR)
DEFINE_EXTRACT_PTR_FROM_STRUCT(element, collection, IDX_COLLECTION_ELEMENT)
DEFINE_EXTRACT_PTR_FROM_STRUCT(count, collection, IDX_COLLECTION_COUNT)
DEFINE_EXTRACT_PTR_FROM_STRUCT(first, collection, IDX_COLLECTION_FIRST)
DEFINE_EXTRACT_PTR_FROM_STRUCT(last, collection, IDX_COLLECTION_LAST)
DEFINE_EXTRACT_PTR_FROM_STRUCT(data, collection, IDX_COLLECTION_DATA)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(type, collection)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(id, collection)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(isnull, collection)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(allocator, collection)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(element, collection)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(count, collection)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(first, collection)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(last, collection)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(data, collection)
DEFINE_EXTRACT_PTR_FROM_STRUCT(capacity, varray, IDX_VARRAY_CAPACITY)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(capacity, varray)
DEFINE_EXTRACT_PTR_FROM_STRUCT(type, record, IDX_RECORD_TYPE)
DEFINE_EXTRACT_PTR_FROM_STRUCT(id, record, IDX_RECORD_ID)
DEFINE_EXTRACT_PTR_FROM_STRUCT(isnull, record, IDX_RECORD_ISNULL)
DEFINE_EXTRACT_PTR_FROM_STRUCT(count, record, IDX_RECORD_COUNT)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(type, record)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(id, record)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(isnull, record)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(count, record)
DEFINE_EXTRACT_PTR_FROM_STRUCT(type, elemdesc, IDX_ELEMDESC_TYPE)
DEFINE_EXTRACT_PTR_FROM_STRUCT(notnull, elemdesc, IDX_ELEMDESC_NOTNULL)
DEFINE_EXTRACT_PTR_FROM_STRUCT(field_count, elemdesc, IDX_ELEMDESC_FIELD_COUNT)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(type, elemdesc)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(notnull, elemdesc)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(field_count, elemdesc)
DEFINE_EXTRACT_VALUE_FROM_STRUCT(notnull, collection)
int ObPLCodeGenerator::extract_notnull_ptr_from_collection(jit::ObLLVMValue &p_collection,
jit::ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_element;
OZ (extract_element_ptr_from_collection(p_collection, p_element));
OZ (extract_notnull_ptr_from_elemdesc(p_element, result));
return ret;
}
DEFINE_EXTRACT_VALUE_FROM_STRUCT(field_count, collection)
int ObPLCodeGenerator::extract_field_count_ptr_from_collection(jit::ObLLVMValue &p_collection,
jit::ObLLVMValue &result)
{
int ret = OB_SUCCESS;
ObLLVMValue p_element;
OZ (extract_element_ptr_from_collection(p_collection, p_element));
OZ (extract_field_count_ptr_from_elemdesc(p_element, result));
return ret;
}
int ObPLCodeGenerator::generate_handle_ref_cursor(const ObPLCursor *cursor, const ObPLStmt &s,
bool is_notfound, bool in_warning)
{
int ret = OB_SUCCESS;
CK (OB_NOT_NULL(cursor));
// 以下一些情况不要关闭cursor:
// function (cur out sys_refcursor), out类型参数
// dup cursor 没有被init,所以不需要关闭,见generate_declare_cursor
// 直接使用subprog 外部的cursor(不是ref cursor),也不要关闭,例如
/*
* DECLARE
CURSOR c (job VARCHAR2, max_sal NUMBER) IS
SELECT employee_name, (salary - max_sal) overpayment FROM emp4 WHERE
job_id = job AND salary > max_sal ORDER BY salary;
PROCEDURE print_overpaid IS
employee_name_ emp4.employee_name%TYPE;
overpayment_ emp4.salary%TYPE;
BEGIN
LOOP
FETCH c INTO employee_name_, overpayment_; //这儿直接使用了外部cursor
EXIT WHEN c%NOTFOUND;
DBMS_OUTPUT.PUT_LINE(employee_name_ || ' (by ' || overpayment_ || ')');
INSERT INTO test2 VALUES(employee_name_, TO_CHAR(overpayment_));
END LOOP;
END print_overpaid;
*/
bool is_pkg_cursor = false;
if (OB_SUCC(ret)) {
// 定义在package spec中的才是package id,这种cursor的routine id是无效的
// 有些定义在package函数中的cursor,它的package id也是有效的,routine id也是有效的。
is_pkg_cursor = OB_INVALID_ID != cursor->get_package_id()
&& OB_INVALID_ID == cursor->get_routine_id();
}
OX (LOG_DEBUG("generate handle ref cursor", K(cursor->get_state()), K(is_pkg_cursor),
K(cursor->get_package_id()), K(cursor->get_routine_id()),K(*cursor)));
if (OB_SUCC(ret) && (pl::ObPLCursor::CursorState::PASSED_IN != cursor->get_state()
&& pl::ObPLCursor::CursorState::DUP_DECL != cursor->get_state()
&& !is_pkg_cursor)
&& cursor->get_routine_id() == s.get_namespace()->get_routine_id()) {
#ifndef NDEBUG
{
ObLLVMValue line_num;
OZ (get_helper().get_int64(s.get_stmt_id(), line_num));
OZ (generate_debug(ObString("close cursor line number"), line_num));
}
#endif
ObSEArray<ObLLVMValue, 6> args;
ObLLVMValue ret_err;
ObLLVMValue arg_value;
OZ (args.push_back(get_vars().at(CTX_IDX)));
OZ (get_helper().get_int64(cursor->get_package_id(), arg_value));
OZ (args.push_back(arg_value));
OZ (get_helper().get_int64(cursor->get_routine_id(), arg_value));
OZ (args.push_back(arg_value));
OZ (get_helper().get_int64(cursor->get_index(), arg_value));
OZ (args.push_back(arg_value));
OZ (get_helper().get_int64(-1, arg_value));
OZ (args.push_back(arg_value));
OZ (get_helper().create_call(ObString("spi_handle_ref_cursor_ref_count"),
get_spi_service().spi_handle_ref_cursor_refcount_,
args, ret_err));
OZ (check_success(ret_err, s.get_stmt_id(), is_notfound, in_warning));
}
return ret;
}
int ObPLCodeGenerator::restart_cg_when_goto_dest(const ObPLStmt &stmt)
{
int ret = OB_SUCCESS;
if (OB_NOT_NULL(get_current().get_v())) {
// do nothing
} else if (stmt.get_is_goto_dst()) {
ObLLVMBasicBlock goto_dst_blk;
OZ (get_helper().create_block(ObString("restart_goto_block"), get_func(), goto_dst_blk));
OZ (set_current(goto_dst_blk));
}
return ret;
}
}
}