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