PostgreSQL Source Code  git master
inherit.c File Reference
Include dependency graph for inherit.c:

Go to the source code of this file.

Functions

static void expand_partitioned_rtentry (PlannerInfo *root, RelOptInfo *relinfo, RangeTblEntry *parentrte, Index parentRTindex, Relation parentrel, Bitmapset *parent_updatedCols, PlanRowMark *top_parentrc, LOCKMODE lockmode)
 
static void expand_single_inheritance_child (PlannerInfo *root, RangeTblEntry *parentrte, Index parentRTindex, Relation parentrel, PlanRowMark *top_parentrc, Relation childrel, RangeTblEntry **childrte_p, Index *childRTindex_p)
 
static Bitmapsettranslate_col_privs (const Bitmapset *parent_privs, List *translated_vars)
 
static Bitmapsettranslate_col_privs_multilevel (PlannerInfo *root, RelOptInfo *rel, RelOptInfo *parent_rel, Bitmapset *parent_cols)
 
static void expand_appendrel_subquery (PlannerInfo *root, RelOptInfo *rel, RangeTblEntry *rte, Index rti)
 
void expand_inherited_rtentry (PlannerInfo *root, RelOptInfo *rel, RangeTblEntry *rte, Index rti)
 
Bitmapsetget_rel_all_updated_cols (PlannerInfo *root, RelOptInfo *rel)
 
bool apply_child_basequals (PlannerInfo *root, RelOptInfo *parentrel, RelOptInfo *childrel, RangeTblEntry *childRTE, AppendRelInfo *appinfo)
 

Function Documentation

◆ apply_child_basequals()

bool apply_child_basequals ( PlannerInfo root,
RelOptInfo parentrel,
RelOptInfo childrel,
RangeTblEntry childRTE,
AppendRelInfo appinfo 
)

Definition at line 832 of file inherit.c.

835 {
836  List *childquals;
837  Index cq_min_security;
838  ListCell *lc;
839 
840  /*
841  * The child rel's targetlist might contain non-Var expressions, which
842  * means that substitution into the quals could produce opportunities for
843  * const-simplification, and perhaps even pseudoconstant quals. Therefore,
844  * transform each RestrictInfo separately to see if it reduces to a
845  * constant or pseudoconstant. (We must process them separately to keep
846  * track of the security level of each qual.)
847  */
848  childquals = NIL;
849  cq_min_security = UINT_MAX;
850  foreach(lc, parentrel->baserestrictinfo)
851  {
852  RestrictInfo *rinfo = (RestrictInfo *) lfirst(lc);
853  Node *childqual;
854  ListCell *lc2;
855 
856  Assert(IsA(rinfo, RestrictInfo));
857  childqual = adjust_appendrel_attrs(root,
858  (Node *) rinfo->clause,
859  1, &appinfo);
860  childqual = eval_const_expressions(root, childqual);
861  /* check for flat-out constant */
862  if (childqual && IsA(childqual, Const))
863  {
864  if (((Const *) childqual)->constisnull ||
865  !DatumGetBool(((Const *) childqual)->constvalue))
866  {
867  /* Restriction reduces to constant FALSE or NULL */
868  return false;
869  }
870  /* Restriction reduces to constant TRUE, so drop it */
871  continue;
872  }
873  /* might have gotten an AND clause, if so flatten it */
874  foreach(lc2, make_ands_implicit((Expr *) childqual))
875  {
876  Node *onecq = (Node *) lfirst(lc2);
877  bool pseudoconstant;
878 
879  /* check for pseudoconstant (no Vars or volatile functions) */
880  pseudoconstant =
881  !contain_vars_of_level(onecq, 0) &&
883  if (pseudoconstant)
884  {
885  /* tell createplan.c to check for gating quals */
886  root->hasPseudoConstantQuals = true;
887  }
888  /* reconstitute RestrictInfo with appropriate properties */
889  childquals = lappend(childquals,
890  make_restrictinfo(root,
891  (Expr *) onecq,
892  rinfo->is_pushed_down,
893  rinfo->has_clone,
894  rinfo->is_clone,
895  pseudoconstant,
896  rinfo->security_level,
897  NULL, NULL, NULL));
898  /* track minimum security level among child quals */
899  cq_min_security = Min(cq_min_security, rinfo->security_level);
900  }
901  }
902 
903  /*
904  * In addition to the quals inherited from the parent, we might have
905  * securityQuals associated with this particular child node. (Currently
906  * this can only happen in appendrels originating from UNION ALL;
907  * inheritance child tables don't have their own securityQuals, see
908  * expand_single_inheritance_child().) Pull any such securityQuals up
909  * into the baserestrictinfo for the child. This is similar to
910  * process_security_barrier_quals() for the parent rel, except that we
911  * can't make any general deductions from such quals, since they don't
912  * hold for the whole appendrel.
913  */
914  if (childRTE->securityQuals)
915  {
916  Index security_level = 0;
917 
918  foreach(lc, childRTE->securityQuals)
919  {
920  List *qualset = (List *) lfirst(lc);
921  ListCell *lc2;
922 
923  foreach(lc2, qualset)
924  {
925  Expr *qual = (Expr *) lfirst(lc2);
926 
927  /* not likely that we'd see constants here, so no check */
928  childquals = lappend(childquals,
929  make_restrictinfo(root, qual,
930  true,
931  false, false,
932  false,
933  security_level,
934  NULL, NULL, NULL));
935  cq_min_security = Min(cq_min_security, security_level);
936  }
937  security_level++;
938  }
939  Assert(security_level <= root->qual_security_level);
940  }
941 
942  /*
943  * OK, we've got all the baserestrictinfo quals for this child.
944  */
945  childrel->baserestrictinfo = childquals;
946  childrel->baserestrict_min_security = cq_min_security;
947 
948  return true;
949 }
Node * adjust_appendrel_attrs(PlannerInfo *root, Node *node, int nappinfos, AppendRelInfo **appinfos)
Definition: appendinfo.c:196
#define Min(x, y)
Definition: c.h:991
unsigned int Index
Definition: c.h:601
Node * eval_const_expressions(PlannerInfo *root, Node *node)
Definition: clauses.c:2234
bool contain_volatile_functions(Node *clause)
Definition: clauses.c:518
Assert(fmt[strlen(fmt) - 1] !='\n')
List * lappend(List *list, void *datum)
Definition: list.c:339
List * make_ands_implicit(Expr *clause)
Definition: makefuncs.c:721
#define IsA(nodeptr, _type_)
Definition: nodes.h:158
#define lfirst(lc)
Definition: pg_list.h:172
#define NIL
Definition: pg_list.h:68
static bool DatumGetBool(Datum X)
Definition: postgres.h:90
RestrictInfo * make_restrictinfo(PlannerInfo *root, Expr *clause, bool is_pushed_down, bool has_clone, bool is_clone, bool pseudoconstant, Index security_level, Relids required_relids, Relids incompatible_relids, Relids outer_relids)
Definition: restrictinfo.c:63
Definition: pg_list.h:54
Definition: nodes.h:129
bool hasPseudoConstantQuals
Definition: pathnodes.h:495
List * securityQuals
Definition: parsenodes.h:1208
List * baserestrictinfo
Definition: pathnodes.h:966
Index baserestrict_min_security
Definition: pathnodes.h:970
bool is_pushed_down
Definition: pathnodes.h:2544
Index security_level
Definition: pathnodes.h:2563
Expr * clause
Definition: pathnodes.h:2541
bool has_clone
Definition: pathnodes.h:2553
bool contain_vars_of_level(Node *node, int levelsup)
Definition: var.c:441

