m3hm3t/pg18_pushdown_group_column
Mehmet Yilmaz 2025-06-25 14:06:38 +00:00
parent f49caf8aca
commit 128a259b15
1 changed files with 49 additions and 41 deletions

View File

@ -136,8 +136,7 @@ static MultiNode * ApplyCartesianProductReferenceJoin(MultiNode *leftNode,
static MultiNode * ApplyCartesianProduct(MultiNode *leftNode, MultiNode *rightNode,
List *partitionColumnList, JoinType joinType,
List *joinClauses);
static bool
ExprMentionsPartitionColumn(Node *node, Query *query);
static bool ExprMentionsPartitionColumn(Node *node, Query *query);
/*
@ -233,7 +232,8 @@ TargetListOnPartitionColumn(Query *query, List *targetEntryList)
RangeTblEntry *rte = NULL;
FindReferencedTableColumn(targetExpression, NIL, query, &column, &rte,
/*skipOuterVars=*/false);
/*skipOuterVars=*/ false);
Oid relationId = rte ? rte->relid : InvalidOid;
/*
@ -308,7 +308,9 @@ static bool
ExprMentionsPartitionColumn(Node *node, Query *query)
{
if (node == NULL)
{
return false;
}
if (IsA(node, Var))
{
@ -323,21 +325,27 @@ ExprMentionsPartitionColumn(Node *node, Query *query)
/* Sanity-check varno */
if (v->varno <= 0 || v->varno > list_length(query->rtable))
{
return false;
}
RangeTblEntry *rte = rt_fetch(v->varno, query->rtable);
#if PG_VERSION_NUM >= 180000
/* Synthetic GROUP RTE examine its expressions instead */
if (rte->rtekind == RTE_GROUP && rte->groupexprs)
{
ListCell *lc;
foreach (lc, rte->groupexprs)
foreach(lc, rte->groupexprs)
if (ExprMentionsPartitionColumn((Node *) lfirst(lc), query))
{
return true;
}
return false;
}
#endif
/* Real table? — compare against its dist key */
if (rte->rtekind == RTE_RELATION && HasDistributionKey(rte->relid))
{