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