References adjust_appendrel_attrs(), Assert(), RelOptInfo::baserestrict_min_security, RelOptInfo::baserestrictinfo, RestrictInfo::clause, contain_vars_of_level(), contain_volatile_functions(), DatumGetBool(), eval_const_expressions(), RestrictInfo::has_clone, PlannerInfo::hasPseudoConstantQuals, RestrictInfo::is_clone, RestrictInfo::is_pushed_down, IsA, lappend(), lfirst, make_ands_implicit(), make_restrictinfo(), Min, NIL, RestrictInfo::security_level, and RangeTblEntry::securityQuals.

Referenced by build_simple_rel().

◆ expand_appendrel_subquery()

static void expand_appendrel_subquery ( PlannerInfo root,
RelOptInfo rel,
RangeTblEntry rte,
Index  rti 
)
static

Definition at line 791 of file inherit.c.

793 {
794  ListCell *l;
795 
796  foreach(l, root->append_rel_list)
797  {
798  AppendRelInfo *appinfo = (AppendRelInfo *) lfirst(l);
799  Index childRTindex = appinfo->child_relid;
800  RangeTblEntry *childrte;
801  RelOptInfo *childrel;
802 
803  /* append_rel_list contains all append rels; ignore others */
804  if (appinfo->parent_relid != rti)
805  continue;
806 
807  /* find the child RTE, which should already exist */
808  Assert(childRTindex < root->simple_rel_array_size);
809  childrte = root->simple_rte_array[childRTindex];
810  Assert(childrte != NULL);
811 
812  /* Build the child RelOptInfo. */
813  childrel = build_simple_rel(root, childRTindex, rel);
814 
815  /* Child may itself be an inherited rel, either table or subquery. */
816  if (childrte->inh)
817  expand_inherited_rtentry(root, childrel, childrte, childRTindex);
818  }
819 }
void expand_inherited_rtentry(PlannerInfo *root, RelOptInfo *rel, RangeTblEntry *rte, Index rti)
Definition: inherit.c:86
RelOptInfo * build_simple_rel(PlannerInfo *root, int relid, RelOptInfo *parent)
Definition: relnode.c:192
Index child_relid
Definition: pathnodes.h:2944
Index parent_relid
Definition: pathnodes.h:2943
List * append_rel_list
Definition: pathnodes.h:362

References PlannerInfo::append_rel_list, Assert(), build_simple_rel(), AppendRelInfo::child_relid, expand_inherited_rtentry(), RangeTblEntry::inh, lfirst, and AppendRelInfo::parent_relid.

Referenced by expand_inherited_rtentry().

◆ expand_inherited_rtentry()

void expand_inherited_rtentry ( PlannerInfo root,
RelOptInfo rel,
RangeTblEntry rte,
Index  rti 
)

Definition at line 86 of file inherit.c.

