Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * heap.c
4 : : * code to create and destroy POSTGRES heap relations
5 : : *
6 : : * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
7 : : * Portions Copyright (c) 1994, Regents of the University of California
8 : : *
9 : : *
10 : : * IDENTIFICATION
11 : : * src/backend/catalog/heap.c
12 : : *
13 : : *
14 : : * INTERFACE ROUTINES
15 : : * heap_create() - Create an uncataloged heap relation
16 : : * heap_create_with_catalog() - Create a cataloged relation
17 : : * heap_drop_with_catalog() - Removes named relation from catalogs
18 : : *
19 : : * NOTES
20 : : * this code taken from access/heap/create.c, which contains
21 : : * the old heap_create_with_catalog, amcreate, and amdestroy.
22 : : * those routines will soon call these routines using the function
23 : : * manager,
24 : : * just like the poorly named "NewXXX" routines do. The
25 : : * "New" routines are all going to die soon, once and for all!
26 : : * -cim 1/13/91
27 : : *
28 : : *-------------------------------------------------------------------------
29 : : */
30 : : #include "postgres.h"
31 : :
32 : : #include "access/genam.h"
33 : : #include "access/multixact.h"
34 : : #include "access/relation.h"
35 : : #include "access/table.h"
36 : : #include "access/tableam.h"
37 : : #include "catalog/binary_upgrade.h"
38 : : #include "catalog/catalog.h"
39 : : #include "catalog/heap.h"
40 : : #include "catalog/index.h"
41 : : #include "catalog/objectaccess.h"
42 : : #include "catalog/partition.h"
43 : : #include "catalog/pg_am.h"
44 : : #include "catalog/pg_attrdef.h"
45 : : #include "catalog/pg_collation.h"
46 : : #include "catalog/pg_constraint.h"
47 : : #include "catalog/pg_foreign_table.h"
48 : : #include "catalog/pg_inherits.h"
49 : : #include "catalog/pg_namespace.h"
50 : : #include "catalog/pg_opclass.h"
51 : : #include "catalog/pg_partitioned_table.h"
52 : : #include "catalog/pg_statistic.h"
53 : : #include "catalog/pg_subscription_rel.h"
54 : : #include "catalog/pg_tablespace.h"
55 : : #include "catalog/pg_type.h"
56 : : #include "catalog/storage.h"
57 : : #include "commands/tablecmds.h"
58 : : #include "commands/typecmds.h"
59 : : #include "common/int.h"
60 : : #include "miscadmin.h"
61 : : #include "nodes/nodeFuncs.h"
62 : : #include "optimizer/optimizer.h"
63 : : #include "parser/parse_coerce.h"
64 : : #include "parser/parse_collate.h"
65 : : #include "parser/parse_expr.h"
66 : : #include "parser/parse_relation.h"
67 : : #include "parser/parsetree.h"
68 : : #include "partitioning/partdesc.h"
69 : : #include "pgstat.h"
70 : : #include "storage/lmgr.h"
71 : : #include "storage/predicate.h"
72 : : #include "utils/array.h"
73 : : #include "utils/builtins.h"
74 : : #include "utils/fmgroids.h"
75 : : #include "utils/inval.h"
76 : : #include "utils/lsyscache.h"
77 : : #include "utils/syscache.h"
78 : :
79 : :
80 : : /* Potentially set by pg_upgrade_support functions */
81 : : Oid binary_upgrade_next_heap_pg_class_oid = InvalidOid;
82 : : Oid binary_upgrade_next_toast_pg_class_oid = InvalidOid;
83 : : RelFileNumber binary_upgrade_next_heap_pg_class_relfilenumber = InvalidRelFileNumber;
84 : : RelFileNumber binary_upgrade_next_toast_pg_class_relfilenumber = InvalidRelFileNumber;
85 : :
86 : : static void AddNewRelationTuple(Relation pg_class_desc,
87 : : Relation new_rel_desc,
88 : : Oid new_rel_oid,
89 : : Oid new_type_oid,
90 : : Oid reloftype,
91 : : Oid relowner,
92 : : char relkind,
93 : : TransactionId relfrozenxid,
94 : : TransactionId relminmxid,
95 : : Datum relacl,
96 : : Datum reloptions);
97 : : static ObjectAddress AddNewRelationType(const char *typeName,
98 : : Oid typeNamespace,
99 : : Oid new_rel_oid,
100 : : char new_rel_kind,
101 : : Oid ownerid,
102 : : Oid new_row_type,
103 : : Oid new_array_type);
104 : : static void RelationRemoveInheritance(Oid relid);
105 : : static Oid StoreRelCheck(Relation rel, const char *ccname, Node *expr,
106 : : bool is_enforced, bool is_validated, bool is_local,
107 : : int16 inhcount, bool is_no_inherit, bool is_internal);
108 : : static void StoreConstraints(Relation rel, List *cooked_constraints,
109 : : bool is_internal);
110 : : static bool MergeWithExistingConstraint(Relation rel, const char *ccname, Node *expr,
111 : : bool allow_merge, bool is_local,
112 : : bool is_enforced,
113 : : bool is_initially_valid,
114 : : bool is_no_inherit);
115 : : static void SetRelationNumChecks(Relation rel, int numchecks);
116 : : static Node *cookConstraint(ParseState *pstate,
117 : : Node *raw_constraint,
118 : : char *relname);
119 : :
120 : :
121 : : /* ----------------------------------------------------------------
122 : : * XXX UGLY HARD CODED BADNESS FOLLOWS XXX
123 : : *
124 : : * these should all be moved to someplace in the lib/catalog
125 : : * module, if not obliterated first.
126 : : * ----------------------------------------------------------------
127 : : */
128 : :
129 : :
130 : : /*
131 : : * Note:
132 : : * Should the system special case these attributes in the future?
133 : : * Advantage: consume much less space in the ATTRIBUTE relation.
134 : : * Disadvantage: special cases will be all over the place.
135 : : */
136 : :
137 : : /*
138 : : * The initializers below do not include trailing variable length fields,
139 : : * but that's OK - we're never going to reference anything beyond the
140 : : * fixed-size portion of the structure anyway. Fields that can default
141 : : * to zeroes are also not mentioned.
142 : : */
143 : :
144 : : static const FormData_pg_attribute a1 = {
145 : : .attname = {"ctid"},
146 : : .atttypid = TIDOID,
147 : : .attlen = sizeof(ItemPointerData),
148 : : .attnum = SelfItemPointerAttributeNumber,
149 : : .atttypmod = -1,
150 : : .attbyval = false,
151 : : .attalign = TYPALIGN_SHORT,
152 : : .attstorage = TYPSTORAGE_PLAIN,
153 : : .attnotnull = true,
154 : : .attislocal = true,
155 : : };
156 : :
157 : : static const FormData_pg_attribute a2 = {
158 : : .attname = {"xmin"},
159 : : .atttypid = XIDOID,
160 : : .attlen = sizeof(TransactionId),
161 : : .attnum = MinTransactionIdAttributeNumber,
162 : : .atttypmod = -1,
163 : : .attbyval = true,
164 : : .attalign = TYPALIGN_INT,
165 : : .attstorage = TYPSTORAGE_PLAIN,
166 : : .attnotnull = true,
167 : : .attislocal = true,
168 : : };
169 : :
170 : : static const FormData_pg_attribute a3 = {
171 : : .attname = {"cmin"},
172 : : .atttypid = CIDOID,
173 : : .attlen = sizeof(CommandId),
174 : : .attnum = MinCommandIdAttributeNumber,
175 : : .atttypmod = -1,
176 : : .attbyval = true,
177 : : .attalign = TYPALIGN_INT,
178 : : .attstorage = TYPSTORAGE_PLAIN,
179 : : .attnotnull = true,
180 : : .attislocal = true,
181 : : };
182 : :
183 : : static const FormData_pg_attribute a4 = {
184 : : .attname = {"xmax"},
185 : : .atttypid = XIDOID,
186 : : .attlen = sizeof(TransactionId),
187 : : .attnum = MaxTransactionIdAttributeNumber,
188 : : .atttypmod = -1,
189 : : .attbyval = true,
190 : : .attalign = TYPALIGN_INT,
191 : : .attstorage = TYPSTORAGE_PLAIN,
192 : : .attnotnull = true,
193 : : .attislocal = true,
194 : : };
195 : :
196 : : static const FormData_pg_attribute a5 = {
197 : : .attname = {"cmax"},
198 : : .atttypid = CIDOID,
199 : : .attlen = sizeof(CommandId),
200 : : .attnum = MaxCommandIdAttributeNumber,
201 : : .atttypmod = -1,
202 : : .attbyval = true,
203 : : .attalign = TYPALIGN_INT,
204 : : .attstorage = TYPSTORAGE_PLAIN,
205 : : .attnotnull = true,
206 : : .attislocal = true,
207 : : };
208 : :
209 : : /*
210 : : * We decided to call this attribute "tableoid" rather than say
211 : : * "classoid" on the basis that in the future there may be more than one
212 : : * table of a particular class/type. In any case table is still the word
213 : : * used in SQL.
214 : : */
215 : : static const FormData_pg_attribute a6 = {
216 : : .attname = {"tableoid"},
217 : : .atttypid = OIDOID,
218 : : .attlen = sizeof(Oid),
219 : : .attnum = TableOidAttributeNumber,
220 : : .atttypmod = -1,
221 : : .attbyval = true,
222 : : .attalign = TYPALIGN_INT,
223 : : .attstorage = TYPSTORAGE_PLAIN,
224 : : .attnotnull = true,
225 : : .attislocal = true,
226 : : };
227 : :
228 : : static const FormData_pg_attribute *const SysAtt[] = {&a1, &a2, &a3, &a4, &a5, &a6};
229 : :
230 : : /*
231 : : * This function returns a Form_pg_attribute pointer for a system attribute.
232 : : * Note that we elog if the presented attno is invalid, which would only
233 : : * happen if there's a problem upstream.
234 : : */
235 : : const FormData_pg_attribute *
2672 andres@anarazel.de 236 :CBC 20061 : SystemAttributeDefinition(AttrNumber attno)
237 : : {
8907 bruce@momjian.us 238 [ + - - + ]: 20061 : if (attno >= 0 || attno < -(int) lengthof(SysAtt))
8273 tgl@sss.pgh.pa.us 239 [ # # ]:UBC 0 : elog(ERROR, "invalid system attribute number %d", attno);
9078 tgl@sss.pgh.pa.us 240 :CBC 20061 : return SysAtt[-attno - 1];
241 : : }
242 : :
243 : : /*
244 : : * If the given name is a system attribute name, return a Form_pg_attribute
245 : : * pointer for a prototype definition. If not, return NULL.
246 : : */
247 : : const FormData_pg_attribute *
2672 andres@anarazel.de 248 : 189019 : SystemAttributeByName(const char *attname)
249 : : {
250 : : int j;
251 : :
8910 tgl@sss.pgh.pa.us 252 [ + + ]: 1259081 : for (j = 0; j < (int) lengthof(SysAtt); j++)
253 : : {
2707 andres@anarazel.de 254 : 1090146 : const FormData_pg_attribute *att = SysAtt[j];
255 : :
2672 256 [ + + ]: 1090146 : if (strcmp(NameStr(att->attname), attname) == 0)
257 : 20084 : return att;
258 : : }
259 : :
8910 tgl@sss.pgh.pa.us 260 : 168935 : return NULL;
261 : : }
262 : :
263 : :
264 : : /* ----------------------------------------------------------------
265 : : * XXX END OF UGLY HARD CODED BADNESS XXX
266 : : * ---------------------------------------------------------------- */
267 : :
268 : :
269 : : /* ----------------------------------------------------------------
270 : : * heap_create - Create an uncataloged heap relation
271 : : *
272 : : * Note API change: the caller must now always provide the OID
273 : : * to use for the relation. The relfilenumber may be (and in
274 : : * the simplest cases is) left unspecified.
275 : : *
276 : : * create_storage indicates whether or not to create the storage.
277 : : * However, even if create_storage is true, no storage will be
278 : : * created if the relkind is one that doesn't have storage.
279 : : *
280 : : * rel->rd_rel is initialized by RelationBuildLocalRelation,
281 : : * and is mostly zeroes at return.
282 : : * ----------------------------------------------------------------
283 : : */
284 : : Relation
8750 285 : 71482 : heap_create(const char *relname,
286 : : Oid relnamespace,
287 : : Oid reltablespace,
288 : : Oid relid,
289 : : RelFileNumber relfilenumber,
290 : : Oid accessmtd,
291 : : TupleDesc tupDesc,
292 : : char relkind,
293 : : char relpersistence,
294 : : bool shared_relation,
295 : : bool mapped_relation,
296 : : bool allow_system_table_mods,
297 : : TransactionId *relfrozenxid,
298 : : MultiXactId *relminmxid,
299 : : bool create_storage)
300 : : {
301 : : Relation rel;
302 : :
303 : : /* The caller must have provided an OID for the relation. */
7520 304 [ - + ]: 71482 : Assert(OidIsValid(relid));
305 : :
306 : : /*
307 : : * Don't allow creating relations in pg_catalog directly, even though it
308 : : * is allowed to move user defined relations there. Semantics with search
309 : : * paths including pg_catalog are too confusing for now.
310 : : *
311 : : * But allow creating indexes on relations in pg_catalog even if
312 : : * allow_system_table_mods = off, upper layers already guarantee it's on a
313 : : * user defined relation, not a system one.
314 : : */
4668 heikki.linnakangas@i 315 [ + + + + ]: 112534 : if (!allow_system_table_mods &&
2503 tgl@sss.pgh.pa.us 316 [ + + - + ]: 88431 : ((IsCatalogNamespace(relnamespace) && relkind != RELKIND_INDEX) ||
4490 rhaas@postgresql.org 317 : 41048 : IsToastNamespace(relnamespace)) &&
4668 heikki.linnakangas@i 318 [ + - ]: 4 : IsNormalProcessingMode())
319 [ + - ]: 4 : ereport(ERROR,
320 : : (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
321 : : errmsg("permission denied to create \"%s.%s\"",
322 : : get_namespace_name(relnamespace), relname),
323 : : errdetail("System catalog modifications are currently disallowed.")));
324 : :
2544 andres@anarazel.de 325 : 71478 : *relfrozenxid = InvalidTransactionId;
326 : 71478 : *relminmxid = InvalidMultiXactId;
327 : :
328 : : /*
329 : : * Force reltablespace to zero if the relation kind does not support
330 : : * tablespaces. This is mainly just for cleanliness' sake.
331 : : */
1563 peter@eisentraut.org 332 [ + + + + : 71478 : if (!RELKIND_HAS_TABLESPACE(relkind))
+ + + + +
+ + + + +
+ + ]
333 : 12103 : reltablespace = InvalidOid;
334 : :
335 : : /* Don't create storage for relkinds without physical storage. */
1518 rhaas@postgresql.org 336 [ + + + + : 71478 : if (!RELKIND_HAS_STORAGE(relkind))
+ + + + +
+ ]
5354 337 : 15019 : create_storage = false;
338 : : else
339 : : {
340 : : /*
341 : : * If relfilenumber is unspecified by the caller then create storage
342 : : * with oid same as relid.
343 : : */
1348 344 [ + + ]: 56459 : if (!RelFileNumberIsValid(relfilenumber))
1264 345 : 54894 : relfilenumber = relid;
346 : : }
347 : :
348 : : /*
349 : : * Never allow a pg_class entry to explicitly specify the database's
350 : : * default tablespace in reltablespace; force it to zero instead. This
351 : : * ensures that if the database is cloned with a different default
352 : : * tablespace, the pg_class entry will still match where CREATE DATABASE
353 : : * will put the physically copied relation.
354 : : *
355 : : * Yes, this is a bit of a hack.
356 : : */
7917 tgl@sss.pgh.pa.us 357 [ + + ]: 71478 : if (reltablespace == MyDatabaseTableSpace)
358 : 3 : reltablespace = InvalidOid;
359 : :
360 : : /*
361 : : * build the relcache entry.
362 : : */
8755 363 : 71478 : rel = RelationBuildLocalRelation(relname,
364 : : relnamespace,
365 : : tupDesc,
366 : : relid,
367 : : accessmtd,
368 : : relfilenumber,
369 : : reltablespace,
370 : : shared_relation,
371 : : mapped_relation,
372 : : relpersistence,
373 : : relkind);
374 : :
375 : : /*
376 : : * Have the storage manager create the relation's disk file, if needed.
377 : : *
378 : : * For tables, the AM callback creates both the main and the init fork.
379 : : * For others, only the main fork is created; the other forks will be
380 : : * created on demand.
381 : : */
7917 382 [ + + ]: 71478 : if (create_storage)
383 : : {
1563 peter@eisentraut.org 384 [ + + + + : 56422 : if (RELKIND_HAS_TABLE_AM(rel->rd_rel->relkind))
+ + ]
1348 rhaas@postgresql.org 385 : 31657 : table_relation_set_new_filelocator(rel, &rel->rd_locator,
386 : : relpersistence,
387 : : relfrozenxid, relminmxid);
1563 peter@eisentraut.org 388 [ + - + + : 24765 : else if (RELKIND_HAS_STORAGE(rel->rd_rel->relkind))
- + - - -
- ]
1348 rhaas@postgresql.org 389 : 24765 : RelationCreateStorage(rel->rd_locator, relpersistence, true);
390 : : else
1563 peter@eisentraut.org 391 :UBC 0 : Assert(false);
392 : : }
393 : :
394 : : /*
395 : : * If a tablespace is specified, removal of that tablespace is normally
396 : : * protected by the existence of a physical file; but for relations with
397 : : * no files, add a pg_shdepend entry to account for that.
398 : : */
1886 alvherre@alvh.no-ip. 399 [ + + + + ]:CBC 71478 : if (!create_storage && reltablespace != InvalidOid)
400 : 65 : recordDependencyOnTablespace(RelationRelationId, relid,
401 : : reltablespace);
402 : :
403 : : /* ensure that stats are dropped if transaction aborts */
1269 andres@anarazel.de 404 : 71478 : pgstat_create_relation(rel);
405 : :
10057 bruce@momjian.us 406 : 71478 : return rel;
407 : : }
408 : :
409 : : /* ----------------------------------------------------------------
410 : : * heap_create_with_catalog - Create a cataloged relation
411 : : *
412 : : * this is done in multiple steps:
413 : : *
414 : : * 1) CheckAttributeNamesTypes() is used to make certain the tuple
415 : : * descriptor contains a valid set of attribute names and types
416 : : *
417 : : * 2) pg_class is opened and get_relname_relid()
418 : : * performs a scan to ensure that no relation with the
419 : : * same name already exists.
420 : : *
421 : : * 3) heap_create() is called to create the new relation on disk.
422 : : *
423 : : * 4) TypeCreate() is called to define a new type corresponding
424 : : * to the new relation.
425 : : *
426 : : * 5) AddNewRelationTuple() is called to register the
427 : : * relation in pg_class.
428 : : *
429 : : * 6) AddNewAttributeTuples() is called to register the
430 : : * new relation's schema in pg_attribute.
431 : : *
432 : : * 7) StoreConstraints() is called - vadim 08/22/97
433 : : *
434 : : * 8) the relations are closed and the new relation's oid
435 : : * is returned.
436 : : *
437 : : * ----------------------------------------------------------------
438 : : */
439 : :
440 : : /* --------------------------------
441 : : * CheckAttributeNamesTypes
442 : : *
443 : : * this is used to make certain the tuple descriptor contains a
444 : : * valid set of attribute names and datatypes. a problem simply
445 : : * generates ereport(ERROR) which aborts the current transaction.
446 : : *
447 : : * relkind is the relkind of the relation to be created.
448 : : * flags controls which datatypes are allowed, cf CheckAttributeType.
449 : : * --------------------------------
450 : : */
451 : : void
5880 tgl@sss.pgh.pa.us 452 : 46779 : CheckAttributeNamesTypes(TupleDesc tupdesc, char relkind,
453 : : int flags)
454 : : {
455 : : int i;
456 : : int j;
10415 bruce@momjian.us 457 : 46779 : int natts = tupdesc->natts;
458 : :
459 : : /* Sanity check on column count */
8490 tgl@sss.pgh.pa.us 460 [ + - - + ]: 46779 : if (natts < 0 || natts > MaxHeapAttributeNumber)
8274 tgl@sss.pgh.pa.us 461 [ # # ]:UBC 0 : ereport(ERROR,
462 : : (errcode(ERRCODE_TOO_MANY_COLUMNS),
463 : : errmsg("tables can have at most %d columns",
464 : : MaxHeapAttributeNumber)));
465 : :
466 : : /*
467 : : * first check for collision with system attribute names
468 : : *
469 : : * Skip this for a view or type relation, since those don't have system
470 : : * attributes.
471 : : */
8613 bruce@momjian.us 472 [ + + + + ]:CBC 46779 : if (relkind != RELKIND_VIEW && relkind != RELKIND_COMPOSITE_TYPE)
473 : : {
8698 inoue@tpf.co.jp 474 [ + + ]: 144435 : for (i = 0; i < natts; i++)
475 : : {
3129 andres@anarazel.de 476 : 108964 : Form_pg_attribute attr = TupleDescAttr(tupdesc, i);
477 : :
2672 478 [ - + ]: 108964 : if (SystemAttributeByName(NameStr(attr->attname)) != NULL)
8273 tgl@sss.pgh.pa.us 479 [ # # ]:UBC 0 : ereport(ERROR,
480 : : (errcode(ERRCODE_DUPLICATE_COLUMN),
481 : : errmsg("column name \"%s\" conflicts with a system column name",
482 : : NameStr(attr->attname))));
483 : : }
484 : : }
485 : :
486 : : /*
487 : : * next check for repeated attribute names
488 : : */
9494 tgl@sss.pgh.pa.us 489 [ + + ]:CBC 194485 : for (i = 1; i < natts; i++)
490 : : {
491 [ + + ]: 3560670 : for (j = 0; j < i; j++)
492 : : {
3129 andres@anarazel.de 493 [ - + ]: 3412964 : if (strcmp(NameStr(TupleDescAttr(tupdesc, j)->attname),
494 : 3412964 : NameStr(TupleDescAttr(tupdesc, i)->attname)) == 0)
8273 tgl@sss.pgh.pa.us 495 [ # # ]:UBC 0 : ereport(ERROR,
496 : : (errcode(ERRCODE_DUPLICATE_COLUMN),
497 : : errmsg("column name \"%s\" specified more than once",
498 : : NameStr(TupleDescAttr(tupdesc, j)->attname))));
499 : : }
500 : : }
501 : :
502 : : /*
503 : : * next check the attribute types
504 : : */
8490 tgl@sss.pgh.pa.us 505 [ + + ]:CBC 239124 : for (i = 0; i < natts; i++)
506 : : {
3129 andres@anarazel.de 507 : 192364 : CheckAttributeType(NameStr(TupleDescAttr(tupdesc, i)->attname),
508 : 192364 : TupleDescAttr(tupdesc, i)->atttypid,
509 : 192364 : TupleDescAttr(tupdesc, i)->attcollation,
510 : : NIL, /* assume we're creating a new rowtype */
401 peter@eisentraut.org 511 [ + + ]: 192364 : flags | (TupleDescAttr(tupdesc, i)->attgenerated == ATTRIBUTE_GENERATED_VIRTUAL ? CHKATYPE_IS_VIRTUAL : 0));
512 : : }
8490 tgl@sss.pgh.pa.us 513 : 46760 : }
514 : :
515 : : /* --------------------------------
516 : : * CheckAttributeType
517 : : *
518 : : * Verify that the proposed datatype of an attribute is legal.
519 : : * This is needed mainly because there are types (and pseudo-types)
520 : : * in the catalogs that we do not support as elements of real tuples.
521 : : * We also check some other properties required of a table column.
522 : : *
523 : : * If the attribute is being proposed for addition to an existing table or
524 : : * composite type, pass a one-element list of the rowtype OID as
525 : : * containing_rowtypes. When checking a to-be-created rowtype, it's
526 : : * sufficient to pass NIL, because there could not be any recursive reference
527 : : * to a not-yet-existing rowtype.
528 : : *
529 : : * flags is a bitmask controlling which datatypes we allow. For the most
530 : : * part, pseudo-types are disallowed as attribute types, but there are some
531 : : * exceptions: ANYARRAYOID, RECORDOID, and RECORDARRAYOID can be allowed
532 : : * in some cases. (This works because values of those type classes are
533 : : * self-identifying to some extent. However, RECORDOID and RECORDARRAYOID
534 : : * are reliably identifiable only within a session, since the identity info
535 : : * may use a typmod that is only locally assigned. The caller is expected
536 : : * to know whether these cases are safe.)
537 : : *
538 : : * flags can also control the phrasing of the error messages. If
539 : : * CHKATYPE_IS_PARTKEY is specified, "attname" should be a partition key
540 : : * column number as text, not a real column name.
541 : : * --------------------------------
542 : : */
543 : : void
5466 544 : 237714 : CheckAttributeType(const char *attname,
545 : : Oid atttypid, Oid attcollation,
546 : : List *containing_rowtypes,
547 : : int flags)
548 : : {
8490 549 : 237714 : char att_typtype = get_typtype(atttypid);
550 : : Oid att_typelem;
551 : :
552 : : /* since this function recurses, it could be driven to stack overflow */
758 akorotkov@postgresql 553 : 237714 : check_stack_depth();
554 : :
3336 tgl@sss.pgh.pa.us 555 [ + + ]: 237714 : if (att_typtype == TYPTYPE_PSEUDO)
556 : : {
557 : : /*
558 : : * We disallow pseudo-type columns, with the exception of ANYARRAY,
559 : : * RECORD, and RECORD[] when the caller says that those are OK.
560 : : *
561 : : * We don't need to worry about recursive containment for RECORD and
562 : : * RECORD[] because (a) no named composite type should be allowed to
563 : : * contain those, and (b) two "anonymous" record types couldn't be
564 : : * considered to be the same type, so infinite recursion isn't
565 : : * possible.
566 : : */
2601 567 [ + + + + : 1043 : if (!((atttypid == ANYARRAYOID && (flags & CHKATYPE_ANYARRAY)) ||
+ + + + ]
568 [ + + ]: 12 : (atttypid == RECORDOID && (flags & CHKATYPE_ANYRECORD)) ||
569 [ - + ]: 3 : (atttypid == RECORDARRAYOID && (flags & CHKATYPE_ANYRECORD))))
570 : : {
2274 571 [ + + ]: 16 : if (flags & CHKATYPE_IS_PARTKEY)
572 [ + - ]: 6 : ereport(ERROR,
573 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
574 : : /* translator: first %s is an integer not a name */
575 : : errmsg("partition key column %s has pseudo-type %s",
576 : : attname, format_type_be(atttypid))));
577 : : else
578 [ + - ]: 10 : ereport(ERROR,
579 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
580 : : errmsg("column \"%s\" has pseudo-type %s",
581 : : attname, format_type_be(atttypid))));
582 : : }
583 : : }
5400 584 [ + + ]: 236677 : else if (att_typtype == TYPTYPE_DOMAIN)
585 : : {
586 : : /*
587 : : * Prevent virtual generated columns from having a domain type. We
588 : : * would have to enforce domain constraints when columns underlying
589 : : * the generated column change. This could possibly be implemented,
590 : : * but it's not.
591 : : */
401 peter@eisentraut.org 592 [ + + ]: 33977 : if (flags & CHKATYPE_IS_VIRTUAL)
593 [ + - ]: 18 : ereport(ERROR,
594 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
595 : : errmsg("virtual generated column \"%s\" cannot have a domain type", attname));
596 : :
597 : : /*
598 : : * If it's a domain, recurse to check its base type.
599 : : */
5400 tgl@sss.pgh.pa.us 600 : 33959 : CheckAttributeType(attname, getBaseType(atttypid), attcollation,
601 : : containing_rowtypes,
602 : : flags);
603 : : }
6883 604 [ + + ]: 202700 : else if (att_typtype == TYPTYPE_COMPOSITE)
605 : : {
606 : : /*
607 : : * For a composite type, recurse into its attributes.
608 : : */
609 : : Relation relation;
610 : : TupleDesc tupdesc;
611 : : int i;
612 : :
613 : : /*
614 : : * Check for self-containment. Eventually we might be able to allow
615 : : * this (just return without complaint, if so) but it's not clear how
616 : : * many other places would require anti-recursion defenses before it
617 : : * would be safe to allow tables to contain their own rowtype.
618 : : */
5466 619 [ + + ]: 389 : if (list_member_oid(containing_rowtypes, atttypid))
620 [ + - ]: 18 : ereport(ERROR,
621 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
622 : : errmsg("composite type %s cannot be made a member of itself",
623 : : format_type_be(atttypid))));
624 : :
2433 625 : 371 : containing_rowtypes = lappend_oid(containing_rowtypes, atttypid);
626 : :
6883 627 : 371 : relation = relation_open(get_typ_typrelid(atttypid), AccessShareLock);
628 : :
629 : 371 : tupdesc = RelationGetDescr(relation);
630 : :
631 [ + + ]: 2733 : for (i = 0; i < tupdesc->natts; i++)
632 : : {
3129 andres@anarazel.de 633 : 2368 : Form_pg_attribute attr = TupleDescAttr(tupdesc, i);
634 : :
6883 tgl@sss.pgh.pa.us 635 [ + + ]: 2368 : if (attr->attisdropped)
636 : 1 : continue;
5466 637 : 2367 : CheckAttributeType(NameStr(attr->attname),
638 : : attr->atttypid, attr->attcollation,
639 : : containing_rowtypes,
640 : : flags & ~CHKATYPE_IS_PARTKEY);
641 : : }
642 : :
6883 643 : 365 : relation_close(relation, AccessShareLock);
644 : :
2433 645 : 365 : containing_rowtypes = list_delete_last(containing_rowtypes);
646 : : }
2274 647 [ + + ]: 202311 : else if (att_typtype == TYPTYPE_RANGE)
648 : : {
649 : : /*
650 : : * If it's a range, recurse to check its subtype.
651 : : */
2235 652 : 1045 : CheckAttributeType(attname, get_range_subtype(atttypid),
653 : : get_range_collation(atttypid),
654 : : containing_rowtypes,
655 : : flags);
656 : : }
5466 657 [ + + ]: 201266 : else if (OidIsValid((att_typelem = get_element_type(atttypid))))
658 : : {
659 : : /*
660 : : * Must recurse into array types, too, in case they are composite.
661 : : */
662 : 5213 : CheckAttributeType(attname, att_typelem, attcollation,
663 : : containing_rowtypes,
664 : : flags);
665 : : }
666 : :
667 : : /*
668 : : * For consistency with check_virtual_generated_security().
669 : : */
263 peter@eisentraut.org 670 [ + + + + ]: 237641 : if ((flags & CHKATYPE_IS_VIRTUAL) && atttypid >= FirstUnpinnedObjectId)
671 [ + - ]: 3 : ereport(ERROR,
672 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
673 : : errmsg("virtual generated column \"%s\" cannot have a user-defined type", attname),
674 : : errdetail("Virtual generated columns that make use of user-defined types are not yet supported."));
675 : :
676 : : /*
677 : : * This might not be strictly invalid per SQL standard, but it is pretty
678 : : * useless, and it cannot be dumped, so we must disallow it.
679 : : */
5466 tgl@sss.pgh.pa.us 680 [ + + - + ]: 237638 : if (!OidIsValid(attcollation) && type_is_collatable(atttypid))
681 : : {
2274 tgl@sss.pgh.pa.us 682 [ # # ]:UBC 0 : if (flags & CHKATYPE_IS_PARTKEY)
683 [ # # ]: 0 : ereport(ERROR,
684 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
685 : : /* translator: first %s is an integer not a name */
686 : : errmsg("no collation was derived for partition key column %s with collatable type %s",
687 : : attname, format_type_be(atttypid)),
688 : : errhint("Use the COLLATE clause to set the collation explicitly.")));
689 : : else
690 [ # # ]: 0 : ereport(ERROR,
691 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
692 : : errmsg("no collation was derived for column \"%s\" with collatable type %s",
693 : : attname, format_type_be(atttypid)),
694 : : errhint("Use the COLLATE clause to set the collation explicitly.")));
695 : : }
10841 scrappy@hub.org 696 :CBC 237638 : }
697 : :
698 : : /*
699 : : * InsertPgAttributeTuples
700 : : * Construct and insert a set of tuples in pg_attribute.
701 : : *
702 : : * Caller has already opened and locked pg_attribute. tupdesc contains the
703 : : * attributes to insert. tupdesc_extra supplies the values for certain
704 : : * variable-length/nullable pg_attribute fields and must contain the same
705 : : * number of elements as tupdesc or be NULL. The other variable-length fields
706 : : * of pg_attribute are always initialized to null values.
707 : : *
708 : : * indstate is the index state for CatalogTupleInsertWithInfo. It can be
709 : : * passed as NULL, in which case we'll fetch the necessary info. (Don't do
710 : : * this when inserting multiple attributes, because it's a tad more
711 : : * expensive.)
712 : : *
713 : : * new_rel_oid is the relation OID assigned to the attributes inserted.
714 : : * If set to InvalidOid, the relation OID from tupdesc is used instead.
715 : : */
716 : : void
2053 michael@paquier.xyz 717 : 108131 : InsertPgAttributeTuples(Relation pg_attribute_rel,
718 : : TupleDesc tupdesc,
719 : : Oid new_rel_oid,
720 : : const FormExtraData_pg_attribute tupdesc_extra[],
721 : : CatalogIndexState indstate)
722 : : {
723 : : TupleTableSlot **slot;
724 : : TupleDesc td;
725 : : int nslots;
726 : 108131 : int natts = 0;
727 : 108131 : int slotCount = 0;
728 : 108131 : bool close_index = false;
729 : :
730 : 108131 : td = RelationGetDescr(pg_attribute_rel);
731 : :
732 : : /* Initialize the number of slots to use */
733 [ + + ]: 108131 : nslots = Min(tupdesc->natts,
734 : : (MAX_CATALOG_MULTI_INSERT_BYTES / sizeof(FormData_pg_attribute)));
95 michael@paquier.xyz 735 :GNC 108131 : slot = palloc_array(TupleTableSlot *, nslots);
2053 michael@paquier.xyz 736 [ + + ]:CBC 552222 : for (int i = 0; i < nslots; i++)
737 : 444091 : slot[i] = MakeSingleTupleTableSlot(td, &TTSOpsHeapTuple);
738 : :
739 [ + + ]: 554010 : while (natts < tupdesc->natts)
740 : : {
741 : 445879 : Form_pg_attribute attrs = TupleDescAttr(tupdesc, natts);
728 peter@eisentraut.org 742 [ + + ]: 445879 : const FormExtraData_pg_attribute *attrs_extra = tupdesc_extra ? &tupdesc_extra[natts] : NULL;
743 : :
2053 michael@paquier.xyz 744 : 445879 : ExecClearTuple(slot[slotCount]);
745 : :
1601 dgustafsson@postgres 746 : 445879 : memset(slot[slotCount]->tts_isnull, false,
747 : 445879 : slot[slotCount]->tts_tupleDescriptor->natts * sizeof(bool));
748 : :
2053 michael@paquier.xyz 749 [ + + ]: 445879 : if (new_rel_oid != InvalidOid)
750 : 405478 : slot[slotCount]->tts_values[Anum_pg_attribute_attrelid - 1] = ObjectIdGetDatum(new_rel_oid);
751 : : else
752 : 40401 : slot[slotCount]->tts_values[Anum_pg_attribute_attrelid - 1] = ObjectIdGetDatum(attrs->attrelid);
753 : :
754 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attname - 1] = NameGetDatum(&attrs->attname);
755 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_atttypid - 1] = ObjectIdGetDatum(attrs->atttypid);
756 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attlen - 1] = Int16GetDatum(attrs->attlen);
757 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attnum - 1] = Int16GetDatum(attrs->attnum);
758 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_atttypmod - 1] = Int32GetDatum(attrs->atttypmod);
1083 peter@eisentraut.org 759 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attndims - 1] = Int16GetDatum(attrs->attndims);
2053 michael@paquier.xyz 760 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attbyval - 1] = BoolGetDatum(attrs->attbyval);
761 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attalign - 1] = CharGetDatum(attrs->attalign);
1757 tgl@sss.pgh.pa.us 762 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attstorage - 1] = CharGetDatum(attrs->attstorage);
763 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attcompression - 1] = CharGetDatum(attrs->attcompression);
2053 michael@paquier.xyz 764 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attnotnull - 1] = BoolGetDatum(attrs->attnotnull);
765 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_atthasdef - 1] = BoolGetDatum(attrs->atthasdef);
766 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_atthasmissing - 1] = BoolGetDatum(attrs->atthasmissing);
767 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attidentity - 1] = CharGetDatum(attrs->attidentity);
768 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attgenerated - 1] = CharGetDatum(attrs->attgenerated);
769 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attisdropped - 1] = BoolGetDatum(attrs->attisdropped);
770 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attislocal - 1] = BoolGetDatum(attrs->attislocal);
1083 peter@eisentraut.org 771 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attinhcount - 1] = Int16GetDatum(attrs->attinhcount);
2053 michael@paquier.xyz 772 : 445879 : slot[slotCount]->tts_values[Anum_pg_attribute_attcollation - 1] = ObjectIdGetDatum(attrs->attcollation);
728 peter@eisentraut.org 773 [ + + ]: 445879 : if (attrs_extra)
774 : : {
775 : 22331 : slot[slotCount]->tts_values[Anum_pg_attribute_attstattarget - 1] = attrs_extra->attstattarget.value;
776 : 22331 : slot[slotCount]->tts_isnull[Anum_pg_attribute_attstattarget - 1] = attrs_extra->attstattarget.isnull;
777 : :
778 : 22331 : slot[slotCount]->tts_values[Anum_pg_attribute_attoptions - 1] = attrs_extra->attoptions.value;
779 : 22331 : slot[slotCount]->tts_isnull[Anum_pg_attribute_attoptions - 1] = attrs_extra->attoptions.isnull;
780 : : }
781 : : else
782 : : {
783 : 423548 : slot[slotCount]->tts_isnull[Anum_pg_attribute_attstattarget - 1] = true;
2053 michael@paquier.xyz 784 : 423548 : slot[slotCount]->tts_isnull[Anum_pg_attribute_attoptions - 1] = true;
785 : : }
786 : :
787 : : /*
788 : : * The remaining fields are not set for new columns.
789 : : */
790 : 445879 : slot[slotCount]->tts_isnull[Anum_pg_attribute_attacl - 1] = true;
791 : 445879 : slot[slotCount]->tts_isnull[Anum_pg_attribute_attfdwoptions - 1] = true;
792 : 445879 : slot[slotCount]->tts_isnull[Anum_pg_attribute_attmissingval - 1] = true;
793 : :
794 : 445879 : ExecStoreVirtualTuple(slot[slotCount]);
795 : 445879 : slotCount++;
796 : :
797 : : /*
798 : : * If slots are full or the end of processing has been reached, insert
799 : : * a batch of tuples.
800 : : */
801 [ + + + + ]: 445879 : if (slotCount == nslots || natts == tupdesc->natts - 1)
802 : : {
803 : : /* fetch index info only when we know we need it */
804 [ + + ]: 106017 : if (!indstate)
805 : : {
806 : 1407 : indstate = CatalogOpenIndexes(pg_attribute_rel);
807 : 1407 : close_index = true;
808 : : }
809 : :
810 : : /* insert the new tuples and update the indexes */
811 : 106017 : CatalogTuplesMultiInsertWithInfo(pg_attribute_rel, slot, slotCount,
812 : : indstate);
813 : 106017 : slotCount = 0;
814 : : }
815 : :
816 : 445879 : natts++;
817 : : }
818 : :
819 [ + + ]: 108131 : if (close_index)
820 : 1407 : CatalogCloseIndexes(indstate);
821 [ + + ]: 552222 : for (int i = 0; i < nslots; i++)
822 : 444091 : ExecDropSingleTupleTableSlot(slot[i]);
823 : 108131 : pfree(slot);
6330 alvherre@alvh.no-ip. 824 : 108131 : }
825 : :
826 : : /* --------------------------------
827 : : * AddNewAttributeTuples
828 : : *
829 : : * this registers the new relation's schema by adding
830 : : * tuples to pg_attribute.
831 : : * --------------------------------
832 : : */
833 : : static void
10841 scrappy@hub.org 834 : 46353 : AddNewAttributeTuples(Oid new_rel_oid,
835 : : TupleDesc tupdesc,
836 : : char relkind)
837 : : {
838 : : Relation rel;
839 : : CatalogIndexState indstate;
10415 bruce@momjian.us 840 : 46353 : int natts = tupdesc->natts;
841 : : ObjectAddress myself,
842 : : referenced;
843 : :
844 : : /*
845 : : * open pg_attribute and its indexes.
846 : : */
2610 andres@anarazel.de 847 : 46353 : rel = table_open(AttributeRelationId, RowExclusiveLock);
848 : :
8623 tgl@sss.pgh.pa.us 849 : 46353 : indstate = CatalogOpenIndexes(rel);
850 : :
2053 michael@paquier.xyz 851 : 46353 : InsertPgAttributeTuples(rel, tupdesc, new_rel_oid, NULL, indstate);
852 : :
853 : : /* add dependencies on their datatypes and collations */
854 [ + + ]: 237724 : for (int i = 0; i < natts; i++)
855 : : {
621 drowley@postgresql.o 856 : 191371 : Form_pg_attribute attr = TupleDescAttr(tupdesc, i);
857 : :
858 : : /* Add dependency info */
2083 michael@paquier.xyz 859 : 191371 : ObjectAddressSubSet(myself, RelationRelationId, new_rel_oid, i + 1);
621 drowley@postgresql.o 860 : 191371 : ObjectAddressSet(referenced, TypeRelationId, attr->atttypid);
8643 tgl@sss.pgh.pa.us 861 : 191371 : recordDependencyOn(&myself, &referenced, DEPENDENCY_NORMAL);
862 : :
863 : : /* The default collation is pinned, so don't bother recording it */
621 drowley@postgresql.o 864 [ + + ]: 191371 : if (OidIsValid(attr->attcollation) &&
865 [ + + ]: 58598 : attr->attcollation != DEFAULT_COLLATION_OID)
866 : : {
2053 michael@paquier.xyz 867 : 41888 : ObjectAddressSet(referenced, CollationRelationId,
868 : : attr->attcollation);
5510 peter_e@gmx.net 869 : 41888 : recordDependencyOn(&myself, &referenced, DEPENDENCY_NORMAL);
870 : : }
871 : : }
872 : :
873 : : /*
874 : : * Next we add the system attributes. Skip all for a view or type
875 : : * relation. We don't bother with making datatype dependencies here,
876 : : * since presumably all these types are pinned.
877 : : */
8613 bruce@momjian.us 878 [ + + + + ]: 46353 : if (relkind != RELKIND_VIEW && relkind != RELKIND_COMPOSITE_TYPE)
879 : : {
880 : : TupleDesc td;
881 : :
2053 michael@paquier.xyz 882 : 35450 : td = CreateTupleDesc(lengthof(SysAtt), (FormData_pg_attribute **) &SysAtt);
883 : :
884 : 35450 : InsertPgAttributeTuples(rel, td, new_rel_oid, NULL, indstate);
885 : 35450 : FreeTupleDesc(td);
886 : : }
887 : :
888 : : /*
889 : : * clean up
890 : : */
8623 tgl@sss.pgh.pa.us 891 : 46353 : CatalogCloseIndexes(indstate);
892 : :
2610 andres@anarazel.de 893 : 46353 : table_close(rel, RowExclusiveLock);
10841 scrappy@hub.org 894 : 46353 : }
895 : :
896 : : /* --------------------------------
897 : : * InsertPgClassTuple
898 : : *
899 : : * Construct and insert a new tuple in pg_class.
900 : : *
901 : : * Caller has already opened and locked pg_class.
902 : : * Tuple data is taken from new_rel_desc->rd_rel, except for the
903 : : * variable-width fields which are not present in a cached reldesc.
904 : : * relacl and reloptions are passed in Datum form (to avoid having
905 : : * to reference the data types in heap.h). Pass (Datum) 0 to set them
906 : : * to NULL.
907 : : * --------------------------------
908 : : */
909 : : void
7195 tgl@sss.pgh.pa.us 910 : 71274 : InsertPgClassTuple(Relation pg_class_desc,
911 : : Relation new_rel_desc,
912 : : Oid new_rel_oid,
913 : : Datum relacl,
914 : : Datum reloptions)
915 : : {
916 : 71274 : Form_pg_class rd_rel = new_rel_desc->rd_rel;
917 : : Datum values[Natts_pg_class];
918 : : bool nulls[Natts_pg_class];
919 : : HeapTuple tup;
920 : :
921 : : /* This is a tad tedious, but way cleaner than what we used to do... */
922 : 71274 : memset(values, 0, sizeof(values));
6342 923 : 71274 : memset(nulls, false, sizeof(nulls));
924 : :
2672 andres@anarazel.de 925 : 71274 : values[Anum_pg_class_oid - 1] = ObjectIdGetDatum(new_rel_oid);
7195 tgl@sss.pgh.pa.us 926 : 71274 : values[Anum_pg_class_relname - 1] = NameGetDatum(&rd_rel->relname);
927 : 71274 : values[Anum_pg_class_relnamespace - 1] = ObjectIdGetDatum(rd_rel->relnamespace);
928 : 71274 : values[Anum_pg_class_reltype - 1] = ObjectIdGetDatum(rd_rel->reltype);
5890 peter_e@gmx.net 929 : 71274 : values[Anum_pg_class_reloftype - 1] = ObjectIdGetDatum(rd_rel->reloftype);
7195 tgl@sss.pgh.pa.us 930 : 71274 : values[Anum_pg_class_relowner - 1] = ObjectIdGetDatum(rd_rel->relowner);
931 : 71274 : values[Anum_pg_class_relam - 1] = ObjectIdGetDatum(rd_rel->relam);
1264 rhaas@postgresql.org 932 : 71274 : values[Anum_pg_class_relfilenode - 1] = ObjectIdGetDatum(rd_rel->relfilenode);
7195 tgl@sss.pgh.pa.us 933 : 71274 : values[Anum_pg_class_reltablespace - 1] = ObjectIdGetDatum(rd_rel->reltablespace);
934 : 71274 : values[Anum_pg_class_relpages - 1] = Int32GetDatum(rd_rel->relpages);
935 : 71274 : values[Anum_pg_class_reltuples - 1] = Float4GetDatum(rd_rel->reltuples);
5266 936 : 71274 : values[Anum_pg_class_relallvisible - 1] = Int32GetDatum(rd_rel->relallvisible);
377 melanieplageman@gmai 937 : 71274 : values[Anum_pg_class_relallfrozen - 1] = Int32GetDatum(rd_rel->relallfrozen);
7195 tgl@sss.pgh.pa.us 938 : 71274 : values[Anum_pg_class_reltoastrelid - 1] = ObjectIdGetDatum(rd_rel->reltoastrelid);
939 : 71274 : values[Anum_pg_class_relhasindex - 1] = BoolGetDatum(rd_rel->relhasindex);
940 : 71274 : values[Anum_pg_class_relisshared - 1] = BoolGetDatum(rd_rel->relisshared);
5571 rhaas@postgresql.org 941 : 71274 : values[Anum_pg_class_relpersistence - 1] = CharGetDatum(rd_rel->relpersistence);
7195 tgl@sss.pgh.pa.us 942 : 71274 : values[Anum_pg_class_relkind - 1] = CharGetDatum(rd_rel->relkind);
943 : 71274 : values[Anum_pg_class_relnatts - 1] = Int16GetDatum(rd_rel->relnatts);
944 : 71274 : values[Anum_pg_class_relchecks - 1] = Int16GetDatum(rd_rel->relchecks);
945 : 71274 : values[Anum_pg_class_relhasrules - 1] = BoolGetDatum(rd_rel->relhasrules);
6335 946 : 71274 : values[Anum_pg_class_relhastriggers - 1] = BoolGetDatum(rd_rel->relhastriggers);
4190 sfrost@snowman.net 947 : 71274 : values[Anum_pg_class_relrowsecurity - 1] = BoolGetDatum(rd_rel->relrowsecurity);
3815 948 : 71274 : values[Anum_pg_class_relforcerowsecurity - 1] = BoolGetDatum(rd_rel->relforcerowsecurity);
7195 tgl@sss.pgh.pa.us 949 : 71274 : values[Anum_pg_class_relhassubclass - 1] = BoolGetDatum(rd_rel->relhassubclass);
4696 950 : 71274 : values[Anum_pg_class_relispopulated - 1] = BoolGetDatum(rd_rel->relispopulated);
4510 rhaas@postgresql.org 951 : 71274 : values[Anum_pg_class_relreplident - 1] = CharGetDatum(rd_rel->relreplident);
3385 952 : 71274 : values[Anum_pg_class_relispartition - 1] = BoolGetDatum(rd_rel->relispartition);
2916 peter_e@gmx.net 953 : 71274 : values[Anum_pg_class_relrewrite - 1] = ObjectIdGetDatum(rd_rel->relrewrite);
7070 tgl@sss.pgh.pa.us 954 : 71274 : values[Anum_pg_class_relfrozenxid - 1] = TransactionIdGetDatum(rd_rel->relfrozenxid);
4799 alvherre@alvh.no-ip. 955 : 71274 : values[Anum_pg_class_relminmxid - 1] = MultiXactIdGetDatum(rd_rel->relminmxid);
6005 tgl@sss.pgh.pa.us 956 [ + + ]: 71274 : if (relacl != (Datum) 0)
957 : 66 : values[Anum_pg_class_relacl - 1] = relacl;
958 : : else
959 : 71208 : nulls[Anum_pg_class_relacl - 1] = true;
7195 960 [ + + ]: 71274 : if (reloptions != (Datum) 0)
961 : 959 : values[Anum_pg_class_reloptions - 1] = reloptions;
962 : : else
6342 963 : 70315 : nulls[Anum_pg_class_reloptions - 1] = true;
964 : :
965 : : /* relpartbound is set by updating this tuple, if necessary */
3385 rhaas@postgresql.org 966 : 71274 : nulls[Anum_pg_class_relpartbound - 1] = true;
967 : :
6342 tgl@sss.pgh.pa.us 968 : 71274 : tup = heap_form_tuple(RelationGetDescr(pg_class_desc), values, nulls);
969 : :
970 : : /* finally insert the new tuple, update the indexes, and clean up */
3330 alvherre@alvh.no-ip. 971 : 71274 : CatalogTupleInsert(pg_class_desc, tup);
972 : :
7195 tgl@sss.pgh.pa.us 973 : 71274 : heap_freetuple(tup);
974 : 71274 : }
975 : :
976 : : /* --------------------------------
977 : : * AddNewRelationTuple
978 : : *
979 : : * this registers the new relation in the catalogs by
980 : : * adding a tuple to pg_class.
981 : : * --------------------------------
982 : : */
983 : : static void
9903 bruce@momjian.us 984 : 46353 : AddNewRelationTuple(Relation pg_class_desc,
985 : : Relation new_rel_desc,
986 : : Oid new_rel_oid,
987 : : Oid new_type_oid,
988 : : Oid reloftype,
989 : : Oid relowner,
990 : : char relkind,
991 : : TransactionId relfrozenxid,
992 : : TransactionId relminmxid,
993 : : Datum relacl,
994 : : Datum reloptions)
995 : : {
996 : : Form_pg_class new_rel_reltup;
997 : :
998 : : /*
999 : : * first we update some of the information in our uncataloged relation's
1000 : : * relation descriptor.
1001 : : */
10416 1002 : 46353 : new_rel_reltup = new_rel_desc->rd_rel;
1003 : :
1004 : : /* The relation is empty */
1563 peter@eisentraut.org 1005 : 46353 : new_rel_reltup->relpages = 0;
1006 : 46353 : new_rel_reltup->reltuples = -1;
1007 : 46353 : new_rel_reltup->relallvisible = 0;
377 melanieplageman@gmai 1008 : 46353 : new_rel_reltup->relallfrozen = 0;
1009 : :
1010 : : /* Sequences always have a known size */
1563 peter@eisentraut.org 1011 [ + + ]: 46353 : if (relkind == RELKIND_SEQUENCE)
1012 : : {
1013 : 951 : new_rel_reltup->relpages = 1;
1014 : 951 : new_rel_reltup->reltuples = 1;
1015 : : }
1016 : :
2544 andres@anarazel.de 1017 : 46353 : new_rel_reltup->relfrozenxid = relfrozenxid;
1018 : 46353 : new_rel_reltup->relminmxid = relminmxid;
7506 tgl@sss.pgh.pa.us 1019 : 46353 : new_rel_reltup->relowner = relowner;
9162 1020 : 46353 : new_rel_reltup->reltype = new_type_oid;
5890 peter_e@gmx.net 1021 : 46353 : new_rel_reltup->reloftype = reloftype;
1022 : :
1023 : : /* relispartition is always set by updating this tuple later */
3385 rhaas@postgresql.org 1024 : 46353 : new_rel_reltup->relispartition = false;
1025 : :
1026 : : /* fill rd_att's type ID with something sane even if reltype is zero */
2077 tgl@sss.pgh.pa.us 1027 [ + + ]: 46353 : new_rel_desc->rd_att->tdtypeid = new_type_oid ? new_type_oid : RECORDOID;
1028 : 46353 : new_rel_desc->rd_att->tdtypmod = -1;
1029 : :
1030 : : /* Now build and insert the tuple */
6005 1031 : 46353 : InsertPgClassTuple(pg_class_desc, new_rel_desc, new_rel_oid,
1032 : : relacl, reloptions);
10841 scrappy@hub.org 1033 : 46353 : }
1034 : :
1035 : :
1036 : : /* --------------------------------
1037 : : * AddNewRelationType -
1038 : : *
1039 : : * define a composite type corresponding to the new relation
1040 : : * --------------------------------
1041 : : */
1042 : : static ObjectAddress
8752 tgl@sss.pgh.pa.us 1043 : 36367 : AddNewRelationType(const char *typeName,
1044 : : Oid typeNamespace,
1045 : : Oid new_rel_oid,
1046 : : char new_rel_kind,
1047 : : Oid ownerid,
1048 : : Oid new_row_type,
1049 : : Oid new_array_type)
1050 : : {
1051 : : return
5861 bruce@momjian.us 1052 : 36367 : TypeCreate(new_row_type, /* optional predetermined OID */
1053 : : typeName, /* type name */
1054 : : typeNamespace, /* type namespace */
1055 : : new_rel_oid, /* relation oid */
1056 : : new_rel_kind, /* relation kind */
1057 : : ownerid, /* owner's ID */
1058 : : -1, /* internal size (varlena) */
1059 : : TYPTYPE_COMPOSITE, /* type-type (composite) */
1060 : : TYPCATEGORY_COMPOSITE, /* type-category (ditto) */
1061 : : false, /* composite types are never preferred */
1062 : : DEFAULT_TYPDELIM, /* default array delimiter */
1063 : : F_RECORD_IN, /* input procedure */
1064 : : F_RECORD_OUT, /* output procedure */
1065 : : F_RECORD_RECV, /* receive procedure */
1066 : : F_RECORD_SEND, /* send procedure */
1067 : : InvalidOid, /* typmodin procedure - none */
1068 : : InvalidOid, /* typmodout procedure - none */
1069 : : InvalidOid, /* analyze procedure - default */
1070 : : InvalidOid, /* subscript procedure - none */
1071 : : InvalidOid, /* array element type - irrelevant */
1072 : : false, /* this is not an array type */
1073 : : new_array_type, /* array type if any */
1074 : : InvalidOid, /* domain base type - irrelevant */
1075 : : NULL, /* default value - none */
1076 : : NULL, /* default binary representation */
1077 : : false, /* passed by reference */
1078 : : TYPALIGN_DOUBLE, /* alignment - must be the largest! */
1079 : : TYPSTORAGE_EXTENDED, /* fully TOASTable */
1080 : : -1, /* typmod */
1081 : : 0, /* array dimensions for typBaseType */
1082 : : false, /* Type NOT NULL */
1083 : : InvalidOid); /* rowtypes never have a collation */
1084 : : }
1085 : :
1086 : : /* --------------------------------
1087 : : * heap_create_with_catalog
1088 : : *
1089 : : * creates a new cataloged relation. see comments above.
1090 : : *
1091 : : * Arguments:
1092 : : * relname: name to give to new rel
1093 : : * relnamespace: OID of namespace it goes in
1094 : : * reltablespace: OID of tablespace it goes in
1095 : : * relid: OID to assign to new rel, or InvalidOid to select a new OID
1096 : : * reltypeid: OID to assign to rel's rowtype, or InvalidOid to select one
1097 : : * reloftypeid: if a typed table, OID of underlying type; else InvalidOid
1098 : : * ownerid: OID of new rel's owner
1099 : : * accessmtd: OID of new rel's access method
1100 : : * tupdesc: tuple descriptor (source of column definitions)
1101 : : * cooked_constraints: list of precooked check constraints and defaults
1102 : : * relkind: relkind for new rel
1103 : : * relpersistence: rel's persistence status (permanent, temp, or unlogged)
1104 : : * shared_relation: true if it's to be a shared relation
1105 : : * mapped_relation: true if the relation will use the relfilenumber map
1106 : : * oncommit: ON COMMIT marking (only relevant if it's a temp table)
1107 : : * reloptions: reloptions in Datum form, or (Datum) 0 if none
1108 : : * use_user_acl: true if should look for user-defined default permissions;
1109 : : * if false, relacl is always set NULL
1110 : : * allow_system_table_mods: true to allow creation in system namespaces
1111 : : * is_internal: is this a system-generated catalog?
1112 : : * relrewrite: link to original relation during a table rewrite
1113 : : *
1114 : : * Output parameters:
1115 : : * typaddress: if not null, gets the object address of the new pg_type entry
1116 : : * (this must be null if the relkind is one that doesn't get a pg_type entry)
1117 : : *
1118 : : * Returns the OID of the new relation
1119 : : * --------------------------------
1120 : : */
1121 : : Oid
8750 tgl@sss.pgh.pa.us 1122 : 46377 : heap_create_with_catalog(const char *relname,
1123 : : Oid relnamespace,
1124 : : Oid reltablespace,
1125 : : Oid relid,
1126 : : Oid reltypeid,
1127 : : Oid reloftypeid,
1128 : : Oid ownerid,
1129 : : Oid accessmtd,
1130 : : TupleDesc tupdesc,
1131 : : List *cooked_constraints,
1132 : : char relkind,
1133 : : char relpersistence,
1134 : : bool shared_relation,
1135 : : bool mapped_relation,
1136 : : OnCommitAction oncommit,
1137 : : Datum reloptions,
1138 : : bool use_user_acl,
1139 : : bool allow_system_table_mods,
1140 : : bool is_internal,
1141 : : Oid relrewrite,
1142 : : ObjectAddress *typaddress)
1143 : : {
1144 : : Relation pg_class_desc;
1145 : : Relation new_rel_desc;
1146 : : Acl *relacl;
1147 : : Oid existing_relid;
1148 : : Oid old_type_oid;
1149 : : Oid new_type_oid;
1150 : :
1151 : : /* By default set to InvalidOid unless overridden by binary-upgrade */
1348 rhaas@postgresql.org 1152 : 46377 : RelFileNumber relfilenumber = InvalidRelFileNumber;
1153 : : TransactionId relfrozenxid;
1154 : : MultiXactId relminmxid;
1155 : :
2610 andres@anarazel.de 1156 : 46377 : pg_class_desc = table_open(RelationRelationId, RowExclusiveLock);
1157 : :
1158 : : /*
1159 : : * sanity checks
1160 : : */
9903 bruce@momjian.us 1161 [ + + - + ]: 46377 : Assert(IsNormalProcessingMode() || IsBootstrapProcessingMode());
1162 : :
1163 : : /*
1164 : : * Validate proposed tupdesc for the desired relkind. If
1165 : : * allow_system_table_mods is on, allow ANYARRAY to be used; this is a
1166 : : * hack to allow creating pg_statistic and cloning it during VACUUM FULL.
1167 : : */
2601 tgl@sss.pgh.pa.us 1168 : 46377 : CheckAttributeNamesTypes(tupdesc, relkind,
1169 : : allow_system_table_mods ? CHKATYPE_ANYARRAY : 0);
1170 : :
1171 : : /*
1172 : : * This would fail later on anyway, if the relation already exists. But
1173 : : * by catching it here we can emit a nicer error message.
1174 : : */
5712 rhaas@postgresql.org 1175 : 46358 : existing_relid = get_relname_relid(relname, relnamespace);
1176 [ + + ]: 46358 : if (existing_relid != InvalidOid)
8273 tgl@sss.pgh.pa.us 1177 [ + - ]: 1 : ereport(ERROR,
1178 : : (errcode(ERRCODE_DUPLICATE_TABLE),
1179 : : errmsg("relation \"%s\" already exists", relname)));
1180 : :
1181 : : /*
1182 : : * Since we are going to create a rowtype as well, also check for
1183 : : * collision with an existing type name. If there is one and it's an
1184 : : * autogenerated array, we can rename it out of the way; otherwise we can
1185 : : * at least give a good error message.
1186 : : */
2672 andres@anarazel.de 1187 : 46357 : old_type_oid = GetSysCacheOid2(TYPENAMENSP, Anum_pg_type_oid,
1188 : : CStringGetDatum(relname),
1189 : : ObjectIdGetDatum(relnamespace));
6882 tgl@sss.pgh.pa.us 1190 [ + + ]: 46357 : if (OidIsValid(old_type_oid))
1191 : : {
1192 [ - + ]: 1 : if (!moveArrayTypeName(old_type_oid, relname, relnamespace))
6882 tgl@sss.pgh.pa.us 1193 [ # # ]:UBC 0 : ereport(ERROR,
1194 : : (errcode(ERRCODE_DUPLICATE_OBJECT),
1195 : : errmsg("type \"%s\" already exists", relname),
1196 : : errhint("A relation has an associated type of the same name, "
1197 : : "so you must use a name that doesn't conflict "
1198 : : "with any existing type.")));
1199 : : }
1200 : :
1201 : : /*
1202 : : * Shared relations must be in pg_global (last-ditch check)
1203 : : */
5880 tgl@sss.pgh.pa.us 1204 [ + + - + ]:CBC 46357 : if (shared_relation && reltablespace != GLOBALTABLESPACE_OID)
5880 tgl@sss.pgh.pa.us 1205 [ # # ]:UBC 0 : elog(ERROR, "shared relations must be placed in pg_global tablespace");
1206 : :
1207 : : /*
1208 : : * Allocate an OID for the relation, unless we were told what to use.
1209 : : *
1210 : : * The OID will be the relfilenumber as well, so make sure it doesn't
1211 : : * collide with either pg_class OIDs or existing physical files.
1212 : : */
5884 tgl@sss.pgh.pa.us 1213 [ + + ]:CBC 46357 : if (!OidIsValid(relid))
1214 : : {
1215 : : /* Use binary-upgrade override for pg_class.oid and relfilenumber */
1563 peter@eisentraut.org 1216 [ + + ]: 41512 : if (IsBinaryUpgrade)
1217 : : {
1218 : : /*
1219 : : * Indexes are not supported here; they use
1220 : : * binary_upgrade_next_index_pg_class_oid.
1221 : : */
1222 [ - + ]: 1153 : Assert(relkind != RELKIND_INDEX);
1223 [ - + ]: 1153 : Assert(relkind != RELKIND_PARTITIONED_INDEX);
1224 : :
1225 [ + + ]: 1153 : if (relkind == RELKIND_TOASTVALUE)
1226 : : {
1227 : : /* There might be no TOAST table, so we have to test for it. */
1228 [ + - ]: 270 : if (OidIsValid(binary_upgrade_next_toast_pg_class_oid))
1229 : : {
1230 : 270 : relid = binary_upgrade_next_toast_pg_class_oid;
1231 : 270 : binary_upgrade_next_toast_pg_class_oid = InvalidOid;
1232 : :
1348 rhaas@postgresql.org 1233 [ - + ]: 270 : if (!RelFileNumberIsValid(binary_upgrade_next_toast_pg_class_relfilenumber))
1518 rhaas@postgresql.org 1234 [ # # ]:UBC 0 : ereport(ERROR,
1235 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
1236 : : errmsg("toast relfilenumber value not set when in binary upgrade mode")));
1237 : :
1348 rhaas@postgresql.org 1238 :CBC 270 : relfilenumber = binary_upgrade_next_toast_pg_class_relfilenumber;
1239 : 270 : binary_upgrade_next_toast_pg_class_relfilenumber = InvalidRelFileNumber;
1240 : : }
1241 : : }
1242 : : else
1243 : : {
1563 peter@eisentraut.org 1244 [ - + ]: 883 : if (!OidIsValid(binary_upgrade_next_heap_pg_class_oid))
1563 peter@eisentraut.org 1245 [ # # ]:UBC 0 : ereport(ERROR,
1246 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
1247 : : errmsg("pg_class heap OID value not set when in binary upgrade mode")));
1248 : :
1563 peter@eisentraut.org 1249 :CBC 883 : relid = binary_upgrade_next_heap_pg_class_oid;
1250 : 883 : binary_upgrade_next_heap_pg_class_oid = InvalidOid;
1251 : :
1518 rhaas@postgresql.org 1252 [ + + + - : 883 : if (RELKIND_HAS_STORAGE(relkind))
+ + + - +
+ ]
1253 : : {
1348 1254 [ - + ]: 727 : if (!RelFileNumberIsValid(binary_upgrade_next_heap_pg_class_relfilenumber))
1518 rhaas@postgresql.org 1255 [ # # ]:UBC 0 : ereport(ERROR,
1256 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
1257 : : errmsg("relfilenumber value not set when in binary upgrade mode")));
1258 : :
1348 rhaas@postgresql.org 1259 :CBC 727 : relfilenumber = binary_upgrade_next_heap_pg_class_relfilenumber;
1260 : 727 : binary_upgrade_next_heap_pg_class_relfilenumber = InvalidRelFileNumber;
1261 : : }
1262 : : }
1263 : : }
1264 : :
1563 peter@eisentraut.org 1265 [ + + ]: 41512 : if (!OidIsValid(relid))
1264 rhaas@postgresql.org 1266 : 40359 : relid = GetNewRelFileNumber(reltablespace, pg_class_desc,
1267 : : relpersistence);
1268 : : }
1269 : :
1270 : : /*
1271 : : * Other sessions' catalog scans can't find this until we commit. Hence,
1272 : : * it doesn't hurt to hold AccessExclusiveLock. Do it here so callers
1273 : : * can't accidentally vary in their lock mode or acquisition timing.
1274 : : */
626 noah@leadboat.com 1275 : 46357 : LockRelationOid(relid, AccessExclusiveLock);
1276 : :
1277 : : /*
1278 : : * Determine the relation's initial permissions.
1279 : : */
6005 tgl@sss.pgh.pa.us 1280 [ + + ]: 46357 : if (use_user_acl)
1281 : : {
1282 [ + + + ]: 33066 : switch (relkind)
1283 : : {
1284 : 29866 : case RELKIND_RELATION:
1285 : : case RELKIND_VIEW:
1286 : : case RELKIND_MATVIEW:
1287 : : case RELKIND_FOREIGN_TABLE:
1288 : : case RELKIND_PARTITIONED_TABLE:
3077 peter_e@gmx.net 1289 : 29866 : relacl = get_user_default_acl(OBJECT_TABLE, ownerid,
1290 : : relnamespace);
6005 tgl@sss.pgh.pa.us 1291 : 29866 : break;
1292 : 951 : case RELKIND_SEQUENCE:
3077 peter_e@gmx.net 1293 : 951 : relacl = get_user_default_acl(OBJECT_SEQUENCE, ownerid,
1294 : : relnamespace);
6005 tgl@sss.pgh.pa.us 1295 : 951 : break;
1296 : 2249 : default:
1297 : 2249 : relacl = NULL;
1298 : 2249 : break;
1299 : : }
1300 : : }
1301 : : else
1302 : 13291 : relacl = NULL;
1303 : :
1304 : : /*
1305 : : * Create the relcache entry (mostly dummy at this point) and the physical
1306 : : * disk file. (If we fail further down, it's the smgr's responsibility to
1307 : : * remove the disk file again.)
1308 : : *
1309 : : * NB: Note that passing create_storage = true is correct even for binary
1310 : : * upgrade. The storage we create here will be replaced later, but we
1311 : : * need to have something on disk in the meanwhile.
1312 : : */
8723 1313 : 46357 : new_rel_desc = heap_create(relname,
1314 : : relnamespace,
1315 : : reltablespace,
1316 : : relid,
1317 : : relfilenumber,
1318 : : accessmtd,
1319 : : tupdesc,
1320 : : relkind,
1321 : : relpersistence,
1322 : : shared_relation,
1323 : : mapped_relation,
1324 : : allow_system_table_mods,
1325 : : &relfrozenxid,
1326 : : &relminmxid,
1327 : : true);
1328 : :
7520 1329 [ - + ]: 46353 : Assert(relid == RelationGetRelid(new_rel_desc));
1330 : :
2916 peter_e@gmx.net 1331 : 46353 : new_rel_desc->rd_rel->relrewrite = relrewrite;
1332 : :
1333 : : /*
1334 : : * Decide whether to create a pg_type entry for the relation's rowtype.
1335 : : * These types are made except where the use of a relation as such is an
1336 : : * implementation detail: toast tables, sequences and indexes.
1337 : : */
2078 tgl@sss.pgh.pa.us 1338 [ + + + + : 82720 : if (!(relkind == RELKIND_SEQUENCE ||
+ - ]
1339 [ + - ]: 36367 : relkind == RELKIND_TOASTVALUE ||
1340 : : relkind == RELKIND_INDEX ||
1341 : : relkind == RELKIND_PARTITIONED_INDEX))
6883 1342 : 36367 : {
1343 : : Oid new_array_oid;
1344 : : ObjectAddress new_type_addr;
1345 : : char *relarrayname;
1346 : :
1347 : : /*
1348 : : * We'll make an array over the composite type, too. For largely
1349 : : * historical reasons, the array type's OID is assigned first.
1350 : : */
2077 1351 : 36367 : new_array_oid = AssignTypeArrayOid();
1352 : :
1353 : : /*
1354 : : * Make the pg_type entry for the composite type. The OID of the
1355 : : * composite type can be preselected by the caller, but if reltypeid
1356 : : * is InvalidOid, we'll generate a new OID for it.
1357 : : *
1358 : : * NOTE: we could get a unique-index failure here, in case someone
1359 : : * else is creating the same type name in parallel but hadn't
1360 : : * committed yet when we checked for a duplicate name above.
1361 : : */
1362 : 36367 : new_type_addr = AddNewRelationType(relname,
1363 : : relnamespace,
1364 : : relid,
1365 : : relkind,
1366 : : ownerid,
1367 : : reltypeid,
1368 : : new_array_oid);
1369 : 36367 : new_type_oid = new_type_addr.objectId;
1370 [ + + ]: 36367 : if (typaddress)
1371 : 2249 : *typaddress = new_type_addr;
1372 : :
1373 : : /* Now create the array type. */
6883 1374 : 36367 : relarrayname = makeArrayTypeName(relname, relnamespace);
1375 : :
3189 1376 : 36367 : TypeCreate(new_array_oid, /* force the type's OID to this */
1377 : : relarrayname, /* Array type name */
1378 : : relnamespace, /* Same namespace as parent */
1379 : : InvalidOid, /* Not composite, no relationOid */
1380 : : 0, /* relkind, also N/A here */
1381 : : ownerid, /* owner's ID */
1382 : : -1, /* Internal size (varlena) */
1383 : : TYPTYPE_BASE, /* Not composite - typelem is */
1384 : : TYPCATEGORY_ARRAY, /* type-category (array) */
1385 : : false, /* array types are never preferred */
1386 : : DEFAULT_TYPDELIM, /* default array delimiter */
1387 : : F_ARRAY_IN, /* array input proc */
1388 : : F_ARRAY_OUT, /* array output proc */
1389 : : F_ARRAY_RECV, /* array recv (bin) proc */
1390 : : F_ARRAY_SEND, /* array send (bin) proc */
1391 : : InvalidOid, /* typmodin procedure - none */
1392 : : InvalidOid, /* typmodout procedure - none */
1393 : : F_ARRAY_TYPANALYZE, /* array analyze procedure */
1394 : : F_ARRAY_SUBSCRIPT_HANDLER, /* array subscript procedure */
1395 : : new_type_oid, /* array element type - the rowtype */
1396 : : true, /* yes, this is an array type */
1397 : : InvalidOid, /* this has no array type */
1398 : : InvalidOid, /* domain base type - irrelevant */
1399 : : NULL, /* default value - none */
1400 : : NULL, /* default binary representation */
1401 : : false, /* passed by reference */
1402 : : TYPALIGN_DOUBLE, /* alignment - must be the largest! */
1403 : : TYPSTORAGE_EXTENDED, /* fully TOASTable */
1404 : : -1, /* typmod */
1405 : : 0, /* array dimensions for typBaseType */
1406 : : false, /* Type NOT NULL */
1407 : : InvalidOid); /* rowtypes never have a collation */
1408 : :
6883 1409 : 36367 : pfree(relarrayname);
1410 : : }
1411 : : else
1412 : : {
1413 : : /* Caller should not be expecting a type to be created. */
2077 1414 [ - + ]: 9986 : Assert(reltypeid == InvalidOid);
1415 [ - + ]: 9986 : Assert(typaddress == NULL);
1416 : :
1417 : 9986 : new_type_oid = InvalidOid;
1418 : : }
1419 : :
1420 : : /*
1421 : : * now create an entry in pg_class for the relation.
1422 : : *
1423 : : * NOTE: we could get a unique-index failure here, in case someone else is
1424 : : * creating the same relation name in parallel but hadn't committed yet
1425 : : * when we checked for a duplicate name above.
1426 : : */
9903 bruce@momjian.us 1427 : 46353 : AddNewRelationTuple(pg_class_desc,
1428 : : new_rel_desc,
1429 : : relid,
1430 : : new_type_oid,
1431 : : reloftypeid,
1432 : : ownerid,
1433 : : relkind,
1434 : : relfrozenxid,
1435 : : relminmxid,
1436 : : PointerGetDatum(relacl),
1437 : : reloptions);
1438 : :
1439 : : /*
1440 : : * now add tuples to pg_attribute for the attributes in our new relation.
1441 : : */
2672 andres@anarazel.de 1442 : 46353 : AddNewAttributeTuples(relid, new_rel_desc->rd_att, relkind);
1443 : :
1444 : : /*
1445 : : * Make a dependency link to force the relation to be deleted if its
1446 : : * namespace is. Also make a dependency link to its owner, as well as
1447 : : * dependencies for any roles mentioned in the default ACL.
1448 : : *
1449 : : * For composite types, these dependencies are tracked for the pg_type
1450 : : * entry, so we needn't record them here. Likewise, TOAST tables don't
1451 : : * need a namespace dependency (they live in a pinned namespace) nor an
1452 : : * owner dependency (they depend indirectly through the parent table), nor
1453 : : * should they have any ACL entries. The same applies for extension
1454 : : * dependencies.
1455 : : *
1456 : : * Also, skip this in bootstrap mode, since we don't make dependencies
1457 : : * while bootstrapping.
1458 : : */
6883 tgl@sss.pgh.pa.us 1459 [ + + + + ]: 46353 : if (relkind != RELKIND_COMPOSITE_TYPE &&
6880 1460 : 35069 : relkind != RELKIND_TOASTVALUE &&
6883 1461 [ + + ]: 35069 : !IsBootstrapProcessingMode())
1462 : : {
1463 : : ObjectAddress myself,
1464 : : referenced;
1465 : : ObjectAddresses *addrs;
1466 : :
2017 michael@paquier.xyz 1467 : 32009 : ObjectAddressSet(myself, RelationRelationId, relid);
1468 : :
6883 tgl@sss.pgh.pa.us 1469 : 32009 : recordDependencyOnOwner(RelationRelationId, relid, ownerid);
1470 : :
2683 1471 : 32009 : recordDependencyOnNewAcl(RelationRelationId, relid, 0, ownerid, relacl);
1472 : :
3390 1473 : 32009 : recordDependencyOnCurrentExtension(&myself, false);
1474 : :
2017 michael@paquier.xyz 1475 : 32009 : addrs = new_object_addresses();
1476 : :
1477 : 32009 : ObjectAddressSet(referenced, NamespaceRelationId, relnamespace);
1478 : 32009 : add_exact_object_address(&referenced, addrs);
1479 : :
5890 peter_e@gmx.net 1480 [ + + ]: 32009 : if (reloftypeid)
1481 : : {
2017 michael@paquier.xyz 1482 : 34 : ObjectAddressSet(referenced, TypeRelationId, reloftypeid);
1483 : 34 : add_exact_object_address(&referenced, addrs);
1484 : : }
1485 : :
1486 : : /*
1487 : : * Make a dependency link to force the relation to be deleted if its
1488 : : * access method is.
1489 : : *
1490 : : * No need to add an explicit dependency for the toast table, as the
1491 : : * main table depends on it. Partitioned tables may not have an
1492 : : * access method set.
1493 : : */
502 1494 [ + + + - : 32009 : if ((RELKIND_HAS_TABLE_AM(relkind) && relkind != RELKIND_TOASTVALUE) ||
+ + - + +
+ ]
1495 [ + + ]: 2797 : (relkind == RELKIND_PARTITIONED_TABLE && OidIsValid(accessmtd)))
1496 : : {
2017 1497 : 19373 : ObjectAddressSet(referenced, AccessMethodRelationId, accessmtd);
1498 : 19373 : add_exact_object_address(&referenced, addrs);
1499 : : }
1500 : :
1501 : 32009 : record_object_address_dependencies(&myself, addrs, DEPENDENCY_NORMAL);
1502 : 32009 : free_object_addresses(addrs);
1503 : : }
1504 : :
1505 : : /* Post creation hook for new relation */
4757 rhaas@postgresql.org 1506 [ + + ]: 46353 : InvokeObjectPostCreateHookArg(RelationRelationId, relid, 0, is_internal);
1507 : :
1508 : : /*
1509 : : * Store any supplied CHECK constraints and defaults.
1510 : : *
1511 : : * NB: this may do a CommandCounterIncrement and rebuild the relcache
1512 : : * entry, so the relation must be valid and self-consistent at this point.
1513 : : * In particular, there are not yet constraints and defaults anywhere.
1514 : : */
4746 1515 : 46353 : StoreConstraints(new_rel_desc, cooked_constraints, is_internal);
1516 : :
1517 : : /*
1518 : : * If there's a special on-commit action, remember it
1519 : : */
8525 tgl@sss.pgh.pa.us 1520 [ + + ]: 46353 : if (oncommit != ONCOMMIT_NOOP)
7520 1521 : 91 : register_on_commit_action(relid, oncommit);
1522 : :
1523 : : /*
1524 : : * ok, the relation has been cataloged, so close our relations and return
1525 : : * the OID of the newly created relation.
1526 : : */
2610 andres@anarazel.de 1527 : 46353 : table_close(new_rel_desc, NoLock); /* do not unlock till end of xact */
1528 : 46353 : table_close(pg_class_desc, RowExclusiveLock);
1529 : :
7520 tgl@sss.pgh.pa.us 1530 : 46353 : return relid;
1531 : : }
1532 : :
1533 : : /*
1534 : : * RelationRemoveInheritance
1535 : : *
1536 : : * Formerly, this routine checked for child relations and aborted the
1537 : : * deletion if any were found. Now we rely on the dependency mechanism
1538 : : * to check for or delete child relations. By the time we get here,
1539 : : * there are no children and we need only remove any pg_inherits rows
1540 : : * linking this relation to its parent(s).
1541 : : */
1542 : : static void
7869 1543 : 26064 : RelationRemoveInheritance(Oid relid)
1544 : : {
1545 : : Relation catalogRelation;
1546 : : SysScanDesc scan;
1547 : : ScanKeyData key;
1548 : : HeapTuple tuple;
1549 : :
2610 andres@anarazel.de 1550 : 26064 : catalogRelation = table_open(InheritsRelationId, RowExclusiveLock);
1551 : :
8159 tgl@sss.pgh.pa.us 1552 : 26064 : ScanKeyInit(&key,
1553 : : Anum_pg_inherits_inhrelid,
1554 : : BTEqualStrategyNumber, F_OIDEQ,
1555 : : ObjectIdGetDatum(relid));
1556 : :
7640 1557 : 26064 : scan = systable_beginscan(catalogRelation, InheritsRelidSeqnoIndexId, true,
1558 : : NULL, 1, &key);
1559 : :
8647 1560 [ + + ]: 31243 : while (HeapTupleIsValid(tuple = systable_getnext(scan)))
3329 1561 : 5179 : CatalogTupleDelete(catalogRelation, &tuple->t_self);
1562 : :
8647 1563 : 26064 : systable_endscan(scan);
2610 andres@anarazel.de 1564 : 26064 : table_close(catalogRelation, RowExclusiveLock);
10841 scrappy@hub.org 1565 : 26064 : }
1566 : :
1567 : : /*
1568 : : * DeleteRelationTuple
1569 : : *
1570 : : * Remove pg_class row for the given relid.
1571 : : *
1572 : : * Note: this is shared by relation deletion and index deletion. It's
1573 : : * not intended for use anyplace else.
1574 : : */
1575 : : void
8645 tgl@sss.pgh.pa.us 1576 : 38794 : DeleteRelationTuple(Oid relid)
1577 : : {
1578 : : Relation pg_class_desc;
1579 : : HeapTuple tup;
1580 : :
1581 : : /* Grab an appropriate lock on the pg_class relation */
2610 andres@anarazel.de 1582 : 38794 : pg_class_desc = table_open(RelationRelationId, RowExclusiveLock);
1583 : :
5873 rhaas@postgresql.org 1584 : 38794 : tup = SearchSysCache1(RELOID, ObjectIdGetDatum(relid));
10070 bruce@momjian.us 1585 [ - + ]: 38794 : if (!HeapTupleIsValid(tup))
8274 tgl@sss.pgh.pa.us 1586 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for relation %u", relid);
1587 : :
1588 : : /* delete the relation tuple from pg_class, and finish up */
3329 tgl@sss.pgh.pa.us 1589 :CBC 38794 : CatalogTupleDelete(pg_class_desc, &tup->t_self);
1590 : :
8645 1591 : 38794 : ReleaseSysCache(tup);
1592 : :
2610 andres@anarazel.de 1593 : 38794 : table_close(pg_class_desc, RowExclusiveLock);
9670 bruce@momjian.us 1594 : 38794 : }
1595 : :
1596 : : /*
1597 : : * DeleteAttributeTuples
1598 : : *
1599 : : * Remove pg_attribute rows for the given relid.
1600 : : *
1601 : : * Note: this is shared by relation deletion and index deletion. It's
1602 : : * not intended for use anyplace else.
1603 : : */
1604 : : void
8645 tgl@sss.pgh.pa.us 1605 : 38794 : DeleteAttributeTuples(Oid relid)
1606 : : {
1607 : : Relation attrel;
1608 : : SysScanDesc scan;
1609 : : ScanKeyData key[1];
1610 : : HeapTuple atttup;
1611 : :
1612 : : /* Grab an appropriate lock on the pg_attribute relation */
2610 andres@anarazel.de 1613 : 38794 : attrel = table_open(AttributeRelationId, RowExclusiveLock);
1614 : :
1615 : : /* Use the index to scan only attributes of the target relation */
8159 tgl@sss.pgh.pa.us 1616 : 38794 : ScanKeyInit(&key[0],
1617 : : Anum_pg_attribute_attrelid,
1618 : : BTEqualStrategyNumber, F_OIDEQ,
1619 : : ObjectIdGetDatum(relid));
1620 : :
7640 1621 : 38794 : scan = systable_beginscan(attrel, AttributeRelidNumIndexId, true,
1622 : : NULL, 1, key);
1623 : :
1624 : : /* Delete all the matching tuples */
8645 1625 [ + + ]: 255694 : while ((atttup = systable_getnext(scan)) != NULL)
3329 1626 : 216900 : CatalogTupleDelete(attrel, &atttup->t_self);
1627 : :
1628 : : /* Clean up after the scan */
8645 1629 : 38794 : systable_endscan(scan);
2610 andres@anarazel.de 1630 : 38794 : table_close(attrel, RowExclusiveLock);
9648 bruce@momjian.us 1631 : 38794 : }
1632 : :
1633 : : /*
1634 : : * DeleteSystemAttributeTuples
1635 : : *
1636 : : * Remove pg_attribute rows for system columns of the given relid.
1637 : : *
1638 : : * Note: this is only used when converting a table to a view. Views don't
1639 : : * have system columns, so we should remove them from pg_attribute.
1640 : : */
1641 : : void
4890 tgl@sss.pgh.pa.us 1642 :UBC 0 : DeleteSystemAttributeTuples(Oid relid)
1643 : : {
1644 : : Relation attrel;
1645 : : SysScanDesc scan;
1646 : : ScanKeyData key[2];
1647 : : HeapTuple atttup;
1648 : :
1649 : : /* Grab an appropriate lock on the pg_attribute relation */
2610 andres@anarazel.de 1650 : 0 : attrel = table_open(AttributeRelationId, RowExclusiveLock);
1651 : :
1652 : : /* Use the index to scan only system attributes of the target relation */
4890 tgl@sss.pgh.pa.us 1653 : 0 : ScanKeyInit(&key[0],
1654 : : Anum_pg_attribute_attrelid,
1655 : : BTEqualStrategyNumber, F_OIDEQ,
1656 : : ObjectIdGetDatum(relid));
1657 : 0 : ScanKeyInit(&key[1],
1658 : : Anum_pg_attribute_attnum,
1659 : : BTLessEqualStrategyNumber, F_INT2LE,
1660 : : Int16GetDatum(0));
1661 : :
1662 : 0 : scan = systable_beginscan(attrel, AttributeRelidNumIndexId, true,
1663 : : NULL, 2, key);
1664 : :
1665 : : /* Delete all the matching tuples */
1666 [ # # ]: 0 : while ((atttup = systable_getnext(scan)) != NULL)
3329 1667 : 0 : CatalogTupleDelete(attrel, &atttup->t_self);
1668 : :
1669 : : /* Clean up after the scan */
4890 1670 : 0 : systable_endscan(scan);
2610 andres@anarazel.de 1671 : 0 : table_close(attrel, RowExclusiveLock);
4890 tgl@sss.pgh.pa.us 1672 : 0 : }
1673 : :
1674 : : /*
1675 : : * RemoveAttributeById
1676 : : *
1677 : : * This is the guts of ALTER TABLE DROP COLUMN: actually mark the attribute
1678 : : * deleted in pg_attribute. We also remove pg_statistic entries for it.
1679 : : * (Everything else needed, such as getting rid of any pg_attrdef entry,
1680 : : * is handled by dependency.c.)
1681 : : */
1682 : : void
8626 tgl@sss.pgh.pa.us 1683 :CBC 1070 : RemoveAttributeById(Oid relid, AttrNumber attnum)
1684 : : {
1685 : : Relation rel;
1686 : : Relation attr_rel;
1687 : : HeapTuple tuple;
1688 : : Form_pg_attribute attStruct;
1689 : : char newattname[NAMEDATALEN];
814 peter@eisentraut.org 1690 : 1070 : Datum valuesAtt[Natts_pg_attribute] = {0};
1691 : 1070 : bool nullsAtt[Natts_pg_attribute] = {0};
1692 : 1070 : bool replacesAtt[Natts_pg_attribute] = {0};
1693 : :
1694 : : /*
1695 : : * Grab an exclusive lock on the target table, which we will NOT release
1696 : : * until end of transaction. (In the simple case where we are directly
1697 : : * dropping this column, ATExecDropColumn already did this ... but when
1698 : : * cascading from a drop of some other object, we may not have any lock.)
1699 : : */
8599 tgl@sss.pgh.pa.us 1700 : 1070 : rel = relation_open(relid, AccessExclusiveLock);
1701 : :
2610 andres@anarazel.de 1702 : 1070 : attr_rel = table_open(AttributeRelationId, RowExclusiveLock);
1703 : :
5873 rhaas@postgresql.org 1704 : 1070 : tuple = SearchSysCacheCopy2(ATTNUM,
1705 : : ObjectIdGetDatum(relid),
1706 : : Int16GetDatum(attnum));
3189 tgl@sss.pgh.pa.us 1707 [ - + ]: 1070 : if (!HeapTupleIsValid(tuple)) /* shouldn't happen */
8273 tgl@sss.pgh.pa.us 1708 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for attribute %d of relation %u",
1709 : : attnum, relid);
8626 tgl@sss.pgh.pa.us 1710 :CBC 1070 : attStruct = (Form_pg_attribute) GETSTRUCT(tuple);
1711 : :
1712 : : /* Mark the attribute as dropped */
977 peter@eisentraut.org 1713 : 1070 : attStruct->attisdropped = true;
1714 : :
1715 : : /*
1716 : : * Set the type OID to invalid. A dropped attribute's type link cannot be
1717 : : * relied on (once the attribute is dropped, the type might be too).
1718 : : * Fortunately we do not need the type row --- the only really essential
1719 : : * information is the type's typlen and typalign, which are preserved in
1720 : : * the attribute's attlen and attalign. We set atttypid to zero here as a
1721 : : * means of catching code that incorrectly expects it to be valid.
1722 : : */
1723 : 1070 : attStruct->atttypid = InvalidOid;
1724 : :
1725 : : /* Remove any not-null constraint the column may have */
1726 : 1070 : attStruct->attnotnull = false;
1727 : :
1728 : : /* Unset this so no one tries to look up the generation expression */
1729 : 1070 : attStruct->attgenerated = '\0';
1730 : :
1731 : : /*
1732 : : * Change the column name to something that isn't likely to conflict
1733 : : */
1734 : 1070 : snprintf(newattname, sizeof(newattname),
1735 : : "........pg.dropped.%d........", attnum);
1736 : 1070 : namestrcpy(&(attStruct->attname), newattname);
1737 : :
1738 : : /* Clear the missing value */
814 1739 : 1070 : attStruct->atthasmissing = false;
1740 : 1070 : nullsAtt[Anum_pg_attribute_attmissingval - 1] = true;
1741 : 1070 : replacesAtt[Anum_pg_attribute_attmissingval - 1] = true;
1742 : :
1743 : : /*
1744 : : * Clear the other nullable fields. This saves some space in pg_attribute
1745 : : * and removes no longer useful information.
1746 : : */
792 1747 : 1070 : nullsAtt[Anum_pg_attribute_attstattarget - 1] = true;
1748 : 1070 : replacesAtt[Anum_pg_attribute_attstattarget - 1] = true;
814 1749 : 1070 : nullsAtt[Anum_pg_attribute_attacl - 1] = true;
1750 : 1070 : replacesAtt[Anum_pg_attribute_attacl - 1] = true;
1751 : 1070 : nullsAtt[Anum_pg_attribute_attoptions - 1] = true;
1752 : 1070 : replacesAtt[Anum_pg_attribute_attoptions - 1] = true;
1753 : 1070 : nullsAtt[Anum_pg_attribute_attfdwoptions - 1] = true;
1754 : 1070 : replacesAtt[Anum_pg_attribute_attfdwoptions - 1] = true;
1755 : :
1756 : 1070 : tuple = heap_modify_tuple(tuple, RelationGetDescr(attr_rel),
1757 : : valuesAtt, nullsAtt, replacesAtt);
1758 : :
977 1759 : 1070 : CatalogTupleUpdate(attr_rel, &tuple->t_self, tuple);
1760 : :
1761 : : /*
1762 : : * Because updating the pg_attribute row will trigger a relcache flush for
1763 : : * the target relation, we need not do anything else to notify other
1764 : : * backends of the change.
1765 : : */
1766 : :
2610 andres@anarazel.de 1767 : 1070 : table_close(attr_rel, RowExclusiveLock);
1768 : :
977 peter@eisentraut.org 1769 : 1070 : RemoveStatistics(relid, attnum);
1770 : :
8599 tgl@sss.pgh.pa.us 1771 : 1070 : relation_close(rel, NoLock);
8626 1772 : 1070 : }
1773 : :
1774 : : /*
1775 : : * heap_drop_with_catalog - removes specified relation from catalogs
1776 : : *
1777 : : * Note that this routine is not responsible for dropping objects that are
1778 : : * linked to the pg_class entry via dependencies (for example, indexes and
1779 : : * constraints). Those are deleted by the dependency-tracing logic in
1780 : : * dependency.c before control gets here. In general, therefore, this routine
1781 : : * should never be called directly; go through performDeletion() instead.
1782 : : */
1783 : : void
7869 1784 : 26067 : heap_drop_with_catalog(Oid relid)
1785 : : {
1786 : : Relation rel;
1787 : : HeapTuple tuple;
3110 rhaas@postgresql.org 1788 : 26067 : Oid parentOid = InvalidOid,
1789 : 26067 : defaultPartOid = InvalidOid;
1790 : :
1791 : : /*
1792 : : * To drop a partition safely, we must grab exclusive lock on its parent,
1793 : : * because another backend might be about to execute a query on the parent
1794 : : * table. If it relies on previously cached partition descriptor, then it
1795 : : * could attempt to access the just-dropped relation as its partition. We
1796 : : * must therefore take a table lock strong enough to prevent all queries
1797 : : * on the table from proceeding until we commit and send out a
1798 : : * shared-cache-inval notice that will make them update their partition
1799 : : * descriptors.
1800 : : */
3260 1801 : 26067 : tuple = SearchSysCache1(RELOID, ObjectIdGetDatum(relid));
3030 tgl@sss.pgh.pa.us 1802 [ - + ]: 26067 : if (!HeapTupleIsValid(tuple))
3030 tgl@sss.pgh.pa.us 1803 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for relation %u", relid);
3260 rhaas@postgresql.org 1804 [ + + ]:CBC 26067 : if (((Form_pg_class) GETSTRUCT(tuple))->relispartition)
1805 : : {
1806 : : /*
1807 : : * We have to lock the parent if the partition is being detached,
1808 : : * because it's possible that some query still has a partition
1809 : : * descriptor that includes this partition.
1810 : : */
1816 alvherre@alvh.no-ip. 1811 : 4221 : parentOid = get_partition_parent(relid, true);
3243 rhaas@postgresql.org 1812 : 4221 : LockRelationOid(parentOid, AccessExclusiveLock);
1813 : :
1814 : : /*
1815 : : * If this is not the default partition, dropping it will change the
1816 : : * default partition's partition constraint, so we must lock it.
1817 : : */
3110 1818 : 4221 : defaultPartOid = get_default_partition_oid(parentOid);
1819 [ + + + + ]: 4221 : if (OidIsValid(defaultPartOid) && relid != defaultPartOid)
1820 : 277 : LockRelationOid(defaultPartOid, AccessExclusiveLock);
1821 : : }
1822 : :
3260 1823 : 26067 : ReleaseSysCache(tuple);
1824 : :
1825 : : /*
1826 : : * Open and lock the relation.
1827 : : */
1828 : 26067 : rel = relation_open(relid, AccessExclusiveLock);
1829 : :
1830 : : /*
1831 : : * There can no longer be anyone *else* touching the relation, but we
1832 : : * might still have open queries or cursors, or pending trigger events, in
1833 : : * our own session.
1834 : : */
5507 tgl@sss.pgh.pa.us 1835 : 26067 : CheckTableNotInUse(rel, "DROP TABLE");
1836 : :
1837 : : /*
1838 : : * This effectively deletes all rows in the table, and may be done in a
1839 : : * serializable transaction. In that case we must record a rw-conflict in
1840 : : * to this transaction from each transaction holding a predicate lock on
1841 : : * the table.
1842 : : */
5394 heikki.linnakangas@i 1843 : 26064 : CheckTableForSerializableConflictIn(rel);
1844 : :
1845 : : /*
1846 : : * Delete pg_foreign_table tuple first.
1847 : : */
5552 rhaas@postgresql.org 1848 [ + + ]: 26064 : if (rel->rd_rel->relkind == RELKIND_FOREIGN_TABLE)
1849 : : {
1850 : : Relation ftrel;
1851 : : HeapTuple fttuple;
1852 : :
1257 drowley@postgresql.o 1853 : 137 : ftrel = table_open(ForeignTableRelationId, RowExclusiveLock);
1854 : :
1855 : 137 : fttuple = SearchSysCache1(FOREIGNTABLEREL, ObjectIdGetDatum(relid));
1856 [ - + ]: 137 : if (!HeapTupleIsValid(fttuple))
5552 rhaas@postgresql.org 1857 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for foreign table %u", relid);
1858 : :
1257 drowley@postgresql.o 1859 :CBC 137 : CatalogTupleDelete(ftrel, &fttuple->t_self);
1860 : :
1861 : 137 : ReleaseSysCache(fttuple);
1862 : 137 : table_close(ftrel, RowExclusiveLock);
1863 : : }
1864 : :
1865 : : /*
1866 : : * If a partitioned table, delete the pg_partitioned_table tuple.
1867 : : */
3385 rhaas@postgresql.org 1868 [ + + ]: 26064 : if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
1869 : 2206 : RemovePartitionKeyByRelId(relid);
1870 : :
1871 : : /*
1872 : : * If the relation being dropped is the default partition itself,
1873 : : * invalidate its entry in pg_partitioned_table.
1874 : : */
3110 1875 [ + + ]: 26064 : if (relid == defaultPartOid)
1876 : 291 : update_default_partition_oid(parentOid, InvalidOid);
1877 : :
1878 : : /*
1879 : : * Schedule unlinking of the relation's physical files at commit.
1880 : : */
2102 peter@eisentraut.org 1881 [ + + + - : 26064 : if (RELKIND_HAS_STORAGE(rel->rd_rel->relkind))
+ + + + +
+ ]
6325 heikki.linnakangas@i 1882 : 20508 : RelationDropStorage(rel);
1883 : :
1884 : : /* ensure that stats are dropped if transaction commits */
1439 andres@anarazel.de 1885 : 26064 : pgstat_drop_relation(rel);
1886 : :
1887 : : /*
1888 : : * Close relcache entry, but *keep* AccessExclusiveLock on the relation
1889 : : * until transaction commit. This ensures no one else will try to do
1890 : : * something with the doomed relation.
1891 : : */
7869 tgl@sss.pgh.pa.us 1892 : 26064 : relation_close(rel, NoLock);
1893 : :
1894 : : /*
1895 : : * Remove any associated relation synchronization states.
1896 : : */
3279 peter_e@gmx.net 1897 : 26064 : RemoveSubscriptionRel(InvalidOid, relid);
1898 : :
1899 : : /*
1900 : : * Forget any ON COMMIT action for the rel
1901 : : */
7869 tgl@sss.pgh.pa.us 1902 : 26064 : remove_on_commit_action(relid);
1903 : :
1904 : : /*
1905 : : * Flush the relation from the relcache. We want to do this before
1906 : : * starting to remove catalog entries, just to be certain that no relcache
1907 : : * entry rebuild will happen partway through. (That should not really
1908 : : * matter, since we don't do CommandCounterIncrement here, but let's be
1909 : : * safe.)
1910 : : */
1911 : 26064 : RelationForgetRelation(relid);
1912 : :
1913 : : /*
1914 : : * remove inheritance information
1915 : : */
1916 : 26064 : RelationRemoveInheritance(relid);
1917 : :
1918 : : /*
1919 : : * delete statistics
1920 : : */
1921 : 26064 : RemoveStatistics(relid, 0);
1922 : :
1923 : : /*
1924 : : * delete attribute tuples
1925 : : */
1926 : 26064 : DeleteAttributeTuples(relid);
1927 : :
1928 : : /*
1929 : : * delete relation tuple
1930 : : */
1931 : 26064 : DeleteRelationTuple(relid);
1932 : :
3243 rhaas@postgresql.org 1933 [ + + ]: 26064 : if (OidIsValid(parentOid))
1934 : : {
1935 : : /*
1936 : : * If this is not the default partition, the partition constraint of
1937 : : * the default partition has changed to include the portion of the key
1938 : : * space previously covered by the dropped partition.
1939 : : */
3110 1940 [ + + + + ]: 4221 : if (OidIsValid(defaultPartOid) && relid != defaultPartOid)
1941 : 277 : CacheInvalidateRelcacheByRelid(defaultPartOid);
1942 : :
1943 : : /*
1944 : : * Invalidate the parent's relcache so that the partition is no longer
1945 : : * included in its partition descriptor.
1946 : : */
3243 1947 : 4221 : CacheInvalidateRelcacheByRelid(parentOid);
1948 : : /* keep the lock */
1949 : : }
10721 scrappy@hub.org 1950 : 26064 : }
1951 : :
1952 : :
1953 : : /*
1954 : : * RelationClearMissing
1955 : : *
1956 : : * Set atthasmissing and attmissingval to false/null for all attributes
1957 : : * where they are currently set. This can be safely and usefully done if
1958 : : * the table is rewritten (e.g. by VACUUM FULL or CLUSTER) where we know there
1959 : : * are no rows left with less than a full complement of attributes.
1960 : : *
1961 : : * The caller must have an AccessExclusive lock on the relation.
1962 : : */
1963 : : void
2909 andrew@dunslane.net 1964 : 1481 : RelationClearMissing(Relation rel)
1965 : : {
1966 : : Relation attr_rel;
1967 : 1481 : Oid relid = RelationGetRelid(rel);
1968 : 1481 : int natts = RelationGetNumberOfAttributes(rel);
1969 : : int attnum;
1970 : : Datum repl_val[Natts_pg_attribute];
1971 : : bool repl_null[Natts_pg_attribute];
1972 : : bool repl_repl[Natts_pg_attribute];
1973 : : Form_pg_attribute attrtuple;
1974 : : HeapTuple tuple,
1975 : : newtuple;
1976 : :
1977 : 1481 : memset(repl_val, 0, sizeof(repl_val));
1978 : 1481 : memset(repl_null, false, sizeof(repl_null));
1979 : 1481 : memset(repl_repl, false, sizeof(repl_repl));
1980 : :
1981 : 1481 : repl_val[Anum_pg_attribute_atthasmissing - 1] = BoolGetDatum(false);
1982 : 1481 : repl_null[Anum_pg_attribute_attmissingval - 1] = true;
1983 : :
1984 : 1481 : repl_repl[Anum_pg_attribute_atthasmissing - 1] = true;
1985 : 1481 : repl_repl[Anum_pg_attribute_attmissingval - 1] = true;
1986 : :
1987 : :
1988 : : /* Get a lock on pg_attribute */
2610 andres@anarazel.de 1989 : 1481 : attr_rel = table_open(AttributeRelationId, RowExclusiveLock);
1990 : :
1991 : : /* process each non-system attribute, including any dropped columns */
2909 andrew@dunslane.net 1992 [ + + ]: 5336 : for (attnum = 1; attnum <= natts; attnum++)
1993 : : {
1994 : 3855 : tuple = SearchSysCache2(ATTNUM,
1995 : : ObjectIdGetDatum(relid),
1996 : : Int16GetDatum(attnum));
1997 [ - + ]: 3855 : if (!HeapTupleIsValid(tuple)) /* shouldn't happen */
2909 andrew@dunslane.net 1998 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for attribute %d of relation %u",
1999 : : attnum, relid);
2000 : :
2909 andrew@dunslane.net 2001 :CBC 3855 : attrtuple = (Form_pg_attribute) GETSTRUCT(tuple);
2002 : :
2003 : : /* ignore any where atthasmissing is not true */
2004 [ + + ]: 3855 : if (attrtuple->atthasmissing)
2005 : : {
2006 : 69 : newtuple = heap_modify_tuple(tuple, RelationGetDescr(attr_rel),
2007 : : repl_val, repl_null, repl_repl);
2008 : :
2009 : 69 : CatalogTupleUpdate(attr_rel, &newtuple->t_self, newtuple);
2010 : :
2011 : 69 : heap_freetuple(newtuple);
2012 : : }
2013 : :
2014 : 3855 : ReleaseSysCache(tuple);
2015 : : }
2016 : :
2017 : : /*
2018 : : * Our update of the pg_attribute rows will force a relcache rebuild, so
2019 : : * there's nothing else to do here.
2020 : : */
2610 andres@anarazel.de 2021 : 1481 : table_close(attr_rel, RowExclusiveLock);
2909 andrew@dunslane.net 2022 : 1481 : }
2023 : :
2024 : : /*
2025 : : * StoreAttrMissingVal
2026 : : *
2027 : : * Set the missing value of a single attribute.
2028 : : */
2029 : : void
377 tgl@sss.pgh.pa.us 2030 : 270 : StoreAttrMissingVal(Relation rel, AttrNumber attnum, Datum missingval)
2031 : : {
2032 : 270 : Datum valuesAtt[Natts_pg_attribute] = {0};
2033 : 270 : bool nullsAtt[Natts_pg_attribute] = {0};
2034 : 270 : bool replacesAtt[Natts_pg_attribute] = {0};
2035 : : Relation attrrel;
2036 : : Form_pg_attribute attStruct;
2037 : : HeapTuple atttup,
2038 : : newtup;
2039 : :
2040 : : /* This is only supported for plain tables */
2041 [ - + ]: 270 : Assert(rel->rd_rel->relkind == RELKIND_RELATION);
2042 : :
2043 : : /* Fetch the pg_attribute row */
2044 : 270 : attrrel = table_open(AttributeRelationId, RowExclusiveLock);
2045 : :
2046 : 270 : atttup = SearchSysCache2(ATTNUM,
2047 : : ObjectIdGetDatum(RelationGetRelid(rel)),
2048 : : Int16GetDatum(attnum));
2049 [ - + ]: 270 : if (!HeapTupleIsValid(atttup)) /* shouldn't happen */
377 tgl@sss.pgh.pa.us 2050 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for attribute %d of relation %u",
2051 : : attnum, RelationGetRelid(rel));
377 tgl@sss.pgh.pa.us 2052 :CBC 270 : attStruct = (Form_pg_attribute) GETSTRUCT(atttup);
2053 : :
2054 : : /* Make a one-element array containing the value */
2055 : 270 : missingval = PointerGetDatum(construct_array(&missingval,
2056 : : 1,
2057 : : attStruct->atttypid,
2058 : 270 : attStruct->attlen,
2059 : 270 : attStruct->attbyval,
2060 : 270 : attStruct->attalign));
2061 : :
2062 : : /* Update the pg_attribute row */
2063 : 270 : valuesAtt[Anum_pg_attribute_atthasmissing - 1] = BoolGetDatum(true);
2064 : 270 : replacesAtt[Anum_pg_attribute_atthasmissing - 1] = true;
2065 : :
2066 : 270 : valuesAtt[Anum_pg_attribute_attmissingval - 1] = missingval;
2067 : 270 : replacesAtt[Anum_pg_attribute_attmissingval - 1] = true;
2068 : :
2069 : 270 : newtup = heap_modify_tuple(atttup, RelationGetDescr(attrrel),
2070 : : valuesAtt, nullsAtt, replacesAtt);
2071 : 270 : CatalogTupleUpdate(attrrel, &newtup->t_self, newtup);
2072 : :
2073 : : /* clean up */
2074 : 270 : ReleaseSysCache(atttup);
2075 : 270 : table_close(attrrel, RowExclusiveLock);
2076 : 270 : }
2077 : :
2078 : : /*
2079 : : * SetAttrMissing
2080 : : *
2081 : : * Set the missing value of a single attribute. This should only be used by
2082 : : * binary upgrade. Takes an AccessExclusive lock on the relation owning the
2083 : : * attribute.
2084 : : */
2085 : : void
2823 andrew@dunslane.net 2086 : 3 : SetAttrMissing(Oid relid, char *attname, char *value)
2087 : : {
1338 peter@eisentraut.org 2088 : 3 : Datum valuesAtt[Natts_pg_attribute] = {0};
2089 : 3 : bool nullsAtt[Natts_pg_attribute] = {0};
2090 : 3 : bool replacesAtt[Natts_pg_attribute] = {0};
2091 : : Datum missingval;
2092 : : Form_pg_attribute attStruct;
2093 : : Relation attrrel,
2094 : : tablerel;
2095 : : HeapTuple atttup,
2096 : : newtup;
2097 : :
2098 : : /* lock the table the attribute belongs to */
2610 andres@anarazel.de 2099 : 3 : tablerel = table_open(relid, AccessExclusiveLock);
2100 : :
2101 : : /* Don't do anything unless it's a plain table */
1731 andrew@dunslane.net 2102 [ - + ]: 3 : if (tablerel->rd_rel->relkind != RELKIND_RELATION)
2103 : : {
1731 andrew@dunslane.net 2104 :UBC 0 : table_close(tablerel, AccessExclusiveLock);
2105 : 0 : return;
2106 : : }
2107 : :
2108 : : /* Lock the attribute row and get the data */
2610 andres@anarazel.de 2109 :CBC 3 : attrrel = table_open(AttributeRelationId, RowExclusiveLock);
2823 andrew@dunslane.net 2110 : 3 : atttup = SearchSysCacheAttName(relid, attname);
2111 [ - + ]: 3 : if (!HeapTupleIsValid(atttup))
2823 andrew@dunslane.net 2112 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for attribute %s of relation %u",
2113 : : attname, relid);
2823 andrew@dunslane.net 2114 :CBC 3 : attStruct = (Form_pg_attribute) GETSTRUCT(atttup);
2115 : :
2116 : : /* get an array value from the value string */
2117 : 3 : missingval = OidFunctionCall3(F_ARRAY_IN,
2118 : : CStringGetDatum(value),
2119 : : ObjectIdGetDatum(attStruct->atttypid),
2120 : : Int32GetDatum(attStruct->atttypmod));
2121 : :
2122 : : /* update the tuple - set atthasmissing and attmissingval */
2123 : 3 : valuesAtt[Anum_pg_attribute_atthasmissing - 1] = BoolGetDatum(true);
2124 : 3 : replacesAtt[Anum_pg_attribute_atthasmissing - 1] = true;
2125 : 3 : valuesAtt[Anum_pg_attribute_attmissingval - 1] = missingval;
2126 : 3 : replacesAtt[Anum_pg_attribute_attmissingval - 1] = true;
2127 : :
2128 : 3 : newtup = heap_modify_tuple(atttup, RelationGetDescr(attrrel),
2129 : : valuesAtt, nullsAtt, replacesAtt);
2130 : 3 : CatalogTupleUpdate(attrrel, &newtup->t_self, newtup);
2131 : :
2132 : : /* clean up */
2133 : 3 : ReleaseSysCache(atttup);
2610 andres@anarazel.de 2134 : 3 : table_close(attrrel, RowExclusiveLock);
2135 : 3 : table_close(tablerel, AccessExclusiveLock);
2136 : : }
2137 : :
2138 : : /*
2139 : : * Store a check-constraint expression for the given relation.
2140 : : *
2141 : : * Caller is responsible for updating the count of constraints
2142 : : * in the pg_class entry for the relation.
2143 : : *
2144 : : * The OID of the new constraint is returned.
2145 : : */
2146 : : static Oid
3057 peter_e@gmx.net 2147 : 1571 : StoreRelCheck(Relation rel, const char *ccname, Node *expr,
2148 : : bool is_enforced, bool is_validated, bool is_local,
2149 : : int16 inhcount, bool is_no_inherit, bool is_internal)
2150 : : {
2151 : : char *ccbin;
2152 : : List *varList;
2153 : : int keycount;
2154 : : int16 *attNos;
2155 : : Oid constrOid;
2156 : :
2157 : : /*
2158 : : * Flatten expression to string form for storage.
2159 : : */
6519 tgl@sss.pgh.pa.us 2160 : 1571 : ccbin = nodeToString(expr);
2161 : :
2162 : : /*
2163 : : * Find columns of rel that are used in expr
2164 : : *
2165 : : * NB: pull_var_clause is okay here only because we don't allow subselects
2166 : : * in check constraints; it would fail to examine the contents of
2167 : : * subselects.
2168 : : */
3657 2169 : 1571 : varList = pull_var_clause(expr, 0);
7963 neilc@samurai.com 2170 : 1571 : keycount = list_length(varList);
2171 : :
8647 tgl@sss.pgh.pa.us 2172 [ + + ]: 1571 : if (keycount > 0)
2173 : : {
2174 : : ListCell *vl;
2175 : 1564 : int i = 0;
2176 : :
2177 : 1564 : attNos = (int16 *) palloc(keycount * sizeof(int16));
2178 [ + - + + : 3362 : foreach(vl, varList)
+ + ]
2179 : : {
8593 bruce@momjian.us 2180 : 1798 : Var *var = (Var *) lfirst(vl);
2181 : : int j;
2182 : :
8647 tgl@sss.pgh.pa.us 2183 [ + + ]: 1923 : for (j = 0; j < i; j++)
2184 [ + + ]: 244 : if (attNos[j] == var->varattno)
2185 : 119 : break;
2186 [ + + ]: 1798 : if (j == i)
2187 : 1679 : attNos[i++] = var->varattno;
2188 : : }
2189 : 1564 : keycount = i;
2190 : : }
2191 : : else
2192 : 7 : attNos = NULL;
2193 : :
2194 : : /*
2195 : : * Partitioned tables do not contain any rows themselves, so a NO INHERIT
2196 : : * constraint makes no sense.
2197 : : */
3385 rhaas@postgresql.org 2198 [ + + ]: 1571 : if (is_no_inherit &&
2199 [ + + ]: 62 : rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
2200 [ + - ]: 12 : ereport(ERROR,
2201 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
2202 : : errmsg("cannot add NO INHERIT constraint to partitioned table \"%s\"",
2203 : : RelationGetRelationName(rel))));
2204 : :
2205 : : /*
2206 : : * Create the Check Constraint
2207 : : */
2208 : : constrOid =
4008 alvherre@alvh.no-ip. 2209 : 1559 : CreateConstraintEntry(ccname, /* Constraint Name */
3189 tgl@sss.pgh.pa.us 2210 : 1559 : RelationGetNamespace(rel), /* namespace */
2211 : : CONSTRAINT_CHECK, /* Constraint Type */
2212 : : false, /* Is Deferrable */
2213 : : false, /* Is Deferred */
2214 : : is_enforced, /* Is Enforced */
2215 : : is_validated,
2216 : : InvalidOid, /* no parent constraint */
2217 : : RelationGetRelid(rel), /* relation */
2218 : : attNos, /* attrs in the constraint */
2219 : : keycount, /* # key attrs in the constraint */
2220 : : keycount, /* # total attrs in the constraint */
2221 : : InvalidOid, /* not a domain constraint */
2222 : : InvalidOid, /* no associated index */
2223 : : InvalidOid, /* Foreign key fields */
2224 : : NULL,
2225 : : NULL,
2226 : : NULL,
2227 : : NULL,
2228 : : 0,
2229 : : ' ',
2230 : : ' ',
2231 : : NULL,
2232 : : 0,
2233 : : ' ',
2234 : : NULL, /* not an exclusion constraint */
2235 : : expr, /* Tree form of check constraint */
2236 : : ccbin, /* Binary form of check constraint */
2237 : : is_local, /* conislocal */
2238 : : inhcount, /* coninhcount */
2239 : : is_no_inherit, /* connoinherit */
2240 : : false, /* conperiod */
2241 : : is_internal); /* internally constructed? */
2242 : :
6519 2243 : 1559 : pfree(ccbin);
2244 : :
4008 alvherre@alvh.no-ip. 2245 : 1559 : return constrOid;
2246 : : }
2247 : :
2248 : : /*
2249 : : * Store a not-null constraint for the given relation
2250 : : *
2251 : : * The OID of the new constraint is returned.
2252 : : */
2253 : : static Oid
492 2254 : 13134 : StoreRelNotNull(Relation rel, const char *nnname, AttrNumber attnum,
2255 : : bool is_validated, bool is_local, int inhcount,
2256 : : bool is_no_inherit)
2257 : : {
2258 : : Oid constrOid;
2259 : :
2260 [ - + ]: 13134 : Assert(attnum > InvalidAttrNumber);
2261 : :
2262 : : constrOid =
2263 : 13134 : CreateConstraintEntry(nnname,
2264 : 13134 : RelationGetNamespace(rel),
2265 : : CONSTRAINT_NOTNULL,
2266 : : false,
2267 : : false,
2268 : : true, /* Is Enforced */
2269 : : is_validated,
2270 : : InvalidOid,
2271 : : RelationGetRelid(rel),
2272 : : &attnum,
2273 : : 1,
2274 : : 1,
2275 : : InvalidOid, /* not a domain constraint */
2276 : : InvalidOid, /* no associated index */
2277 : : InvalidOid, /* Foreign key fields */
2278 : : NULL,
2279 : : NULL,
2280 : : NULL,
2281 : : NULL,
2282 : : 0,
2283 : : ' ',
2284 : : ' ',
2285 : : NULL,
2286 : : 0,
2287 : : ' ',
2288 : : NULL, /* not an exclusion constraint */
2289 : : NULL,
2290 : : NULL,
2291 : : is_local,
2292 : : inhcount,
2293 : : is_no_inherit,
2294 : : false,
2295 : : false);
2296 : 13134 : return constrOid;
2297 : : }
2298 : :
2299 : : /*
2300 : : * Store defaults and CHECK constraints (passed as a list of CookedConstraint).
2301 : : *
2302 : : * Each CookedConstraint struct is modified to store the new catalog tuple OID.
2303 : : *
2304 : : * NOTE: only pre-cooked expressions will be passed this way, which is to
2305 : : * say expressions inherited from an existing relation. Newly parsed
2306 : : * expressions can be added later, by direct calls to StoreAttrDefault
2307 : : * and StoreRelCheck (see AddRelationNewConstraints()).
2308 : : */
2309 : : static void
4746 rhaas@postgresql.org 2310 : 46353 : StoreConstraints(Relation rel, List *cooked_constraints, bool is_internal)
2311 : : {
6519 tgl@sss.pgh.pa.us 2312 : 46353 : int numchecks = 0;
2313 : : ListCell *lc;
2314 : :
4008 alvherre@alvh.no-ip. 2315 [ + + ]: 46353 : if (cooked_constraints == NIL)
8778 tgl@sss.pgh.pa.us 2316 : 46011 : return; /* nothing to do */
2317 : :
2318 : : /*
2319 : : * Deparsing of constraint expressions will fail unless the just-created
2320 : : * pg_attribute tuples for this relation are made visible. So, bump the
2321 : : * command counter. CAUTION: this will cause a relcache entry rebuild.
2322 : : */
9555 2323 : 342 : CommandCounterIncrement();
2324 : :
6519 2325 [ + - + + : 891 : foreach(lc, cooked_constraints)
+ + ]
2326 : : {
2327 : 549 : CookedConstraint *con = (CookedConstraint *) lfirst(lc);
2328 : :
2329 [ + + - ]: 549 : switch (con->contype)
2330 : : {
2331 : 271 : case CONSTR_DEFAULT:
4008 alvherre@alvh.no-ip. 2332 : 271 : con->conoid = StoreAttrDefault(rel, con->attnum, con->expr,
2333 : : is_internal);
6519 tgl@sss.pgh.pa.us 2334 : 271 : break;
2335 : 278 : case CONSTR_CHECK:
4008 alvherre@alvh.no-ip. 2336 : 278 : con->conoid =
2337 : 278 : StoreRelCheck(rel, con->name, con->expr,
428 peter@eisentraut.org 2338 : 278 : con->is_enforced, !con->skip_validation,
2339 : 278 : con->is_local, con->inhcount,
2340 : 278 : con->is_no_inherit, is_internal);
6519 tgl@sss.pgh.pa.us 2341 : 278 : numchecks++;
2342 : 278 : break;
2343 : :
6519 tgl@sss.pgh.pa.us 2344 :UBC 0 : default:
2345 [ # # ]: 0 : elog(ERROR, "unrecognized constraint type: %d",
2346 : : (int) con->contype);
2347 : : }
2348 : : }
2349 : :
6519 tgl@sss.pgh.pa.us 2350 [ + + ]:CBC 342 : if (numchecks > 0)
2351 : 131 : SetRelationNumChecks(rel, numchecks);
2352 : : }
2353 : :
2354 : : /*
2355 : : * AddRelationNewConstraints
2356 : : *
2357 : : * Add new column default expressions and/or constraint check expressions
2358 : : * to an existing relation. This is defined to do both for efficiency in
2359 : : * DefineRelation, but of course you can do just one or the other by passing
2360 : : * empty lists.
2361 : : *
2362 : : * rel: relation to be modified
2363 : : * newColDefaults: list of RawColumnDefault structures
2364 : : * newConstraints: list of Constraint nodes
2365 : : * allow_merge: true if check constraints may be merged with existing ones
2366 : : * is_local: true if definition is local, false if it's inherited
2367 : : * is_internal: true if result of some internal process, not a user request
2368 : : * queryString: used during expression transformation of default values and
2369 : : * cooked CHECK constraints
2370 : : *
2371 : : * All entries in newColDefaults will be processed. Entries in newConstraints
2372 : : * will be processed only if they are CONSTR_CHECK or CONSTR_NOTNULL types.
2373 : : *
2374 : : * Returns a list of CookedConstraint nodes that shows the cooked form of
2375 : : * the default and constraint expressions added to the relation.
2376 : : *
2377 : : * NB: caller should have opened rel with some self-conflicting lock mode,
2378 : : * and should hold that lock till end of transaction; for normal cases that'll
2379 : : * be AccessExclusiveLock, but if caller knows that the constraint is already
2380 : : * enforced by some other means, it can be ShareUpdateExclusiveLock. Also, we
2381 : : * assume the caller has done a CommandCounterIncrement if necessary to make
2382 : : * the relation's catalog tuples visible.
2383 : : */
2384 : : List *
2385 : 8800 : AddRelationNewConstraints(Relation rel,
2386 : : List *newColDefaults,
2387 : : List *newConstraints,
2388 : : bool allow_merge,
2389 : : bool is_local,
2390 : : bool is_internal,
2391 : : const char *queryString)
2392 : : {
7984 2393 : 8800 : List *cookedConstraints = NIL;
2394 : : TupleDesc tupleDesc;
2395 : : TupleConstr *oldconstr;
2396 : : int numoldchecks;
2397 : : ParseState *pstate;
2398 : : ParseNamespaceItem *nsitem;
2399 : : int numchecks;
2400 : : List *checknames;
2401 : : List *nnnames;
2402 : : Node *expr;
2403 : : CookedConstraint *cooked;
2404 : :
2405 : : /*
2406 : : * Get info about existing constraints.
2407 : : */
9660 2408 : 8800 : tupleDesc = RelationGetDescr(rel);
2409 : 8800 : oldconstr = tupleDesc->constr;
2410 [ + + ]: 8800 : if (oldconstr)
2411 : 7627 : numoldchecks = oldconstr->num_check;
2412 : : else
2413 : 1173 : numoldchecks = 0;
2414 : :
2415 : : /*
2416 : : * Create a dummy ParseState and insert the target relation as its sole
2417 : : * rangetable entry. We need a ParseState for transformExpr.
2418 : : */
2419 : 8800 : pstate = make_parsestate(NULL);
2762 peter_e@gmx.net 2420 : 8800 : pstate->p_sourcetext = queryString;
2264 tgl@sss.pgh.pa.us 2421 : 8800 : nsitem = addRangeTableEntryForRelation(pstate,
2422 : : rel,
2423 : : AccessShareLock,
2424 : : NULL,
2425 : : false,
2426 : : true);
2427 : 8800 : addNSItemToQuery(pstate, nsitem, true, true, true);
2428 : :
2429 : : /*
2430 : : * Process column default expressions.
2431 : : */
671 alvherre@alvh.no-ip. 2432 [ + + + + : 19835 : foreach_ptr(RawColumnDefault, colDef, newColDefaults)
+ + ]
2433 : : {
3129 andres@anarazel.de 2434 : 2457 : Form_pg_attribute atp = TupleDescAttr(rel->rd_att, colDef->attnum - 1);
2435 : : Oid defOid;
2436 : :
8761 tgl@sss.pgh.pa.us 2437 : 2457 : expr = cookDefault(pstate, colDef->raw_default,
2438 : : atp->atttypid, atp->atttypmod,
2542 peter@eisentraut.org 2439 : 2457 : NameStr(atp->attname),
2440 : 2457 : atp->attgenerated);
2441 : :
2442 : : /*
2443 : : * If the expression is just a NULL constant, we do not bother to make
2444 : : * an explicit pg_attrdef entry, since the default behavior is
2445 : : * equivalent. This applies to column defaults, but not for
2446 : : * generation expressions.
2447 : : *
2448 : : * Note a nonobvious property of this test: if the column is of a
2449 : : * domain type, what we'll get is not a bare null Const but a
2450 : : * CoerceToDomain expr, so we will not discard the default. This is
2451 : : * critical because the column default needs to be retained to
2452 : : * override any default that the domain might have.
2453 : : */
6712 tgl@sss.pgh.pa.us 2454 [ + - ]: 2352 : if (expr == NULL ||
2542 peter@eisentraut.org 2455 [ + + ]: 2352 : (!colDef->generated &&
2456 [ + + ]: 1445 : IsA(expr, Const) &&
2457 [ + + ]: 686 : castNode(Const, expr)->constisnull))
6712 tgl@sss.pgh.pa.us 2458 : 68 : continue;
2459 : :
377 2460 : 2284 : defOid = StoreAttrDefault(rel, colDef->attnum, expr, is_internal);
2461 : :
95 michael@paquier.xyz 2462 :GNC 2278 : cooked = palloc_object(CookedConstraint);
7984 tgl@sss.pgh.pa.us 2463 :CBC 2278 : cooked->contype = CONSTR_DEFAULT;
4008 alvherre@alvh.no-ip. 2464 : 2278 : cooked->conoid = defOid;
7984 tgl@sss.pgh.pa.us 2465 : 2278 : cooked->name = NULL;
2466 : 2278 : cooked->attnum = colDef->attnum;
2467 : 2278 : cooked->expr = expr;
428 peter@eisentraut.org 2468 : 2278 : cooked->is_enforced = true;
5401 alvherre@alvh.no-ip. 2469 : 2278 : cooked->skip_validation = false;
6519 tgl@sss.pgh.pa.us 2470 : 2278 : cooked->is_local = is_local;
2471 : 2278 : cooked->inhcount = is_local ? 0 : 1;
5077 alvherre@alvh.no-ip. 2472 : 2278 : cooked->is_no_inherit = false;
7984 tgl@sss.pgh.pa.us 2473 : 2278 : cookedConstraints = lappend(cookedConstraints, cooked);
2474 : : }
2475 : :
2476 : : /*
2477 : : * Process constraint expressions.
2478 : : */
9660 2479 : 8689 : numchecks = numoldchecks;
7948 2480 : 8689 : checknames = NIL;
492 alvherre@alvh.no-ip. 2481 : 8689 : nnnames = NIL;
671 2482 [ + + + + : 23815 : foreach_node(Constraint, cdef, newConstraints)
+ + ]
2483 : : {
2484 : : Oid constrOid;
2485 : :
933 2486 [ + + ]: 6611 : if (cdef->contype == CONSTR_CHECK)
2487 : : {
2488 : : char *ccname;
2489 : :
2490 [ + + ]: 1406 : if (cdef->raw_expr != NULL)
2491 : : {
2492 [ - + ]: 1280 : Assert(cdef->cooked_expr == NULL);
2493 : :
2494 : : /*
2495 : : * Transform raw parsetree to executable expression, and
2496 : : * verify it's valid as a CHECK constraint.
2497 : : */
2498 : 1280 : expr = cookConstraint(pstate, cdef->raw_expr,
2499 : 1280 : RelationGetRelationName(rel));
2500 : : }
2501 : : else
2502 : : {
2503 [ - + ]: 126 : Assert(cdef->cooked_expr != NULL);
2504 : :
2505 : : /*
2506 : : * Here, we assume the parser will only pass us valid CHECK
2507 : : * expressions, so we do no particular checking.
2508 : : */
2509 : 126 : expr = stringToNode(cdef->cooked_expr);
2510 : : }
2511 : :
2512 : : /*
2513 : : * Check name uniqueness, or generate a name if none was given.
2514 : : */
2515 [ + + ]: 1391 : if (cdef->conname != NULL)
2516 : : {
2517 : 1083 : ccname = cdef->conname;
2518 : : /* Check against other new constraints */
2519 : : /* Needed because we don't do CommandCounterIncrement in loop */
671 2520 [ + + + + : 2263 : foreach_ptr(char, chkname, checknames)
+ + ]
2521 : : {
2522 [ - + ]: 97 : if (strcmp(chkname, ccname) == 0)
933 alvherre@alvh.no-ip. 2523 [ # # ]:UBC 0 : ereport(ERROR,
2524 : : (errcode(ERRCODE_DUPLICATE_OBJECT),
2525 : : errmsg("check constraint \"%s\" already exists",
2526 : : ccname)));
2527 : : }
2528 : :
2529 : : /* save name for future checks */
933 alvherre@alvh.no-ip. 2530 :CBC 1083 : checknames = lappend(checknames, ccname);
2531 : :
2532 : : /*
2533 : : * Check against pre-existing constraints. If we are allowed
2534 : : * to merge with an existing constraint, there's no more to do
2535 : : * here. (We omit the duplicate constraint from the result,
2536 : : * which is what ATAddCheckNNConstraint wants.)
2537 : : */
2538 [ + + ]: 1059 : if (MergeWithExistingConstraint(rel, ccname, expr,
2539 : : allow_merge, is_local,
428 peter@eisentraut.org 2540 : 1083 : cdef->is_enforced,
933 alvherre@alvh.no-ip. 2541 : 1083 : cdef->initially_valid,
2542 : 1083 : cdef->is_no_inherit))
2543 : 74 : continue;
2544 : : }
2545 : : else
2546 : : {
2547 : : /*
2548 : : * When generating a name, we want to create "tab_col_check"
2549 : : * for a column constraint and "tab_check" for a table
2550 : : * constraint. We no longer have any info about the syntactic
2551 : : * positioning of the constraint phrase, so we approximate
2552 : : * this by seeing whether the expression references more than
2553 : : * one column. (If the user played by the rules, the result
2554 : : * is the same...)
2555 : : *
2556 : : * Note: pull_var_clause() doesn't descend into sublinks, but
2557 : : * we eliminated those above; and anyway this only needs to be
2558 : : * an approximate answer.
2559 : : */
2560 : : List *vars;
2561 : : char *colname;
2562 : :
2563 : 308 : vars = pull_var_clause(expr, 0);
2564 : :
2565 : : /* eliminate duplicates */
2566 : 308 : vars = list_union(NIL, vars);
2567 : :
2568 [ + + ]: 308 : if (list_length(vars) == 1)
2569 : 272 : colname = get_attname(RelationGetRelid(rel),
2570 : 272 : ((Var *) linitial(vars))->varattno,
2571 : : true);
2572 : : else
2573 : 36 : colname = NULL;
2574 : :
2575 : 308 : ccname = ChooseConstraintName(RelationGetRelationName(rel),
2576 : : colname,
2577 : : "check",
2578 : 308 : RelationGetNamespace(rel),
2579 : : checknames);
2580 : :
2581 : : /* save name for future checks */
2582 : 308 : checknames = lappend(checknames, ccname);
2583 : : }
2584 : :
2585 : : /*
2586 : : * OK, store it.
2587 : : */
2588 : : constrOid =
428 peter@eisentraut.org 2589 : 1293 : StoreRelCheck(rel, ccname, expr, cdef->is_enforced,
2590 : 1293 : cdef->initially_valid, is_local,
2591 : 1293 : is_local ? 0 : 1, cdef->is_no_inherit,
2592 : : is_internal);
2593 : :
933 alvherre@alvh.no-ip. 2594 : 1281 : numchecks++;
2595 : :
95 michael@paquier.xyz 2596 :GNC 1281 : cooked = palloc_object(CookedConstraint);
933 alvherre@alvh.no-ip. 2597 :CBC 1281 : cooked->contype = CONSTR_CHECK;
2598 : 1281 : cooked->conoid = constrOid;
2599 : 1281 : cooked->name = ccname;
2600 : 1281 : cooked->attnum = 0;
2601 : 1281 : cooked->expr = expr;
428 peter@eisentraut.org 2602 : 1281 : cooked->is_enforced = cdef->is_enforced;
933 alvherre@alvh.no-ip. 2603 : 1281 : cooked->skip_validation = cdef->skip_validation;
2604 : 1281 : cooked->is_local = is_local;
2605 : 1281 : cooked->inhcount = is_local ? 0 : 1;
2606 : 1281 : cooked->is_no_inherit = cdef->is_no_inherit;
2607 : 1281 : cookedConstraints = lappend(cookedConstraints, cooked);
2608 : : }
492 2609 [ + - ]: 5205 : else if (cdef->contype == CONSTR_NOTNULL)
2610 : : {
2611 : : CookedConstraint *nncooked;
2612 : : AttrNumber colnum;
2613 : 5205 : int16 inhcount = is_local ? 0 : 1;
2614 : : char *nnname;
2615 : :
2616 : : /* Determine which column to modify */
2617 : 5205 : colnum = get_attnum(RelationGetRelid(rel), strVal(linitial(cdef->keys)));
2618 [ + + ]: 5205 : if (colnum == InvalidAttrNumber)
2619 [ + - ]: 9 : ereport(ERROR,
2620 : : errcode(ERRCODE_UNDEFINED_COLUMN),
2621 : : errmsg("column \"%s\" of relation \"%s\" does not exist",
2622 : : strVal(linitial(cdef->keys)), RelationGetRelationName(rel)));
2623 [ - + ]: 5196 : if (colnum < InvalidAttrNumber)
492 alvherre@alvh.no-ip. 2624 [ # # ]:UBC 0 : ereport(ERROR,
2625 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
2626 : : errmsg("cannot add not-null constraint on system column \"%s\"",
2627 : : strVal(linitial(cdef->keys))));
2628 : :
342 alvherre@alvh.no-ip. 2629 [ - + ]:CBC 5196 : Assert(cdef->initially_valid != cdef->skip_validation);
2630 : :
2631 : : /*
2632 : : * If the column already has a not-null constraint, we don't want
2633 : : * to add another one; adjust inheritance status as needed. This
2634 : : * also checks whether the existing constraint matches the
2635 : : * requested validity.
2636 : : */
492 2637 [ + + ]: 5172 : if (AdjustNotNullInheritance(RelationGetRelid(rel), colnum,
40 alvherre@kurilemu.de 2638 : 5196 : cdef->conname,
342 alvherre@alvh.no-ip. 2639 : 5196 : is_local, cdef->is_no_inherit,
2640 : 5196 : cdef->skip_validation))
492 2641 : 47 : continue;
2642 : :
2643 : : /*
2644 : : * If a constraint name is specified, check that it isn't already
2645 : : * used. Otherwise, choose a non-conflicting one ourselves.
2646 : : */
2647 [ + + ]: 5125 : if (cdef->conname)
2648 : : {
2649 [ + + ]: 672 : if (ConstraintNameIsUsed(CONSTRAINT_RELATION,
2650 : : RelationGetRelid(rel),
2651 : 672 : cdef->conname))
2652 [ + - ]: 3 : ereport(ERROR,
2653 : : errcode(ERRCODE_DUPLICATE_OBJECT),
2654 : : errmsg("constraint \"%s\" for relation \"%s\" already exists",
2655 : : cdef->conname, RelationGetRelationName(rel)));
2656 : 669 : nnname = cdef->conname;
2657 : : }
2658 : : else
2659 : 8906 : nnname = ChooseConstraintName(RelationGetRelationName(rel),
2660 : 4453 : strVal(linitial(cdef->keys)),
2661 : : "not_null",
2662 : 4453 : RelationGetNamespace(rel),
2663 : : nnnames);
2664 : 5122 : nnnames = lappend(nnnames, nnname);
2665 : :
2666 : : constrOid =
2667 : 5122 : StoreRelNotNull(rel, nnname, colnum,
2668 : 5122 : cdef->initially_valid,
2669 : : is_local,
2670 : : inhcount,
2671 : 5122 : cdef->is_no_inherit);
2672 : :
95 michael@paquier.xyz 2673 :GNC 5122 : nncooked = palloc_object(CookedConstraint);
492 alvherre@alvh.no-ip. 2674 :CBC 5122 : nncooked->contype = CONSTR_NOTNULL;
2675 : 5122 : nncooked->conoid = constrOid;
2676 : 5122 : nncooked->name = nnname;
2677 : 5122 : nncooked->attnum = colnum;
2678 : 5122 : nncooked->expr = NULL;
428 peter@eisentraut.org 2679 : 5122 : nncooked->is_enforced = true;
492 alvherre@alvh.no-ip. 2680 : 5122 : nncooked->skip_validation = cdef->skip_validation;
2681 : 5122 : nncooked->is_local = is_local;
2682 : 5122 : nncooked->inhcount = inhcount;
2683 : 5122 : nncooked->is_no_inherit = cdef->is_no_inherit;
2684 : :
2685 : 5122 : cookedConstraints = lappend(cookedConstraints, nncooked);
2686 : : }
2687 : : }
2688 : :
2689 : : /*
2690 : : * Update the count of constraints in the relation's pg_class tuple. We do
2691 : : * this even if there was no change, in order to ensure that an SI update
2692 : : * message is sent out for the pg_class tuple, which will force other
2693 : : * backends to rebuild their relcache entries for the rel. (This is
2694 : : * critical if we added defaults but not constraints.)
2695 : : */
8778 tgl@sss.pgh.pa.us 2696 : 8602 : SetRelationNumChecks(rel, numchecks);
2697 : :
7984 2698 : 8602 : return cookedConstraints;
2699 : : }
2700 : :
2701 : : /*
2702 : : * Check for a pre-existing check constraint that conflicts with a proposed
2703 : : * new one, and either adjust its conislocal/coninhcount settings or throw
2704 : : * error as needed.
2705 : : *
2706 : : * Returns true if merged (constraint is a duplicate), or false if it's
2707 : : * got a so-far-unique name, or throws error if conflict.
2708 : : *
2709 : : * XXX See MergeConstraintsIntoExisting too if you change this code.
2710 : : */
2711 : : static bool
3057 peter_e@gmx.net 2712 : 1083 : MergeWithExistingConstraint(Relation rel, const char *ccname, Node *expr,
2713 : : bool allow_merge, bool is_local,
2714 : : bool is_enforced,
2715 : : bool is_initially_valid,
2716 : : bool is_no_inherit)
2717 : : {
2718 : : bool found;
2719 : : Relation conDesc;
2720 : : SysScanDesc conscan;
2721 : : ScanKeyData skey[3];
2722 : : HeapTuple tup;
2723 : :
2724 : : /* Search for a pg_constraint entry with same name and relation */
2610 andres@anarazel.de 2725 : 1083 : conDesc = table_open(ConstraintRelationId, RowExclusiveLock);
2726 : :
6519 tgl@sss.pgh.pa.us 2727 : 1083 : found = false;
2728 : :
2729 : 1083 : ScanKeyInit(&skey[0],
2730 : : Anum_pg_constraint_conrelid,
2731 : : BTEqualStrategyNumber, F_OIDEQ,
2732 : : ObjectIdGetDatum(RelationGetRelid(rel)));
2749 2733 : 1083 : ScanKeyInit(&skey[1],
2734 : : Anum_pg_constraint_contypid,
2735 : : BTEqualStrategyNumber, F_OIDEQ,
2736 : : ObjectIdGetDatum(InvalidOid));
2737 : 1083 : ScanKeyInit(&skey[2],
2738 : : Anum_pg_constraint_conname,
2739 : : BTEqualStrategyNumber, F_NAMEEQ,
2740 : : CStringGetDatum(ccname));
2741 : :
2742 : 1083 : conscan = systable_beginscan(conDesc, ConstraintRelidTypidNameIndexId, true,
2743 : : NULL, 3, skey);
2744 : :
2745 : : /* There can be at most one matching row */
2746 [ + + ]: 1083 : if (HeapTupleIsValid(tup = systable_getnext(conscan)))
2747 : : {
6519 2748 : 98 : Form_pg_constraint con = (Form_pg_constraint) GETSTRUCT(tup);
2749 : :
2750 : : /* Found it. Conflicts if not identical check constraint */
2749 2751 [ + + ]: 98 : if (con->contype == CONSTRAINT_CHECK)
2752 : : {
2753 : : Datum val;
2754 : : bool isnull;
2755 : :
2756 : 95 : val = fastgetattr(tup,
2757 : : Anum_pg_constraint_conbin,
2758 : : conDesc->rd_att, &isnull);
2759 [ - + ]: 95 : if (isnull)
2749 tgl@sss.pgh.pa.us 2760 [ # # ]:UBC 0 : elog(ERROR, "null conbin for rel %s",
2761 : : RelationGetRelationName(rel));
2749 tgl@sss.pgh.pa.us 2762 [ + + ]:CBC 95 : if (equal(expr, stringToNode(TextDatumGetCString(val))))
2763 : 92 : found = true;
2764 : : }
2765 : :
2766 : : /*
2767 : : * If the existing constraint is purely inherited (no local
2768 : : * definition) then interpret addition of a local constraint as a
2769 : : * legal merge. This allows ALTER ADD CONSTRAINT on parent and child
2770 : : * tables to be given in either order with same end state. However if
2771 : : * the relation is a partition, all inherited constraints are always
2772 : : * non-local, including those that were merged.
2773 : : */
2774 [ + + + + : 98 : if (is_local && !con->conislocal && !rel->rd_rel->relispartition)
+ + ]
2775 : 54 : allow_merge = true;
2776 : :
2777 [ + + - + ]: 98 : if (!found || !allow_merge)
2778 [ + - ]: 6 : ereport(ERROR,
2779 : : (errcode(ERRCODE_DUPLICATE_OBJECT),
2780 : : errmsg("constraint \"%s\" for relation \"%s\" already exists",
2781 : : ccname, RelationGetRelationName(rel))));
2782 : :
2783 : : /* If the child constraint is "no inherit" then cannot merge */
2784 [ - + ]: 92 : if (con->connoinherit)
2749 tgl@sss.pgh.pa.us 2785 [ # # ]:UBC 0 : ereport(ERROR,
2786 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2787 : : errmsg("constraint \"%s\" conflicts with non-inherited constraint on relation \"%s\"",
2788 : : ccname, RelationGetRelationName(rel))));
2789 : :
2790 : : /*
2791 : : * Must not change an existing inherited constraint to "no inherit"
2792 : : * status. That's because inherited constraints should be able to
2793 : : * propagate to lower-level children.
2794 : : */
2749 tgl@sss.pgh.pa.us 2795 [ + + + + ]:CBC 92 : if (con->coninhcount > 0 && is_no_inherit)
2796 [ + - ]: 3 : ereport(ERROR,
2797 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2798 : : errmsg("constraint \"%s\" conflicts with inherited constraint on relation \"%s\"",
2799 : : ccname, RelationGetRelationName(rel))));
2800 : :
2801 : : /*
2802 : : * If the child constraint is "not valid" then cannot merge with a
2803 : : * valid parent constraint.
2804 : : */
428 peter@eisentraut.org 2805 [ + + + + : 89 : if (is_initially_valid && con->conenforced && !con->convalidated)
+ + ]
2749 tgl@sss.pgh.pa.us 2806 [ + - ]: 3 : ereport(ERROR,
2807 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2808 : : errmsg("constraint \"%s\" conflicts with NOT VALID constraint on relation \"%s\"",
2809 : : ccname, RelationGetRelationName(rel))));
2810 : :
2811 : : /*
2812 : : * A non-enforced child constraint cannot be merged with an enforced
2813 : : * parent constraint. However, the reverse is allowed, where the child
2814 : : * constraint is enforced.
2815 : : */
428 peter@eisentraut.org 2816 [ + + + + : 86 : if ((!is_local && is_enforced && !con->conenforced) ||
+ + + + ]
2817 [ + + + + ]: 54 : (is_local && !is_enforced && con->conenforced))
2818 [ + - ]: 12 : ereport(ERROR,
2819 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2820 : : errmsg("constraint \"%s\" conflicts with NOT ENFORCED constraint on relation \"%s\"",
2821 : : ccname, RelationGetRelationName(rel))));
2822 : :
2823 : : /* OK to update the tuple */
2749 tgl@sss.pgh.pa.us 2824 [ + + ]: 74 : ereport(NOTICE,
2825 : : (errmsg("merging constraint \"%s\" with inherited definition",
2826 : : ccname)));
2827 : :
2828 : 74 : tup = heap_copytuple(tup);
2829 : 74 : con = (Form_pg_constraint) GETSTRUCT(tup);
2830 : :
2831 : : /*
2832 : : * In case of partitions, an inherited constraint must be inherited
2833 : : * only once since it cannot have multiple parents and it is never
2834 : : * considered local.
2835 : : */
2836 [ + + ]: 74 : if (rel->rd_rel->relispartition)
2837 : : {
2838 : 6 : con->coninhcount = 1;
2839 : 6 : con->conislocal = false;
2840 : : }
2841 : : else
2842 : : {
2843 [ + + ]: 68 : if (is_local)
2844 : 42 : con->conislocal = true;
521 alvherre@alvh.no-ip. 2845 [ - + ]: 26 : else if (pg_add_s16_overflow(con->coninhcount, 1,
2846 : : &con->coninhcount))
1083 peter@eisentraut.org 2847 [ # # ]:UBC 0 : ereport(ERROR,
2848 : : errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
2849 : : errmsg("too many inheritance parents"));
2850 : : }
2851 : :
2749 tgl@sss.pgh.pa.us 2852 [ - + ]:CBC 74 : if (is_no_inherit)
2853 : : {
2749 tgl@sss.pgh.pa.us 2854 [ # # ]:UBC 0 : Assert(is_local);
2855 : 0 : con->connoinherit = true;
2856 : : }
2857 : :
2858 : : /*
2859 : : * If the child constraint is required to be enforced while the parent
2860 : : * constraint is not, this should be allowed by marking the child
2861 : : * constraint as enforced. In the reverse case, an error would have
2862 : : * already been thrown before reaching this point.
2863 : : */
428 peter@eisentraut.org 2864 [ + + + + ]:CBC 74 : if (is_enforced && !con->conenforced)
2865 : : {
2866 [ - + ]: 9 : Assert(is_local);
2867 : 9 : con->conenforced = true;
2868 : 9 : con->convalidated = true;
2869 : : }
2870 : :
2749 tgl@sss.pgh.pa.us 2871 : 74 : CatalogTupleUpdate(conDesc, &tup->t_self, tup);
2872 : : }
2873 : :
6519 2874 : 1059 : systable_endscan(conscan);
2610 andres@anarazel.de 2875 : 1059 : table_close(conDesc, RowExclusiveLock);
2876 : :
6519 tgl@sss.pgh.pa.us 2877 : 1059 : return found;
2878 : : }
2879 : :
2880 : : /*
2881 : : * Create the not-null constraints when creating a new relation
2882 : : *
2883 : : * These come from two sources: the 'constraints' list (of Constraint) is
2884 : : * specified directly by the user; the 'old_notnulls' list (of
2885 : : * CookedConstraint) comes from inheritance. We create one constraint
2886 : : * for each column, giving priority to user-specified ones, and setting
2887 : : * inhcount according to how many parents cause each column to get a
2888 : : * not-null constraint. If a user-specified name clashes with another
2889 : : * user-specified name, an error is raised. 'existing_constraints'
2890 : : * is a list of already defined constraint names, which should be avoided
2891 : : * when generating further ones.
2892 : : *
2893 : : * Returns a list of AttrNumber for columns that need to have the attnotnull
2894 : : * flag set.
2895 : : */
2896 : : List *
492 alvherre@alvh.no-ip. 2897 : 32360 : AddRelationNotNullConstraints(Relation rel, List *constraints,
2898 : : List *old_notnulls, List *existing_constraints)
2899 : : {
2900 : : List *givennames;
2901 : : List *nnnames;
2902 : 32360 : List *nncols = NIL;
2903 : :
2904 : : /*
2905 : : * We track two lists of names: nnnames keeps all the constraint names,
2906 : : * givennames tracks user-generated names. The distinction is important,
2907 : : * because we must raise error for user-generated name conflicts, but for
2908 : : * system-generated name conflicts we just generate another.
2909 : : */
22 alvherre@kurilemu.de 2910 : 32360 : nnnames = list_copy(existing_constraints); /* don't scribble on input */
492 alvherre@alvh.no-ip. 2911 : 32360 : givennames = NIL;
2912 : :
2913 : : /*
2914 : : * First, create all not-null constraints that are directly specified by
2915 : : * the user. Note that inheritance might have given us another source for
2916 : : * each, so we must scan the old_notnulls list and increment inhcount for
2917 : : * each element with identical attnum. We delete from there any element
2918 : : * that we process.
2919 : : *
2920 : : * We don't use foreach() here because we have two nested loops over the
2921 : : * constraint list, with possible element deletions in the inner one. If
2922 : : * we used foreach_delete_current() it could only fix up the state of one
2923 : : * of the loops, so it seems cleaner to use looping over list indexes for
2924 : : * both loops. Note that any deletion will happen beyond where the outer
2925 : : * loop is, so its index never needs adjustment.
2926 : : */
2927 [ + + ]: 39213 : for (int outerpos = 0; outerpos < list_length(constraints); outerpos++)
2928 : : {
2929 : : Constraint *constr;
2930 : : AttrNumber attnum;
2931 : : char *conname;
2932 : 6892 : int inhcount = 0;
2933 : :
2934 : 6892 : constr = list_nth_node(Constraint, constraints, outerpos);
2935 : :
2936 [ - + ]: 6892 : Assert(constr->contype == CONSTR_NOTNULL);
2937 : :
2938 : 6892 : attnum = get_attnum(RelationGetRelid(rel),
2939 : 6892 : strVal(linitial(constr->keys)));
2940 [ - + ]: 6892 : if (attnum == InvalidAttrNumber)
492 alvherre@alvh.no-ip. 2941 [ # # ]:UBC 0 : ereport(ERROR,
2942 : : errcode(ERRCODE_UNDEFINED_COLUMN),
2943 : : errmsg("column \"%s\" of relation \"%s\" does not exist",
2944 : : strVal(linitial(constr->keys)),
2945 : : RelationGetRelationName(rel)));
492 alvherre@alvh.no-ip. 2946 [ - + ]:CBC 6892 : if (attnum < InvalidAttrNumber)
492 alvherre@alvh.no-ip. 2947 [ # # ]:UBC 0 : ereport(ERROR,
2948 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
2949 : : errmsg("cannot add not-null constraint on system column \"%s\"",
2950 : : strVal(linitial(constr->keys))));
2951 : :
2952 : : /*
2953 : : * A column can only have one not-null constraint, so discard any
2954 : : * additional ones that appear for columns we already saw; but check
2955 : : * that the NO INHERIT flags match.
2956 : : */
492 alvherre@alvh.no-ip. 2957 [ + + ]:CBC 9044 : for (int restpos = outerpos + 1; restpos < list_length(constraints);)
2958 : : {
2959 : : Constraint *other;
2960 : :
2961 : 2182 : other = list_nth_node(Constraint, constraints, restpos);
2962 [ + + ]: 2182 : if (strcmp(strVal(linitial(constr->keys)),
2963 : 2182 : strVal(linitial(other->keys))) == 0)
2964 : : {
2965 [ + + ]: 48 : if (other->is_no_inherit != constr->is_no_inherit)
2966 [ + - ]: 21 : ereport(ERROR,
2967 : : errcode(ERRCODE_SYNTAX_ERROR),
2968 : : errmsg("conflicting NO INHERIT declaration for not-null constraint on column \"%s\"",
2969 : : strVal(linitial(constr->keys))));
2970 : :
2971 : : /*
2972 : : * Preserve constraint name if one is specified, but raise an
2973 : : * error if conflicting ones are specified.
2974 : : */
2975 [ + + ]: 27 : if (other->conname)
2976 : : {
2977 [ + + ]: 18 : if (!constr->conname)
2978 : 6 : constr->conname = pstrdup(other->conname);
2979 [ + + ]: 12 : else if (strcmp(constr->conname, other->conname) != 0)
2980 [ + - ]: 9 : ereport(ERROR,
2981 : : errcode(ERRCODE_SYNTAX_ERROR),
2982 : : errmsg("conflicting not-null constraint names \"%s\" and \"%s\"",
2983 : : constr->conname, other->conname));
2984 : : }
2985 : :
2986 : : /* XXX do we need to verify any other fields? */
2987 : 18 : constraints = list_delete_nth_cell(constraints, restpos);
2988 : : }
2989 : : else
2990 : 2134 : restpos++;
2991 : : }
2992 : :
2993 : : /*
2994 : : * Search in the list of inherited constraints for any entries on the
2995 : : * same column; determine an inheritance count from that. Also, if at
2996 : : * least one parent has a constraint for this column, then we must not
2997 : : * accept a user specification for a NO INHERIT one. Any constraint
2998 : : * from parents that we process here is deleted from the list: we no
2999 : : * longer need to process it in the loop below.
3000 : : */
3001 [ + + + + : 13800 : foreach_ptr(CookedConstraint, old, old_notnulls)
+ + ]
3002 : : {
3003 [ + + ]: 88 : if (old->attnum == attnum)
3004 : : {
3005 : : /*
3006 : : * If we get a constraint from the parent, having a local NO
3007 : : * INHERIT one doesn't work.
3008 : : */
3009 [ + + ]: 73 : if (constr->is_no_inherit)
3010 [ + - ]: 6 : ereport(ERROR,
3011 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
3012 : : errmsg("cannot define not-null constraint with NO INHERIT on column \"%s\"",
3013 : : strVal(linitial(constr->keys))),
3014 : : errdetail("The column has an inherited not-null constraint.")));
3015 : :
3016 : 67 : inhcount++;
3017 : 67 : old_notnulls = foreach_delete_current(old_notnulls, old);
3018 : : }
3019 : : }
3020 : :
3021 : : /*
3022 : : * Determine a constraint name, which may have been specified by the
3023 : : * user, or raise an error if a conflict exists with another
3024 : : * user-specified name.
3025 : : */
3026 [ + + ]: 6856 : if (constr->conname)
3027 : : {
3028 [ + + + + : 791 : foreach_ptr(char, thisname, givennames)
+ + ]
3029 : : {
3030 [ + + ]: 71 : if (strcmp(thisname, constr->conname) == 0)
3031 [ + - ]: 3 : ereport(ERROR,
3032 : : errcode(ERRCODE_DUPLICATE_OBJECT),
3033 : : errmsg("constraint \"%s\" for relation \"%s\" already exists",
3034 : : constr->conname,
3035 : : RelationGetRelationName(rel)));
3036 : : }
3037 : :
3038 : 360 : conname = constr->conname;
3039 : 360 : givennames = lappend(givennames, conname);
3040 : : }
3041 : : else
3042 : 6493 : conname = ChooseConstraintName(RelationGetRelationName(rel),
3043 : 6493 : get_attname(RelationGetRelid(rel),
3044 : : attnum, false),
3045 : : "not_null",
3046 : 6493 : RelationGetNamespace(rel),
3047 : : nnnames);
3048 : 6853 : nnnames = lappend(nnnames, conname);
3049 : :
3050 : 6853 : StoreRelNotNull(rel, conname,
3051 : : attnum, true, true,
3052 : 6853 : inhcount, constr->is_no_inherit);
3053 : :
3054 : 6853 : nncols = lappend_int(nncols, attnum);
3055 : : }
3056 : :
3057 : : /*
3058 : : * If any column remains in the old_notnulls list, we must create a not-
3059 : : * null constraint marked not-local for that column. Because multiple
3060 : : * parents could specify a not-null constraint for the same column, we
3061 : : * must count how many there are and set an appropriate inhcount
3062 : : * accordingly, deleting elements we've already processed.
3063 : : *
3064 : : * We don't use foreach() here because we have two nested loops over the
3065 : : * constraint list, with possible element deletions in the inner one. If
3066 : : * we used foreach_delete_current() it could only fix up the state of one
3067 : : * of the loops, so it seems cleaner to use looping over list indexes for
3068 : : * both loops. Note that any deletion will happen beyond where the outer
3069 : : * loop is, so its index never needs adjustment.
3070 : : */
3071 [ + + ]: 33480 : for (int outerpos = 0; outerpos < list_length(old_notnulls); outerpos++)
3072 : : {
3073 : : CookedConstraint *cooked;
3074 : 1159 : char *conname = NULL;
3075 : 1159 : int inhcount = 1;
3076 : :
3077 : 1159 : cooked = (CookedConstraint *) list_nth(old_notnulls, outerpos);
3078 [ - + ]: 1159 : Assert(cooked->contype == CONSTR_NOTNULL);
3079 [ - + ]: 1159 : Assert(cooked->name);
3080 : :
3081 : : /*
3082 : : * Preserve the first non-conflicting constraint name we come across.
3083 : : */
3084 [ + - ]: 1159 : if (conname == NULL)
3085 : 1159 : conname = cooked->name;
3086 : :
3087 [ + + ]: 1410 : for (int restpos = outerpos + 1; restpos < list_length(old_notnulls);)
3088 : : {
3089 : : CookedConstraint *other;
3090 : :
3091 : 251 : other = (CookedConstraint *) list_nth(old_notnulls, restpos);
3092 [ - + ]: 251 : Assert(other->name);
3093 [ + + ]: 251 : if (other->attnum == cooked->attnum)
3094 : : {
3095 [ - + ]: 19 : if (conname == NULL)
492 alvherre@alvh.no-ip. 3096 :UBC 0 : conname = other->name;
3097 : :
492 alvherre@alvh.no-ip. 3098 :CBC 19 : inhcount++;
3099 : 19 : old_notnulls = list_delete_nth_cell(old_notnulls, restpos);
3100 : : }
3101 : : else
3102 : 232 : restpos++;
3103 : : }
3104 : :
3105 : : /* If we got a name, make sure it isn't one we've already used */
3106 [ + - ]: 1159 : if (conname != NULL)
3107 : : {
3108 [ + + + + : 2579 : foreach_ptr(char, thisname, nnnames)
+ + ]
3109 : : {
3110 [ + + ]: 264 : if (strcmp(thisname, conname) == 0)
3111 : : {
3112 : 3 : conname = NULL;
3113 : 3 : break;
3114 : : }
3115 : : }
3116 : : }
3117 : :
3118 : : /* and choose a name, if needed */
3119 [ + + ]: 1159 : if (conname == NULL)
3120 : 3 : conname = ChooseConstraintName(RelationGetRelationName(rel),
3121 : 3 : get_attname(RelationGetRelid(rel),
3122 : 3 : cooked->attnum, false),
3123 : : "not_null",
3124 : 3 : RelationGetNamespace(rel),
3125 : : nnnames);
3126 : 1159 : nnnames = lappend(nnnames, conname);
3127 : :
3128 : : /* ignore the origin constraint's is_local and inhcount */
3129 : 1159 : StoreRelNotNull(rel, conname, cooked->attnum, true,
3130 : : false, inhcount, false);
3131 : :
3132 : 1159 : nncols = lappend_int(nncols, cooked->attnum);
3133 : : }
3134 : :
3135 : 32321 : return nncols;
3136 : : }
3137 : :
3138 : : /*
3139 : : * Update the count of constraints in the relation's pg_class tuple.
3140 : : *
3141 : : * Caller had better hold exclusive lock on the relation.
3142 : : *
3143 : : * An important side effect is that a SI update message will be sent out for
3144 : : * the pg_class tuple, which will force other backends to rebuild their
3145 : : * relcache entries for the rel. Also, this backend will rebuild its
3146 : : * own relcache entry at the next CommandCounterIncrement.
3147 : : */
3148 : : static void
8778 tgl@sss.pgh.pa.us 3149 : 8733 : SetRelationNumChecks(Relation rel, int numchecks)
3150 : : {
3151 : : Relation relrel;
3152 : : HeapTuple reltup;
3153 : : Form_pg_class relStruct;
3154 : :
2610 andres@anarazel.de 3155 : 8733 : relrel = table_open(RelationRelationId, RowExclusiveLock);
5873 rhaas@postgresql.org 3156 : 8733 : reltup = SearchSysCacheCopy1(RELOID,
3157 : : ObjectIdGetDatum(RelationGetRelid(rel)));
9660 tgl@sss.pgh.pa.us 3158 [ - + ]: 8733 : if (!HeapTupleIsValid(reltup))
8274 tgl@sss.pgh.pa.us 3159 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for relation %u",
3160 : : RelationGetRelid(rel));
9660 tgl@sss.pgh.pa.us 3161 :CBC 8733 : relStruct = (Form_pg_class) GETSTRUCT(reltup);
3162 : :
8778 3163 [ + + ]: 8733 : if (relStruct->relchecks != numchecks)
3164 : : {
3165 : 1317 : relStruct->relchecks = numchecks;
3166 : :
3330 alvherre@alvh.no-ip. 3167 : 1317 : CatalogTupleUpdate(relrel, &reltup->t_self, reltup);
3168 : : }
3169 : : else
3170 : : {
3171 : : /* Skip the disk update, but force relcache inval anyway */
8069 tgl@sss.pgh.pa.us 3172 : 7416 : CacheInvalidateRelcache(rel);
3173 : : }
3174 : :
9586 JanWieck@Yahoo.com 3175 : 8733 : heap_freetuple(reltup);
2610 andres@anarazel.de 3176 : 8733 : table_close(relrel, RowExclusiveLock);
10432 vadim4o@yahoo.com 3177 : 8733 : }
3178 : :
3179 : : /*
3180 : : * Check for references to generated columns
3181 : : */
3182 : : static bool
2542 peter@eisentraut.org 3183 : 3205 : check_nested_generated_walker(Node *node, void *context)
3184 : : {
3185 : 3205 : ParseState *pstate = context;
3186 : :
3187 [ - + ]: 3205 : if (node == NULL)
2542 peter@eisentraut.org 3188 :UBC 0 : return false;
2542 peter@eisentraut.org 3189 [ + + ]:CBC 3205 : else if (IsA(node, Var))
3190 : : {
3191 : 1098 : Var *var = (Var *) node;
3192 : : Oid relid;
3193 : : AttrNumber attnum;
3194 : :
3195 : 1098 : relid = rt_fetch(var->varno, pstate->p_rtable)->relid;
1759 tgl@sss.pgh.pa.us 3196 [ - + ]: 1098 : if (!OidIsValid(relid))
1759 tgl@sss.pgh.pa.us 3197 :UBC 0 : return false; /* XXX shouldn't we raise an error? */
3198 : :
2542 peter@eisentraut.org 3199 :CBC 1098 : attnum = var->varattno;
3200 : :
1759 tgl@sss.pgh.pa.us 3201 [ + + + + ]: 1098 : if (attnum > 0 && get_attgenerated(relid, attnum))
2542 peter@eisentraut.org 3202 [ + - ]: 18 : ereport(ERROR,
3203 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3204 : : errmsg("cannot use generated column \"%s\" in column generation expression",
3205 : : get_attname(relid, attnum, false)),
3206 : : errdetail("A generated column cannot reference another generated column."),
3207 : : parser_errposition(pstate, var->location)));
3208 : : /* A whole-row Var is necessarily self-referential, so forbid it */
1759 tgl@sss.pgh.pa.us 3209 [ + + ]: 1080 : if (attnum == 0)
3210 [ + - ]: 6 : ereport(ERROR,
3211 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3212 : : errmsg("cannot use whole-row variable in column generation expression"),
3213 : : errdetail("This would cause the generated column to depend on its own value."),
3214 : : parser_errposition(pstate, var->location)));
3215 : : /* System columns were already checked in the parser */
3216 : :
2542 peter@eisentraut.org 3217 : 1074 : return false;
3218 : : }
3219 : : else
3220 : 2107 : return expression_tree_walker(node, check_nested_generated_walker,
3221 : : context);
3222 : : }
3223 : :
3224 : : static void
3225 : 946 : check_nested_generated(ParseState *pstate, Node *node)
3226 : : {
3227 : 946 : check_nested_generated_walker(node, pstate);
3228 : 922 : }
3229 : :
3230 : : /*
3231 : : * Check security of virtual generated column expression.
3232 : : *
3233 : : * Just like selecting from a view is exploitable (CVE-2024-7348), selecting
3234 : : * from a table with virtual generated columns is exploitable. Users who are
3235 : : * concerned about this can avoid selecting from views, but telling them to
3236 : : * avoid selecting from tables is less practical.
3237 : : *
3238 : : * To address this, this restricts generation expressions for virtual
3239 : : * generated columns are restricted to using built-in functions and types. We
3240 : : * assume that built-in functions and types cannot be exploited for this
3241 : : * purpose. Note the overall security also requires that all functions in use
3242 : : * a immutable. (For example, there are some built-in non-immutable functions
3243 : : * that can run arbitrary SQL.) The immutability is checked elsewhere, since
3244 : : * that is a property that needs to hold independent of security
3245 : : * considerations.
3246 : : *
3247 : : * In the future, this could be expanded by some new mechanism to declare
3248 : : * other functions and types as safe or trusted for this purpose, but that is
3249 : : * to be designed.
3250 : : */
3251 : :
3252 : : /*
3253 : : * Callback for check_functions_in_node() that determines whether a function
3254 : : * is user-defined.
3255 : : */
3256 : : static bool
263 3257 : 461 : contains_user_functions_checker(Oid func_id, void *context)
3258 : : {
3259 : 461 : return (func_id >= FirstUnpinnedObjectId);
3260 : : }
3261 : :
3262 : : /*
3263 : : * Checks for all the things we don't want in the generation expressions of
3264 : : * virtual generated columns for security reasons. Errors out if it finds
3265 : : * one.
3266 : : */
3267 : : static bool
3268 : 1311 : check_virtual_generated_security_walker(Node *node, void *context)
3269 : : {
3270 : 1311 : ParseState *pstate = context;
3271 : :
3272 [ - + ]: 1311 : if (node == NULL)
263 peter@eisentraut.org 3273 :UBC 0 : return false;
3274 : :
263 peter@eisentraut.org 3275 [ + + ]:CBC 1311 : if (!IsA(node, List))
3276 : : {
3277 [ + + ]: 1308 : if (check_functions_in_node(node, contains_user_functions_checker, NULL))
3278 [ + - ]: 6 : ereport(ERROR,
3279 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3280 : : errmsg("generation expression uses user-defined function"),
3281 : : errdetail("Virtual generated columns that make use of user-defined functions are not yet supported."),
3282 : : parser_errposition(pstate, exprLocation(node)));
3283 : :
3284 : : /*
3285 : : * check_functions_in_node() doesn't check some node types (see
3286 : : * comment there). We handle CoerceToDomain and MinMaxExpr by
3287 : : * checking for built-in types. The other listed node types cannot
3288 : : * call user-definable SQL-visible functions.
3289 : : *
3290 : : * We furthermore need this type check to handle built-in, immutable
3291 : : * polymorphic functions such as array_eq().
3292 : : */
3293 [ + + ]: 1302 : if (exprType(node) >= FirstUnpinnedObjectId)
3294 [ + - ]: 3 : ereport(ERROR,
3295 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3296 : : errmsg("generation expression uses user-defined type"),
3297 : : errdetail("Virtual generated columns that make use of user-defined types are not yet supported."),
3298 : : parser_errposition(pstate, exprLocation(node)));
3299 : : }
3300 : :
3301 : 1302 : return expression_tree_walker(node, check_virtual_generated_security_walker, context);
3302 : : }
3303 : :
3304 : : static void
3305 : 383 : check_virtual_generated_security(ParseState *pstate, Node *node)
3306 : : {
3307 : 383 : check_virtual_generated_security_walker(node, pstate);
3308 : 374 : }
3309 : :
3310 : : /*
3311 : : * Take a raw default and convert it to a cooked format ready for
3312 : : * storage.
3313 : : *
3314 : : * Parse state should be set up to recognize any vars that might appear
3315 : : * in the expression. (Even though we plan to reject vars, it's more
3316 : : * user-friendly to give the correct error message than "unknown var".)
3317 : : *
3318 : : * If atttypid is not InvalidOid, coerce the expression to the specified
3319 : : * type (and typmod atttypmod). attname is only needed in this case:
3320 : : * it is used in the error message, if any.
3321 : : */
3322 : : Node *
8762 bruce@momjian.us 3323 : 2555 : cookDefault(ParseState *pstate,
3324 : : Node *raw_default,
3325 : : Oid atttypid,
3326 : : int32 atttypmod,
3327 : : const char *attname,
3328 : : char attgenerated)
3329 : : {
3330 : : Node *expr;
3331 : :
3332 [ - + ]: 2555 : Assert(raw_default != NULL);
3333 : :
3334 : : /*
3335 : : * Transform raw parsetree to executable expression.
3336 : : */
2542 peter@eisentraut.org 3337 [ + + ]: 2555 : expr = transformExpr(pstate, raw_default, attgenerated ? EXPR_KIND_GENERATED_COLUMN : EXPR_KIND_COLUMN_DEFAULT);
3338 : :
3339 [ + + ]: 2489 : if (attgenerated)
3340 : : {
3341 : : /* Disallow refs to other generated columns */
3342 : 946 : check_nested_generated(pstate, expr);
3343 : :
3344 : : /* Disallow mutable functions */
850 tgl@sss.pgh.pa.us 3345 [ + + ]: 922 : if (contain_mutable_functions_after_planning((Expr *) expr))
2542 peter@eisentraut.org 3346 [ + - ]: 6 : ereport(ERROR,
3347 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3348 : : errmsg("generation expression is not immutable")));
3349 : :
3350 : : /* Check security of expressions for virtual generated column */
263 3351 [ + + ]: 916 : if (attgenerated == ATTRIBUTE_GENERATED_VIRTUAL)
3352 : 383 : check_virtual_generated_security(pstate, expr);
3353 : : }
3354 : : else
3355 : : {
3356 : : /*
3357 : : * For a default expression, transformExpr() should have rejected
3358 : : * column references.
3359 : : */
2542 3360 [ - + ]: 1543 : Assert(!contain_var_clause(expr));
3361 : : }
3362 : :
3363 : : /*
3364 : : * Coerce the expression to the correct type and typmod, if given. This
3365 : : * should match the parser's processing of non-defaulted expressions ---
3366 : : * see transformAssignedExpr().
3367 : : */
8761 tgl@sss.pgh.pa.us 3368 [ + - ]: 2450 : if (OidIsValid(atttypid))
3369 : : {
8593 bruce@momjian.us 3370 : 2450 : Oid type_id = exprType(expr);
3371 : :
8265 tgl@sss.pgh.pa.us 3372 : 2450 : expr = coerce_to_target_type(pstate, expr, type_id,
3373 : : atttypid, atttypmod,
3374 : : COERCION_ASSIGNMENT,
3375 : : COERCE_IMPLICIT_CAST,
3376 : : -1);
3377 [ - + ]: 2447 : if (expr == NULL)
8273 tgl@sss.pgh.pa.us 3378 [ # # ]:UBC 0 : ereport(ERROR,
3379 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
3380 : : errmsg("column \"%s\" is of type %s"
3381 : : " but default expression is of type %s",
3382 : : attname,
3383 : : format_type_be(atttypid),
3384 : : format_type_be(type_id)),
3385 : : errhint("You will need to rewrite or cast the expression.")));
3386 : : }
3387 : :
3388 : : /*
3389 : : * Finally, take care of collations in the finished expression.
3390 : : */
5475 tgl@sss.pgh.pa.us 3391 :CBC 2447 : assign_expr_collations(pstate, expr);
3392 : :
8265 3393 : 2447 : return expr;
3394 : : }
3395 : :
3396 : : /*
3397 : : * Take a raw CHECK constraint expression and convert it to a cooked format
3398 : : * ready for storage.
3399 : : *
3400 : : * Parse state must be set up to recognize any vars that might appear
3401 : : * in the expression.
3402 : : */
3403 : : static Node *
6519 3404 : 1280 : cookConstraint(ParseState *pstate,
3405 : : Node *raw_constraint,
3406 : : char *relname)
3407 : : {
3408 : : Node *expr;
3409 : :
3410 : : /*
3411 : : * Transform raw parsetree to executable expression.
3412 : : */
4965 3413 : 1280 : expr = transformExpr(pstate, raw_constraint, EXPR_KIND_CHECK_CONSTRAINT);
3414 : :
3415 : : /*
3416 : : * Make sure it yields a boolean result.
3417 : : */
6519 3418 : 1265 : expr = coerce_to_boolean(pstate, expr, "CHECK");
3419 : :
3420 : : /*
3421 : : * Take care of collations.
3422 : : */
5475 3423 : 1265 : assign_expr_collations(pstate, expr);
3424 : :
3425 : : /*
3426 : : * Make sure no outside relations are referred to (this is probably dead
3427 : : * code now that add_missing_from is history).
3428 : : */
6519 3429 [ - + ]: 1265 : if (list_length(pstate->p_rtable) != 1)
6519 tgl@sss.pgh.pa.us 3430 [ # # ]:UBC 0 : ereport(ERROR,
3431 : : (errcode(ERRCODE_INVALID_COLUMN_REFERENCE),
3432 : : errmsg("only table \"%s\" can be referenced in check constraint",
3433 : : relname)));
3434 : :
6519 tgl@sss.pgh.pa.us 3435 :CBC 1265 : return expr;
3436 : : }
3437 : :
3438 : : /*
3439 : : * CopyStatistics --- copy entries in pg_statistic from one rel to another
3440 : : */
3441 : : void
1960 michael@paquier.xyz 3442 : 278 : CopyStatistics(Oid fromrelid, Oid torelid)
3443 : : {
3444 : : HeapTuple tup;
3445 : : SysScanDesc scan;
3446 : : ScanKeyData key[1];
3447 : : Relation statrel;
1215 3448 : 278 : CatalogIndexState indstate = NULL;
3449 : :
1960 3450 : 278 : statrel = table_open(StatisticRelationId, RowExclusiveLock);
3451 : :
3452 : : /* Now search for stat records */
3453 : 278 : ScanKeyInit(&key[0],
3454 : : Anum_pg_statistic_starelid,
3455 : : BTEqualStrategyNumber, F_OIDEQ,
3456 : : ObjectIdGetDatum(fromrelid));
3457 : :
3458 : 278 : scan = systable_beginscan(statrel, StatisticRelidAttnumInhIndexId,
3459 : : true, NULL, 1, key);
3460 : :
3461 [ + + ]: 281 : while (HeapTupleIsValid((tup = systable_getnext(scan))))
3462 : : {
3463 : : Form_pg_statistic statform;
3464 : :
3465 : : /* make a modifiable copy */
3466 : 3 : tup = heap_copytuple(tup);
3467 : 3 : statform = (Form_pg_statistic) GETSTRUCT(tup);
3468 : :
3469 : : /* update the copy of the tuple and insert it */
3470 : 3 : statform->starelid = torelid;
3471 : :
3472 : : /* fetch index information when we know we need it */
1215 3473 [ + - ]: 3 : if (indstate == NULL)
3474 : 3 : indstate = CatalogOpenIndexes(statrel);
3475 : :
3476 : 3 : CatalogTupleInsertWithInfo(statrel, tup, indstate);
3477 : :
1960 3478 : 3 : heap_freetuple(tup);
3479 : : }
3480 : :
3481 : 278 : systable_endscan(scan);
3482 : :
1215 3483 [ + + ]: 278 : if (indstate != NULL)
3484 : 3 : CatalogCloseIndexes(indstate);
1960 3485 : 278 : table_close(statrel, RowExclusiveLock);
3486 : 278 : }
3487 : :
3488 : : /*
3489 : : * RemoveStatistics --- remove entries in pg_statistic for a rel or column
3490 : : *
3491 : : * If attnum is zero, remove all entries for rel; else remove only the one(s)
3492 : : * for that column.
3493 : : */
3494 : : void
7869 tgl@sss.pgh.pa.us 3495 : 28223 : RemoveStatistics(Oid relid, AttrNumber attnum)
3496 : : {
3497 : : Relation pgstatistic;
3498 : : SysScanDesc scan;
3499 : : ScanKeyData key[2];
3500 : : int nkeys;
3501 : : HeapTuple tuple;
3502 : :
2610 andres@anarazel.de 3503 : 28223 : pgstatistic = table_open(StatisticRelationId, RowExclusiveLock);
3504 : :
8159 tgl@sss.pgh.pa.us 3505 : 28223 : ScanKeyInit(&key[0],
3506 : : Anum_pg_statistic_starelid,
3507 : : BTEqualStrategyNumber, F_OIDEQ,
3508 : : ObjectIdGetDatum(relid));
3509 : :
8343 3510 [ + + ]: 28223 : if (attnum == 0)
3511 : 26474 : nkeys = 1;
3512 : : else
3513 : : {
8159 3514 : 1749 : ScanKeyInit(&key[1],
3515 : : Anum_pg_statistic_staattnum,
3516 : : BTEqualStrategyNumber, F_INT2EQ,
3517 : : Int16GetDatum(attnum));
8343 3518 : 1749 : nkeys = 2;
3519 : : }
3520 : :
5920 3521 : 28223 : scan = systable_beginscan(pgstatistic, StatisticRelidAttnumInhIndexId, true,
3522 : : NULL, nkeys, key);
3523 : :
3524 : : /* we must loop even when attnum != 0, in case of inherited stats */
8640 3525 [ + + ]: 30202 : while (HeapTupleIsValid(tuple = systable_getnext(scan)))
3329 3526 : 1979 : CatalogTupleDelete(pgstatistic, &tuple->t_self);
3527 : :
8640 3528 : 28223 : systable_endscan(scan);
3529 : :
2610 andres@anarazel.de 3530 : 28223 : table_close(pgstatistic, RowExclusiveLock);
9604 tgl@sss.pgh.pa.us 3531 : 28223 : }
3532 : :
3533 : :
3534 : : /*
3535 : : * RelationTruncateIndexes - truncate all indexes associated
3536 : : * with the heap relation to zero tuples.
3537 : : *
3538 : : * The routine will truncate and then reconstruct the indexes on
3539 : : * the specified relation. Caller must hold exclusive lock on rel.
3540 : : */
3541 : : static void
7167 3542 : 286 : RelationTruncateIndexes(Relation heapRelation)
3543 : : {
3544 : : ListCell *indlist;
3545 : :
3546 : : /* Ask the relcache to produce a list of the indexes of the rel */
8327 3547 [ + + + + : 401 : foreach(indlist, RelationGetIndexList(heapRelation))
+ + ]
3548 : : {
7963 neilc@samurai.com 3549 : 115 : Oid indexId = lfirst_oid(indlist);
3550 : : Relation currentIndex;
3551 : : IndexInfo *indexInfo;
3552 : :
3553 : : /* Open the index relation; use exclusive lock, just to be sure */
7167 tgl@sss.pgh.pa.us 3554 : 115 : currentIndex = index_open(indexId, AccessExclusiveLock);
3555 : :
3556 : : /*
3557 : : * Fetch info needed for index_build. Since we know there are no
3558 : : * tuples that actually need indexing, we can use a dummy IndexInfo.
3559 : : * This is slightly cheaper to build, but the real point is to avoid
3560 : : * possibly running user-defined code in index expressions or
3561 : : * predicates. We might be getting invoked during ON COMMIT
3562 : : * processing, and we don't want to run any such code then.
3563 : : */
2296 3564 : 115 : indexInfo = BuildDummyIndexInfo(currentIndex);
3565 : :
3566 : : /*
3567 : : * Now truncate the actual file (and discard buffers).
3568 : : */
7981 3569 : 115 : RelationTruncate(currentIndex, 0);
3570 : :
3571 : : /* Initialize the index and rebuild */
3572 : : /* Note: we do not need to re-establish pkey setting */
2607 michael@paquier.xyz 3573 : 115 : index_build(heapRelation, currentIndex, indexInfo, true, false);
3574 : :
3575 : : /* We're done with this index */
7167 tgl@sss.pgh.pa.us 3576 : 115 : index_close(currentIndex, NoLock);
3577 : : }
8645 3578 : 286 : }
3579 : :
3580 : : /*
3581 : : * heap_truncate
3582 : : *
3583 : : * This routine deletes all data within all the specified relations.
3584 : : *
3585 : : * This is not transaction-safe! There is another, transaction-safe
3586 : : * implementation in commands/tablecmds.c. We now use this only for
3587 : : * ON COMMIT truncation of temporary tables, where it doesn't matter.
3588 : : */
3589 : : void
7717 3590 : 191 : heap_truncate(List *relids)
3591 : : {
3592 : 191 : List *relations = NIL;
3593 : : ListCell *cell;
3594 : :
3595 : : /* Open relations for processing, and grab exclusive access on each */
3596 [ + - + + : 415 : foreach(cell, relids)
+ + ]
3597 : : {
3598 : 224 : Oid rid = lfirst_oid(cell);
3599 : : Relation rel;
3600 : :
2610 andres@anarazel.de 3601 : 224 : rel = table_open(rid, AccessExclusiveLock);
7717 tgl@sss.pgh.pa.us 3602 : 224 : relations = lappend(relations, rel);
3603 : : }
3604 : :
3605 : : /* Don't allow truncate on tables that are referenced by foreign keys */
3606 : 191 : heap_truncate_check_FKs(relations, true);
3607 : :
3608 : : /* OK to do it */
3609 [ + - + + : 406 : foreach(cell, relations)
+ + ]
3610 : : {
3611 : 218 : Relation rel = lfirst(cell);
3612 : :
3613 : : /* Truncate the relation */
6048 3614 : 218 : heap_truncate_one_rel(rel);
3615 : :
3616 : : /* Close the relation, but keep exclusive lock on it until commit */
2610 andres@anarazel.de 3617 : 218 : table_close(rel, NoLock);
3618 : : }
8645 tgl@sss.pgh.pa.us 3619 : 188 : }
3620 : :
3621 : : /*
3622 : : * heap_truncate_one_rel
3623 : : *
3624 : : * This routine deletes all data within the specified relation.
3625 : : *
3626 : : * This is not transaction-safe, because the truncation is done immediately
3627 : : * and cannot be rolled back later. Caller is responsible for having
3628 : : * checked permissions etc, and must have obtained AccessExclusiveLock.
3629 : : */
3630 : : void
6048 3631 : 242 : heap_truncate_one_rel(Relation rel)
3632 : : {
3633 : : Oid toastrelid;
3634 : :
3635 : : /*
3636 : : * Truncate the relation. Partitioned tables have no storage, so there is
3637 : : * nothing to do for them here.
3638 : : */
2687 michael@paquier.xyz 3639 [ + + ]: 242 : if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
3640 : 12 : return;
3641 : :
3642 : : /* Truncate the underlying relation */
2544 andres@anarazel.de 3643 : 230 : table_relation_nontransactional_truncate(rel);
3644 : :
3645 : : /* If the relation has indexes, truncate the indexes too */
6048 tgl@sss.pgh.pa.us 3646 : 230 : RelationTruncateIndexes(rel);
3647 : :
3648 : : /* If there is a toast table, truncate that too */
3649 : 230 : toastrelid = rel->rd_rel->reltoastrelid;
3650 [ + + ]: 230 : if (OidIsValid(toastrelid))
3651 : : {
2610 andres@anarazel.de 3652 : 56 : Relation toastrel = table_open(toastrelid, AccessExclusiveLock);
3653 : :
2544 3654 : 56 : table_relation_nontransactional_truncate(toastrel);
6048 tgl@sss.pgh.pa.us 3655 : 56 : RelationTruncateIndexes(toastrel);
3656 : : /* keep the lock... */
2610 andres@anarazel.de 3657 : 56 : table_close(toastrel, NoLock);
3658 : : }
3659 : : }
3660 : :
3661 : : /*
3662 : : * heap_truncate_check_FKs
3663 : : * Check for foreign keys referencing a list of relations that
3664 : : * are to be truncated, and raise error if there are any
3665 : : *
3666 : : * We disallow such FKs (except self-referential ones) since the whole point
3667 : : * of TRUNCATE is to not scan the individual rows to be thrown away.
3668 : : *
3669 : : * This is split out so it can be shared by both implementations of truncate.
3670 : : * Caller should already hold a suitable lock on the relations.
3671 : : *
3672 : : * tempTables is only used to select an appropriate error message.
3673 : : */
3674 : : void
7717 tgl@sss.pgh.pa.us 3675 : 1081 : heap_truncate_check_FKs(List *relations, bool tempTables)
3676 : : {
3677 : 1081 : List *oids = NIL;
3678 : : List *dependents;
3679 : : ListCell *cell;
3680 : :
3681 : : /*
3682 : : * Build a list of OIDs of the interesting relations.
3683 : : *
3684 : : * If a relation has no triggers, then it can neither have FKs nor be
3685 : : * referenced by a FK from another table, so we can ignore it. For
3686 : : * partitioned tables, FKs have no triggers, so we must include them
3687 : : * anyway.
3688 : : */
3689 [ + - + + : 3312 : foreach(cell, relations)
+ + ]
3690 : : {
3691 : 2231 : Relation rel = lfirst(cell);
3692 : :
2803 alvherre@alvh.no-ip. 3693 [ + + ]: 2231 : if (rel->rd_rel->relhastriggers ||
3694 [ + + ]: 1535 : rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
7717 tgl@sss.pgh.pa.us 3695 : 955 : oids = lappend_oid(oids, RelationGetRelid(rel));
3696 : : }
3697 : :
3698 : : /*
3699 : : * Fast path: if no relation has triggers, none has FKs either.
3700 : : */
3701 [ + + ]: 1081 : if (oids == NIL)
8213 3702 : 687 : return;
3703 : :
3704 : : /*
3705 : : * Otherwise, must scan pg_constraint. We make one pass with all the
3706 : : * relations considered; if this finds nothing, then all is well.
3707 : : */
7199 3708 : 394 : dependents = heap_truncate_find_FKs(oids);
3709 [ + + ]: 394 : if (dependents == NIL)
3710 : 354 : return;
3711 : :
3712 : : /*
3713 : : * Otherwise we repeat the scan once per relation to identify a particular
3714 : : * pair of relations to complain about. This is pretty slow, but
3715 : : * performance shouldn't matter much in a failure path. The reason for
3716 : : * doing things this way is to ensure that the message produced is not
3717 : : * dependent on chance row locations within pg_constraint.
3718 : : */
3719 [ + - + - : 52 : foreach(cell, oids)
+ - ]
3720 : : {
7102 bruce@momjian.us 3721 : 52 : Oid relid = lfirst_oid(cell);
3722 : : ListCell *cell2;
3723 : :
7199 tgl@sss.pgh.pa.us 3724 : 52 : dependents = heap_truncate_find_FKs(list_make1_oid(relid));
3725 : :
3726 [ + + + + : 82 : foreach(cell2, dependents)
+ + ]
3727 : : {
7102 bruce@momjian.us 3728 : 70 : Oid relid2 = lfirst_oid(cell2);
3729 : :
7199 tgl@sss.pgh.pa.us 3730 [ + + ]: 70 : if (!list_member_oid(oids, relid2))
3731 : : {
7102 bruce@momjian.us 3732 : 40 : char *relname = get_rel_name(relid);
3733 : 40 : char *relname2 = get_rel_name(relid2);
3734 : :
7199 tgl@sss.pgh.pa.us 3735 [ + + ]: 40 : if (tempTables)
3736 [ + - ]: 3 : ereport(ERROR,
3737 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3738 : : errmsg("unsupported ON COMMIT and foreign key combination"),
3739 : : errdetail("Table \"%s\" references \"%s\", but they do not have the same ON COMMIT setting.",
3740 : : relname2, relname)));
3741 : : else
3742 [ + - ]: 37 : ereport(ERROR,
3743 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3744 : : errmsg("cannot truncate a table referenced in a foreign key constraint"),
3745 : : errdetail("Table \"%s\" references \"%s\".",
3746 : : relname2, relname),
3747 : : errhint("Truncate table \"%s\" at the same time, "
3748 : : "or use TRUNCATE ... CASCADE.",
3749 : : relname2)));
3750 : : }
3751 : : }
3752 : : }
3753 : : }
3754 : :
3755 : : /*
3756 : : * heap_truncate_find_FKs
3757 : : * Find relations having foreign keys referencing any of the given rels
3758 : : *
3759 : : * Input and result are both lists of relation OIDs. The result contains
3760 : : * no duplicates, does *not* include any rels that were already in the input
3761 : : * list, and is sorted in OID order. (The last property is enforced mainly
3762 : : * to guarantee consistent behavior in the regression tests; we don't want
3763 : : * behavior to change depending on chance locations of rows in pg_constraint.)
3764 : : *
3765 : : * Note: caller should already have appropriate lock on all rels mentioned
3766 : : * in relationIds. Since adding or dropping an FK requires exclusive lock
3767 : : * on both rels, this ensures that the answer will be stable.
3768 : : */
3769 : : List *
7317 3770 : 486 : heap_truncate_find_FKs(List *relationIds)
3771 : : {
3772 : 486 : List *result = NIL;
3773 : : List *oids;
3774 : : List *parent_cons;
3775 : : ListCell *cell;
3776 : : ScanKeyData key;
3777 : : Relation fkeyRel;
3778 : : SysScanDesc fkeyScan;
3779 : : HeapTuple tuple;
3780 : : bool restart;
3781 : :
2228 alvherre@alvh.no-ip. 3782 : 486 : oids = list_copy(relationIds);
3783 : :
3784 : : /*
3785 : : * Must scan pg_constraint. Right now, it is a seqscan because there is
3786 : : * no available index on confrelid.
3787 : : */
2610 andres@anarazel.de 3788 : 486 : fkeyRel = table_open(ConstraintRelationId, AccessShareLock);
3789 : :
2228 alvherre@alvh.no-ip. 3790 : 504 : restart:
3791 : 504 : restart = false;
3792 : 504 : parent_cons = NIL;
3793 : :
7317 tgl@sss.pgh.pa.us 3794 : 504 : fkeyScan = systable_beginscan(fkeyRel, InvalidOid, false,
3795 : : NULL, 0, NULL);
3796 : :
3797 [ + + ]: 259274 : while (HeapTupleIsValid(tuple = systable_getnext(fkeyScan)))
3798 : : {
3799 : 258770 : Form_pg_constraint con = (Form_pg_constraint) GETSTRUCT(tuple);
3800 : :
3801 : : /* Not a foreign key */
3802 [ + + ]: 258770 : if (con->contype != CONSTRAINT_FOREIGN)
3803 : 238142 : continue;
3804 : :
3805 : : /* Not referencing one of our list of tables */
2228 alvherre@alvh.no-ip. 3806 [ + + ]: 20628 : if (!list_member_oid(oids, con->confrelid))
7317 tgl@sss.pgh.pa.us 3807 : 20072 : continue;
3808 : :
3809 : : /*
3810 : : * If this constraint has a parent constraint which we have not seen
3811 : : * yet, keep track of it for the second loop, below. Tracking parent
3812 : : * constraints allows us to climb up to the top-level constraint and
3813 : : * look for all possible relations referencing the partitioned table.
3814 : : */
2228 alvherre@alvh.no-ip. 3815 [ + + ]: 556 : if (OidIsValid(con->conparentid) &&
3816 [ + + ]: 174 : !list_member_oid(parent_cons, con->conparentid))
3817 : 93 : parent_cons = lappend_oid(parent_cons, con->conparentid);
3818 : :
3819 : : /*
3820 : : * Add referencer to result, unless present in input list. (Don't
3821 : : * worry about dupes: we'll fix that below).
3822 : : */
7317 tgl@sss.pgh.pa.us 3823 [ + + ]: 556 : if (!list_member_oid(relationIds, con->conrelid))
2434 3824 : 250 : result = lappend_oid(result, con->conrelid);
3825 : : }
3826 : :
7317 3827 : 504 : systable_endscan(fkeyScan);
3828 : :
3829 : : /*
3830 : : * Process each parent constraint we found to add the list of referenced
3831 : : * relations by them to the oids list. If we do add any new such
3832 : : * relations, redo the first loop above. Also, if we see that the parent
3833 : : * constraint in turn has a parent, add that so that we process all
3834 : : * relations in a single additional pass.
3835 : : */
2228 alvherre@alvh.no-ip. 3836 [ + + + + : 606 : foreach(cell, parent_cons)
+ + ]
3837 : : {
2131 tgl@sss.pgh.pa.us 3838 : 102 : Oid parent = lfirst_oid(cell);
3839 : :
2228 alvherre@alvh.no-ip. 3840 : 102 : ScanKeyInit(&key,
3841 : : Anum_pg_constraint_oid,
3842 : : BTEqualStrategyNumber, F_OIDEQ,
3843 : : ObjectIdGetDatum(parent));
3844 : :
3845 : 102 : fkeyScan = systable_beginscan(fkeyRel, ConstraintOidIndexId,
3846 : : true, NULL, 1, &key);
3847 : :
3848 : 102 : tuple = systable_getnext(fkeyScan);
3849 [ + - ]: 102 : if (HeapTupleIsValid(tuple))
3850 : : {
3851 : 102 : Form_pg_constraint con = (Form_pg_constraint) GETSTRUCT(tuple);
3852 : :
3853 : : /*
3854 : : * pg_constraint rows always appear for partitioned hierarchies
3855 : : * this way: on the each side of the constraint, one row appears
3856 : : * for each partition that points to the top-most table on the
3857 : : * other side.
3858 : : *
3859 : : * Because of this arrangement, we can correctly catch all
3860 : : * relevant relations by adding to 'parent_cons' all rows with
3861 : : * valid conparentid, and to the 'oids' list all rows with a zero
3862 : : * conparentid. If any oids are added to 'oids', redo the first
3863 : : * loop above by setting 'restart'.
3864 : : */
3865 [ + + ]: 102 : if (OidIsValid(con->conparentid))
3866 : 36 : parent_cons = list_append_unique_oid(parent_cons,
3867 : : con->conparentid);
3868 [ + + ]: 66 : else if (!list_member_oid(oids, con->confrelid))
3869 : : {
3870 : 18 : oids = lappend_oid(oids, con->confrelid);
3871 : 18 : restart = true;
3872 : : }
3873 : : }
3874 : :
3875 : 102 : systable_endscan(fkeyScan);
3876 : : }
3877 : :
3878 : 504 : list_free(parent_cons);
3879 [ + + ]: 504 : if (restart)
3880 : 18 : goto restart;
3881 : :
2610 andres@anarazel.de 3882 : 486 : table_close(fkeyRel, AccessShareLock);
2228 alvherre@alvh.no-ip. 3883 : 486 : list_free(oids);
3884 : :
3885 : : /* Now sort and de-duplicate the result list */
2434 tgl@sss.pgh.pa.us 3886 : 486 : list_sort(result, list_oid_cmp);
3887 : 486 : list_deduplicate_oid(result);
3888 : :
3889 : 486 : return result;
3890 : : }
3891 : :
3892 : : /*
3893 : : * StorePartitionKey
3894 : : * Store information about the partition key rel into the catalog
3895 : : */
3896 : : void
3385 rhaas@postgresql.org 3897 : 2716 : StorePartitionKey(Relation rel,
3898 : : char strategy,
3899 : : int16 partnatts,
3900 : : AttrNumber *partattrs,
3901 : : List *partexprs,
3902 : : Oid *partopclass,
3903 : : Oid *partcollation)
3904 : : {
3905 : : int i;
3906 : : int2vector *partattrs_vec;
3907 : : oidvector *partopclass_vec;
3908 : : oidvector *partcollation_vec;
3909 : : Datum partexprDatum;
3910 : : Relation pg_partitioned_table;
3911 : : HeapTuple tuple;
3912 : : Datum values[Natts_pg_partitioned_table];
1338 peter@eisentraut.org 3913 : 2716 : bool nulls[Natts_pg_partitioned_table] = {0};
3914 : : ObjectAddress myself;
3915 : : ObjectAddress referenced;
3916 : : ObjectAddresses *addrs;
3917 : :
3385 rhaas@postgresql.org 3918 [ - + ]: 2716 : Assert(rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE);
3919 : :
3920 : : /* Copy the partition attribute numbers, opclass OIDs into arrays */
3921 : 2716 : partattrs_vec = buildint2vector(partattrs, partnatts);
3922 : 2716 : partopclass_vec = buildoidvector(partopclass, partnatts);
3923 : 2716 : partcollation_vec = buildoidvector(partcollation, partnatts);
3924 : :
3925 : : /* Convert the expressions (if any) to a text datum */
3926 [ + + ]: 2716 : if (partexprs)
3927 : : {
3928 : : char *exprString;
3929 : :
3930 : 113 : exprString = nodeToString(partexprs);
3931 : 113 : partexprDatum = CStringGetTextDatum(exprString);
3932 : 113 : pfree(exprString);
3933 : : }
3934 : : else
3935 : 2603 : partexprDatum = (Datum) 0;
3936 : :
2610 andres@anarazel.de 3937 : 2716 : pg_partitioned_table = table_open(PartitionedRelationId, RowExclusiveLock);
3938 : :
3939 : : /* Only this can ever be NULL */
3385 rhaas@postgresql.org 3940 [ + + ]: 2716 : if (!partexprDatum)
3941 : 2603 : nulls[Anum_pg_partitioned_table_partexprs - 1] = true;
3942 : :
3943 : 2716 : values[Anum_pg_partitioned_table_partrelid - 1] = ObjectIdGetDatum(RelationGetRelid(rel));
3944 : 2716 : values[Anum_pg_partitioned_table_partstrat - 1] = CharGetDatum(strategy);
3945 : 2716 : values[Anum_pg_partitioned_table_partnatts - 1] = Int16GetDatum(partnatts);
3110 3946 : 2716 : values[Anum_pg_partitioned_table_partdefid - 1] = ObjectIdGetDatum(InvalidOid);
3337 3947 : 2716 : values[Anum_pg_partitioned_table_partattrs - 1] = PointerGetDatum(partattrs_vec);
3385 3948 : 2716 : values[Anum_pg_partitioned_table_partclass - 1] = PointerGetDatum(partopclass_vec);
3949 : 2716 : values[Anum_pg_partitioned_table_partcollation - 1] = PointerGetDatum(partcollation_vec);
3950 : 2716 : values[Anum_pg_partitioned_table_partexprs - 1] = partexprDatum;
3951 : :
3952 : 2716 : tuple = heap_form_tuple(RelationGetDescr(pg_partitioned_table), values, nulls);
3953 : :
3330 alvherre@alvh.no-ip. 3954 : 2716 : CatalogTupleInsert(pg_partitioned_table, tuple);
2610 andres@anarazel.de 3955 : 2716 : table_close(pg_partitioned_table, RowExclusiveLock);
3956 : :
3957 : : /* Mark this relation as dependent on a few things as follows */
2017 michael@paquier.xyz 3958 : 2716 : addrs = new_object_addresses();
3959 : 2716 : ObjectAddressSet(myself, RelationRelationId, RelationGetRelid(rel));
3960 : :
3961 : : /* Operator class and collation per key column */
3385 rhaas@postgresql.org 3962 [ + + ]: 5669 : for (i = 0; i < partnatts; i++)
3963 : : {
2017 michael@paquier.xyz 3964 : 2953 : ObjectAddressSet(referenced, OperatorClassRelationId, partopclass[i]);
3965 : 2953 : add_exact_object_address(&referenced, addrs);
3966 : :
3967 : : /* The default collation is pinned, so don't bother recording it */
3204 rhaas@postgresql.org 3968 [ + + ]: 2953 : if (OidIsValid(partcollation[i]) &&
3969 [ + + ]: 330 : partcollation[i] != DEFAULT_COLLATION_OID)
3970 : : {
2017 michael@paquier.xyz 3971 : 60 : ObjectAddressSet(referenced, CollationRelationId, partcollation[i]);
3972 : 60 : add_exact_object_address(&referenced, addrs);
3973 : : }
3974 : : }
3975 : :
3976 : 2716 : record_object_address_dependencies(&myself, addrs, DEPENDENCY_NORMAL);
3977 : 2716 : free_object_addresses(addrs);
3978 : :
3979 : : /*
3980 : : * The partitioning columns are made internally dependent on the table,
3981 : : * because we cannot drop any of them without dropping the whole table.
3982 : : * (ATExecDropColumn independently enforces that, but it's not bulletproof
3983 : : * so we need the dependencies too.)
3984 : : */
2428 tgl@sss.pgh.pa.us 3985 [ + + ]: 5669 : for (i = 0; i < partnatts; i++)
3986 : : {
3987 [ + + ]: 2953 : if (partattrs[i] == 0)
3988 : 122 : continue; /* ignore expressions here */
3989 : :
2017 michael@paquier.xyz 3990 : 2831 : ObjectAddressSubSet(referenced, RelationRelationId,
3991 : : RelationGetRelid(rel), partattrs[i]);
2428 tgl@sss.pgh.pa.us 3992 : 2831 : recordDependencyOn(&referenced, &myself, DEPENDENCY_INTERNAL);
3993 : : }
3994 : :
3995 : : /*
3996 : : * Also consider anything mentioned in partition expressions. External
3997 : : * references (e.g. functions) get NORMAL dependencies. Table columns
3998 : : * mentioned in the expressions are handled the same as plain partitioning
3999 : : * columns, i.e. they become internally dependent on the whole table.
4000 : : */
3385 rhaas@postgresql.org 4001 [ + + ]: 2716 : if (partexprs)
4002 : 113 : recordDependencyOnSingleRelExpr(&myself,
4003 : : (Node *) partexprs,
4004 : : RelationGetRelid(rel),
4005 : : DEPENDENCY_NORMAL,
4006 : : DEPENDENCY_INTERNAL,
4007 : : true /* reverse the self-deps */ );
4008 : :
4009 : : /*
4010 : : * We must invalidate the relcache so that the next
4011 : : * CommandCounterIncrement() will cause the same to be rebuilt using the
4012 : : * information in just created catalog entry.
4013 : : */
4014 : 2716 : CacheInvalidateRelcache(rel);
4015 : 2716 : }
4016 : :
4017 : : /*
4018 : : * RemovePartitionKeyByRelId
4019 : : * Remove pg_partitioned_table entry for a relation
4020 : : */
4021 : : void
4022 : 2206 : RemovePartitionKeyByRelId(Oid relid)
4023 : : {
4024 : : Relation rel;
4025 : : HeapTuple tuple;
4026 : :
2610 andres@anarazel.de 4027 : 2206 : rel = table_open(PartitionedRelationId, RowExclusiveLock);
4028 : :
3385 rhaas@postgresql.org 4029 : 2206 : tuple = SearchSysCache1(PARTRELID, ObjectIdGetDatum(relid));
4030 [ - + ]: 2206 : if (!HeapTupleIsValid(tuple))
3385 rhaas@postgresql.org 4031 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for partition key of relation %u",
4032 : : relid);
4033 : :
3329 tgl@sss.pgh.pa.us 4034 :CBC 2206 : CatalogTupleDelete(rel, &tuple->t_self);
4035 : :
3385 rhaas@postgresql.org 4036 : 2206 : ReleaseSysCache(tuple);
2610 andres@anarazel.de 4037 : 2206 : table_close(rel, RowExclusiveLock);
3385 rhaas@postgresql.org 4038 : 2206 : }
4039 : :
4040 : : /*
4041 : : * StorePartitionBound
4042 : : * Update pg_class tuple of rel to store the partition bound and set
4043 : : * relispartition to true
4044 : : *
4045 : : * If this is the default partition, also update the default partition OID in
4046 : : * pg_partitioned_table.
4047 : : *
4048 : : * Also, invalidate the parent's relcache, so that the next rebuild will load
4049 : : * the new partition's info into its partition descriptor. If there is a
4050 : : * default partition, we must invalidate its relcache entry as well.
4051 : : */
4052 : : void
3213 tgl@sss.pgh.pa.us 4053 : 5847 : StorePartitionBound(Relation rel, Relation parent, PartitionBoundSpec *bound)
4054 : : {
4055 : : Relation classRel;
4056 : : HeapTuple tuple,
4057 : : newtuple;
4058 : : Datum new_val[Natts_pg_class];
4059 : : bool new_null[Natts_pg_class],
4060 : : new_repl[Natts_pg_class];
4061 : : Oid defaultPartOid;
4062 : :
4063 : : /* Update pg_class tuple */
2610 andres@anarazel.de 4064 : 5847 : classRel = table_open(RelationRelationId, RowExclusiveLock);
3385 rhaas@postgresql.org 4065 : 5847 : tuple = SearchSysCacheCopy1(RELOID,
4066 : : ObjectIdGetDatum(RelationGetRelid(rel)));
3357 4067 [ - + ]: 5847 : if (!HeapTupleIsValid(tuple))
3357 rhaas@postgresql.org 4068 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for relation %u",
4069 : : RelationGetRelid(rel));
4070 : :
4071 : : #ifdef USE_ASSERT_CHECKING
4072 : : {
4073 : : Form_pg_class classForm;
4074 : : bool isnull;
4075 : :
3385 rhaas@postgresql.org 4076 :CBC 5847 : classForm = (Form_pg_class) GETSTRUCT(tuple);
4077 [ - + ]: 5847 : Assert(!classForm->relispartition);
4078 : 5847 : (void) SysCacheGetAttr(RELOID, tuple, Anum_pg_class_relpartbound,
4079 : : &isnull);
4080 [ - + ]: 5847 : Assert(isnull);
4081 : : }
4082 : : #endif
4083 : :
4084 : : /* Fill in relpartbound value */
4085 : 5847 : memset(new_val, 0, sizeof(new_val));
4086 : 5847 : memset(new_null, false, sizeof(new_null));
4087 : 5847 : memset(new_repl, false, sizeof(new_repl));
4088 : 5847 : new_val[Anum_pg_class_relpartbound - 1] = CStringGetTextDatum(nodeToString(bound));
4089 : 5847 : new_null[Anum_pg_class_relpartbound - 1] = false;
4090 : 5847 : new_repl[Anum_pg_class_relpartbound - 1] = true;
4091 : 5847 : newtuple = heap_modify_tuple(tuple, RelationGetDescr(classRel),
4092 : : new_val, new_null, new_repl);
4093 : : /* Also set the flag */
4094 : 5847 : ((Form_pg_class) GETSTRUCT(newtuple))->relispartition = true;
4095 : :
4096 : : /*
4097 : : * We already checked for no inheritance children, but reset
4098 : : * relhassubclass in case it was left over.
4099 : : */
599 alvherre@alvh.no-ip. 4100 [ + + + + ]: 5847 : if (rel->rd_rel->relkind == RELKIND_RELATION && rel->rd_rel->relhassubclass)
4101 : 3 : ((Form_pg_class) GETSTRUCT(newtuple))->relhassubclass = false;
4102 : :
3330 4103 : 5847 : CatalogTupleUpdate(classRel, &newtuple->t_self, newtuple);
3385 rhaas@postgresql.org 4104 : 5847 : heap_freetuple(newtuple);
2610 andres@anarazel.de 4105 : 5847 : table_close(classRel, RowExclusiveLock);
4106 : :
4107 : : /*
4108 : : * If we're storing bounds for the default partition, update
4109 : : * pg_partitioned_table too.
4110 : : */
2916 alvherre@alvh.no-ip. 4111 [ + + ]: 5847 : if (bound->is_default)
4112 : 370 : update_default_partition_oid(RelationGetRelid(parent),
4113 : : RelationGetRelid(rel));
4114 : :
4115 : : /* Make these updates visible */
2917 4116 : 5847 : CommandCounterIncrement();
4117 : :
4118 : : /*
4119 : : * The partition constraint for the default partition depends on the
4120 : : * partition bounds of every other partition, so we must invalidate the
4121 : : * relcache entry for that partition every time a partition is added or
4122 : : * removed.
4123 : : */
4124 : : defaultPartOid =
1788 4125 : 5847 : get_default_oid_from_partdesc(RelationGetPartitionDesc(parent, true));
3110 rhaas@postgresql.org 4126 [ + + ]: 5847 : if (OidIsValid(defaultPartOid))
4127 : 441 : CacheInvalidateRelcacheByRelid(defaultPartOid);
4128 : :
3373 4129 : 5847 : CacheInvalidateRelcache(parent);
3385 4130 : 5847 : }
|
