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

wip

test_fixunion
Sait Talha Nisanci 2021-07-05 14:45:54 +03:00
parent e4b8c7c183
commit f98efe6c2c
1 changed files with 24 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/backend/distributed/planner/relation_restriction_equivalence.c
View File

@ -1793,13 +1793,35 @@ RelationRestrictionPartitionKeyIndex(Query* originalQuery, RelationRestriction *
partitionKeyTargetAttrIndex++; partitionKeyTargetAttrIndex++;
if (!targetEntry->resjunk && if (!targetEntry->resjunk &&
IsA(targetExpression, Var) && IsA(targetExpression, Var))
IsPartitionColumn(targetExpression, relationPlannerParseQuery))
{ {
Var *targetColumn = (Var *) targetExpression; Var *targetColumn = (Var *) targetExpression;
// elog(WARNING, "%d %d", targetColumn->varno, targetColumn->varattno);
if (!IsPartitionColumn(targetExpression, relationPlannerParseQuery)) {
continue;
}
Oid relationId = InvalidOid;
Var *column = NULL;
RangeTblEntry* res = (RangeTblEntry*) list_nth(relationPlannerParseQuery->rtable, targetColumn->varno - 1); RangeTblEntry* res = (RangeTblEntry*) list_nth(relationPlannerParseQuery->rtable, targetColumn->varno - 1);
if (res->rtekind == RTE_JOIN) {
Var* cur = NULL;
int index = 1;
foreach_ptr(cur, res->joinaliasvars) {
if (index == targetColumn->varattno) {
res = (RangeTblEntry*) list_nth(relationPlannerParseQuery->rtable, cur->varno - 1);
break;
}
index++;
}
}
if (res->rtekind == RTE_RELATION && GetRTEIdentity(res) == findRteIdentityContext->rteIdentity) if (res->rtekind == RTE_RELATION && GetRTEIdentity(res) == findRteIdentityContext->rteIdentity)
{ {
FindReferencedTableColumn(targetExpression, NIL, relationPlannerParseQuery, &relationId, &column);
targetColumn = column;
*partitionKeyIndex = partitionKeyTargetAttrIndex; *partitionKeyIndex = partitionKeyTargetAttrIndex;
return (Var*)copyObject(targetColumn); return (Var*)copyObject(targetColumn);
} }