88 {
89  Oid parentOID;
90  Relation oldrelation;
91  LOCKMODE lockmode;
92  PlanRowMark *oldrc;
93  bool old_isParent = false;
94  int old_allMarkTypes = 0;
95 
96  Assert(rte->inh); /* else caller error */
97 
98  if (rte->rtekind == RTE_SUBQUERY)
99  {
100  expand_appendrel_subquery(root, rel, rte, rti);
101  return;
102  }
103 
104  Assert(rte->rtekind == RTE_RELATION);
105 
106  parentOID = rte->relid;
107 
108  /*
109  * We used to check has_subclass() here, but there's no longer any need
110  * to, because subquery_planner already did.
111  */
112 
113  /*
114  * The rewriter should already have obtained an appropriate lock on each
115  * relation named in the query, so we can open the parent relation without
116  * locking it. However, for each child relation we add to the query, we
117  * must obtain an appropriate lock, because this will be the first use of
118  * those relations in the parse/rewrite/plan pipeline. Child rels should
119  * use the same lockmode as their parent.
120  */
121  oldrelation = table_open(parentOID, NoLock);
122  lockmode = rte->rellockmode;
123 
124  /*
125  * If parent relation is selected FOR UPDATE/SHARE, we need to mark its
126  * PlanRowMark as isParent = true, and generate a new PlanRowMark for each
127  * child.
128  */
129  oldrc = get_plan_rowmark(root->rowMarks, rti);
130  if (oldrc)
131  {
132  old_isParent = oldrc->isParent;
133  oldrc->isParent = true;
134  /* Save initial value of allMarkTypes before children add to it */
135  old_allMarkTypes = oldrc->allMarkTypes;
136  }
137 
138  /* Scan the inheritance set and expand it */
139  if (oldrelation->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
140  {
141  RTEPermissionInfo *perminfo;
142 
143  perminfo = getRTEPermissionInfo(root->parse->rteperminfos, rte);
144 
145  /*
146  * Partitioned table, so set up for partitioning.
147  */
148  Assert(rte->relkind == RELKIND_PARTITIONED_TABLE);
149 
150  /*
151  * Recursively expand and lock the partitions. While at it, also
152  * extract the partition key columns of all the partitioned tables.
153  */
154  expand_partitioned_rtentry(root, rel, rte, rti,
155  oldrelation,
156  perminfo->updatedCols,
157  oldrc, lockmode);
158  }
159  else
160  {
161  /*
162  * Ordinary table, so process traditional-inheritance children. (Note
163  * that partitioned tables are not allowed to have inheritance
164  * children, so it's not possible for both cases to apply.)
165  */
166  List *inhOIDs;
167  ListCell *l;
168 
169  /* Scan for all members of inheritance set, acquire needed locks */
170  inhOIDs = find_all_inheritors(parentOID, lockmode, NULL);
171 
172  /*
173  * We used to special-case the situation where the table no longer has
174  * any children, by clearing rte->inh and exiting. That no longer
175  * works, because this function doesn't get run until after decisions
176  * have been made that depend on rte->inh. We have to treat such
177  * situations as normal inheritance. The table itself should always
178  * have been found, though.
179  */
180  Assert(inhOIDs != NIL);
181  Assert(linitial_oid(inhOIDs) == parentOID);
182 
183  /* Expand simple_rel_array and friends to hold child objects. */
184  expand_planner_arrays(root, list_length(inhOIDs));
185 
186  /*
187  * Expand inheritance children in the order the OIDs were returned by
188  * find_all_inheritors.
189  */
190  foreach(l, inhOIDs)
191  {
192  Oid childOID = lfirst_oid(l);
193  Relation newrelation;
194  RangeTblEntry *childrte;
195  Index childRTindex;
196 
197  /* Open rel if needed; we already have required locks */
198  if (childOID != parentOID)
199  newrelation = table_open(childOID, NoLock);
200  else
201  newrelation = oldrelation;
202 
203  /*
204  * It is possible that the parent table has children that are temp
205  * tables of other backends. We cannot safely access such tables
206  * (because of buffering issues), and the best thing to do seems
207  * to be to silently ignore them.
208  */
209  if (childOID != parentOID && RELATION_IS_OTHER_TEMP(newrelation))
210  {
211  table_close(newrelation, lockmode);
212  continue;
213  }
214 
215  /* Create RTE and AppendRelInfo, plus PlanRowMark if needed. */
216  expand_single_inheritance_child(root, rte, rti, oldrelation,
217  oldrc, newrelation,
218  &childrte, &childRTindex);
219 
220  /* Create the otherrel RelOptInfo too. */
221  (void) build_simple_rel(root, childRTindex, rel);
222 
223  /* Close child relations, but keep locks */
224  if (childOID != parentOID)
225  table_close(newrelation, NoLock);
226  }
227  }
228 
229  /*
230  * Some children might require different mark types, which would've been
231  * reported into oldrc. If so, add relevant entries to the top-level
232  * targetlist and update parent rel's reltarget. This should match what
233  * preprocess_targetlist() would have added if the mark types had been
234  * requested originally.
235  *
236  * (Someday it might be useful to fold these resjunk columns into the
237  * row-identity-column management used for UPDATE/DELETE. Today is not
238  * that day, however.)
239  */
240  if (oldrc)
241  {
242  int new_allMarkTypes = oldrc->allMarkTypes;
243  Var *var;
244  TargetEntry *tle;
245  char resname[32];
246  List *newvars = NIL;
247 
248  /* Add TID junk Var if needed, unless we had it already */
249  if (new_allMarkTypes & ~(1 << ROW_MARK_COPY) &&
250  !(old_allMarkTypes & ~(1 << ROW_MARK_COPY)))
251  {
252  /* Need to fetch TID */
253  var = makeVar(oldrc->rti,
255  TIDOID,
256  -1,
257  InvalidOid,
258  0);
259  snprintf(resname, sizeof(resname), "ctid%u", oldrc->rowmarkId);
260  tle = makeTargetEntry((Expr *) var,
261  list_length(root->processed_tlist) + 1,
262  pstrdup(resname),
263  true);
264  root->processed_tlist = lappend(root->processed_tlist, tle);
265  newvars = lappend(newvars, var);
266  }
267 
268  /* Add whole-row junk Var if needed, unless we had it already */
269  if ((new_allMarkTypes & (1 << ROW_MARK_COPY)) &&
270  !(old_allMarkTypes & (1 << ROW_MARK_COPY)))
271  {
272  var = makeWholeRowVar(planner_rt_fetch(oldrc->rti, root),
273  oldrc->rti,
274  0,
275  false);
276  snprintf(resname, sizeof(resname), "wholerow%u", oldrc->rowmarkId);
277  tle = makeTargetEntry((Expr *) var,
278  list_length(root->processed_tlist) + 1,
279  pstrdup(resname),
280  true);
281  root->processed_tlist = lappend(root->processed_tlist, tle);
282  newvars = lappend(newvars, var);
283  }
284 
285  /* Add tableoid junk Var, unless we had it already */
286  if (!old_isParent)
287  {
288  var = makeVar(oldrc->rti,
290  OIDOID,
291  -1,
292  InvalidOid,
293  0);
294  snprintf(resname, sizeof(resname), "tableoid%u", oldrc->rowmarkId);
295  tle = makeTargetEntry((Expr *) var,
296  list_length(root->processed_tlist) + 1,
297  pstrdup(resname),
298  true);
299  root->processed_tlist = lappend(root->processed_tlist, tle);
300  newvars = lappend(newvars, var);
301  }
302 
303  /*
304  * Add the newly added Vars to parent's reltarget. We needn't worry
305  * about the children's reltargets, they'll be made later.
306  */
307  add_vars_to_targetlist(root, newvars, bms_make_singleton(0));
308  }
309 
310  table_close(oldrelation, NoLock);
311 }
Bitmapset * bms_make_singleton(int x)
Definition: bitmapset.c:216
static void expand_partitioned_rtentry(PlannerInfo *root, RelOptInfo *relinfo, RangeTblEntry *parentrte, Index parentRTindex, Relation parentrel, Bitmapset *parent_updatedCols, PlanRowMark *top_parentrc, LOCKMODE lockmode)
Definition: inherit.c:318
static void expand_appendrel_subquery(PlannerInfo *root, RelOptInfo *rel, RangeTblEntry *rte, Index rti)
Definition: inherit.c:791
static void expand_single_inheritance_child(PlannerInfo *root, RangeTblEntry *parentrte, Index parentRTindex, Relation parentrel, PlanRowMark *top_parentrc, Relation childrel, RangeTblEntry **childrte_p, Index *childRTindex_p)
Definition: inherit.c:453
void add_vars_to_targetlist(PlannerInfo *root, List *vars, Relids where_needed)
Definition: initsplan.c:278
int LOCKMODE
Definition: lockdefs.h:26
#define NoLock
Definition: lockdefs.h:34
Var * makeWholeRowVar(RangeTblEntry *rte, int varno, Index varlevelsup, bool allowScalar)
Definition: makefuncs.c:135
TargetEntry * makeTargetEntry(Expr *expr, AttrNumber resno, char *resname, bool resjunk)
Definition: makefuncs.c:240
Var * makeVar(int varno, AttrNumber varattno, Oid vartype, int32 vartypmod, Oid varcollid, Index varlevelsup)
Definition: makefuncs.c:66
char * pstrdup(const char *in)
Definition: mcxt.c:1683
RTEPermissionInfo * getRTEPermissionInfo(List *rteperminfos, RangeTblEntry *rte)
@ RTE_SUBQUERY
Definition: parsenodes.h:1012
@ RTE_RELATION
Definition: parsenodes.h:1011
#define planner_rt_fetch(rti, root)
Definition: pathnodes.h:555
List * find_all_inheritors(Oid parentrelId, LOCKMODE lockmode, List **numparents)
Definition: pg_inherits.c:255
static int list_length(const List *l)
Definition: pg_list.h:152
#define linitial_oid(l)
Definition: pg_list.h:180
#define lfirst_oid(lc)
Definition: pg_list.h:174
@ ROW_MARK_COPY
Definition: plannodes.h:1332
#define snprintf
Definition: port.h:238
#define InvalidOid
Definition: postgres_ext.h:36
unsigned int Oid
Definition: postgres_ext.h:31
PlanRowMark * get_plan_rowmark(List *rowmarks, Index rtindex)
Definition: preptlist.c:485
#define RELATION_IS_OTHER_TEMP(relation)
Definition: rel.h:658
void expand_planner_arrays(PlannerInfo *root, int add_size)
Definition: relnode.c:163
bool isParent
Definition: plannodes.h:1387
Index rowmarkId
Definition: plannodes.h:1382
int allMarkTypes
Definition: plannodes.h:1384
List * processed_tlist
Definition: pathnodes.h:453
Query * parse
Definition: pathnodes.h:199
List * rowMarks
Definition: pathnodes.h:368
Bitmapset * updatedCols
Definition: parsenodes.h:1254
RTEKind rtekind
Definition: parsenodes.h:1030
Form_pg_class rd_rel
Definition: rel.h:111
Definition: primnodes.h:234
#define TableOidAttributeNumber
Definition: sysattr.h:26
#define SelfItemPointerAttributeNumber
Definition: sysattr.h:21
void table_close(Relation relation, LOCKMODE lockmode)
Definition: table.c:126
Relation table_open(Oid relationId, LOCKMODE lockmode)
Definition: table.c:40

References add_vars_to_targetlist(), PlanRowMark::allMarkTypes, Assert(), bms_make_singleton(), build_simple_rel(), expand_appendrel_subquery(), expand_partitioned_rtentry(), expand_planner_arrays(), expand_single_inheritance_child(), find_all_inheritors(), get_plan_rowmark(), getRTEPermissionInfo(), RangeTblEntry::inh, InvalidOid, PlanRowMark::isParent, lappend(), lfirst_oid, linitial_oid, list_length(), makeTargetEntry(), makeVar(), makeWholeRowVar(), NIL, NoLock, PlannerInfo::parse, planner_rt_fetch, PlannerInfo::processed_tlist, pstrdup(), RelationData::rd_rel, RELATION_IS_OTHER_TEMP, RangeTblEntry::relid, RangeTblEntry::relkind, RangeTblEntry::rellockmode, ROW_MARK_COPY, PlanRowMark::rowmarkId, PlannerInfo::rowMarks, RTE_RELATION, RTE_SUBQUERY, RangeTblEntry::rtekind, PlanRowMark::rti, SelfItemPointerAttributeNumber, snprintf, table_close(), table_open(), TableOidAttributeNumber, and RTEPermissionInfo::updatedCols.

Referenced by add_other_rels_to_query(), and expand_appendrel_subquery().

◆ expand_partitioned_rtentry()

static void expand_partitioned_rtentry ( PlannerInfo root,
RelOptInfo relinfo,
RangeTblEntry parentrte,
Index  parentRTindex,
Relation  parentrel,
Bitmapset parent_updatedCols,
PlanRowMark top_parentrc,
LOCKMODE  lockmode 
)
static

Definition at line 318 of file inherit.c.

323 {
324  PartitionDesc partdesc;
325  Bitmapset *live_parts;
326  int num_live_parts;
327  int i;
328 
330 
331  Assert(parentrte->inh);
332 
333  partdesc = PartitionDirectoryLookup(root->glob->partition_directory,
334  parentrel);
335 
336  /* A partitioned table should always have a partition descriptor. */
337  Assert(partdesc);
338 
339  /*
340  * Note down whether any partition key cols are being updated. Though it's
341  * the root partitioned table's updatedCols we are interested in,
342  * parent_updatedCols provided by the caller contains the root partrel's
343  * updatedCols translated to match the attribute ordering of parentrel.
344  */
345  if (!root->partColsUpdated)
346  root->partColsUpdated =
347  has_partition_attrs(parentrel, parent_updatedCols, NULL);
348 
349  /* Nothing further to do here if there are no partitions. */
350  if (partdesc->nparts == 0)
351  return;
352 
353  /*
354  * Perform partition pruning using restriction clauses assigned to parent
355  * relation. live_parts will contain PartitionDesc indexes of partitions
356  * that survive pruning. Below, we will initialize child objects for the
357  * surviving partitions.
358  */
359  relinfo->live_parts = live_parts = prune_append_rel_partitions(relinfo);
360 
361  /* Expand simple_rel_array and friends to hold child objects. */
362  num_live_parts = bms_num_members(live_parts);
363  if (num_live_parts > 0)
364  expand_planner_arrays(root, num_live_parts);
365 
366  /*
367  * We also store partition RelOptInfo pointers in the parent relation.
368  * Since we're palloc0'ing, slots corresponding to pruned partitions will
369  * contain NULL.
370  */
371  Assert(relinfo->part_rels == NULL);
372  relinfo->part_rels = (RelOptInfo **)
373  palloc0(relinfo->nparts * sizeof(RelOptInfo *));
374 
375  /*
376  * Create a child RTE for each live partition. Note that unlike
377  * traditional inheritance, we don't need a child RTE for the partitioned
378  * table itself, because it's not going to be scanned.
379  */
380  i = -1;
381  while ((i = bms_next_member(live_parts, i)) >= 0)
382  {
383  Oid childOID = partdesc->oids[i];
384  Relation childrel;
385  RangeTblEntry *childrte;
386  Index childRTindex;
387  RelOptInfo *childrelinfo;
388 
389  /* Open rel, acquiring required locks */
390  childrel = table_open(childOID, lockmode);
391 
392  /*
393  * Temporary partitions belonging to other sessions should have been
394  * disallowed at definition, but for paranoia's sake, let's double
395  * check.
396  */
397  if (RELATION_IS_OTHER_TEMP(childrel))
398  elog(ERROR, "temporary relation from another session found as partition");
399 
400  /* Create RTE and AppendRelInfo, plus PlanRowMark if needed. */
401  expand_single_inheritance_child(root, parentrte, parentRTindex,
402  parentrel, top_parentrc, childrel,
403  &childrte, &childRTindex);
404 
405  /* Create the otherrel RelOptInfo too. */
406  childrelinfo = build_simple_rel(root, childRTindex, relinfo);
407  relinfo->part_rels[i] = childrelinfo;
408  relinfo->all_partrels = bms_add_members(relinfo->all_partrels,
409  childrelinfo->relids);
410 
411  /* If this child is itself partitioned, recurse */
412  if (childrel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
413  {
414  AppendRelInfo *appinfo = root->append_rel_array[childRTindex];
415  Bitmapset *child_updatedCols;
416 
417  child_updatedCols = translate_col_privs(parent_updatedCols,
418  appinfo->translated_vars);
419 
420  expand_partitioned_rtentry(root, childrelinfo,
421  childrte, childRTindex,
422  childrel,
423  child_updatedCols,
424  top_parentrc, lockmode);
425  }
426 
427  /* Close child relation, but keep locks */
428  table_close(childrel, NoLock);
429  }
430 }
int bms_next_member(const Bitmapset *a, int prevbit)
Definition: bitmapset.c:1306
int bms_num_members(const Bitmapset *a)
Definition: bitmapset.c:751
Bitmapset * bms_add_members(Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:917
#define ERROR
Definition: elog.h:39
#define elog(elevel,...)
Definition: elog.h:224
static Bitmapset * translate_col_privs(const Bitmapset *parent_privs, List *translated_vars)
Definition: inherit.c:702
int i
Definition: isn.c:73
void * palloc0(Size size)
Definition: mcxt.c:1334
PartitionDesc PartitionDirectoryLookup(PartitionDirectory pdir, Relation rel)
Definition: partdesc.c:414
bool has_partition_attrs(Relation rel, Bitmapset *attnums, bool *used_in_expr)
Definition: partition.c:255
Bitmapset * prune_append_rel_partitions(RelOptInfo *rel)
Definition: partprune.c:750
void check_stack_depth(void)
Definition: postgres.c:3531
List * translated_vars
Definition: pathnodes.h:2971
bool partColsUpdated
Definition: pathnodes.h:546
PlannerGlobal * glob
Definition: pathnodes.h:202
Relids relids
Definition: pathnodes.h:856
Relids all_partrels
Definition: pathnodes.h:1022
Bitmapset * live_parts
Definition: pathnodes.h:1020

References RelOptInfo::all_partrels, Assert(), bms_add_members(), bms_next_member(), bms_num_members(), build_simple_rel(), check_stack_depth(), elog, ERROR, expand_planner_arrays(), expand_single_inheritance_child(), PlannerInfo::glob, has_partition_attrs(), i, RangeTblEntry::inh, RelOptInfo::live_parts, NoLock, RelOptInfo::nparts, PartitionDescData::nparts, PartitionDescData::oids, palloc0(), PlannerInfo::partColsUpdated, PartitionDirectoryLookup(), prune_append_rel_partitions(), RelationData::rd_rel, RELATION_IS_OTHER_TEMP, RelOptInfo::relids, table_close(), table_open(), translate_col_privs(), and AppendRelInfo::translated_vars.

Referenced by expand_inherited_rtentry().

◆ expand_single_inheritance_child()

static void expand_single_inheritance_child ( PlannerInfo root,
RangeTblEntry parentrte,
Index  parentRTindex,
Relation  parentrel,
PlanRowMark top_parentrc,
Relation  childrel,
RangeTblEntry **  childrte_p,
Index childRTindex_p 
)
static

Definition at line 453 of file inherit.c.

458 {
459  Query *parse = root->parse;
460  Oid parentOID PG_USED_FOR_ASSERTS_ONLY =
461  RelationGetRelid(parentrel);
462  Oid childOID = RelationGetRelid(childrel);
463  RangeTblEntry *childrte;
464  Index childRTindex;
465  AppendRelInfo *appinfo;
466  TupleDesc child_tupdesc;
467  List *parent_colnames;
468  List *child_colnames;
469 
470  /*
471  * Build an RTE for the child, and attach to query's rangetable list. We
472  * copy most scalar fields of the parent's RTE, but replace relation OID,
473  * relkind, and inh for the child. Set the child's securityQuals to
474  * empty, because we only want to apply the parent's RLS conditions
475  * regardless of what RLS properties individual children may have. (This
476  * is an intentional choice to make inherited RLS work like regular
477  * permissions checks.) The parent securityQuals will be propagated to
478  * children along with other base restriction clauses, so we don't need to
479  * do it here. Other infrastructure of the parent RTE has to be
480  * translated to match the child table's column ordering, which we do
481  * below, so a "flat" copy is sufficient to start with.
482  */
483  childrte = makeNode(RangeTblEntry);
484  memcpy(childrte, parentrte, sizeof(RangeTblEntry));
485  Assert(parentrte->rtekind == RTE_RELATION); /* else this is dubious */
486  childrte->relid = childOID;
487  childrte->relkind = childrel->rd_rel->relkind;
488  /* A partitioned child will need to be expanded further. */
489  if (childrte->relkind == RELKIND_PARTITIONED_TABLE)
490  {
491  Assert(childOID != parentOID);
492  childrte->inh = true;
493  }
494  else
495  childrte->inh = false;
496  childrte->securityQuals = NIL;
497 
498  /* No permission checking for child RTEs. */
499  childrte->perminfoindex = 0;
500 
501  /* Link not-yet-fully-filled child RTE into data structures */
502  parse->rtable = lappend(parse->rtable, childrte);
503  childRTindex = list_length(parse->rtable);
504  *childrte_p = childrte;
505  *childRTindex_p = childRTindex;
506 
507  /*
508  * Build an AppendRelInfo struct for each parent/child pair.
509  */
510  appinfo = make_append_rel_info(parentrel, childrel,
511  parentRTindex, childRTindex);
512  root->append_rel_list = lappend(root->append_rel_list, appinfo);
513 
514  /* tablesample is probably null, but copy it */
515  childrte->tablesample = copyObject(parentrte->tablesample);
516 
517  /*
518  * Construct an alias clause for the child, which we can also use as eref.
519  * This is important so that EXPLAIN will print the right column aliases
520  * for child-table columns. (Since ruleutils.c doesn't have any easy way
521  * to reassociate parent and child columns, we must get the child column
522  * aliases right to start with. Note that setting childrte->alias forces
523  * ruleutils.c to use these column names, which it otherwise would not.)
524  */
525  child_tupdesc = RelationGetDescr(childrel);
526  parent_colnames = parentrte->eref->colnames;
527  child_colnames = NIL;
528  for (int cattno = 0; cattno < child_tupdesc->natts; cattno++)
529  {
530  Form_pg_attribute att = TupleDescAttr(child_tupdesc, cattno);
531  const char *attname;
532 
533  if (att->attisdropped)
534  {
535  /* Always insert an empty string for a dropped column */
536  attname = "";
537  }
538  else if (appinfo->parent_colnos[cattno] > 0 &&
539  appinfo->parent_colnos[cattno] <= list_length(parent_colnames))
540  {
541  /* Duplicate the query-assigned name for the parent column */
542  attname = strVal(list_nth(parent_colnames,
543  appinfo->parent_colnos[cattno] - 1));
544  }
545  else
546  {
547  /* New column, just use its real name */
548  attname = NameStr(att->attname);
549  }
550  child_colnames = lappend(child_colnames, makeString(pstrdup(attname)));
551  }
552 
553  /*
554  * We just duplicate the parent's table alias name for each child. If the
555  * plan gets printed, ruleutils.c has to sort out unique table aliases to
556  * use, which it can handle.
557  */
558  childrte->alias = childrte->eref = makeAlias(parentrte->eref->aliasname,
559  child_colnames);
560 
561  /*
562  * Store the RTE and appinfo in the respective PlannerInfo arrays, which
563  * the caller must already have allocated space for.
564  */
565  Assert(childRTindex < root->simple_rel_array_size);
566  Assert(root->simple_rte_array[childRTindex] == NULL);
567  root->simple_rte_array[childRTindex] = childrte;
568  Assert(root->append_rel_array[childRTindex] == NULL);
569  root->append_rel_array[childRTindex] = appinfo;
570 
571  /*
572  * Build a PlanRowMark if parent is marked FOR UPDATE/SHARE.
573  */
574  if (top_parentrc)
575  {
576  PlanRowMark *childrc = makeNode(PlanRowMark);
577 
578  childrc->rti = childRTindex;
579  childrc->prti = top_parentrc->rti;
580  childrc->rowmarkId = top_parentrc->rowmarkId;
581  /* Reselect rowmark type, because relkind might not match parent */
582  childrc->markType = select_rowmark_type(childrte,
583  top_parentrc->strength);
584  childrc->allMarkTypes = (1 << childrc->markType);
585  childrc->strength = top_parentrc->strength;
586  childrc->waitPolicy = top_parentrc->waitPolicy;
587 
588  /*
589  * We mark RowMarks for partitioned child tables as parent RowMarks so
590  * that the executor ignores them (except their existence means that
591  * the child tables will be locked using the appropriate mode).
592  */
593  childrc->isParent = (childrte->relkind == RELKIND_PARTITIONED_TABLE);
594 
595  /* Include child's rowmark type in top parent's allMarkTypes */
596  top_parentrc->allMarkTypes |= childrc->allMarkTypes;
597 
598  root->rowMarks = lappend(root->rowMarks, childrc);
599  }
600 
601  /*
602  * If we are creating a child of the query target relation (only possible
603  * in UPDATE/DELETE/MERGE), add it to all_result_relids, as well as
604  * leaf_result_relids if appropriate, and make sure that we generate
605  * required row-identity data.
606  */
607  if (bms_is_member(parentRTindex, root->all_result_relids))
608  {
609  /* OK, record the child as a result rel too. */
611  childRTindex);
612 
613  /* Non-leaf partitions don't need any row identity info. */
614  if (childrte->relkind != RELKIND_PARTITIONED_TABLE)
615  {
616  Var *rrvar;
617 
619  childRTindex);
620 
621  /*
622  * If we have any child target relations, assume they all need to
623  * generate a junk "tableoid" column. (If only one child survives
624  * pruning, we wouldn't really need this, but it's not worth
625  * thrashing about to avoid it.)
626  */
627  rrvar = makeVar(childRTindex,
629  OIDOID,
630  -1,
631  InvalidOid,
632  0);
633  add_row_identity_var(root, rrvar, childRTindex, "tableoid");
634 
635  /* Register any row-identity columns needed by this child. */
636  add_row_identity_columns(root, childRTindex,
637  childrte, childrel);
638  }
639  }
640 }
void add_row_identity_columns(PlannerInfo *root, Index rtindex, RangeTblEntry *target_rte, Relation target_relation)
Definition: appendinfo.c:884
AppendRelInfo * make_append_rel_info(Relation parentrel, Relation childrel, Index parentRTindex, Index childRTindex)
Definition: appendinfo.c:51
void add_row_identity_var(PlannerInfo *root, Var *orig_var, Index rtindex, const char *rowid_name)
Definition: appendinfo.c:789
bool bms_is_member(int x, const Bitmapset *a)
Definition: bitmapset.c:510
Bitmapset * bms_add_member(Bitmapset *a, int x)
Definition: bitmapset.c:815
#define NameStr(name)
Definition: c.h:733
#define PG_USED_FOR_ASSERTS_ONLY
Definition: c.h:169
Alias * makeAlias(const char *aliasname, List *colnames)
Definition: makefuncs.c:389
#define copyObject(obj)
Definition: nodes.h:223
#define makeNode(_type_)
Definition: nodes.h:155
NameData attname
Definition: pg_attribute.h:41
FormData_pg_attribute * Form_pg_attribute
Definition: pg_attribute.h:209
static void * list_nth(const List *list, int n)
Definition: pg_list.h:299
RowMarkType select_rowmark_type(RangeTblEntry *rte, LockClauseStrength strength)
Definition: planner.c:2345
static struct subre * parse(struct vars *v, int stopper, int type, struct state *init, struct state *final)
Definition: regcomp.c:715
#define RelationGetRelid(relation)
Definition: rel.h:505
#define RelationGetDescr(relation)
Definition: rel.h:531
char * aliasname
Definition: primnodes.h:50
List * colnames
Definition: primnodes.h:51
LockClauseStrength strength
Definition: plannodes.h:1385
Index prti
Definition: plannodes.h:1381
RowMarkType markType
Definition: plannodes.h:1383
LockWaitPolicy waitPolicy
Definition: plannodes.h:1386
Relids all_result_relids
Definition: pathnodes.h:351
Relids leaf_result_relids
Definition: pathnodes.h:353
struct TableSampleClause * tablesample
Definition: parsenodes.h:1081
Alias * eref
Definition: parsenodes.h:1205
Alias * alias
Definition: parsenodes.h:1204
Index perminfoindex
Definition: parsenodes.h:1080
#define TupleDescAttr(tupdesc, i)
Definition: tupdesc.h:92
String * makeString(char *str)
Definition: value.c:63
#define strVal(v)
Definition: value.h:82

References add_row_identity_columns(), add_row_identity_var(), RangeTblEntry::alias, Alias::aliasname, PlannerInfo::all_result_relids, PlanRowMark::allMarkTypes, PlannerInfo::append_rel_list, Assert(), attname, bms_add_member(), bms_is_member(), Alias::colnames, copyObject, RangeTblEntry::eref, RangeTblEntry::inh, InvalidOid, PlanRowMark::isParent, lappend(), PlannerInfo::leaf_result_relids, list_length(), list_nth(), make_append_rel_info(), makeAlias(), makeNode, makeString(), makeVar(), PlanRowMark::markType, NameStr, TupleDescData::natts, NIL, parse(), PlannerInfo::parse, RangeTblEntry::perminfoindex, PG_USED_FOR_ASSERTS_ONLY, PlanRowMark::prti, pstrdup(), RelationData::rd_rel, RelationGetDescr, RelationGetRelid, RangeTblEntry::relid, RangeTblEntry::relkind, PlanRowMark::rowmarkId, PlannerInfo::rowMarks, RTE_RELATION, RangeTblEntry::rtekind, PlanRowMark::rti, RangeTblEntry::securityQuals, select_rowmark_type(), PlanRowMark::strength, strVal, TableOidAttributeNumber, RangeTblEntry::tablesample, TupleDescAttr, and PlanRowMark::waitPolicy.

Referenced by expand_inherited_rtentry(), and expand_partitioned_rtentry().

◆ get_rel_all_updated_cols()

Bitmapset* get_rel_all_updated_cols ( PlannerInfo root,
RelOptInfo rel 
)

Definition at line 648 of file inherit.c.

649 {
650  Index relid;
651  RangeTblEntry *rte;
652  RTEPermissionInfo *perminfo;
653  Bitmapset *updatedCols,
654  *extraUpdatedCols;
655 
656  Assert(root->parse->commandType == CMD_UPDATE);
657  Assert(IS_SIMPLE_REL(rel));
658 
659  /*
660  * We obtain updatedCols for the query's result relation. Then, if
661  * necessary, we map it to the column numbers of the relation for which
662  * they were requested.
663  */
664  relid = root->parse->resultRelation;
665  rte = planner_rt_fetch(relid, root);
666  perminfo = getRTEPermissionInfo(root->parse->rteperminfos, rte);
667 
668  updatedCols = perminfo->updatedCols;
669 
670  if (rel->relid != relid)
671  {
672  RelOptInfo *top_parent_rel = find_base_rel(root, relid);
673 
674  Assert(IS_OTHER_REL(rel));
675 
676  updatedCols = translate_col_privs_multilevel(root, rel, top_parent_rel,
677  updatedCols);
678  }
679 
680  /*
681  * Now we must check to see if there are any generated columns that depend
682  * on the updatedCols, and add them to the result.
683  */
684  extraUpdatedCols = get_dependent_generated_columns(root, rel->relid,
685  updatedCols);
686 
687  return bms_union(updatedCols, extraUpdatedCols);
688 }
Bitmapset * bms_union(const Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:251
static Bitmapset * translate_col_privs_multilevel(PlannerInfo *root, RelOptInfo *rel, RelOptInfo *parent_rel, Bitmapset *parent_cols)
Definition: inherit.c:752
@ CMD_UPDATE
Definition: nodes.h:256
#define IS_SIMPLE_REL(rel)
Definition: pathnodes.h:824
#define IS_OTHER_REL(rel)
Definition: pathnodes.h:839
Bitmapset * get_dependent_generated_columns(PlannerInfo *root, Index rti, Bitmapset *target_cols)
Definition: plancat.c:2289
RelOptInfo * find_base_rel(PlannerInfo *root, int relid)
Definition: relnode.c:407
CmdType commandType
Definition: parsenodes.h:121
Index relid
Definition: pathnodes.h:903

References Assert(), bms_union(), CMD_UPDATE, Query::commandType, find_base_rel(), get_dependent_generated_columns(), getRTEPermissionInfo(), IS_OTHER_REL, IS_SIMPLE_REL, PlannerInfo::parse, planner_rt_fetch, RelOptInfo::relid, translate_col_privs_multilevel(), and RTEPermissionInfo::updatedCols.

Referenced by postgresPlanForeignModify().

◆ translate_col_privs()

static Bitmapset * translate_col_privs ( const Bitmapset parent_privs,
List translated_vars 
)
static

Definition at line 702 of file inherit.c.

704 {
705  Bitmapset *child_privs = NULL;
706  bool whole_row;
707  int attno;
708  ListCell *lc;
709 
710  /* System attributes have the same numbers in all tables */
711  for (attno = FirstLowInvalidHeapAttributeNumber + 1; attno < 0; attno++)
712  {
714  parent_privs))
715  child_privs = bms_add_member(child_privs,
717  }
718 
719  /* Check if parent has whole-row reference */
721  parent_privs);
722 
723  /* And now translate the regular user attributes, using the vars list */
724  attno = InvalidAttrNumber;
725  foreach(lc, translated_vars)
726  {
727  Var *var = lfirst_node(Var, lc);
728 
729  attno++;
730  if (var == NULL) /* ignore dropped columns */
731  continue;
732  if (whole_row ||
734  parent_privs))
735  child_privs = bms_add_member(child_privs,
737  }
738 
739  return child_privs;
740 }
#define InvalidAttrNumber
Definition: attnum.h:23
#define lfirst_node(type, lc)
Definition: pg_list.h:176
AttrNumber varattno
Definition: primnodes.h:246
#define FirstLowInvalidHeapAttributeNumber
Definition: sysattr.h:27

