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

83
src/sql/resolver/expr/ob_raw_expr_replacer.cpp
View File

@ -10,50 +10,82 @@
* See the Mulan PubL v2 for more details.
*/
#define USING_LOG_PREFIX SQL_RESV
#include "ob_raw_expr_replacer.h"
using namespace oceanbase::sql;
using namespace oceanbase::common;
namespace oceanbase {
namespace sql {
ObRawExprReplacer::ObRawExprReplacer(ObRawExpr* old_expr, ObRawExpr* new_expr)
: old_expr_(old_expr), new_expr_(new_expr)
namespace oceanbase
{
namespace sql
{
ObRawExprReplacer::ObRawExprReplacer(ObRawExpr *old_expr,
ObRawExpr *new_expr)
: old_expr_(old_expr), new_expr_(new_expr)
{}
ObRawExprReplacer::~ObRawExprReplacer()
{}
int ObRawExprReplacer::replace(ObRawExpr& expr)
int ObRawExprReplacer::replace(ObRawExpr &expr)
{
return expr.preorder_accept(*this);
}
int ObRawExprReplacer::visit(ObConstRawExpr& expr)
int ObRawExprReplacer::visit(ObConstRawExpr &expr)
{
UNUSED(expr);
return OB_SUCCESS;
}
int ObRawExprReplacer::visit(ObVarRawExpr& expr)
int ObRawExprReplacer::visit(ObExecParamRawExpr &expr)
{
int ret = OB_SUCCESS;
if (&expr != new_expr_) {
if (OB_ISNULL(expr.get_ref_expr())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("ref expr is null", K(ret));
} else if (expr.get_ref_expr() == old_expr_) {
expr.set_ref_expr(new_expr_);
} else if (OB_FAIL(expr.get_ref_expr()->preorder_accept(*this))) {
LOG_WARN("failed to visit ref expr", K(ret));
}
}
return ret;
}
int ObRawExprReplacer::visit(ObVarRawExpr &expr)
{
UNUSED(expr);
return OB_SUCCESS;
}
int ObRawExprReplacer::visit(ObQueryRefRawExpr& expr)
int ObRawExprReplacer::visit(ObOpPseudoColumnRawExpr &expr)
{
UNUSED(expr);
return OB_SUCCESS;
}
int ObRawExprReplacer::visit(ObColumnRefRawExpr& expr)
int ObRawExprReplacer::visit(ObQueryRefRawExpr &expr)
{
UNUSED(expr);
return OB_SUCCESS;
}
int ObRawExprReplacer::visit(ObOpRawExpr& expr)
int ObRawExprReplacer::visit(ObPlQueryRefRawExpr &expr)
{
UNUSED(expr);
return OB_SUCCESS;
}
int ObRawExprReplacer::visit(ObColumnRefRawExpr &expr)
{
UNUSED(expr);
return OB_SUCCESS;
}
int ObRawExprReplacer::visit(ObOpRawExpr &expr)
{
int ret = OB_SUCCESS;
if (&expr != new_expr_) {
@ -67,7 +99,7 @@ int ObRawExprReplacer::visit(ObOpRawExpr& expr)
return ret;
}
int ObRawExprReplacer::visit(ObCaseOpRawExpr& expr)
int ObRawExprReplacer::visit(ObCaseOpRawExpr &expr)
{
int ret = OB_SUCCESS;
if (expr.get_when_expr_size() != expr.get_then_expr_size()) {
@ -84,7 +116,7 @@ int ObRawExprReplacer::visit(ObCaseOpRawExpr& expr)
ret = expr.replace_when_param_expr(i, new_expr_);
}
if (OB_SUCCESS == ret) {
if (OB_SUCCESS == ret ) {
if (expr.get_then_param_expr(i) == old_expr_) {
ret = expr.replace_then_param_expr(i, new_expr_);
}
@ -101,24 +133,18 @@ int ObRawExprReplacer::visit(ObCaseOpRawExpr& expr)
return ret;
}
int ObRawExprReplacer::visit(ObAggFunRawExpr& expr)
int ObRawExprReplacer::visit(ObAggFunRawExpr &expr)
{
int ret = OB_SUCCESS;
if (&expr != new_expr_) {
ObIArray<ObRawExpr*>& real_param_exprs = expr.get_real_param_exprs_for_update();
ObIArray<ObRawExpr*> &real_param_exprs = expr.get_real_param_exprs_for_update();
for (int64_t i = 0; OB_SUCC(ret) && i < real_param_exprs.count(); ++i) {
if (real_param_exprs.at(i) == old_expr_) {
real_param_exprs.at(i) = new_expr_;
}
}
if (OB_SUCC(ret)) {
if (expr.get_push_down_sum_expr() == old_expr_) {
expr.set_push_down_sum_expr(new_expr_);
}
if (expr.get_push_down_count_expr() == old_expr_) {
expr.set_push_down_count_expr(new_expr_);
}
ObIArray<OrderItem>& order_items = expr.get_order_items_for_update();
ObIArray<OrderItem> &order_items = expr.get_order_items_for_update();
for (int64_t i = 0; OB_SUCC(ret) && i < order_items.count(); ++i) {
if (order_items.at(i).expr_ == old_expr_) {
order_items.at(i).expr_ = new_expr_;
@ -129,7 +155,7 @@ int ObRawExprReplacer::visit(ObAggFunRawExpr& expr)
return ret;
}
int ObRawExprReplacer::visit(ObSysFunRawExpr& expr)
int ObRawExprReplacer::visit(ObSysFunRawExpr &expr)
{
int ret = OB_SUCCESS;
if (&expr != new_expr_) {
@ -143,13 +169,13 @@ int ObRawExprReplacer::visit(ObSysFunRawExpr& expr)
return OB_SUCCESS;
}
int ObRawExprReplacer::visit(ObSetOpRawExpr& expr)
int ObRawExprReplacer::visit(ObSetOpRawExpr &expr)
{
UNUSED(expr);
return OB_SUCCESS;
}
int ObRawExprReplacer::visit(ObWinFunRawExpr& expr)
int ObRawExprReplacer::visit(ObWinFunRawExpr &expr)
{
int ret = OB_SUCCESS;
if (&expr != new_expr_) {
@ -163,5 +189,10 @@ int ObRawExprReplacer::visit(ObWinFunRawExpr& expr)
return ret;
}
} // end of namespace sql
} // end of namespace oceanbase
bool ObRawExprReplacer::skip_child(ObRawExpr &expr)
{
return &expr == new_expr_;
}
}//end of namespace sql
}//end of namespace oceanbase