database/openGauss-server
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加insert语句支持右值字段引用功能

Browse Source
This commit is contained in:
Laishenghao
2022-11-14 15:12:08 +08:00
parent a31f86e583
commit adf53e20bb
16 changed files with 643 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/gausskernel/runtime/executor/nodeModifyTable.cpp
View File

@ -3832,7 +3832,8 @@ ModifyTableState* ExecInitModifyTable(ModifyTable* node, EState* estate, int efl
i = 0;
foreach (l, node->plans) {
sub_plan = (Plan*)lfirst(l);
sub_plan->rightRefState = node->plan.rightRefState;
/*
* Verify result relation is a valid target for the current operation
*/
@ -3977,6 +3978,7 @@ ModifyTableState* ExecInitModifyTable(ModifyTable* node, EState* estate, int efl
/* Need an econtext too */
econtext = CreateExprContext(estate);
mt_state->ps.ps_ExprContext = econtext;
ATTACH_RIGHT_REF_STATE(&(mt_state->ps));
/*
* Build a projection for each result rel.
@ -4020,6 +4022,7 @@ ModifyTableState* ExecInitModifyTable(ModifyTable* node, EState* estate, int efl
}
econtext = mt_state->ps.ps_ExprContext;
ATTACH_RIGHT_REF_STATE(&(mt_state->ps));
/* initialize slot for the existing tuple */
upsertState->us_existing =
@ -4039,7 +4042,8 @@ ModifyTableState* ExecInitModifyTable(ModifyTable* node, EState* estate, int efl
result_rel_info->ri_updateProj =
ExecBuildProjectionInfo((List*)setexpr, econtext,
upsertState->us_updateproj, result_rel_info->ri_RelationDesc->rd_att);
result_rel_info->ri_updateProj->isUpsertHasRightRef =
IS_ENABLE_RIGHT_REF(econtext->rightRefState) && econtext->rightRefState->isUpsertHasRightRef;
/* initialize expression state to evaluate update where clause if exists */
if (node->upsertWhere) {
upsertState->us_updateWhere = (List*)ExecInitExpr((Expr*)node->upsertWhere, &mt_state->ps);