References bms_add_member(), bms_is_member(), FirstLowInvalidHeapAttributeNumber, InvalidAttrNumber, lfirst_node, and Var::varattno.

Referenced by expand_partitioned_rtentry(), and translate_col_privs_multilevel().

◆ translate_col_privs_multilevel()

static Bitmapset * translate_col_privs_multilevel ( PlannerInfo root,
RelOptInfo rel,
RelOptInfo parent_rel,
Bitmapset parent_cols 
)
static

Definition at line 752 of file inherit.c.

755 {
756  AppendRelInfo *appinfo;
757 
758  /* Fast path for easy case. */
759  if (parent_cols == NULL)
760  return NULL;
761 
762  /* Recurse if immediate parent is not the top parent. */
763  if (rel->parent != parent_rel)
764  {
765  if (rel->parent)
766  parent_cols = translate_col_privs_multilevel(root, rel->parent,
767  parent_rel,
768  parent_cols);
769  else
770  elog(ERROR, "rel with relid %u is not a child rel", rel->relid);
771  }
772 
773  /* Now translate for this child. */
774  Assert(root->append_rel_array != NULL);
775  appinfo = root->append_rel_array[rel->relid];
776  Assert(appinfo != NULL);
777 
778  return translate_col_privs(parent_cols, appinfo->translated_vars);
779 }

References Assert(), elog, ERROR, RelOptInfo::relid, translate_col_privs(), and AppendRelInfo::translated_vars.

Referenced by get_rel_all_updated_cols().