Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * parse_utilcmd.c
4 : : * Perform parse analysis work for various utility commands
5 : : *
6 : : * Formerly we did this work during parse_analyze_*() in analyze.c. However
7 : : * that is fairly unsafe in the presence of querytree caching, since any
8 : : * database state that we depend on in making the transformations might be
9 : : * obsolete by the time the utility command is executed; and utility commands
10 : : * have no infrastructure for holding locks or rechecking plan validity.
11 : : * Hence these functions are now called at the start of execution of their
12 : : * respective utility commands.
13 : : *
14 : : *
15 : : * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
16 : : * Portions Copyright (c) 1994, Regents of the University of California
17 : : *
18 : : * src/backend/parser/parse_utilcmd.c
19 : : *
20 : : *-------------------------------------------------------------------------
21 : : */
22 : :
23 : : #include "postgres.h"
24 : :
25 : : #include "access/amapi.h"
26 : : #include "access/attmap.h"
27 : : #include "access/htup_details.h"
28 : : #include "access/relation.h"
29 : : #include "access/reloptions.h"
30 : : #include "access/table.h"
31 : : #include "access/toast_compression.h"
32 : : #include "catalog/dependency.h"
33 : : #include "catalog/heap.h"
34 : : #include "catalog/index.h"
35 : : #include "catalog/namespace.h"
36 : : #include "catalog/partition.h"
37 : : #include "catalog/pg_am.h"
38 : : #include "catalog/pg_collation.h"
39 : : #include "catalog/pg_constraint.h"
40 : : #include "catalog/pg_opclass.h"
41 : : #include "catalog/pg_operator.h"
42 : : #include "catalog/pg_statistic_ext.h"
43 : : #include "catalog/pg_type.h"
44 : : #include "commands/comment.h"
45 : : #include "commands/defrem.h"
46 : : #include "commands/sequence.h"
47 : : #include "commands/tablecmds.h"
48 : : #include "commands/tablespace.h"
49 : : #include "miscadmin.h"
50 : : #include "nodes/makefuncs.h"
51 : : #include "nodes/nodeFuncs.h"
52 : : #include "optimizer/optimizer.h"
53 : : #include "parser/analyze.h"
54 : : #include "parser/parse_clause.h"
55 : : #include "parser/parse_coerce.h"
56 : : #include "parser/parse_collate.h"
57 : : #include "parser/parse_expr.h"
58 : : #include "parser/parse_relation.h"
59 : : #include "parser/parse_target.h"
60 : : #include "parser/parse_type.h"
61 : : #include "parser/parse_utilcmd.h"
62 : : #include "parser/parser.h"
63 : : #include "partitioning/partbounds.h"
64 : : #include "partitioning/partdesc.h"
65 : : #include "rewrite/rewriteManip.h"
66 : : #include "utils/acl.h"
67 : : #include "utils/builtins.h"
68 : : #include "utils/lsyscache.h"
69 : : #include "utils/partcache.h"
70 : : #include "utils/rel.h"
71 : : #include "utils/ruleutils.h"
72 : : #include "utils/syscache.h"
73 : : #include "utils/typcache.h"
74 : :
75 : :
76 : : /* State shared by transformCreateStmt and its subroutines */
77 : : typedef struct
78 : : {
79 : : ParseState *pstate; /* overall parser state */
80 : : const char *stmtType; /* "CREATE [FOREIGN] TABLE" or "ALTER TABLE" */
81 : : RangeVar *relation; /* relation to create */
82 : : Relation rel; /* opened/locked rel, if ALTER */
83 : : List *inhRelations; /* relations to inherit from */
84 : : bool isforeign; /* true if CREATE/ALTER FOREIGN TABLE */
85 : : bool isalter; /* true if altering existing table */
86 : : List *columns; /* ColumnDef items */
87 : : List *ckconstraints; /* CHECK constraints */
88 : : List *nnconstraints; /* NOT NULL constraints */
89 : : List *fkconstraints; /* FOREIGN KEY constraints */
90 : : List *ixconstraints; /* index-creating constraints */
91 : : List *likeclauses; /* LIKE clauses that need post-processing */
92 : : List *blist; /* "before list" of things to do before
93 : : * creating the table */
94 : : List *alist; /* "after list" of things to do after creating
95 : : * the table */
96 : : IndexStmt *pkey; /* PRIMARY KEY index, if any */
97 : : bool ispartitioned; /* true if table is partitioned */
98 : : PartitionBoundSpec *partbound; /* transformed FOR VALUES */
99 : : bool ofType; /* true if statement contains OF typename */
100 : : } CreateStmtContext;
101 : :
102 : :
103 : : static void transformColumnDefinition(CreateStmtContext *cxt,
104 : : ColumnDef *column);
105 : : static void transformTableConstraint(CreateStmtContext *cxt,
106 : : Constraint *constraint);
107 : : static void transformTableLikeClause(CreateStmtContext *cxt,
108 : : TableLikeClause *table_like_clause);
109 : : static void transformOfType(CreateStmtContext *cxt,
110 : : TypeName *ofTypename);
111 : : static CreateStatsStmt *generateClonedExtStatsStmt(RangeVar *heapRel,
112 : : Oid heapRelid,
113 : : Oid source_statsid,
114 : : const AttrMap *attmap);
115 : : static List *get_collation(Oid collation, Oid actual_datatype);
116 : : static List *get_opclass(Oid opclass, Oid actual_datatype);
117 : : static void transformIndexConstraints(CreateStmtContext *cxt);
118 : : static IndexStmt *transformIndexConstraint(Constraint *constraint,
119 : : CreateStmtContext *cxt);
120 : : static void transformFKConstraints(CreateStmtContext *cxt,
121 : : bool skipValidation,
122 : : bool isAddConstraint);
123 : : static void transformCheckConstraints(CreateStmtContext *cxt,
124 : : bool skipValidation);
125 : : static void transformConstraintAttrs(ParseState *pstate,
126 : : List *constraintList);
127 : : static void transformColumnType(CreateStmtContext *cxt, ColumnDef *column);
128 : : static void checkSchemaNameRV(ParseState *pstate, const char *context_schema,
129 : : RangeVar *relation);
130 : : static void checkSchemaNameList(const char *context_schema,
131 : : List *qualified_name);
132 : : static CreateStmt *transformCreateSchemaCreateTable(ParseState *pstate,
133 : : CreateStmt *stmt,
134 : : List **fk_elements);
135 : : static void transformPartitionCmd(CreateStmtContext *cxt, PartitionBoundSpec *bound);
136 : : static List *transformPartitionRangeBounds(ParseState *pstate, List *blist,
137 : : Relation parent);
138 : : static void validateInfiniteBounds(ParseState *pstate, List *blist);
139 : : static Const *transformPartitionBoundValue(ParseState *pstate, Node *val,
140 : : const char *colName, Oid colType, int32 colTypmod,
141 : : Oid partCollation);
142 : :
143 : :
144 : : /*
145 : : * transformCreateStmt -
146 : : * parse analysis for CREATE TABLE
147 : : *
148 : : * Returns a List of utility commands to be done in sequence. One of these
149 : : * will be the transformed CreateStmt, but there may be additional actions
150 : : * to be done before and after the actual DefineRelation() call.
151 : : * In addition to normal utility commands such as AlterTableStmt and
152 : : * IndexStmt, the result list may contain TableLikeClause(s), representing
153 : : * the need to perform additional parse analysis after DefineRelation().
154 : : *
155 : : * SQL allows constraints to be scattered all over, so thumb through
156 : : * the columns and collect all constraints into one place.
157 : : * If there are any implied indices (e.g. UNIQUE or PRIMARY KEY)
158 : : * then expand those into multiple IndexStmt blocks.
159 : : * - thomas 1997-12-02
160 : : */
161 : : List *
6891 tgl@sss.pgh.pa.us 162 :CBC 26410 : transformCreateStmt(CreateStmt *stmt, const char *queryString)
163 : : {
164 : : ParseState *pstate;
165 : : CreateStmtContext cxt;
166 : : List *result;
167 : : List *save_alist;
168 : : ListCell *elements;
169 : : Oid namespaceid;
170 : : Oid existing_relid;
171 : : ParseCallbackState pcbstate;
172 : :
173 : : /* Set up pstate */
4066 alvherre@alvh.no-ip. 174 : 26410 : pstate = make_parsestate(NULL);
175 : 26410 : pstate->p_sourcetext = queryString;
176 : :
177 : : /*
178 : : * Look up the creation namespace. This also checks permissions on the
179 : : * target namespace, locks it against concurrent drops, checks for a
180 : : * preexisting relation in that namespace with the same name, and updates
181 : : * stmt->relation->relpersistence if the selected namespace is temporary.
182 : : */
183 : 26410 : setup_parser_errposition_callback(&pcbstate, pstate,
184 : 26410 : stmt->relation->location);
185 : : namespaceid =
5223 rhaas@postgresql.org 186 : 26410 : RangeVarGetAndCheckCreationNamespace(stmt->relation, NoLock,
187 : : &existing_relid);
4066 alvherre@alvh.no-ip. 188 : 26402 : cancel_parser_errposition_callback(&pcbstate);
189 : :
190 : : /*
191 : : * If the relation already exists and the user specified "IF NOT EXISTS",
192 : : * bail out with a NOTICE.
193 : : */
5223 rhaas@postgresql.org 194 [ + + + + ]: 26402 : if (stmt->if_not_exists && OidIsValid(existing_relid))
195 : : {
196 : : /*
197 : : * If we are in an extension script, insist that the pre-existing
198 : : * object be a member of the extension, to avoid security risks.
199 : : */
200 : : ObjectAddress address;
201 : :
1366 tgl@sss.pgh.pa.us 202 : 6 : ObjectAddressSet(address, RelationRelationId, existing_relid);
203 : 6 : checkMembershipInCurrentExtension(&address);
204 : :
205 : : /* OK to skip */
5223 rhaas@postgresql.org 206 [ + + ]: 5 : ereport(NOTICE,
207 : : (errcode(ERRCODE_DUPLICATE_TABLE),
208 : : errmsg("relation \"%s\" already exists, skipping",
209 : : stmt->relation->relname)));
210 : 5 : return NIL;
211 : : }
212 : :
213 : : /*
214 : : * If the target relation name isn't schema-qualified, make it so. This
215 : : * prevents some corner cases in which added-on rewritten commands might
216 : : * think they should apply to other relations that have the same name and
217 : : * are earlier in the search path. But a local temp table is effectively
218 : : * specified to be in pg_temp, so no need for anything extra in that case.
219 : : */
5622 220 [ + + ]: 26396 : if (stmt->relation->schemaname == NULL
221 [ + + ]: 24736 : && stmt->relation->relpersistence != RELPERSISTENCE_TEMP)
6826 tgl@sss.pgh.pa.us 222 : 23029 : stmt->relation->schemaname = get_namespace_name(namespaceid);
223 : :
224 : : /* Set up CreateStmtContext */
5579 225 : 26396 : cxt.pstate = pstate;
5603 rhaas@postgresql.org 226 [ + + ]: 26396 : if (IsA(stmt, CreateForeignTableStmt))
227 : : {
228 : 299 : cxt.stmtType = "CREATE FOREIGN TABLE";
4802 tgl@sss.pgh.pa.us 229 : 299 : cxt.isforeign = true;
230 : : }
231 : : else
232 : : {
5603 rhaas@postgresql.org 233 : 26097 : cxt.stmtType = "CREATE TABLE";
4802 tgl@sss.pgh.pa.us 234 : 26097 : cxt.isforeign = false;
235 : : }
6891 236 : 26396 : cxt.relation = stmt->relation;
237 : 26396 : cxt.rel = NULL;
238 : 26396 : cxt.inhRelations = stmt->inhRelations;
239 : 26396 : cxt.isalter = false;
240 : 26396 : cxt.columns = NIL;
241 : 26396 : cxt.ckconstraints = NIL;
543 alvherre@alvh.no-ip. 242 : 26396 : cxt.nnconstraints = NIL;
6891 tgl@sss.pgh.pa.us 243 : 26396 : cxt.fkconstraints = NIL;
244 : 26396 : cxt.ixconstraints = NIL;
1993 245 : 26396 : cxt.likeclauses = NIL;
6891 246 : 26396 : cxt.blist = NIL;
247 : 26396 : cxt.alist = NIL;
248 : 26396 : cxt.pkey = NULL;
3436 rhaas@postgresql.org 249 : 26396 : cxt.ispartitioned = stmt->partspec != NULL;
3070 peter_e@gmx.net 250 : 26396 : cxt.partbound = stmt->partbound;
251 : 26396 : cxt.ofType = (stmt->ofTypename != NULL);
252 : :
5912 bruce@momjian.us 253 [ + + - + ]: 26396 : Assert(!stmt->ofTypename || !stmt->inhRelations); /* grammar enforces */
254 : :
5941 peter_e@gmx.net 255 [ + + ]: 26396 : if (stmt->ofTypename)
5579 tgl@sss.pgh.pa.us 256 : 81 : transformOfType(&cxt, stmt->ofTypename);
257 : :
3436 rhaas@postgresql.org 258 [ + + ]: 26384 : if (stmt->partspec)
259 : : {
260 [ + + + + ]: 3613 : if (stmt->inhRelations && !stmt->partbound)
261 [ + - ]: 4 : ereport(ERROR,
262 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
263 : : errmsg("cannot create partitioned table as inheritance child")));
264 : : }
265 : :
266 : : /*
267 : : * Run through each primary element in the table creation clause. Separate
268 : : * column defs from constraints, and do preliminary analysis.
269 : : */
6891 tgl@sss.pgh.pa.us 270 [ + + + + : 72140 : foreach(elements, stmt->tableElts)
+ + ]
271 : : {
272 : 45921 : Node *element = lfirst(elements);
273 : :
274 [ + + + - ]: 45921 : switch (nodeTag(element))
275 : : {
276 : 43395 : case T_ColumnDef:
5579 277 : 43395 : transformColumnDefinition(&cxt, (ColumnDef *) element);
6891 278 : 43250 : break;
279 : :
3421 280 : 2001 : case T_Constraint:
281 : 2001 : transformTableConstraint(&cxt, (Constraint *) element);
6891 282 : 1993 : break;
283 : :
3421 284 : 525 : case T_TableLikeClause:
285 : 525 : transformTableLikeClause(&cxt, (TableLikeClause *) element);
3865 bruce@momjian.us 286 : 517 : break;
287 : :
6891 tgl@sss.pgh.pa.us 288 :UBC 0 : default:
289 [ # # ]: 0 : elog(ERROR, "unrecognized node type: %d",
290 : : (int) nodeTag(element));
291 : : break;
292 : : }
293 : : }
294 : :
295 : : /*
296 : : * Transfer anything we already have in cxt.alist into save_alist, to keep
297 : : * it separate from the output of transformIndexConstraints. (This may
298 : : * not be necessary anymore, but we'll keep doing it to preserve the
299 : : * historical order of execution of the alist commands.)
300 : : */
6891 tgl@sss.pgh.pa.us 301 :CBC 26219 : save_alist = cxt.alist;
302 : 26219 : cxt.alist = NIL;
303 : :
304 [ - + ]: 26219 : Assert(stmt->constraints == NIL);
305 : :
306 : : /*
307 : : * Before processing index constraints, which could include a primary key,
308 : : * we must scan all not-null constraints to propagate the is_not_null flag
309 : : * to each corresponding ColumnDef. This is necessary because table-level
310 : : * not-null constraints have not been marked in each ColumnDef, and the PK
311 : : * processing code needs to know whether one constraint has already been
312 : : * declared in order not to declare a redundant one.
313 : : */
543 alvherre@alvh.no-ip. 314 [ + + + + : 60273 : foreach_node(Constraint, nn, cxt.nnconstraints)
+ + ]
315 : : {
316 : 7835 : char *colname = strVal(linitial(nn->keys));
317 : :
318 [ + + + + : 18657 : foreach_node(ColumnDef, cd, cxt.columns)
+ + ]
319 : : {
320 : : /* not our column? */
321 [ + + ]: 10805 : if (strcmp(cd->colname, colname) != 0)
322 : 2987 : continue;
323 : : /* Already marked not-null? Nothing to do */
324 [ + + ]: 7818 : if (cd->is_not_null)
325 : 7482 : break;
326 : : /* Bingo, we're done for this constraint */
327 : 336 : cd->is_not_null = true;
328 : 336 : break;
329 : : }
330 : : }
331 : :
332 : : /*
333 : : * Postprocess constraints that give rise to index definitions.
334 : : */
5579 tgl@sss.pgh.pa.us 335 : 26219 : transformIndexConstraints(&cxt);
336 : :
337 : : /*
338 : : * Re-consideration of LIKE clauses should happen after creation of
339 : : * indexes, but before creation of foreign keys. This order is critical
340 : : * because a LIKE clause may attempt to create a primary key. If there's
341 : : * also a pkey in the main CREATE TABLE list, creation of that will not
342 : : * check for a duplicate at runtime (since index_check_primary_key()
343 : : * expects that we rejected dups here). Creation of the LIKE-generated
344 : : * pkey behaves like ALTER TABLE ADD, so it will check, but obviously that
345 : : * only works if it happens second. On the other hand, we want to make
346 : : * pkeys before foreign key constraints, in case the user tries to make a
347 : : * self-referential FK.
348 : : */
1993 349 : 26191 : cxt.alist = list_concat(cxt.alist, cxt.likeclauses);
350 : :
351 : : /*
352 : : * Postprocess foreign-key constraints.
353 : : */
5579 354 : 26191 : transformFKConstraints(&cxt, true, false);
355 : :
356 : : /*
357 : : * Postprocess check constraints.
358 : : *
359 : : * For regular tables all constraints can be marked valid immediately,
360 : : * because the table is new therefore empty. Not so for foreign tables.
361 : : */
1825 alvherre@alvh.no-ip. 362 : 26191 : transformCheckConstraints(&cxt, !cxt.isforeign);
363 : :
364 : : /*
365 : : * Output results.
366 : : */
6891 tgl@sss.pgh.pa.us 367 : 26191 : stmt->tableElts = cxt.columns;
368 : 26191 : stmt->constraints = cxt.ckconstraints;
543 alvherre@alvh.no-ip. 369 : 26191 : stmt->nnconstraints = cxt.nnconstraints;
370 : :
6891 tgl@sss.pgh.pa.us 371 : 26191 : result = lappend(cxt.blist, stmt);
372 : 26191 : result = list_concat(result, cxt.alist);
373 : 26191 : result = list_concat(result, save_alist);
374 : :
375 : 26191 : return result;
376 : : }
377 : :
378 : : /*
379 : : * generateSerialExtraStmts
380 : : * Generate CREATE SEQUENCE and ALTER SEQUENCE ... OWNED BY statements
381 : : * to create the sequence for a serial or identity column.
382 : : *
383 : : * This includes determining the name the sequence will have. The caller
384 : : * can ask to get back the name components by passing non-null pointers
385 : : * for snamespace_p and sname_p.
386 : : */
387 : : static void
3316 peter_e@gmx.net 388 : 869 : generateSerialExtraStmts(CreateStmtContext *cxt, ColumnDef *column,
389 : : Oid seqtypid, List *seqoptions,
390 : : bool for_identity, bool col_exists,
391 : : char **snamespace_p, char **sname_p)
392 : : {
393 : : ListCell *option;
3275 bruce@momjian.us 394 : 869 : DefElem *nameEl = NULL;
595 tgl@sss.pgh.pa.us 395 : 869 : DefElem *loggedEl = NULL;
396 : : Oid snamespaceid;
397 : : char *snamespace;
398 : : char *sname;
399 : : char seqpersistence;
400 : : CreateSeqStmt *seqstmt;
401 : : AlterSeqStmt *altseqstmt;
402 : : List *attnamelist;
403 : :
404 : : /* Make a copy of this as we may end up modifying it in the code below */
1131 drowley@postgresql.o 405 : 869 : seqoptions = list_copy(seqoptions);
406 : :
407 : : /*
408 : : * Check for non-SQL-standard options (not supported within CREATE
409 : : * SEQUENCE, because they'd be redundant), and remove them from the
410 : : * seqoptions list if found.
411 : : */
3316 peter_e@gmx.net 412 [ + + + + : 1075 : foreach(option, seqoptions)
+ + ]
413 : : {
3312 tgl@sss.pgh.pa.us 414 : 206 : DefElem *defel = lfirst_node(DefElem, option);
415 : :
3316 peter_e@gmx.net 416 [ + + ]: 206 : if (strcmp(defel->defname, "sequence_name") == 0)
417 : : {
418 [ - + ]: 22 : if (nameEl)
1755 dean.a.rasheed@gmail 419 :UBC 0 : errorConflictingDefElem(defel, cxt->pstate);
3316 peter_e@gmx.net 420 :CBC 22 : nameEl = defel;
595 tgl@sss.pgh.pa.us 421 : 22 : seqoptions = foreach_delete_current(seqoptions, option);
422 : : }
423 [ + + ]: 184 : else if (strcmp(defel->defname, "logged") == 0 ||
424 [ + + ]: 183 : strcmp(defel->defname, "unlogged") == 0)
425 : : {
426 [ - + ]: 2 : if (loggedEl)
595 tgl@sss.pgh.pa.us 427 :UBC 0 : errorConflictingDefElem(defel, cxt->pstate);
595 tgl@sss.pgh.pa.us 428 :CBC 2 : loggedEl = defel;
429 : 2 : seqoptions = foreach_delete_current(seqoptions, option);
430 : : }
431 : : }
432 : :
433 : : /*
434 : : * Determine namespace and name to use for the sequence.
435 : : */
3316 peter_e@gmx.net 436 [ + + ]: 869 : if (nameEl)
437 : : {
438 : : /* Use specified name */
3275 bruce@momjian.us 439 : 22 : RangeVar *rv = makeRangeVarFromNameList(castNode(List, nameEl->arg));
440 : :
3316 peter_e@gmx.net 441 : 22 : snamespace = rv->schemaname;
3250 tgl@sss.pgh.pa.us 442 [ - + ]: 22 : if (!snamespace)
443 : : {
444 : : /* Given unqualified SEQUENCE NAME, select namespace */
3250 tgl@sss.pgh.pa.us 445 [ # # ]:UBC 0 : if (cxt->rel)
446 : 0 : snamespaceid = RelationGetNamespace(cxt->rel);
447 : : else
448 : 0 : snamespaceid = RangeVarGetCreationNamespace(cxt->relation);
449 : 0 : snamespace = get_namespace_name(snamespaceid);
450 : : }
3316 peter_e@gmx.net 451 :CBC 22 : sname = rv->relname;
452 : : }
453 : : else
454 : : {
455 : : /*
456 : : * Generate a name.
457 : : *
458 : : * Although we use ChooseRelationName, it's not guaranteed that the
459 : : * selected sequence name won't conflict; given sufficiently long
460 : : * field names, two different serial columns in the same table could
461 : : * be assigned the same sequence name, and we'd not notice since we
462 : : * aren't creating the sequence quite yet. In practice this seems
463 : : * quite unlikely to be a problem, especially since few people would
464 : : * need two serial columns in one table.
465 : : */
466 [ + + ]: 847 : if (cxt->rel)
467 : 137 : snamespaceid = RelationGetNamespace(cxt->rel);
468 : : else
469 : : {
470 : 710 : snamespaceid = RangeVarGetCreationNamespace(cxt->relation);
471 : 710 : RangeVarAdjustRelationPersistence(cxt->relation, snamespaceid);
472 : : }
473 : 847 : snamespace = get_namespace_name(snamespaceid);
474 : 847 : sname = ChooseRelationName(cxt->relation->relname,
475 : 847 : column->colname,
476 : : "seq",
477 : : snamespaceid,
478 : : false);
479 : : }
480 : :
481 [ + + ]: 869 : ereport(DEBUG1,
482 : : (errmsg_internal("%s will create implicit sequence \"%s\" for serial column \"%s.%s\"",
483 : : cxt->stmtType, sname,
484 : : cxt->relation->relname, column->colname)));
485 : :
486 : : /*
487 : : * Determine the persistence of the sequence. By default we copy the
488 : : * persistence of the table, but if LOGGED or UNLOGGED was specified, use
489 : : * that (as long as the table isn't TEMP).
490 : : *
491 : : * For CREATE TABLE, we get the persistence from cxt->relation, which
492 : : * comes from the CreateStmt in progress. For ALTER TABLE, the parser
493 : : * won't set cxt->relation->relpersistence, but we have cxt->rel as the
494 : : * existing table, so we copy the persistence from there.
495 : : */
595 tgl@sss.pgh.pa.us 496 [ + + ]: 869 : seqpersistence = cxt->rel ? cxt->rel->rd_rel->relpersistence : cxt->relation->relpersistence;
497 [ + + ]: 869 : if (loggedEl)
498 : : {
499 [ - + ]: 2 : if (seqpersistence == RELPERSISTENCE_TEMP)
595 tgl@sss.pgh.pa.us 500 [ # # ]:UBC 0 : ereport(ERROR,
501 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
502 : : errmsg("cannot set logged status of a temporary sequence"),
503 : : parser_errposition(cxt->pstate, loggedEl->location)));
595 tgl@sss.pgh.pa.us 504 [ + + ]:CBC 2 : else if (strcmp(loggedEl->defname, "logged") == 0)
505 : 1 : seqpersistence = RELPERSISTENCE_PERMANENT;
506 : : else
507 : 1 : seqpersistence = RELPERSISTENCE_UNLOGGED;
508 : : }
509 : :
510 : : /*
511 : : * Build a CREATE SEQUENCE command to create the sequence object, and add
512 : : * it to the list of things to be done before this CREATE/ALTER TABLE.
513 : : */
3316 peter_e@gmx.net 514 : 869 : seqstmt = makeNode(CreateSeqStmt);
515 : 869 : seqstmt->for_identity = for_identity;
516 : 869 : seqstmt->sequence = makeRangeVar(snamespace, sname, -1);
595 tgl@sss.pgh.pa.us 517 : 869 : seqstmt->sequence->relpersistence = seqpersistence;
3316 peter_e@gmx.net 518 : 869 : seqstmt->options = seqoptions;
519 : :
520 : : /*
521 : : * If a sequence data type was specified, add it to the options. Prepend
522 : : * to the list rather than append; in case a user supplied their own AS
523 : : * clause, the "redundant options" error will point to their occurrence,
524 : : * not our synthetic one.
525 : : */
526 [ + + ]: 869 : if (seqtypid)
3250 tgl@sss.pgh.pa.us 527 : 861 : seqstmt->options = lcons(makeDefElem("as",
3240 528 : 861 : (Node *) makeTypeNameFromOid(seqtypid, -1),
529 : : -1),
530 : : seqstmt->options);
531 : :
532 : : /*
533 : : * If this is ALTER ADD COLUMN, make sure the sequence will be owned by
534 : : * the table's owner. The current user might be someone else (perhaps a
535 : : * superuser, or someone who's only a member of the owning role), but the
536 : : * SEQUENCE OWNED BY mechanisms will bleat unless table and sequence have
537 : : * exactly the same owning role.
538 : : */
3316 peter_e@gmx.net 539 [ + + ]: 869 : if (cxt->rel)
540 : 159 : seqstmt->ownerId = cxt->rel->rd_rel->relowner;
541 : : else
542 : 710 : seqstmt->ownerId = InvalidOid;
543 : :
544 : 869 : cxt->blist = lappend(cxt->blist, seqstmt);
545 : :
546 : : /*
547 : : * Store the identity sequence name that we decided on. ALTER TABLE ...
548 : : * ADD COLUMN ... IDENTITY needs this so that it can fill the new column
549 : : * with values from the sequence, while the association of the sequence
550 : : * with the table is not set until after the ALTER TABLE.
551 : : */
3014 552 : 869 : column->identitySequence = seqstmt->sequence;
553 : :
554 : : /*
555 : : * Build an ALTER SEQUENCE ... OWNED BY command to mark the sequence as
556 : : * owned by this column, and add it to the appropriate list of things to
557 : : * be done along with this CREATE/ALTER TABLE. In a CREATE or ALTER ADD
558 : : * COLUMN, it must be done after the statement because we don't know the
559 : : * column's attnum yet. But if we do have the attnum (in AT_AddIdentity),
560 : : * we can do the marking immediately, which improves some ALTER TABLE
561 : : * behaviors.
562 : : */
3316 563 : 869 : altseqstmt = makeNode(AlterSeqStmt);
564 : 869 : altseqstmt->sequence = makeRangeVar(snamespace, sname, -1);
565 : 869 : attnamelist = list_make3(makeString(snamespace),
566 : : makeString(cxt->relation->relname),
567 : : makeString(column->colname));
568 : 869 : altseqstmt->options = list_make1(makeDefElem("owned_by",
569 : : (Node *) attnamelist, -1));
570 : 869 : altseqstmt->for_identity = for_identity;
571 : :
2302 tgl@sss.pgh.pa.us 572 [ + + ]: 869 : if (col_exists)
573 : 103 : cxt->blist = lappend(cxt->blist, altseqstmt);
574 : : else
575 : 766 : cxt->alist = lappend(cxt->alist, altseqstmt);
576 : :
3316 peter_e@gmx.net 577 [ + + ]: 869 : if (snamespace_p)
578 : 538 : *snamespace_p = snamespace;
579 [ + + ]: 869 : if (sname_p)
580 : 538 : *sname_p = sname;
581 : 869 : }
582 : :
583 : : /*
584 : : * transformColumnDefinition -
585 : : * transform a single ColumnDef within CREATE TABLE
586 : : * Also used in ALTER TABLE ADD COLUMN
587 : : */
588 : : static void
5579 tgl@sss.pgh.pa.us 589 : 44911 : transformColumnDefinition(CreateStmtContext *cxt, ColumnDef *column)
590 : : {
591 : : bool is_serial;
592 : : bool saw_nullable;
593 : : bool saw_default;
594 : : bool saw_identity;
595 : : bool saw_generated;
543 alvherre@alvh.no-ip. 596 : 44911 : bool need_notnull = false;
597 : 44911 : bool disallow_noinherit_notnull = false;
598 : 44911 : Constraint *notnull_constraint = NULL;
599 : :
6891 tgl@sss.pgh.pa.us 600 : 44911 : cxt->columns = lappend(cxt->columns, column);
601 : :
602 : : /* Check for SERIAL pseudo-types */
603 : 44911 : is_serial = false;
5941 peter_e@gmx.net 604 [ + + ]: 44911 : if (column->typeName
605 [ + + ]: 44691 : && list_length(column->typeName->names) == 1
606 [ + - ]: 18902 : && !column->typeName->pct_type)
607 : : {
6137 608 : 18902 : char *typname = strVal(linitial(column->typeName->names));
609 : :
5432 rhaas@postgresql.org 610 [ + + ]: 18902 : if (strcmp(typname, "smallserial") == 0 ||
611 [ + + ]: 18897 : strcmp(typname, "serial2") == 0)
612 : : {
613 : 9 : is_serial = true;
614 : 9 : column->typeName->names = NIL;
615 : 9 : column->typeName->typeOid = INT2OID;
616 : : }
617 [ + + ]: 18893 : else if (strcmp(typname, "serial") == 0 ||
5077 bruce@momjian.us 618 [ - + ]: 18382 : strcmp(typname, "serial4") == 0)
619 : : {
6891 tgl@sss.pgh.pa.us 620 : 511 : is_serial = true;
6137 peter_e@gmx.net 621 : 511 : column->typeName->names = NIL;
622 : 511 : column->typeName->typeOid = INT4OID;
623 : : }
6891 tgl@sss.pgh.pa.us 624 [ + + ]: 18382 : else if (strcmp(typname, "bigserial") == 0 ||
625 [ + + ]: 18372 : strcmp(typname, "serial8") == 0)
626 : : {
627 : 18 : is_serial = true;
6137 peter_e@gmx.net 628 : 18 : column->typeName->names = NIL;
629 : 18 : column->typeName->typeOid = INT8OID;
630 : : }
631 : :
632 : : /*
633 : : * We have to reject "serial[]" explicitly, because once we've set
634 : : * typeid, LookupTypeName won't notice arrayBounds. We don't need any
635 : : * special coding for serial(typmod) though.
636 : : */
637 [ + + - + ]: 18902 : if (is_serial && column->typeName->arrayBounds != NIL)
6619 tgl@sss.pgh.pa.us 638 [ # # ]:UBC 0 : ereport(ERROR,
639 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
640 : : errmsg("array of serial is not implemented"),
641 : : parser_errposition(cxt->pstate,
642 : : column->typeName->location)));
643 : : }
644 : :
645 : : /* Do necessary work on the column type declaration */
5941 peter_e@gmx.net 646 [ + + ]:CBC 44911 : if (column->typeName)
5579 tgl@sss.pgh.pa.us 647 : 44691 : transformColumnType(cxt, column);
648 : :
649 : : /* Special actions for SERIAL pseudo-types */
6891 650 [ + + ]: 44886 : if (is_serial)
651 : : {
652 : : char *snamespace;
653 : : char *sname;
654 : : char *qstring;
655 : : A_Const *snamenode;
656 : : TypeCast *castnode;
657 : : FuncCall *funccallnode;
658 : : Constraint *constraint;
659 : :
3316 peter_e@gmx.net 660 : 538 : generateSerialExtraStmts(cxt, column,
2302 tgl@sss.pgh.pa.us 661 : 538 : column->typeName->typeOid, NIL,
662 : : false, false,
663 : : &snamespace, &sname);
664 : :
665 : : /*
666 : : * Create appropriate constraints for SERIAL. We do this in full,
667 : : * rather than shortcutting, so that we will detect any conflicting
668 : : * constraints the user wrote (like a different DEFAULT).
669 : : *
670 : : * Create an expression tree representing the function call
671 : : * nextval('sequencename'). We cannot reduce the raw tree to cooked
672 : : * form until after the sequence is created, but there's no need to do
673 : : * so.
674 : : */
6891 675 : 538 : qstring = quote_qualified_identifier(snamespace, sname);
676 : 538 : snamenode = makeNode(A_Const);
1699 peter@eisentraut.org 677 : 538 : snamenode->val.node.type = T_String;
1572 678 : 538 : snamenode->val.sval.sval = qstring;
6459 tgl@sss.pgh.pa.us 679 : 538 : snamenode->location = -1;
6580 alvherre@alvh.no-ip. 680 : 538 : castnode = makeNode(TypeCast);
6137 peter_e@gmx.net 681 : 538 : castnode->typeName = SystemTypeName("regclass");
6580 alvherre@alvh.no-ip. 682 : 538 : castnode->arg = (Node *) snamenode;
6459 tgl@sss.pgh.pa.us 683 : 538 : castnode->location = -1;
4691 rhaas@postgresql.org 684 : 538 : funccallnode = makeFuncCall(SystemFuncName("nextval"),
4691 rhaas@postgresql.org 685 :ECB (407) : list_make1(castnode),
686 : : COERCE_EXPLICIT_CALL,
687 : : -1);
6891 tgl@sss.pgh.pa.us 688 :CBC 538 : constraint = makeNode(Constraint);
689 : 538 : constraint->contype = CONSTR_DEFAULT;
6123 690 : 538 : constraint->location = -1;
6891 691 : 538 : constraint->raw_expr = (Node *) funccallnode;
692 : 538 : constraint->cooked_expr = NULL;
693 : 538 : column->constraints = lappend(column->constraints, constraint);
694 : :
695 : : /* have a not-null constraint added later */
543 alvherre@alvh.no-ip. 696 : 538 : need_notnull = true;
697 : 538 : disallow_noinherit_notnull = true;
698 : : }
699 : :
700 : : /* Process column constraints, if any... */
29 tgl@sss.pgh.pa.us 701 :GNC 44886 : transformConstraintAttrs(cxt->pstate, column->constraints);
702 : :
703 : : /*
704 : : * First, scan the column's constraints to see if a not-null constraint
705 : : * that we add must be prevented from being NO INHERIT. This should be
706 : : * enforced only for PRIMARY KEY, not IDENTITY or SERIAL. However, if the
707 : : * not-null constraint is specified as a table constraint rather than as a
708 : : * column constraint, AddRelationNotNullConstraints would raise an error
709 : : * if a NO INHERIT mismatch is found. To avoid inconsistently disallowing
710 : : * it in the table constraint case but not the column constraint case, we
711 : : * disallow it here as well. Maybe AddRelationNotNullConstraints can be
712 : : * improved someday, so that it doesn't complain, and then we can remove
713 : : * the restriction for SERIAL and IDENTITY here as well.
714 : : */
543 alvherre@alvh.no-ip. 715 [ + + ]:CBC 44870 : if (!disallow_noinherit_notnull)
716 : : {
717 [ + + + + : 100254 : foreach_node(Constraint, constraint, column->constraints)
+ + ]
718 : : {
719 [ + + ]: 11590 : switch (constraint->contype)
720 : : {
721 : 3570 : case CONSTR_IDENTITY:
722 : : case CONSTR_PRIMARY:
723 : 3570 : disallow_noinherit_notnull = true;
724 : 3570 : break;
725 : 8020 : default:
726 : 8020 : break;
727 : : }
728 : : }
729 : : }
730 : :
731 : : /* Now scan them again to do full processing */
6891 tgl@sss.pgh.pa.us 732 : 44870 : saw_nullable = false;
733 : 44870 : saw_default = false;
3316 peter_e@gmx.net 734 : 44870 : saw_identity = false;
2593 peter@eisentraut.org 735 : 44870 : saw_generated = false;
736 : :
543 alvherre@alvh.no-ip. 737 [ + + + + : 102000 : foreach_node(Constraint, constraint, column->constraints)
+ + ]
738 : : {
6891 tgl@sss.pgh.pa.us 739 [ + + + + : 12476 : switch (constraint->contype)
+ + + + -
+ + - ]
740 : : {
741 : 14 : case CONSTR_NULL:
543 alvherre@alvh.no-ip. 742 [ - + - - : 14 : if ((saw_nullable && column->is_not_null) || need_notnull)
+ + ]
6891 tgl@sss.pgh.pa.us 743 [ + - ]: 4 : ereport(ERROR,
744 : : (errcode(ERRCODE_SYNTAX_ERROR),
745 : : errmsg("conflicting NULL/NOT NULL declarations for column \"%s\" of table \"%s\"",
746 : : column->colname, cxt->relation->relname),
747 : : parser_errposition(cxt->pstate,
748 : : constraint->location)));
3184 peter_e@gmx.net 749 : 10 : column->is_not_null = false;
6891 tgl@sss.pgh.pa.us 750 : 10 : saw_nullable = true;
751 : 10 : break;
752 : :
753 : 4352 : case CONSTR_NOTNULL:
543 alvherre@alvh.no-ip. 754 [ + + + + ]: 4352 : if (cxt->ispartitioned && constraint->is_no_inherit)
755 [ + - ]: 4 : ereport(ERROR,
756 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
757 : : errmsg("not-null constraints on partitioned tables cannot be NO INHERIT"));
758 : :
759 : : /* Disallow conflicting [NOT] NULL markings */
1119 760 [ + + - + ]: 4348 : if (saw_nullable && !column->is_not_null)
1119 alvherre@alvh.no-ip. 761 [ # # ]:UBC 0 : ereport(ERROR,
762 : : (errcode(ERRCODE_SYNTAX_ERROR),
763 : : errmsg("conflicting NULL/NOT NULL declarations for column \"%s\" of table \"%s\"",
764 : : column->colname, cxt->relation->relname),
765 : : parser_errposition(cxt->pstate,
766 : : constraint->location)));
767 : :
543 alvherre@alvh.no-ip. 768 [ + + + + ]:CBC 4348 : if (disallow_noinherit_notnull && constraint->is_no_inherit)
769 [ + - ]: 20 : ereport(ERROR,
770 : : errcode(ERRCODE_SYNTAX_ERROR),
771 : : errmsg("conflicting NO INHERIT declarations for not-null constraints on column \"%s\"",
772 : : column->colname));
773 : :
774 : : /*
775 : : * If this is the first time we see this column being marked
776 : : * not-null, add the constraint entry and keep track of it.
777 : : * Also, remove previous markings that we need one.
778 : : *
779 : : * If this is a redundant not-null specification, just check
780 : : * that it doesn't conflict with what was specified earlier.
781 : : *
782 : : * Any conflicts with table constraints will be further
783 : : * checked in AddRelationNotNullConstraints().
784 : : */
785 [ + + ]: 4328 : if (!column->is_not_null)
786 : : {
787 : 4312 : column->is_not_null = true;
788 : 4312 : saw_nullable = true;
789 : 4312 : need_notnull = false;
790 : :
791 : 4312 : constraint->keys = list_make1(makeString(column->colname));
792 : 4312 : notnull_constraint = constraint;
793 : 4312 : cxt->nnconstraints = lappend(cxt->nnconstraints, constraint);
794 : : }
795 [ + - ]: 16 : else if (notnull_constraint)
796 : : {
797 [ + + ]: 16 : if (constraint->conname &&
798 [ + + ]: 12 : notnull_constraint->conname &&
799 [ + + ]: 8 : strcmp(notnull_constraint->conname, constraint->conname) != 0)
800 [ + - ]: 4 : elog(ERROR, "conflicting not-null constraint names \"%s\" and \"%s\"",
801 : : notnull_constraint->conname, constraint->conname);
802 : :
803 [ - + ]: 12 : if (notnull_constraint->is_no_inherit != constraint->is_no_inherit)
543 alvherre@alvh.no-ip. 804 [ # # ]:UBC 0 : ereport(ERROR,
805 : : errcode(ERRCODE_SYNTAX_ERROR),
806 : : errmsg("conflicting NO INHERIT declarations for not-null constraints on column \"%s\"",
807 : : column->colname));
808 : :
543 alvherre@alvh.no-ip. 809 [ + + + + ]:CBC 12 : if (!notnull_constraint->conname && constraint->conname)
810 : 4 : notnull_constraint->conname = constraint->conname;
811 : : }
812 : :
6891 tgl@sss.pgh.pa.us 813 : 4324 : break;
814 : :
815 : 1632 : case CONSTR_DEFAULT:
816 [ - + ]: 1632 : if (saw_default)
6891 tgl@sss.pgh.pa.us 817 [ # # ]:UBC 0 : ereport(ERROR,
818 : : (errcode(ERRCODE_SYNTAX_ERROR),
819 : : errmsg("multiple default values specified for column \"%s\" of table \"%s\"",
820 : : column->colname, cxt->relation->relname),
821 : : parser_errposition(cxt->pstate,
822 : : constraint->location)));
6891 tgl@sss.pgh.pa.us 823 :CBC 1632 : column->raw_default = constraint->raw_expr;
824 [ - + ]: 1632 : Assert(constraint->cooked_expr == NULL);
825 : 1632 : saw_default = true;
826 : 1632 : break;
827 : :
3316 peter_e@gmx.net 828 : 244 : case CONSTR_IDENTITY:
829 : : {
830 : : Type ctype;
831 : : Oid typeOid;
832 : :
3070 833 [ + + ]: 244 : if (cxt->ofType)
834 [ + - ]: 4 : ereport(ERROR,
835 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
836 : : errmsg("identity columns are not supported on typed tables")));
837 [ + + ]: 240 : if (cxt->partbound)
838 [ + - ]: 16 : ereport(ERROR,
839 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
840 : : errmsg("identity columns are not supported on partitions")));
841 : :
3275 bruce@momjian.us 842 : 224 : ctype = typenameType(cxt->pstate, column->typeName, NULL);
2723 andres@anarazel.de 843 : 224 : typeOid = ((Form_pg_type) GETSTRUCT(ctype))->oid;
3275 bruce@momjian.us 844 : 224 : ReleaseSysCache(ctype);
845 : :
846 [ + + ]: 224 : if (saw_identity)
847 [ + - ]: 4 : ereport(ERROR,
848 : : (errcode(ERRCODE_SYNTAX_ERROR),
849 : : errmsg("multiple identity specifications for column \"%s\" of table \"%s\"",
850 : : column->colname, cxt->relation->relname),
851 : : parser_errposition(cxt->pstate,
852 : : constraint->location)));
853 : :
854 : 220 : generateSerialExtraStmts(cxt, column,
855 : : typeOid, constraint->options,
856 : : true, false,
857 : : NULL, NULL);
858 : :
859 : 220 : column->identity = constraint->generated_when;
860 : 220 : saw_identity = true;
861 : :
862 : : /*
863 : : * Identity columns are always NOT NULL, but we may have a
864 : : * constraint already.
865 : : */
543 alvherre@alvh.no-ip. 866 [ + + ]: 220 : if (!saw_nullable)
867 : 204 : need_notnull = true;
868 [ + + ]: 16 : else if (!column->is_not_null)
1880 tgl@sss.pgh.pa.us 869 [ + - ]: 4 : ereport(ERROR,
870 : : (errcode(ERRCODE_SYNTAX_ERROR),
871 : : errmsg("conflicting NULL/NOT NULL declarations for column \"%s\" of table \"%s\"",
872 : : column->colname, cxt->relation->relname),
873 : : parser_errposition(cxt->pstate,
874 : : constraint->location)));
3275 bruce@momjian.us 875 : 216 : break;
876 : : }
877 : :
2593 peter@eisentraut.org 878 : 1309 : case CONSTR_GENERATED:
879 [ + + ]: 1309 : if (cxt->ofType)
880 [ + - ]: 8 : ereport(ERROR,
881 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
882 : : errmsg("generated columns are not supported on typed tables")));
883 [ + + ]: 1301 : if (saw_generated)
884 [ + - ]: 8 : ereport(ERROR,
885 : : (errcode(ERRCODE_SYNTAX_ERROR),
886 : : errmsg("multiple generation clauses specified for column \"%s\" of table \"%s\"",
887 : : column->colname, cxt->relation->relname),
888 : : parser_errposition(cxt->pstate,
889 : : constraint->location)));
452 890 : 1293 : column->generated = constraint->generated_kind;
2593 891 : 1293 : column->raw_default = constraint->raw_expr;
892 [ - + ]: 1293 : Assert(constraint->cooked_expr == NULL);
893 : 1293 : saw_generated = true;
894 : 1293 : break;
895 : :
5993 tgl@sss.pgh.pa.us 896 : 340 : case CONSTR_CHECK:
4157 897 : 340 : cxt->ckconstraints = lappend(cxt->ckconstraints, constraint);
898 : 340 : break;
899 : :
900 : 3597 : case CONSTR_PRIMARY:
543 alvherre@alvh.no-ip. 901 [ + + - + ]: 3597 : if (saw_nullable && !column->is_not_null)
543 alvherre@alvh.no-ip. 902 [ # # ]:UBC 0 : ereport(ERROR,
903 : : (errcode(ERRCODE_SYNTAX_ERROR),
904 : : errmsg("conflicting NULL/NOT NULL declarations for column \"%s\" of table \"%s\"",
905 : : column->colname, cxt->relation->relname),
906 : : parser_errposition(cxt->pstate,
907 : : constraint->location)));
543 alvherre@alvh.no-ip. 908 :CBC 3597 : need_notnull = true;
909 : :
4802 tgl@sss.pgh.pa.us 910 [ + + ]: 3597 : if (cxt->isforeign)
911 [ + - ]: 4 : ereport(ERROR,
912 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
913 : : errmsg("primary key constraints are not supported on foreign tables"),
914 : : parser_errposition(cxt->pstate,
915 : : constraint->location)));
916 : : pg_fallthrough;
917 : :
918 : : case CONSTR_UNIQUE:
919 [ - + ]: 3835 : if (cxt->isforeign)
4802 tgl@sss.pgh.pa.us 920 [ # # ]:UBC 0 : ereport(ERROR,
921 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
922 : : errmsg("unique constraints are not supported on foreign tables"),
923 : : parser_errposition(cxt->pstate,
924 : : constraint->location)));
6891 tgl@sss.pgh.pa.us 925 [ + - ]:CBC 3835 : if (constraint->keys == NIL)
926 : 3835 : constraint->keys = list_make1(makeString(column->colname));
927 : 3835 : cxt->ixconstraints = lappend(cxt->ixconstraints, constraint);
928 : 3835 : break;
929 : :
5993 tgl@sss.pgh.pa.us 930 :UBC 0 : case CONSTR_EXCLUSION:
931 : : /* grammar does not allow EXCLUDE as a column constraint */
932 [ # # ]: 0 : elog(ERROR, "column exclusion constraints are not supported");
933 : : break;
934 : :
6123 tgl@sss.pgh.pa.us 935 :CBC 594 : case CONSTR_FOREIGN:
4802 936 [ + + ]: 594 : if (cxt->isforeign)
937 [ + - ]: 4 : ereport(ERROR,
938 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
939 : : errmsg("foreign key constraints are not supported on foreign tables"),
940 : : parser_errposition(cxt->pstate,
941 : : constraint->location)));
942 : :
943 : : /*
944 : : * Fill in the current attribute's name and throw it into the
945 : : * list of FK constraints to be processed later.
946 : : */
6123 947 : 590 : constraint->fk_attrs = list_make1(makeString(column->colname));
948 : 590 : cxt->fkconstraints = lappend(cxt->fkconstraints, constraint);
949 : 590 : break;
950 : :
6891 951 : 152 : case CONSTR_ATTR_DEFERRABLE:
952 : : case CONSTR_ATTR_NOT_DEFERRABLE:
953 : : case CONSTR_ATTR_DEFERRED:
954 : : case CONSTR_ATTR_IMMEDIATE:
955 : : case CONSTR_ATTR_ENFORCED:
956 : : case CONSTR_ATTR_NOT_ENFORCED:
957 : : /* transformConstraintAttrs took care of these */
958 : 152 : break;
959 : :
6891 tgl@sss.pgh.pa.us 960 :UBC 0 : default:
961 [ # # ]: 0 : elog(ERROR, "unrecognized constraint type: %d",
962 : : constraint->contype);
963 : : break;
964 : : }
965 : :
3316 peter_e@gmx.net 966 [ + + + + ]:CBC 12392 : if (saw_default && saw_identity)
967 [ + - ]: 8 : ereport(ERROR,
968 : : (errcode(ERRCODE_SYNTAX_ERROR),
969 : : errmsg("both default and identity specified for column \"%s\" of table \"%s\"",
970 : : column->colname, cxt->relation->relname),
971 : : parser_errposition(cxt->pstate,
972 : : constraint->location)));
973 : :
2593 peter@eisentraut.org 974 [ + + + + ]: 12384 : if (saw_default && saw_generated)
975 [ + - ]: 8 : ereport(ERROR,
976 : : (errcode(ERRCODE_SYNTAX_ERROR),
977 : : errmsg("both default and generation expression specified for column \"%s\" of table \"%s\"",
978 : : column->colname, cxt->relation->relname),
979 : : parser_errposition(cxt->pstate,
980 : : constraint->location)));
981 : :
982 [ + + + + ]: 12376 : if (saw_identity && saw_generated)
983 [ + - ]: 8 : ereport(ERROR,
984 : : (errcode(ERRCODE_SYNTAX_ERROR),
985 : : errmsg("both identity and generation expression specified for column \"%s\" of table \"%s\"",
986 : : column->colname, cxt->relation->relname),
987 : : parser_errposition(cxt->pstate,
988 : : constraint->location)));
989 : : }
990 : :
991 : : /*
992 : : * If we need a not-null constraint for PRIMARY KEY, SERIAL or IDENTITY,
993 : : * and one was not explicitly specified, add one now.
994 : : */
543 alvherre@alvh.no-ip. 995 [ + + + + : 44762 : if (need_notnull && !(saw_nullable && column->is_not_null))
- + ]
996 : : {
997 : 3391 : column->is_not_null = true;
998 : 3391 : notnull_constraint = makeNotNullConstraint(makeString(column->colname));
999 : 3391 : cxt->nnconstraints = lappend(cxt->nnconstraints, notnull_constraint);
1000 : : }
1001 : :
1002 : : /*
1003 : : * If needed, generate ALTER FOREIGN TABLE ALTER COLUMN statement to add
1004 : : * per-column foreign data wrapper options to this column after creation.
1005 : : */
5387 rhaas@postgresql.org 1006 [ + + ]: 44762 : if (column->fdwoptions != NIL)
1007 : : {
1008 : : AlterTableStmt *stmt;
1009 : : AlterTableCmd *cmd;
1010 : :
1011 : 85 : cmd = makeNode(AlterTableCmd);
1012 : 85 : cmd->subtype = AT_AlterColumnGenericOptions;
1013 : 85 : cmd->name = column->colname;
1014 : 85 : cmd->def = (Node *) column->fdwoptions;
1015 : 85 : cmd->behavior = DROP_RESTRICT;
1016 : 85 : cmd->missing_ok = false;
1017 : :
1018 : 85 : stmt = makeNode(AlterTableStmt);
1019 : 85 : stmt->relation = cxt->relation;
1020 : 85 : stmt->cmds = NIL;
2124 michael@paquier.xyz 1021 : 85 : stmt->objtype = OBJECT_FOREIGN_TABLE;
5387 rhaas@postgresql.org 1022 : 85 : stmt->cmds = lappend(stmt->cmds, cmd);
1023 : :
1024 : 85 : cxt->alist = lappend(cxt->alist, stmt);
1025 : : }
6891 tgl@sss.pgh.pa.us 1026 : 44762 : }
1027 : :
1028 : : /*
1029 : : * transformTableConstraint
1030 : : * transform a Constraint node within CREATE TABLE or ALTER TABLE
1031 : : */
1032 : : static void
5579 1033 : 12765 : transformTableConstraint(CreateStmtContext *cxt, Constraint *constraint)
1034 : : {
6891 1035 [ + + + + : 12765 : switch (constraint->contype)
+ + - - ]
1036 : : {
1037 : 5249 : case CONSTR_PRIMARY:
4157 1038 [ + + ]: 5249 : if (cxt->isforeign)
1039 [ + - ]: 4 : ereport(ERROR,
1040 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1041 : : errmsg("primary key constraints are not supported on foreign tables"),
1042 : : parser_errposition(cxt->pstate,
1043 : : constraint->location)));
1044 : 5245 : cxt->ixconstraints = lappend(cxt->ixconstraints, constraint);
1045 : 5245 : break;
1046 : :
6891 1047 : 3337 : case CONSTR_UNIQUE:
4157 1048 [ + + ]: 3337 : if (cxt->isforeign)
1049 [ + - ]: 4 : ereport(ERROR,
1050 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1051 : : errmsg("unique constraints are not supported on foreign tables"),
1052 : : parser_errposition(cxt->pstate,
1053 : : constraint->location)));
1054 : 3333 : cxt->ixconstraints = lappend(cxt->ixconstraints, constraint);
1055 : 3333 : break;
1056 : :
5993 1057 : 153 : case CONSTR_EXCLUSION:
4157 1058 [ - + ]: 153 : if (cxt->isforeign)
4157 tgl@sss.pgh.pa.us 1059 [ # # ]:UBC 0 : ereport(ERROR,
1060 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1061 : : errmsg("exclusion constraints are not supported on foreign tables"),
1062 : : parser_errposition(cxt->pstate,
1063 : : constraint->location)));
6891 tgl@sss.pgh.pa.us 1064 :CBC 153 : cxt->ixconstraints = lappend(cxt->ixconstraints, constraint);
1065 : 153 : break;
1066 : :
1067 : 1037 : case CONSTR_CHECK:
1068 : 1037 : cxt->ckconstraints = lappend(cxt->ckconstraints, constraint);
1069 : 1037 : break;
1070 : :
543 alvherre@alvh.no-ip. 1071 : 752 : case CONSTR_NOTNULL:
1072 [ + + + + ]: 752 : if (cxt->ispartitioned && constraint->is_no_inherit)
1073 [ + - ]: 4 : ereport(ERROR,
1074 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1075 : : errmsg("not-null constraints on partitioned tables cannot be NO INHERIT"));
1076 : :
1077 : 748 : cxt->nnconstraints = lappend(cxt->nnconstraints, constraint);
1078 : 748 : break;
1079 : :
6123 tgl@sss.pgh.pa.us 1080 : 2237 : case CONSTR_FOREIGN:
4157 1081 [ - + ]: 2237 : if (cxt->isforeign)
4157 tgl@sss.pgh.pa.us 1082 [ # # ]:UBC 0 : ereport(ERROR,
1083 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1084 : : errmsg("foreign key constraints are not supported on foreign tables"),
1085 : : parser_errposition(cxt->pstate,
1086 : : constraint->location)));
6123 tgl@sss.pgh.pa.us 1087 :CBC 2237 : cxt->fkconstraints = lappend(cxt->fkconstraints, constraint);
1088 : 2237 : break;
1089 : :
6891 tgl@sss.pgh.pa.us 1090 :UBC 0 : case CONSTR_NULL:
1091 : : case CONSTR_DEFAULT:
1092 : : case CONSTR_ATTR_DEFERRABLE:
1093 : : case CONSTR_ATTR_NOT_DEFERRABLE:
1094 : : case CONSTR_ATTR_DEFERRED:
1095 : : case CONSTR_ATTR_IMMEDIATE:
1096 : : case CONSTR_ATTR_ENFORCED:
1097 : : case CONSTR_ATTR_NOT_ENFORCED:
1098 [ # # ]: 0 : elog(ERROR, "invalid context for constraint type %d",
1099 : : constraint->contype);
1100 : : break;
1101 : :
1102 : 0 : default:
1103 [ # # ]: 0 : elog(ERROR, "unrecognized constraint type: %d",
1104 : : constraint->contype);
1105 : : break;
1106 : : }
6891 tgl@sss.pgh.pa.us 1107 :CBC 12753 : }
1108 : :
1109 : : /*
1110 : : * transformTableLikeClause
1111 : : *
1112 : : * Change the LIKE <srctable> portion of a CREATE TABLE statement into
1113 : : * column definitions that recreate the user defined column portions of
1114 : : * <srctable>. Also, if there are any LIKE options that we can't fully
1115 : : * process at this point, add the TableLikeClause to cxt->likeclauses, which
1116 : : * will cause utility.c to call expandTableLikeClause() after the new
1117 : : * table has been created.
1118 : : *
1119 : : * Some options are ignored. For example, as foreign tables have no storage,
1120 : : * these INCLUDING options have no effect: STORAGE, COMPRESSION, IDENTITY
1121 : : * and INDEXES. Similarly, INCLUDING INDEXES is ignored from a view.
1122 : : */
1123 : : static void
5232 peter_e@gmx.net 1124 : 525 : transformTableLikeClause(CreateStmtContext *cxt, TableLikeClause *table_like_clause)
1125 : : {
1126 : : AttrNumber parent_attno;
1127 : : Relation relation;
1128 : : TupleDesc tupleDesc;
1129 : : AclResult aclresult;
1130 : : char *comment;
1131 : : ParseCallbackState pcbstate;
1132 : :
4802 tgl@sss.pgh.pa.us 1133 : 525 : setup_parser_errposition_callback(&pcbstate, cxt->pstate,
1134 : 525 : table_like_clause->relation->location);
1135 : :
1136 : : /* Open the relation referenced by the LIKE clause */
5176 peter_e@gmx.net 1137 : 525 : relation = relation_openrv(table_like_clause->relation, AccessShareLock);
1138 : :
5057 tgl@sss.pgh.pa.us 1139 [ + + ]: 521 : if (relation->rd_rel->relkind != RELKIND_RELATION &&
1140 [ + + ]: 254 : relation->rd_rel->relkind != RELKIND_VIEW &&
4811 kgrittn@postgresql.o 1141 [ + - ]: 246 : relation->rd_rel->relkind != RELKIND_MATVIEW &&
5057 tgl@sss.pgh.pa.us 1142 [ + + ]: 246 : relation->rd_rel->relkind != RELKIND_COMPOSITE_TYPE &&
3436 rhaas@postgresql.org 1143 [ + - ]: 242 : relation->rd_rel->relkind != RELKIND_FOREIGN_TABLE &&
1144 [ + + ]: 242 : relation->rd_rel->relkind != RELKIND_PARTITIONED_TABLE)
6891 tgl@sss.pgh.pa.us 1145 [ + - ]: 4 : ereport(ERROR,
1146 : : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
1147 : : errmsg("relation \"%s\" is invalid in LIKE clause",
1148 : : RelationGetRelationName(relation)),
1149 : : errdetail_relkind_not_supported(relation->rd_rel->relkind)));
1150 : :
5176 peter_e@gmx.net 1151 : 517 : cancel_parser_errposition_callback(&pcbstate);
1152 : :
1153 : : /*
1154 : : * Check for privileges
1155 : : */
1156 [ + + ]: 517 : if (relation->rd_rel->relkind == RELKIND_COMPOSITE_TYPE)
1157 : : {
1269 peter@eisentraut.org 1158 : 4 : aclresult = object_aclcheck(TypeRelationId, relation->rd_rel->reltype, GetUserId(),
1159 : : ACL_USAGE);
5176 peter_e@gmx.net 1160 [ - + ]: 4 : if (aclresult != ACLCHECK_OK)
3076 peter_e@gmx.net 1161 :UBC 0 : aclcheck_error(aclresult, OBJECT_TYPE,
5176 1162 : 0 : RelationGetRelationName(relation));
1163 : : }
1164 : : else
1165 : : {
5176 peter_e@gmx.net 1166 :CBC 513 : aclresult = pg_class_aclcheck(RelationGetRelid(relation), GetUserId(),
1167 : : ACL_SELECT);
1168 [ - + ]: 513 : if (aclresult != ACLCHECK_OK)
3076 peter_e@gmx.net 1169 :UBC 0 : aclcheck_error(aclresult, get_relkind_objtype(relation->rd_rel->relkind),
5176 1170 : 0 : RelationGetRelationName(relation));
1171 : : }
1172 : :
6891 tgl@sss.pgh.pa.us 1173 :CBC 517 : tupleDesc = RelationGetDescr(relation);
1174 : :
1175 : : /*
1176 : : * Insert the copied attributes into the cxt for the new table definition.
1177 : : * We must do this now so that they appear in the table in the relative
1178 : : * position where the LIKE clause is, as required by SQL99.
1179 : : */
1180 [ + + ]: 1683 : for (parent_attno = 1; parent_attno <= tupleDesc->natts;
1181 : 1166 : parent_attno++)
1182 : : {
3180 andres@anarazel.de 1183 : 1166 : Form_pg_attribute attribute = TupleDescAttr(tupleDesc,
1184 : : parent_attno - 1);
1185 : : ColumnDef *def;
1186 : :
1187 : : /*
1188 : : * Ignore dropped columns in the parent.
1189 : : */
6891 tgl@sss.pgh.pa.us 1190 [ + + ]: 1166 : if (attribute->attisdropped)
1191 : 24 : continue;
1192 : :
1193 : : /*
1194 : : * Create a new column definition
1195 : : */
980 peter@eisentraut.org 1196 : 1142 : def = makeColumnDef(NameStr(attribute->attname), attribute->atttypid,
1197 : : attribute->atttypmod, attribute->attcollation);
1198 : :
1199 : : /*
1200 : : * Add to column list
1201 : : */
6891 tgl@sss.pgh.pa.us 1202 : 1142 : cxt->columns = lappend(cxt->columns, def);
1203 : :
1204 : : /*
1205 : : * Although we don't transfer the column's default/generation
1206 : : * expression now, we need to mark it GENERATED if appropriate.
1207 : : */
2083 1208 [ + + + + ]: 1142 : if (attribute->atthasdef && attribute->attgenerated &&
1209 [ + + ]: 52 : (table_like_clause->options & CREATE_TABLE_LIKE_GENERATED))
2414 1210 : 32 : def->generated = attribute->attgenerated;
1211 : :
1212 : : /*
1213 : : * Copy identity if requested
1214 : : */
3316 peter_e@gmx.net 1215 [ + + ]: 1142 : if (attribute->attidentity &&
440 michael@paquier.xyz 1216 [ + + ]: 28 : (table_like_clause->options & CREATE_TABLE_LIKE_IDENTITY) &&
1217 [ + + ]: 12 : !cxt->isforeign)
1218 : : {
1219 : : Oid seq_relid;
1220 : : List *seq_options;
1221 : :
1222 : : /*
1223 : : * find sequence owned by old column; extract sequence parameters;
1224 : : * build new create sequence command
1225 : : */
728 peter@eisentraut.org 1226 : 8 : seq_relid = getIdentitySequence(relation, attribute->attnum, false);
3316 peter_e@gmx.net 1227 : 8 : seq_options = sequence_options(seq_relid);
1228 : 8 : generateSerialExtraStmts(cxt, def,
1229 : : InvalidOid, seq_options,
1230 : : true, false,
1231 : : NULL, NULL);
1232 : 8 : def->identity = attribute->attidentity;
1233 : : }
1234 : :
1235 : : /* Likewise, copy storage if requested */
440 michael@paquier.xyz 1236 [ + + ]: 1142 : if ((table_like_clause->options & CREATE_TABLE_LIKE_STORAGE) &&
1237 [ + + ]: 130 : !cxt->isforeign)
805 peter@eisentraut.org 1238 : 110 : def->storage = attribute->attstorage;
1239 : : else
1240 : 1032 : def->storage = 0;
1241 : :
1242 : : /* Likewise, copy compression if requested */
440 michael@paquier.xyz 1243 [ + + ]: 1142 : if ((table_like_clause->options & CREATE_TABLE_LIKE_COMPRESSION) != 0 &&
1244 [ + + ]: 100 : CompressionMethodIsValid(attribute->attcompression) &&
1245 [ + + ]: 8 : !cxt->isforeign)
805 peter@eisentraut.org 1246 : 4 : def->compression =
1247 : 4 : pstrdup(GetCompressionMethodName(attribute->attcompression));
1248 : : else
1873 rhaas@postgresql.org 1249 : 1138 : def->compression = NULL;
1250 : :
1251 : : /* Likewise, copy comment if requested */
5232 peter_e@gmx.net 1252 [ + + + + ]: 1278 : if ((table_like_clause->options & CREATE_TABLE_LIKE_COMMENTS) &&
6017 tgl@sss.pgh.pa.us 1253 : 136 : (comment = GetComment(attribute->attrelid,
1254 : : RelationRelationId,
1255 : 136 : attribute->attnum)) != NULL)
1256 : : {
6049 andrew@dunslane.net 1257 : 56 : CommentStmt *stmt = makeNode(CommentStmt);
1258 : :
1259 : 56 : stmt->objtype = OBJECT_COLUMN;
3461 peter_e@gmx.net 1260 : 56 : stmt->object = (Node *) list_make3(makeString(cxt->relation->schemaname),
1261 : : makeString(cxt->relation->relname),
1262 : : makeString(def->colname));
6049 andrew@dunslane.net 1263 : 56 : stmt->comment = comment;
1264 : :
1265 : 56 : cxt->alist = lappend(cxt->alist, stmt);
1266 : : }
1267 : : }
1268 : :
1269 : : /*
1270 : : * Reproduce not-null constraints, if any, by copying them. We do this
1271 : : * regardless of options given.
1272 : : */
543 alvherre@alvh.no-ip. 1273 [ + + + + ]: 517 : if (tupleDesc->constr && tupleDesc->constr->has_not_null)
1274 : : {
1275 : : List *lst;
1276 : :
1277 : 226 : lst = RelationGetNotNullConstraints(RelationGetRelid(relation), false,
1278 : : true);
1279 : 226 : cxt->nnconstraints = list_concat(cxt->nnconstraints, lst);
1280 : :
1281 : : /* Copy comments on not-null constraints */
313 fujii@postgresql.org 1282 [ + + ]: 226 : if (table_like_clause->options & CREATE_TABLE_LIKE_COMMENTS)
1283 : : {
1284 [ + - + + : 148 : foreach_node(Constraint, nnconstr, lst)
+ + ]
1285 : : {
1286 [ + + ]: 60 : if ((comment = GetComment(get_relation_constraint_oid(RelationGetRelid(relation),
1287 : 60 : nnconstr->conname, false),
1288 : : ConstraintRelationId,
1289 : : 0)) != NULL)
1290 : : {
1291 : 20 : CommentStmt *stmt = makeNode(CommentStmt);
1292 : :
1293 : 20 : stmt->objtype = OBJECT_TABCONSTRAINT;
1294 : 20 : stmt->object = (Node *) list_make3(makeString(cxt->relation->schemaname),
1295 : : makeString(cxt->relation->relname),
1296 : : makeString(nnconstr->conname));
1297 : 20 : stmt->comment = comment;
1298 : 20 : cxt->alist = lappend(cxt->alist, stmt);
1299 : : }
1300 : : }
1301 : : }
1302 : : }
1303 : :
1304 : : /*
1305 : : * We cannot yet deal with defaults, CHECK constraints, indexes, or
1306 : : * statistics, since we don't yet know what column numbers the copied
1307 : : * columns will have in the finished table. If any of those options are
1308 : : * specified, add the LIKE clause to cxt->likeclauses so that
1309 : : * expandTableLikeClause will be called after we do know that.
1310 : : *
1311 : : * In order for this to work, we remember the relation OID so that
1312 : : * expandTableLikeClause is certain to open the same table.
1313 : : */
722 alvherre@alvh.no-ip. 1314 [ + + ]: 517 : if (table_like_clause->options &
1315 : : (CREATE_TABLE_LIKE_DEFAULTS |
1316 : : CREATE_TABLE_LIKE_GENERATED |
1317 : : CREATE_TABLE_LIKE_CONSTRAINTS |
1318 : : CREATE_TABLE_LIKE_INDEXES |
1319 : : CREATE_TABLE_LIKE_STATISTICS))
1320 : : {
1981 tgl@sss.pgh.pa.us 1321 : 137 : table_like_clause->relationOid = RelationGetRelid(relation);
1993 1322 : 137 : cxt->likeclauses = lappend(cxt->likeclauses, table_like_clause);
1323 : : }
1324 : :
1325 : : /*
1326 : : * Close the parent rel, but keep our AccessShareLock on it until xact
1327 : : * commit. That will prevent someone else from deleting or ALTERing the
1328 : : * parent before we can run expandTableLikeClause.
1329 : : */
2083 1330 : 517 : table_close(relation, NoLock);
1331 : 517 : }
1332 : :
1333 : : /*
1334 : : * expandTableLikeClause
1335 : : *
1336 : : * Process LIKE options that require knowing the final column numbers
1337 : : * assigned to the new table's columns. This executes after we have
1338 : : * run DefineRelation for the new table. It returns a list of utility
1339 : : * commands that should be run to generate indexes etc.
1340 : : */
1341 : : List *
1342 : 137 : expandTableLikeClause(RangeVar *heapRel, TableLikeClause *table_like_clause)
1343 : : {
1344 : 137 : List *result = NIL;
1345 : 137 : List *atsubcmds = NIL;
1346 : : AttrNumber parent_attno;
1347 : : Relation relation;
1348 : : Relation childrel;
1349 : : TupleDesc tupleDesc;
1350 : : TupleConstr *constr;
1351 : : AttrMap *attmap;
1352 : : char *comment;
1353 : :
1354 : : /*
1355 : : * Open the relation referenced by the LIKE clause. We should still have
1356 : : * the table lock obtained by transformTableLikeClause (and this'll throw
1357 : : * an assertion failure if not). Hence, no need to recheck privileges
1358 : : * etc. We must open the rel by OID not name, to be sure we get the same
1359 : : * table.
1360 : : */
1981 1361 [ - + ]: 137 : if (!OidIsValid(table_like_clause->relationOid))
1981 tgl@sss.pgh.pa.us 1362 [ # # ]:UBC 0 : elog(ERROR, "expandTableLikeClause called on untransformed LIKE clause");
1363 : :
1981 tgl@sss.pgh.pa.us 1364 :CBC 137 : relation = relation_open(table_like_clause->relationOid, NoLock);
1365 : :
2083 1366 : 137 : tupleDesc = RelationGetDescr(relation);
1367 : 137 : constr = tupleDesc->constr;
1368 : :
1369 : : /*
1370 : : * Open the newly-created child relation; we have lock on that too.
1371 : : */
1372 : 137 : childrel = relation_openrv(heapRel, NoLock);
1373 : :
1374 : : /*
1375 : : * Construct a map from the LIKE relation's attnos to the child rel's.
1376 : : * This re-checks type match etc, although it shouldn't be possible to
1377 : : * have a failure since both tables are locked.
1378 : : */
1379 : 137 : attmap = build_attrmap_by_name(RelationGetDescr(childrel),
1380 : : tupleDesc,
1381 : : false);
1382 : :
1383 : : /*
1384 : : * Process defaults, if required.
1385 : : */
1386 [ + + ]: 137 : if ((table_like_clause->options &
1387 [ + + ]: 69 : (CREATE_TABLE_LIKE_DEFAULTS | CREATE_TABLE_LIKE_GENERATED)) &&
1388 : : constr != NULL)
1389 : : {
1390 [ + + ]: 231 : for (parent_attno = 1; parent_attno <= tupleDesc->natts;
1391 : 170 : parent_attno++)
1392 : : {
1393 : 170 : Form_pg_attribute attribute = TupleDescAttr(tupleDesc,
1394 : : parent_attno - 1);
1395 : :
1396 : : /*
1397 : : * Ignore dropped columns in the parent.
1398 : : */
1399 [ + + ]: 170 : if (attribute->attisdropped)
1400 : 8 : continue;
1401 : :
1402 : : /*
1403 : : * Copy default, if present and it should be copied. We have
1404 : : * separate options for plain default expressions and GENERATED
1405 : : * defaults.
1406 : : */
1407 [ + + + + ]: 223 : if (attribute->atthasdef &&
1408 [ + + ]: 61 : (attribute->attgenerated ?
1409 : 36 : (table_like_clause->options & CREATE_TABLE_LIKE_GENERATED) :
1410 : 25 : (table_like_clause->options & CREATE_TABLE_LIKE_DEFAULTS)))
1411 : : {
1412 : : Node *this_default;
1413 : : AlterTableCmd *atsubcmd;
1414 : : bool found_whole_row;
1415 : :
951 peter@eisentraut.org 1416 : 53 : this_default = TupleDescGetDefault(tupleDesc, parent_attno);
1855 tgl@sss.pgh.pa.us 1417 [ - + ]: 53 : if (this_default == NULL)
1855 tgl@sss.pgh.pa.us 1418 [ # # ]:UBC 0 : elog(ERROR, "default expression not found for attribute %d of relation \"%s\"",
1419 : : parent_attno, RelationGetRelationName(relation));
1420 : :
2083 tgl@sss.pgh.pa.us 1421 :CBC 53 : atsubcmd = makeNode(AlterTableCmd);
1422 : 53 : atsubcmd->subtype = AT_CookedColumnDefault;
1423 : 53 : atsubcmd->num = attmap->attnums[parent_attno - 1];
1424 : 53 : atsubcmd->def = map_variable_attnos(this_default,
1425 : : 1, 0,
1426 : : attmap,
1427 : : InvalidOid,
1428 : : &found_whole_row);
1429 : :
1430 : : /*
1431 : : * Prevent this for the same reason as for constraints below.
1432 : : * Note that defaults cannot contain any vars, so it's OK that
1433 : : * the error message refers to generated columns.
1434 : : */
1435 [ - + ]: 53 : if (found_whole_row)
2083 tgl@sss.pgh.pa.us 1436 [ # # ]:UBC 0 : ereport(ERROR,
1437 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1438 : : errmsg("cannot convert whole-row table reference"),
1439 : : errdetail("Generation expression for column \"%s\" contains a whole-row reference to table \"%s\".",
1440 : : NameStr(attribute->attname),
1441 : : RelationGetRelationName(relation))));
1442 : :
2083 tgl@sss.pgh.pa.us 1443 :CBC 53 : atsubcmds = lappend(atsubcmds, atsubcmd);
1444 : : }
1445 : : }
1446 : : }
1447 : :
1448 : : /*
1449 : : * Copy CHECK constraints if requested, being careful to adjust attribute
1450 : : * numbers so they match the child.
1451 : : */
5232 peter_e@gmx.net 1452 [ + + + + ]: 137 : if ((table_like_clause->options & CREATE_TABLE_LIKE_CONSTRAINTS) &&
1453 : : constr != NULL)
1454 : : {
1455 : : int ccnum;
1456 : :
2247 tgl@sss.pgh.pa.us 1457 [ + + ]: 168 : for (ccnum = 0; ccnum < constr->num_check; ccnum++)
1458 : : {
1459 : 100 : char *ccname = constr->check[ccnum].ccname;
1460 : 100 : char *ccbin = constr->check[ccnum].ccbin;
479 peter@eisentraut.org 1461 : 100 : bool ccenforced = constr->check[ccnum].ccenforced;
2247 tgl@sss.pgh.pa.us 1462 : 100 : bool ccnoinherit = constr->check[ccnum].ccnoinherit;
1463 : : Node *ccbin_node;
1464 : : bool found_whole_row;
1465 : : Constraint *n;
1466 : : AlterTableCmd *atsubcmd;
1467 : :
5057 1468 : 100 : ccbin_node = map_variable_attnos(stringToNode(ccbin),
1469 : : 1, 0,
1470 : : attmap,
1471 : : InvalidOid, &found_whole_row);
1472 : :
1473 : : /*
1474 : : * We reject whole-row variables because the whole point of LIKE
1475 : : * is that the new table's rowtype might later diverge from the
1476 : : * parent's. So, while translation might be possible right now,
1477 : : * it wouldn't be possible to guarantee it would work in future.
1478 : : */
1479 [ - + ]: 100 : if (found_whole_row)
5057 tgl@sss.pgh.pa.us 1480 [ # # ]:UBC 0 : ereport(ERROR,
1481 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1482 : : errmsg("cannot convert whole-row table reference"),
1483 : : errdetail("Constraint \"%s\" contains a whole-row reference to table \"%s\".",
1484 : : ccname,
1485 : : RelationGetRelationName(relation))));
1486 : :
2083 tgl@sss.pgh.pa.us 1487 :CBC 100 : n = makeNode(Constraint);
6891 1488 : 100 : n->contype = CONSTR_CHECK;
6123 1489 : 100 : n->conname = pstrdup(ccname);
2247 1490 : 100 : n->location = -1;
479 peter@eisentraut.org 1491 : 100 : n->is_enforced = ccenforced;
237 1492 : 100 : n->initially_valid = ccenforced; /* sic */
2247 tgl@sss.pgh.pa.us 1493 : 100 : n->is_no_inherit = ccnoinherit;
6891 1494 : 100 : n->raw_expr = NULL;
1495 : 100 : n->cooked_expr = nodeToString(ccbin_node);
1496 : :
1497 : : /* We can skip validation, since the new table should be empty. */
2083 1498 : 100 : n->skip_validation = true;
1499 : :
1500 : 100 : atsubcmd = makeNode(AlterTableCmd);
1501 : 100 : atsubcmd->subtype = AT_AddConstraint;
1502 : 100 : atsubcmd->def = (Node *) n;
1503 : 100 : atsubcmds = lappend(atsubcmds, atsubcmd);
1504 : :
1505 : : /* Copy comment on constraint */
5232 peter_e@gmx.net 1506 [ + + + + ]: 176 : if ((table_like_clause->options & CREATE_TABLE_LIKE_COMMENTS) &&
5145 1507 : 76 : (comment = GetComment(get_relation_constraint_oid(RelationGetRelid(relation),
3240 tgl@sss.pgh.pa.us 1508 : 76 : n->conname, false),
1509 : : ConstraintRelationId,
1510 : : 0)) != NULL)
1511 : : {
6049 andrew@dunslane.net 1512 : 20 : CommentStmt *stmt = makeNode(CommentStmt);
1513 : :
4151 alvherre@alvh.no-ip. 1514 : 20 : stmt->objtype = OBJECT_TABCONSTRAINT;
2083 tgl@sss.pgh.pa.us 1515 : 20 : stmt->object = (Node *) list_make3(makeString(heapRel->schemaname),
1516 : : makeString(heapRel->relname),
1517 : : makeString(n->conname));
6049 andrew@dunslane.net 1518 : 20 : stmt->comment = comment;
1519 : :
2083 tgl@sss.pgh.pa.us 1520 : 20 : result = lappend(result, stmt);
1521 : : }
1522 : : }
1523 : : }
1524 : :
1525 : : /*
1526 : : * If we generated any ALTER TABLE actions above, wrap them into a single
1527 : : * ALTER TABLE command. Stick it at the front of the result, so it runs
1528 : : * before any CommentStmts we made above.
1529 : : */
1530 [ + + ]: 137 : if (atsubcmds)
1531 : : {
1532 : 89 : AlterTableStmt *atcmd = makeNode(AlterTableStmt);
1533 : :
1534 : 89 : atcmd->relation = copyObject(heapRel);
1535 : 89 : atcmd->cmds = atsubcmds;
1536 : 89 : atcmd->objtype = OBJECT_TABLE;
1537 : 89 : atcmd->missing_ok = false;
1538 : 89 : result = lcons(atcmd, result);
1539 : : }
1540 : :
1541 : : /*
1542 : : * Process indexes if required.
1543 : : */
5232 peter_e@gmx.net 1544 [ + + ]: 137 : if ((table_like_clause->options & CREATE_TABLE_LIKE_INDEXES) &&
440 michael@paquier.xyz 1545 [ + + ]: 73 : relation->rd_rel->relhasindex &&
1546 [ + + ]: 57 : childrel->rd_rel->relkind != RELKIND_FOREIGN_TABLE)
1547 : : {
1548 : : List *parent_indexes;
1549 : : ListCell *l;
1550 : :
6867 neilc@samurai.com 1551 : 53 : parent_indexes = RelationGetIndexList(relation);
1552 : :
1553 [ + - + + : 139 : foreach(l, parent_indexes)
+ + ]
1554 : : {
6746 bruce@momjian.us 1555 : 86 : Oid parent_index_oid = lfirst_oid(l);
1556 : : Relation parent_index;
1557 : : IndexStmt *index_stmt;
1558 : :
6867 neilc@samurai.com 1559 : 86 : parent_index = index_open(parent_index_oid, AccessShareLock);
1560 : :
1561 : : /* Build CREATE INDEX statement to recreate the parent_index */
2083 tgl@sss.pgh.pa.us 1562 : 86 : index_stmt = generateClonedIndexStmt(heapRel,
1563 : : parent_index,
1564 : : attmap,
1565 : : NULL);
1566 : :
1567 : : /* Copy comment on index, if requested */
5232 peter_e@gmx.net 1568 [ + + ]: 86 : if (table_like_clause->options & CREATE_TABLE_LIKE_COMMENTS)
1569 : : {
6049 andrew@dunslane.net 1570 : 48 : comment = GetComment(parent_index_oid, RelationRelationId, 0);
1571 : :
1572 : : /*
1573 : : * We make use of IndexStmt's idxcomment option, so as not to
1574 : : * need to know now what name the index will have.
1575 : : */
5041 tgl@sss.pgh.pa.us 1576 : 48 : index_stmt->idxcomment = comment;
1577 : : }
1578 : :
2083 1579 : 86 : result = lappend(result, index_stmt);
1580 : :
6730 1581 : 86 : index_close(parent_index, AccessShareLock);
1582 : : }
1583 : : }
1584 : :
1585 : : /*
1586 : : * Process extended statistics if required.
1587 : : */
713 1588 [ + + ]: 137 : if (table_like_clause->options & CREATE_TABLE_LIKE_STATISTICS)
1589 : : {
1590 : : List *parent_extstats;
1591 : : ListCell *l;
1592 : :
1593 : 48 : parent_extstats = RelationGetStatExtList(relation);
1594 : :
1595 [ + + + + : 88 : foreach(l, parent_extstats)
+ + ]
1596 : : {
1597 : 40 : Oid parent_stat_oid = lfirst_oid(l);
1598 : : CreateStatsStmt *stats_stmt;
1599 : :
1600 : 40 : stats_stmt = generateClonedExtStatsStmt(heapRel,
1601 : : RelationGetRelid(childrel),
1602 : : parent_stat_oid,
1603 : : attmap);
1604 : :
1605 : : /* Copy comment on statistics object, if requested */
1606 [ + + ]: 40 : if (table_like_clause->options & CREATE_TABLE_LIKE_COMMENTS)
1607 : : {
1608 : 32 : comment = GetComment(parent_stat_oid, StatisticExtRelationId, 0);
1609 : :
1610 : : /*
1611 : : * We make use of CreateStatsStmt's stxcomment option, so as
1612 : : * not to need to know now what name the statistics will have.
1613 : : */
1614 : 32 : stats_stmt->stxcomment = comment;
1615 : : }
1616 : :
1617 : 40 : result = lappend(result, stats_stmt);
1618 : : }
1619 : :
1620 : 48 : list_free(parent_extstats);
1621 : : }
1622 : :
1623 : : /* Done with child rel */
2083 1624 : 137 : table_close(childrel, NoLock);
1625 : :
1626 : : /*
1627 : : * Close the parent rel, but keep our AccessShareLock on it until xact
1628 : : * commit. That will prevent someone else from deleting or ALTERing the
1629 : : * parent before the child is committed.
1630 : : */
2661 andres@anarazel.de 1631 : 137 : table_close(relation, NoLock);
1632 : :
2083 tgl@sss.pgh.pa.us 1633 : 137 : return result;
1634 : : }
1635 : :
1636 : : static void
5579 1637 : 81 : transformOfType(CreateStmtContext *cxt, TypeName *ofTypename)
1638 : : {
1639 : : HeapTuple tuple;
1640 : : TupleDesc tupdesc;
1641 : : int i;
1642 : : Oid ofTypeId;
1643 : :
1285 peter@eisentraut.org 1644 [ - + ]: 81 : Assert(ofTypename);
1645 : :
504 michael@paquier.xyz 1646 : 81 : tuple = typenameType(cxt->pstate, ofTypename, NULL);
5494 rhaas@postgresql.org 1647 : 77 : check_of_type(tuple);
2723 andres@anarazel.de 1648 : 69 : ofTypeId = ((Form_pg_type) GETSTRUCT(tuple))->oid;
3240 tgl@sss.pgh.pa.us 1649 : 69 : ofTypename->typeOid = ofTypeId; /* cached for later */
1650 : :
5941 peter_e@gmx.net 1651 : 69 : tupdesc = lookup_rowtype_tupdesc(ofTypeId, -1);
1652 [ + + ]: 207 : for (i = 0; i < tupdesc->natts; i++)
1653 : : {
3180 andres@anarazel.de 1654 : 138 : Form_pg_attribute attr = TupleDescAttr(tupdesc, i);
1655 : : ColumnDef *n;
1656 : :
5700 peter_e@gmx.net 1657 [ - + ]: 138 : if (attr->attisdropped)
5700 peter_e@gmx.net 1658 :UBC 0 : continue;
1659 : :
980 peter@eisentraut.org 1660 :CBC 138 : n = makeColumnDef(NameStr(attr->attname), attr->atttypid,
1661 : : attr->atttypmod, attr->attcollation);
5941 peter_e@gmx.net 1662 : 138 : n->is_from_type = true;
1663 : :
1664 : 138 : cxt->columns = lappend(cxt->columns, n);
1665 : : }
1602 tgl@sss.pgh.pa.us 1666 [ + - ]: 69 : ReleaseTupleDesc(tupdesc);
1667 : :
5941 peter_e@gmx.net 1668 : 69 : ReleaseSysCache(tuple);
1669 : 69 : }
1670 : :
1671 : : /*
1672 : : * Generate an IndexStmt node using information from an already existing index
1673 : : * "source_idx".
1674 : : *
1675 : : * heapRel is stored into the IndexStmt's relation field, but we don't use it
1676 : : * otherwise; some callers pass NULL, if they don't need it to be valid.
1677 : : * (The target relation might not exist yet, so we mustn't try to access it.)
1678 : : *
1679 : : * Attribute numbers in expression Vars are adjusted according to attmap.
1680 : : *
1681 : : * If constraintOid isn't NULL, we store the OID of any constraint associated
1682 : : * with the index there.
1683 : : *
1684 : : * Unlike transformIndexConstraint, we don't make any effort to force primary
1685 : : * key columns to be not-null. The larger cloning process this is part of
1686 : : * should have cloned their not-null status separately (and DefineIndex will
1687 : : * complain if that fails to happen).
1688 : : */
1689 : : IndexStmt *
2569 tgl@sss.pgh.pa.us 1690 : 2176 : generateClonedIndexStmt(RangeVar *heapRel, Relation source_idx,
1691 : : const AttrMap *attmap,
1692 : : Oid *constraintOid)
1693 : : {
6730 1694 : 2176 : Oid source_relid = RelationGetRelid(source_idx);
1695 : : HeapTuple ht_idxrel;
1696 : : HeapTuple ht_idx;
1697 : : HeapTuple ht_am;
1698 : : Form_pg_class idxrelrec;
1699 : : Form_pg_index idxrec;
1700 : : Form_pg_am amrec;
1701 : : oidvector *indcollation;
1702 : : oidvector *indclass;
1703 : : IndexStmt *index;
1704 : : List *indexprs;
1705 : : ListCell *indexpr_item;
1706 : : Oid indrelid;
1707 : : int keyno;
1708 : : Oid keycoltype;
1709 : : Datum datum;
1710 : : bool isnull;
1711 : :
2569 1712 [ + + ]: 2176 : if (constraintOid)
1713 : 1417 : *constraintOid = InvalidOid;
1714 : :
1715 : : /*
1716 : : * Fetch pg_class tuple of source index. We can't use the copy in the
1717 : : * relcache entry because it doesn't include optional fields.
1718 : : */
5924 rhaas@postgresql.org 1719 : 2176 : ht_idxrel = SearchSysCache1(RELOID, ObjectIdGetDatum(source_relid));
6730 tgl@sss.pgh.pa.us 1720 [ - + ]: 2176 : if (!HeapTupleIsValid(ht_idxrel))
6730 tgl@sss.pgh.pa.us 1721 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for relation %u", source_relid);
6730 tgl@sss.pgh.pa.us 1722 :CBC 2176 : idxrelrec = (Form_pg_class) GETSTRUCT(ht_idxrel);
1723 : :
1724 : : /* Fetch pg_index tuple for source index from relcache entry */
1725 : 2176 : ht_idx = source_idx->rd_indextuple;
6867 neilc@samurai.com 1726 : 2176 : idxrec = (Form_pg_index) GETSTRUCT(ht_idx);
1727 : 2176 : indrelid = idxrec->indrelid;
1728 : :
1729 : : /* Fetch the pg_am tuple of the index' access method */
3761 tgl@sss.pgh.pa.us 1730 : 2176 : ht_am = SearchSysCache1(AMOID, ObjectIdGetDatum(idxrelrec->relam));
1731 [ - + ]: 2176 : if (!HeapTupleIsValid(ht_am))
3761 tgl@sss.pgh.pa.us 1732 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for access method %u",
1733 : : idxrelrec->relam);
3761 tgl@sss.pgh.pa.us 1734 :CBC 2176 : amrec = (Form_pg_am) GETSTRUCT(ht_am);
1735 : :
1736 : : /* Extract indcollation from the pg_index tuple */
1137 dgustafsson@postgres 1737 : 2176 : datum = SysCacheGetAttrNotNull(INDEXRELID, ht_idx,
1738 : : Anum_pg_index_indcollation);
5519 tgl@sss.pgh.pa.us 1739 : 2176 : indcollation = (oidvector *) DatumGetPointer(datum);
1740 : :
1741 : : /* Extract indclass from the pg_index tuple */
1137 dgustafsson@postgres 1742 : 2176 : datum = SysCacheGetAttrNotNull(INDEXRELID, ht_idx, Anum_pg_index_indclass);
6730 tgl@sss.pgh.pa.us 1743 : 2176 : indclass = (oidvector *) DatumGetPointer(datum);
1744 : :
1745 : : /* Begin building the IndexStmt */
6867 neilc@samurai.com 1746 : 2176 : index = makeNode(IndexStmt);
3028 alvherre@alvh.no-ip. 1747 : 2176 : index->relation = heapRel;
6730 tgl@sss.pgh.pa.us 1748 : 2176 : index->accessMethod = pstrdup(NameStr(amrec->amname));
6662 1749 [ + + ]: 2176 : if (OidIsValid(idxrelrec->reltablespace))
1750 : 58 : index->tableSpace = get_tablespace_name(idxrelrec->reltablespace);
1751 : : else
1752 : 2118 : index->tableSpace = NULL;
5041 1753 : 2176 : index->excludeOpNames = NIL;
1754 : 2176 : index->idxcomment = NULL;
5579 1755 : 2176 : index->indexOid = InvalidOid;
1399 rhaas@postgresql.org 1756 : 2176 : index->oldNumber = InvalidRelFileNumber;
2222 noah@leadboat.com 1757 : 2176 : index->oldCreateSubid = InvalidSubTransactionId;
1399 rhaas@postgresql.org 1758 : 2176 : index->oldFirstRelfilelocatorSubid = InvalidSubTransactionId;
6867 neilc@samurai.com 1759 : 2176 : index->unique = idxrec->indisunique;
1552 peter@eisentraut.org 1760 : 2176 : index->nulls_not_distinct = idxrec->indnullsnotdistinct;
6867 neilc@samurai.com 1761 : 2176 : index->primary = idxrec->indisprimary;
595 peter@eisentraut.org 1762 [ + + + + : 2176 : index->iswithoutoverlaps = (idxrec->indisprimary || idxrec->indisunique) && idxrec->indisexclusion;
+ + ]
4090 tgl@sss.pgh.pa.us 1763 : 2176 : index->transformed = true; /* don't need transformIndexStmt */
6730 1764 : 2176 : index->concurrent = false;
4090 1765 : 2176 : index->if_not_exists = false;
2567 alvherre@alvh.no-ip. 1766 : 2176 : index->reset_default_tblspc = false;
1767 : :
1768 : : /*
1769 : : * We don't try to preserve the name of the source index; instead, just
1770 : : * let DefineIndex() choose a reasonable name. (If we tried to preserve
1771 : : * the name, we'd get duplicate-relation-name failures unless the source
1772 : : * table was in a different schema.)
1773 : : */
6867 neilc@samurai.com 1774 : 2176 : index->idxname = NULL;
1775 : :
1776 : : /*
1777 : : * If the index is marked PRIMARY or has an exclusion condition, it's
1778 : : * certainly from a constraint; else, if it's not marked UNIQUE, it
1779 : : * certainly isn't. If it is or might be from a constraint, we have to
1780 : : * fetch the pg_constraint record.
1781 : : */
5579 tgl@sss.pgh.pa.us 1782 [ + + + + : 2176 : if (index->primary || index->unique || idxrec->indisexclusion)
+ + ]
6124 1783 : 1160 : {
5912 bruce@momjian.us 1784 : 1160 : Oid constraintId = get_index_constraint(source_relid);
1785 : :
6124 tgl@sss.pgh.pa.us 1786 [ + + ]: 1160 : if (OidIsValid(constraintId))
1787 : : {
1788 : : HeapTuple ht_constr;
1789 : : Form_pg_constraint conrec;
1790 : :
2997 alvherre@alvh.no-ip. 1791 [ + + ]: 1127 : if (constraintOid)
1792 : 1009 : *constraintOid = constraintId;
1793 : :
5924 rhaas@postgresql.org 1794 : 1127 : ht_constr = SearchSysCache1(CONSTROID,
1795 : : ObjectIdGetDatum(constraintId));
6124 tgl@sss.pgh.pa.us 1796 [ - + ]: 1127 : if (!HeapTupleIsValid(ht_constr))
6124 tgl@sss.pgh.pa.us 1797 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for constraint %u",
1798 : : constraintId);
6124 tgl@sss.pgh.pa.us 1799 :CBC 1127 : conrec = (Form_pg_constraint) GETSTRUCT(ht_constr);
1800 : :
1801 : 1127 : index->isconstraint = true;
1802 : 1127 : index->deferrable = conrec->condeferrable;
1803 : 1127 : index->initdeferred = conrec->condeferred;
1804 : :
1805 : : /* If it's an exclusion constraint, we need the operator names */
5579 1806 [ + + ]: 1127 : if (idxrec->indisexclusion)
1807 : : {
1808 : : Datum *elems;
1809 : : int nElems;
1810 : : int i;
1811 : :
595 peter@eisentraut.org 1812 [ + + + - : 75 : Assert(conrec->contype == CONSTRAINT_EXCLUSION ||
+ + - + ]
1813 : : (index->iswithoutoverlaps &&
1814 : : (conrec->contype == CONSTRAINT_PRIMARY || conrec->contype == CONSTRAINT_UNIQUE)));
1815 : : /* Extract operator OIDs from the pg_constraint tuple */
1137 dgustafsson@postgres 1816 : 75 : datum = SysCacheGetAttrNotNull(CONSTROID, ht_constr,
1817 : : Anum_pg_constraint_conexclop);
1404 peter@eisentraut.org 1818 : 75 : deconstruct_array_builtin(DatumGetArrayTypeP(datum), OIDOID, &elems, NULL, &nElems);
1819 : :
5993 tgl@sss.pgh.pa.us 1820 [ + + ]: 224 : for (i = 0; i < nElems; i++)
1821 : : {
1822 : 149 : Oid operid = DatumGetObjectId(elems[i]);
1823 : : HeapTuple opertup;
1824 : : Form_pg_operator operform;
1825 : : char *oprname;
1826 : : char *nspname;
1827 : : List *namelist;
1828 : :
5924 rhaas@postgresql.org 1829 : 149 : opertup = SearchSysCache1(OPEROID,
1830 : : ObjectIdGetDatum(operid));
5993 tgl@sss.pgh.pa.us 1831 [ - + ]: 149 : if (!HeapTupleIsValid(opertup))
5993 tgl@sss.pgh.pa.us 1832 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for operator %u",
1833 : : operid);
5993 tgl@sss.pgh.pa.us 1834 :CBC 149 : operform = (Form_pg_operator) GETSTRUCT(opertup);
1835 : 149 : oprname = pstrdup(NameStr(operform->oprname));
1836 : : /* For simplicity we always schema-qualify the op name */
1837 : 149 : nspname = get_namespace_name(operform->oprnamespace);
1838 : 149 : namelist = list_make2(makeString(nspname),
1839 : : makeString(oprname));
1840 : 149 : index->excludeOpNames = lappend(index->excludeOpNames,
1841 : : namelist);
1842 : 149 : ReleaseSysCache(opertup);
1843 : : }
1844 : : }
1845 : :
6124 1846 : 1127 : ReleaseSysCache(ht_constr);
1847 : : }
1848 : : else
1849 : 33 : index->isconstraint = false;
1850 : : }
1851 : : else
1852 : 1016 : index->isconstraint = false;
1853 : :
1854 : : /* Get the index expressions, if any */
6730 1855 : 2176 : datum = SysCacheGetAttr(INDEXRELID, ht_idx,
1856 : : Anum_pg_index_indexprs, &isnull);
1857 [ + + ]: 2176 : if (!isnull)
1858 : : {
1859 : : char *exprsString;
1860 : :
6615 1861 : 122 : exprsString = TextDatumGetCString(datum);
6867 neilc@samurai.com 1862 : 122 : indexprs = (List *) stringToNode(exprsString);
1863 : : }
1864 : : else
6730 tgl@sss.pgh.pa.us 1865 : 2054 : indexprs = NIL;
1866 : :
1867 : : /* Build the list of IndexElem */
1868 : 2176 : index->indexParams = NIL;
2950 teodor@sigaev.ru 1869 : 2176 : index->indexIncludingParams = NIL;
1870 : :
6730 tgl@sss.pgh.pa.us 1871 : 2176 : indexpr_item = list_head(indexprs);
2950 teodor@sigaev.ru 1872 [ + + ]: 4727 : for (keyno = 0; keyno < idxrec->indnkeyatts; keyno++)
1873 : : {
1874 : : IndexElem *iparam;
6867 neilc@samurai.com 1875 : 2551 : AttrNumber attnum = idxrec->indkey.values[keyno];
3180 andres@anarazel.de 1876 : 2551 : Form_pg_attribute attr = TupleDescAttr(RelationGetDescr(source_idx),
1877 : : keyno);
6730 tgl@sss.pgh.pa.us 1878 : 2551 : int16 opt = source_idx->rd_indoption[keyno];
1879 : :
6867 neilc@samurai.com 1880 : 2551 : iparam = makeNode(IndexElem);
1881 : :
1882 [ + + ]: 2551 : if (AttributeNumberIsValid(attnum))
1883 : : {
1884 : : /* Simple index column */
1885 : : char *attname;
1886 : :
3004 alvherre@alvh.no-ip. 1887 : 2429 : attname = get_attname(indrelid, attnum, false);
6867 neilc@samurai.com 1888 : 2429 : keycoltype = get_atttype(indrelid, attnum);
1889 : :
1890 : 2429 : iparam->name = attname;
1891 : 2429 : iparam->expr = NULL;
1892 : : }
1893 : : else
1894 : : {
1895 : : /* Expressional index */
1896 : : Node *indexkey;
1897 : : bool found_whole_row;
1898 : :
1899 [ - + ]: 122 : if (indexpr_item == NULL)
6867 neilc@samurai.com 1900 [ # # ]:UBC 0 : elog(ERROR, "too few entries in indexprs list");
6867 neilc@samurai.com 1901 :CBC 122 : indexkey = (Node *) lfirst(indexpr_item);
2486 tgl@sss.pgh.pa.us 1902 : 122 : indexpr_item = lnext(indexprs, indexpr_item);
1903 : :
1904 : : /* Adjust Vars to match new table's column numbering */
5057 1905 : 122 : indexkey = map_variable_attnos(indexkey,
1906 : : 1, 0,
1907 : : attmap,
1908 : : InvalidOid, &found_whole_row);
1909 : :
1910 : : /* As in expandTableLikeClause, reject whole-row variables */
1911 [ - + ]: 122 : if (found_whole_row)
5057 tgl@sss.pgh.pa.us 1912 [ # # ]:UBC 0 : ereport(ERROR,
1913 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1914 : : errmsg("cannot convert whole-row table reference"),
1915 : : errdetail("Index \"%s\" contains a whole-row table reference.",
1916 : : RelationGetRelationName(source_idx))));
1917 : :
6867 neilc@samurai.com 1918 :CBC 122 : iparam->name = NULL;
1919 : 122 : iparam->expr = indexkey;
1920 : :
1921 : 122 : keycoltype = exprType(indexkey);
1922 : : }
1923 : :
1924 : : /* Copy the original index column name */
3180 andres@anarazel.de 1925 : 2551 : iparam->indexcolname = pstrdup(NameStr(attr->attname));
1926 : :
1927 : : /* Add the collation name, if non-default */
5519 tgl@sss.pgh.pa.us 1928 : 2551 : iparam->collation = get_collation(indcollation->values[keyno], keycoltype);
1929 : :
1930 : : /* Add the operator class name, if non-default */
6867 neilc@samurai.com 1931 : 2551 : iparam->opclass = get_opclass(indclass->values[keyno], keycoltype);
2227 akorotkov@postgresql 1932 : 2551 : iparam->opclassopts =
1933 : 2551 : untransformRelOptions(get_attoptions(source_relid, keyno + 1));
1934 : :
6867 neilc@samurai.com 1935 : 2551 : iparam->ordering = SORTBY_DEFAULT;
1936 : 2551 : iparam->nulls_ordering = SORTBY_NULLS_DEFAULT;
1937 : :
1938 : : /* Adjust options if necessary */
2661 andres@anarazel.de 1939 [ + + ]: 2551 : if (source_idx->rd_indam->amcanorder)
1940 : : {
1941 : : /*
1942 : : * If it supports sort ordering, copy DESC and NULLS opts. Don't
1943 : : * set non-default settings unnecessarily, though, so as to
1944 : : * improve the chance of recognizing equivalence to constraint
1945 : : * indexes.
1946 : : */
6867 neilc@samurai.com 1947 [ - + ]: 2385 : if (opt & INDOPTION_DESC)
1948 : : {
6867 neilc@samurai.com 1949 :UBC 0 : iparam->ordering = SORTBY_DESC;
6730 tgl@sss.pgh.pa.us 1950 [ # # ]: 0 : if ((opt & INDOPTION_NULLS_FIRST) == 0)
1951 : 0 : iparam->nulls_ordering = SORTBY_NULLS_LAST;
1952 : : }
1953 : : else
1954 : : {
6730 tgl@sss.pgh.pa.us 1955 [ - + ]:CBC 2385 : if (opt & INDOPTION_NULLS_FIRST)
6730 tgl@sss.pgh.pa.us 1956 :UBC 0 : iparam->nulls_ordering = SORTBY_NULLS_FIRST;
1957 : : }
1958 : : }
1959 : :
121 tgl@sss.pgh.pa.us 1960 :GNC 2551 : iparam->location = -1;
1961 : :
6867 neilc@samurai.com 1962 :CBC 2551 : index->indexParams = lappend(index->indexParams, iparam);
1963 : : }
1964 : :
1965 : : /* Handle included columns separately */
2950 teodor@sigaev.ru 1966 [ + + ]: 2188 : for (keyno = idxrec->indnkeyatts; keyno < idxrec->indnatts; keyno++)
1967 : : {
1968 : : IndexElem *iparam;
1969 : 12 : AttrNumber attnum = idxrec->indkey.values[keyno];
1970 : 12 : Form_pg_attribute attr = TupleDescAttr(RelationGetDescr(source_idx),
1971 : : keyno);
1972 : :
1973 : 12 : iparam = makeNode(IndexElem);
1974 : :
1975 [ + - ]: 12 : if (AttributeNumberIsValid(attnum))
1976 : : {
1977 : : /* Simple index column */
1978 : : char *attname;
1979 : :
1980 : 12 : attname = get_attname(indrelid, attnum, false);
1981 : :
1982 : 12 : iparam->name = attname;
1983 : 12 : iparam->expr = NULL;
1984 : : }
1985 : : else
2950 teodor@sigaev.ru 1986 [ # # ]:UBC 0 : ereport(ERROR,
1987 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1988 : : errmsg("expressions are not supported in included columns")));
1989 : :
1990 : : /* Copy the original index column name */
2950 teodor@sigaev.ru 1991 :CBC 12 : iparam->indexcolname = pstrdup(NameStr(attr->attname));
1992 : :
121 tgl@sss.pgh.pa.us 1993 :GNC 12 : iparam->location = -1;
1994 : :
2950 teodor@sigaev.ru 1995 :CBC 12 : index->indexIncludingParams = lappend(index->indexIncludingParams, iparam);
1996 : : }
1997 : : /* Copy reloptions if any */
6730 tgl@sss.pgh.pa.us 1998 : 2176 : datum = SysCacheGetAttr(RELOID, ht_idxrel,
1999 : : Anum_pg_class_reloptions, &isnull);
2000 [ - + ]: 2176 : if (!isnull)
6730 tgl@sss.pgh.pa.us 2001 :UBC 0 : index->options = untransformRelOptions(datum);
2002 : :
2003 : : /* If it's a partial index, decompile and append the predicate */
6730 tgl@sss.pgh.pa.us 2004 :CBC 2176 : datum = SysCacheGetAttr(INDEXRELID, ht_idx,
2005 : : Anum_pg_index_indpred, &isnull);
2006 [ + + ]: 2176 : if (!isnull)
2007 : : {
2008 : : char *pred_str;
2009 : : Node *pred_tree;
2010 : : bool found_whole_row;
2011 : :
2012 : : /* Convert text string to node tree */
6615 2013 : 20 : pred_str = TextDatumGetCString(datum);
5057 2014 : 20 : pred_tree = (Node *) stringToNode(pred_str);
2015 : :
2016 : : /* Adjust Vars to match new table's column numbering */
2017 : 20 : pred_tree = map_variable_attnos(pred_tree,
2018 : : 1, 0,
2019 : : attmap,
2020 : : InvalidOid, &found_whole_row);
2021 : :
2022 : : /* As in expandTableLikeClause, reject whole-row variables */
2023 [ - + ]: 20 : if (found_whole_row)
5057 tgl@sss.pgh.pa.us 2024 [ # # ]:UBC 0 : ereport(ERROR,
2025 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
2026 : : errmsg("cannot convert whole-row table reference"),
2027 : : errdetail("Index \"%s\" contains a whole-row table reference.",
2028 : : RelationGetRelationName(source_idx))));
2029 : :
5057 tgl@sss.pgh.pa.us 2030 :CBC 20 : index->whereClause = pred_tree;
2031 : : }
2032 : :
2033 : : /* Clean up */
6867 neilc@samurai.com 2034 : 2176 : ReleaseSysCache(ht_idxrel);
3761 tgl@sss.pgh.pa.us 2035 : 2176 : ReleaseSysCache(ht_am);
2036 : :
6867 neilc@samurai.com 2037 : 2176 : return index;
2038 : : }
2039 : :
2040 : : /*
2041 : : * Generate a CreateStatsStmt node using information from an already existing
2042 : : * extended statistic "source_statsid", for the rel identified by heapRel and
2043 : : * heapRelid.
2044 : : *
2045 : : * stxkeys in the source statistic holds attribute numbers from the parent
2046 : : * relation. Those attnums, along with the attribute numbers referenced by
2047 : : * Vars inside the expression tree, are remapped to the new relation's
2048 : : * numbering according to attmap.
2049 : : */
2050 : : static CreateStatsStmt *
2983 alvherre@alvh.no-ip. 2051 : 40 : generateClonedExtStatsStmt(RangeVar *heapRel, Oid heapRelid,
2052 : : Oid source_statsid, const AttrMap *attmap)
2053 : : {
2054 : : HeapTuple ht_stats;
2055 : : Form_pg_statistic_ext statsrec;
2056 : : CreateStatsStmt *stats;
2931 tgl@sss.pgh.pa.us 2057 : 40 : List *stat_types = NIL;
2058 : 40 : List *def_names = NIL;
2059 : : bool isnull;
2060 : : Datum datum;
2061 : : ArrayType *arr;
2062 : : char *enabled;
2063 : : int i;
2064 : :
2983 alvherre@alvh.no-ip. 2065 [ - + ]: 40 : Assert(OidIsValid(heapRelid));
2066 [ - + ]: 40 : Assert(heapRel != NULL);
2067 : :
2068 : : /*
2069 : : * Fetch pg_statistic_ext tuple of source statistics object.
2070 : : */
2071 : 40 : ht_stats = SearchSysCache1(STATEXTOID, ObjectIdGetDatum(source_statsid));
2072 [ - + ]: 40 : if (!HeapTupleIsValid(ht_stats))
2983 alvherre@alvh.no-ip. 2073 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for statistics object %u", source_statsid);
2983 alvherre@alvh.no-ip. 2074 :CBC 40 : statsrec = (Form_pg_statistic_ext) GETSTRUCT(ht_stats);
2075 : :
2076 : : /* Determine which statistics types exist */
1137 dgustafsson@postgres 2077 : 40 : datum = SysCacheGetAttrNotNull(STATEXTOID, ht_stats,
2078 : : Anum_pg_statistic_ext_stxkind);
2983 alvherre@alvh.no-ip. 2079 : 40 : arr = DatumGetArrayTypeP(datum);
2080 [ + - ]: 40 : if (ARR_NDIM(arr) != 1 ||
2081 [ + - ]: 40 : ARR_HASNULL(arr) ||
2082 [ - + ]: 40 : ARR_ELEMTYPE(arr) != CHAROID)
2983 alvherre@alvh.no-ip. 2083 [ # # ]:UBC 0 : elog(ERROR, "stxkind is not a 1-D char array");
2983 alvherre@alvh.no-ip. 2084 [ - + ]:CBC 40 : enabled = (char *) ARR_DATA_PTR(arr);
2085 [ + + ]: 128 : for (i = 0; i < ARR_DIMS(arr)[0]; i++)
2086 : : {
2087 [ + + ]: 88 : if (enabled[i] == STATS_EXT_NDISTINCT)
2088 : 24 : stat_types = lappend(stat_types, makeString("ndistinct"));
2089 [ + + ]: 64 : else if (enabled[i] == STATS_EXT_DEPENDENCIES)
2090 : 24 : stat_types = lappend(stat_types, makeString("dependencies"));
2596 tomas.vondra@postgre 2091 [ + + ]: 40 : else if (enabled[i] == STATS_EXT_MCV)
2092 : 24 : stat_types = lappend(stat_types, makeString("mcv"));
1866 2093 [ + - ]: 16 : else if (enabled[i] == STATS_EXT_EXPRESSIONS)
2094 : : /* expression stats are not exposed to users */
2095 : 16 : continue;
2096 : : else
2983 alvherre@alvh.no-ip. 2097 [ # # ]:UBC 0 : elog(ERROR, "unrecognized statistics kind %c", enabled[i]);
2098 : : }
2099 : :
2100 : : /* Determine which columns the statistics are on */
2983 alvherre@alvh.no-ip. 2101 [ + + ]:CBC 88 : for (i = 0; i < statsrec->stxkeys.dim1; i++)
2102 : : {
1866 tomas.vondra@postgre 2103 : 48 : StatsElem *selem = makeNode(StatsElem);
2983 alvherre@alvh.no-ip. 2104 : 48 : AttrNumber attnum = statsrec->stxkeys.values[i];
2105 : :
5 andrew@dunslane.net 2106 : 48 : selem->name =
2107 : 48 : get_attname(heapRelid, attmap->attnums[attnum - 1], false);
1866 tomas.vondra@postgre 2108 : 48 : selem->expr = NULL;
2109 : :
2110 : 48 : def_names = lappend(def_names, selem);
2111 : : }
2112 : :
2113 : : /*
2114 : : * Now handle expressions, if there are any. The order (with respect to
2115 : : * regular attributes) does not really matter for extended stats, so we
2116 : : * simply append them after simple column references.
2117 : : *
2118 : : * XXX Some places during build/estimation treat expressions as if they
2119 : : * are before attributes, but for the CREATE command that's entirely
2120 : : * irrelevant.
2121 : : */
2122 : 40 : datum = SysCacheGetAttr(STATEXTOID, ht_stats,
2123 : : Anum_pg_statistic_ext_stxexprs, &isnull);
2124 : :
2125 [ + + ]: 40 : if (!isnull)
2126 : : {
2127 : : ListCell *lc;
2128 : 16 : List *exprs = NIL;
2129 : : char *exprsString;
2130 : :
2131 : 16 : exprsString = TextDatumGetCString(datum);
2132 : 16 : exprs = (List *) stringToNode(exprsString);
2133 : :
2134 [ + - + + : 32 : foreach(lc, exprs)
+ + ]
2135 : : {
713 tgl@sss.pgh.pa.us 2136 : 16 : Node *expr = (Node *) lfirst(lc);
1866 tomas.vondra@postgre 2137 : 16 : StatsElem *selem = makeNode(StatsElem);
2138 : : bool found_whole_row;
2139 : :
2140 : : /* Adjust Vars to match new table's column numbering */
713 tgl@sss.pgh.pa.us 2141 : 16 : expr = map_variable_attnos(expr,
2142 : : 1, 0,
2143 : : attmap,
2144 : : InvalidOid,
2145 : : &found_whole_row);
2146 : :
1866 tomas.vondra@postgre 2147 : 16 : selem->name = NULL;
713 tgl@sss.pgh.pa.us 2148 : 16 : selem->expr = expr;
2149 : :
1866 tomas.vondra@postgre 2150 : 16 : def_names = lappend(def_names, selem);
2151 : : }
2152 : :
2153 : 16 : pfree(exprsString);
2154 : : }
2155 : :
2156 : : /* finally, build the output node */
2983 alvherre@alvh.no-ip. 2157 : 40 : stats = makeNode(CreateStatsStmt);
2158 : 40 : stats->defnames = NULL;
2159 : 40 : stats->stat_types = stat_types;
2160 : 40 : stats->exprs = def_names;
2161 : 40 : stats->relations = list_make1(heapRel);
2162 : 40 : stats->stxcomment = NULL;
1866 tomas.vondra@postgre 2163 : 40 : stats->transformed = true; /* don't need transformStatsStmt again */
1790 noah@leadboat.com 2164 : 40 : stats->if_not_exists = false;
2165 : :
2166 : : /* Clean up */
2983 alvherre@alvh.no-ip. 2167 : 40 : ReleaseSysCache(ht_stats);
2168 : :
2169 : 40 : return stats;
2170 : : }
2171 : :
2172 : : /*
2173 : : * get_collation - fetch qualified name of a collation
2174 : : *
2175 : : * If collation is InvalidOid or is the default for the given actual_datatype,
2176 : : * then the return value is NIL.
2177 : : */
2178 : : static List *
5519 tgl@sss.pgh.pa.us 2179 : 2551 : get_collation(Oid collation, Oid actual_datatype)
2180 : : {
2181 : : List *result;
2182 : : HeapTuple ht_coll;
2183 : : Form_pg_collation coll_rec;
2184 : : char *nsp_name;
2185 : : char *coll_name;
2186 : :
2187 [ + + ]: 2551 : if (!OidIsValid(collation))
2188 : 2280 : return NIL; /* easy case */
2189 [ + + ]: 271 : if (collation == get_typcollation(actual_datatype))
2190 : 257 : return NIL; /* just let it default */
2191 : :
2192 : 14 : ht_coll = SearchSysCache1(COLLOID, ObjectIdGetDatum(collation));
2193 [ - + ]: 14 : if (!HeapTupleIsValid(ht_coll))
5519 tgl@sss.pgh.pa.us 2194 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for collation %u", collation);
5519 tgl@sss.pgh.pa.us 2195 :CBC 14 : coll_rec = (Form_pg_collation) GETSTRUCT(ht_coll);
2196 : :
2197 : : /* For simplicity, we always schema-qualify the name */
2198 : 14 : nsp_name = get_namespace_name(coll_rec->collnamespace);
2199 : 14 : coll_name = pstrdup(NameStr(coll_rec->collname));
2200 : 14 : result = list_make2(makeString(nsp_name), makeString(coll_name));
2201 : :
2202 : 14 : ReleaseSysCache(ht_coll);
2203 : 14 : return result;
2204 : : }
2205 : :
2206 : : /*
2207 : : * get_opclass - fetch qualified name of an index operator class
2208 : : *
2209 : : * If the opclass is the default for the given actual_datatype, then
2210 : : * the return value is NIL.
2211 : : */
2212 : : static List *
6867 neilc@samurai.com 2213 : 2551 : get_opclass(Oid opclass, Oid actual_datatype)
2214 : : {
5519 tgl@sss.pgh.pa.us 2215 : 2551 : List *result = NIL;
2216 : : HeapTuple ht_opc;
2217 : : Form_pg_opclass opc_rec;
2218 : :
5924 rhaas@postgresql.org 2219 : 2551 : ht_opc = SearchSysCache1(CLAOID, ObjectIdGetDatum(opclass));
6867 neilc@samurai.com 2220 [ - + ]: 2551 : if (!HeapTupleIsValid(ht_opc))
6867 neilc@samurai.com 2221 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for opclass %u", opclass);
6867 neilc@samurai.com 2222 :CBC 2551 : opc_rec = (Form_pg_opclass) GETSTRUCT(ht_opc);
2223 : :
6730 tgl@sss.pgh.pa.us 2224 [ + + ]: 2551 : if (GetDefaultOpClass(actual_datatype, opc_rec->opcmethod) != opclass)
2225 : : {
2226 : : /* For simplicity, we always schema-qualify the name */
6746 bruce@momjian.us 2227 : 16 : char *nsp_name = get_namespace_name(opc_rec->opcnamespace);
6730 tgl@sss.pgh.pa.us 2228 : 16 : char *opc_name = pstrdup(NameStr(opc_rec->opcname));
2229 : :
6867 neilc@samurai.com 2230 : 16 : result = list_make2(makeString(nsp_name), makeString(opc_name));
2231 : : }
2232 : :
2233 : 2551 : ReleaseSysCache(ht_opc);
2234 : 2551 : return result;
2235 : : }
2236 : :
2237 : :
2238 : : /*
2239 : : * transformIndexConstraints
2240 : : * Handle UNIQUE, PRIMARY KEY, EXCLUDE constraints, which create indexes.
2241 : : * We also merge in any index definitions arising from
2242 : : * LIKE ... INCLUDING INDEXES.
2243 : : */
2244 : : static void
5579 tgl@sss.pgh.pa.us 2245 : 42088 : transformIndexConstraints(CreateStmtContext *cxt)
2246 : : {
2247 : : IndexStmt *index;
6867 neilc@samurai.com 2248 : 42088 : List *indexlist = NIL;
2569 tgl@sss.pgh.pa.us 2249 : 42088 : List *finalindexlist = NIL;
2250 : : ListCell *lc;
2251 : :
2252 : : /*
2253 : : * Run through the constraints that need to generate an index, and do so.
2254 : : *
2255 : : * For PRIMARY KEY, this queues not-null constraints for each column, if
2256 : : * needed.
2257 : : */
6867 neilc@samurai.com 2258 [ + + + + : 54582 : foreach(lc, cxt->ixconstraints)
+ + ]
2259 : : {
3312 tgl@sss.pgh.pa.us 2260 : 12538 : Constraint *constraint = lfirst_node(Constraint, lc);
2261 : :
6730 2262 [ + + + + : 12538 : Assert(constraint->contype == CONSTR_PRIMARY ||
- + ]
2263 : : constraint->contype == CONSTR_UNIQUE ||
2264 : : constraint->contype == CONSTR_EXCLUSION);
2265 : :
6867 neilc@samurai.com 2266 : 12538 : index = transformIndexConstraint(constraint, cxt);
2267 : :
6730 tgl@sss.pgh.pa.us 2268 : 12494 : indexlist = lappend(indexlist, index);
2269 : : }
2270 : :
2271 : : /*
2272 : : * Scan the index list and remove any redundant index specifications. This
2273 : : * can happen if, for instance, the user writes UNIQUE PRIMARY KEY. A
2274 : : * strict reading of SQL would suggest raising an error instead, but that
2275 : : * strikes me as too anal-retentive. - tgl 2001-02-14
2276 : : *
2277 : : * XXX in ALTER TABLE case, it'd be nice to look for duplicate
2278 : : * pre-existing indexes, too.
2279 : : */
6891 2280 [ + + ]: 42044 : if (cxt->pkey != NULL)
2281 : : {
2282 : : /* Make sure we keep the PKEY index in preference to others... */
2569 2283 : 8790 : finalindexlist = list_make1(cxt->pkey);
2284 : : }
2285 : :
6867 neilc@samurai.com 2286 [ + + + + : 54538 : foreach(lc, indexlist)
+ + ]
2287 : : {
6891 tgl@sss.pgh.pa.us 2288 : 12494 : bool keep = true;
2289 : : ListCell *k;
2290 : :
6867 neilc@samurai.com 2291 : 12494 : index = lfirst(lc);
2292 : :
2293 : : /* if it's pkey, it's already in finalindexlist */
6891 tgl@sss.pgh.pa.us 2294 [ + + ]: 12494 : if (index == cxt->pkey)
2295 : 8790 : continue;
2296 : :
2569 2297 [ + + + + : 3828 : foreach(k, finalindexlist)
+ + ]
2298 : : {
6891 2299 : 124 : IndexStmt *priorindex = lfirst(k);
2300 : :
6730 2301 [ + + + - ]: 128 : if (equal(index->indexParams, priorindex->indexParams) &&
2950 teodor@sigaev.ru 2302 [ + - ]: 8 : equal(index->indexIncludingParams, priorindex->indexIncludingParams) &&
6730 tgl@sss.pgh.pa.us 2303 [ + - ]: 8 : equal(index->whereClause, priorindex->whereClause) &&
5993 2304 : 4 : equal(index->excludeOpNames, priorindex->excludeOpNames) &&
6124 2305 [ + - ]: 4 : strcmp(index->accessMethod, priorindex->accessMethod) == 0 &&
1552 peter@eisentraut.org 2306 [ + - ]: 4 : index->nulls_not_distinct == priorindex->nulls_not_distinct &&
6124 tgl@sss.pgh.pa.us 2307 [ - + ]: 4 : index->deferrable == priorindex->deferrable &&
6124 tgl@sss.pgh.pa.us 2308 [ # # ]:UBC 0 : index->initdeferred == priorindex->initdeferred)
2309 : : {
6730 2310 : 0 : priorindex->unique |= index->unique;
2311 : :
2312 : : /*
2313 : : * If the prior index is as yet unnamed, and this one is
2314 : : * named, then transfer the name to the prior index. This
2315 : : * ensures that if we have named and unnamed constraints,
2316 : : * we'll use (at least one of) the names for the index.
2317 : : */
6891 2318 [ # # ]: 0 : if (priorindex->idxname == NULL)
2319 : 0 : priorindex->idxname = index->idxname;
2320 : 0 : keep = false;
2321 : 0 : break;
2322 : : }
2323 : : }
2324 : :
6891 tgl@sss.pgh.pa.us 2325 [ + - ]:CBC 3704 : if (keep)
2569 2326 : 3704 : finalindexlist = lappend(finalindexlist, index);
2327 : : }
2328 : :
2329 : : /*
2330 : : * Now append all the IndexStmts to cxt->alist.
2331 : : */
2332 : 42044 : cxt->alist = list_concat(cxt->alist, finalindexlist);
6867 neilc@samurai.com 2333 : 42044 : }
2334 : :
2335 : : /*
2336 : : * transformIndexConstraint
2337 : : * Transform one UNIQUE, PRIMARY KEY, or EXCLUDE constraint for
2338 : : * transformIndexConstraints. An IndexStmt is returned.
2339 : : *
2340 : : * For a PRIMARY KEY constraint, we additionally create not-null constraints
2341 : : * for columns that don't already have them.
2342 : : */
2343 : : static IndexStmt *
2344 : 12538 : transformIndexConstraint(Constraint *constraint, CreateStmtContext *cxt)
2345 : : {
2346 : : IndexStmt *index;
2347 : : ListCell *lc;
2348 : :
2349 : 12538 : index = makeNode(IndexStmt);
2350 : :
5993 tgl@sss.pgh.pa.us 2351 : 12538 : index->unique = (constraint->contype != CONSTR_EXCLUSION);
6867 neilc@samurai.com 2352 : 12538 : index->primary = (constraint->contype == CONSTR_PRIMARY);
2353 [ + + ]: 12538 : if (index->primary)
2354 : : {
2355 [ - + ]: 8810 : if (cxt->pkey != NULL)
6867 neilc@samurai.com 2356 [ # # ]:UBC 0 : ereport(ERROR,
2357 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
2358 : : errmsg("multiple primary keys for table \"%s\" are not allowed",
2359 : : cxt->relation->relname),
2360 : : parser_errposition(cxt->pstate, constraint->location)));
6867 neilc@samurai.com 2361 :CBC 8810 : cxt->pkey = index;
2362 : :
2363 : : /*
2364 : : * In ALTER TABLE case, a primary index might already exist, but
2365 : : * DefineIndex will check for it.
2366 : : */
2367 : : }
1552 peter@eisentraut.org 2368 : 12538 : index->nulls_not_distinct = constraint->nulls_not_distinct;
6867 neilc@samurai.com 2369 : 12538 : index->isconstraint = true;
595 peter@eisentraut.org 2370 : 12538 : index->iswithoutoverlaps = constraint->without_overlaps;
6124 tgl@sss.pgh.pa.us 2371 : 12538 : index->deferrable = constraint->deferrable;
2372 : 12538 : index->initdeferred = constraint->initdeferred;
2373 : :
6123 2374 [ + + ]: 12538 : if (constraint->conname != NULL)
2375 : 980 : index->idxname = pstrdup(constraint->conname);
2376 : : else
6746 bruce@momjian.us 2377 : 11558 : index->idxname = NULL; /* DefineIndex will choose name */
2378 : :
6867 neilc@samurai.com 2379 : 12538 : index->relation = cxt->relation;
5993 tgl@sss.pgh.pa.us 2380 [ + + ]: 12538 : index->accessMethod = constraint->access_method ? constraint->access_method : DEFAULT_INDEX_TYPE;
6867 neilc@samurai.com 2381 : 12538 : index->options = constraint->options;
2382 : 12538 : index->tableSpace = constraint->indexspace;
5993 tgl@sss.pgh.pa.us 2383 : 12538 : index->whereClause = constraint->where_clause;
6867 neilc@samurai.com 2384 : 12538 : index->indexParams = NIL;
2950 teodor@sigaev.ru 2385 : 12538 : index->indexIncludingParams = NIL;
5993 tgl@sss.pgh.pa.us 2386 : 12538 : index->excludeOpNames = NIL;
5041 2387 : 12538 : index->idxcomment = NULL;
5579 2388 : 12538 : index->indexOid = InvalidOid;
1399 rhaas@postgresql.org 2389 : 12538 : index->oldNumber = InvalidRelFileNumber;
2222 noah@leadboat.com 2390 : 12538 : index->oldCreateSubid = InvalidSubTransactionId;
1399 rhaas@postgresql.org 2391 : 12538 : index->oldFirstRelfilelocatorSubid = InvalidSubTransactionId;
4090 tgl@sss.pgh.pa.us 2392 : 12538 : index->transformed = false;
6867 neilc@samurai.com 2393 : 12538 : index->concurrent = false;
4090 tgl@sss.pgh.pa.us 2394 : 12538 : index->if_not_exists = false;
2567 alvherre@alvh.no-ip. 2395 : 12538 : index->reset_default_tblspc = constraint->reset_default_tblspc;
2396 : :
2397 : : /*
2398 : : * If it's ALTER TABLE ADD CONSTRAINT USING INDEX, look up the index and
2399 : : * verify it's usable, then extract the implied column name list. (We
2400 : : * will not actually need the column name list at runtime, but we need it
2401 : : * now to check for duplicate column entries below.)
2402 : : */
5579 tgl@sss.pgh.pa.us 2403 [ + + ]: 12538 : if (constraint->indexname != NULL)
2404 : : {
2405 : 6669 : char *index_name = constraint->indexname;
2406 : 6669 : Relation heap_rel = cxt->rel;
2407 : : Oid index_oid;
2408 : : Relation index_rel;
2409 : : Form_pg_index index_form;
2410 : : oidvector *indclass;
2411 : : Datum indclassDatum;
2412 : : int i;
2413 : :
2414 : : /* Grammar should not allow this with explicit column list */
2415 [ - + ]: 6669 : Assert(constraint->keys == NIL);
2416 : :
2417 : : /* Grammar should only allow PRIMARY and UNIQUE constraints */
2418 [ + + - + ]: 6669 : Assert(constraint->contype == CONSTR_PRIMARY ||
2419 : : constraint->contype == CONSTR_UNIQUE);
2420 : :
2421 : : /* Must be ALTER, not CREATE, but grammar doesn't enforce that */
2422 [ - + ]: 6669 : if (!cxt->isalter)
5579 tgl@sss.pgh.pa.us 2423 [ # # ]:UBC 0 : ereport(ERROR,
2424 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
2425 : : errmsg("cannot use an existing index in CREATE TABLE"),
2426 : : parser_errposition(cxt->pstate, constraint->location)));
2427 : :
2428 : : /* Look for the index in the same schema as the table */
5579 tgl@sss.pgh.pa.us 2429 :CBC 6669 : index_oid = get_relname_relid(index_name, RelationGetNamespace(heap_rel));
2430 : :
2431 [ - + ]: 6669 : if (!OidIsValid(index_oid))
5579 tgl@sss.pgh.pa.us 2432 [ # # ]:UBC 0 : ereport(ERROR,
2433 : : (errcode(ERRCODE_UNDEFINED_OBJECT),
2434 : : errmsg("index \"%s\" does not exist", index_name),
2435 : : parser_errposition(cxt->pstate, constraint->location)));
2436 : :
2437 : : /* Open the index (this will throw an error if it is not an index) */
5579 tgl@sss.pgh.pa.us 2438 :CBC 6669 : index_rel = index_open(index_oid, AccessShareLock);
2439 : 6669 : index_form = index_rel->rd_index;
2440 : :
2441 : : /* Check that it does not have an associated constraint already */
2442 [ - + ]: 6669 : if (OidIsValid(get_index_constraint(index_oid)))
5579 tgl@sss.pgh.pa.us 2443 [ # # ]:UBC 0 : ereport(ERROR,
2444 : : (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
2445 : : errmsg("index \"%s\" is already associated with a constraint",
2446 : : index_name),
2447 : : parser_errposition(cxt->pstate, constraint->location)));
2448 : :
2449 : : /* Perform validity checks on the index */
5579 tgl@sss.pgh.pa.us 2450 [ - + ]:CBC 6669 : if (index_form->indrelid != RelationGetRelid(heap_rel))
5579 tgl@sss.pgh.pa.us 2451 [ # # ]:UBC 0 : ereport(ERROR,
2452 : : (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
2453 : : errmsg("index \"%s\" does not belong to table \"%s\"",
2454 : : index_name, RelationGetRelationName(heap_rel)),
2455 : : parser_errposition(cxt->pstate, constraint->location)));
2456 : :
2686 peter_e@gmx.net 2457 [ - + ]:CBC 6669 : if (!index_form->indisvalid)
5579 tgl@sss.pgh.pa.us 2458 [ # # ]:UBC 0 : ereport(ERROR,
2459 : : (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
2460 : : errmsg("index \"%s\" is not valid", index_name),
2461 : : parser_errposition(cxt->pstate, constraint->location)));
2462 : :
2463 : : /*
2464 : : * Today we forbid non-unique indexes, but we could permit GiST
2465 : : * indexes whose last entry is a range type and use that to create a
2466 : : * WITHOUT OVERLAPS constraint (i.e. a temporal constraint).
2467 : : */
5579 tgl@sss.pgh.pa.us 2468 [ + + ]:CBC 6669 : if (!index_form->indisunique)
2469 [ + - ]: 8 : ereport(ERROR,
2470 : : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
2471 : : errmsg("\"%s\" is not a unique index", index_name),
2472 : : errdetail("Cannot create a primary key or unique constraint using such an index."),
2473 : : parser_errposition(cxt->pstate, constraint->location)));
2474 : :
2475 [ - + ]: 6661 : if (RelationGetIndexExpressions(index_rel) != NIL)
5579 tgl@sss.pgh.pa.us 2476 [ # # ]:UBC 0 : ereport(ERROR,
2477 : : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
2478 : : errmsg("index \"%s\" contains expressions", index_name),
2479 : : errdetail("Cannot create a primary key or unique constraint using such an index."),
2480 : : parser_errposition(cxt->pstate, constraint->location)));
2481 : :
5579 tgl@sss.pgh.pa.us 2482 [ - + ]:CBC 6661 : if (RelationGetIndexPredicate(index_rel) != NIL)
5579 tgl@sss.pgh.pa.us 2483 [ # # ]:UBC 0 : ereport(ERROR,
2484 : : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
2485 : : errmsg("\"%s\" is a partial index", index_name),
2486 : : errdetail("Cannot create a primary key or unique constraint using such an index."),
2487 : : parser_errposition(cxt->pstate, constraint->location)));
2488 : :
2489 : : /*
2490 : : * It's probably unsafe to change a deferred index to non-deferred. (A
2491 : : * non-constraint index couldn't be deferred anyway, so this case
2492 : : * should never occur; no need to sweat, but let's check it.)
2493 : : */
5579 tgl@sss.pgh.pa.us 2494 [ - + - - ]:CBC 6661 : if (!index_form->indimmediate && !constraint->deferrable)
5579 tgl@sss.pgh.pa.us 2495 [ # # ]:UBC 0 : ereport(ERROR,
2496 : : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
2497 : : errmsg("\"%s\" is a deferrable index", index_name),
2498 : : errdetail("Cannot create a non-deferrable constraint using a deferrable index."),
2499 : : parser_errposition(cxt->pstate, constraint->location)));
2500 : :
2501 : : /*
2502 : : * Insist on it being a btree. We must have an index that exactly
2503 : : * matches what you'd get from plain ADD CONSTRAINT syntax, else dump
2504 : : * and reload will produce a different index (breaking pg_upgrade in
2505 : : * particular).
2506 : : */
3695 alvherre@alvh.no-ip. 2507 [ - + ]:CBC 6661 : if (index_rel->rd_rel->relam != get_index_am_oid(DEFAULT_INDEX_TYPE, false))
5579 tgl@sss.pgh.pa.us 2508 [ # # ]:UBC 0 : ereport(ERROR,
2509 : : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
2510 : : errmsg("index \"%s\" is not a btree", index_name),
2511 : : parser_errposition(cxt->pstate, constraint->location)));
2512 : :
2513 : : /* Must get indclass the hard way */
1137 dgustafsson@postgres 2514 :CBC 6661 : indclassDatum = SysCacheGetAttrNotNull(INDEXRELID,
2515 : 6661 : index_rel->rd_indextuple,
2516 : : Anum_pg_index_indclass);
5579 tgl@sss.pgh.pa.us 2517 : 6661 : indclass = (oidvector *) DatumGetPointer(indclassDatum);
2518 : :
2519 [ + + ]: 17538 : for (i = 0; i < index_form->indnatts; i++)
2520 : : {
5062 peter_e@gmx.net 2521 : 10885 : int16 attnum = index_form->indkey.values[i];
2522 : : const FormData_pg_attribute *attform;
2523 : : char *attname;
2524 : : Oid defopclass;
2525 : :
2526 : : /*
2527 : : * We shouldn't see attnum == 0 here, since we already rejected
2528 : : * expression indexes. If we do, SystemAttributeDefinition will
2529 : : * throw an error.
2530 : : */
5579 tgl@sss.pgh.pa.us 2531 [ + - ]: 10885 : if (attnum > 0)
2532 : : {
2533 [ - + ]: 10885 : Assert(attnum <= heap_rel->rd_att->natts);
3180 andres@anarazel.de 2534 : 10885 : attform = TupleDescAttr(heap_rel->rd_att, attnum - 1);
2535 : : }
2536 : : else
2723 andres@anarazel.de 2537 :UBC 0 : attform = SystemAttributeDefinition(attnum);
5579 tgl@sss.pgh.pa.us 2538 :CBC 10885 : attname = pstrdup(NameStr(attform->attname));
2539 : :
2950 teodor@sigaev.ru 2540 [ + + ]: 10885 : if (i < index_form->indnkeyatts)
2541 : : {
2542 : : /*
2543 : : * Insist on default opclass, collation, and sort options.
2544 : : * While the index would still work as a constraint with
2545 : : * non-default settings, it might not provide exactly the same
2546 : : * uniqueness semantics as you'd get from a normally-created
2547 : : * constraint; and there's also the dump/reload problem
2548 : : * mentioned above.
2549 : : */
2550 : : Datum attoptions =
1082 tgl@sss.pgh.pa.us 2551 : 10865 : get_attoptions(RelationGetRelid(index_rel), i + 1);
2552 : :
2950 teodor@sigaev.ru 2553 : 10865 : defopclass = GetDefaultOpClass(attform->atttypid,
2554 : 10865 : index_rel->rd_rel->relam);
2555 [ + - ]: 10865 : if (indclass->values[i] != defopclass ||
2342 tgl@sss.pgh.pa.us 2556 [ + + + - ]: 10865 : attform->attcollation != index_rel->rd_indcollation[i] ||
2227 akorotkov@postgresql 2557 : 10861 : attoptions != (Datum) 0 ||
2950 teodor@sigaev.ru 2558 [ + + ]: 10861 : index_rel->rd_indoption[i] != 0)
2559 [ + - ]: 8 : ereport(ERROR,
2560 : : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
2561 : : errmsg("index \"%s\" column number %d does not have default sorting behavior", index_name, i + 1),
2562 : : errdetail("Cannot create a primary key or unique constraint using such an index."),
2563 : : parser_errposition(cxt->pstate, constraint->location)));
2564 : :
2565 : : /* If a PK, ensure the columns get not null constraints */
543 alvherre@alvh.no-ip. 2566 [ + + ]: 10857 : if (constraint->contype == CONSTR_PRIMARY)
2567 : 4841 : cxt->nnconstraints =
2568 : 4841 : lappend(cxt->nnconstraints,
2569 : 4841 : makeNotNullConstraint(makeString(attname)));
2570 : :
2950 teodor@sigaev.ru 2571 : 10857 : constraint->keys = lappend(constraint->keys, makeString(attname));
2572 : : }
2573 : : else
2574 : 20 : constraint->including = lappend(constraint->including, makeString(attname));
2575 : : }
2576 : :
2577 : : /* Close the index relation but keep the lock */
137 michael@paquier.xyz 2578 :GNC 6653 : index_close(index_rel, NoLock);
2579 : :
5579 tgl@sss.pgh.pa.us 2580 :CBC 6653 : index->indexOid = index_oid;
2581 : : }
2582 : :
2583 : : /*
2584 : : * If it's an EXCLUDE constraint, the grammar returns a list of pairs of
2585 : : * IndexElems and operator names. We have to break that apart into
2586 : : * separate lists.
2587 : : */
5993 2588 [ + + ]: 12522 : if (constraint->contype == CONSTR_EXCLUSION)
2589 : : {
2590 [ + - + + : 375 : foreach(lc, constraint->exclusions)
+ + ]
2591 : : {
5912 bruce@momjian.us 2592 : 222 : List *pair = (List *) lfirst(lc);
2593 : : IndexElem *elem;
2594 : : List *opname;
2595 : :
5993 tgl@sss.pgh.pa.us 2596 [ - + ]: 222 : Assert(list_length(pair) == 2);
3312 2597 : 222 : elem = linitial_node(IndexElem, pair);
2598 : 222 : opname = lsecond_node(List, pair);
2599 : :
5993 2600 : 222 : index->indexParams = lappend(index->indexParams, elem);
2601 : 222 : index->excludeOpNames = lappend(index->excludeOpNames, opname);
2602 : : }
2603 : : }
2604 : :
2605 : : /*
2606 : : * For UNIQUE and PRIMARY KEY, we just have a list of column names.
2607 : : *
2608 : : * Make sure referenced keys exist. If we are making a PRIMARY KEY index,
2609 : : * also make sure they are not-null. For WITHOUT OVERLAPS constraints, we
2610 : : * make sure the last part is a range or multirange.
2611 : : */
2612 : : else
2613 : : {
2950 teodor@sigaev.ru 2614 [ + - + + : 30067 : foreach(lc, constraint->keys)
+ + ]
2615 : : {
2616 : 17718 : char *key = strVal(lfirst(lc));
2617 : 17718 : bool found = false;
2618 : 17718 : ColumnDef *column = NULL;
2619 : : ListCell *columns;
2620 : : IndexElem *iparam;
595 peter@eisentraut.org 2621 : 17718 : Oid typid = InvalidOid;
2622 : :
2623 : : /* Make sure referenced column exists. */
2950 teodor@sigaev.ru 2624 [ + + + + : 19004 : foreach(columns, cxt->columns)
+ + ]
2625 : : {
1751 peter@eisentraut.org 2626 : 7013 : column = lfirst_node(ColumnDef, columns);
2950 teodor@sigaev.ru 2627 [ + + ]: 7013 : if (strcmp(column->colname, key) == 0)
2628 : : {
2629 : 5727 : found = true;
2630 : 5727 : break;
2631 : : }
2632 : : }
595 peter@eisentraut.org 2633 [ + + ]: 17718 : if (!found)
2634 : 11991 : column = NULL;
2635 : :
2950 teodor@sigaev.ru 2636 [ + + ]: 17718 : if (found)
2637 : : {
2638 : : /*
2639 : : * column is defined in the new table. For CREATE TABLE with
2640 : : * a PRIMARY KEY, we can apply the not-null constraint cheaply
2641 : : * here. If the not-null constraint already exists, we can
2642 : : * (albeit not so cheaply) verify that it's not a NO INHERIT
2643 : : * constraint.
2644 : : *
2645 : : * Note that ALTER TABLE never needs either check, because
2646 : : * those constraints have already been added by
2647 : : * ATPrepAddPrimaryKey.
2648 : : */
2569 tgl@sss.pgh.pa.us 2649 [ + + ]: 5727 : if (constraint->contype == CONSTR_PRIMARY &&
543 alvherre@alvh.no-ip. 2650 [ + + ]: 5192 : !cxt->isalter)
2651 : : {
2652 [ + + ]: 5173 : if (column->is_not_null)
2653 : : {
2654 [ + - + - : 8335 : foreach_node(Constraint, nn, cxt->nnconstraints)
+ + ]
2655 : : {
2656 [ + + ]: 4284 : if (strcmp(strVal(linitial(nn->keys)), key) == 0)
2657 : : {
2658 [ + + ]: 4055 : if (nn->is_no_inherit)
2659 [ + - ]: 4 : ereport(ERROR,
2660 : : errcode(ERRCODE_SYNTAX_ERROR),
2661 : : errmsg("conflicting NO INHERIT declaration for not-null constraint on column \"%s\"",
2662 : : key));
2663 : 4051 : break;
2664 : : }
2665 : : }
2666 : : }
2667 : : else
2668 : : {
2669 : 1118 : column->is_not_null = true;
2670 : 1118 : cxt->nnconstraints =
2671 : 1118 : lappend(cxt->nnconstraints,
2672 : 1118 : makeNotNullConstraint(makeString(key)));
2673 : : }
2674 : : }
2675 [ + + ]: 554 : else if (constraint->contype == CONSTR_PRIMARY)
2676 [ - + ]: 19 : Assert(column->is_not_null);
2677 : : }
2723 andres@anarazel.de 2678 [ - + ]: 11991 : else if (SystemAttributeByName(key) != NULL)
2679 : : {
2680 : : /*
2681 : : * column will be a system column in the new table, so accept
2682 : : * it. System columns can't ever be null, so no need to worry
2683 : : * about PRIMARY/NOT NULL constraint.
2684 : : */
2950 teodor@sigaev.ru 2685 :UBC 0 : found = true;
2686 : : }
2950 teodor@sigaev.ru 2687 [ + + ]:CBC 11991 : else if (cxt->inhRelations)
2688 : : {
2689 : : /* try inherited tables */
2690 : : ListCell *inher;
2691 : :
2692 [ + - + - : 64 : foreach(inher, cxt->inhRelations)
+ - ]
2693 : : {
1751 peter@eisentraut.org 2694 : 64 : RangeVar *inh = lfirst_node(RangeVar, inher);
2695 : : Relation rel;
2696 : : int count;
2697 : :
2661 andres@anarazel.de 2698 : 64 : rel = table_openrv(inh, AccessShareLock);
2699 : : /* check user requested inheritance from valid relkind */
2950 teodor@sigaev.ru 2700 [ - + ]: 64 : if (rel->rd_rel->relkind != RELKIND_RELATION &&
2950 teodor@sigaev.ru 2701 [ # # ]:UBC 0 : rel->rd_rel->relkind != RELKIND_FOREIGN_TABLE &&
2702 [ # # ]: 0 : rel->rd_rel->relkind != RELKIND_PARTITIONED_TABLE)
2703 [ # # ]: 0 : ereport(ERROR,
2704 : : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
2705 : : errmsg("inherited relation \"%s\" is not a table or foreign table",
2706 : : inh->relname)));
2950 teodor@sigaev.ru 2707 [ + - ]:CBC 68 : for (count = 0; count < rel->rd_att->natts; count++)
2708 : : {
2709 : 68 : Form_pg_attribute inhattr = TupleDescAttr(rel->rd_att,
2710 : : count);
2711 : 68 : char *inhname = NameStr(inhattr->attname);
2712 : :
2713 [ - + ]: 68 : if (inhattr->attisdropped)
2950 teodor@sigaev.ru 2714 :UBC 0 : continue;
2950 teodor@sigaev.ru 2715 [ + + ]:CBC 68 : if (strcmp(key, inhname) == 0)
2716 : : {
2717 : 64 : found = true;
595 peter@eisentraut.org 2718 : 64 : typid = inhattr->atttypid;
2719 : :
543 alvherre@alvh.no-ip. 2720 [ + + ]: 64 : if (constraint->contype == CONSTR_PRIMARY)
2721 : 56 : cxt->nnconstraints =
2722 : 56 : lappend(cxt->nnconstraints,
2723 : 56 : makeNotNullConstraint(makeString(pstrdup(inhname))));
2950 teodor@sigaev.ru 2724 : 64 : break;
2725 : : }
2726 : : }
2661 andres@anarazel.de 2727 : 64 : table_close(rel, NoLock);
2950 teodor@sigaev.ru 2728 [ + - ]: 64 : if (found)
2729 : 64 : break;
2730 : : }
2731 : : }
2732 : :
2733 : : /*
2734 : : * In the ALTER TABLE case, don't complain about index keys not
2735 : : * created in the command; they may well exist already.
2736 : : * DefineIndex will complain about them if not.
2737 : : */
2738 [ + + + + ]: 17714 : if (!found && !cxt->isalter)
2739 [ + - ]: 8 : ereport(ERROR,
2740 : : (errcode(ERRCODE_UNDEFINED_COLUMN),
2741 : : errmsg("column \"%s\" named in key does not exist", key),
2742 : : parser_errposition(cxt->pstate, constraint->location)));
2743 : :
2744 : : /* Check for PRIMARY KEY(foo, foo) */
2745 [ + + + + : 24719 : foreach(columns, index->indexParams)
+ + ]
2746 : : {
2747 : 7013 : iparam = (IndexElem *) lfirst(columns);
2748 [ + - - + ]: 7013 : if (iparam->name && strcmp(key, iparam->name) == 0)
2749 : : {
2950 teodor@sigaev.ru 2750 [ # # ]:UBC 0 : if (index->primary)
2751 [ # # ]: 0 : ereport(ERROR,
2752 : : (errcode(ERRCODE_DUPLICATE_COLUMN),
2753 : : errmsg("column \"%s\" appears twice in primary key constraint",
2754 : : key),
2755 : : parser_errposition(cxt->pstate, constraint->location)));
2756 : : else
2757 [ # # ]: 0 : ereport(ERROR,
2758 : : (errcode(ERRCODE_DUPLICATE_COLUMN),
2759 : : errmsg("column \"%s\" appears twice in unique constraint",
2760 : : key),
2761 : : parser_errposition(cxt->pstate, constraint->location)));
2762 : : }
2763 : : }
2764 : :
2765 : : /*
2766 : : * The WITHOUT OVERLAPS part (if any) must be a range or
2767 : : * multirange type, or a domain over such a type.
2768 : : */
595 peter@eisentraut.org 2769 [ + + + + ]:CBC 17706 : if (constraint->without_overlaps && lc == list_last_cell(constraint->keys))
2770 : : {
2771 [ + + + - ]: 531 : if (!found && cxt->isalter)
2772 : : {
2773 : : /*
2774 : : * Look up the column type on existing table. If we can't
2775 : : * find it, let things fail in DefineIndex.
2776 : : */
2777 : 109 : Relation rel = cxt->rel;
2778 : :
2779 [ + - ]: 220 : for (int i = 0; i < rel->rd_att->natts; i++)
2780 : : {
2781 : 220 : Form_pg_attribute attr = TupleDescAttr(rel->rd_att, i);
2782 : : const char *attname;
2783 : :
2784 [ - + ]: 220 : if (attr->attisdropped)
28 tgl@sss.pgh.pa.us 2785 :UBC 0 : continue;
595 peter@eisentraut.org 2786 :CBC 220 : attname = NameStr(attr->attname);
2787 [ + + ]: 220 : if (strcmp(attname, key) == 0)
2788 : : {
2789 : 109 : found = true;
2790 : 109 : typid = attr->atttypid;
2791 : 109 : break;
2792 : : }
2793 : : }
2794 : : }
2795 [ + - ]: 531 : if (found)
2796 : : {
2797 : : /* Look up column type if we didn't already */
2798 [ + + + - ]: 531 : if (!OidIsValid(typid) && column)
28 tgl@sss.pgh.pa.us 2799 : 418 : typid = typenameTypeId(cxt->pstate,
2800 : 418 : column->typeName);
2801 : : /* Look through any domain */
2802 [ + - ]: 531 : if (OidIsValid(typid))
2803 : 531 : typid = getBaseType(typid);
2804 : : /* Complain if not range/multirange */
2805 [ + - ]: 531 : if (!OidIsValid(typid) ||
2806 [ + + + + ]: 531 : !(type_is_range(typid) || type_is_multirange(typid)))
595 peter@eisentraut.org 2807 [ + - ]: 8 : ereport(ERROR,
2808 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2809 : : errmsg("column \"%s\" in WITHOUT OVERLAPS is not a range or multirange type", key),
2810 : : parser_errposition(cxt->pstate, constraint->location)));
2811 : : }
2812 : : }
2813 : :
2814 : : /* OK, add it to the index definition */
2950 teodor@sigaev.ru 2815 : 17698 : iparam = makeNode(IndexElem);
2816 : 17698 : iparam->name = pstrdup(key);
2817 : 17698 : iparam->expr = NULL;
2818 : 17698 : iparam->indexcolname = NULL;
2819 : 17698 : iparam->collation = NIL;
2820 : 17698 : iparam->opclass = NIL;
2227 akorotkov@postgresql 2821 : 17698 : iparam->opclassopts = NIL;
2950 teodor@sigaev.ru 2822 : 17698 : iparam->ordering = SORTBY_DEFAULT;
2823 : 17698 : iparam->nulls_ordering = SORTBY_NULLS_DEFAULT;
121 tgl@sss.pgh.pa.us 2824 :GNC 17698 : iparam->location = -1;
2950 teodor@sigaev.ru 2825 :CBC 17698 : index->indexParams = lappend(index->indexParams, iparam);
2826 : : }
2827 : :
595 peter@eisentraut.org 2828 [ + + ]: 12349 : if (constraint->without_overlaps)
2829 : : {
2830 : : /*
2831 : : * This enforces that there is at least one equality column
2832 : : * besides the WITHOUT OVERLAPS columns. This is per SQL
2833 : : * standard. XXX Do we need this?
2834 : : */
2835 [ + + ]: 523 : if (list_length(constraint->keys) < 2)
2836 [ + - ]: 8 : ereport(ERROR,
2837 : : errcode(ERRCODE_SYNTAX_ERROR),
2838 : : errmsg("constraint using WITHOUT OVERLAPS needs at least two columns"));
2839 : :
2840 : : /* WITHOUT OVERLAPS requires a GiST index */
2841 : 515 : index->accessMethod = "gist";
2842 : : }
2843 : :
2844 : : }
2845 : :
2846 : : /*
2847 : : * Add included columns to index definition. This is much like the
2848 : : * simple-column-name-list code above, except that we don't worry about
2849 : : * NOT NULL marking; included columns in a primary key should not be
2850 : : * forced NOT NULL. We don't complain about duplicate columns, either,
2851 : : * though maybe we should?
2852 : : */
2950 teodor@sigaev.ru 2853 [ + + + + : 12686 : foreach(lc, constraint->including)
+ + ]
2854 : : {
5993 tgl@sss.pgh.pa.us 2855 : 192 : char *key = strVal(lfirst(lc));
6867 neilc@samurai.com 2856 : 192 : bool found = false;
2857 : 192 : ColumnDef *column = NULL;
2858 : : ListCell *columns;
2859 : : IndexElem *iparam;
2860 : :
2861 [ + + + - : 417 : foreach(columns, cxt->columns)
+ + ]
2862 : : {
3312 tgl@sss.pgh.pa.us 2863 : 341 : column = lfirst_node(ColumnDef, columns);
6867 neilc@samurai.com 2864 [ + + ]: 341 : if (strcmp(column->colname, key) == 0)
2865 : : {
2866 : 116 : found = true;
2867 : 116 : break;
2868 : : }
2869 : : }
2870 : :
2950 teodor@sigaev.ru 2871 [ + + ]: 192 : if (!found)
2872 : : {
2723 andres@anarazel.de 2873 [ - + ]: 76 : if (SystemAttributeByName(key) != NULL)
2874 : : {
2875 : : /*
2876 : : * column will be a system column in the new table, so accept
2877 : : * it.
2878 : : */
2950 teodor@sigaev.ru 2879 :UBC 0 : found = true;
2880 : : }
2950 teodor@sigaev.ru 2881 [ - + ]:CBC 76 : else if (cxt->inhRelations)
2882 : : {
2883 : : /* try inherited tables */
2884 : : ListCell *inher;
2885 : :
2950 teodor@sigaev.ru 2886 [ # # # # :UBC 0 : foreach(inher, cxt->inhRelations)
# # ]
2887 : : {
2888 : 0 : RangeVar *inh = lfirst_node(RangeVar, inher);
2889 : : Relation rel;
2890 : : int count;
2891 : :
2661 andres@anarazel.de 2892 : 0 : rel = table_openrv(inh, AccessShareLock);
2893 : : /* check user requested inheritance from valid relkind */
2950 teodor@sigaev.ru 2894 [ # # ]: 0 : if (rel->rd_rel->relkind != RELKIND_RELATION &&
2895 [ # # ]: 0 : rel->rd_rel->relkind != RELKIND_FOREIGN_TABLE &&
2896 [ # # ]: 0 : rel->rd_rel->relkind != RELKIND_PARTITIONED_TABLE)
2897 [ # # ]: 0 : ereport(ERROR,
2898 : : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
2899 : : errmsg("inherited relation \"%s\" is not a table or foreign table",
2900 : : inh->relname)));
2901 [ # # ]: 0 : for (count = 0; count < rel->rd_att->natts; count++)
2902 : : {
2903 : 0 : Form_pg_attribute inhattr = TupleDescAttr(rel->rd_att,
2904 : : count);
2905 : 0 : char *inhname = NameStr(inhattr->attname);
2906 : :
2907 [ # # ]: 0 : if (inhattr->attisdropped)
2908 : 0 : continue;
2909 [ # # ]: 0 : if (strcmp(key, inhname) == 0)
2910 : : {
2911 : 0 : found = true;
2912 : 0 : break;
2913 : : }
2914 : : }
2661 andres@anarazel.de 2915 : 0 : table_close(rel, NoLock);
2950 teodor@sigaev.ru 2916 [ # # ]: 0 : if (found)
2917 : 0 : break;
2918 : : }
2919 : : }
2920 : : }
2921 : :
2922 : : /*
2923 : : * In the ALTER TABLE case, don't complain about index keys not
2924 : : * created in the command; they may well exist already. DefineIndex
2925 : : * will complain about them if not.
2926 : : */
6867 neilc@samurai.com 2927 [ + + - + ]:CBC 192 : if (!found && !cxt->isalter)
6867 neilc@samurai.com 2928 [ # # ]:UBC 0 : ereport(ERROR,
2929 : : (errcode(ERRCODE_UNDEFINED_COLUMN),
2930 : : errmsg("column \"%s\" named in key does not exist", key),
2931 : : parser_errposition(cxt->pstate, constraint->location)));
2932 : :
2933 : : /* OK, add it to the index definition */
6867 neilc@samurai.com 2934 :CBC 192 : iparam = makeNode(IndexElem);
2935 : 192 : iparam->name = pstrdup(key);
2936 : 192 : iparam->expr = NULL;
5977 tgl@sss.pgh.pa.us 2937 : 192 : iparam->indexcolname = NULL;
5519 2938 : 192 : iparam->collation = NIL;
6867 neilc@samurai.com 2939 : 192 : iparam->opclass = NIL;
2227 akorotkov@postgresql 2940 : 192 : iparam->opclassopts = NIL;
121 tgl@sss.pgh.pa.us 2941 :GNC 192 : iparam->location = -1;
2950 teodor@sigaev.ru 2942 :CBC 192 : index->indexIncludingParams = lappend(index->indexIncludingParams, iparam);
2943 : : }
2944 : :
6867 neilc@samurai.com 2945 : 12494 : return index;
2946 : : }
2947 : :
2948 : : /*
2949 : : * transformCheckConstraints
2950 : : * handle CHECK constraints
2951 : : *
2952 : : * Right now, there's nothing to do here when called from ALTER TABLE,
2953 : : * but the other constraint-transformation functions are called in both
2954 : : * the CREATE TABLE and ALTER TABLE paths, so do the same here, and just
2955 : : * don't do anything if we're not authorized to skip validation.
2956 : : */
2957 : : static void
3793 rhaas@postgresql.org 2958 : 42044 : transformCheckConstraints(CreateStmtContext *cxt, bool skipValidation)
2959 : : {
2960 : : ListCell *ckclist;
2961 : :
2962 [ + + ]: 42044 : if (cxt->ckconstraints == NIL)
2963 : 40774 : return;
2964 : :
2965 : : /*
2966 : : * When creating a new table (but not a foreign table), we can safely skip
2967 : : * the validation of check constraints and mark them as valid based on the
2968 : : * constraint enforcement flag, since NOT ENFORCED constraints must always
2969 : : * be marked as NOT VALID. (This will override any user-supplied NOT VALID
2970 : : * flag.)
2971 : : */
2972 [ + + ]: 1270 : if (skipValidation)
2973 : : {
2974 [ + - + + : 1082 : foreach(ckclist, cxt->ckconstraints)
+ + ]
2975 : : {
2976 : 592 : Constraint *constraint = (Constraint *) lfirst(ckclist);
2977 : :
2978 : 592 : constraint->skip_validation = true;
479 peter@eisentraut.org 2979 : 592 : constraint->initially_valid = constraint->is_enforced;
2980 : : }
2981 : : }
2982 : : }
2983 : :
2984 : : /*
2985 : : * transformFKConstraints
2986 : : * handle FOREIGN KEY constraints
2987 : : */
2988 : : static void
5579 tgl@sss.pgh.pa.us 2989 : 42044 : transformFKConstraints(CreateStmtContext *cxt,
2990 : : bool skipValidation, bool isAddConstraint)
2991 : : {
2992 : : ListCell *fkclist;
2993 : :
6891 2994 [ + + ]: 42044 : if (cxt->fkconstraints == NIL)
2995 : 39293 : return;
2996 : :
2997 : : /*
2998 : : * If CREATE TABLE or adding a column with NULL default, we can safely
2999 : : * skip validation of FK constraints, and mark them as valid based on the
3000 : : * constraint enforcement flag, since NOT ENFORCED constraints must always
3001 : : * be marked as NOT VALID. (This will override any user-supplied NOT VALID
3002 : : * flag.)
3003 : : */
3004 [ + + ]: 2751 : if (skipValidation)
3005 : : {
3006 [ + - + + : 1922 : foreach(fkclist, cxt->fkconstraints)
+ + ]
3007 : : {
6123 3008 : 999 : Constraint *constraint = (Constraint *) lfirst(fkclist);
3009 : :
3010 : 999 : constraint->skip_validation = true;
398 peter@eisentraut.org 3011 : 999 : constraint->initially_valid = constraint->is_enforced;
3012 : : }
3013 : : }
3014 : :
3015 : : /*
3016 : : * For CREATE TABLE or ALTER TABLE ADD COLUMN, gin up an ALTER TABLE ADD
3017 : : * CONSTRAINT command to execute after the basic command is complete. (If
3018 : : * called from ADD CONSTRAINT, that routine will add the FK constraints to
3019 : : * its own subcommand list.)
3020 : : *
3021 : : * Note: the ADD CONSTRAINT command must also execute after any index
3022 : : * creation commands. Thus, this should run after
3023 : : * transformIndexConstraints, so that the CREATE INDEX commands are
3024 : : * already in cxt->alist. See also the handling of cxt->likeclauses.
3025 : : */
6891 tgl@sss.pgh.pa.us 3026 [ + + ]: 2751 : if (!isAddConstraint)
3027 : : {
3028 : 919 : AlterTableStmt *alterstmt = makeNode(AlterTableStmt);
3029 : :
3030 : 919 : alterstmt->relation = cxt->relation;
3031 : 919 : alterstmt->cmds = NIL;
2124 michael@paquier.xyz 3032 : 919 : alterstmt->objtype = OBJECT_TABLE;
3033 : :
6891 tgl@sss.pgh.pa.us 3034 [ + - + + : 1914 : foreach(fkclist, cxt->fkconstraints)
+ + ]
3035 : : {
6123 3036 : 995 : Constraint *constraint = (Constraint *) lfirst(fkclist);
6891 3037 : 995 : AlterTableCmd *altercmd = makeNode(AlterTableCmd);
3038 : :
2302 3039 : 995 : altercmd->subtype = AT_AddConstraint;
6891 3040 : 995 : altercmd->name = NULL;
6123 3041 : 995 : altercmd->def = (Node *) constraint;
6891 3042 : 995 : alterstmt->cmds = lappend(alterstmt->cmds, altercmd);
3043 : : }
3044 : :
3045 : 919 : cxt->alist = lappend(cxt->alist, alterstmt);
3046 : : }
3047 : : }
3048 : :
3049 : : /*
3050 : : * transformIndexStmt - parse analysis for CREATE INDEX and ALTER TABLE
3051 : : *
3052 : : * Note: this is a no-op for an index not using either index expressions or
3053 : : * a predicate expression. There are several code paths that create indexes
3054 : : * without bothering to call this, because they know they don't have any
3055 : : * such expressions to deal with.
3056 : : *
3057 : : * To avoid race conditions, it's important that this function rely only on
3058 : : * the passed-in relid (and not on stmt->relation) to determine the target
3059 : : * relation.
3060 : : */
3061 : : IndexStmt *
4460 rhaas@postgresql.org 3062 : 16878 : transformIndexStmt(Oid relid, IndexStmt *stmt, const char *queryString)
3063 : : {
3064 : : ParseState *pstate;
3065 : : ParseNamespaceItem *nsitem;
3066 : : ListCell *l;
3067 : : Relation rel;
3068 : :
3069 : : /* Nothing to do if statement already transformed. */
4090 tgl@sss.pgh.pa.us 3070 [ + + ]: 16878 : if (stmt->transformed)
3071 : 86 : return stmt;
3072 : :
3073 : : /* Set up pstate */
6891 3074 : 16792 : pstate = make_parsestate(NULL);
3075 : 16792 : pstate->p_sourcetext = queryString;
3076 : :
3077 : : /*
3078 : : * Put the parent table into the rtable so that the expressions can refer
3079 : : * to its fields without qualification. Caller is responsible for locking
3080 : : * relation, but we still need to open it.
3081 : : */
4460 rhaas@postgresql.org 3082 : 16792 : rel = relation_open(relid, NoLock);
2315 tgl@sss.pgh.pa.us 3083 : 16792 : nsitem = addRangeTableEntryForRelation(pstate, rel,
3084 : : AccessShareLock,
3085 : : NULL, false, true);
3086 : :
3087 : : /* no to join list, yes to namespaces */
3088 : 16792 : addNSItemToQuery(pstate, nsitem, false, true, true);
3089 : :
3090 : : /* take care of the where clause */
6891 3091 [ + + ]: 16792 : if (stmt->whereClause)
3092 : : {
3093 : 262 : stmt->whereClause = transformWhereClause(pstate,
3094 : : stmt->whereClause,
3095 : : EXPR_KIND_INDEX_PREDICATE,
3096 : : "WHERE");
3097 : : /* we have to fix its collations too */
5507 3098 : 262 : assign_expr_collations(pstate, stmt->whereClause);
3099 : : }
3100 : :
3101 : : /* take care of any index expressions */
6891 3102 [ + - + + : 40055 : foreach(l, stmt->indexParams)
+ + ]
3103 : : {
3104 : 23287 : IndexElem *ielem = (IndexElem *) lfirst(l);
3105 : :
3106 [ + + ]: 23287 : if (ielem->expr)
3107 : : {
3108 : : /* Extract preliminary index col name before transforming expr */
5977 3109 [ + - ]: 811 : if (ielem->indexcolname == NULL)
3110 : 811 : ielem->indexcolname = FigureIndexColname(ielem->expr);
3111 : :
3112 : : /* Now do parse transformation of the expression */
5016 3113 : 811 : ielem->expr = transformExpr(pstate, ielem->expr,
3114 : : EXPR_KIND_INDEX_EXPRESSION);
3115 : :
3116 : : /* We have to fix its collations too */
5526 3117 : 787 : assign_expr_collations(pstate, ielem->expr);
3118 : :
3119 : : /*
3120 : : * transformExpr() should have already rejected subqueries,
3121 : : * aggregates, window functions, and SRFs, based on the EXPR_KIND_
3122 : : * for an index expression.
3123 : : *
3124 : : * DefineIndex() will make more checks.
3125 : : */
3126 : : }
3127 : : }
3128 : :
3129 : : /*
3130 : : * Check that only the base rel is mentioned. (This should be dead code
3131 : : * now that add_missing_from is history.)
3132 : : */
6891 3133 [ - + ]: 16768 : if (list_length(pstate->p_rtable) != 1)
6891 tgl@sss.pgh.pa.us 3134 [ # # ]:UBC 0 : ereport(ERROR,
3135 : : (errcode(ERRCODE_INVALID_COLUMN_REFERENCE),
3136 : : errmsg("index expressions and predicates can refer only to the table being indexed")));
3137 : :
6891 tgl@sss.pgh.pa.us 3138 :CBC 16768 : free_parsestate(pstate);
3139 : :
3140 : : /* Close relation */
2661 andres@anarazel.de 3141 : 16768 : table_close(rel, NoLock);
3142 : :
3143 : : /* Mark statement as successfully transformed */
4090 tgl@sss.pgh.pa.us 3144 : 16768 : stmt->transformed = true;
3145 : :
6891 3146 : 16768 : return stmt;
3147 : : }
3148 : :
3149 : : /*
3150 : : * transformStatsStmt - parse analysis for CREATE STATISTICS
3151 : : *
3152 : : * To avoid race conditions, it's important that this function relies only on
3153 : : * the passed-in relid (and not on stmt->relation) to determine the target
3154 : : * relation.
3155 : : */
3156 : : CreateStatsStmt *
1866 tomas.vondra@postgre 3157 : 665 : transformStatsStmt(Oid relid, CreateStatsStmt *stmt, const char *queryString)
3158 : : {
3159 : : ParseState *pstate;
3160 : : ParseNamespaceItem *nsitem;
3161 : : ListCell *l;
3162 : : Relation rel;
3163 : :
3164 : : /* Nothing to do if statement already transformed. */
3165 [ + + ]: 665 : if (stmt->transformed)
3166 : 40 : return stmt;
3167 : :
3168 : : /* Set up pstate */
3169 : 625 : pstate = make_parsestate(NULL);
3170 : 625 : pstate->p_sourcetext = queryString;
3171 : :
3172 : : /*
3173 : : * Put the parent table into the rtable so that the expressions can refer
3174 : : * to its fields without qualification. Caller is responsible for locking
3175 : : * relation, but we still need to open it.
3176 : : */
3177 : 625 : rel = relation_open(relid, NoLock);
3178 : 625 : nsitem = addRangeTableEntryForRelation(pstate, rel,
3179 : : AccessShareLock,
3180 : : NULL, false, true);
3181 : :
3182 : : /* no to join list, yes to namespaces */
3183 : 625 : addNSItemToQuery(pstate, nsitem, false, true, true);
3184 : :
3185 : : /* take care of any expressions */
3186 [ + - + + : 2151 : foreach(l, stmt->exprs)
+ + ]
3187 : : {
3188 : 1526 : StatsElem *selem = (StatsElem *) lfirst(l);
3189 : :
3190 [ + + ]: 1526 : if (selem->expr)
3191 : : {
3192 : : /* Now do parse transformation of the expression */
3193 : 423 : selem->expr = transformExpr(pstate, selem->expr,
3194 : : EXPR_KIND_STATS_EXPRESSION);
3195 : :
3196 : : /* We have to fix its collations too */
3197 : 423 : assign_expr_collations(pstate, selem->expr);
3198 : : }
3199 : : }
3200 : :
3201 : : /*
3202 : : * Check that only the base rel is mentioned. (This should be dead code
3203 : : * now that add_missing_from is history.)
3204 : : */
3205 [ - + ]: 625 : if (list_length(pstate->p_rtable) != 1)
1866 tomas.vondra@postgre 3206 [ # # ]:UBC 0 : ereport(ERROR,
3207 : : (errcode(ERRCODE_INVALID_COLUMN_REFERENCE),
3208 : : errmsg("statistics expressions can refer only to the table being referenced")));
3209 : :
1866 tomas.vondra@postgre 3210 :CBC 625 : free_parsestate(pstate);
3211 : :
3212 : : /* Close relation */
3213 : 625 : table_close(rel, NoLock);
3214 : :
3215 : : /* Mark statement as successfully transformed */
3216 : 625 : stmt->transformed = true;
3217 : :
3218 : 625 : return stmt;
3219 : : }
3220 : :
3221 : :
3222 : : /*
3223 : : * transformRuleStmt -
3224 : : * transform a CREATE RULE Statement. The action is a list of parse
3225 : : * trees which is transformed into a list of query trees, and we also
3226 : : * transform the WHERE clause if any.
3227 : : *
3228 : : * actions and whereClause are output parameters that receive the
3229 : : * transformed results.
3230 : : */
3231 : : void
6891 tgl@sss.pgh.pa.us 3232 : 746 : transformRuleStmt(RuleStmt *stmt, const char *queryString,
3233 : : List **actions, Node **whereClause)
3234 : : {
3235 : : Relation rel;
3236 : : ParseState *pstate;
3237 : : ParseNamespaceItem *oldnsitem;
3238 : : ParseNamespaceItem *newnsitem;
3239 : :
3240 : : /*
3241 : : * To avoid deadlock, make sure the first thing we do is grab
3242 : : * AccessExclusiveLock on the target relation. This will be needed by
3243 : : * DefineQueryRewrite(), and we don't want to grab a lesser lock
3244 : : * beforehand.
3245 : : */
2661 andres@anarazel.de 3246 : 746 : rel = table_openrv(stmt->relation, AccessExclusiveLock);
3247 : :
4811 kgrittn@postgresql.o 3248 [ - + ]: 746 : if (rel->rd_rel->relkind == RELKIND_MATVIEW)
4811 kgrittn@postgresql.o 3249 [ # # ]:UBC 0 : ereport(ERROR,
3250 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3251 : : errmsg("rules on materialized views are not supported")));
3252 : :
3253 : : /* Set up pstate */
6891 tgl@sss.pgh.pa.us 3254 :CBC 746 : pstate = make_parsestate(NULL);
3255 : 746 : pstate->p_sourcetext = queryString;
3256 : :
3257 : : /*
3258 : : * NOTE: 'OLD' must always have a varno equal to 1 and 'NEW' equal to 2.
3259 : : * Set up their ParseNamespaceItems in the main pstate for use in parsing
3260 : : * the rule qualification.
3261 : : */
2315 3262 : 746 : oldnsitem = addRangeTableEntryForRelation(pstate, rel,
3263 : : AccessShareLock,
3264 : : makeAlias("old", NIL),
3265 : : false, false);
3266 : 746 : newnsitem = addRangeTableEntryForRelation(pstate, rel,
3267 : : AccessShareLock,
3268 : : makeAlias("new", NIL),
3269 : : false, false);
3270 : :
3271 : : /*
3272 : : * They must be in the namespace too for lookup purposes, but only add the
3273 : : * one(s) that are relevant for the current kind of rule. In an UPDATE
3274 : : * rule, quals must refer to OLD.field or NEW.field to be unambiguous, but
3275 : : * there's no need to be so picky for INSERT & DELETE. We do not add them
3276 : : * to the joinlist.
3277 : : */
6891 3278 [ + + + + : 746 : switch (stmt->event)
- ]
3279 : : {
3280 : 12 : case CMD_SELECT:
2315 3281 : 12 : addNSItemToQuery(pstate, oldnsitem, false, true, true);
6891 3282 : 12 : break;
3283 : 290 : case CMD_UPDATE:
2315 3284 : 290 : addNSItemToQuery(pstate, oldnsitem, false, true, true);
3285 : 290 : addNSItemToQuery(pstate, newnsitem, false, true, true);
6891 3286 : 290 : break;
3287 : 334 : case CMD_INSERT:
2315 3288 : 334 : addNSItemToQuery(pstate, newnsitem, false, true, true);
6891 3289 : 334 : break;
3290 : 110 : case CMD_DELETE:
2315 3291 : 110 : addNSItemToQuery(pstate, oldnsitem, false, true, true);
6891 3292 : 110 : break;
6891 tgl@sss.pgh.pa.us 3293 :UBC 0 : default:
3294 [ # # ]: 0 : elog(ERROR, "unrecognized event type: %d",
3295 : : (int) stmt->event);
3296 : : break;
3297 : : }
3298 : :
3299 : : /* take care of the where clause */
6891 tgl@sss.pgh.pa.us 3300 :CBC 746 : *whereClause = transformWhereClause(pstate,
3301 : : stmt->whereClause,
3302 : : EXPR_KIND_WHERE,
3303 : : "WHERE");
3304 : : /* we have to fix its collations too */
5507 3305 : 746 : assign_expr_collations(pstate, *whereClause);
3306 : :
3307 : : /* this is probably dead code without add_missing_from: */
3240 3308 [ - + ]: 746 : if (list_length(pstate->p_rtable) != 2) /* naughty, naughty... */
6891 tgl@sss.pgh.pa.us 3309 [ # # ]:UBC 0 : ereport(ERROR,
3310 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3311 : : errmsg("rule WHERE condition cannot contain references to other relations")));
3312 : :
3313 : : /*
3314 : : * 'instead nothing' rules with a qualification need a query rangetable so
3315 : : * the rewrite handler can add the negated rule qualification to the
3316 : : * original query. We create a query with the new command type CMD_NOTHING
3317 : : * here that is treated specially by the rewrite system.
3318 : : */
6891 tgl@sss.pgh.pa.us 3319 [ + + ]:CBC 746 : if (stmt->actions == NIL)
3320 : : {
3321 : 104 : Query *nothing_qry = makeNode(Query);
3322 : :
3323 : 104 : nothing_qry->commandType = CMD_NOTHING;
3324 : 104 : nothing_qry->rtable = pstate->p_rtable;
1246 alvherre@alvh.no-ip. 3325 : 104 : nothing_qry->rteperminfos = pstate->p_rteperminfos;
3240 tgl@sss.pgh.pa.us 3326 : 104 : nothing_qry->jointree = makeFromExpr(NIL, NULL); /* no join wanted */
3327 : :
6891 3328 : 104 : *actions = list_make1(nothing_qry);
3329 : : }
3330 : : else
3331 : : {
3332 : : ListCell *l;
3333 : 642 : List *newactions = NIL;
3334 : :
3335 : : /*
3336 : : * transform each statement, like parse_sub_analyze()
3337 : : */
3338 [ + - + + : 1302 : foreach(l, stmt->actions)
+ + ]
3339 : : {
3340 : 672 : Node *action = (Node *) lfirst(l);
3341 : 672 : ParseState *sub_pstate = make_parsestate(NULL);
3342 : : Query *sub_qry,
3343 : : *top_subqry;
3344 : : bool has_old,
3345 : : has_new;
3346 : :
3347 : : /*
3348 : : * Since outer ParseState isn't parent of inner, have to pass down
3349 : : * the query text by hand.
3350 : : */
3351 : 672 : sub_pstate->p_sourcetext = queryString;
3352 : :
3353 : : /*
3354 : : * Set up OLD/NEW in the rtable for this statement. The entries
3355 : : * are added only to relnamespace, not varnamespace, because we
3356 : : * don't want them to be referred to by unqualified field names
3357 : : * nor "*" in the rule actions. We decide later whether to put
3358 : : * them in the joinlist.
3359 : : */
2315 3360 : 672 : oldnsitem = addRangeTableEntryForRelation(sub_pstate, rel,
3361 : : AccessShareLock,
3362 : : makeAlias("old", NIL),
3363 : : false, false);
3364 : 672 : newnsitem = addRangeTableEntryForRelation(sub_pstate, rel,
3365 : : AccessShareLock,
3366 : : makeAlias("new", NIL),
3367 : : false, false);
3368 : 672 : addNSItemToQuery(sub_pstate, oldnsitem, false, true, false);
3369 : 672 : addNSItemToQuery(sub_pstate, newnsitem, false, true, false);
3370 : :
3371 : : /* Transform the rule action statement */
1782 3372 : 672 : top_subqry = transformStmt(sub_pstate, action);
3373 : :
3374 : : /*
3375 : : * We cannot support utility-statement actions (eg NOTIFY) with
3376 : : * nonempty rule WHERE conditions, because there's no way to make
3377 : : * the utility action execute conditionally.
3378 : : */
6891 3379 [ + + ]: 664 : if (top_subqry->commandType == CMD_UTILITY &&
3380 [ - + ]: 26 : *whereClause != NULL)
6891 tgl@sss.pgh.pa.us 3381 [ # # ]:UBC 0 : ereport(ERROR,
3382 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3383 : : errmsg("rules with WHERE conditions can only have SELECT, INSERT, UPDATE, or DELETE actions")));
3384 : :
3385 : : /*
3386 : : * If the action is INSERT...SELECT, OLD/NEW have been pushed down
3387 : : * into the SELECT, and that's what we need to look at. (Ugly
3388 : : * kluge ... try to fix this when we redesign querytrees.)
3389 : : */
6891 tgl@sss.pgh.pa.us 3390 :CBC 664 : sub_qry = getInsertSelectQuery(top_subqry, NULL);
3391 : :
3392 : : /*
3393 : : * If the sub_qry is a setop, we cannot attach any qualifications
3394 : : * to it, because the planner won't notice them. This could
3395 : : * perhaps be relaxed someday, but for now, we may as well reject
3396 : : * such a rule immediately.
3397 : : */
3398 [ - + - - ]: 664 : if (sub_qry->setOperations != NULL && *whereClause != NULL)
6891 tgl@sss.pgh.pa.us 3399 [ # # ]:UBC 0 : ereport(ERROR,
3400 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3401 : : errmsg("conditional UNION/INTERSECT/EXCEPT statements are not implemented")));
3402 : :
3403 : : /*
3404 : : * Validate action's use of OLD/NEW, qual too
3405 : : */
6891 tgl@sss.pgh.pa.us 3406 :CBC 664 : has_old =
3407 [ + + + + ]: 1076 : rangeTableEntry_used((Node *) sub_qry, PRS2_OLD_VARNO, 0) ||
3408 : 412 : rangeTableEntry_used(*whereClause, PRS2_OLD_VARNO, 0);
3409 : 664 : has_new =
3410 [ + + + + ]: 899 : rangeTableEntry_used((Node *) sub_qry, PRS2_NEW_VARNO, 0) ||
3411 : 235 : rangeTableEntry_used(*whereClause, PRS2_NEW_VARNO, 0);
3412 : :
3413 [ + + + + : 664 : switch (stmt->event)
- ]
3414 : : {
3415 : 12 : case CMD_SELECT:
3416 [ - + ]: 12 : if (has_old)
6891 tgl@sss.pgh.pa.us 3417 [ # # ]:UBC 0 : ereport(ERROR,
3418 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3419 : : errmsg("ON SELECT rule cannot use OLD")));
6891 tgl@sss.pgh.pa.us 3420 [ - + ]:CBC 12 : if (has_new)
6891 tgl@sss.pgh.pa.us 3421 [ # # ]:UBC 0 : ereport(ERROR,
3422 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3423 : : errmsg("ON SELECT rule cannot use NEW")));
6891 tgl@sss.pgh.pa.us 3424 :CBC 12 : break;
3425 : 232 : case CMD_UPDATE:
3426 : : /* both are OK */
3427 : 232 : break;
3428 : 307 : case CMD_INSERT:
3429 [ - + ]: 307 : if (has_old)
6891 tgl@sss.pgh.pa.us 3430 [ # # ]:UBC 0 : ereport(ERROR,
3431 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3432 : : errmsg("ON INSERT rule cannot use OLD")));
6891 tgl@sss.pgh.pa.us 3433 :CBC 307 : break;
3434 : 113 : case CMD_DELETE:
3435 [ - + ]: 113 : if (has_new)
6891 tgl@sss.pgh.pa.us 3436 [ # # ]:UBC 0 : ereport(ERROR,
3437 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3438 : : errmsg("ON DELETE rule cannot use NEW")));
6891 tgl@sss.pgh.pa.us 3439 :CBC 113 : break;
6891 tgl@sss.pgh.pa.us 3440 :UBC 0 : default:
3441 [ # # ]: 0 : elog(ERROR, "unrecognized event type: %d",
3442 : : (int) stmt->event);
3443 : : break;
3444 : : }
3445 : :
3446 : : /*
3447 : : * OLD/NEW are not allowed in WITH queries, because they would
3448 : : * amount to outer references for the WITH, which we disallow.
3449 : : * However, they were already in the outer rangetable when we
3450 : : * analyzed the query, so we have to check.
3451 : : *
3452 : : * Note that in the INSERT...SELECT case, we need to examine the
3453 : : * CTE lists of both top_subqry and sub_qry.
3454 : : *
3455 : : * Note that we aren't digging into the body of the query looking
3456 : : * for WITHs in nested sub-SELECTs. A WITH down there can
3457 : : * legitimately refer to OLD/NEW, because it'd be an
3458 : : * indirect-correlated outer reference.
3459 : : */
5681 tgl@sss.pgh.pa.us 3460 [ + + ]:CBC 664 : if (rangeTableEntry_used((Node *) top_subqry->cteList,
3461 [ - + ]: 660 : PRS2_OLD_VARNO, 0) ||
3462 : 660 : rangeTableEntry_used((Node *) sub_qry->cteList,
3463 : : PRS2_OLD_VARNO, 0))
3464 [ + - ]: 4 : ereport(ERROR,
3465 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3466 : : errmsg("cannot refer to OLD within WITH query")));
3467 [ + - ]: 660 : if (rangeTableEntry_used((Node *) top_subqry->cteList,
3468 [ - + ]: 660 : PRS2_NEW_VARNO, 0) ||
3469 : 660 : rangeTableEntry_used((Node *) sub_qry->cteList,
3470 : : PRS2_NEW_VARNO, 0))
5681 tgl@sss.pgh.pa.us 3471 [ # # ]:UBC 0 : ereport(ERROR,
3472 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3473 : : errmsg("cannot refer to NEW within WITH query")));
3474 : :
3475 : : /*
3476 : : * For efficiency's sake, add OLD to the rule action's jointree
3477 : : * only if it was actually referenced in the statement or qual.
3478 : : *
3479 : : * For INSERT, NEW is not really a relation (only a reference to
3480 : : * the to-be-inserted tuple) and should never be added to the
3481 : : * jointree.
3482 : : *
3483 : : * For UPDATE, we treat NEW as being another kind of reference to
3484 : : * OLD, because it represents references to *transformed* tuples
3485 : : * of the existing relation. It would be wrong to enter NEW
3486 : : * separately in the jointree, since that would cause a double
3487 : : * join of the updated relation. It's also wrong to fail to make
3488 : : * a jointree entry if only NEW and not OLD is mentioned.
3489 : : */
6891 tgl@sss.pgh.pa.us 3490 [ + + + + :CBC 660 : if (has_old || (has_new && stmt->event == CMD_UPDATE))
+ + ]
3491 : : {
3492 : : RangeTblRef *rtr;
3493 : :
3494 : : /*
3495 : : * If sub_qry is a setop, manipulating its jointree will do no
3496 : : * good at all, because the jointree is dummy. (This should be
3497 : : * a can't-happen case because of prior tests.)
3498 : : */
3499 [ - + ]: 279 : if (sub_qry->setOperations != NULL)
6891 tgl@sss.pgh.pa.us 3500 [ # # ]:UBC 0 : ereport(ERROR,
3501 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3502 : : errmsg("conditional UNION/INTERSECT/EXCEPT statements are not implemented")));
3503 : : /* hackishly add OLD to the already-built FROM clause */
2315 tgl@sss.pgh.pa.us 3504 :CBC 279 : rtr = makeNode(RangeTblRef);
3505 : 279 : rtr->rtindex = oldnsitem->p_rtindex;
3506 : 279 : sub_qry->jointree->fromlist =
3507 : 279 : lappend(sub_qry->jointree->fromlist, rtr);
3508 : : }
3509 : :
6891 3510 : 660 : newactions = lappend(newactions, top_subqry);
3511 : :
3512 : 660 : free_parsestate(sub_pstate);
3513 : : }
3514 : :
3515 : 630 : *actions = newactions;
3516 : : }
3517 : :
3518 : 734 : free_parsestate(pstate);
3519 : :
3520 : : /* Close relation, but keep the exclusive lock */
2661 andres@anarazel.de 3521 : 734 : table_close(rel, NoLock);
6891 tgl@sss.pgh.pa.us 3522 : 734 : }
3523 : :
3524 : :
3525 : : /*
3526 : : * checkPartition
3527 : : * Check whether partRelOid is a leaf partition of the parent table (rel).
3528 : : * isMerge: true indicates the operation is "ALTER TABLE ... MERGE PARTITIONS";
3529 : : * false indicates the operation is "ALTER TABLE ... SPLIT PARTITION".
3530 : : */
3531 : : static void
142 akorotkov@postgresql 3532 :GNC 597 : checkPartition(Relation rel, Oid partRelOid, bool isMerge)
3533 : : {
3534 : : Relation partRel;
3535 : :
3536 : 597 : partRel = table_open(partRelOid, NoLock);
3537 : :
3538 [ + + ]: 597 : if (partRel->rd_rel->relkind != RELKIND_RELATION)
3539 [ + - + - ]: 4 : ereport(ERROR,
3540 : : errcode(ERRCODE_WRONG_OBJECT_TYPE),
3541 : : errmsg("\"%s\" is not a table", RelationGetRelationName(partRel)),
3542 : : isMerge
3543 : : ? errhint("ALTER TABLE ... MERGE PARTITIONS can only merge partitions don't have sub-partitions.")
3544 : : : errhint("ALTER TABLE ... SPLIT PARTITION can only split partitions don't have sub-partitions."));
3545 : :
3546 [ + + ]: 593 : if (!partRel->rd_rel->relispartition)
3547 [ + - + - ]: 12 : ereport(ERROR,
3548 : : errcode(ERRCODE_WRONG_OBJECT_TYPE),
3549 : : errmsg("\"%s\" is not a partition of partitioned table \"%s\"",
3550 : : RelationGetRelationName(partRel), RelationGetRelationName(rel)),
3551 : : isMerge
3552 : : ? errhint("ALTER TABLE ... MERGE PARTITIONS can only merge partitions don't have sub-partitions.")
3553 : : : errhint("ALTER TABLE ... SPLIT PARTITION can only split partitions don't have sub-partitions."));
3554 : :
3555 [ + + ]: 581 : if (get_partition_parent(partRelOid, false) != RelationGetRelid(rel))
3556 [ + - + + ]: 12 : ereport(ERROR,
3557 : : errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
3558 : : errmsg("relation \"%s\" is not a partition of relation \"%s\"",
3559 : : RelationGetRelationName(partRel), RelationGetRelationName(rel)),
3560 : : isMerge
3561 : : ? errhint("ALTER TABLE ... MERGE PARTITIONS can only merge partitions don't have sub-partitions.")
3562 : : : errhint("ALTER TABLE ... SPLIT PARTITION can only split partitions don't have sub-partitions."));
3563 : :
3564 : 569 : table_close(partRel, NoLock);
3565 : 569 : }
3566 : :
3567 : : /*
3568 : : * transformPartitionCmdForSplit -
3569 : : * analyze the ALTER TABLE ... SPLIT PARTITION command
3570 : : *
3571 : : * For each new partition, sps->bound is set to the transformed value of bound.
3572 : : * Does checks for bounds of new partitions.
3573 : : */
3574 : : static void
3575 : 241 : transformPartitionCmdForSplit(CreateStmtContext *cxt, PartitionCmd *partcmd)
3576 : : {
3577 : 241 : Relation parent = cxt->rel;
3578 : : PartitionKey key;
3579 : : char strategy;
3580 : : Oid splitPartOid;
3581 : : Oid defaultPartOid;
3582 : 241 : int default_index = -1;
3583 : : bool isSplitPartDefault;
3584 : : ListCell *listptr,
3585 : : *listptr2;
3586 : : List *splitlist;
3587 : :
3588 : 241 : splitlist = partcmd->partlist;
3589 : 241 : key = RelationGetPartitionKey(parent);
3590 : 241 : strategy = get_partition_strategy(key);
3591 : 241 : defaultPartOid = get_default_oid_from_partdesc(RelationGetPartitionDesc(parent, true));
3592 : :
3593 : : /* Transform partition bounds for all partitions in the list: */
3594 [ + - + + : 1168 : foreach_node(SinglePartitionSpec, sps, splitlist)
+ + ]
3595 : : {
3596 : 694 : cxt->partbound = NULL;
3597 : 694 : transformPartitionCmd(cxt, sps->bound);
3598 : : /* Assign the transformed value of the partition bound. */
3599 : 690 : sps->bound = cxt->partbound;
3600 : : }
3601 : :
3602 : : /*
3603 : : * Open and lock the partition, check ownership along the way. We need to
3604 : : * use AccessExclusiveLock here because this split partition will be
3605 : : * detached, then dropped in ATExecSplitPartition.
3606 : : */
3607 : 237 : splitPartOid = RangeVarGetRelidExtended(partcmd->name, AccessExclusiveLock,
3608 : : 0, RangeVarCallbackOwnsRelation,
3609 : : NULL);
3610 : :
3611 : 229 : checkPartition(parent, splitPartOid, false);
3612 : :
3613 [ + + - ]: 225 : switch (strategy)
3614 : : {
3615 : 221 : case PARTITION_STRATEGY_LIST:
3616 : : case PARTITION_STRATEGY_RANGE:
3617 : : {
3618 [ + - + + : 1088 : foreach_node(SinglePartitionSpec, sps, splitlist)
+ + ]
3619 : : {
3620 [ + + ]: 654 : if (sps->bound->is_default)
3621 : : {
3622 [ + + ]: 56 : if (default_index != -1)
3623 [ + - ]: 4 : ereport(ERROR,
3624 : : errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3625 : : errmsg("DEFAULT partition should be one"),
3626 : : parser_errposition(cxt->pstate, sps->name->location));
3627 : :
3628 : 52 : default_index = foreach_current_index(sps);
3629 : : }
3630 : : }
3631 : : }
3632 : 217 : break;
3633 : :
3634 : 4 : case PARTITION_STRATEGY_HASH:
3635 [ + - ]: 4 : ereport(ERROR,
3636 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3637 : : errmsg("partition of hash-partitioned table cannot be split"));
3638 : : break;
3639 : :
142 akorotkov@postgresql 3640 :UNC 0 : default:
3641 [ # # ]: 0 : elog(ERROR, "unexpected partition strategy: %d",
3642 : : (int) key->strategy);
3643 : : break;
3644 : : }
3645 : :
3646 : : /* isSplitPartDefault: is the being split partition a DEFAULT partition? */
142 akorotkov@postgresql 3647 :GNC 217 : isSplitPartDefault = (defaultPartOid == splitPartOid);
3648 : :
3649 [ + + + + ]: 217 : if (isSplitPartDefault && default_index == -1)
3650 [ + - ]: 4 : ereport(ERROR,
3651 : : errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3652 : : errmsg("can not split DEFAULT partition \"%s\"",
3653 : : get_rel_name(splitPartOid)),
3654 : : errhint("To split DEFAULT partition one of the new partition must be DEFAULT."),
3655 : : parser_errposition(cxt->pstate, ((SinglePartitionSpec *) linitial(splitlist))->name->location));
3656 : :
3657 : : /*
3658 : : * If the partition being split is not the DEFAULT partition, but the
3659 : : * DEFAULT partition exists, then none of the resulting split partitions
3660 : : * can be the DEFAULT.
3661 : : */
3662 [ + + + + : 213 : if (!isSplitPartDefault && (default_index != -1) && OidIsValid(defaultPartOid))
+ + ]
3663 : : {
3664 : : SinglePartitionSpec *spsDef =
3665 : 4 : (SinglePartitionSpec *) list_nth(splitlist, default_index);
3666 : :
3667 [ + - ]: 4 : ereport(ERROR,
3668 : : errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3669 : : errmsg("can not split non-DEFAULT partition \"%s\"",
3670 : : get_rel_name(splitPartOid)),
3671 : : errmsg("new partition cannot be DEFAULT because DEFAULT partition \"%s\" already exists",
3672 : : get_rel_name(defaultPartOid)),
3673 : : parser_errposition(cxt->pstate, spsDef->name->location));
3674 : : }
3675 : :
3676 [ + - + + : 779 : foreach(listptr, splitlist)
+ + ]
3677 : : {
3678 : : Oid nspid;
3679 : 582 : SinglePartitionSpec *sps = (SinglePartitionSpec *) lfirst(listptr);
3680 : 582 : RangeVar *name = sps->name;
3681 : :
3682 : 582 : nspid = RangeVarGetCreationNamespace(sps->name);
3683 : :
3684 : : /* Partitions in the list should have different names. */
3685 [ + - + + : 1183 : for_each_cell(listptr2, splitlist, lnext(splitlist, listptr))
+ + ]
3686 : : {
3687 : : Oid nspid2;
3688 : 613 : SinglePartitionSpec *sps2 = (SinglePartitionSpec *) lfirst(listptr2);
3689 : 613 : RangeVar *name2 = sps2->name;
3690 : :
3691 [ + + ]: 613 : if (equal(name, name2))
3692 [ + - ]: 8 : ereport(ERROR,
3693 : : errcode(ERRCODE_DUPLICATE_TABLE),
3694 : : errmsg("partition with name \"%s\" is already used", name->relname),
3695 : : parser_errposition(cxt->pstate, name2->location));
3696 : :
3697 : 605 : nspid2 = RangeVarGetCreationNamespace(sps2->name);
3698 : :
3699 [ + + + + ]: 605 : if (nspid2 == nspid && strcmp(name->relname, name2->relname) == 0)
3700 [ + - ]: 4 : ereport(ERROR,
3701 : : errcode(ERRCODE_DUPLICATE_TABLE),
3702 : : errmsg("partition with name \"%s\" is already used", name->relname),
3703 : : parser_errposition(cxt->pstate, name2->location));
3704 : : }
3705 : : }
3706 : :
3707 : : /* Then we should check partitions with transformed bounds. */
3708 : 197 : check_partitions_for_split(parent, splitPartOid, splitlist, cxt->pstate);
3709 : 129 : }
3710 : :
3711 : :
3712 : : /*
3713 : : * transformPartitionCmdForMerge -
3714 : : * analyze the ALTER TABLE ... MERGE PARTITIONS command
3715 : : *
3716 : : * Does simple checks for merged partitions. Calculates bound of the resulting
3717 : : * partition.
3718 : : */
3719 : : static void
3720 : 176 : transformPartitionCmdForMerge(CreateStmtContext *cxt, PartitionCmd *partcmd)
3721 : : {
3722 : : Oid defaultPartOid;
3723 : : Oid partOid;
3724 : 176 : Relation parent = cxt->rel;
3725 : : PartitionKey key;
3726 : : char strategy;
3727 : : ListCell *listptr,
3728 : : *listptr2;
3729 : 176 : bool isDefaultPart = false;
3730 : 176 : List *partOids = NIL;
3731 : :
3732 : 176 : key = RelationGetPartitionKey(parent);
3733 : 176 : strategy = get_partition_strategy(key);
3734 : :
3735 [ + + ]: 176 : if (strategy == PARTITION_STRATEGY_HASH)
3736 [ + - ]: 4 : ereport(ERROR,
3737 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3738 : : errmsg("partition of hash-partitioned table cannot be merged"));
3739 : :
3740 : : /* Does the partitioned table (parent) have a default partition? */
3741 : 172 : defaultPartOid = get_default_oid_from_partdesc(RelationGetPartitionDesc(parent, true));
3742 : :
3743 [ + - + + : 516 : foreach(listptr, partcmd->partlist)
+ + ]
3744 : : {
3745 : 384 : RangeVar *name = (RangeVar *) lfirst(listptr);
3746 : :
3747 : : /* Partitions in the list should have different names. */
3748 [ + - + + : 668 : for_each_cell(listptr2, partcmd->partlist, lnext(partcmd->partlist, listptr))
+ + ]
3749 : : {
3750 : 288 : RangeVar *name2 = (RangeVar *) lfirst(listptr2);
3751 : :
3752 [ + + ]: 288 : if (equal(name, name2))
3753 [ + - ]: 4 : ereport(ERROR,
3754 : : errcode(ERRCODE_DUPLICATE_TABLE),
3755 : : errmsg("partition with name \"%s\" is already used", name->relname),
3756 : : parser_errposition(cxt->pstate, name2->location));
3757 : : }
3758 : :
3759 : : /*
3760 : : * Search the DEFAULT partition in the list. Open and lock partitions
3761 : : * before calculating the boundary for resulting partition, we also
3762 : : * check for ownership along the way. We need to use
3763 : : * AccessExclusiveLock here, because these merged partitions will be
3764 : : * detached and then dropped in ATExecMergePartitions.
3765 : : */
3766 : 380 : partOid = RangeVarGetRelidExtended(name, AccessExclusiveLock, 0,
3767 : : RangeVarCallbackOwnsRelation,
3768 : : NULL);
3769 : : /* Is the current partition a DEFAULT partition? */
3770 [ + + ]: 372 : if (partOid == defaultPartOid)
3771 : 4 : isDefaultPart = true;
3772 : :
3773 : : /*
3774 : : * Extended check because the same partition can have different names
3775 : : * (for example, "part_name" and "public.part_name").
3776 : : */
3777 [ + + + + : 624 : foreach(listptr2, partOids)
+ + ]
3778 : : {
3779 : 256 : Oid curOid = lfirst_oid(listptr2);
3780 : :
3781 [ + + ]: 256 : if (curOid == partOid)
3782 [ + - ]: 4 : ereport(ERROR,
3783 : : errcode(ERRCODE_DUPLICATE_TABLE),
3784 : : errmsg("partition with name \"%s\" is already used", name->relname),
3785 : : parser_errposition(cxt->pstate, name->location));
3786 : : }
3787 : :
3788 : 368 : checkPartition(parent, partOid, true);
3789 : :
3790 : 344 : partOids = lappend_oid(partOids, partOid);
3791 : : }
3792 : :
3793 : : /* Allocate the bound of the resulting partition. */
3794 [ - + ]: 132 : Assert(partcmd->bound == NULL);
3795 : 132 : partcmd->bound = makeNode(PartitionBoundSpec);
3796 : :
3797 : : /* Fill the partition bound. */
3798 : 132 : partcmd->bound->strategy = strategy;
3799 : 132 : partcmd->bound->location = -1;
3800 : 132 : partcmd->bound->is_default = isDefaultPart;
3801 [ + + ]: 132 : if (!isDefaultPart)
3802 : 128 : calculate_partition_bound_for_merge(parent, partcmd->partlist,
3803 : : partOids, partcmd->bound,
3804 : : cxt->pstate);
3805 : 124 : }
3806 : :
3807 : : /*
3808 : : * transformAlterTableStmt -
3809 : : * parse analysis for ALTER TABLE
3810 : : *
3811 : : * Returns the transformed AlterTableStmt. There may be additional actions
3812 : : * to be done before and after the transformed statement, which are returned
3813 : : * in *beforeStmts and *afterStmts as lists of utility command parsetrees.
3814 : : *
3815 : : * To avoid race conditions, it's important that this function rely only on
3816 : : * the passed-in relid (and not on stmt->relation) to determine the target
3817 : : * relation.
3818 : : */
3819 : : AlterTableStmt *
4460 rhaas@postgresql.org 3820 :CBC 16073 : transformAlterTableStmt(Oid relid, AlterTableStmt *stmt,
3821 : : const char *queryString,
3822 : : List **beforeStmts, List **afterStmts)
3823 : : {
3824 : : Relation rel;
3825 : : TupleDesc tupdesc;
3826 : : ParseState *pstate;
3827 : : CreateStmtContext cxt;
3828 : : List *save_alist;
3829 : : ListCell *lcmd,
3830 : : *l;
6891 tgl@sss.pgh.pa.us 3831 : 16073 : List *newcmds = NIL;
3832 : 16073 : bool skipValidation = true;
3833 : : AlterTableCmd *newcmd;
3834 : : ParseNamespaceItem *nsitem;
3835 : :
3836 : : /* Caller is responsible for locking the relation */
4460 rhaas@postgresql.org 3837 : 16073 : rel = relation_open(relid, NoLock);
2751 peter_e@gmx.net 3838 : 16073 : tupdesc = RelationGetDescr(rel);
3839 : :
3840 : : /* Set up pstate */
6891 tgl@sss.pgh.pa.us 3841 : 16073 : pstate = make_parsestate(NULL);
3842 : 16073 : pstate->p_sourcetext = queryString;
2315 3843 : 16073 : nsitem = addRangeTableEntryForRelation(pstate,
3844 : : rel,
3845 : : AccessShareLock,
3846 : : NULL,
3847 : : false,
3848 : : true);
3849 : 16073 : addNSItemToQuery(pstate, nsitem, false, true, true);
3850 : :
3851 : : /* Set up CreateStmtContext */
5579 3852 : 16073 : cxt.pstate = pstate;
2302 3853 [ + + ]: 16073 : if (rel->rd_rel->relkind == RELKIND_FOREIGN_TABLE)
3854 : : {
4802 3855 : 120 : cxt.stmtType = "ALTER FOREIGN TABLE";
3856 : 120 : cxt.isforeign = true;
3857 : : }
3858 : : else
3859 : : {
3860 : 15953 : cxt.stmtType = "ALTER TABLE";
3861 : 15953 : cxt.isforeign = false;
3862 : : }
6891 3863 : 16073 : cxt.relation = stmt->relation;
3864 : 16073 : cxt.rel = rel;
3865 : 16073 : cxt.inhRelations = NIL;
3866 : 16073 : cxt.isalter = true;
3867 : 16073 : cxt.columns = NIL;
3868 : 16073 : cxt.ckconstraints = NIL;
543 alvherre@alvh.no-ip. 3869 : 16073 : cxt.nnconstraints = NIL;
6891 tgl@sss.pgh.pa.us 3870 : 16073 : cxt.fkconstraints = NIL;
3871 : 16073 : cxt.ixconstraints = NIL;
1993 3872 : 16073 : cxt.likeclauses = NIL;
6891 3873 : 16073 : cxt.blist = NIL;
3874 : 16073 : cxt.alist = NIL;
3875 : 16073 : cxt.pkey = NULL;
3436 rhaas@postgresql.org 3876 : 16073 : cxt.ispartitioned = (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE);
3877 : 16073 : cxt.partbound = NULL;
3070 peter_e@gmx.net 3878 : 16073 : cxt.ofType = false;
3879 : :
3880 : : /*
3881 : : * Transform ALTER subcommands that need it (most don't). These largely
3882 : : * re-use code from CREATE TABLE.
3883 : : */
6891 tgl@sss.pgh.pa.us 3884 [ + - + + : 31942 : foreach(lcmd, stmt->cmds)
+ + ]
3885 : : {
3886 : 16073 : AlterTableCmd *cmd = (AlterTableCmd *) lfirst(lcmd);
3887 : :
3888 [ + + + + : 16073 : switch (cmd->subtype)
+ + + +
- ]
3889 : : {
3890 : 1516 : case AT_AddColumn:
3891 : : {
3360 peter_e@gmx.net 3892 : 1516 : ColumnDef *def = castNode(ColumnDef, cmd->def);
3893 : :
5579 tgl@sss.pgh.pa.us 3894 : 1516 : transformColumnDefinition(&cxt, def);
3895 : :
3896 : : /*
3897 : : * If the column has a non-null default, we can't skip
3898 : : * validation of foreign keys.
3899 : : */
6585 3900 [ + + ]: 1512 : if (def->raw_default != NULL)
6891 3901 : 669 : skipValidation = false;
3902 : :
3903 : : /*
3904 : : * All constraints are processed in other ways. Remove the
3905 : : * original list
3906 : : */
3907 : 1512 : def->constraints = NIL;
3908 : :
6585 3909 : 1512 : newcmds = lappend(newcmds, cmd);
6891 3910 : 1512 : break;
3911 : : }
3912 : :
3913 : 10764 : case AT_AddConstraint:
3914 : :
3915 : : /*
3916 : : * The original AddConstraint cmd node doesn't go to newcmds
3917 : : */
3918 [ + - ]: 10764 : if (IsA(cmd->def, Constraint))
3919 : : {
5579 3920 : 10764 : transformTableConstraint(&cxt, (Constraint *) cmd->def);
6123 3921 [ + + ]: 10760 : if (((Constraint *) cmd->def)->contype == CONSTR_FOREIGN)
3922 : 1828 : skipValidation = false;
3923 : : }
3924 : : else
6891 tgl@sss.pgh.pa.us 3925 [ # # ]:UBC 0 : elog(ERROR, "unrecognized node type: %d",
3926 : : (int) nodeTag(cmd->def));
6891 tgl@sss.pgh.pa.us 3927 :CBC 10760 : break;
3928 : :
4050 alvherre@alvh.no-ip. 3929 : 949 : case AT_AlterColumnType:
3930 : : {
2302 tgl@sss.pgh.pa.us 3931 : 949 : ColumnDef *def = castNode(ColumnDef, cmd->def);
3932 : : AttrNumber attnum;
3933 : :
3934 : : /*
3935 : : * For ALTER COLUMN TYPE, transform the USING clause if
3936 : : * one was specified.
3937 : : */
4050 alvherre@alvh.no-ip. 3938 [ + + ]: 949 : if (def->raw_default)
3939 : : {
3940 : 167 : def->cooked_default =
3941 : 167 : transformExpr(pstate, def->raw_default,
3942 : : EXPR_KIND_ALTER_COL_TRANSFORM);
3943 : : }
3944 : :
3945 : : /*
3946 : : * For identity column, create ALTER SEQUENCE command to
3947 : : * change the data type of the sequence. Identity sequence
3948 : : * is associated with the top level partitioned table.
3949 : : * Hence ignore partitions.
3950 : : */
728 peter@eisentraut.org 3951 [ + + ]: 949 : if (!RelationGetForm(rel)->relispartition)
3952 : : {
3953 : 881 : attnum = get_attnum(relid, cmd->name);
3954 [ - + ]: 881 : if (attnum == InvalidAttrNumber)
728 peter@eisentraut.org 3955 [ # # ]:UBC 0 : ereport(ERROR,
3956 : : (errcode(ERRCODE_UNDEFINED_COLUMN),
3957 : : errmsg("column \"%s\" of relation \"%s\" does not exist",
3958 : : cmd->name, RelationGetRelationName(rel))));
3959 : :
728 peter@eisentraut.org 3960 [ + + ]:CBC 881 : if (attnum > 0 &&
3961 [ + + ]: 877 : TupleDescAttr(tupdesc, attnum - 1)->attidentity)
3962 : : {
3963 : 24 : Oid seq_relid = getIdentitySequence(rel, attnum, false);
3964 : 24 : Oid typeOid = typenameTypeId(pstate, def->typeName);
3965 : 24 : AlterSeqStmt *altseqstmt = makeNode(AlterSeqStmt);
3966 : :
3967 : : altseqstmt->sequence
3968 : 24 : = makeRangeVar(get_namespace_name(get_rel_namespace(seq_relid)),
3969 : : get_rel_name(seq_relid),
3970 : : -1);
3971 : 24 : altseqstmt->options = list_make1(makeDefElem("as",
3972 : : (Node *) makeTypeNameFromOid(typeOid, -1),
3973 : : -1));
3974 : 24 : altseqstmt->for_identity = true;
3975 : 24 : cxt.blist = lappend(cxt.blist, altseqstmt);
3976 : : }
3977 : : }
3978 : :
3316 peter_e@gmx.net 3979 : 949 : newcmds = lappend(newcmds, cmd);
3980 : 949 : break;
3981 : : }
3982 : :
3983 : 107 : case AT_AddIdentity:
3984 : : {
3275 bruce@momjian.us 3985 : 107 : Constraint *def = castNode(Constraint, cmd->def);
3986 : 107 : ColumnDef *newdef = makeNode(ColumnDef);
3987 : : AttrNumber attnum;
3988 : :
3316 peter_e@gmx.net 3989 : 107 : newdef->colname = cmd->name;
3990 : 107 : newdef->identity = def->generated_when;
3991 : 107 : cmd->def = (Node *) newdef;
3992 : :
3993 : 107 : attnum = get_attnum(relid, cmd->name);
2302 tgl@sss.pgh.pa.us 3994 [ + + ]: 107 : if (attnum == InvalidAttrNumber)
3995 [ + - ]: 4 : ereport(ERROR,
3996 : : (errcode(ERRCODE_UNDEFINED_COLUMN),
3997 : : errmsg("column \"%s\" of relation \"%s\" does not exist",
3998 : : cmd->name, RelationGetRelationName(rel))));
3999 : :
4000 : 103 : generateSerialExtraStmts(&cxt, newdef,
4001 : : get_atttype(relid, attnum),
4002 : : def->options, true, true,
4003 : : NULL, NULL);
4004 : :
3316 peter_e@gmx.net 4005 : 103 : newcmds = lappend(newcmds, cmd);
4006 : 103 : break;
4007 : : }
4008 : :
4009 : 41 : case AT_SetIdentity:
4010 : : {
4011 : : /*
4012 : : * Create an ALTER SEQUENCE statement for the internal
4013 : : * sequence of the identity column.
4014 : : */
4015 : : ListCell *lc;
4016 : 41 : List *newseqopts = NIL;
4017 : 41 : List *newdef = NIL;
4018 : : AttrNumber attnum;
4019 : : Oid seq_relid;
4020 : :
4021 : : /*
4022 : : * Split options into those handled by ALTER SEQUENCE and
4023 : : * those for ALTER TABLE proper.
4024 : : */
4025 [ + - + + : 122 : foreach(lc, castNode(List, cmd->def))
+ + ]
4026 : : {
3275 bruce@momjian.us 4027 : 81 : DefElem *def = lfirst_node(DefElem, lc);
4028 : :
3316 peter_e@gmx.net 4029 [ + + ]: 81 : if (strcmp(def->defname, "generated") == 0)
4030 : 29 : newdef = lappend(newdef, def);
4031 : : else
4032 : 52 : newseqopts = lappend(newseqopts, def);
4033 : : }
4034 : :
4035 : 41 : attnum = get_attnum(relid, cmd->name);
2302 tgl@sss.pgh.pa.us 4036 [ - + ]: 41 : if (attnum == InvalidAttrNumber)
2302 tgl@sss.pgh.pa.us 4037 [ # # ]:UBC 0 : ereport(ERROR,
4038 : : (errcode(ERRCODE_UNDEFINED_COLUMN),
4039 : : errmsg("column \"%s\" of relation \"%s\" does not exist",
4040 : : cmd->name, RelationGetRelationName(rel))));
4041 : :
728 peter@eisentraut.org 4042 :CBC 41 : seq_relid = getIdentitySequence(rel, attnum, true);
4043 : :
2302 tgl@sss.pgh.pa.us 4044 [ + + ]: 41 : if (seq_relid)
4045 : : {
4046 : : AlterSeqStmt *seqstmt;
4047 : :
4048 : 33 : seqstmt = makeNode(AlterSeqStmt);
4049 : 33 : seqstmt->sequence = makeRangeVar(get_namespace_name(get_rel_namespace(seq_relid)),
4050 : : get_rel_name(seq_relid), -1);
4051 : 33 : seqstmt->options = newseqopts;
4052 : 33 : seqstmt->for_identity = true;
4053 : 33 : seqstmt->missing_ok = false;
4054 : :
4055 : 33 : cxt.blist = lappend(cxt.blist, seqstmt);
4056 : : }
4057 : :
4058 : : /*
4059 : : * If column was not an identity column, we just let the
4060 : : * ALTER TABLE command error out later. (There are cases
4061 : : * this fails to cover, but we'll need to restructure
4062 : : * where creation of the sequence dependency linkage
4063 : : * happens before we can fix it.)
4064 : : */
4065 : :
3316 peter_e@gmx.net 4066 : 41 : cmd->def = (Node *) newdef;
4050 alvherre@alvh.no-ip. 4067 : 41 : newcmds = lappend(newcmds, cmd);
4068 : 41 : break;
4069 : : }
4070 : :
3436 rhaas@postgresql.org 4071 : 2267 : case AT_AttachPartition:
4072 : : case AT_DetachPartition:
4073 : : {
4074 : 2267 : PartitionCmd *partcmd = (PartitionCmd *) cmd->def;
4075 : :
142 akorotkov@postgresql 4076 :GNC 2267 : transformPartitionCmd(&cxt, partcmd->bound);
4077 : : /* assign the transformed value of the partition bound */
3436 rhaas@postgresql.org 4078 :CBC 2251 : partcmd->bound = cxt.partbound;
4079 : : }
4080 : :
4081 : 2251 : newcmds = lappend(newcmds, cmd);
4082 : 2251 : break;
4083 : :
142 akorotkov@postgresql 4084 :GNC 180 : case AT_MergePartitions:
4085 : : {
4086 : 180 : PartitionCmd *partcmd = (PartitionCmd *) cmd->def;
4087 : :
4088 [ + + ]: 180 : if (list_length(partcmd->partlist) < 2)
4089 [ + - ]: 4 : ereport(ERROR,
4090 : : errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
4091 : : errmsg("list of partitions to be merged should include at least two partitions"));
4092 : :
4093 : 176 : transformPartitionCmdForMerge(&cxt, partcmd);
4094 : 124 : newcmds = lappend(newcmds, cmd);
4095 : 124 : break;
4096 : : }
4097 : :
4098 : 249 : case AT_SplitPartition:
4099 : : {
4100 : 249 : PartitionCmd *partcmd = (PartitionCmd *) cmd->def;
4101 : :
4102 [ + + ]: 249 : if (list_length(partcmd->partlist) < 2)
4103 [ + - ]: 8 : ereport(ERROR,
4104 : : errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
4105 : : errmsg("list of new partitions should contain at least two partitions"));
4106 : :
4107 : 241 : transformPartitionCmdForSplit(&cxt, partcmd);
4108 : 129 : newcmds = lappend(newcmds, cmd);
4109 : 129 : break;
4110 : : }
4111 : :
6891 tgl@sss.pgh.pa.us 4112 :UBC 0 : default:
4113 : :
4114 : : /*
4115 : : * Currently, we shouldn't actually get here for the
4116 : : * subcommand types that don't require transformation; but if
4117 : : * we do, just emit them unchanged.
4118 : : */
4119 : 0 : newcmds = lappend(newcmds, cmd);
4120 : 0 : break;
4121 : : }
4122 : : }
4123 : :
4124 : : /*
4125 : : * Transfer anything we already have in cxt.alist into save_alist, to keep
4126 : : * it separate from the output of transformIndexConstraints.
4127 : : */
6891 tgl@sss.pgh.pa.us 4128 :CBC 15869 : save_alist = cxt.alist;
4129 : 15869 : cxt.alist = NIL;
4130 : :
4131 : : /* Postprocess constraints */
5579 4132 : 15869 : transformIndexConstraints(&cxt);
4133 : 15853 : transformFKConstraints(&cxt, skipValidation, true);
3793 rhaas@postgresql.org 4134 : 15853 : transformCheckConstraints(&cxt, false);
4135 : :
4136 : : /*
4137 : : * Push any index-creation commands into the ALTER, so that they can be
4138 : : * scheduled nicely by tablecmds.c. Note that tablecmds.c assumes that
4139 : : * the IndexStmt attached to an AT_AddIndex or AT_AddIndexConstraint
4140 : : * subcommand has already been through transformIndexStmt.
4141 : : */
6891 tgl@sss.pgh.pa.us 4142 [ + + + + : 23433 : foreach(l, cxt.alist)
+ + ]
4143 : : {
2569 4144 : 7580 : Node *istmt = (Node *) lfirst(l);
4145 : :
4146 : : /*
4147 : : * We assume here that cxt.alist contains only IndexStmts generated
4148 : : * from primary key constraints.
4149 : : */
4150 [ + - ]: 7580 : if (IsA(istmt, IndexStmt))
4151 : : {
4152 : 7580 : IndexStmt *idxstmt = (IndexStmt *) istmt;
4153 : :
4154 : 7580 : idxstmt = transformIndexStmt(relid, idxstmt, queryString);
4155 : 7580 : newcmd = makeNode(AlterTableCmd);
4156 [ + + ]: 7580 : newcmd->subtype = OidIsValid(idxstmt->indexOid) ? AT_AddIndexConstraint : AT_AddIndex;
4157 : 7580 : newcmd->def = (Node *) idxstmt;
4158 : 7580 : newcmds = lappend(newcmds, newcmd);
4159 : : }
4160 : : else
2569 tgl@sss.pgh.pa.us 4161 [ # # ]:UBC 0 : elog(ERROR, "unexpected stmt type %d", (int) nodeTag(istmt));
4162 : : }
6891 tgl@sss.pgh.pa.us 4163 :CBC 15853 : cxt.alist = NIL;
4164 : :
4165 : : /* Append any CHECK, NOT NULL or FK constraints to the commands list */
543 alvherre@alvh.no-ip. 4166 [ + + + + : 32474 : foreach_node(Constraint, def, cxt.ckconstraints)
+ + ]
4167 : : {
4168 : 768 : newcmd = makeNode(AlterTableCmd);
4169 : 768 : newcmd->subtype = AT_AddConstraint;
4170 : 768 : newcmd->def = (Node *) def;
4171 : 768 : newcmds = lappend(newcmds, newcmd);
4172 : : }
4173 [ + + + + : 37407 : foreach_node(Constraint, def, cxt.nnconstraints)
+ + ]
4174 : : {
6891 tgl@sss.pgh.pa.us 4175 : 5701 : newcmd = makeNode(AlterTableCmd);
4176 : 5701 : newcmd->subtype = AT_AddConstraint;
543 alvherre@alvh.no-ip. 4177 : 5701 : newcmd->def = (Node *) def;
984 4178 : 5701 : newcmds = lappend(newcmds, newcmd);
4179 : : }
543 4180 [ + + + + : 33538 : foreach_node(Constraint, def, cxt.fkconstraints)
+ + ]
4181 : : {
6891 tgl@sss.pgh.pa.us 4182 : 1832 : newcmd = makeNode(AlterTableCmd);
4183 : 1832 : newcmd->subtype = AT_AddConstraint;
543 alvherre@alvh.no-ip. 4184 : 1832 : newcmd->def = (Node *) def;
6891 tgl@sss.pgh.pa.us 4185 : 1832 : newcmds = lappend(newcmds, newcmd);
4186 : : }
4187 : :
4188 : : /* Close rel */
4189 : 15853 : relation_close(rel, NoLock);
4190 : :
4191 : : /*
4192 : : * Output results.
4193 : : */
4194 : 15853 : stmt->cmds = newcmds;
4195 : :
2302 4196 : 15853 : *beforeStmts = cxt.blist;
4197 : 15853 : *afterStmts = list_concat(cxt.alist, save_alist);
4198 : :
4199 : 15853 : return stmt;
4200 : : }
4201 : :
4202 : :
4203 : : /*
4204 : : * Preprocess a list of column constraint clauses
4205 : : * to attach constraint attributes to their primary constraint nodes
4206 : : * and detect inconsistent/misplaced constraint attributes.
4207 : : *
4208 : : * NOTE: currently, attributes are only supported for FOREIGN KEY, UNIQUE,
4209 : : * EXCLUSION, and PRIMARY KEY constraints, but someday they ought to be
4210 : : * supported for other constraint types.
4211 : : *
4212 : : * NOTE: this must be idempotent in non-error cases; see
4213 : : * transformCreateSchemaCreateTable.
4214 : : */
4215 : : static void
29 tgl@sss.pgh.pa.us 4216 :GNC 45202 : transformConstraintAttrs(ParseState *pstate, List *constraintList)
4217 : : {
6123 tgl@sss.pgh.pa.us 4218 :CBC 45202 : Constraint *lastprimarycon = NULL;
6891 4219 : 45202 : bool saw_deferrability = false;
4220 : 45202 : bool saw_initially = false;
479 peter@eisentraut.org 4221 : 45202 : bool saw_enforced = false;
4222 : : ListCell *clist;
4223 : :
4224 : : #define SUPPORTS_ATTRS(node) \
4225 : : ((node) != NULL && \
4226 : : ((node)->contype == CONSTR_PRIMARY || \
4227 : : (node)->contype == CONSTR_UNIQUE || \
4228 : : (node)->contype == CONSTR_EXCLUSION || \
4229 : : (node)->contype == CONSTR_FOREIGN))
4230 : :
6891 tgl@sss.pgh.pa.us 4231 [ + + + + : 57955 : foreach(clist, constraintList)
+ + ]
4232 : : {
6123 4233 : 12769 : Constraint *con = (Constraint *) lfirst(clist);
4234 : :
4235 [ - + ]: 12769 : if (!IsA(con, Constraint))
6123 tgl@sss.pgh.pa.us 4236 [ # # ]:UBC 0 : elog(ERROR, "unrecognized node type: %d",
4237 : : (int) nodeTag(con));
6123 tgl@sss.pgh.pa.us 4238 [ + + + + :CBC 12769 : switch (con->contype)
+ + + ]
4239 : : {
4240 : 83 : case CONSTR_ATTR_DEFERRABLE:
4241 [ + - + + : 83 : if (!SUPPORTS_ATTRS(lastprimarycon))
+ + + - -
+ ]
6123 tgl@sss.pgh.pa.us 4242 [ # # ]:UBC 0 : ereport(ERROR,
4243 : : (errcode(ERRCODE_SYNTAX_ERROR),
4244 : : errmsg("misplaced DEFERRABLE clause"),
4245 : : parser_errposition(pstate, con->location)));
6123 tgl@sss.pgh.pa.us 4246 [ - + ]:CBC 83 : if (saw_deferrability)
6123 tgl@sss.pgh.pa.us 4247 [ # # ]:UBC 0 : ereport(ERROR,
4248 : : (errcode(ERRCODE_SYNTAX_ERROR),
4249 : : errmsg("multiple DEFERRABLE/NOT DEFERRABLE clauses not allowed"),
4250 : : parser_errposition(pstate, con->location)));
6123 tgl@sss.pgh.pa.us 4251 :CBC 83 : saw_deferrability = true;
4252 : 83 : lastprimarycon->deferrable = true;
4253 : 83 : break;
4254 : :
6123 tgl@sss.pgh.pa.us 4255 :GBC 4 : case CONSTR_ATTR_NOT_DEFERRABLE:
4256 [ + - + - : 4 : if (!SUPPORTS_ATTRS(lastprimarycon))
+ - + - -
+ ]
6123 tgl@sss.pgh.pa.us 4257 [ # # ]:UBC 0 : ereport(ERROR,
4258 : : (errcode(ERRCODE_SYNTAX_ERROR),
4259 : : errmsg("misplaced NOT DEFERRABLE clause"),
4260 : : parser_errposition(pstate, con->location)));
6123 tgl@sss.pgh.pa.us 4261 [ - + ]:GBC 4 : if (saw_deferrability)
6123 tgl@sss.pgh.pa.us 4262 [ # # ]:UBC 0 : ereport(ERROR,
4263 : : (errcode(ERRCODE_SYNTAX_ERROR),
4264 : : errmsg("multiple DEFERRABLE/NOT DEFERRABLE clauses not allowed"),
4265 : : parser_errposition(pstate, con->location)));
6123 tgl@sss.pgh.pa.us 4266 :GBC 4 : saw_deferrability = true;
4267 : 4 : lastprimarycon->deferrable = false;
4268 [ - + ]: 4 : if (saw_initially &&
6123 tgl@sss.pgh.pa.us 4269 [ # # ]:UBC 0 : lastprimarycon->initdeferred)
4270 [ # # ]: 0 : ereport(ERROR,
4271 : : (errcode(ERRCODE_SYNTAX_ERROR),
4272 : : errmsg("constraint declared INITIALLY DEFERRED must be DEFERRABLE"),
4273 : : parser_errposition(pstate, con->location)));
6123 tgl@sss.pgh.pa.us 4274 :GBC 4 : break;
4275 : :
6123 tgl@sss.pgh.pa.us 4276 :CBC 65 : case CONSTR_ATTR_DEFERRED:
4277 [ + - + + : 65 : if (!SUPPORTS_ATTRS(lastprimarycon))
+ + + - -
+ ]
6123 tgl@sss.pgh.pa.us 4278 [ # # ]:UBC 0 : ereport(ERROR,
4279 : : (errcode(ERRCODE_SYNTAX_ERROR),
4280 : : errmsg("misplaced INITIALLY DEFERRED clause"),
4281 : : parser_errposition(pstate, con->location)));
6123 tgl@sss.pgh.pa.us 4282 [ - + ]:CBC 65 : if (saw_initially)
6123 tgl@sss.pgh.pa.us 4283 [ # # ]:UBC 0 : ereport(ERROR,
4284 : : (errcode(ERRCODE_SYNTAX_ERROR),
4285 : : errmsg("multiple INITIALLY IMMEDIATE/DEFERRED clauses not allowed"),
4286 : : parser_errposition(pstate, con->location)));
6123 tgl@sss.pgh.pa.us 4287 :CBC 65 : saw_initially = true;
4288 : 65 : lastprimarycon->initdeferred = true;
4289 : :
4290 : : /*
4291 : : * If only INITIALLY DEFERRED appears, assume DEFERRABLE
4292 : : */
4293 [ + + ]: 65 : if (!saw_deferrability)
4294 : 15 : lastprimarycon->deferrable = true;
4295 [ - + ]: 50 : else if (!lastprimarycon->deferrable)
6123 tgl@sss.pgh.pa.us 4296 [ # # ]:UBC 0 : ereport(ERROR,
4297 : : (errcode(ERRCODE_SYNTAX_ERROR),
4298 : : errmsg("constraint declared INITIALLY DEFERRED must be DEFERRABLE"),
4299 : : parser_errposition(pstate, con->location)));
6123 tgl@sss.pgh.pa.us 4300 :CBC 65 : break;
4301 : :
4302 : 8 : case CONSTR_ATTR_IMMEDIATE:
4303 [ + - + - : 8 : if (!SUPPORTS_ATTRS(lastprimarycon))
+ - + - -
+ ]
6123 tgl@sss.pgh.pa.us 4304 [ # # ]:UBC 0 : ereport(ERROR,
4305 : : (errcode(ERRCODE_SYNTAX_ERROR),
4306 : : errmsg("misplaced INITIALLY IMMEDIATE clause"),
4307 : : parser_errposition(pstate, con->location)));
6123 tgl@sss.pgh.pa.us 4308 [ - + ]:CBC 8 : if (saw_initially)
6123 tgl@sss.pgh.pa.us 4309 [ # # ]:UBC 0 : ereport(ERROR,
4310 : : (errcode(ERRCODE_SYNTAX_ERROR),
4311 : : errmsg("multiple INITIALLY IMMEDIATE/DEFERRED clauses not allowed"),
4312 : : parser_errposition(pstate, con->location)));
6123 tgl@sss.pgh.pa.us 4313 :CBC 8 : saw_initially = true;
4314 : 8 : lastprimarycon->initdeferred = false;
4315 : 8 : break;
4316 : :
479 peter@eisentraut.org 4317 : 28 : case CONSTR_ATTR_ENFORCED:
4318 [ + - ]: 28 : if (lastprimarycon == NULL ||
398 4319 [ + + ]: 28 : (lastprimarycon->contype != CONSTR_CHECK &&
4320 [ + + ]: 12 : lastprimarycon->contype != CONSTR_FOREIGN))
479 4321 [ + - ]: 4 : ereport(ERROR,
4322 : : (errcode(ERRCODE_SYNTAX_ERROR),
4323 : : errmsg("misplaced ENFORCED clause"),
4324 : : parser_errposition(pstate, con->location)));
4325 [ + + ]: 24 : if (saw_enforced)
4326 [ + - ]: 4 : ereport(ERROR,
4327 : : (errcode(ERRCODE_SYNTAX_ERROR),
4328 : : errmsg("multiple ENFORCED/NOT ENFORCED clauses not allowed"),
4329 : : parser_errposition(pstate, con->location)));
4330 : 20 : saw_enforced = true;
4331 : 20 : lastprimarycon->is_enforced = true;
4332 : 20 : break;
4333 : :
4334 : 53 : case CONSTR_ATTR_NOT_ENFORCED:
4335 [ + - ]: 53 : if (lastprimarycon == NULL ||
398 4336 [ + + ]: 53 : (lastprimarycon->contype != CONSTR_CHECK &&
4337 [ + + ]: 17 : lastprimarycon->contype != CONSTR_FOREIGN))
479 4338 [ + - ]: 4 : ereport(ERROR,
4339 : : (errcode(ERRCODE_SYNTAX_ERROR),
4340 : : errmsg("misplaced NOT ENFORCED clause"),
4341 : : parser_errposition(pstate, con->location)));
4342 [ + + ]: 49 : if (saw_enforced)
4343 [ + - ]: 4 : ereport(ERROR,
4344 : : (errcode(ERRCODE_SYNTAX_ERROR),
4345 : : errmsg("multiple ENFORCED/NOT ENFORCED clauses not allowed"),
4346 : : parser_errposition(pstate, con->location)));
4347 : 45 : saw_enforced = true;
4348 : 45 : lastprimarycon->is_enforced = false;
4349 : :
4350 : : /* A NOT ENFORCED constraint must be marked as invalid. */
4351 : 45 : lastprimarycon->skip_validation = true;
4352 : 45 : lastprimarycon->initially_valid = false;
4353 : 45 : break;
4354 : :
6123 tgl@sss.pgh.pa.us 4355 : 12528 : default:
4356 : : /* Otherwise it's not an attribute */
4357 : 12528 : lastprimarycon = con;
4358 : : /* reset flags for new primary node */
4359 : 12528 : saw_deferrability = false;
4360 : 12528 : saw_initially = false;
479 peter@eisentraut.org 4361 : 12528 : saw_enforced = false;
6123 tgl@sss.pgh.pa.us 4362 : 12528 : break;
4363 : : }
4364 : : }
6891 4365 : 45186 : }
4366 : :
4367 : : /*
4368 : : * Special handling of type definition for a column
4369 : : */
4370 : : static void
5579 4371 : 44691 : transformColumnType(CreateStmtContext *cxt, ColumnDef *column)
4372 : : {
4373 : : /*
4374 : : * All we really need to do here is verify that the type is valid,
4375 : : * including any collation spec that might be present.
4376 : : */
5536 4377 : 44691 : Type ctype = typenameType(cxt->pstate, column->typeName, NULL);
4378 : :
4379 [ + + ]: 44682 : if (column->collClause)
4380 : : {
4381 : 355 : Form_pg_type typtup = (Form_pg_type) GETSTRUCT(ctype);
4382 : :
5503 peter_e@gmx.net 4383 : 355 : LookupCollation(cxt->pstate,
5444 bruce@momjian.us 4384 : 355 : column->collClause->collname,
4385 : 355 : column->collClause->location);
4386 : : /* Complain if COLLATE is applied to an uncollatable type */
5536 tgl@sss.pgh.pa.us 4387 [ + + ]: 347 : if (!OidIsValid(typtup->typcollation))
4388 [ + - ]: 8 : ereport(ERROR,
4389 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
4390 : : errmsg("collations are not supported by type %s",
4391 : : format_type_be(typtup->oid)),
4392 : : parser_errposition(cxt->pstate,
4393 : : column->collClause->location)));
4394 : : }
4395 : :
6891 4396 : 44666 : ReleaseSysCache(ctype);
4397 : 44666 : }
4398 : :
4399 : :
4400 : : /*
4401 : : * transformCreateSchemaStmtElements -
4402 : : * analyzes the elements of a CREATE SCHEMA statement
4403 : : *
4404 : : * This presently has two responsibilities. We verify that no subcommands are
4405 : : * trying to create objects outside the new schema. We also pull out any
4406 : : * foreign-key constraint clauses embedded in CREATE TABLE subcommands, and
4407 : : * convert them to ALTER TABLE ADD CONSTRAINT commands appended to the list.
4408 : : * This supports forward references in foreign keys, which is required by the
4409 : : * SQL standard.
4410 : : *
4411 : : * We used to try to re-order the commands in a way that would work even if
4412 : : * the user-written order would not, but that's too hard (perhaps impossible)
4413 : : * to do correctly with not-yet-parse-analyzed commands. Now we'll just
4414 : : * execute the elements in the order given, except for foreign keys.
4415 : : *
4416 : : * "schemaName" is the name of the schema that will be used for the creation
4417 : : * of the objects listed. It may be obtained from the schema name defined
4418 : : * in the statement or a role specification.
4419 : : *
4420 : : * The result is a list of parse nodes that still need to be analyzed ---
4421 : : * but we can't analyze the later commands until we've executed the earlier
4422 : : * ones, because of possible inter-object references.
4423 : : *
4424 : : * Note it's important that we not modify the input data structure. We create
4425 : : * a new result List, and we copy any CREATE TABLE subcommands that we might
4426 : : * modify.
4427 : : */
4428 : : List *
29 tgl@sss.pgh.pa.us 4429 :GNC 710 : transformCreateSchemaStmtElements(ParseState *pstate, List *schemaElts,
4430 : : const char *schemaName)
4431 : : {
4432 : 710 : List *elements = NIL;
4433 : 710 : List *fk_elements = NIL;
4434 : : ListCell *lc;
4435 : :
4436 : : /*
4437 : : * Run through each schema element in the schema element list. Check
4438 : : * target schema names, and collect the list of actions to be done.
4439 : : */
4440 [ + + + + : 1150 : foreach(lc, schemaElts)
+ + ]
4441 : : {
4442 : 508 : Node *element = lfirst(lc);
4443 : :
6891 tgl@sss.pgh.pa.us 4444 [ + + + + :CBC 508 : switch (nodeTag(element))
+ + + + +
+ + + - ]
4445 : : {
4446 : 12 : case T_CreateSeqStmt:
4447 : : {
4448 : 12 : CreateSeqStmt *elp = (CreateSeqStmt *) element;
4449 : :
29 tgl@sss.pgh.pa.us 4450 :GNC 12 : checkSchemaNameRV(pstate, schemaName, elp->sequence);
29 tgl@sss.pgh.pa.us 4451 :UNC 0 : elements = lappend(elements, element);
4452 : : }
6891 tgl@sss.pgh.pa.us 4453 :UBC 0 : break;
4454 : :
6891 tgl@sss.pgh.pa.us 4455 :CBC 336 : case T_CreateStmt:
4456 : : {
4457 : 336 : CreateStmt *elp = (CreateStmt *) element;
4458 : :
29 tgl@sss.pgh.pa.us 4459 :GNC 336 : checkSchemaNameRV(pstate, schemaName, elp->relation);
4460 : : /* Pull out any foreign key clauses, add to fk_elements */
4461 : 324 : elp = transformCreateSchemaCreateTable(pstate,
4462 : : elp,
4463 : : &fk_elements);
4464 : 324 : elements = lappend(elements, elp);
4465 : : }
6891 tgl@sss.pgh.pa.us 4466 :CBC 324 : break;
4467 : :
4468 : 37 : case T_ViewStmt:
4469 : : {
4470 : 37 : ViewStmt *elp = (ViewStmt *) element;
4471 : :
29 tgl@sss.pgh.pa.us 4472 :GNC 37 : checkSchemaNameRV(pstate, schemaName, elp->view);
4473 : 21 : elements = lappend(elements, element);
4474 : : }
6891 tgl@sss.pgh.pa.us 4475 :CBC 21 : break;
4476 : :
4477 : 26 : case T_IndexStmt:
4478 : : {
4479 : 26 : IndexStmt *elp = (IndexStmt *) element;
4480 : :
29 tgl@sss.pgh.pa.us 4481 :GNC 26 : checkSchemaNameRV(pstate, schemaName, elp->relation);
4482 : 14 : elements = lappend(elements, element);
4483 : : }
6891 tgl@sss.pgh.pa.us 4484 :CBC 14 : break;
4485 : :
4486 : 12 : case T_CreateTrigStmt:
4487 : : {
4488 : 12 : CreateTrigStmt *elp = (CreateTrigStmt *) element;
4489 : :
29 tgl@sss.pgh.pa.us 4490 :GNC 12 : checkSchemaNameRV(pstate, schemaName, elp->relation);
29 tgl@sss.pgh.pa.us 4491 :UNC 0 : elements = lappend(elements, element);
4492 : : }
6891 4493 : 0 : break;
4494 : :
29 tgl@sss.pgh.pa.us 4495 :GNC 5 : case T_CreateDomainStmt:
4496 : : {
4497 : 5 : CreateDomainStmt *elp = (CreateDomainStmt *) element;
4498 : :
4499 : 5 : checkSchemaNameList(schemaName, elp->domainname);
4500 : 5 : elements = lappend(elements, element);
4501 : : }
4502 : 5 : break;
4503 : :
4504 : 22 : case T_CreateFunctionStmt:
4505 : : {
4506 : 22 : CreateFunctionStmt *elp = (CreateFunctionStmt *) element;
4507 : :
4508 : 22 : checkSchemaNameList(schemaName, elp->funcname);
4509 : 18 : elements = lappend(elements, element);
4510 : : }
4511 : 18 : break;
4512 : :
4513 : : /*
4514 : : * CREATE TYPE can produce a DefineStmt, but also
4515 : : * CreateEnumStmt, CreateRangeStmt, and CompositeTypeStmt.
4516 : : * Allowing DefineStmt also provides support for several other
4517 : : * commands: currently, CREATE AGGREGATE, CREATE COLLATION,
4518 : : * CREATE OPERATOR, and text search objects.
4519 : : */
4520 : :
4521 : 39 : case T_DefineStmt:
4522 : : {
4523 : 39 : DefineStmt *elp = (DefineStmt *) element;
4524 : :
4525 : 39 : checkSchemaNameList(schemaName, elp->defnames);
4526 : 39 : elements = lappend(elements, element);
4527 : : }
4528 : 39 : break;
4529 : :
4530 : 5 : case T_CreateEnumStmt:
4531 : : {
4532 : 5 : CreateEnumStmt *elp = (CreateEnumStmt *) element;
4533 : :
4534 : 5 : checkSchemaNameList(schemaName, elp->typeName);
4535 : 5 : elements = lappend(elements, element);
4536 : : }
4537 : 5 : break;
4538 : :
4539 : 5 : case T_CreateRangeStmt:
4540 : : {
4541 : 5 : CreateRangeStmt *elp = (CreateRangeStmt *) element;
4542 : :
4543 : 5 : checkSchemaNameList(schemaName, elp->typeName);
4544 : 5 : elements = lappend(elements, element);
4545 : : }
4546 : 5 : break;
4547 : :
4548 : 5 : case T_CompositeTypeStmt:
4549 : : {
4550 : 5 : CompositeTypeStmt *elp = (CompositeTypeStmt *) element;
4551 : :
4552 : 5 : checkSchemaNameRV(pstate, schemaName, elp->typevar);
4553 : 5 : elements = lappend(elements, element);
4554 : : }
29 tgl@sss.pgh.pa.us 4555 :GBC 5 : break;
4556 : :
6891 4557 : 4 : case T_GrantStmt:
29 tgl@sss.pgh.pa.us 4558 :GNC 4 : elements = lappend(elements, element);
6891 tgl@sss.pgh.pa.us 4559 :GBC 4 : break;
4560 : :
6891 tgl@sss.pgh.pa.us 4561 :UBC 0 : default:
4562 [ # # ]: 0 : elog(ERROR, "unrecognized node type: %d",
4563 : : (int) nodeTag(element));
4564 : : }
4565 : : }
4566 : :
29 tgl@sss.pgh.pa.us 4567 :GNC 642 : return list_concat(elements, fk_elements);
4568 : : }
4569 : :
4570 : : /*
4571 : : * checkSchemaNameRV
4572 : : * Check schema name in an element of a CREATE SCHEMA command,
4573 : : * where the element's name is given by a RangeVar
4574 : : *
4575 : : * It's okay if the command doesn't specify a target schema name, because
4576 : : * CreateSchemaCommand will set up the default creation schema to be the
4577 : : * new schema. But if a target schema name is given, it had better match.
4578 : : * We also have to check that the command doesn't say CREATE TEMP, since
4579 : : * that would likewise put the object into the wrong schema.
4580 : : */
4581 : : static void
4582 : 428 : checkSchemaNameRV(ParseState *pstate, const char *context_schema,
4583 : : RangeVar *relation)
4584 : : {
4585 [ + + ]: 428 : if (relation->schemaname != NULL &&
4586 [ + + ]: 84 : strcmp(context_schema, relation->schemaname) != 0)
6891 4587 [ + - ]: 60 : ereport(ERROR,
4588 : : (errcode(ERRCODE_INVALID_SCHEMA_DEFINITION),
4589 : : errmsg("CREATE specifies a schema (%s) "
4590 : : "different from the one being created (%s)",
4591 : : relation->schemaname, context_schema),
4592 : : parser_errposition(pstate, relation->location)));
4593 : :
29 4594 [ + + ]: 368 : if (relation->relpersistence == RELPERSISTENCE_TEMP)
4595 : : {
4596 : : /* spell this error the same as in RangeVarAdjustRelationPersistence */
4597 [ + - ]: 4 : ereport(ERROR,
4598 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
4599 : : errmsg("cannot create temporary relation in non-temporary schema"),
4600 : : parser_errposition(pstate, relation->location)));
4601 : : }
6891 tgl@sss.pgh.pa.us 4602 :GIC 364 : }
4603 : :
4604 : : /*
4605 : : * checkSchemaNameList
4606 : : * Check schema name in an element of a CREATE SCHEMA command,
4607 : : * where the element's name is given by a List
4608 : : *
4609 : : * Much as above, but we don't have to worry about TEMP.
4610 : : * Sadly, this also means we don't have a parse location to report.
4611 : : */
4612 : : static void
29 tgl@sss.pgh.pa.us 4613 :GNC 76 : checkSchemaNameList(const char *context_schema, List *qualified_name)
4614 : : {
4615 : : char *obj_schema;
4616 : : char *obj_name;
4617 : :
4618 : 76 : DeconstructQualifiedName(qualified_name, &obj_schema, &obj_name);
4619 [ + + ]: 76 : if (obj_schema != NULL &&
4620 [ + + ]: 8 : strcmp(context_schema, obj_schema) != 0)
29 tgl@sss.pgh.pa.us 4621 [ + - ]:CBC 4 : ereport(ERROR,
4622 : : (errcode(ERRCODE_INVALID_SCHEMA_DEFINITION),
4623 : : errmsg("CREATE specifies a schema (%s) "
4624 : : "different from the one being created (%s)",
4625 : : obj_schema, context_schema)));
29 tgl@sss.pgh.pa.us 4626 :GNC 72 : }
4627 : :
4628 : : /*
4629 : : * transformCreateSchemaCreateTable
4630 : : * Process one CreateStmt for transformCreateSchemaStmtElements.
4631 : : *
4632 : : * We remove any foreign-key clauses in the statement and convert them into
4633 : : * ALTER TABLE commands, which we append to *fk_elements.
4634 : : */
4635 : : static CreateStmt *
4636 : 324 : transformCreateSchemaCreateTable(ParseState *pstate,
4637 : : CreateStmt *stmt,
4638 : : List **fk_elements)
4639 : : {
4640 : : CreateStmt *newstmt;
4641 : 324 : List *newElts = NIL;
4642 : : ListCell *lc;
4643 : :
4644 : : /*
4645 : : * Flat-copy the CreateStmt node, allowing us to replace its tableElts
4646 : : * list without damaging the input data structure. Most sub-nodes will be
4647 : : * shared with the input, though.
4648 : : */
4649 : 324 : newstmt = makeNode(CreateStmt);
4650 : 324 : memcpy(newstmt, stmt, sizeof(CreateStmt));
4651 : :
4652 : : /* Scan for foreign-key constraints */
4653 [ + + + + : 699 : foreach(lc, stmt->tableElts)
+ + ]
4654 : : {
4655 : 375 : Node *element = lfirst(lc);
4656 : : AlterTableStmt *alterstmt;
4657 : : AlterTableCmd *altercmd;
4658 : :
4659 [ + + ]: 375 : if (IsA(element, Constraint))
4660 : : {
4661 : 59 : Constraint *constr = (Constraint *) element;
4662 : :
4663 [ + + ]: 59 : if (constr->contype != CONSTR_FOREIGN)
4664 : : {
4665 : : /* Other constraint types pass through unchanged */
4666 : 14 : newElts = lappend(newElts, constr);
4667 : 14 : continue;
4668 : : }
4669 : :
4670 : : /* Make it into an ALTER TABLE ADD CONSTRAINT command */
4671 : 45 : altercmd = makeNode(AlterTableCmd);
4672 : 45 : altercmd->subtype = AT_AddConstraint;
4673 : 45 : altercmd->name = NULL;
4674 : 45 : altercmd->def = (Node *) copyObject(constr);
4675 : :
4676 : 45 : alterstmt = makeNode(AlterTableStmt);
4677 : 45 : alterstmt->relation = copyObject(stmt->relation);
4678 : 45 : alterstmt->cmds = list_make1(altercmd);
4679 : 45 : alterstmt->objtype = OBJECT_TABLE;
4680 : :
4681 : 45 : *fk_elements = lappend(*fk_elements, alterstmt);
4682 : : }
4683 [ + - ]: 316 : else if (IsA(element, ColumnDef))
4684 : : {
4685 : 316 : ColumnDef *entry = (ColumnDef *) element;
4686 : : ColumnDef *newentry;
4687 : : List *entryconstraints;
4688 : 316 : bool afterFK = false;
4689 : :
4690 : : /*
4691 : : * We must preprocess the list of column constraints to attach
4692 : : * attributes such as DEFERRED to the appropriate constraint node.
4693 : : * Do this on a copy. (But execution of the CreateStmt will run
4694 : : * transformConstraintAttrs on the copy, so we are nonetheless
4695 : : * relying on transformConstraintAttrs to be idempotent.)
4696 : : */
4697 : 316 : entryconstraints = copyObject(entry->constraints);
4698 : 316 : transformConstraintAttrs(pstate, entryconstraints);
4699 : :
4700 : : /* Scan the column constraints ... */
4701 [ + + + + : 865 : foreach_node(Constraint, colconstr, entryconstraints)
+ + ]
4702 : : {
4703 [ + + + ]: 233 : switch (colconstr->contype)
4704 : : {
4705 : 38 : case CONSTR_FOREIGN:
4706 : : /* colconstr is already a copy, OK to modify */
4707 : 38 : colconstr->fk_attrs = list_make1(makeString(entry->colname));
4708 : :
4709 : : /* Make it into an ALTER TABLE ADD CONSTRAINT command */
4710 : 38 : altercmd = makeNode(AlterTableCmd);
4711 : 38 : altercmd->subtype = AT_AddConstraint;
4712 : 38 : altercmd->name = NULL;
4713 : 38 : altercmd->def = (Node *) colconstr;
4714 : :
4715 : 38 : alterstmt = makeNode(AlterTableStmt);
4716 : 38 : alterstmt->relation = copyObject(stmt->relation);
4717 : 38 : alterstmt->cmds = list_make1(altercmd);
4718 : 38 : alterstmt->objtype = OBJECT_TABLE;
4719 : :
4720 : 38 : *fk_elements = lappend(*fk_elements, alterstmt);
4721 : :
4722 : : /* Remove the Constraint node from entryconstraints */
4723 : 38 : entryconstraints =
4724 : 38 : foreach_delete_current(entryconstraints, colconstr);
4725 : :
4726 : : /*
4727 : : * Immediately-following attribute constraints should
4728 : : * be dropped, too.
4729 : : */
4730 : 38 : afterFK = true;
4731 : 38 : break;
4732 : :
4733 : : /*
4734 : : * Column constraint lists separate a Constraint node
4735 : : * from its attributes (e.g. NOT ENFORCED); so a
4736 : : * column-level foreign key constraint may be
4737 : : * represented by multiple Constraint nodes. After
4738 : : * transformConstraintAttrs, the foreign key
4739 : : * Constraint node contains all required information,
4740 : : * making it okay to put into *fk_elements as a
4741 : : * stand-alone Constraint. But since we removed the
4742 : : * foreign key Constraint node from entryconstraints,
4743 : : * we must remove any dependent attribute nodes too,
4744 : : * else the later re-execution of
4745 : : * transformConstraintAttrs will misbehave.
4746 : : */
4747 : 65 : case CONSTR_ATTR_DEFERRABLE:
4748 : : case CONSTR_ATTR_NOT_DEFERRABLE:
4749 : : case CONSTR_ATTR_DEFERRED:
4750 : : case CONSTR_ATTR_IMMEDIATE:
4751 : : case CONSTR_ATTR_ENFORCED:
4752 : : case CONSTR_ATTR_NOT_ENFORCED:
4753 [ + - ]: 65 : if (afterFK)
4754 : 65 : entryconstraints =
4755 : 65 : foreach_delete_current(entryconstraints,
4756 : : colconstr);
4757 : 65 : break;
4758 : :
4759 : 130 : default:
4760 : : /* Any following constraint attributes are unrelated */
4761 : 130 : afterFK = false;
4762 : 130 : break;
4763 : : }
4764 : : }
4765 : :
4766 : : /* Now make a modified ColumnDef to put into newElts */
4767 : 316 : newentry = makeNode(ColumnDef);
4768 : 316 : memcpy(newentry, entry, sizeof(ColumnDef));
4769 : 316 : newentry->constraints = entryconstraints;
4770 : 316 : newElts = lappend(newElts, newentry);
4771 : : }
4772 : : else
4773 : : {
4774 : : /* Other node types pass through unchanged */
29 tgl@sss.pgh.pa.us 4775 :UNC 0 : newElts = lappend(newElts, element);
4776 : : }
4777 : : }
4778 : :
29 tgl@sss.pgh.pa.us 4779 :GNC 324 : newstmt->tableElts = newElts;
4780 : 324 : return newstmt;
29 tgl@sss.pgh.pa.us 4781 :ECB (237) : }
4782 : :
4783 : : /*
4784 : : * transformPartitionCmd
4785 : : * Analyze the ATTACH/DETACH/SPLIT PARTITION command
4786 : : *
4787 : : * In case of the ATTACH/SPLIT PARTITION command, cxt->partbound is set to the
4788 : : * transformed value of bound.
4789 : : */
4790 : : static void
142 akorotkov@postgresql 4791 :GNC 2961 : transformPartitionCmd(CreateStmtContext *cxt, PartitionBoundSpec *bound)
4792 : : {
3436 rhaas@postgresql.org 4793 :CBC 2961 : Relation parentRel = cxt->rel;
4794 : :
3028 alvherre@alvh.no-ip. 4795 [ + + - - : 2961 : switch (parentRel->rd_rel->relkind)
- ]
4796 : : {
4797 : 2683 : case RELKIND_PARTITIONED_TABLE:
4798 : : /* transform the partition bound, if any */
4799 [ - + ]: 2683 : Assert(RelationGetPartitionKey(parentRel) != NULL);
142 akorotkov@postgresql 4800 [ + + ]:GNC 2683 : if (bound != NULL)
3028 alvherre@alvh.no-ip. 4801 :CBC 2310 : cxt->partbound = transformPartitionBound(cxt->pstate, parentRel,
4802 : : bound);
4803 : 2667 : break;
4804 : 278 : case RELKIND_PARTITIONED_INDEX:
4805 : :
4806 : : /*
4807 : : * A partitioned index cannot have a partition bound set. ALTER
4808 : : * INDEX prevents that with its grammar, but not ALTER TABLE.
4809 : : */
142 akorotkov@postgresql 4810 [ + + ]:GNC 278 : if (bound != NULL)
2254 michael@paquier.xyz 4811 [ + - ]:CBC 4 : ereport(ERROR,
4812 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
4813 : : errmsg("\"%s\" is not a partitioned table",
4814 : : RelationGetRelationName(parentRel))));
3028 alvherre@alvh.no-ip. 4815 : 274 : break;
3028 alvherre@alvh.no-ip. 4816 :UBC 0 : case RELKIND_RELATION:
4817 : : /* the table must be partitioned */
4818 [ # # ]: 0 : ereport(ERROR,
4819 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
4820 : : errmsg("table \"%s\" is not partitioned",
4821 : : RelationGetRelationName(parentRel))));
4822 : : break;
4823 : 0 : case RELKIND_INDEX:
4824 : : /* the index must be partitioned */
4825 [ # # ]: 0 : ereport(ERROR,
4826 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
4827 : : errmsg("index \"%s\" is not partitioned",
4828 : : RelationGetRelationName(parentRel))));
4829 : : break;
4830 : 0 : default:
4831 : : /* parser shouldn't let this case through */
4832 [ # # ]: 0 : elog(ERROR, "\"%s\" is not a partitioned table or index",
4833 : : RelationGetRelationName(parentRel));
4834 : : break;
4835 : : }
3436 rhaas@postgresql.org 4836 :CBC 2941 : }
4837 : :
4838 : : /*
4839 : : * transformPartitionBound
4840 : : *
4841 : : * Transform a partition bound specification
4842 : : */
4843 : : PartitionBoundSpec *
3264 tgl@sss.pgh.pa.us 4844 : 8716 : transformPartitionBound(ParseState *pstate, Relation parent,
4845 : : PartitionBoundSpec *spec)
4846 : : {
4847 : : PartitionBoundSpec *result_spec;
3436 rhaas@postgresql.org 4848 : 8716 : PartitionKey key = RelationGetPartitionKey(parent);
4849 : 8716 : char strategy = get_partition_strategy(key);
4850 : 8716 : int partnatts = get_partition_natts(key);
4851 : 8716 : List *partexprs = get_partition_exprs(key);
4852 : :
4853 : : /* Avoid scribbling on input */
4854 : 8716 : result_spec = copyObject(spec);
4855 : :
3161 4856 [ + + ]: 8716 : if (spec->is_default)
4857 : : {
4858 : : /*
4859 : : * Hash partitioning does not support a default partition; there's no
4860 : : * use case for it (since the set of partitions to create is perfectly
4861 : : * defined), and if users do get into it accidentally, it's hard to
4862 : : * back out from it afterwards.
4863 : : */
3099 4864 [ + + ]: 536 : if (strategy == PARTITION_STRATEGY_HASH)
4865 [ + - ]: 4 : ereport(ERROR,
4866 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
4867 : : errmsg("a hash-partitioned table may not have a default partition")));
4868 : :
4869 : : /*
4870 : : * In case of the default partition, parser had no way to identify the
4871 : : * partition strategy. Assign the parent's strategy to the default
4872 : : * partition bound spec.
4873 : : */
3161 4874 : 532 : result_spec->strategy = strategy;
4875 : :
4876 : 532 : return result_spec;
4877 : : }
4878 : :
3099 4879 [ + + ]: 8180 : if (strategy == PARTITION_STRATEGY_HASH)
4880 : : {
4881 [ + + ]: 488 : if (spec->strategy != PARTITION_STRATEGY_HASH)
4882 [ + - ]: 8 : ereport(ERROR,
4883 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
4884 : : errmsg("invalid bound specification for a hash partition"),
4885 : : parser_errposition(pstate, exprLocation((Node *) spec))));
4886 : :
4887 [ + + ]: 480 : if (spec->modulus <= 0)
4888 [ + - ]: 8 : ereport(ERROR,
4889 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
4890 : : errmsg("modulus for hash partition must be an integer value greater than zero")));
4891 : :
4892 [ - + ]: 472 : Assert(spec->remainder >= 0);
4893 : :
4894 [ + + ]: 472 : if (spec->remainder >= spec->modulus)
4895 [ + - ]: 8 : ereport(ERROR,
4896 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
4897 : : errmsg("remainder for hash partition must be less than modulus")));
4898 : : }
4899 [ + + ]: 7692 : else if (strategy == PARTITION_STRATEGY_LIST)
4900 : : {
4901 : : ListCell *cell;
4902 : : char *colname;
4903 : : Oid coltype;
4904 : : int32 coltypmod;
4905 : : Oid partcollation;
4906 : :
3264 tgl@sss.pgh.pa.us 4907 [ + + ]: 3308 : if (spec->strategy != PARTITION_STRATEGY_LIST)
4908 [ + - ]: 12 : ereport(ERROR,
4909 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
4910 : : errmsg("invalid bound specification for a list partition"),
4911 : : parser_errposition(pstate, exprLocation((Node *) spec))));
4912 : :
4913 : : /* Get the only column's name in case we need to output an error */
3436 rhaas@postgresql.org 4914 [ + + ]: 3296 : if (key->partattrs[0] != 0)
3004 alvherre@alvh.no-ip. 4915 : 3209 : colname = get_attname(RelationGetRelid(parent),
4916 : 3209 : key->partattrs[0], false);
4917 : : else
3436 rhaas@postgresql.org 4918 : 87 : colname = deparse_expression((Node *) linitial(partexprs),
3240 tgl@sss.pgh.pa.us 4919 : 87 : deparse_context_for(RelationGetRelationName(parent),
4920 : : RelationGetRelid(parent)),
4921 : : false, false);
4922 : : /* Need its type data too */
3264 4923 : 3296 : coltype = get_partition_col_typid(key, 0);
4924 : 3296 : coltypmod = get_partition_col_typmod(key, 0);
2657 peter@eisentraut.org 4925 : 3296 : partcollation = get_partition_col_collation(key, 0);
4926 : :
3436 rhaas@postgresql.org 4927 : 3296 : result_spec->listdatums = NIL;
4928 [ + - + + : 8499 : foreach(cell, spec->listdatums)
+ + ]
4929 : : {
2657 peter@eisentraut.org 4930 : 5243 : Node *expr = lfirst(cell);
4931 : : Const *value;
4932 : : ListCell *cell2;
4933 : : bool duplicate;
4934 : :
4935 : 5243 : value = transformPartitionBoundValue(pstate, expr,
4936 : : colname, coltype, coltypmod,
4937 : : partcollation);
4938 : :
4939 : : /* Don't add to the result if the value is a duplicate */
3436 rhaas@postgresql.org 4940 : 5203 : duplicate = false;
4941 [ + + + + : 9557 : foreach(cell2, result_spec->listdatums)
+ + ]
4942 : : {
1751 peter@eisentraut.org 4943 : 4354 : Const *value2 = lfirst_node(Const, cell2);
4944 : :
3436 rhaas@postgresql.org 4945 [ - + ]: 4354 : if (equal(value, value2))
4946 : : {
3436 rhaas@postgresql.org 4947 :UBC 0 : duplicate = true;
4948 : 0 : break;
4949 : : }
4950 : : }
3436 rhaas@postgresql.org 4951 [ - + ]:CBC 5203 : if (duplicate)
3436 rhaas@postgresql.org 4952 :UBC 0 : continue;
4953 : :
3436 rhaas@postgresql.org 4954 :CBC 5203 : result_spec->listdatums = lappend(result_spec->listdatums,
4955 : : value);
4956 : : }
4957 : : }
4958 [ + - ]: 4384 : else if (strategy == PARTITION_STRATEGY_RANGE)
4959 : : {
4960 [ + + ]: 4384 : if (spec->strategy != PARTITION_STRATEGY_RANGE)
4961 [ + - ]: 16 : ereport(ERROR,
4962 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
4963 : : errmsg("invalid bound specification for a range partition"),
4964 : : parser_errposition(pstate, exprLocation((Node *) spec))));
4965 : :
4966 [ + + ]: 4368 : if (list_length(spec->lowerdatums) != partnatts)
4967 [ + - ]: 4 : ereport(ERROR,
4968 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
4969 : : errmsg("FROM must specify exactly one value per partitioning column")));
4970 [ + + ]: 4364 : if (list_length(spec->upperdatums) != partnatts)
4971 [ + - ]: 4 : ereport(ERROR,
4972 : : (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
4973 : : errmsg("TO must specify exactly one value per partitioning column")));
4974 : :
4975 : : /*
4976 : : * Convert raw parse nodes into PartitionRangeDatum nodes and perform
4977 : : * any necessary validation.
4978 : : */
2657 peter@eisentraut.org 4979 : 4316 : result_spec->lowerdatums =
2540 tgl@sss.pgh.pa.us 4980 : 4360 : transformPartitionRangeBounds(pstate, spec->lowerdatums,
4981 : : parent);
2657 peter@eisentraut.org 4982 : 4312 : result_spec->upperdatums =
2540 tgl@sss.pgh.pa.us 4983 : 4316 : transformPartitionRangeBounds(pstate, spec->upperdatums,
4984 : : parent);
4985 : : }
4986 : : else
2657 peter@eisentraut.org 4987 [ # # ]:UBC 0 : elog(ERROR, "unexpected partition strategy: %d", (int) strategy);
4988 : :
2657 peter@eisentraut.org 4989 :CBC 8032 : return result_spec;
4990 : : }
4991 : :
4992 : : /*
4993 : : * transformPartitionRangeBounds
4994 : : * This converts the expressions for range partition bounds from the raw
4995 : : * grammar representation to PartitionRangeDatum structs
4996 : : */
4997 : : static List *
4998 : 8676 : transformPartitionRangeBounds(ParseState *pstate, List *blist,
4999 : : Relation parent)
5000 : : {
5001 : 8676 : List *result = NIL;
5002 : 8676 : PartitionKey key = RelationGetPartitionKey(parent);
5003 : 8676 : List *partexprs = get_partition_exprs(key);
5004 : : ListCell *lc;
5005 : : int i,
5006 : : j;
5007 : :
116 drowley@postgresql.o 5008 : 8676 : j = 0;
2657 peter@eisentraut.org 5009 [ + - + + : 18806 : foreach(lc, blist)
+ + ]
5010 : : {
2540 tgl@sss.pgh.pa.us 5011 : 10166 : Node *expr = lfirst(lc);
2657 peter@eisentraut.org 5012 : 10166 : PartitionRangeDatum *prd = NULL;
5013 : :
116 drowley@postgresql.o 5014 : 10166 : i = foreach_current_index(lc);
5015 : :
5016 : : /*
5017 : : * Infinite range bounds -- "minvalue" and "maxvalue" -- get passed in
5018 : : * as ColumnRefs.
5019 : : */
2657 peter@eisentraut.org 5020 [ + + ]: 10166 : if (IsA(expr, ColumnRef))
5021 : : {
2540 tgl@sss.pgh.pa.us 5022 : 523 : ColumnRef *cref = (ColumnRef *) expr;
5023 : 523 : char *cname = NULL;
5024 : :
5025 : : /*
5026 : : * There should be a single field named either "minvalue" or
5027 : : * "maxvalue".
5028 : : */
2657 peter@eisentraut.org 5029 [ + + ]: 523 : if (list_length(cref->fields) == 1 &&
5030 [ + - ]: 519 : IsA(linitial(cref->fields), String))
5031 : 519 : cname = strVal(linitial(cref->fields));
5032 : :
2597 michael@paquier.xyz 5033 [ + + ]: 523 : if (cname == NULL)
5034 : : {
5035 : : /*
5036 : : * ColumnRef is not in the desired single-field-name form. For
5037 : : * consistency between all partition strategies, let the
5038 : : * expression transformation report any errors rather than
5039 : : * doing it ourselves.
5040 : : */
5041 : : }
5042 [ + + ]: 519 : else if (strcmp("minvalue", cname) == 0)
5043 : : {
2657 peter@eisentraut.org 5044 : 252 : prd = makeNode(PartitionRangeDatum);
5045 : 252 : prd->kind = PARTITION_RANGE_DATUM_MINVALUE;
5046 : 252 : prd->value = NULL;
5047 : : }
5048 [ + + ]: 267 : else if (strcmp("maxvalue", cname) == 0)
5049 : : {
5050 : 259 : prd = makeNode(PartitionRangeDatum);
5051 : 259 : prd->kind = PARTITION_RANGE_DATUM_MAXVALUE;
5052 : 259 : prd->value = NULL;
5053 : : }
5054 : : }
5055 : :
5056 [ + + ]: 10166 : if (prd == NULL)
5057 : : {
5058 : : char *colname;
5059 : : Oid coltype;
5060 : : int32 coltypmod;
5061 : : Oid partcollation;
5062 : : Const *value;
5063 : :
5064 : : /* Get the column's name in case we need to output an error */
3436 rhaas@postgresql.org 5065 [ + + ]: 9655 : if (key->partattrs[i] != 0)
3004 alvherre@alvh.no-ip. 5066 : 9077 : colname = get_attname(RelationGetRelid(parent),
5067 : 9077 : key->partattrs[i], false);
5068 : : else
5069 : : {
3436 rhaas@postgresql.org 5070 : 578 : colname = deparse_expression((Node *) list_nth(partexprs, j),
3240 tgl@sss.pgh.pa.us 5071 : 578 : deparse_context_for(RelationGetRelationName(parent),
5072 : : RelationGetRelid(parent)),
5073 : : false, false);
3436 rhaas@postgresql.org 5074 : 578 : ++j;
5075 : : }
5076 : :
5077 : : /* Need its type data too */
3264 tgl@sss.pgh.pa.us 5078 : 9655 : coltype = get_partition_col_typid(key, i);
5079 : 9655 : coltypmod = get_partition_col_typmod(key, i);
2657 peter@eisentraut.org 5080 : 9655 : partcollation = get_partition_col_collation(key, i);
5081 : :
5082 : 9655 : value = transformPartitionBoundValue(pstate, expr,
5083 : : colname,
5084 : : coltype, coltypmod,
5085 : : partcollation);
5086 [ + + ]: 9623 : if (value->constisnull)
5087 [ + - ]: 4 : ereport(ERROR,
5088 : : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
5089 : : errmsg("cannot specify NULL in range bound")));
5090 : 9619 : prd = makeNode(PartitionRangeDatum);
5091 : 9619 : prd->kind = PARTITION_RANGE_DATUM_VALUE;
5092 : 9619 : prd->value = (Node *) value;
5093 : : }
5094 : :
5095 : 10130 : prd->location = exprLocation(expr);
5096 : :
5097 : 10130 : result = lappend(result, prd);
5098 : : }
5099 : :
5100 : : /*
5101 : : * Once we see MINVALUE or MAXVALUE for one column, the remaining columns
5102 : : * must be the same.
5103 : : */
5104 : 8640 : validateInfiniteBounds(pstate, result);
5105 : :
5106 : 8628 : return result;
5107 : : }
5108 : :
5109 : : /*
5110 : : * validateInfiniteBounds
5111 : : *
5112 : : * Check that a MAXVALUE or MINVALUE specification in a partition bound is
5113 : : * followed only by more of the same.
5114 : : */
5115 : : static void
3154 rhaas@postgresql.org 5116 : 8640 : validateInfiniteBounds(ParseState *pstate, List *blist)
5117 : : {
5118 : : ListCell *lc;
5119 : 8640 : PartitionRangeDatumKind kind = PARTITION_RANGE_DATUM_VALUE;
5120 : :
5121 [ + - + + : 18754 : foreach(lc, blist)
+ + ]
5122 : : {
1751 peter@eisentraut.org 5123 : 10126 : PartitionRangeDatum *prd = lfirst_node(PartitionRangeDatum, lc);
5124 : :
3154 rhaas@postgresql.org 5125 [ + + ]: 10126 : if (kind == prd->kind)
5126 : 9759 : continue;
5127 : :
5128 [ + + + - ]: 367 : switch (kind)
5129 : : {
5130 : 355 : case PARTITION_RANGE_DATUM_VALUE:
5131 : 355 : kind = prd->kind;
5132 : 355 : break;
5133 : :
5134 : 4 : case PARTITION_RANGE_DATUM_MAXVALUE:
5135 [ + - ]: 4 : ereport(ERROR,
5136 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
5137 : : errmsg("every bound following MAXVALUE must also be MAXVALUE"),
5138 : : parser_errposition(pstate, exprLocation((Node *) prd))));
5139 : : break;
5140 : :
5141 : 8 : case PARTITION_RANGE_DATUM_MINVALUE:
5142 [ + - ]: 8 : ereport(ERROR,
5143 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
5144 : : errmsg("every bound following MINVALUE must also be MINVALUE"),
5145 : : parser_errposition(pstate, exprLocation((Node *) prd))));
5146 : : break;
5147 : : }
5148 : : }
5149 : 8628 : }
5150 : :
5151 : : /*
5152 : : * Transform one entry in a partition bound spec, producing a constant.
5153 : : */
5154 : : static Const *
2657 peter@eisentraut.org 5155 : 14898 : transformPartitionBoundValue(ParseState *pstate, Node *val,
5156 : : const char *colName, Oid colType, int32 colTypmod,
5157 : : Oid partCollation)
5158 : : {
5159 : : Node *value;
5160 : :
5161 : : /* Transform raw parsetree */
5162 : 14898 : value = transformExpr(pstate, val, EXPR_KIND_PARTITION_BOUND);
5163 : :
5164 : : /*
5165 : : * transformExpr() should have already rejected column references,
5166 : : * subqueries, aggregates, window functions, and SRFs, based on the
5167 : : * EXPR_KIND_ of a partition bound expression.
5168 : : */
2050 tgl@sss.pgh.pa.us 5169 [ - + ]: 14830 : Assert(!contain_var_clause(value));
5170 : :
5171 : : /*
5172 : : * Coerce to the correct type. This might cause an explicit coercion step
5173 : : * to be added on top of the expression, which must be evaluated before
5174 : : * returning the result to the caller.
5175 : : */
3264 5176 : 14830 : value = coerce_to_target_type(pstate,
5177 : : value, exprType(value),
5178 : : colType,
5179 : : colTypmod,
5180 : : COERCION_ASSIGNMENT,
5181 : : COERCE_IMPLICIT_CAST,
5182 : : -1);
5183 : :
5184 [ + + ]: 14830 : if (value == NULL)
5185 [ + - ]: 4 : ereport(ERROR,
5186 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
5187 : : errmsg("specified value cannot be cast to type %s for column \"%s\"",
5188 : : format_type_be(colType), colName),
5189 : : parser_errposition(pstate, exprLocation(val))));
5190 : :
5191 : : /*
5192 : : * Evaluate the expression, if needed, assigning the partition key's data
5193 : : * type and collation to the resulting Const node.
5194 : : */
2050 5195 [ + + ]: 14826 : if (!IsA(value, Const))
5196 : : {
2045 5197 : 970 : assign_expr_collations(pstate, value);
2050 5198 : 970 : value = (Node *) expression_planner((Expr *) value);
5199 : 970 : value = (Node *) evaluate_expr((Expr *) value, colType, colTypmod,
5200 : : partCollation);
5201 [ - + ]: 970 : if (!IsA(value, Const))
2050 tgl@sss.pgh.pa.us 5202 [ # # ]:UBC 0 : elog(ERROR, "could not evaluate partition bound expression");
5203 : : }
5204 : : else
5205 : : {
5206 : : /*
5207 : : * If the expression is already a Const, as is often the case, we can
5208 : : * skip the rather expensive steps above. But we still have to insert
5209 : : * the right collation, since coerce_to_target_type doesn't handle
5210 : : * that.
5211 : : */
2050 tgl@sss.pgh.pa.us 5212 :CBC 13856 : ((Const *) value)->constcollid = partCollation;
5213 : : }
5214 : :
5215 : : /*
5216 : : * Attach original expression's parse location to the Const, so that
5217 : : * that's what will be reported for any later errors related to this
5218 : : * partition bound.
5219 : : */
5220 : 14826 : ((Const *) value)->location = exprLocation(val);
5221 : :
3264 5222 : 14826 : return (Const *) value;
5223 : : }
|
