mirror of https://github.com/citusdata/citus.git
Enhance set_relation_column_names to handle relid validation and support for synthetic PG 18 RTEs
parent
0f5116e127
commit
f90f018951
|
@ -1163,8 +1163,9 @@ set_relation_column_names(deparse_namespace *dpns, RangeTblEntry *rte,
|
||||||
* real_colnames[] will be indexed by physical column number, with NULL
|
* real_colnames[] will be indexed by physical column number, with NULL
|
||||||
* entries for dropped columns.
|
* entries for dropped columns.
|
||||||
*/
|
*/
|
||||||
if (rte->rtekind == RTE_RELATION ||
|
if ((rte->rtekind == RTE_RELATION ||
|
||||||
GetRangeTblKind(rte) == CITUS_RTE_SHARD)
|
GetRangeTblKind(rte) == CITUS_RTE_SHARD) &&
|
||||||
|
OidIsValid(rte->relid))
|
||||||
{
|
{
|
||||||
/* Relation --- look to the system catalogs for up-to-date info */
|
/* Relation --- look to the system catalogs for up-to-date info */
|
||||||
Relation rel;
|
Relation rel;
|
||||||
|
@ -1186,6 +1187,28 @@ set_relation_column_names(deparse_namespace *dpns, RangeTblEntry *rte,
|
||||||
real_colnames[i] = pstrdup(NameStr(attr->attname));
|
real_colnames[i] = pstrdup(NameStr(attr->attname));
|
||||||
}
|
}
|
||||||
relation_close(rel, AccessShareLock);
|
relation_close(rel, AccessShareLock);
|
||||||
|
}
|
||||||
|
else if (GetRangeTblKind(rte) == CITUS_RTE_SHARD)
|
||||||
|
{
|
||||||
|
/* shard RTE without relid (pulled-up clone in PG18) */
|
||||||
|
/* use the column aliases already stored in rte->eref->colnames */
|
||||||
|
|
||||||
|
ncolumns = list_length(rte->eref->colnames);
|
||||||
|
real_colnames = (char **) palloc0(ncolumns * sizeof(char *));
|
||||||
|
|
||||||
|
for (i = 0; i < ncolumns; i++)
|
||||||
|
real_colnames[i] = pstrdup(strVal(list_nth(rte->eref->colnames, i)));
|
||||||
|
|
||||||
|
/* keep changed_any / has_anonymous defaults */
|
||||||
|
}
|
||||||
|
else if (rte->rtekind == RTE_GROUP)
|
||||||
|
{
|
||||||
|
/* ----- synthetic PG 18 RTE for GROUP BY / HAVING ----- */
|
||||||
|
ncolumns = list_length(rte->eref->colnames);
|
||||||
|
real_colnames = (char **) palloc0(ncolumns * sizeof(char *));
|
||||||
|
|
||||||
|
for (i = 0; i < ncolumns; i++)
|
||||||
|
real_colnames[i] = pstrdup(strVal(list_nth(rte->eref->colnames, i)));
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
|
|
Loading…
Reference in New Issue