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

patch 4.0

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

130
src/sql/optimizer/ob_explain_log_plan.cpp
View File

@ -16,6 +16,7 @@
#include "sql/optimizer/ob_log_operator_factory.h"
#include "sql/optimizer/ob_log_plan_factory.h"
#include "sql/optimizer/ob_log_values.h"
#include "sql/code_generator/ob_code_generator.h"
using namespace oceanbase;
using namespace sql;
@ -33,58 +34,65 @@ using namespace oceanbase::sql::log_op_def;
int ObExplainLogPlan::generate_raw_plan()
{
int ret = OB_SUCCESS;
ObLogPlan* child_plan = NULL;
ObDMLStmt* child_stmt = NULL;
ObLogValues* values = NULL;
if (NULL == get_stmt() || !get_stmt()->is_explain_stmt()) {
if (OB_ISNULL(get_stmt()) || OB_UNLIKELY(!get_stmt()->is_explain_stmt())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("invalid stmt for explain", "stmt", get_stmt());
LOG_WARN("get unexpected error", K(get_stmt()), K(ret));
} else {
const ObExplainStmt* explain_stmt = static_cast<const ObExplainStmt*>(get_stmt());
ObLogPlan *child_plan = NULL;
const ObDMLStmt *child_stmt = NULL;
ObLogicalOperator *top = NULL;
ObLogValues *values_op = NULL;
int64_t batch_size = 0;
const ObExplainStmt *explain_stmt = static_cast<const ObExplainStmt*>(get_stmt());
if (OB_ISNULL(child_stmt = explain_stmt->get_explain_query_stmt())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("no child stmt for explain");
} else if (OB_ISNULL(
child_plan = optimizer_context_.get_log_plan_factory().create(optimizer_context_, *child_stmt))) {
LOG_WARN("get unexpected null", K(ret));
} else if (OB_ISNULL(child_plan = optimizer_context_.get_log_plan_factory().
create(optimizer_context_, *child_stmt))) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_ERROR("failed to create log plan for explain stmt");
} else if (OB_FAIL(child_plan->init_plan_info())) {
LOG_WARN("failed to init equal_sets");
} else if (OB_FAIL(child_plan->generate_plan())) {
LOG_WARN("failed to generate plan tree for explain", K(ret));
} else if (OB_FAIL(ObCodeGenerator::detect_batch_size(*child_plan, batch_size))) {
LOG_WARN("detect batch size failed", K(ret));
} else if (OB_FAIL(allocate_values_as_top(top))) {
LOG_WARN("failed to allocate expr values_op as top", K(ret));
} else if (OB_ISNULL(top) || OB_UNLIKELY(LOG_VALUES != top->get_type())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("get unexpected error", K(top), K(ret));
} else {
char* buf = NULL;
values = static_cast<ObLogValues*>(log_op_factory_.allocate(*this, LOG_VALUES));
if (NULL == values) {
set_plan_root(top);
top->mark_is_plan_root();
child_plan->get_optimizer_context().set_batch_size(batch_size);
char *buf = NULL;
values_op = static_cast<ObLogValues*>(top);
if (OB_ISNULL(
buf = static_cast<char*>(get_allocator().alloc(ObLogValues::MAX_EXPLAIN_BUFFER_SIZE)))) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_ERROR("failed to allocate UPDATE operator");
} else if (NULL == (buf = static_cast<char*>(get_allocator().alloc(ObLogValues::MAX_EXPLAIN_BUFFER_SIZE)))) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_ERROR("Failed to allocate buffer",
"buffer size",
static_cast<int64_t>(ObLogValues::MAX_EXPLAIN_BUFFER_SIZE),
K(ret));
LOG_ERROR("Failed to allocate buffer", "buffer size",
static_cast<int64_t>(ObLogValues::MAX_EXPLAIN_BUFFER_SIZE), K(ret));
} else {
// NO MORE FAILURE FROM NOW ON!!!
int64_t pos = 0;
values->set_explain_plan(child_plan);
values_op->set_explain_plan(child_plan);
if (EXPLAIN_FORMAT_JSON == explain_stmt->get_explain_type()) {
char* pre_buf = NULL;
if (NULL == (pre_buf = static_cast<char*>(get_allocator().alloc(ObLogValues::MAX_EXPLAIN_BUFFER_SIZE)))) {
char *pre_buf = NULL;
if (NULL
== (pre_buf = static_cast<char*>(get_allocator().alloc(
ObLogValues::MAX_EXPLAIN_BUFFER_SIZE)))) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_ERROR("Failed to allocate buffer",
"buffer size",
static_cast<int64_t>(ObLogValues::MAX_EXPLAIN_BUFFER_SIZE),
K(ret));
LOG_ERROR("Failed to allocate buffer", "buffer size",
static_cast<int64_t>(ObLogValues::MAX_EXPLAIN_BUFFER_SIZE), K(ret));
} else if (NULL == child_plan->get_plan_root()) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("invalid argument", K(ret));
} else {
Value* child_value = NULL;
if (OB_FAIL(child_plan->get_plan_root()->to_json(
pre_buf, ObLogValues::MAX_EXPLAIN_BUFFER_SIZE, pos, child_value))) {
Value *child_value = NULL;
if (OB_FAIL(
child_plan->get_plan_root()->to_json(pre_buf, ObLogValues::MAX_EXPLAIN_BUFFER_SIZE,
pos, child_value))) {
LOG_WARN("to_json fails", K(ret), K(child_value));
} else {
Tidy tidy(child_value);
@ -101,28 +109,31 @@ int ObExplainLogPlan::generate_raw_plan()
pre_buf = NULL;
}
} else {
pos = child_plan->to_string(buf, ObLogValues::MAX_EXPLAIN_BUFFER_SIZE, explain_stmt->get_explain_type());
pos = child_plan->to_string(buf, ObLogValues::MAX_EXPLAIN_BUFFER_SIZE,
explain_stmt->get_explain_type(),
explain_stmt->get_display_opt());
if (pos < ObLogValues::MAX_EXPLAIN_BUFFER_SIZE - 2) {
buf[pos + 1] = '\0';
} else {
buf[ObLogValues::MAX_EXPLAIN_BUFFER_SIZE - 1] = '\0';
}
}
if (OB_SUCC(ret)) {
//For explain stmt, we can do pack at the stage of expr alloc,
//But we need to use the FALSE flag to tell the driver
//to use normal encoding instead of memcopy
optimizer_context_.set_packed(false);
ObObj obj;
obj.set_varchar(ObString::make_string(buf));
ObNewRow row;
row.cells_ = &obj;
row.count_ = 1;
if (OB_FAIL(values->add_row(row))) {
if (OB_FAIL(values_op->add_row(row))) {
LOG_WARN("failed to add row", K(ret));
} else {
set_phy_plan_type(OB_PHY_PLAN_LOCAL);
set_plan_root(values);
values->mark_is_plan_root();
// set values operator id && set max operator id for LogPlan
values->set_op_id(0);
// set values_op operator id && set max operator id for LogPlan
values_op->set_op_id(0);
set_max_op_id(1);
}
}
@ -132,42 +143,15 @@ int ObExplainLogPlan::generate_raw_plan()
}
}
}
get_optimizer_context().get_all_exprs().reuse();
if (OB_FAIL(ret)) {
} else if (OB_ISNULL(optimizer_context_.get_session_info())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("session is NULL", K(ret));
} else if (optimizer_context_.get_session_info()->use_static_typing_engine()) {
// Static typing engine need expr to output rows. We generate a const output expr
// for values operator.
ObConstRawExpr* output = NULL;
if (OB_FAIL(optimizer_context_.get_expr_factory().create_raw_expr(T_VARCHAR, output))) {
LOG_WARN("create const expr failed", K(ret));
} else {
ObObj v;
v.set_varchar(" ");
v.set_collation_type(ObCharset::get_system_collation());
output->set_param(v);
output->set_value(v);
if (OB_FAIL(output->formalize(optimizer_context_.get_session_info()))) {
LOG_WARN("const expr formalize failed", K(ret));
} else if (OB_FAIL(values->add_expr_to_output(output))) {
LOG_WARN("add output expr failed", K(ret));
}
}
if (OB_SUCC(ret)) {
if (OB_FAIL(plan_traverse_loop(OPERATOR_NUMBERING, ALLOC_EXPR))) {
LOG_WARN("failed to do plan traverse", K(ret));
}
}
} else if (OB_FAIL(allocate_output_expr_for_values_op(*values_op))) {
LOG_WARN("failed ot allocate output expr", K(ret));
} else {
get_optimizer_context().set_plan_type(ObPhyPlanType::OB_PHY_PLAN_LOCAL,
ObPhyPlanType::OB_PHY_PLAN_LOCAL,
false);
}
}
return ret;
}
int ObExplainLogPlan::generate_plan()
{
return generate_raw_plan();
}