external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Get partition column via updatedCols

pg14_support_after_rebase
Sait Talha Nisanci 2021-08-23 17:50:41 +03:00
parent 75a0bc5755
commit 3b92284a9a
1 changed files with 17 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/backend/distributed/planner/multi_router_planner.c
View File

@ -1134,6 +1134,23 @@ ErrorIfOnConflictNotSupported(Query *queryTree)
Node *arbiterWhere = queryTree->onConflict->arbiterWhere;
Node *onConflictWhere = queryTree->onConflict->onConflictWhere;
bool setTargetEntryPartitionColumn = false;
if (partitionColumn) {
RangeTblEntry* resultRTE = ExtractResultRelationRTE(queryTree);
/*
* FirstLowInvalidHeapAttributeNumber is added as an offset to rte->updatedCols.
* So we substract that to get the column no for an updated column that matches
* resultRTE->updatedcols.
*/
int updatedColNoWithOffset = partitionColumn->varattno - FirstLowInvalidHeapAttributeNumber;
if (bms_is_member(updatedColNoWithOffset, resultRTE->updatedCols)) {
setTargetEntryPartitionColumn = true;
}
}
/*
* onConflictSet is expanded via expand_targetlist() on the standard planner.
* This ends up adding all the columns to the onConflictSet even if the user
@ -1146,17 +1163,6 @@ ErrorIfOnConflictNotSupported(Query *queryTree)
foreach(setTargetCell, onConflictSet)
{
TargetEntry *setTargetEntry = (TargetEntry *) lfirst(setTargetCell);
bool setTargetEntryPartitionColumn = false;
/* reference tables do not have partition column */
if (partitionColumn == NULL)
{
setTargetEntryPartitionColumn = false;
}
else if (setTargetEntry->resno == partitionColumn->varattno)
{
setTargetEntryPartitionColumn = true;
}
if (setTargetEntryPartitionColumn)
{