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

some variable renameing

pull/7643/head
paragjain 2024-07-14 10:45:16 +00:00
parent 2fc5e14ee8
commit 346422e7be
4 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/backend/distributed/planner/merge_planner.c
View File

@ -857,8 +857,11 @@ ConvertRelationRTEIntoSubquery(Query *mergeQuery, RangeTblEntry *sourceRte,
/* set the FROM expression to the subquery */
newRangeTableRef->rtindex = SINGLE_RTE_INDEX;
sourceResultsQuery->jointree = makeFromExpr(list_make1(newRangeTableRef), NULL);
bool isMergeQuery = true;
sourceResultsQuery->targetList =
CreateAllTargetListForRelation(sourceRte->relid, requiredAttributes);
CreateAllTargetListForRelation(sourceRte->relid, requiredAttributes,
isMergeQuery);
List *restrictionList =
GetRestrictInfoListForRelation(sourceRte, plannerRestrictionContext);
List *copyRestrictionList = copyObject(restrictionList);

7
src/backend/distributed/planner/query_colocation_checker.c
View File

@ -325,12 +325,14 @@ WrapRteRelationIntoSubquery(RangeTblEntry *rteRelation,
* as a NULL column.
*/
List *
CreateAllTargetListForRelation(Oid relationId, List *requiredAttributes)
CreateAllTargetListForRelation(Oid relationId, List *requiredAttributes, bool
isMergeQuery)
{
Relation relation = relation_open(relationId, AccessShareLock);
int numberOfAttributes = RelationGetNumberOfAttributes(relation);
List *targetList = NIL;
int colAppendIdx = 1;
for (int attrNum = 1; attrNum <= numberOfAttributes; attrNum++)
{
@ -360,8 +362,9 @@ CreateAllTargetListForRelation(Oid relationId, List *requiredAttributes)
}
else
{
int varAttNum = isMergeQuery ? attrNum : colAppendIdx++;
TargetEntry *targetEntry =
CreateTargetEntryForColumn(attributeTuple, SINGLE_RTE_INDEX, attrNum,
CreateTargetEntryForColumn(attributeTuple, SINGLE_RTE_INDEX, varAttNum,
resNo);
targetList = lappend(targetList, targetEntry);
}

4
src/backend/distributed/planner/recursive_planning.c
View File

@ -1767,8 +1767,10 @@ ReplaceRTERelationWithRteSubquery(RangeTblEntry *rangeTableEntry,
{
Query *subquery = WrapRteRelationIntoSubquery(rangeTableEntry, requiredAttrNumbers,
perminfo);
bool isMergeQuery = false;
List *outerQueryTargetList = CreateAllTargetListForRelation(rangeTableEntry->relid,
requiredAttrNumbers);
requiredAttrNumbers,
isMergeQuery);
List *restrictionList =
GetRestrictInfoListForRelation(rangeTableEntry,

3
src/include/distributed/query_colocation_checker.h
View File

@ -38,6 +38,7 @@ extern bool SubqueryColocated(Query *subquery, ColocatedJoinChecker *context);
extern Query * WrapRteRelationIntoSubquery(RangeTblEntry *rteRelation,
List *requiredAttributes,
RTEPermissionInfo *perminfo);
extern List * CreateAllTargetListForRelation(Oid relationId, List *requiredAttributes);
extern List * CreateAllTargetListForRelation(Oid relationId, List *requiredAttributes,
bool isMergeQuery);
#endif /* QUERY_COLOCATION_CHECKER_H */