Line data Source code
1 : /*-------------------------------------------------------------------------
2 : *
3 : * typecmds.c
4 : * Routines for SQL commands that manipulate types (and domains).
5 : *
6 : * Portions Copyright (c) 1996-2025, PostgreSQL Global Development Group
7 : * Portions Copyright (c) 1994, Regents of the University of California
8 : *
9 : *
10 : * IDENTIFICATION
11 : * src/backend/commands/typecmds.c
12 : *
13 : * DESCRIPTION
14 : * The "DefineFoo" routines take the parse tree and pick out the
15 : * appropriate arguments/flags, passing the results to the
16 : * corresponding "FooCreate" routines (in src/backend/catalog) that do
17 : * the actual catalog-munging. These routines also verify permission
18 : * of the user to execute the command.
19 : *
20 : * NOTES
21 : * These things must be defined and committed in the following order:
22 : * "create function":
23 : * input/output, recv/send functions
24 : * "create type":
25 : * type
26 : * "create operator":
27 : * operators
28 : *
29 : *
30 : *-------------------------------------------------------------------------
31 : */
32 : #include "postgres.h"
33 :
34 : #include "access/genam.h"
35 : #include "access/htup_details.h"
36 : #include "access/relation.h"
37 : #include "access/table.h"
38 : #include "access/tableam.h"
39 : #include "access/xact.h"
40 : #include "catalog/binary_upgrade.h"
41 : #include "catalog/catalog.h"
42 : #include "catalog/heap.h"
43 : #include "catalog/objectaccess.h"
44 : #include "catalog/pg_am.h"
45 : #include "catalog/pg_authid.h"
46 : #include "catalog/pg_cast.h"
47 : #include "catalog/pg_collation.h"
48 : #include "catalog/pg_constraint.h"
49 : #include "catalog/pg_depend.h"
50 : #include "catalog/pg_enum.h"
51 : #include "catalog/pg_language.h"
52 : #include "catalog/pg_namespace.h"
53 : #include "catalog/pg_proc.h"
54 : #include "catalog/pg_range.h"
55 : #include "catalog/pg_type.h"
56 : #include "commands/defrem.h"
57 : #include "commands/tablecmds.h"
58 : #include "commands/typecmds.h"
59 : #include "executor/executor.h"
60 : #include "miscadmin.h"
61 : #include "nodes/makefuncs.h"
62 : #include "optimizer/optimizer.h"
63 : #include "parser/parse_coerce.h"
64 : #include "parser/parse_collate.h"
65 : #include "parser/parse_expr.h"
66 : #include "parser/parse_func.h"
67 : #include "parser/parse_type.h"
68 : #include "utils/builtins.h"
69 : #include "utils/fmgroids.h"
70 : #include "utils/inval.h"
71 : #include "utils/lsyscache.h"
72 : #include "utils/rel.h"
73 : #include "utils/ruleutils.h"
74 : #include "utils/snapmgr.h"
75 : #include "utils/syscache.h"
76 :
77 :
78 : /* result structure for get_rels_with_domain() */
79 : typedef struct
80 : {
81 : Relation rel; /* opened and locked relation */
82 : int natts; /* number of attributes of interest */
83 : int *atts; /* attribute numbers */
84 : /* atts[] is of allocated length RelationGetNumberOfAttributes(rel) */
85 : } RelToCheck;
86 :
87 : /* parameter structure for AlterTypeRecurse() */
88 : typedef struct
89 : {
90 : /* Flags indicating which type attributes to update */
91 : bool updateStorage;
92 : bool updateReceive;
93 : bool updateSend;
94 : bool updateTypmodin;
95 : bool updateTypmodout;
96 : bool updateAnalyze;
97 : bool updateSubscript;
98 : /* New values for relevant attributes */
99 : char storage;
100 : Oid receiveOid;
101 : Oid sendOid;
102 : Oid typmodinOid;
103 : Oid typmodoutOid;
104 : Oid analyzeOid;
105 : Oid subscriptOid;
106 : } AlterTypeRecurseParams;
107 :
108 : /* Potentially set by pg_upgrade_support functions */
109 : Oid binary_upgrade_next_array_pg_type_oid = InvalidOid;
110 : Oid binary_upgrade_next_mrng_pg_type_oid = InvalidOid;
111 : Oid binary_upgrade_next_mrng_array_pg_type_oid = InvalidOid;
112 :
113 : static void makeRangeConstructors(const char *name, Oid namespace,
114 : Oid rangeOid, Oid subtype);
115 : static void makeMultirangeConstructors(const char *name, Oid namespace,
116 : Oid multirangeOid, Oid rangeOid,
117 : Oid rangeArrayOid, Oid *castFuncOid);
118 : static Oid findTypeInputFunction(List *procname, Oid typeOid);
119 : static Oid findTypeOutputFunction(List *procname, Oid typeOid);
120 : static Oid findTypeReceiveFunction(List *procname, Oid typeOid);
121 : static Oid findTypeSendFunction(List *procname, Oid typeOid);
122 : static Oid findTypeTypmodinFunction(List *procname);
123 : static Oid findTypeTypmodoutFunction(List *procname);
124 : static Oid findTypeAnalyzeFunction(List *procname, Oid typeOid);
125 : static Oid findTypeSubscriptingFunction(List *procname, Oid typeOid);
126 : static Oid findRangeSubOpclass(List *opcname, Oid subtype);
127 : static Oid findRangeCanonicalFunction(List *procname, Oid typeOid);
128 : static Oid findRangeSubtypeDiffFunction(List *procname, Oid subtype);
129 : static void validateDomainCheckConstraint(Oid domainoid, const char *ccbin, LOCKMODE lockmode);
130 : static void validateDomainNotNullConstraint(Oid domainoid);
131 : static List *get_rels_with_domain(Oid domainOid, LOCKMODE lockmode);
132 : static void checkEnumOwner(HeapTuple tup);
133 : static char *domainAddCheckConstraint(Oid domainOid, Oid domainNamespace,
134 : Oid baseTypeOid,
135 : int typMod, Constraint *constr,
136 : const char *domainName, ObjectAddress *constrAddr);
137 : static Node *replace_domain_constraint_value(ParseState *pstate,
138 : ColumnRef *cref);
139 : static void domainAddNotNullConstraint(Oid domainOid, Oid domainNamespace, Oid baseTypeOid,
140 : int typMod, Constraint *constr,
141 : const char *domainName, ObjectAddress *constrAddr);
142 : static void AlterTypeRecurse(Oid typeOid, bool isImplicitArray,
143 : HeapTuple tup, Relation catalog,
144 : AlterTypeRecurseParams *atparams);
145 :
146 :
147 : /*
148 : * DefineType
149 : * Registers a new base type.
150 : */
151 : ObjectAddress
152 396 : DefineType(ParseState *pstate, List *names, List *parameters)
153 : {
154 : char *typeName;
155 : Oid typeNamespace;
156 396 : int16 internalLength = -1; /* default: variable-length */
157 396 : List *inputName = NIL;
158 396 : List *outputName = NIL;
159 396 : List *receiveName = NIL;
160 396 : List *sendName = NIL;
161 396 : List *typmodinName = NIL;
162 396 : List *typmodoutName = NIL;
163 396 : List *analyzeName = NIL;
164 396 : List *subscriptName = NIL;
165 396 : char category = TYPCATEGORY_USER;
166 396 : bool preferred = false;
167 396 : char delimiter = DEFAULT_TYPDELIM;
168 396 : Oid elemType = InvalidOid;
169 396 : char *defaultValue = NULL;
170 396 : bool byValue = false;
171 396 : char alignment = TYPALIGN_INT; /* default alignment */
172 396 : char storage = TYPSTORAGE_PLAIN; /* default TOAST storage method */
173 396 : Oid collation = InvalidOid;
174 396 : DefElem *likeTypeEl = NULL;
175 396 : DefElem *internalLengthEl = NULL;
176 396 : DefElem *inputNameEl = NULL;
177 396 : DefElem *outputNameEl = NULL;
178 396 : DefElem *receiveNameEl = NULL;
179 396 : DefElem *sendNameEl = NULL;
180 396 : DefElem *typmodinNameEl = NULL;
181 396 : DefElem *typmodoutNameEl = NULL;
182 396 : DefElem *analyzeNameEl = NULL;
183 396 : DefElem *subscriptNameEl = NULL;
184 396 : DefElem *categoryEl = NULL;
185 396 : DefElem *preferredEl = NULL;
186 396 : DefElem *delimiterEl = NULL;
187 396 : DefElem *elemTypeEl = NULL;
188 396 : DefElem *defaultValueEl = NULL;
189 396 : DefElem *byValueEl = NULL;
190 396 : DefElem *alignmentEl = NULL;
191 396 : DefElem *storageEl = NULL;
192 396 : DefElem *collatableEl = NULL;
193 : Oid inputOid;
194 : Oid outputOid;
195 396 : Oid receiveOid = InvalidOid;
196 396 : Oid sendOid = InvalidOid;
197 396 : Oid typmodinOid = InvalidOid;
198 396 : Oid typmodoutOid = InvalidOid;
199 396 : Oid analyzeOid = InvalidOid;
200 396 : Oid subscriptOid = InvalidOid;
201 : char *array_type;
202 : Oid array_oid;
203 : Oid typoid;
204 : ListCell *pl;
205 : ObjectAddress address;
206 :
207 : /*
208 : * As of Postgres 8.4, we require superuser privilege to create a base
209 : * type. This is simple paranoia: there are too many ways to mess up the
210 : * system with an incorrect type definition (for instance, representation
211 : * parameters that don't match what the C code expects). In practice it
212 : * takes superuser privilege to create the I/O functions, and so the
213 : * former requirement that you own the I/O functions pretty much forced
214 : * superuserness anyway. We're just making doubly sure here.
215 : *
216 : * XXX re-enable NOT_USED code sections below if you remove this test.
217 : */
218 396 : if (!superuser())
219 0 : ereport(ERROR,
220 : (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
221 : errmsg("must be superuser to create a base type")));
222 :
223 : /* Convert list of names to a name and namespace */
224 396 : typeNamespace = QualifiedNameGetCreationNamespace(names, &typeName);
225 :
226 : #ifdef NOT_USED
227 : /* XXX this is unnecessary given the superuser check above */
228 : /* Check we have creation rights in target namespace */
229 : aclresult = object_aclcheck(NamespaceRelationId, typeNamespace, GetUserId(), ACL_CREATE);
230 : if (aclresult != ACLCHECK_OK)
231 : aclcheck_error(aclresult, OBJECT_SCHEMA,
232 : get_namespace_name(typeNamespace));
233 : #endif
234 :
235 : /*
236 : * Look to see if type already exists.
237 : */
238 396 : typoid = GetSysCacheOid2(TYPENAMENSP, Anum_pg_type_oid,
239 : CStringGetDatum(typeName),
240 : ObjectIdGetDatum(typeNamespace));
241 :
242 : /*
243 : * If it's not a shell, see if it's an autogenerated array type, and if so
244 : * rename it out of the way.
245 : */
246 396 : if (OidIsValid(typoid) && get_typisdefined(typoid))
247 : {
248 6 : if (moveArrayTypeName(typoid, typeName, typeNamespace))
249 0 : typoid = InvalidOid;
250 : else
251 6 : ereport(ERROR,
252 : (errcode(ERRCODE_DUPLICATE_OBJECT),
253 : errmsg("type \"%s\" already exists", typeName)));
254 : }
255 :
256 : /*
257 : * If this command is a parameterless CREATE TYPE, then we're just here to
258 : * make a shell type, so do that (or fail if there already is a shell).
259 : */
260 390 : if (parameters == NIL)
261 : {
262 150 : if (OidIsValid(typoid))
263 6 : ereport(ERROR,
264 : (errcode(ERRCODE_DUPLICATE_OBJECT),
265 : errmsg("type \"%s\" already exists", typeName)));
266 :
267 144 : address = TypeShellMake(typeName, typeNamespace, GetUserId());
268 144 : return address;
269 : }
270 :
271 : /*
272 : * Otherwise, we must already have a shell type, since there is no other
273 : * way that the I/O functions could have been created.
274 : */
275 240 : if (!OidIsValid(typoid))
276 6 : ereport(ERROR,
277 : (errcode(ERRCODE_DUPLICATE_OBJECT),
278 : errmsg("type \"%s\" does not exist", typeName),
279 : errhint("Create the type as a shell type, then create its I/O functions, then do a full CREATE TYPE.")));
280 :
281 : /* Extract the parameters from the parameter list */
282 1182 : foreach(pl, parameters)
283 : {
284 948 : DefElem *defel = (DefElem *) lfirst(pl);
285 : DefElem **defelp;
286 :
287 948 : if (strcmp(defel->defname, "like") == 0)
288 56 : defelp = &likeTypeEl;
289 892 : else if (strcmp(defel->defname, "internallength") == 0)
290 156 : defelp = &internalLengthEl;
291 736 : else if (strcmp(defel->defname, "input") == 0)
292 228 : defelp = &inputNameEl;
293 508 : else if (strcmp(defel->defname, "output") == 0)
294 228 : defelp = &outputNameEl;
295 280 : else if (strcmp(defel->defname, "receive") == 0)
296 18 : defelp = &receiveNameEl;
297 262 : else if (strcmp(defel->defname, "send") == 0)
298 18 : defelp = &sendNameEl;
299 244 : else if (strcmp(defel->defname, "typmod_in") == 0)
300 8 : defelp = &typmodinNameEl;
301 236 : else if (strcmp(defel->defname, "typmod_out") == 0)
302 8 : defelp = &typmodoutNameEl;
303 228 : else if (strcmp(defel->defname, "analyze") == 0 ||
304 228 : strcmp(defel->defname, "analyse") == 0)
305 0 : defelp = &analyzeNameEl;
306 228 : else if (strcmp(defel->defname, "subscript") == 0)
307 2 : defelp = &subscriptNameEl;
308 226 : else if (strcmp(defel->defname, "category") == 0)
309 12 : defelp = &categoryEl;
310 214 : else if (strcmp(defel->defname, "preferred") == 0)
311 12 : defelp = &preferredEl;
312 202 : else if (strcmp(defel->defname, "delimiter") == 0)
313 0 : defelp = &delimiterEl;
314 202 : else if (strcmp(defel->defname, "element") == 0)
315 14 : defelp = &elemTypeEl;
316 188 : else if (strcmp(defel->defname, "default") == 0)
317 18 : defelp = &defaultValueEl;
318 170 : else if (strcmp(defel->defname, "passedbyvalue") == 0)
319 14 : defelp = &byValueEl;
320 156 : else if (strcmp(defel->defname, "alignment") == 0)
321 56 : defelp = &alignmentEl;
322 100 : else if (strcmp(defel->defname, "storage") == 0)
323 60 : defelp = &storageEl;
324 40 : else if (strcmp(defel->defname, "collatable") == 0)
325 4 : defelp = &collatableEl;
326 : else
327 : {
328 : /* WARNING, not ERROR, for historical backwards-compatibility */
329 36 : ereport(WARNING,
330 : (errcode(ERRCODE_SYNTAX_ERROR),
331 : errmsg("type attribute \"%s\" not recognized",
332 : defel->defname),
333 : parser_errposition(pstate, defel->location)));
334 36 : continue;
335 : }
336 912 : if (*defelp != NULL)
337 0 : errorConflictingDefElem(defel, pstate);
338 912 : *defelp = defel;
339 : }
340 :
341 : /*
342 : * Now interpret the options; we do this separately so that LIKE can be
343 : * overridden by other options regardless of the ordering in the parameter
344 : * list.
345 : */
346 234 : if (likeTypeEl)
347 : {
348 : Type likeType;
349 : Form_pg_type likeForm;
350 :
351 56 : likeType = typenameType(pstate, defGetTypeName(likeTypeEl), NULL);
352 50 : likeForm = (Form_pg_type) GETSTRUCT(likeType);
353 50 : internalLength = likeForm->typlen;
354 50 : byValue = likeForm->typbyval;
355 50 : alignment = likeForm->typalign;
356 50 : storage = likeForm->typstorage;
357 50 : ReleaseSysCache(likeType);
358 : }
359 228 : if (internalLengthEl)
360 156 : internalLength = defGetTypeLength(internalLengthEl);
361 228 : if (inputNameEl)
362 222 : inputName = defGetQualifiedName(inputNameEl);
363 228 : if (outputNameEl)
364 222 : outputName = defGetQualifiedName(outputNameEl);
365 228 : if (receiveNameEl)
366 18 : receiveName = defGetQualifiedName(receiveNameEl);
367 228 : if (sendNameEl)
368 18 : sendName = defGetQualifiedName(sendNameEl);
369 228 : if (typmodinNameEl)
370 8 : typmodinName = defGetQualifiedName(typmodinNameEl);
371 228 : if (typmodoutNameEl)
372 8 : typmodoutName = defGetQualifiedName(typmodoutNameEl);
373 228 : if (analyzeNameEl)
374 0 : analyzeName = defGetQualifiedName(analyzeNameEl);
375 228 : if (subscriptNameEl)
376 2 : subscriptName = defGetQualifiedName(subscriptNameEl);
377 228 : if (categoryEl)
378 : {
379 12 : char *p = defGetString(categoryEl);
380 :
381 12 : category = p[0];
382 : /* restrict to non-control ASCII */
383 12 : if (category < 32 || category > 126)
384 0 : ereport(ERROR,
385 : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
386 : errmsg("invalid type category \"%s\": must be simple ASCII",
387 : p)));
388 : }
389 228 : if (preferredEl)
390 12 : preferred = defGetBoolean(preferredEl);
391 228 : if (delimiterEl)
392 : {
393 0 : char *p = defGetString(delimiterEl);
394 :
395 0 : delimiter = p[0];
396 : /* XXX shouldn't we restrict the delimiter? */
397 : }
398 228 : if (elemTypeEl)
399 : {
400 14 : elemType = typenameTypeId(NULL, defGetTypeName(elemTypeEl));
401 : /* disallow arrays of pseudotypes */
402 14 : if (get_typtype(elemType) == TYPTYPE_PSEUDO)
403 0 : ereport(ERROR,
404 : (errcode(ERRCODE_DATATYPE_MISMATCH),
405 : errmsg("array element type cannot be %s",
406 : format_type_be(elemType))));
407 : }
408 228 : if (defaultValueEl)
409 18 : defaultValue = defGetString(defaultValueEl);
410 228 : if (byValueEl)
411 14 : byValue = defGetBoolean(byValueEl);
412 228 : if (alignmentEl)
413 : {
414 56 : char *a = defGetString(alignmentEl);
415 :
416 : /*
417 : * Note: if argument was an unquoted identifier, parser will have
418 : * applied translations to it, so be prepared to recognize translated
419 : * type names as well as the nominal form.
420 : */
421 94 : if (pg_strcasecmp(a, "double") == 0 ||
422 76 : pg_strcasecmp(a, "float8") == 0 ||
423 38 : pg_strcasecmp(a, "pg_catalog.float8") == 0)
424 18 : alignment = TYPALIGN_DOUBLE;
425 44 : else if (pg_strcasecmp(a, "int4") == 0 ||
426 6 : pg_strcasecmp(a, "pg_catalog.int4") == 0)
427 38 : alignment = TYPALIGN_INT;
428 0 : else if (pg_strcasecmp(a, "int2") == 0 ||
429 0 : pg_strcasecmp(a, "pg_catalog.int2") == 0)
430 0 : alignment = TYPALIGN_SHORT;
431 0 : else if (pg_strcasecmp(a, "char") == 0 ||
432 0 : pg_strcasecmp(a, "pg_catalog.bpchar") == 0)
433 0 : alignment = TYPALIGN_CHAR;
434 : else
435 0 : ereport(ERROR,
436 : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
437 : errmsg("alignment \"%s\" not recognized", a)));
438 : }
439 228 : if (storageEl)
440 : {
441 60 : char *a = defGetString(storageEl);
442 :
443 60 : if (pg_strcasecmp(a, "plain") == 0)
444 18 : storage = TYPSTORAGE_PLAIN;
445 42 : else if (pg_strcasecmp(a, "external") == 0)
446 0 : storage = TYPSTORAGE_EXTERNAL;
447 42 : else if (pg_strcasecmp(a, "extended") == 0)
448 36 : storage = TYPSTORAGE_EXTENDED;
449 6 : else if (pg_strcasecmp(a, "main") == 0)
450 6 : storage = TYPSTORAGE_MAIN;
451 : else
452 0 : ereport(ERROR,
453 : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
454 : errmsg("storage \"%s\" not recognized", a)));
455 : }
456 228 : if (collatableEl)
457 4 : collation = defGetBoolean(collatableEl) ? DEFAULT_COLLATION_OID : InvalidOid;
458 :
459 : /*
460 : * make sure we have our required definitions
461 : */
462 228 : if (inputName == NIL)
463 6 : ereport(ERROR,
464 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
465 : errmsg("type input function must be specified")));
466 222 : if (outputName == NIL)
467 0 : ereport(ERROR,
468 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
469 : errmsg("type output function must be specified")));
470 :
471 222 : if (typmodinName == NIL && typmodoutName != NIL)
472 0 : ereport(ERROR,
473 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
474 : errmsg("type modifier output function is useless without a type modifier input function")));
475 :
476 : /*
477 : * Convert I/O proc names to OIDs
478 : */
479 222 : inputOid = findTypeInputFunction(inputName, typoid);
480 216 : outputOid = findTypeOutputFunction(outputName, typoid);
481 216 : if (receiveName)
482 18 : receiveOid = findTypeReceiveFunction(receiveName, typoid);
483 216 : if (sendName)
484 18 : sendOid = findTypeSendFunction(sendName, typoid);
485 :
486 : /*
487 : * Convert typmodin/out function proc names to OIDs.
488 : */
489 216 : if (typmodinName)
490 8 : typmodinOid = findTypeTypmodinFunction(typmodinName);
491 216 : if (typmodoutName)
492 8 : typmodoutOid = findTypeTypmodoutFunction(typmodoutName);
493 :
494 : /*
495 : * Convert analysis function proc name to an OID. If no analysis function
496 : * is specified, we'll use zero to select the built-in default algorithm.
497 : */
498 216 : if (analyzeName)
499 0 : analyzeOid = findTypeAnalyzeFunction(analyzeName, typoid);
500 :
501 : /*
502 : * Likewise look up the subscripting function if any. If it is not
503 : * specified, but a typelem is specified, allow that if
504 : * raw_array_subscript_handler can be used. (This is for backwards
505 : * compatibility; maybe someday we should throw an error instead.)
506 : */
507 216 : if (subscriptName)
508 2 : subscriptOid = findTypeSubscriptingFunction(subscriptName, typoid);
509 214 : else if (OidIsValid(elemType))
510 : {
511 6 : if (internalLength > 0 && !byValue && get_typlen(elemType) > 0)
512 6 : subscriptOid = F_RAW_ARRAY_SUBSCRIPT_HANDLER;
513 : else
514 0 : ereport(ERROR,
515 : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
516 : errmsg("element type cannot be specified without a subscripting function")));
517 : }
518 :
519 : /*
520 : * Check permissions on functions. We choose to require the creator/owner
521 : * of a type to also own the underlying functions. Since creating a type
522 : * is tantamount to granting public execute access on the functions, the
523 : * minimum sane check would be for execute-with-grant-option. But we
524 : * don't have a way to make the type go away if the grant option is
525 : * revoked, so ownership seems better.
526 : *
527 : * XXX For now, this is all unnecessary given the superuser check above.
528 : * If we ever relax that, these calls likely should be moved into
529 : * findTypeInputFunction et al, where they could be shared by AlterType.
530 : */
531 : #ifdef NOT_USED
532 : if (inputOid && !object_ownercheck(ProcedureRelationId, inputOid, GetUserId()))
533 : aclcheck_error(ACLCHECK_NOT_OWNER, OBJECT_FUNCTION,
534 : NameListToString(inputName));
535 : if (outputOid && !object_ownercheck(ProcedureRelationId, outputOid, GetUserId()))
536 : aclcheck_error(ACLCHECK_NOT_OWNER, OBJECT_FUNCTION,
537 : NameListToString(outputName));
538 : if (receiveOid && !object_ownercheck(ProcedureRelationId, receiveOid, GetUserId()))
539 : aclcheck_error(ACLCHECK_NOT_OWNER, OBJECT_FUNCTION,
540 : NameListToString(receiveName));
541 : if (sendOid && !object_ownercheck(ProcedureRelationId, sendOid, GetUserId()))
542 : aclcheck_error(ACLCHECK_NOT_OWNER, OBJECT_FUNCTION,
543 : NameListToString(sendName));
544 : if (typmodinOid && !object_ownercheck(ProcedureRelationId, typmodinOid, GetUserId()))
545 : aclcheck_error(ACLCHECK_NOT_OWNER, OBJECT_FUNCTION,
546 : NameListToString(typmodinName));
547 : if (typmodoutOid && !object_ownercheck(ProcedureRelationId, typmodoutOid, GetUserId()))
548 : aclcheck_error(ACLCHECK_NOT_OWNER, OBJECT_FUNCTION,
549 : NameListToString(typmodoutName));
550 : if (analyzeOid && !object_ownercheck(ProcedureRelationId, analyzeOid, GetUserId()))
551 : aclcheck_error(ACLCHECK_NOT_OWNER, OBJECT_FUNCTION,
552 : NameListToString(analyzeName));
553 : if (subscriptOid && !object_ownercheck(ProcedureRelationId, subscriptOid, GetUserId()))
554 : aclcheck_error(ACLCHECK_NOT_OWNER, OBJECT_FUNCTION,
555 : NameListToString(subscriptName));
556 : #endif
557 :
558 : /*
559 : * OK, we're done checking, time to make the type. We must assign the
560 : * array type OID ahead of calling TypeCreate, since the base type and
561 : * array type each refer to the other.
562 : */
563 216 : array_oid = AssignTypeArrayOid();
564 :
565 : /*
566 : * now have TypeCreate do all the real work.
567 : *
568 : * Note: the pg_type.oid is stored in user tables as array elements (base
569 : * types) in ArrayType and in composite types in DatumTupleFields. This
570 : * oid must be preserved by binary upgrades.
571 : */
572 : address =
573 216 : TypeCreate(InvalidOid, /* no predetermined type OID */
574 : typeName, /* type name */
575 : typeNamespace, /* namespace */
576 : InvalidOid, /* relation oid (n/a here) */
577 : 0, /* relation kind (ditto) */
578 : GetUserId(), /* owner's ID */
579 : internalLength, /* internal size */
580 : TYPTYPE_BASE, /* type-type (base type) */
581 : category, /* type-category */
582 : preferred, /* is it a preferred type? */
583 : delimiter, /* array element delimiter */
584 : inputOid, /* input procedure */
585 : outputOid, /* output procedure */
586 : receiveOid, /* receive procedure */
587 : sendOid, /* send procedure */
588 : typmodinOid, /* typmodin procedure */
589 : typmodoutOid, /* typmodout procedure */
590 : analyzeOid, /* analyze procedure */
591 : subscriptOid, /* subscript procedure */
592 : elemType, /* element type ID */
593 : false, /* this is not an implicit array type */
594 : array_oid, /* array type we are about to create */
595 : InvalidOid, /* base type ID (only for domains) */
596 : defaultValue, /* default type value */
597 : NULL, /* no binary form available */
598 : byValue, /* passed by value */
599 : alignment, /* required alignment */
600 : storage, /* TOAST strategy */
601 : -1, /* typMod (Domains only) */
602 : 0, /* Array Dimensions of typbasetype */
603 : false, /* Type NOT NULL */
604 : collation); /* type's collation */
605 : Assert(typoid == address.objectId);
606 :
607 : /*
608 : * Create the array type that goes with it.
609 : */
610 216 : array_type = makeArrayTypeName(typeName, typeNamespace);
611 :
612 : /* alignment must be TYPALIGN_INT or TYPALIGN_DOUBLE for arrays */
613 216 : alignment = (alignment == TYPALIGN_DOUBLE) ? TYPALIGN_DOUBLE : TYPALIGN_INT;
614 :
615 216 : TypeCreate(array_oid, /* force assignment of this type OID */
616 : array_type, /* type name */
617 : typeNamespace, /* namespace */
618 : InvalidOid, /* relation oid (n/a here) */
619 : 0, /* relation kind (ditto) */
620 : GetUserId(), /* owner's ID */
621 : -1, /* internal size (always varlena) */
622 : TYPTYPE_BASE, /* type-type (base type) */
623 : TYPCATEGORY_ARRAY, /* type-category (array) */
624 : false, /* array types are never preferred */
625 : delimiter, /* array element delimiter */
626 : F_ARRAY_IN, /* input procedure */
627 : F_ARRAY_OUT, /* output procedure */
628 : F_ARRAY_RECV, /* receive procedure */
629 : F_ARRAY_SEND, /* send procedure */
630 : typmodinOid, /* typmodin procedure */
631 : typmodoutOid, /* typmodout procedure */
632 : F_ARRAY_TYPANALYZE, /* analyze procedure */
633 : F_ARRAY_SUBSCRIPT_HANDLER, /* array subscript procedure */
634 : typoid, /* element type ID */
635 : true, /* yes this is an array type */
636 : InvalidOid, /* no further array type */
637 : InvalidOid, /* base type ID */
638 : NULL, /* never a default type value */
639 : NULL, /* binary default isn't sent either */
640 : false, /* never passed by value */
641 : alignment, /* see above */
642 : TYPSTORAGE_EXTENDED, /* ARRAY is always toastable */
643 : -1, /* typMod (Domains only) */
644 : 0, /* Array dimensions of typbasetype */
645 : false, /* Type NOT NULL */
646 : collation); /* type's collation */
647 :
648 216 : pfree(array_type);
649 :
650 216 : return address;
651 : }
652 :
653 : /*
654 : * Guts of type deletion.
655 : */
656 : void
657 78314 : RemoveTypeById(Oid typeOid)
658 : {
659 : Relation relation;
660 : HeapTuple tup;
661 :
662 78314 : relation = table_open(TypeRelationId, RowExclusiveLock);
663 :
664 78314 : tup = SearchSysCache1(TYPEOID, ObjectIdGetDatum(typeOid));
665 78314 : if (!HeapTupleIsValid(tup))
666 0 : elog(ERROR, "cache lookup failed for type %u", typeOid);
667 :
668 78314 : CatalogTupleDelete(relation, &tup->t_self);
669 :
670 : /*
671 : * If it is an enum, delete the pg_enum entries too; we don't bother with
672 : * making dependency entries for those, so it has to be done "by hand"
673 : * here.
674 : */
675 78314 : if (((Form_pg_type) GETSTRUCT(tup))->typtype == TYPTYPE_ENUM)
676 324 : EnumValuesDelete(typeOid);
677 :
678 : /*
679 : * If it is a range type, delete the pg_range entry too; we don't bother
680 : * with making a dependency entry for that, so it has to be done "by hand"
681 : * here.
682 : */
683 78314 : if (((Form_pg_type) GETSTRUCT(tup))->typtype == TYPTYPE_RANGE)
684 104 : RangeDelete(typeOid);
685 :
686 78314 : ReleaseSysCache(tup);
687 :
688 78314 : table_close(relation, RowExclusiveLock);
689 78314 : }
690 :
691 :
692 : /*
693 : * DefineDomain
694 : * Registers a new domain.
695 : */
696 : ObjectAddress
697 1464 : DefineDomain(ParseState *pstate, CreateDomainStmt *stmt)
698 : {
699 : char *domainName;
700 : char *domainArrayName;
701 : Oid domainNamespace;
702 : AclResult aclresult;
703 : int16 internalLength;
704 : Oid inputProcedure;
705 : Oid outputProcedure;
706 : Oid receiveProcedure;
707 : Oid sendProcedure;
708 : Oid analyzeProcedure;
709 : bool byValue;
710 : char category;
711 : char delimiter;
712 : char alignment;
713 : char storage;
714 : char typtype;
715 : Datum datum;
716 : bool isnull;
717 1464 : char *defaultValue = NULL;
718 1464 : char *defaultValueBin = NULL;
719 1464 : bool saw_default = false;
720 1464 : bool typNotNull = false;
721 1464 : bool nullDefined = false;
722 1464 : int32 typNDims = list_length(stmt->typeName->arrayBounds);
723 : HeapTuple typeTup;
724 1464 : List *schema = stmt->constraints;
725 : ListCell *listptr;
726 : Oid basetypeoid;
727 : Oid old_type_oid;
728 : Oid domaincoll;
729 : Oid domainArrayOid;
730 : Form_pg_type baseType;
731 : int32 basetypeMod;
732 : Oid baseColl;
733 : ObjectAddress address;
734 :
735 : /* Convert list of names to a name and namespace */
736 1464 : domainNamespace = QualifiedNameGetCreationNamespace(stmt->domainname,
737 : &domainName);
738 :
739 : /* Check we have creation rights in target namespace */
740 1464 : aclresult = object_aclcheck(NamespaceRelationId, domainNamespace, GetUserId(),
741 : ACL_CREATE);
742 1464 : if (aclresult != ACLCHECK_OK)
743 0 : aclcheck_error(aclresult, OBJECT_SCHEMA,
744 0 : get_namespace_name(domainNamespace));
745 :
746 : /*
747 : * Check for collision with an existing type name. If there is one and
748 : * it's an autogenerated array, we can rename it out of the way.
749 : */
750 1464 : old_type_oid = GetSysCacheOid2(TYPENAMENSP, Anum_pg_type_oid,
751 : CStringGetDatum(domainName),
752 : ObjectIdGetDatum(domainNamespace));
753 1464 : if (OidIsValid(old_type_oid))
754 : {
755 0 : if (!moveArrayTypeName(old_type_oid, domainName, domainNamespace))
756 0 : ereport(ERROR,
757 : (errcode(ERRCODE_DUPLICATE_OBJECT),
758 : errmsg("type \"%s\" already exists", domainName)));
759 : }
760 :
761 : /*
762 : * Look up the base type.
763 : */
764 1464 : typeTup = typenameType(pstate, stmt->typeName, &basetypeMod);
765 1458 : baseType = (Form_pg_type) GETSTRUCT(typeTup);
766 1458 : basetypeoid = baseType->oid;
767 :
768 : /*
769 : * Base type must be a plain base type, a composite type, another domain,
770 : * an enum or a range type. Domains over pseudotypes would create a
771 : * security hole. (It would be shorter to code this to just check for
772 : * pseudotypes; but it seems safer to call out the specific typtypes that
773 : * are supported, rather than assume that all future typtypes would be
774 : * automatically supported.)
775 : */
776 1458 : typtype = baseType->typtype;
777 1458 : if (typtype != TYPTYPE_BASE &&
778 82 : typtype != TYPTYPE_COMPOSITE &&
779 32 : typtype != TYPTYPE_DOMAIN &&
780 18 : typtype != TYPTYPE_ENUM &&
781 12 : typtype != TYPTYPE_RANGE &&
782 : typtype != TYPTYPE_MULTIRANGE)
783 6 : ereport(ERROR,
784 : (errcode(ERRCODE_DATATYPE_MISMATCH),
785 : errmsg("\"%s\" is not a valid base type for a domain",
786 : TypeNameToString(stmt->typeName)),
787 : parser_errposition(pstate, stmt->typeName->location)));
788 :
789 1452 : aclresult = object_aclcheck(TypeRelationId, basetypeoid, GetUserId(), ACL_USAGE);
790 1452 : if (aclresult != ACLCHECK_OK)
791 6 : aclcheck_error_type(aclresult, basetypeoid);
792 :
793 : /*
794 : * Collect the properties of the new domain. Some are inherited from the
795 : * base type, some are not. If you change any of this inheritance
796 : * behavior, be sure to update AlterTypeRecurse() to match!
797 : */
798 :
799 : /*
800 : * Identify the collation if any
801 : */
802 1446 : baseColl = baseType->typcollation;
803 1446 : if (stmt->collClause)
804 242 : domaincoll = get_collation_oid(stmt->collClause->collname, false);
805 : else
806 1204 : domaincoll = baseColl;
807 :
808 : /* Complain if COLLATE is applied to an uncollatable type */
809 1446 : if (OidIsValid(domaincoll) && !OidIsValid(baseColl))
810 18 : ereport(ERROR,
811 : (errcode(ERRCODE_DATATYPE_MISMATCH),
812 : errmsg("collations are not supported by type %s",
813 : format_type_be(basetypeoid)),
814 : parser_errposition(pstate, stmt->typeName->location)));
815 :
816 : /* passed by value */
817 1428 : byValue = baseType->typbyval;
818 :
819 : /* Required Alignment */
820 1428 : alignment = baseType->typalign;
821 :
822 : /* TOAST Strategy */
823 1428 : storage = baseType->typstorage;
824 :
825 : /* Storage Length */
826 1428 : internalLength = baseType->typlen;
827 :
828 : /* Type Category */
829 1428 : category = baseType->typcategory;
830 :
831 : /* Array element Delimiter */
832 1428 : delimiter = baseType->typdelim;
833 :
834 : /* I/O Functions */
835 1428 : inputProcedure = F_DOMAIN_IN;
836 1428 : outputProcedure = baseType->typoutput;
837 1428 : receiveProcedure = F_DOMAIN_RECV;
838 1428 : sendProcedure = baseType->typsend;
839 :
840 : /* Domains never accept typmods, so no typmodin/typmodout needed */
841 :
842 : /* Analysis function */
843 1428 : analyzeProcedure = baseType->typanalyze;
844 :
845 : /*
846 : * Domains don't need a subscript function, since they are not
847 : * subscriptable on their own. If the base type is subscriptable, the
848 : * parser will reduce the type to the base type before subscripting.
849 : */
850 :
851 : /* Inherited default value */
852 1428 : datum = SysCacheGetAttr(TYPEOID, typeTup,
853 : Anum_pg_type_typdefault, &isnull);
854 1428 : if (!isnull)
855 0 : defaultValue = TextDatumGetCString(datum);
856 :
857 : /* Inherited default binary value */
858 1428 : datum = SysCacheGetAttr(TYPEOID, typeTup,
859 : Anum_pg_type_typdefaultbin, &isnull);
860 1428 : if (!isnull)
861 0 : defaultValueBin = TextDatumGetCString(datum);
862 :
863 : /*
864 : * Run through constraints manually to avoid the additional processing
865 : * conducted by DefineRelation() and friends.
866 : */
867 2264 : foreach(listptr, schema)
868 : {
869 914 : Constraint *constr = lfirst(listptr);
870 :
871 914 : if (!IsA(constr, Constraint))
872 0 : elog(ERROR, "unrecognized node type: %d",
873 : (int) nodeTag(constr));
874 914 : switch (constr->contype)
875 : {
876 180 : case CONSTR_DEFAULT:
877 :
878 : /*
879 : * The inherited default value may be overridden by the user
880 : * with the DEFAULT <expr> clause ... but only once.
881 : */
882 180 : if (saw_default)
883 6 : ereport(ERROR,
884 : errcode(ERRCODE_SYNTAX_ERROR),
885 : errmsg("multiple default expressions"),
886 : parser_errposition(pstate, constr->location));
887 174 : saw_default = true;
888 :
889 174 : if (constr->raw_expr)
890 : {
891 : Node *defaultExpr;
892 :
893 : /*
894 : * Cook the constr->raw_expr into an expression. Note:
895 : * name is strictly for error message
896 : */
897 174 : defaultExpr = cookDefault(pstate, constr->raw_expr,
898 : basetypeoid,
899 : basetypeMod,
900 : domainName,
901 : 0);
902 :
903 : /*
904 : * If the expression is just a NULL constant, we treat it
905 : * like not having a default.
906 : *
907 : * Note that if the basetype is another domain, we'll see
908 : * a CoerceToDomain expr here and not discard the default.
909 : * This is critical because the domain default needs to be
910 : * retained to override any default that the base domain
911 : * might have.
912 : */
913 168 : if (defaultExpr == NULL ||
914 168 : (IsA(defaultExpr, Const) &&
915 36 : ((Const *) defaultExpr)->constisnull))
916 : {
917 0 : defaultValue = NULL;
918 0 : defaultValueBin = NULL;
919 : }
920 : else
921 : {
922 : /*
923 : * Expression must be stored as a nodeToString result,
924 : * but we also require a valid textual representation
925 : * (mainly to make life easier for pg_dump).
926 : */
927 : defaultValue =
928 168 : deparse_expression(defaultExpr,
929 : NIL, false, false);
930 168 : defaultValueBin = nodeToString(defaultExpr);
931 : }
932 : }
933 : else
934 : {
935 : /* No default (can this still happen?) */
936 0 : defaultValue = NULL;
937 0 : defaultValueBin = NULL;
938 : }
939 168 : break;
940 :
941 112 : case CONSTR_NOTNULL:
942 112 : if (nullDefined)
943 : {
944 6 : if (!typNotNull)
945 0 : ereport(ERROR,
946 : errcode(ERRCODE_SYNTAX_ERROR),
947 : errmsg("conflicting NULL/NOT NULL constraints"),
948 : parser_errposition(pstate, constr->location));
949 :
950 6 : ereport(ERROR,
951 : errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
952 : errmsg("redundant NOT NULL constraint definition"),
953 : parser_errposition(pstate, constr->location));
954 : }
955 106 : if (constr->is_no_inherit)
956 6 : ereport(ERROR,
957 : errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
958 : errmsg("not-null constraints for domains cannot be marked NO INHERIT"),
959 : parser_errposition(pstate, constr->location));
960 100 : typNotNull = true;
961 100 : nullDefined = true;
962 100 : break;
963 :
964 6 : case CONSTR_NULL:
965 6 : if (nullDefined && typNotNull)
966 6 : ereport(ERROR,
967 : errcode(ERRCODE_SYNTAX_ERROR),
968 : errmsg("conflicting NULL/NOT NULL constraints"),
969 : parser_errposition(pstate, constr->location));
970 0 : typNotNull = false;
971 0 : nullDefined = true;
972 0 : break;
973 :
974 574 : case CONSTR_CHECK:
975 :
976 : /*
977 : * Check constraints are handled after domain creation, as
978 : * they require the Oid of the domain; at this point we can
979 : * only check that they're not marked NO INHERIT, because that
980 : * would be bogus.
981 : */
982 574 : if (constr->is_no_inherit)
983 6 : ereport(ERROR,
984 : errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
985 : errmsg("check constraints for domains cannot be marked NO INHERIT"),
986 : parser_errposition(pstate, constr->location));
987 :
988 568 : break;
989 :
990 : /*
991 : * All else are error cases
992 : */
993 6 : case CONSTR_UNIQUE:
994 6 : ereport(ERROR,
995 : errcode(ERRCODE_SYNTAX_ERROR),
996 : errmsg("unique constraints not possible for domains"),
997 : parser_errposition(pstate, constr->location));
998 : break;
999 :
1000 6 : case CONSTR_PRIMARY:
1001 6 : ereport(ERROR,
1002 : (errcode(ERRCODE_SYNTAX_ERROR),
1003 : errmsg("primary key constraints not possible for domains"),
1004 : parser_errposition(pstate, constr->location)));
1005 : break;
1006 :
1007 0 : case CONSTR_EXCLUSION:
1008 0 : ereport(ERROR,
1009 : (errcode(ERRCODE_SYNTAX_ERROR),
1010 : errmsg("exclusion constraints not possible for domains"),
1011 : parser_errposition(pstate, constr->location)));
1012 : break;
1013 :
1014 6 : case CONSTR_FOREIGN:
1015 6 : ereport(ERROR,
1016 : (errcode(ERRCODE_SYNTAX_ERROR),
1017 : errmsg("foreign key constraints not possible for domains"),
1018 : parser_errposition(pstate, constr->location)));
1019 : break;
1020 :
1021 6 : case CONSTR_ATTR_DEFERRABLE:
1022 : case CONSTR_ATTR_NOT_DEFERRABLE:
1023 : case CONSTR_ATTR_DEFERRED:
1024 : case CONSTR_ATTR_IMMEDIATE:
1025 6 : ereport(ERROR,
1026 : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1027 : errmsg("specifying constraint deferrability not supported for domains"),
1028 : parser_errposition(pstate, constr->location)));
1029 : break;
1030 :
1031 6 : case CONSTR_GENERATED:
1032 : case CONSTR_IDENTITY:
1033 6 : ereport(ERROR,
1034 : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1035 : errmsg("specifying GENERATED not supported for domains"),
1036 : parser_errposition(pstate, constr->location)));
1037 : break;
1038 :
1039 12 : case CONSTR_ATTR_ENFORCED:
1040 : case CONSTR_ATTR_NOT_ENFORCED:
1041 12 : ereport(ERROR,
1042 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
1043 : errmsg("specifying constraint enforceability not supported for domains"),
1044 : parser_errposition(pstate, constr->location)));
1045 : break;
1046 :
1047 : /* no default, to let compiler warn about missing case */
1048 : }
1049 : }
1050 :
1051 : /* Allocate OID for array type */
1052 1350 : domainArrayOid = AssignTypeArrayOid();
1053 :
1054 : /*
1055 : * Have TypeCreate do all the real work.
1056 : */
1057 : address =
1058 1350 : TypeCreate(InvalidOid, /* no predetermined type OID */
1059 : domainName, /* type name */
1060 : domainNamespace, /* namespace */
1061 : InvalidOid, /* relation oid (n/a here) */
1062 : 0, /* relation kind (ditto) */
1063 : GetUserId(), /* owner's ID */
1064 : internalLength, /* internal size */
1065 : TYPTYPE_DOMAIN, /* type-type (domain type) */
1066 : category, /* type-category */
1067 : false, /* domain types are never preferred */
1068 : delimiter, /* array element delimiter */
1069 : inputProcedure, /* input procedure */
1070 : outputProcedure, /* output procedure */
1071 : receiveProcedure, /* receive procedure */
1072 : sendProcedure, /* send procedure */
1073 : InvalidOid, /* typmodin procedure - none */
1074 : InvalidOid, /* typmodout procedure - none */
1075 : analyzeProcedure, /* analyze procedure */
1076 : InvalidOid, /* subscript procedure - none */
1077 : InvalidOid, /* no array element type */
1078 : false, /* this isn't an array */
1079 : domainArrayOid, /* array type we are about to create */
1080 : basetypeoid, /* base type ID */
1081 : defaultValue, /* default type value (text) */
1082 : defaultValueBin, /* default type value (binary) */
1083 : byValue, /* passed by value */
1084 : alignment, /* required alignment */
1085 : storage, /* TOAST strategy */
1086 : basetypeMod, /* typeMod value */
1087 : typNDims, /* Array dimensions for base type */
1088 : typNotNull, /* Type NOT NULL */
1089 : domaincoll); /* type's collation */
1090 :
1091 : /*
1092 : * Create the array type that goes with it.
1093 : */
1094 1350 : domainArrayName = makeArrayTypeName(domainName, domainNamespace);
1095 :
1096 : /* alignment must be TYPALIGN_INT or TYPALIGN_DOUBLE for arrays */
1097 1350 : alignment = (alignment == TYPALIGN_DOUBLE) ? TYPALIGN_DOUBLE : TYPALIGN_INT;
1098 :
1099 1350 : TypeCreate(domainArrayOid, /* force assignment of this type OID */
1100 : domainArrayName, /* type name */
1101 : domainNamespace, /* namespace */
1102 : InvalidOid, /* relation oid (n/a here) */
1103 : 0, /* relation kind (ditto) */
1104 : GetUserId(), /* owner's ID */
1105 : -1, /* internal size (always varlena) */
1106 : TYPTYPE_BASE, /* type-type (base type) */
1107 : TYPCATEGORY_ARRAY, /* type-category (array) */
1108 : false, /* array types are never preferred */
1109 : delimiter, /* array element delimiter */
1110 : F_ARRAY_IN, /* input procedure */
1111 : F_ARRAY_OUT, /* output procedure */
1112 : F_ARRAY_RECV, /* receive procedure */
1113 : F_ARRAY_SEND, /* send procedure */
1114 : InvalidOid, /* typmodin procedure - none */
1115 : InvalidOid, /* typmodout procedure - none */
1116 : F_ARRAY_TYPANALYZE, /* analyze procedure */
1117 : F_ARRAY_SUBSCRIPT_HANDLER, /* array subscript procedure */
1118 : address.objectId, /* element type ID */
1119 : true, /* yes this is an array type */
1120 : InvalidOid, /* no further array type */
1121 : InvalidOid, /* base type ID */
1122 : NULL, /* never a default type value */
1123 : NULL, /* binary default isn't sent either */
1124 : false, /* never passed by value */
1125 : alignment, /* see above */
1126 : TYPSTORAGE_EXTENDED, /* ARRAY is always toastable */
1127 : -1, /* typMod (Domains only) */
1128 : 0, /* Array dimensions of typbasetype */
1129 : false, /* Type NOT NULL */
1130 : domaincoll); /* type's collation */
1131 :
1132 1350 : pfree(domainArrayName);
1133 :
1134 : /*
1135 : * Process constraints which refer to the domain ID returned by TypeCreate
1136 : */
1137 2144 : foreach(listptr, schema)
1138 : {
1139 794 : Constraint *constr = lfirst(listptr);
1140 :
1141 : /* it must be a Constraint, per check above */
1142 :
1143 794 : switch (constr->contype)
1144 : {
1145 550 : case CONSTR_CHECK:
1146 550 : domainAddCheckConstraint(address.objectId, domainNamespace,
1147 : basetypeoid, basetypeMod,
1148 : constr, domainName, NULL);
1149 550 : break;
1150 :
1151 82 : case CONSTR_NOTNULL:
1152 82 : domainAddNotNullConstraint(address.objectId, domainNamespace,
1153 : basetypeoid, basetypeMod,
1154 : constr, domainName, NULL);
1155 82 : break;
1156 :
1157 : /* Other constraint types were fully processed above */
1158 :
1159 162 : default:
1160 162 : break;
1161 : }
1162 :
1163 : /* CCI so we can detect duplicate constraint names */
1164 794 : CommandCounterIncrement();
1165 : }
1166 :
1167 : /*
1168 : * Now we can clean up.
1169 : */
1170 1350 : ReleaseSysCache(typeTup);
1171 :
1172 1350 : return address;
1173 : }
1174 :
1175 :
1176 : /*
1177 : * DefineEnum
1178 : * Registers a new enum.
1179 : */
1180 : ObjectAddress
1181 444 : DefineEnum(CreateEnumStmt *stmt)
1182 : {
1183 : char *enumName;
1184 : char *enumArrayName;
1185 : Oid enumNamespace;
1186 : AclResult aclresult;
1187 : Oid old_type_oid;
1188 : Oid enumArrayOid;
1189 : ObjectAddress enumTypeAddr;
1190 :
1191 : /* Convert list of names to a name and namespace */
1192 444 : enumNamespace = QualifiedNameGetCreationNamespace(stmt->typeName,
1193 : &enumName);
1194 :
1195 : /* Check we have creation rights in target namespace */
1196 444 : aclresult = object_aclcheck(NamespaceRelationId, enumNamespace, GetUserId(), ACL_CREATE);
1197 444 : if (aclresult != ACLCHECK_OK)
1198 0 : aclcheck_error(aclresult, OBJECT_SCHEMA,
1199 0 : get_namespace_name(enumNamespace));
1200 :
1201 : /*
1202 : * Check for collision with an existing type name. If there is one and
1203 : * it's an autogenerated array, we can rename it out of the way.
1204 : */
1205 444 : old_type_oid = GetSysCacheOid2(TYPENAMENSP, Anum_pg_type_oid,
1206 : CStringGetDatum(enumName),
1207 : ObjectIdGetDatum(enumNamespace));
1208 444 : if (OidIsValid(old_type_oid))
1209 : {
1210 8 : if (!moveArrayTypeName(old_type_oid, enumName, enumNamespace))
1211 0 : ereport(ERROR,
1212 : (errcode(ERRCODE_DUPLICATE_OBJECT),
1213 : errmsg("type \"%s\" already exists", enumName)));
1214 : }
1215 :
1216 : /* Allocate OID for array type */
1217 444 : enumArrayOid = AssignTypeArrayOid();
1218 :
1219 : /* Create the pg_type entry */
1220 : enumTypeAddr =
1221 444 : TypeCreate(InvalidOid, /* no predetermined type OID */
1222 : enumName, /* type name */
1223 : enumNamespace, /* namespace */
1224 : InvalidOid, /* relation oid (n/a here) */
1225 : 0, /* relation kind (ditto) */
1226 : GetUserId(), /* owner's ID */
1227 : sizeof(Oid), /* internal size */
1228 : TYPTYPE_ENUM, /* type-type (enum type) */
1229 : TYPCATEGORY_ENUM, /* type-category (enum type) */
1230 : false, /* enum types are never preferred */
1231 : DEFAULT_TYPDELIM, /* array element delimiter */
1232 : F_ENUM_IN, /* input procedure */
1233 : F_ENUM_OUT, /* output procedure */
1234 : F_ENUM_RECV, /* receive procedure */
1235 : F_ENUM_SEND, /* send procedure */
1236 : InvalidOid, /* typmodin procedure - none */
1237 : InvalidOid, /* typmodout procedure - none */
1238 : InvalidOid, /* analyze procedure - default */
1239 : InvalidOid, /* subscript procedure - none */
1240 : InvalidOid, /* element type ID */
1241 : false, /* this is not an array type */
1242 : enumArrayOid, /* array type we are about to create */
1243 : InvalidOid, /* base type ID (only for domains) */
1244 : NULL, /* never a default type value */
1245 : NULL, /* binary default isn't sent either */
1246 : true, /* always passed by value */
1247 : TYPALIGN_INT, /* int alignment */
1248 : TYPSTORAGE_PLAIN, /* TOAST strategy always plain */
1249 : -1, /* typMod (Domains only) */
1250 : 0, /* Array dimensions of typbasetype */
1251 : false, /* Type NOT NULL */
1252 : InvalidOid); /* type's collation */
1253 :
1254 : /* Enter the enum's values into pg_enum */
1255 442 : EnumValuesCreate(enumTypeAddr.objectId, stmt->vals);
1256 :
1257 : /*
1258 : * Create the array type that goes with it.
1259 : */
1260 436 : enumArrayName = makeArrayTypeName(enumName, enumNamespace);
1261 :
1262 436 : TypeCreate(enumArrayOid, /* force assignment of this type OID */
1263 : enumArrayName, /* type name */
1264 : enumNamespace, /* namespace */
1265 : InvalidOid, /* relation oid (n/a here) */
1266 : 0, /* relation kind (ditto) */
1267 : GetUserId(), /* owner's ID */
1268 : -1, /* internal size (always varlena) */
1269 : TYPTYPE_BASE, /* type-type (base type) */
1270 : TYPCATEGORY_ARRAY, /* type-category (array) */
1271 : false, /* array types are never preferred */
1272 : DEFAULT_TYPDELIM, /* array element delimiter */
1273 : F_ARRAY_IN, /* input procedure */
1274 : F_ARRAY_OUT, /* output procedure */
1275 : F_ARRAY_RECV, /* receive procedure */
1276 : F_ARRAY_SEND, /* send procedure */
1277 : InvalidOid, /* typmodin procedure - none */
1278 : InvalidOid, /* typmodout procedure - none */
1279 : F_ARRAY_TYPANALYZE, /* analyze procedure */
1280 : F_ARRAY_SUBSCRIPT_HANDLER, /* array subscript procedure */
1281 : enumTypeAddr.objectId, /* element type ID */
1282 : true, /* yes this is an array type */
1283 : InvalidOid, /* no further array type */
1284 : InvalidOid, /* base type ID */
1285 : NULL, /* never a default type value */
1286 : NULL, /* binary default isn't sent either */
1287 : false, /* never passed by value */
1288 : TYPALIGN_INT, /* enums have int align, so do their arrays */
1289 : TYPSTORAGE_EXTENDED, /* ARRAY is always toastable */
1290 : -1, /* typMod (Domains only) */
1291 : 0, /* Array dimensions of typbasetype */
1292 : false, /* Type NOT NULL */
1293 : InvalidOid); /* type's collation */
1294 :
1295 436 : pfree(enumArrayName);
1296 :
1297 436 : return enumTypeAddr;
1298 : }
1299 :
1300 : /*
1301 : * AlterEnum
1302 : * Adds a new label to an existing enum.
1303 : */
1304 : ObjectAddress
1305 396 : AlterEnum(AlterEnumStmt *stmt)
1306 : {
1307 : Oid enum_type_oid;
1308 : TypeName *typename;
1309 : HeapTuple tup;
1310 : ObjectAddress address;
1311 :
1312 : /* Make a TypeName so we can use standard type lookup machinery */
1313 396 : typename = makeTypeNameFromNameList(stmt->typeName);
1314 396 : enum_type_oid = typenameTypeId(NULL, typename);
1315 :
1316 396 : tup = SearchSysCache1(TYPEOID, ObjectIdGetDatum(enum_type_oid));
1317 396 : if (!HeapTupleIsValid(tup))
1318 0 : elog(ERROR, "cache lookup failed for type %u", enum_type_oid);
1319 :
1320 : /* Check it's an enum and check user has permission to ALTER the enum */
1321 396 : checkEnumOwner(tup);
1322 :
1323 396 : ReleaseSysCache(tup);
1324 :
1325 396 : if (stmt->oldVal)
1326 : {
1327 : /* Rename an existing label */
1328 24 : RenameEnumLabel(enum_type_oid, stmt->oldVal, stmt->newVal);
1329 : }
1330 : else
1331 : {
1332 : /* Add a new label */
1333 372 : AddEnumLabel(enum_type_oid, stmt->newVal,
1334 372 : stmt->newValNeighbor, stmt->newValIsAfter,
1335 372 : stmt->skipIfNewValExists);
1336 : }
1337 :
1338 366 : InvokeObjectPostAlterHook(TypeRelationId, enum_type_oid, 0);
1339 :
1340 366 : ObjectAddressSet(address, TypeRelationId, enum_type_oid);
1341 :
1342 366 : return address;
1343 : }
1344 :
1345 :
1346 : /*
1347 : * checkEnumOwner
1348 : *
1349 : * Check that the type is actually an enum and that the current user
1350 : * has permission to do ALTER TYPE on it. Throw an error if not.
1351 : */
1352 : static void
1353 396 : checkEnumOwner(HeapTuple tup)
1354 : {
1355 396 : Form_pg_type typTup = (Form_pg_type) GETSTRUCT(tup);
1356 :
1357 : /* Check that this is actually an enum */
1358 396 : if (typTup->typtype != TYPTYPE_ENUM)
1359 0 : ereport(ERROR,
1360 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
1361 : errmsg("%s is not an enum",
1362 : format_type_be(typTup->oid))));
1363 :
1364 : /* Permission check: must own type */
1365 396 : if (!object_ownercheck(TypeRelationId, typTup->oid, GetUserId()))
1366 0 : aclcheck_error_type(ACLCHECK_NOT_OWNER, typTup->oid);
1367 396 : }
1368 :
1369 :
1370 : /*
1371 : * DefineRange
1372 : * Registers a new range type.
1373 : *
1374 : * Perhaps it might be worthwhile to set pg_type.typelem to the base type,
1375 : * and likewise on multiranges to set it to the range type. But having a
1376 : * non-zero typelem is treated elsewhere as a synonym for being an array,
1377 : * and users might have queries with that same assumption.
1378 : */
1379 : ObjectAddress
1380 184 : DefineRange(ParseState *pstate, CreateRangeStmt *stmt)
1381 : {
1382 : char *typeName;
1383 : Oid typeNamespace;
1384 : Oid typoid;
1385 : char *rangeArrayName;
1386 184 : char *multirangeTypeName = NULL;
1387 : char *multirangeArrayName;
1388 184 : Oid multirangeNamespace = InvalidOid;
1389 : Oid rangeArrayOid;
1390 : Oid multirangeOid;
1391 : Oid multirangeArrayOid;
1392 184 : Oid rangeSubtype = InvalidOid;
1393 184 : List *rangeSubOpclassName = NIL;
1394 184 : List *rangeCollationName = NIL;
1395 184 : List *rangeCanonicalName = NIL;
1396 184 : List *rangeSubtypeDiffName = NIL;
1397 : Oid rangeSubOpclass;
1398 : Oid rangeCollation;
1399 : regproc rangeCanonical;
1400 : regproc rangeSubtypeDiff;
1401 : int16 subtyplen;
1402 : bool subtypbyval;
1403 : char subtypalign;
1404 : char alignment;
1405 : AclResult aclresult;
1406 : ListCell *lc;
1407 : ObjectAddress address;
1408 : ObjectAddress mltrngaddress PG_USED_FOR_ASSERTS_ONLY;
1409 : Oid castFuncOid;
1410 :
1411 : /* Convert list of names to a name and namespace */
1412 184 : typeNamespace = QualifiedNameGetCreationNamespace(stmt->typeName,
1413 : &typeName);
1414 :
1415 : /* Check we have creation rights in target namespace */
1416 184 : aclresult = object_aclcheck(NamespaceRelationId, typeNamespace, GetUserId(), ACL_CREATE);
1417 184 : if (aclresult != ACLCHECK_OK)
1418 0 : aclcheck_error(aclresult, OBJECT_SCHEMA,
1419 0 : get_namespace_name(typeNamespace));
1420 :
1421 : /*
1422 : * Look to see if type already exists.
1423 : */
1424 184 : typoid = GetSysCacheOid2(TYPENAMENSP, Anum_pg_type_oid,
1425 : CStringGetDatum(typeName),
1426 : ObjectIdGetDatum(typeNamespace));
1427 :
1428 : /*
1429 : * If it's not a shell, see if it's an autogenerated array type, and if so
1430 : * rename it out of the way.
1431 : */
1432 184 : if (OidIsValid(typoid) && get_typisdefined(typoid))
1433 : {
1434 0 : if (moveArrayTypeName(typoid, typeName, typeNamespace))
1435 0 : typoid = InvalidOid;
1436 : else
1437 0 : ereport(ERROR,
1438 : (errcode(ERRCODE_DUPLICATE_OBJECT),
1439 : errmsg("type \"%s\" already exists", typeName)));
1440 : }
1441 :
1442 : /*
1443 : * Unlike DefineType(), we don't insist on a shell type existing first, as
1444 : * it's only needed if the user wants to specify a canonical function.
1445 : */
1446 :
1447 : /* Extract the parameters from the parameter list */
1448 496 : foreach(lc, stmt->params)
1449 : {
1450 312 : DefElem *defel = (DefElem *) lfirst(lc);
1451 :
1452 312 : if (strcmp(defel->defname, "subtype") == 0)
1453 : {
1454 184 : if (OidIsValid(rangeSubtype))
1455 0 : errorConflictingDefElem(defel, pstate);
1456 : /* we can look up the subtype name immediately */
1457 184 : rangeSubtype = typenameTypeId(NULL, defGetTypeName(defel));
1458 : }
1459 128 : else if (strcmp(defel->defname, "subtype_opclass") == 0)
1460 : {
1461 8 : if (rangeSubOpclassName != NIL)
1462 0 : errorConflictingDefElem(defel, pstate);
1463 8 : rangeSubOpclassName = defGetQualifiedName(defel);
1464 : }
1465 120 : else if (strcmp(defel->defname, "collation") == 0)
1466 : {
1467 70 : if (rangeCollationName != NIL)
1468 0 : errorConflictingDefElem(defel, pstate);
1469 70 : rangeCollationName = defGetQualifiedName(defel);
1470 : }
1471 50 : else if (strcmp(defel->defname, "canonical") == 0)
1472 : {
1473 0 : if (rangeCanonicalName != NIL)
1474 0 : errorConflictingDefElem(defel, pstate);
1475 0 : rangeCanonicalName = defGetQualifiedName(defel);
1476 : }
1477 50 : else if (strcmp(defel->defname, "subtype_diff") == 0)
1478 : {
1479 14 : if (rangeSubtypeDiffName != NIL)
1480 0 : errorConflictingDefElem(defel, pstate);
1481 14 : rangeSubtypeDiffName = defGetQualifiedName(defel);
1482 : }
1483 36 : else if (strcmp(defel->defname, "multirange_type_name") == 0)
1484 : {
1485 36 : if (multirangeTypeName != NULL)
1486 0 : errorConflictingDefElem(defel, pstate);
1487 : /* we can look up the subtype name immediately */
1488 36 : multirangeNamespace = QualifiedNameGetCreationNamespace(defGetQualifiedName(defel),
1489 : &multirangeTypeName);
1490 : }
1491 : else
1492 0 : ereport(ERROR,
1493 : (errcode(ERRCODE_SYNTAX_ERROR),
1494 : errmsg("type attribute \"%s\" not recognized",
1495 : defel->defname)));
1496 : }
1497 :
1498 : /* Must have a subtype */
1499 184 : if (!OidIsValid(rangeSubtype))
1500 0 : ereport(ERROR,
1501 : (errcode(ERRCODE_SYNTAX_ERROR),
1502 : errmsg("type attribute \"subtype\" is required")));
1503 : /* disallow ranges of pseudotypes */
1504 184 : if (get_typtype(rangeSubtype) == TYPTYPE_PSEUDO)
1505 0 : ereport(ERROR,
1506 : (errcode(ERRCODE_DATATYPE_MISMATCH),
1507 : errmsg("range subtype cannot be %s",
1508 : format_type_be(rangeSubtype))));
1509 :
1510 : /* Identify subopclass */
1511 184 : rangeSubOpclass = findRangeSubOpclass(rangeSubOpclassName, rangeSubtype);
1512 :
1513 : /* Identify collation to use, if any */
1514 184 : if (type_is_collatable(rangeSubtype))
1515 : {
1516 78 : if (rangeCollationName != NIL)
1517 70 : rangeCollation = get_collation_oid(rangeCollationName, false);
1518 : else
1519 8 : rangeCollation = get_typcollation(rangeSubtype);
1520 : }
1521 : else
1522 : {
1523 106 : if (rangeCollationName != NIL)
1524 0 : ereport(ERROR,
1525 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
1526 : errmsg("range collation specified but subtype does not support collation")));
1527 106 : rangeCollation = InvalidOid;
1528 : }
1529 :
1530 : /* Identify support functions, if provided */
1531 184 : if (rangeCanonicalName != NIL)
1532 : {
1533 0 : if (!OidIsValid(typoid))
1534 0 : ereport(ERROR,
1535 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
1536 : errmsg("cannot specify a canonical function without a pre-created shell type"),
1537 : errhint("Create the type as a shell type, then create its canonicalization function, then do a full CREATE TYPE.")));
1538 0 : rangeCanonical = findRangeCanonicalFunction(rangeCanonicalName,
1539 : typoid);
1540 : }
1541 : else
1542 184 : rangeCanonical = InvalidOid;
1543 :
1544 184 : if (rangeSubtypeDiffName != NIL)
1545 14 : rangeSubtypeDiff = findRangeSubtypeDiffFunction(rangeSubtypeDiffName,
1546 : rangeSubtype);
1547 : else
1548 170 : rangeSubtypeDiff = InvalidOid;
1549 :
1550 178 : get_typlenbyvalalign(rangeSubtype,
1551 : &subtyplen, &subtypbyval, &subtypalign);
1552 :
1553 : /* alignment must be TYPALIGN_INT or TYPALIGN_DOUBLE for ranges */
1554 178 : alignment = (subtypalign == TYPALIGN_DOUBLE) ? TYPALIGN_DOUBLE : TYPALIGN_INT;
1555 :
1556 : /* Allocate OID for array type, its multirange, and its multirange array */
1557 178 : rangeArrayOid = AssignTypeArrayOid();
1558 178 : multirangeOid = AssignTypeMultirangeOid();
1559 178 : multirangeArrayOid = AssignTypeMultirangeArrayOid();
1560 :
1561 : /* Create the pg_type entry */
1562 : address =
1563 178 : TypeCreate(InvalidOid, /* no predetermined type OID */
1564 : typeName, /* type name */
1565 : typeNamespace, /* namespace */
1566 : InvalidOid, /* relation oid (n/a here) */
1567 : 0, /* relation kind (ditto) */
1568 : GetUserId(), /* owner's ID */
1569 : -1, /* internal size (always varlena) */
1570 : TYPTYPE_RANGE, /* type-type (range type) */
1571 : TYPCATEGORY_RANGE, /* type-category (range type) */
1572 : false, /* range types are never preferred */
1573 : DEFAULT_TYPDELIM, /* array element delimiter */
1574 : F_RANGE_IN, /* input procedure */
1575 : F_RANGE_OUT, /* output procedure */
1576 : F_RANGE_RECV, /* receive procedure */
1577 : F_RANGE_SEND, /* send procedure */
1578 : InvalidOid, /* typmodin procedure - none */
1579 : InvalidOid, /* typmodout procedure - none */
1580 : F_RANGE_TYPANALYZE, /* analyze procedure */
1581 : InvalidOid, /* subscript procedure - none */
1582 : InvalidOid, /* element type ID - none */
1583 : false, /* this is not an array type */
1584 : rangeArrayOid, /* array type we are about to create */
1585 : InvalidOid, /* base type ID (only for domains) */
1586 : NULL, /* never a default type value */
1587 : NULL, /* no binary form available either */
1588 : false, /* never passed by value */
1589 : alignment, /* alignment */
1590 : TYPSTORAGE_EXTENDED, /* TOAST strategy (always extended) */
1591 : -1, /* typMod (Domains only) */
1592 : 0, /* Array dimensions of typbasetype */
1593 : false, /* Type NOT NULL */
1594 : InvalidOid); /* type's collation (ranges never have one) */
1595 : Assert(typoid == InvalidOid || typoid == address.objectId);
1596 178 : typoid = address.objectId;
1597 :
1598 : /* Create the multirange that goes with it */
1599 178 : if (multirangeTypeName)
1600 : {
1601 : Oid old_typoid;
1602 :
1603 : /*
1604 : * Look to see if multirange type already exists.
1605 : */
1606 36 : old_typoid = GetSysCacheOid2(TYPENAMENSP, Anum_pg_type_oid,
1607 : CStringGetDatum(multirangeTypeName),
1608 : ObjectIdGetDatum(multirangeNamespace));
1609 :
1610 : /*
1611 : * If it's not a shell, see if it's an autogenerated array type, and
1612 : * if so rename it out of the way.
1613 : */
1614 36 : if (OidIsValid(old_typoid) && get_typisdefined(old_typoid))
1615 : {
1616 12 : if (!moveArrayTypeName(old_typoid, multirangeTypeName, multirangeNamespace))
1617 6 : ereport(ERROR,
1618 : (errcode(ERRCODE_DUPLICATE_OBJECT),
1619 : errmsg("type \"%s\" already exists", multirangeTypeName)));
1620 : }
1621 : }
1622 : else
1623 : {
1624 : /* Generate multirange name automatically */
1625 142 : multirangeNamespace = typeNamespace;
1626 142 : multirangeTypeName = makeMultirangeTypeName(typeName, multirangeNamespace);
1627 : }
1628 :
1629 : mltrngaddress =
1630 160 : TypeCreate(multirangeOid, /* force assignment of this type OID */
1631 : multirangeTypeName, /* type name */
1632 : multirangeNamespace, /* namespace */
1633 : InvalidOid, /* relation oid (n/a here) */
1634 : 0, /* relation kind (ditto) */
1635 : GetUserId(), /* owner's ID */
1636 : -1, /* internal size (always varlena) */
1637 : TYPTYPE_MULTIRANGE, /* type-type (multirange type) */
1638 : TYPCATEGORY_RANGE, /* type-category (range type) */
1639 : false, /* multirange types are never preferred */
1640 : DEFAULT_TYPDELIM, /* array element delimiter */
1641 : F_MULTIRANGE_IN, /* input procedure */
1642 : F_MULTIRANGE_OUT, /* output procedure */
1643 : F_MULTIRANGE_RECV, /* receive procedure */
1644 : F_MULTIRANGE_SEND, /* send procedure */
1645 : InvalidOid, /* typmodin procedure - none */
1646 : InvalidOid, /* typmodout procedure - none */
1647 : F_MULTIRANGE_TYPANALYZE, /* analyze procedure */
1648 : InvalidOid, /* subscript procedure - none */
1649 : InvalidOid, /* element type ID - none */
1650 : false, /* this is not an array type */
1651 : multirangeArrayOid, /* array type we are about to create */
1652 : InvalidOid, /* base type ID (only for domains) */
1653 : NULL, /* never a default type value */
1654 : NULL, /* no binary form available either */
1655 : false, /* never passed by value */
1656 : alignment, /* alignment */
1657 : 'x', /* TOAST strategy (always extended) */
1658 : -1, /* typMod (Domains only) */
1659 : 0, /* Array dimensions of typbasetype */
1660 : false, /* Type NOT NULL */
1661 : InvalidOid); /* type's collation (ranges never have one) */
1662 : Assert(multirangeOid == mltrngaddress.objectId);
1663 :
1664 : /* Create the entry in pg_range */
1665 160 : RangeCreate(typoid, rangeSubtype, rangeCollation, rangeSubOpclass,
1666 : rangeCanonical, rangeSubtypeDiff, multirangeOid);
1667 :
1668 : /*
1669 : * Create the array type that goes with it.
1670 : */
1671 160 : rangeArrayName = makeArrayTypeName(typeName, typeNamespace);
1672 :
1673 160 : TypeCreate(rangeArrayOid, /* force assignment of this type OID */
1674 : rangeArrayName, /* type name */
1675 : typeNamespace, /* namespace */
1676 : InvalidOid, /* relation oid (n/a here) */
1677 : 0, /* relation kind (ditto) */
1678 : GetUserId(), /* owner's ID */
1679 : -1, /* internal size (always varlena) */
1680 : TYPTYPE_BASE, /* type-type (base type) */
1681 : TYPCATEGORY_ARRAY, /* type-category (array) */
1682 : false, /* array types are never preferred */
1683 : DEFAULT_TYPDELIM, /* array element delimiter */
1684 : F_ARRAY_IN, /* input procedure */
1685 : F_ARRAY_OUT, /* output procedure */
1686 : F_ARRAY_RECV, /* receive procedure */
1687 : F_ARRAY_SEND, /* send procedure */
1688 : InvalidOid, /* typmodin procedure - none */
1689 : InvalidOid, /* typmodout procedure - none */
1690 : F_ARRAY_TYPANALYZE, /* analyze procedure */
1691 : F_ARRAY_SUBSCRIPT_HANDLER, /* array subscript procedure */
1692 : typoid, /* element type ID */
1693 : true, /* yes this is an array type */
1694 : InvalidOid, /* no further array type */
1695 : InvalidOid, /* base type ID */
1696 : NULL, /* never a default type value */
1697 : NULL, /* binary default isn't sent either */
1698 : false, /* never passed by value */
1699 : alignment, /* alignment - same as range's */
1700 : TYPSTORAGE_EXTENDED, /* ARRAY is always toastable */
1701 : -1, /* typMod (Domains only) */
1702 : 0, /* Array dimensions of typbasetype */
1703 : false, /* Type NOT NULL */
1704 : InvalidOid); /* typcollation */
1705 :
1706 160 : pfree(rangeArrayName);
1707 :
1708 : /* Create the multirange's array type */
1709 :
1710 160 : multirangeArrayName = makeArrayTypeName(multirangeTypeName, typeNamespace);
1711 :
1712 160 : TypeCreate(multirangeArrayOid, /* force assignment of this type OID */
1713 : multirangeArrayName, /* type name */
1714 : multirangeNamespace, /* namespace */
1715 : InvalidOid, /* relation oid (n/a here) */
1716 : 0, /* relation kind (ditto) */
1717 : GetUserId(), /* owner's ID */
1718 : -1, /* internal size (always varlena) */
1719 : TYPTYPE_BASE, /* type-type (base type) */
1720 : TYPCATEGORY_ARRAY, /* type-category (array) */
1721 : false, /* array types are never preferred */
1722 : DEFAULT_TYPDELIM, /* array element delimiter */
1723 : F_ARRAY_IN, /* input procedure */
1724 : F_ARRAY_OUT, /* output procedure */
1725 : F_ARRAY_RECV, /* receive procedure */
1726 : F_ARRAY_SEND, /* send procedure */
1727 : InvalidOid, /* typmodin procedure - none */
1728 : InvalidOid, /* typmodout procedure - none */
1729 : F_ARRAY_TYPANALYZE, /* analyze procedure */
1730 : F_ARRAY_SUBSCRIPT_HANDLER, /* array subscript procedure */
1731 : multirangeOid, /* element type ID */
1732 : true, /* yes this is an array type */
1733 : InvalidOid, /* no further array type */
1734 : InvalidOid, /* base type ID */
1735 : NULL, /* never a default type value */
1736 : NULL, /* binary default isn't sent either */
1737 : false, /* never passed by value */
1738 : alignment, /* alignment - same as range's */
1739 : 'x', /* ARRAY is always toastable */
1740 : -1, /* typMod (Domains only) */
1741 : 0, /* Array dimensions of typbasetype */
1742 : false, /* Type NOT NULL */
1743 : InvalidOid); /* typcollation */
1744 :
1745 : /* And create the constructor functions for this range type */
1746 160 : makeRangeConstructors(typeName, typeNamespace, typoid, rangeSubtype);
1747 160 : makeMultirangeConstructors(multirangeTypeName, typeNamespace,
1748 : multirangeOid, typoid, rangeArrayOid,
1749 : &castFuncOid);
1750 :
1751 : /* Create cast from the range type to its multirange type */
1752 160 : CastCreate(typoid, multirangeOid, castFuncOid, InvalidOid, InvalidOid,
1753 : COERCION_CODE_EXPLICIT, COERCION_METHOD_FUNCTION,
1754 : DEPENDENCY_INTERNAL);
1755 :
1756 160 : pfree(multirangeArrayName);
1757 :
1758 160 : return address;
1759 : }
1760 :
1761 : /*
1762 : * Because there may exist several range types over the same subtype, the
1763 : * range type can't be uniquely determined from the subtype. So it's
1764 : * impossible to define a polymorphic constructor; we have to generate new
1765 : * constructor functions explicitly for each range type.
1766 : *
1767 : * We actually define 4 functions, with 0 through 3 arguments. This is just
1768 : * to offer more convenience for the user.
1769 : */
1770 : static void
1771 160 : makeRangeConstructors(const char *name, Oid namespace,
1772 : Oid rangeOid, Oid subtype)
1773 : {
1774 : static const char *const prosrc[2] = {"range_constructor2",
1775 : "range_constructor3"};
1776 : static const int pronargs[2] = {2, 3};
1777 :
1778 : Oid constructorArgTypes[3];
1779 : ObjectAddress myself,
1780 : referenced;
1781 : int i;
1782 :
1783 160 : constructorArgTypes[0] = subtype;
1784 160 : constructorArgTypes[1] = subtype;
1785 160 : constructorArgTypes[2] = TEXTOID;
1786 :
1787 160 : referenced.classId = TypeRelationId;
1788 160 : referenced.objectId = rangeOid;
1789 160 : referenced.objectSubId = 0;
1790 :
1791 480 : for (i = 0; i < lengthof(prosrc); i++)
1792 : {
1793 : oidvector *constructorArgTypesVector;
1794 :
1795 320 : constructorArgTypesVector = buildoidvector(constructorArgTypes,
1796 320 : pronargs[i]);
1797 :
1798 320 : myself = ProcedureCreate(name, /* name: same as range type */
1799 : namespace, /* namespace */
1800 : false, /* replace */
1801 : false, /* returns set */
1802 : rangeOid, /* return type */
1803 : BOOTSTRAP_SUPERUSERID, /* proowner */
1804 : INTERNALlanguageId, /* language */
1805 : F_FMGR_INTERNAL_VALIDATOR, /* language validator */
1806 320 : prosrc[i], /* prosrc */
1807 : NULL, /* probin */
1808 : NULL, /* prosqlbody */
1809 : PROKIND_FUNCTION,
1810 : false, /* security_definer */
1811 : false, /* leakproof */
1812 : false, /* isStrict */
1813 : PROVOLATILE_IMMUTABLE, /* volatility */
1814 : PROPARALLEL_SAFE, /* parallel safety */
1815 : constructorArgTypesVector, /* parameterTypes */
1816 : PointerGetDatum(NULL), /* allParameterTypes */
1817 : PointerGetDatum(NULL), /* parameterModes */
1818 : PointerGetDatum(NULL), /* parameterNames */
1819 : NIL, /* parameterDefaults */
1820 : PointerGetDatum(NULL), /* trftypes */
1821 : NIL, /* trfoids */
1822 : PointerGetDatum(NULL), /* proconfig */
1823 : InvalidOid, /* prosupport */
1824 : 1.0, /* procost */
1825 : 0.0); /* prorows */
1826 :
1827 : /*
1828 : * Make the constructors internally-dependent on the range type so
1829 : * that they go away silently when the type is dropped. Note that
1830 : * pg_dump depends on this choice to avoid dumping the constructors.
1831 : */
1832 320 : recordDependencyOn(&myself, &referenced, DEPENDENCY_INTERNAL);
1833 : }
1834 160 : }
1835 :
1836 : /*
1837 : * We make a separate multirange constructor for each range type
1838 : * so its name can include the base type, like range constructors do.
1839 : * If we had an anyrangearray polymorphic type we could use it here,
1840 : * but since each type has its own constructor name there's no need.
1841 : *
1842 : * Sets castFuncOid to the oid of the new constructor that can be used
1843 : * to cast from a range to a multirange.
1844 : */
1845 : static void
1846 160 : makeMultirangeConstructors(const char *name, Oid namespace,
1847 : Oid multirangeOid, Oid rangeOid, Oid rangeArrayOid,
1848 : Oid *castFuncOid)
1849 : {
1850 : ObjectAddress myself,
1851 : referenced;
1852 : oidvector *argtypes;
1853 : Datum allParamTypes;
1854 : ArrayType *allParameterTypes;
1855 : Datum paramModes;
1856 : ArrayType *parameterModes;
1857 :
1858 160 : referenced.classId = TypeRelationId;
1859 160 : referenced.objectId = multirangeOid;
1860 160 : referenced.objectSubId = 0;
1861 :
1862 : /* 0-arg constructor - for empty multiranges */
1863 160 : argtypes = buildoidvector(NULL, 0);
1864 160 : myself = ProcedureCreate(name, /* name: same as multirange type */
1865 : namespace,
1866 : false, /* replace */
1867 : false, /* returns set */
1868 : multirangeOid, /* return type */
1869 : BOOTSTRAP_SUPERUSERID, /* proowner */
1870 : INTERNALlanguageId, /* language */
1871 : F_FMGR_INTERNAL_VALIDATOR,
1872 : "multirange_constructor0", /* prosrc */
1873 : NULL, /* probin */
1874 : NULL, /* prosqlbody */
1875 : PROKIND_FUNCTION,
1876 : false, /* security_definer */
1877 : false, /* leakproof */
1878 : true, /* isStrict */
1879 : PROVOLATILE_IMMUTABLE, /* volatility */
1880 : PROPARALLEL_SAFE, /* parallel safety */
1881 : argtypes, /* parameterTypes */
1882 : PointerGetDatum(NULL), /* allParameterTypes */
1883 : PointerGetDatum(NULL), /* parameterModes */
1884 : PointerGetDatum(NULL), /* parameterNames */
1885 : NIL, /* parameterDefaults */
1886 : PointerGetDatum(NULL), /* trftypes */
1887 : NIL, /* trfoids */
1888 : PointerGetDatum(NULL), /* proconfig */
1889 : InvalidOid, /* prosupport */
1890 : 1.0, /* procost */
1891 : 0.0); /* prorows */
1892 :
1893 : /*
1894 : * Make the constructor internally-dependent on the multirange type so
1895 : * that they go away silently when the type is dropped. Note that pg_dump
1896 : * depends on this choice to avoid dumping the constructors.
1897 : */
1898 160 : recordDependencyOn(&myself, &referenced, DEPENDENCY_INTERNAL);
1899 160 : pfree(argtypes);
1900 :
1901 : /*
1902 : * 1-arg constructor - for casts
1903 : *
1904 : * In theory we shouldn't need both this and the vararg (n-arg)
1905 : * constructor, but having a separate 1-arg function lets us define casts
1906 : * against it.
1907 : */
1908 160 : argtypes = buildoidvector(&rangeOid, 1);
1909 160 : myself = ProcedureCreate(name, /* name: same as multirange type */
1910 : namespace,
1911 : false, /* replace */
1912 : false, /* returns set */
1913 : multirangeOid, /* return type */
1914 : BOOTSTRAP_SUPERUSERID, /* proowner */
1915 : INTERNALlanguageId, /* language */
1916 : F_FMGR_INTERNAL_VALIDATOR,
1917 : "multirange_constructor1", /* prosrc */
1918 : NULL, /* probin */
1919 : NULL, /* prosqlbody */
1920 : PROKIND_FUNCTION,
1921 : false, /* security_definer */
1922 : false, /* leakproof */
1923 : true, /* isStrict */
1924 : PROVOLATILE_IMMUTABLE, /* volatility */
1925 : PROPARALLEL_SAFE, /* parallel safety */
1926 : argtypes, /* parameterTypes */
1927 : PointerGetDatum(NULL), /* allParameterTypes */
1928 : PointerGetDatum(NULL), /* parameterModes */
1929 : PointerGetDatum(NULL), /* parameterNames */
1930 : NIL, /* parameterDefaults */
1931 : PointerGetDatum(NULL), /* trftypes */
1932 : NIL, /* trfoids */
1933 : PointerGetDatum(NULL), /* proconfig */
1934 : InvalidOid, /* prosupport */
1935 : 1.0, /* procost */
1936 : 0.0); /* prorows */
1937 : /* ditto */
1938 160 : recordDependencyOn(&myself, &referenced, DEPENDENCY_INTERNAL);
1939 160 : pfree(argtypes);
1940 160 : *castFuncOid = myself.objectId;
1941 :
1942 : /* n-arg constructor - vararg */
1943 160 : argtypes = buildoidvector(&rangeArrayOid, 1);
1944 160 : allParamTypes = ObjectIdGetDatum(rangeArrayOid);
1945 160 : allParameterTypes = construct_array_builtin(&allParamTypes, 1, OIDOID);
1946 160 : paramModes = CharGetDatum(FUNC_PARAM_VARIADIC);
1947 160 : parameterModes = construct_array_builtin(¶mModes, 1, CHAROID);
1948 160 : myself = ProcedureCreate(name, /* name: same as multirange type */
1949 : namespace,
1950 : false, /* replace */
1951 : false, /* returns set */
1952 : multirangeOid, /* return type */
1953 : BOOTSTRAP_SUPERUSERID, /* proowner */
1954 : INTERNALlanguageId, /* language */
1955 : F_FMGR_INTERNAL_VALIDATOR,
1956 : "multirange_constructor2", /* prosrc */
1957 : NULL, /* probin */
1958 : NULL, /* prosqlbody */
1959 : PROKIND_FUNCTION,
1960 : false, /* security_definer */
1961 : false, /* leakproof */
1962 : true, /* isStrict */
1963 : PROVOLATILE_IMMUTABLE, /* volatility */
1964 : PROPARALLEL_SAFE, /* parallel safety */
1965 : argtypes, /* parameterTypes */
1966 : PointerGetDatum(allParameterTypes), /* allParameterTypes */
1967 : PointerGetDatum(parameterModes), /* parameterModes */
1968 : PointerGetDatum(NULL), /* parameterNames */
1969 : NIL, /* parameterDefaults */
1970 : PointerGetDatum(NULL), /* trftypes */
1971 : NIL, /* trfoids */
1972 : PointerGetDatum(NULL), /* proconfig */
1973 : InvalidOid, /* prosupport */
1974 : 1.0, /* procost */
1975 : 0.0); /* prorows */
1976 : /* ditto */
1977 160 : recordDependencyOn(&myself, &referenced, DEPENDENCY_INTERNAL);
1978 160 : pfree(argtypes);
1979 160 : pfree(allParameterTypes);
1980 160 : pfree(parameterModes);
1981 160 : }
1982 :
1983 : /*
1984 : * Find suitable I/O and other support functions for a type.
1985 : *
1986 : * typeOid is the type's OID (which will already exist, if only as a shell
1987 : * type).
1988 : */
1989 :
1990 : static Oid
1991 222 : findTypeInputFunction(List *procname, Oid typeOid)
1992 : {
1993 : Oid argList[3];
1994 : Oid procOid;
1995 : Oid procOid2;
1996 :
1997 : /*
1998 : * Input functions can take a single argument of type CSTRING, or three
1999 : * arguments (string, typioparam OID, typmod). Whine about ambiguity if
2000 : * both forms exist.
2001 : */
2002 222 : argList[0] = CSTRINGOID;
2003 222 : argList[1] = OIDOID;
2004 222 : argList[2] = INT4OID;
2005 :
2006 222 : procOid = LookupFuncName(procname, 1, argList, true);
2007 222 : procOid2 = LookupFuncName(procname, 3, argList, true);
2008 222 : if (OidIsValid(procOid))
2009 : {
2010 206 : if (OidIsValid(procOid2))
2011 0 : ereport(ERROR,
2012 : (errcode(ERRCODE_AMBIGUOUS_FUNCTION),
2013 : errmsg("type input function %s has multiple matches",
2014 : NameListToString(procname))));
2015 : }
2016 : else
2017 : {
2018 16 : procOid = procOid2;
2019 : /* If not found, reference the 1-argument signature in error msg */
2020 16 : if (!OidIsValid(procOid))
2021 0 : ereport(ERROR,
2022 : (errcode(ERRCODE_UNDEFINED_FUNCTION),
2023 : errmsg("function %s does not exist",
2024 : func_signature_string(procname, 1, NIL, argList))));
2025 : }
2026 :
2027 : /* Input functions must return the target type. */
2028 222 : if (get_func_rettype(procOid) != typeOid)
2029 6 : ereport(ERROR,
2030 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2031 : errmsg("type input function %s must return type %s",
2032 : NameListToString(procname), format_type_be(typeOid))));
2033 :
2034 : /*
2035 : * Print warnings if any of the type's I/O functions are marked volatile.
2036 : * There is a general assumption that I/O functions are stable or
2037 : * immutable; this allows us for example to mark record_in/record_out
2038 : * stable rather than volatile. Ideally we would throw errors not just
2039 : * warnings here; but since this check is new as of 9.5, and since the
2040 : * volatility marking might be just an error-of-omission and not a true
2041 : * indication of how the function behaves, we'll let it pass as a warning
2042 : * for now.
2043 : */
2044 216 : if (func_volatile(procOid) == PROVOLATILE_VOLATILE)
2045 0 : ereport(WARNING,
2046 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2047 : errmsg("type input function %s should not be volatile",
2048 : NameListToString(procname))));
2049 :
2050 216 : return procOid;
2051 : }
2052 :
2053 : static Oid
2054 216 : findTypeOutputFunction(List *procname, Oid typeOid)
2055 : {
2056 : Oid argList[1];
2057 : Oid procOid;
2058 :
2059 : /*
2060 : * Output functions always take a single argument of the type and return
2061 : * cstring.
2062 : */
2063 216 : argList[0] = typeOid;
2064 :
2065 216 : procOid = LookupFuncName(procname, 1, argList, true);
2066 216 : if (!OidIsValid(procOid))
2067 0 : ereport(ERROR,
2068 : (errcode(ERRCODE_UNDEFINED_FUNCTION),
2069 : errmsg("function %s does not exist",
2070 : func_signature_string(procname, 1, NIL, argList))));
2071 :
2072 216 : if (get_func_rettype(procOid) != CSTRINGOID)
2073 0 : ereport(ERROR,
2074 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2075 : errmsg("type output function %s must return type %s",
2076 : NameListToString(procname), "cstring")));
2077 :
2078 : /* Just a warning for now, per comments in findTypeInputFunction */
2079 216 : if (func_volatile(procOid) == PROVOLATILE_VOLATILE)
2080 0 : ereport(WARNING,
2081 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2082 : errmsg("type output function %s should not be volatile",
2083 : NameListToString(procname))));
2084 :
2085 216 : return procOid;
2086 : }
2087 :
2088 : static Oid
2089 46 : findTypeReceiveFunction(List *procname, Oid typeOid)
2090 : {
2091 : Oid argList[3];
2092 : Oid procOid;
2093 : Oid procOid2;
2094 :
2095 : /*
2096 : * Receive functions can take a single argument of type INTERNAL, or three
2097 : * arguments (internal, typioparam OID, typmod). Whine about ambiguity if
2098 : * both forms exist.
2099 : */
2100 46 : argList[0] = INTERNALOID;
2101 46 : argList[1] = OIDOID;
2102 46 : argList[2] = INT4OID;
2103 :
2104 46 : procOid = LookupFuncName(procname, 1, argList, true);
2105 46 : procOid2 = LookupFuncName(procname, 3, argList, true);
2106 46 : if (OidIsValid(procOid))
2107 : {
2108 36 : if (OidIsValid(procOid2))
2109 0 : ereport(ERROR,
2110 : (errcode(ERRCODE_AMBIGUOUS_FUNCTION),
2111 : errmsg("type receive function %s has multiple matches",
2112 : NameListToString(procname))));
2113 : }
2114 : else
2115 : {
2116 10 : procOid = procOid2;
2117 : /* If not found, reference the 1-argument signature in error msg */
2118 10 : if (!OidIsValid(procOid))
2119 0 : ereport(ERROR,
2120 : (errcode(ERRCODE_UNDEFINED_FUNCTION),
2121 : errmsg("function %s does not exist",
2122 : func_signature_string(procname, 1, NIL, argList))));
2123 : }
2124 :
2125 : /* Receive functions must return the target type. */
2126 46 : if (get_func_rettype(procOid) != typeOid)
2127 0 : ereport(ERROR,
2128 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2129 : errmsg("type receive function %s must return type %s",
2130 : NameListToString(procname), format_type_be(typeOid))));
2131 :
2132 : /* Just a warning for now, per comments in findTypeInputFunction */
2133 46 : if (func_volatile(procOid) == PROVOLATILE_VOLATILE)
2134 0 : ereport(WARNING,
2135 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2136 : errmsg("type receive function %s should not be volatile",
2137 : NameListToString(procname))));
2138 :
2139 46 : return procOid;
2140 : }
2141 :
2142 : static Oid
2143 46 : findTypeSendFunction(List *procname, Oid typeOid)
2144 : {
2145 : Oid argList[1];
2146 : Oid procOid;
2147 :
2148 : /*
2149 : * Send functions always take a single argument of the type and return
2150 : * bytea.
2151 : */
2152 46 : argList[0] = typeOid;
2153 :
2154 46 : procOid = LookupFuncName(procname, 1, argList, true);
2155 46 : if (!OidIsValid(procOid))
2156 0 : ereport(ERROR,
2157 : (errcode(ERRCODE_UNDEFINED_FUNCTION),
2158 : errmsg("function %s does not exist",
2159 : func_signature_string(procname, 1, NIL, argList))));
2160 :
2161 46 : if (get_func_rettype(procOid) != BYTEAOID)
2162 0 : ereport(ERROR,
2163 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2164 : errmsg("type send function %s must return type %s",
2165 : NameListToString(procname), "bytea")));
2166 :
2167 : /* Just a warning for now, per comments in findTypeInputFunction */
2168 46 : if (func_volatile(procOid) == PROVOLATILE_VOLATILE)
2169 0 : ereport(WARNING,
2170 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2171 : errmsg("type send function %s should not be volatile",
2172 : NameListToString(procname))));
2173 :
2174 46 : return procOid;
2175 : }
2176 :
2177 : static Oid
2178 14 : findTypeTypmodinFunction(List *procname)
2179 : {
2180 : Oid argList[1];
2181 : Oid procOid;
2182 :
2183 : /*
2184 : * typmodin functions always take one cstring[] argument and return int4.
2185 : */
2186 14 : argList[0] = CSTRINGARRAYOID;
2187 :
2188 14 : procOid = LookupFuncName(procname, 1, argList, true);
2189 14 : if (!OidIsValid(procOid))
2190 0 : ereport(ERROR,
2191 : (errcode(ERRCODE_UNDEFINED_FUNCTION),
2192 : errmsg("function %s does not exist",
2193 : func_signature_string(procname, 1, NIL, argList))));
2194 :
2195 14 : if (get_func_rettype(procOid) != INT4OID)
2196 0 : ereport(ERROR,
2197 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2198 : errmsg("typmod_in function %s must return type %s",
2199 : NameListToString(procname), "integer")));
2200 :
2201 : /* Just a warning for now, per comments in findTypeInputFunction */
2202 14 : if (func_volatile(procOid) == PROVOLATILE_VOLATILE)
2203 0 : ereport(WARNING,
2204 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2205 : errmsg("type modifier input function %s should not be volatile",
2206 : NameListToString(procname))));
2207 :
2208 14 : return procOid;
2209 : }
2210 :
2211 : static Oid
2212 14 : findTypeTypmodoutFunction(List *procname)
2213 : {
2214 : Oid argList[1];
2215 : Oid procOid;
2216 :
2217 : /*
2218 : * typmodout functions always take one int4 argument and return cstring.
2219 : */
2220 14 : argList[0] = INT4OID;
2221 :
2222 14 : procOid = LookupFuncName(procname, 1, argList, true);
2223 14 : if (!OidIsValid(procOid))
2224 0 : ereport(ERROR,
2225 : (errcode(ERRCODE_UNDEFINED_FUNCTION),
2226 : errmsg("function %s does not exist",
2227 : func_signature_string(procname, 1, NIL, argList))));
2228 :
2229 14 : if (get_func_rettype(procOid) != CSTRINGOID)
2230 0 : ereport(ERROR,
2231 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2232 : errmsg("typmod_out function %s must return type %s",
2233 : NameListToString(procname), "cstring")));
2234 :
2235 : /* Just a warning for now, per comments in findTypeInputFunction */
2236 14 : if (func_volatile(procOid) == PROVOLATILE_VOLATILE)
2237 0 : ereport(WARNING,
2238 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2239 : errmsg("type modifier output function %s should not be volatile",
2240 : NameListToString(procname))));
2241 :
2242 14 : return procOid;
2243 : }
2244 :
2245 : static Oid
2246 6 : findTypeAnalyzeFunction(List *procname, Oid typeOid)
2247 : {
2248 : Oid argList[1];
2249 : Oid procOid;
2250 :
2251 : /*
2252 : * Analyze functions always take one INTERNAL argument and return bool.
2253 : */
2254 6 : argList[0] = INTERNALOID;
2255 :
2256 6 : procOid = LookupFuncName(procname, 1, argList, true);
2257 6 : if (!OidIsValid(procOid))
2258 0 : ereport(ERROR,
2259 : (errcode(ERRCODE_UNDEFINED_FUNCTION),
2260 : errmsg("function %s does not exist",
2261 : func_signature_string(procname, 1, NIL, argList))));
2262 :
2263 6 : if (get_func_rettype(procOid) != BOOLOID)
2264 0 : ereport(ERROR,
2265 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2266 : errmsg("type analyze function %s must return type %s",
2267 : NameListToString(procname), "boolean")));
2268 :
2269 6 : return procOid;
2270 : }
2271 :
2272 : static Oid
2273 22 : findTypeSubscriptingFunction(List *procname, Oid typeOid)
2274 : {
2275 : Oid argList[1];
2276 : Oid procOid;
2277 :
2278 : /*
2279 : * Subscripting support functions always take one INTERNAL argument and
2280 : * return INTERNAL. (The argument is not used, but we must have it to
2281 : * maintain type safety.)
2282 : */
2283 22 : argList[0] = INTERNALOID;
2284 :
2285 22 : procOid = LookupFuncName(procname, 1, argList, true);
2286 22 : if (!OidIsValid(procOid))
2287 0 : ereport(ERROR,
2288 : (errcode(ERRCODE_UNDEFINED_FUNCTION),
2289 : errmsg("function %s does not exist",
2290 : func_signature_string(procname, 1, NIL, argList))));
2291 :
2292 22 : if (get_func_rettype(procOid) != INTERNALOID)
2293 0 : ereport(ERROR,
2294 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2295 : errmsg("type subscripting function %s must return type %s",
2296 : NameListToString(procname), "internal")));
2297 :
2298 : /*
2299 : * We disallow array_subscript_handler() from being selected explicitly,
2300 : * since that must only be applied to autogenerated array types.
2301 : */
2302 22 : if (procOid == F_ARRAY_SUBSCRIPT_HANDLER)
2303 0 : ereport(ERROR,
2304 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2305 : errmsg("user-defined types cannot use subscripting function %s",
2306 : NameListToString(procname))));
2307 :
2308 22 : return procOid;
2309 : }
2310 :
2311 : /*
2312 : * Find suitable support functions and opclasses for a range type.
2313 : */
2314 :
2315 : /*
2316 : * Find named btree opclass for subtype, or default btree opclass if
2317 : * opcname is NIL.
2318 : */
2319 : static Oid
2320 184 : findRangeSubOpclass(List *opcname, Oid subtype)
2321 : {
2322 : Oid opcid;
2323 : Oid opInputType;
2324 :
2325 184 : if (opcname != NIL)
2326 : {
2327 8 : opcid = get_opclass_oid(BTREE_AM_OID, opcname, false);
2328 :
2329 : /*
2330 : * Verify that the operator class accepts this datatype. Note we will
2331 : * accept binary compatibility.
2332 : */
2333 8 : opInputType = get_opclass_input_type(opcid);
2334 8 : if (!IsBinaryCoercible(subtype, opInputType))
2335 0 : ereport(ERROR,
2336 : (errcode(ERRCODE_DATATYPE_MISMATCH),
2337 : errmsg("operator class \"%s\" does not accept data type %s",
2338 : NameListToString(opcname),
2339 : format_type_be(subtype))));
2340 : }
2341 : else
2342 : {
2343 176 : opcid = GetDefaultOpClass(subtype, BTREE_AM_OID);
2344 176 : if (!OidIsValid(opcid))
2345 : {
2346 : /* We spell the error message identically to ResolveOpClass */
2347 0 : ereport(ERROR,
2348 : (errcode(ERRCODE_UNDEFINED_OBJECT),
2349 : errmsg("data type %s has no default operator class for access method \"%s\"",
2350 : format_type_be(subtype), "btree"),
2351 : errhint("You must specify an operator class for the range type or define a default operator class for the subtype.")));
2352 : }
2353 : }
2354 :
2355 184 : return opcid;
2356 : }
2357 :
2358 : static Oid
2359 0 : findRangeCanonicalFunction(List *procname, Oid typeOid)
2360 : {
2361 : Oid argList[1];
2362 : Oid procOid;
2363 : AclResult aclresult;
2364 :
2365 : /*
2366 : * Range canonical functions must take and return the range type, and must
2367 : * be immutable.
2368 : */
2369 0 : argList[0] = typeOid;
2370 :
2371 0 : procOid = LookupFuncName(procname, 1, argList, true);
2372 :
2373 0 : if (!OidIsValid(procOid))
2374 0 : ereport(ERROR,
2375 : (errcode(ERRCODE_UNDEFINED_FUNCTION),
2376 : errmsg("function %s does not exist",
2377 : func_signature_string(procname, 1, NIL, argList))));
2378 :
2379 0 : if (get_func_rettype(procOid) != typeOid)
2380 0 : ereport(ERROR,
2381 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2382 : errmsg("range canonical function %s must return range type",
2383 : func_signature_string(procname, 1, NIL, argList))));
2384 :
2385 0 : if (func_volatile(procOid) != PROVOLATILE_IMMUTABLE)
2386 0 : ereport(ERROR,
2387 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2388 : errmsg("range canonical function %s must be immutable",
2389 : func_signature_string(procname, 1, NIL, argList))));
2390 :
2391 : /* Also, range type's creator must have permission to call function */
2392 0 : aclresult = object_aclcheck(ProcedureRelationId, procOid, GetUserId(), ACL_EXECUTE);
2393 0 : if (aclresult != ACLCHECK_OK)
2394 0 : aclcheck_error(aclresult, OBJECT_FUNCTION, get_func_name(procOid));
2395 :
2396 0 : return procOid;
2397 : }
2398 :
2399 : static Oid
2400 14 : findRangeSubtypeDiffFunction(List *procname, Oid subtype)
2401 : {
2402 : Oid argList[2];
2403 : Oid procOid;
2404 : AclResult aclresult;
2405 :
2406 : /*
2407 : * Range subtype diff functions must take two arguments of the subtype,
2408 : * must return float8, and must be immutable.
2409 : */
2410 14 : argList[0] = subtype;
2411 14 : argList[1] = subtype;
2412 :
2413 14 : procOid = LookupFuncName(procname, 2, argList, true);
2414 :
2415 14 : if (!OidIsValid(procOid))
2416 6 : ereport(ERROR,
2417 : (errcode(ERRCODE_UNDEFINED_FUNCTION),
2418 : errmsg("function %s does not exist",
2419 : func_signature_string(procname, 2, NIL, argList))));
2420 :
2421 8 : if (get_func_rettype(procOid) != FLOAT8OID)
2422 0 : ereport(ERROR,
2423 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2424 : errmsg("range subtype diff function %s must return type %s",
2425 : func_signature_string(procname, 2, NIL, argList),
2426 : "double precision")));
2427 :
2428 8 : if (func_volatile(procOid) != PROVOLATILE_IMMUTABLE)
2429 0 : ereport(ERROR,
2430 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2431 : errmsg("range subtype diff function %s must be immutable",
2432 : func_signature_string(procname, 2, NIL, argList))));
2433 :
2434 : /* Also, range type's creator must have permission to call function */
2435 8 : aclresult = object_aclcheck(ProcedureRelationId, procOid, GetUserId(), ACL_EXECUTE);
2436 8 : if (aclresult != ACLCHECK_OK)
2437 0 : aclcheck_error(aclresult, OBJECT_FUNCTION, get_func_name(procOid));
2438 :
2439 8 : return procOid;
2440 : }
2441 :
2442 : /*
2443 : * AssignTypeArrayOid
2444 : *
2445 : * Pre-assign the type's array OID for use in pg_type.typarray
2446 : */
2447 : Oid
2448 71512 : AssignTypeArrayOid(void)
2449 : {
2450 : Oid type_array_oid;
2451 :
2452 : /* Use binary-upgrade override for pg_type.typarray? */
2453 71512 : if (IsBinaryUpgrade)
2454 : {
2455 1724 : if (!OidIsValid(binary_upgrade_next_array_pg_type_oid))
2456 0 : ereport(ERROR,
2457 : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
2458 : errmsg("pg_type array OID value not set when in binary upgrade mode")));
2459 :
2460 1724 : type_array_oid = binary_upgrade_next_array_pg_type_oid;
2461 1724 : binary_upgrade_next_array_pg_type_oid = InvalidOid;
2462 : }
2463 : else
2464 : {
2465 69788 : Relation pg_type = table_open(TypeRelationId, AccessShareLock);
2466 :
2467 69788 : type_array_oid = GetNewOidWithIndex(pg_type, TypeOidIndexId,
2468 : Anum_pg_type_oid);
2469 69788 : table_close(pg_type, AccessShareLock);
2470 : }
2471 :
2472 71512 : return type_array_oid;
2473 : }
2474 :
2475 : /*
2476 : * AssignTypeMultirangeOid
2477 : *
2478 : * Pre-assign the range type's multirange OID for use in pg_type.oid
2479 : */
2480 : Oid
2481 178 : AssignTypeMultirangeOid(void)
2482 : {
2483 : Oid type_multirange_oid;
2484 :
2485 : /* Use binary-upgrade override for pg_type.oid? */
2486 178 : if (IsBinaryUpgrade)
2487 : {
2488 12 : if (!OidIsValid(binary_upgrade_next_mrng_pg_type_oid))
2489 0 : ereport(ERROR,
2490 : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
2491 : errmsg("pg_type multirange OID value not set when in binary upgrade mode")));
2492 :
2493 12 : type_multirange_oid = binary_upgrade_next_mrng_pg_type_oid;
2494 12 : binary_upgrade_next_mrng_pg_type_oid = InvalidOid;
2495 : }
2496 : else
2497 : {
2498 166 : Relation pg_type = table_open(TypeRelationId, AccessShareLock);
2499 :
2500 166 : type_multirange_oid = GetNewOidWithIndex(pg_type, TypeOidIndexId,
2501 : Anum_pg_type_oid);
2502 166 : table_close(pg_type, AccessShareLock);
2503 : }
2504 :
2505 178 : return type_multirange_oid;
2506 : }
2507 :
2508 : /*
2509 : * AssignTypeMultirangeArrayOid
2510 : *
2511 : * Pre-assign the range type's multirange array OID for use in pg_type.typarray
2512 : */
2513 : Oid
2514 178 : AssignTypeMultirangeArrayOid(void)
2515 : {
2516 : Oid type_multirange_array_oid;
2517 :
2518 : /* Use binary-upgrade override for pg_type.oid? */
2519 178 : if (IsBinaryUpgrade)
2520 : {
2521 12 : if (!OidIsValid(binary_upgrade_next_mrng_array_pg_type_oid))
2522 0 : ereport(ERROR,
2523 : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
2524 : errmsg("pg_type multirange array OID value not set when in binary upgrade mode")));
2525 :
2526 12 : type_multirange_array_oid = binary_upgrade_next_mrng_array_pg_type_oid;
2527 12 : binary_upgrade_next_mrng_array_pg_type_oid = InvalidOid;
2528 : }
2529 : else
2530 : {
2531 166 : Relation pg_type = table_open(TypeRelationId, AccessShareLock);
2532 :
2533 166 : type_multirange_array_oid = GetNewOidWithIndex(pg_type, TypeOidIndexId,
2534 : Anum_pg_type_oid);
2535 166 : table_close(pg_type, AccessShareLock);
2536 : }
2537 :
2538 178 : return type_multirange_array_oid;
2539 : }
2540 :
2541 :
2542 : /*-------------------------------------------------------------------
2543 : * DefineCompositeType
2544 : *
2545 : * Create a Composite Type relation.
2546 : * `DefineRelation' does all the work, we just provide the correct
2547 : * arguments!
2548 : *
2549 : * If the relation already exists, then 'DefineRelation' will abort
2550 : * the xact...
2551 : *
2552 : * Return type is the new type's object address.
2553 : *-------------------------------------------------------------------
2554 : */
2555 : ObjectAddress
2556 4502 : DefineCompositeType(RangeVar *typevar, List *coldeflist)
2557 : {
2558 4502 : CreateStmt *createStmt = makeNode(CreateStmt);
2559 : Oid old_type_oid;
2560 : Oid typeNamespace;
2561 : ObjectAddress address;
2562 :
2563 : /*
2564 : * now set the parameters for keys/inheritance etc. All of these are
2565 : * uninteresting for composite types...
2566 : */
2567 4502 : createStmt->relation = typevar;
2568 4502 : createStmt->tableElts = coldeflist;
2569 4502 : createStmt->inhRelations = NIL;
2570 4502 : createStmt->constraints = NIL;
2571 4502 : createStmt->options = NIL;
2572 4502 : createStmt->oncommit = ONCOMMIT_NOOP;
2573 4502 : createStmt->tablespacename = NULL;
2574 4502 : createStmt->if_not_exists = false;
2575 :
2576 : /*
2577 : * Check for collision with an existing type name. If there is one and
2578 : * it's an autogenerated array, we can rename it out of the way. This
2579 : * check is here mainly to get a better error message about a "type"
2580 : * instead of below about a "relation".
2581 : */
2582 4502 : typeNamespace = RangeVarGetAndCheckCreationNamespace(createStmt->relation,
2583 : NoLock, NULL);
2584 4502 : RangeVarAdjustRelationPersistence(createStmt->relation, typeNamespace);
2585 : old_type_oid =
2586 4502 : GetSysCacheOid2(TYPENAMENSP, Anum_pg_type_oid,
2587 : CStringGetDatum(createStmt->relation->relname),
2588 : ObjectIdGetDatum(typeNamespace));
2589 4502 : if (OidIsValid(old_type_oid))
2590 : {
2591 0 : if (!moveArrayTypeName(old_type_oid, createStmt->relation->relname, typeNamespace))
2592 0 : ereport(ERROR,
2593 : (errcode(ERRCODE_DUPLICATE_OBJECT),
2594 : errmsg("type \"%s\" already exists", createStmt->relation->relname)));
2595 : }
2596 :
2597 : /*
2598 : * Finally create the relation. This also creates the type.
2599 : */
2600 4502 : DefineRelation(createStmt, RELKIND_COMPOSITE_TYPE, InvalidOid, &address,
2601 : NULL);
2602 :
2603 4490 : return address;
2604 : }
2605 :
2606 : /*
2607 : * AlterDomainDefault
2608 : *
2609 : * Routine implementing ALTER DOMAIN SET/DROP DEFAULT statements.
2610 : *
2611 : * Returns ObjectAddress of the modified domain.
2612 : */
2613 : ObjectAddress
2614 14 : AlterDomainDefault(List *names, Node *defaultRaw)
2615 : {
2616 : TypeName *typename;
2617 : Oid domainoid;
2618 : HeapTuple tup;
2619 : ParseState *pstate;
2620 : Relation rel;
2621 : char *defaultValue;
2622 14 : Node *defaultExpr = NULL; /* NULL if no default specified */
2623 14 : Datum new_record[Natts_pg_type] = {0};
2624 14 : bool new_record_nulls[Natts_pg_type] = {0};
2625 14 : bool new_record_repl[Natts_pg_type] = {0};
2626 : HeapTuple newtuple;
2627 : Form_pg_type typTup;
2628 : ObjectAddress address;
2629 :
2630 : /* Make a TypeName so we can use standard type lookup machinery */
2631 14 : typename = makeTypeNameFromNameList(names);
2632 14 : domainoid = typenameTypeId(NULL, typename);
2633 :
2634 : /* Look up the domain in the type table */
2635 14 : rel = table_open(TypeRelationId, RowExclusiveLock);
2636 :
2637 14 : tup = SearchSysCacheCopy1(TYPEOID, ObjectIdGetDatum(domainoid));
2638 14 : if (!HeapTupleIsValid(tup))
2639 0 : elog(ERROR, "cache lookup failed for type %u", domainoid);
2640 14 : typTup = (Form_pg_type) GETSTRUCT(tup);
2641 :
2642 : /* Check it's a domain and check user has permission for ALTER DOMAIN */
2643 14 : checkDomainOwner(tup);
2644 :
2645 : /* Setup new tuple */
2646 :
2647 : /* Store the new default into the tuple */
2648 14 : if (defaultRaw)
2649 : {
2650 : /* Create a dummy ParseState for transformExpr */
2651 8 : pstate = make_parsestate(NULL);
2652 :
2653 : /*
2654 : * Cook the colDef->raw_expr into an expression. Note: Name is
2655 : * strictly for error message
2656 : */
2657 8 : defaultExpr = cookDefault(pstate, defaultRaw,
2658 : typTup->typbasetype,
2659 : typTup->typtypmod,
2660 8 : NameStr(typTup->typname),
2661 : 0);
2662 :
2663 : /*
2664 : * If the expression is just a NULL constant, we treat the command
2665 : * like ALTER ... DROP DEFAULT. (But see note for same test in
2666 : * DefineDomain.)
2667 : */
2668 8 : if (defaultExpr == NULL ||
2669 8 : (IsA(defaultExpr, Const) && ((Const *) defaultExpr)->constisnull))
2670 : {
2671 : /* Default is NULL, drop it */
2672 0 : defaultExpr = NULL;
2673 0 : new_record_nulls[Anum_pg_type_typdefaultbin - 1] = true;
2674 0 : new_record_repl[Anum_pg_type_typdefaultbin - 1] = true;
2675 0 : new_record_nulls[Anum_pg_type_typdefault - 1] = true;
2676 0 : new_record_repl[Anum_pg_type_typdefault - 1] = true;
2677 : }
2678 : else
2679 : {
2680 : /*
2681 : * Expression must be stored as a nodeToString result, but we also
2682 : * require a valid textual representation (mainly to make life
2683 : * easier for pg_dump).
2684 : */
2685 8 : defaultValue = deparse_expression(defaultExpr,
2686 : NIL, false, false);
2687 :
2688 : /*
2689 : * Form an updated tuple with the new default and write it back.
2690 : */
2691 8 : new_record[Anum_pg_type_typdefaultbin - 1] = CStringGetTextDatum(nodeToString(defaultExpr));
2692 :
2693 8 : new_record_repl[Anum_pg_type_typdefaultbin - 1] = true;
2694 8 : new_record[Anum_pg_type_typdefault - 1] = CStringGetTextDatum(defaultValue);
2695 8 : new_record_repl[Anum_pg_type_typdefault - 1] = true;
2696 : }
2697 : }
2698 : else
2699 : {
2700 : /* ALTER ... DROP DEFAULT */
2701 6 : new_record_nulls[Anum_pg_type_typdefaultbin - 1] = true;
2702 6 : new_record_repl[Anum_pg_type_typdefaultbin - 1] = true;
2703 6 : new_record_nulls[Anum_pg_type_typdefault - 1] = true;
2704 6 : new_record_repl[Anum_pg_type_typdefault - 1] = true;
2705 : }
2706 :
2707 14 : newtuple = heap_modify_tuple(tup, RelationGetDescr(rel),
2708 : new_record, new_record_nulls,
2709 : new_record_repl);
2710 :
2711 14 : CatalogTupleUpdate(rel, &tup->t_self, newtuple);
2712 :
2713 : /* Rebuild dependencies */
2714 14 : GenerateTypeDependencies(newtuple,
2715 : rel,
2716 : defaultExpr,
2717 : NULL, /* don't have typacl handy */
2718 : 0, /* relation kind is n/a */
2719 : false, /* a domain isn't an implicit array */
2720 : false, /* nor is it any kind of dependent type */
2721 : false, /* don't touch extension membership */
2722 : true); /* We do need to rebuild dependencies */
2723 :
2724 14 : InvokeObjectPostAlterHook(TypeRelationId, domainoid, 0);
2725 :
2726 14 : ObjectAddressSet(address, TypeRelationId, domainoid);
2727 :
2728 : /* Clean up */
2729 14 : table_close(rel, RowExclusiveLock);
2730 14 : heap_freetuple(newtuple);
2731 :
2732 14 : return address;
2733 : }
2734 :
2735 : /*
2736 : * AlterDomainNotNull
2737 : *
2738 : * Routine implementing ALTER DOMAIN SET/DROP NOT NULL statements.
2739 : *
2740 : * Returns ObjectAddress of the modified domain.
2741 : */
2742 : ObjectAddress
2743 36 : AlterDomainNotNull(List *names, bool notNull)
2744 : {
2745 : TypeName *typename;
2746 : Oid domainoid;
2747 : Relation typrel;
2748 : HeapTuple tup;
2749 : Form_pg_type typTup;
2750 36 : ObjectAddress address = InvalidObjectAddress;
2751 :
2752 : /* Make a TypeName so we can use standard type lookup machinery */
2753 36 : typename = makeTypeNameFromNameList(names);
2754 36 : domainoid = typenameTypeId(NULL, typename);
2755 :
2756 : /* Look up the domain in the type table */
2757 36 : typrel = table_open(TypeRelationId, RowExclusiveLock);
2758 :
2759 36 : tup = SearchSysCacheCopy1(TYPEOID, ObjectIdGetDatum(domainoid));
2760 36 : if (!HeapTupleIsValid(tup))
2761 0 : elog(ERROR, "cache lookup failed for type %u", domainoid);
2762 36 : typTup = (Form_pg_type) GETSTRUCT(tup);
2763 :
2764 : /* Check it's a domain and check user has permission for ALTER DOMAIN */
2765 36 : checkDomainOwner(tup);
2766 :
2767 : /* Is the domain already set to the desired constraint? */
2768 36 : if (typTup->typnotnull == notNull)
2769 : {
2770 0 : table_close(typrel, RowExclusiveLock);
2771 0 : return address;
2772 : }
2773 :
2774 36 : if (notNull)
2775 : {
2776 : Constraint *constr;
2777 :
2778 24 : constr = makeNode(Constraint);
2779 24 : constr->contype = CONSTR_NOTNULL;
2780 24 : constr->initially_valid = true;
2781 24 : constr->location = -1;
2782 :
2783 24 : domainAddNotNullConstraint(domainoid, typTup->typnamespace,
2784 : typTup->typbasetype, typTup->typtypmod,
2785 24 : constr, NameStr(typTup->typname), NULL);
2786 :
2787 24 : validateDomainNotNullConstraint(domainoid);
2788 : }
2789 : else
2790 : {
2791 : HeapTuple conTup;
2792 : ObjectAddress conobj;
2793 :
2794 12 : conTup = findDomainNotNullConstraint(domainoid);
2795 12 : if (conTup == NULL)
2796 0 : elog(ERROR, "could not find not-null constraint on domain \"%s\"", NameStr(typTup->typname));
2797 :
2798 12 : ObjectAddressSet(conobj, ConstraintRelationId, ((Form_pg_constraint) GETSTRUCT(conTup))->oid);
2799 12 : performDeletion(&conobj, DROP_RESTRICT, 0);
2800 : }
2801 :
2802 : /*
2803 : * Okay to update pg_type row. We can scribble on typTup because it's a
2804 : * copy.
2805 : */
2806 24 : typTup->typnotnull = notNull;
2807 :
2808 24 : CatalogTupleUpdate(typrel, &tup->t_self, tup);
2809 :
2810 24 : InvokeObjectPostAlterHook(TypeRelationId, domainoid, 0);
2811 :
2812 24 : ObjectAddressSet(address, TypeRelationId, domainoid);
2813 :
2814 : /* Clean up */
2815 24 : heap_freetuple(tup);
2816 24 : table_close(typrel, RowExclusiveLock);
2817 :
2818 24 : return address;
2819 : }
2820 :
2821 : /*
2822 : * AlterDomainDropConstraint
2823 : *
2824 : * Implements the ALTER DOMAIN DROP CONSTRAINT statement
2825 : *
2826 : * Returns ObjectAddress of the modified domain.
2827 : */
2828 : ObjectAddress
2829 60 : AlterDomainDropConstraint(List *names, const char *constrName,
2830 : DropBehavior behavior, bool missing_ok)
2831 : {
2832 : TypeName *typename;
2833 : Oid domainoid;
2834 : HeapTuple tup;
2835 : Relation rel;
2836 : Relation conrel;
2837 : SysScanDesc conscan;
2838 : ScanKeyData skey[3];
2839 : HeapTuple contup;
2840 60 : bool found = false;
2841 : ObjectAddress address;
2842 :
2843 : /* Make a TypeName so we can use standard type lookup machinery */
2844 60 : typename = makeTypeNameFromNameList(names);
2845 60 : domainoid = typenameTypeId(NULL, typename);
2846 :
2847 : /* Look up the domain in the type table */
2848 60 : rel = table_open(TypeRelationId, RowExclusiveLock);
2849 :
2850 60 : tup = SearchSysCacheCopy1(TYPEOID, ObjectIdGetDatum(domainoid));
2851 60 : if (!HeapTupleIsValid(tup))
2852 0 : elog(ERROR, "cache lookup failed for type %u", domainoid);
2853 :
2854 : /* Check it's a domain and check user has permission for ALTER DOMAIN */
2855 60 : checkDomainOwner(tup);
2856 :
2857 : /* Grab an appropriate lock on the pg_constraint relation */
2858 60 : conrel = table_open(ConstraintRelationId, RowExclusiveLock);
2859 :
2860 : /* Find and remove the target constraint */
2861 60 : ScanKeyInit(&skey[0],
2862 : Anum_pg_constraint_conrelid,
2863 : BTEqualStrategyNumber, F_OIDEQ,
2864 : ObjectIdGetDatum(InvalidOid));
2865 60 : ScanKeyInit(&skey[1],
2866 : Anum_pg_constraint_contypid,
2867 : BTEqualStrategyNumber, F_OIDEQ,
2868 : ObjectIdGetDatum(domainoid));
2869 60 : ScanKeyInit(&skey[2],
2870 : Anum_pg_constraint_conname,
2871 : BTEqualStrategyNumber, F_NAMEEQ,
2872 : CStringGetDatum(constrName));
2873 :
2874 60 : conscan = systable_beginscan(conrel, ConstraintRelidTypidNameIndexId, true,
2875 : NULL, 3, skey);
2876 :
2877 : /* There can be at most one matching row */
2878 60 : if ((contup = systable_getnext(conscan)) != NULL)
2879 : {
2880 48 : Form_pg_constraint construct = (Form_pg_constraint) GETSTRUCT(contup);
2881 : ObjectAddress conobj;
2882 :
2883 48 : if (construct->contype == CONSTRAINT_NOTNULL)
2884 : {
2885 12 : ((Form_pg_type) GETSTRUCT(tup))->typnotnull = false;
2886 12 : CatalogTupleUpdate(rel, &tup->t_self, tup);
2887 : }
2888 :
2889 48 : conobj.classId = ConstraintRelationId;
2890 48 : conobj.objectId = construct->oid;
2891 48 : conobj.objectSubId = 0;
2892 :
2893 48 : performDeletion(&conobj, behavior, 0);
2894 48 : found = true;
2895 : }
2896 :
2897 : /* Clean up after the scan */
2898 60 : systable_endscan(conscan);
2899 60 : table_close(conrel, RowExclusiveLock);
2900 :
2901 60 : if (!found)
2902 : {
2903 12 : if (!missing_ok)
2904 6 : ereport(ERROR,
2905 : (errcode(ERRCODE_UNDEFINED_OBJECT),
2906 : errmsg("constraint \"%s\" of domain \"%s\" does not exist",
2907 : constrName, TypeNameToString(typename))));
2908 : else
2909 6 : ereport(NOTICE,
2910 : (errmsg("constraint \"%s\" of domain \"%s\" does not exist, skipping",
2911 : constrName, TypeNameToString(typename))));
2912 : }
2913 :
2914 : /*
2915 : * We must send out an sinval message for the domain, to ensure that any
2916 : * dependent plans get rebuilt. Since this command doesn't change the
2917 : * domain's pg_type row, that won't happen automatically; do it manually.
2918 : */
2919 54 : CacheInvalidateHeapTuple(rel, tup, NULL);
2920 :
2921 54 : ObjectAddressSet(address, TypeRelationId, domainoid);
2922 :
2923 : /* Clean up */
2924 54 : table_close(rel, RowExclusiveLock);
2925 :
2926 54 : return address;
2927 : }
2928 :
2929 : /*
2930 : * AlterDomainAddConstraint
2931 : *
2932 : * Implements the ALTER DOMAIN .. ADD CONSTRAINT statement.
2933 : */
2934 : ObjectAddress
2935 182 : AlterDomainAddConstraint(List *names, Node *newConstraint,
2936 : ObjectAddress *constrAddr)
2937 : {
2938 : TypeName *typename;
2939 : Oid domainoid;
2940 : Relation typrel;
2941 : HeapTuple tup;
2942 : Form_pg_type typTup;
2943 : Constraint *constr;
2944 : char *ccbin;
2945 182 : ObjectAddress address = InvalidObjectAddress;
2946 :
2947 : /* Make a TypeName so we can use standard type lookup machinery */
2948 182 : typename = makeTypeNameFromNameList(names);
2949 182 : domainoid = typenameTypeId(NULL, typename);
2950 :
2951 : /* Look up the domain in the type table */
2952 182 : typrel = table_open(TypeRelationId, RowExclusiveLock);
2953 :
2954 182 : tup = SearchSysCacheCopy1(TYPEOID, ObjectIdGetDatum(domainoid));
2955 182 : if (!HeapTupleIsValid(tup))
2956 0 : elog(ERROR, "cache lookup failed for type %u", domainoid);
2957 182 : typTup = (Form_pg_type) GETSTRUCT(tup);
2958 :
2959 : /* Check it's a domain and check user has permission for ALTER DOMAIN */
2960 182 : checkDomainOwner(tup);
2961 :
2962 182 : if (!IsA(newConstraint, Constraint))
2963 0 : elog(ERROR, "unrecognized node type: %d",
2964 : (int) nodeTag(newConstraint));
2965 :
2966 182 : constr = (Constraint *) newConstraint;
2967 :
2968 : /* enforced by parser */
2969 : Assert(constr->contype == CONSTR_CHECK || constr->contype == CONSTR_NOTNULL);
2970 :
2971 182 : if (constr->contype == CONSTR_CHECK)
2972 : {
2973 : /*
2974 : * First, process the constraint expression and add an entry to
2975 : * pg_constraint.
2976 : */
2977 :
2978 152 : ccbin = domainAddCheckConstraint(domainoid, typTup->typnamespace,
2979 : typTup->typbasetype, typTup->typtypmod,
2980 152 : constr, NameStr(typTup->typname), constrAddr);
2981 :
2982 :
2983 : /*
2984 : * If requested to validate the constraint, test all values stored in
2985 : * the attributes based on the domain the constraint is being added
2986 : * to.
2987 : */
2988 146 : if (!constr->skip_validation)
2989 132 : validateDomainCheckConstraint(domainoid, ccbin, ShareLock);
2990 :
2991 : /*
2992 : * We must send out an sinval message for the domain, to ensure that
2993 : * any dependent plans get rebuilt. Since this command doesn't change
2994 : * the domain's pg_type row, that won't happen automatically; do it
2995 : * manually.
2996 : */
2997 86 : CacheInvalidateHeapTuple(typrel, tup, NULL);
2998 : }
2999 30 : else if (constr->contype == CONSTR_NOTNULL)
3000 : {
3001 : /* Is the domain already set NOT NULL? */
3002 30 : if (typTup->typnotnull)
3003 : {
3004 6 : table_close(typrel, RowExclusiveLock);
3005 6 : return address;
3006 : }
3007 24 : domainAddNotNullConstraint(domainoid, typTup->typnamespace,
3008 : typTup->typbasetype, typTup->typtypmod,
3009 24 : constr, NameStr(typTup->typname), constrAddr);
3010 :
3011 24 : if (!constr->skip_validation)
3012 24 : validateDomainNotNullConstraint(domainoid);
3013 :
3014 12 : typTup->typnotnull = true;
3015 12 : CatalogTupleUpdate(typrel, &tup->t_self, tup);
3016 : }
3017 :
3018 98 : ObjectAddressSet(address, TypeRelationId, domainoid);
3019 :
3020 : /* Clean up */
3021 98 : table_close(typrel, RowExclusiveLock);
3022 :
3023 98 : return address;
3024 : }
3025 :
3026 : /*
3027 : * AlterDomainValidateConstraint
3028 : *
3029 : * Implements the ALTER DOMAIN .. VALIDATE CONSTRAINT statement.
3030 : */
3031 : ObjectAddress
3032 12 : AlterDomainValidateConstraint(List *names, const char *constrName)
3033 : {
3034 : TypeName *typename;
3035 : Oid domainoid;
3036 : Relation typrel;
3037 : Relation conrel;
3038 : HeapTuple tup;
3039 : Form_pg_constraint con;
3040 : Form_pg_constraint copy_con;
3041 : char *conbin;
3042 : SysScanDesc scan;
3043 : Datum val;
3044 : HeapTuple tuple;
3045 : HeapTuple copyTuple;
3046 : ScanKeyData skey[3];
3047 : ObjectAddress address;
3048 :
3049 : /* Make a TypeName so we can use standard type lookup machinery */
3050 12 : typename = makeTypeNameFromNameList(names);
3051 12 : domainoid = typenameTypeId(NULL, typename);
3052 :
3053 : /* Look up the domain in the type table */
3054 12 : typrel = table_open(TypeRelationId, AccessShareLock);
3055 :
3056 12 : tup = SearchSysCache1(TYPEOID, ObjectIdGetDatum(domainoid));
3057 12 : if (!HeapTupleIsValid(tup))
3058 0 : elog(ERROR, "cache lookup failed for type %u", domainoid);
3059 :
3060 : /* Check it's a domain and check user has permission for ALTER DOMAIN */
3061 12 : checkDomainOwner(tup);
3062 :
3063 : /*
3064 : * Find and check the target constraint
3065 : */
3066 12 : conrel = table_open(ConstraintRelationId, RowExclusiveLock);
3067 :
3068 12 : ScanKeyInit(&skey[0],
3069 : Anum_pg_constraint_conrelid,
3070 : BTEqualStrategyNumber, F_OIDEQ,
3071 : ObjectIdGetDatum(InvalidOid));
3072 12 : ScanKeyInit(&skey[1],
3073 : Anum_pg_constraint_contypid,
3074 : BTEqualStrategyNumber, F_OIDEQ,
3075 : ObjectIdGetDatum(domainoid));
3076 12 : ScanKeyInit(&skey[2],
3077 : Anum_pg_constraint_conname,
3078 : BTEqualStrategyNumber, F_NAMEEQ,
3079 : CStringGetDatum(constrName));
3080 :
3081 12 : scan = systable_beginscan(conrel, ConstraintRelidTypidNameIndexId, true,
3082 : NULL, 3, skey);
3083 :
3084 : /* There can be at most one matching row */
3085 12 : if (!HeapTupleIsValid(tuple = systable_getnext(scan)))
3086 0 : ereport(ERROR,
3087 : (errcode(ERRCODE_UNDEFINED_OBJECT),
3088 : errmsg("constraint \"%s\" of domain \"%s\" does not exist",
3089 : constrName, TypeNameToString(typename))));
3090 :
3091 12 : con = (Form_pg_constraint) GETSTRUCT(tuple);
3092 12 : if (con->contype != CONSTRAINT_CHECK)
3093 0 : ereport(ERROR,
3094 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
3095 : errmsg("constraint \"%s\" of domain \"%s\" is not a check constraint",
3096 : constrName, TypeNameToString(typename))));
3097 :
3098 12 : val = SysCacheGetAttrNotNull(CONSTROID, tuple, Anum_pg_constraint_conbin);
3099 12 : conbin = TextDatumGetCString(val);
3100 :
3101 : /*
3102 : * Locking related relations with ShareUpdateExclusiveLock is ok because
3103 : * not-yet-valid constraints are still enforced against concurrent inserts
3104 : * or updates.
3105 : */
3106 12 : validateDomainCheckConstraint(domainoid, conbin, ShareUpdateExclusiveLock);
3107 :
3108 : /*
3109 : * Now update the catalog, while we have the door open.
3110 : */
3111 6 : copyTuple = heap_copytuple(tuple);
3112 6 : copy_con = (Form_pg_constraint) GETSTRUCT(copyTuple);
3113 6 : copy_con->convalidated = true;
3114 6 : CatalogTupleUpdate(conrel, ©Tuple->t_self, copyTuple);
3115 :
3116 6 : InvokeObjectPostAlterHook(ConstraintRelationId, con->oid, 0);
3117 :
3118 6 : ObjectAddressSet(address, TypeRelationId, domainoid);
3119 :
3120 6 : heap_freetuple(copyTuple);
3121 :
3122 6 : systable_endscan(scan);
3123 :
3124 6 : table_close(typrel, AccessShareLock);
3125 6 : table_close(conrel, RowExclusiveLock);
3126 :
3127 6 : ReleaseSysCache(tup);
3128 :
3129 6 : return address;
3130 : }
3131 :
3132 : /*
3133 : * Verify that all columns currently using the domain are not null.
3134 : */
3135 : static void
3136 48 : validateDomainNotNullConstraint(Oid domainoid)
3137 : {
3138 : List *rels;
3139 : ListCell *rt;
3140 :
3141 : /* Fetch relation list with attributes based on this domain */
3142 : /* ShareLock is sufficient to prevent concurrent data changes */
3143 :
3144 48 : rels = get_rels_with_domain(domainoid, ShareLock);
3145 :
3146 66 : foreach(rt, rels)
3147 : {
3148 42 : RelToCheck *rtc = (RelToCheck *) lfirst(rt);
3149 42 : Relation testrel = rtc->rel;
3150 42 : TupleDesc tupdesc = RelationGetDescr(testrel);
3151 : TupleTableSlot *slot;
3152 : TableScanDesc scan;
3153 : Snapshot snapshot;
3154 :
3155 : /* Scan all tuples in this relation */
3156 42 : snapshot = RegisterSnapshot(GetLatestSnapshot());
3157 42 : scan = table_beginscan(testrel, snapshot, 0, NULL);
3158 42 : slot = table_slot_create(testrel, NULL);
3159 60 : while (table_scan_getnextslot(scan, ForwardScanDirection, slot))
3160 : {
3161 : int i;
3162 :
3163 : /* Test attributes that are of the domain */
3164 90 : for (i = 0; i < rtc->natts; i++)
3165 : {
3166 72 : int attnum = rtc->atts[i];
3167 72 : Form_pg_attribute attr = TupleDescAttr(tupdesc, attnum - 1);
3168 :
3169 72 : if (slot_attisnull(slot, attnum))
3170 : {
3171 : /*
3172 : * In principle the auxiliary information for this error
3173 : * should be errdatatype(), but errtablecol() seems
3174 : * considerably more useful in practice. Since this code
3175 : * only executes in an ALTER DOMAIN command, the client
3176 : * should already know which domain is in question.
3177 : */
3178 24 : ereport(ERROR,
3179 : (errcode(ERRCODE_NOT_NULL_VIOLATION),
3180 : errmsg("column \"%s\" of table \"%s\" contains null values",
3181 : NameStr(attr->attname),
3182 : RelationGetRelationName(testrel)),
3183 : errtablecol(testrel, attnum)));
3184 : }
3185 : }
3186 : }
3187 18 : ExecDropSingleTupleTableSlot(slot);
3188 18 : table_endscan(scan);
3189 18 : UnregisterSnapshot(snapshot);
3190 :
3191 : /* Close each rel after processing, but keep lock */
3192 18 : table_close(testrel, NoLock);
3193 : }
3194 24 : }
3195 :
3196 : /*
3197 : * Verify that all columns currently using the domain satisfy the given check
3198 : * constraint expression.
3199 : *
3200 : * It is used to validate existing constraints and to add newly created check
3201 : * constraints to a domain.
3202 : *
3203 : * The lockmode is used for relations using the domain. It should be
3204 : * ShareLock when adding a new constraint to domain. It can be
3205 : * ShareUpdateExclusiveLock when validating an existing constraint.
3206 : */
3207 : static void
3208 144 : validateDomainCheckConstraint(Oid domainoid, const char *ccbin, LOCKMODE lockmode)
3209 : {
3210 144 : Expr *expr = (Expr *) stringToNode(ccbin);
3211 : List *rels;
3212 : ListCell *rt;
3213 : EState *estate;
3214 : ExprContext *econtext;
3215 : ExprState *exprstate;
3216 :
3217 : /* Need an EState to run ExecEvalExpr */
3218 144 : estate = CreateExecutorState();
3219 144 : econtext = GetPerTupleExprContext(estate);
3220 :
3221 : /* build execution state for expr */
3222 144 : exprstate = ExecPrepareExpr(expr, estate);
3223 :
3224 : /* Fetch relation list with attributes based on this domain */
3225 144 : rels = get_rels_with_domain(domainoid, lockmode);
3226 :
3227 150 : foreach(rt, rels)
3228 : {
3229 72 : RelToCheck *rtc = (RelToCheck *) lfirst(rt);
3230 72 : Relation testrel = rtc->rel;
3231 72 : TupleDesc tupdesc = RelationGetDescr(testrel);
3232 : TupleTableSlot *slot;
3233 : TableScanDesc scan;
3234 : Snapshot snapshot;
3235 :
3236 : /* Scan all tuples in this relation */
3237 72 : snapshot = RegisterSnapshot(GetLatestSnapshot());
3238 72 : scan = table_beginscan(testrel, snapshot, 0, NULL);
3239 72 : slot = table_slot_create(testrel, NULL);
3240 168 : while (table_scan_getnextslot(scan, ForwardScanDirection, slot))
3241 : {
3242 : int i;
3243 :
3244 : /* Test attributes that are of the domain */
3245 228 : for (i = 0; i < rtc->natts; i++)
3246 : {
3247 132 : int attnum = rtc->atts[i];
3248 : Datum d;
3249 : bool isNull;
3250 : Datum conResult;
3251 :
3252 132 : d = slot_getattr(slot, attnum, &isNull);
3253 :
3254 132 : econtext->domainValue_datum = d;
3255 132 : econtext->domainValue_isNull = isNull;
3256 :
3257 132 : conResult = ExecEvalExprSwitchContext(exprstate,
3258 : econtext,
3259 : &isNull);
3260 :
3261 132 : if (!isNull && !DatumGetBool(conResult))
3262 : {
3263 36 : Form_pg_attribute attr = TupleDescAttr(tupdesc, attnum - 1);
3264 :
3265 : /*
3266 : * In principle the auxiliary information for this error
3267 : * should be errdomainconstraint(), but errtablecol()
3268 : * seems considerably more useful in practice. Since this
3269 : * code only executes in an ALTER DOMAIN command, the
3270 : * client should already know which domain is in question,
3271 : * and which constraint too.
3272 : */
3273 36 : ereport(ERROR,
3274 : (errcode(ERRCODE_CHECK_VIOLATION),
3275 : errmsg("column \"%s\" of table \"%s\" contains values that violate the new constraint",
3276 : NameStr(attr->attname),
3277 : RelationGetRelationName(testrel)),
3278 : errtablecol(testrel, attnum)));
3279 : }
3280 : }
3281 :
3282 96 : ResetExprContext(econtext);
3283 : }
3284 36 : ExecDropSingleTupleTableSlot(slot);
3285 36 : table_endscan(scan);
3286 36 : UnregisterSnapshot(snapshot);
3287 :
3288 : /* Hold relation lock till commit (XXX bad for concurrency) */
3289 36 : table_close(testrel, NoLock);
3290 : }
3291 :
3292 78 : FreeExecutorState(estate);
3293 78 : }
3294 :
3295 : /*
3296 : * get_rels_with_domain
3297 : *
3298 : * Fetch all relations / attributes which are using the domain
3299 : *
3300 : * The result is a list of RelToCheck structs, one for each distinct
3301 : * relation, each containing one or more attribute numbers that are of
3302 : * the domain type. We have opened each rel and acquired the specified lock
3303 : * type on it.
3304 : *
3305 : * We support nested domains by including attributes that are of derived
3306 : * domain types. Current callers do not need to distinguish between attributes
3307 : * that are of exactly the given domain and those that are of derived domains.
3308 : *
3309 : * XXX this is completely broken because there is no way to lock the domain
3310 : * to prevent columns from being added or dropped while our command runs.
3311 : * We can partially protect against column drops by locking relations as we
3312 : * come across them, but there is still a race condition (the window between
3313 : * seeing a pg_depend entry and acquiring lock on the relation it references).
3314 : * Also, holding locks on all these relations simultaneously creates a non-
3315 : * trivial risk of deadlock. We can minimize but not eliminate the deadlock
3316 : * risk by using the weakest suitable lock (ShareLock for most callers).
3317 : *
3318 : * XXX the API for this is not sufficient to support checking domain values
3319 : * that are inside container types, such as composite types, arrays, or
3320 : * ranges. Currently we just error out if a container type containing the
3321 : * target domain is stored anywhere.
3322 : *
3323 : * Generally used for retrieving a list of tests when adding
3324 : * new constraints to a domain.
3325 : */
3326 : static List *
3327 204 : get_rels_with_domain(Oid domainOid, LOCKMODE lockmode)
3328 : {
3329 204 : List *result = NIL;
3330 204 : char *domainTypeName = format_type_be(domainOid);
3331 : Relation depRel;
3332 : ScanKeyData key[2];
3333 : SysScanDesc depScan;
3334 : HeapTuple depTup;
3335 :
3336 : Assert(lockmode != NoLock);
3337 :
3338 : /* since this function recurses, it could be driven to stack overflow */
3339 204 : check_stack_depth();
3340 :
3341 : /*
3342 : * We scan pg_depend to find those things that depend on the domain. (We
3343 : * assume we can ignore refobjsubid for a domain.)
3344 : */
3345 204 : depRel = table_open(DependRelationId, AccessShareLock);
3346 :
3347 204 : ScanKeyInit(&key[0],
3348 : Anum_pg_depend_refclassid,
3349 : BTEqualStrategyNumber, F_OIDEQ,
3350 : ObjectIdGetDatum(TypeRelationId));
3351 204 : ScanKeyInit(&key[1],
3352 : Anum_pg_depend_refobjid,
3353 : BTEqualStrategyNumber, F_OIDEQ,
3354 : ObjectIdGetDatum(domainOid));
3355 :
3356 204 : depScan = systable_beginscan(depRel, DependReferenceIndexId, true,
3357 : NULL, 2, key);
3358 :
3359 692 : while (HeapTupleIsValid(depTup = systable_getnext(depScan)))
3360 : {
3361 518 : Form_pg_depend pg_depend = (Form_pg_depend) GETSTRUCT(depTup);
3362 518 : RelToCheck *rtc = NULL;
3363 : ListCell *rellist;
3364 : Form_pg_attribute pg_att;
3365 : int ptr;
3366 :
3367 : /* Check for directly dependent types */
3368 518 : if (pg_depend->classid == TypeRelationId)
3369 : {
3370 222 : if (get_typtype(pg_depend->objid) == TYPTYPE_DOMAIN)
3371 : {
3372 : /*
3373 : * This is a sub-domain, so recursively add dependent columns
3374 : * to the output list. This is a bit inefficient since we may
3375 : * fail to combine RelToCheck entries when attributes of the
3376 : * same rel have different derived domain types, but it's
3377 : * probably not worth improving.
3378 : */
3379 12 : result = list_concat(result,
3380 12 : get_rels_with_domain(pg_depend->objid,
3381 : lockmode));
3382 : }
3383 : else
3384 : {
3385 : /*
3386 : * Otherwise, it is some container type using the domain, so
3387 : * fail if there are any columns of this type.
3388 : */
3389 210 : find_composite_type_dependencies(pg_depend->objid,
3390 : NULL,
3391 : domainTypeName);
3392 : }
3393 216 : continue;
3394 : }
3395 :
3396 : /* Else, ignore dependees that aren't user columns of relations */
3397 : /* (we assume system columns are never of domain types) */
3398 296 : if (pg_depend->classid != RelationRelationId ||
3399 216 : pg_depend->objsubid <= 0)
3400 80 : continue;
3401 :
3402 : /* See if we already have an entry for this relation */
3403 216 : foreach(rellist, result)
3404 : {
3405 42 : RelToCheck *rt = (RelToCheck *) lfirst(rellist);
3406 :
3407 42 : if (RelationGetRelid(rt->rel) == pg_depend->objid)
3408 : {
3409 42 : rtc = rt;
3410 42 : break;
3411 : }
3412 : }
3413 :
3414 216 : if (rtc == NULL)
3415 : {
3416 : /* First attribute found for this relation */
3417 : Relation rel;
3418 :
3419 : /* Acquire requested lock on relation */
3420 174 : rel = relation_open(pg_depend->objid, lockmode);
3421 :
3422 : /*
3423 : * Check to see if rowtype is stored anyplace as a composite-type
3424 : * column; if so we have to fail, for now anyway.
3425 : */
3426 174 : if (OidIsValid(rel->rd_rel->reltype))
3427 174 : find_composite_type_dependencies(rel->rd_rel->reltype,
3428 : NULL,
3429 : domainTypeName);
3430 :
3431 : /*
3432 : * Otherwise, we can ignore relations except those with both
3433 : * storage and user-chosen column types.
3434 : *
3435 : * XXX If an index-only scan could satisfy "col::some_domain" from
3436 : * a suitable expression index, this should also check expression
3437 : * index columns.
3438 : */
3439 150 : if (rel->rd_rel->relkind != RELKIND_RELATION &&
3440 36 : rel->rd_rel->relkind != RELKIND_MATVIEW)
3441 : {
3442 36 : relation_close(rel, lockmode);
3443 36 : continue;
3444 : }
3445 :
3446 : /* Build the RelToCheck entry with enough space for all atts */
3447 114 : rtc = (RelToCheck *) palloc(sizeof(RelToCheck));
3448 114 : rtc->rel = rel;
3449 114 : rtc->natts = 0;
3450 114 : rtc->atts = (int *) palloc(sizeof(int) * RelationGetNumberOfAttributes(rel));
3451 114 : result = lappend(result, rtc);
3452 : }
3453 :
3454 : /*
3455 : * Confirm column has not been dropped, and is of the expected type.
3456 : * This defends against an ALTER DROP COLUMN occurring just before we
3457 : * acquired lock ... but if the whole table were dropped, we'd still
3458 : * have a problem.
3459 : */
3460 156 : if (pg_depend->objsubid > RelationGetNumberOfAttributes(rtc->rel))
3461 0 : continue;
3462 156 : pg_att = TupleDescAttr(rtc->rel->rd_att, pg_depend->objsubid - 1);
3463 156 : if (pg_att->attisdropped || pg_att->atttypid != domainOid)
3464 0 : continue;
3465 :
3466 : /*
3467 : * Okay, add column to result. We store the columns in column-number
3468 : * order; this is just a hack to improve predictability of regression
3469 : * test output ...
3470 : */
3471 : Assert(rtc->natts < RelationGetNumberOfAttributes(rtc->rel));
3472 :
3473 156 : ptr = rtc->natts++;
3474 156 : while (ptr > 0 && rtc->atts[ptr - 1] > pg_depend->objsubid)
3475 : {
3476 0 : rtc->atts[ptr] = rtc->atts[ptr - 1];
3477 0 : ptr--;
3478 : }
3479 156 : rtc->atts[ptr] = pg_depend->objsubid;
3480 : }
3481 :
3482 174 : systable_endscan(depScan);
3483 :
3484 174 : relation_close(depRel, AccessShareLock);
3485 :
3486 174 : return result;
3487 : }
3488 :
3489 : /*
3490 : * checkDomainOwner
3491 : *
3492 : * Check that the type is actually a domain and that the current user
3493 : * has permission to do ALTER DOMAIN on it. Throw an error if not.
3494 : */
3495 : void
3496 310 : checkDomainOwner(HeapTuple tup)
3497 : {
3498 310 : Form_pg_type typTup = (Form_pg_type) GETSTRUCT(tup);
3499 :
3500 : /* Check that this is actually a domain */
3501 310 : if (typTup->typtype != TYPTYPE_DOMAIN)
3502 0 : ereport(ERROR,
3503 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
3504 : errmsg("%s is not a domain",
3505 : format_type_be(typTup->oid))));
3506 :
3507 : /* Permission check: must own type */
3508 310 : if (!object_ownercheck(TypeRelationId, typTup->oid, GetUserId()))
3509 0 : aclcheck_error_type(ACLCHECK_NOT_OWNER, typTup->oid);
3510 310 : }
3511 :
3512 : /*
3513 : * domainAddCheckConstraint - code shared between CREATE and ALTER DOMAIN
3514 : */
3515 : static char *
3516 702 : domainAddCheckConstraint(Oid domainOid, Oid domainNamespace, Oid baseTypeOid,
3517 : int typMod, Constraint *constr,
3518 : const char *domainName, ObjectAddress *constrAddr)
3519 : {
3520 : Node *expr;
3521 : char *ccbin;
3522 : ParseState *pstate;
3523 : CoerceToDomainValue *domVal;
3524 : Oid ccoid;
3525 :
3526 : Assert(constr->contype == CONSTR_CHECK);
3527 :
3528 : /*
3529 : * Assign or validate constraint name
3530 : */
3531 702 : if (constr->conname)
3532 : {
3533 396 : if (ConstraintNameIsUsed(CONSTRAINT_DOMAIN,
3534 : domainOid,
3535 396 : constr->conname))
3536 0 : ereport(ERROR,
3537 : (errcode(ERRCODE_DUPLICATE_OBJECT),
3538 : errmsg("constraint \"%s\" for domain \"%s\" already exists",
3539 : constr->conname, domainName)));
3540 : }
3541 : else
3542 306 : constr->conname = ChooseConstraintName(domainName,
3543 : NULL,
3544 : "check",
3545 : domainNamespace,
3546 : NIL);
3547 :
3548 : /*
3549 : * Convert the A_EXPR in raw_expr into an EXPR
3550 : */
3551 702 : pstate = make_parsestate(NULL);
3552 :
3553 : /*
3554 : * Set up a CoerceToDomainValue to represent the occurrence of VALUE in
3555 : * the expression. Note that it will appear to have the type of the base
3556 : * type, not the domain. This seems correct since within the check
3557 : * expression, we should not assume the input value can be considered a
3558 : * member of the domain.
3559 : */
3560 702 : domVal = makeNode(CoerceToDomainValue);
3561 702 : domVal->typeId = baseTypeOid;
3562 702 : domVal->typeMod = typMod;
3563 702 : domVal->collation = get_typcollation(baseTypeOid);
3564 702 : domVal->location = -1; /* will be set when/if used */
3565 :
3566 702 : pstate->p_pre_columnref_hook = replace_domain_constraint_value;
3567 702 : pstate->p_ref_hook_state = domVal;
3568 :
3569 702 : expr = transformExpr(pstate, constr->raw_expr, EXPR_KIND_DOMAIN_CHECK);
3570 :
3571 : /*
3572 : * Make sure it yields a boolean result.
3573 : */
3574 696 : expr = coerce_to_boolean(pstate, expr, "CHECK");
3575 :
3576 : /*
3577 : * Fix up collation information.
3578 : */
3579 696 : assign_expr_collations(pstate, expr);
3580 :
3581 : /*
3582 : * Domains don't allow variables (this is probably dead code now that
3583 : * add_missing_from is history, but let's be sure).
3584 : */
3585 1392 : if (pstate->p_rtable != NIL ||
3586 696 : contain_var_clause(expr))
3587 0 : ereport(ERROR,
3588 : (errcode(ERRCODE_INVALID_COLUMN_REFERENCE),
3589 : errmsg("cannot use table references in domain check constraint")));
3590 :
3591 : /*
3592 : * Convert to string form for storage.
3593 : */
3594 696 : ccbin = nodeToString(expr);
3595 :
3596 : /*
3597 : * Store the constraint in pg_constraint
3598 : */
3599 : ccoid =
3600 696 : CreateConstraintEntry(constr->conname, /* Constraint Name */
3601 : domainNamespace, /* namespace */
3602 : CONSTRAINT_CHECK, /* Constraint Type */
3603 : false, /* Is Deferrable */
3604 : false, /* Is Deferred */
3605 : true, /* Is Enforced */
3606 696 : !constr->skip_validation, /* Is Validated */
3607 : InvalidOid, /* no parent constraint */
3608 : InvalidOid, /* not a relation constraint */
3609 : NULL,
3610 : 0,
3611 : 0,
3612 : domainOid, /* domain constraint */
3613 : InvalidOid, /* no associated index */
3614 : InvalidOid, /* Foreign key fields */
3615 : NULL,
3616 : NULL,
3617 : NULL,
3618 : NULL,
3619 : 0,
3620 : ' ',
3621 : ' ',
3622 : NULL,
3623 : 0,
3624 : ' ',
3625 : NULL, /* not an exclusion constraint */
3626 : expr, /* Tree form of check constraint */
3627 : ccbin, /* Binary form of check constraint */
3628 : true, /* is local */
3629 : 0, /* inhcount */
3630 : false, /* connoinherit */
3631 : false, /* conperiod */
3632 696 : false); /* is_internal */
3633 696 : if (constrAddr)
3634 138 : ObjectAddressSet(*constrAddr, ConstraintRelationId, ccoid);
3635 :
3636 : /*
3637 : * Return the compiled constraint expression so the calling routine can
3638 : * perform any additional required tests.
3639 : */
3640 696 : return ccbin;
3641 : }
3642 :
3643 : /* Parser pre_columnref_hook for domain CHECK constraint parsing */
3644 : static Node *
3645 806 : replace_domain_constraint_value(ParseState *pstate, ColumnRef *cref)
3646 : {
3647 : /*
3648 : * Check for a reference to "value", and if that's what it is, replace
3649 : * with a CoerceToDomainValue as prepared for us by
3650 : * domainAddCheckConstraint. (We handle VALUE as a name, not a keyword, to
3651 : * avoid breaking a lot of applications that have used VALUE as a column
3652 : * name in the past.)
3653 : */
3654 806 : if (list_length(cref->fields) == 1)
3655 : {
3656 806 : Node *field1 = (Node *) linitial(cref->fields);
3657 : char *colname;
3658 :
3659 806 : colname = strVal(field1);
3660 806 : if (strcmp(colname, "value") == 0)
3661 : {
3662 806 : CoerceToDomainValue *domVal = copyObject(pstate->p_ref_hook_state);
3663 :
3664 : /* Propagate location knowledge, if any */
3665 806 : domVal->location = cref->location;
3666 806 : return (Node *) domVal;
3667 : }
3668 : }
3669 0 : return NULL;
3670 : }
3671 :
3672 : /*
3673 : * domainAddNotNullConstraint - code shared between CREATE and ALTER DOMAIN
3674 : */
3675 : static void
3676 130 : domainAddNotNullConstraint(Oid domainOid, Oid domainNamespace, Oid baseTypeOid,
3677 : int typMod, Constraint *constr,
3678 : const char *domainName, ObjectAddress *constrAddr)
3679 : {
3680 : Oid ccoid;
3681 :
3682 : Assert(constr->contype == CONSTR_NOTNULL);
3683 :
3684 : /*
3685 : * Assign or validate constraint name
3686 : */
3687 130 : if (constr->conname)
3688 : {
3689 14 : if (ConstraintNameIsUsed(CONSTRAINT_DOMAIN,
3690 : domainOid,
3691 14 : constr->conname))
3692 0 : ereport(ERROR,
3693 : (errcode(ERRCODE_DUPLICATE_OBJECT),
3694 : errmsg("constraint \"%s\" for domain \"%s\" already exists",
3695 : constr->conname, domainName)));
3696 : }
3697 : else
3698 116 : constr->conname = ChooseConstraintName(domainName,
3699 : NULL,
3700 : "not_null",
3701 : domainNamespace,
3702 : NIL);
3703 :
3704 : /*
3705 : * Store the constraint in pg_constraint
3706 : */
3707 : ccoid =
3708 130 : CreateConstraintEntry(constr->conname, /* Constraint Name */
3709 : domainNamespace, /* namespace */
3710 : CONSTRAINT_NOTNULL, /* Constraint Type */
3711 : false, /* Is Deferrable */
3712 : false, /* Is Deferred */
3713 : true, /* Is Enforced */
3714 130 : !constr->skip_validation, /* Is Validated */
3715 : InvalidOid, /* no parent constraint */
3716 : InvalidOid, /* not a relation constraint */
3717 : NULL,
3718 : 0,
3719 : 0,
3720 : domainOid, /* domain constraint */
3721 : InvalidOid, /* no associated index */
3722 : InvalidOid, /* Foreign key fields */
3723 : NULL,
3724 : NULL,
3725 : NULL,
3726 : NULL,
3727 : 0,
3728 : ' ',
3729 : ' ',
3730 : NULL,
3731 : 0,
3732 : ' ',
3733 : NULL, /* not an exclusion constraint */
3734 : NULL,
3735 : NULL,
3736 : true, /* is local */
3737 : 0, /* inhcount */
3738 : false, /* connoinherit */
3739 : false, /* conperiod */
3740 130 : false); /* is_internal */
3741 :
3742 130 : if (constrAddr)
3743 24 : ObjectAddressSet(*constrAddr, ConstraintRelationId, ccoid);
3744 130 : }
3745 :
3746 :
3747 : /*
3748 : * Execute ALTER TYPE RENAME
3749 : */
3750 : ObjectAddress
3751 32 : RenameType(RenameStmt *stmt)
3752 : {
3753 32 : List *names = castNode(List, stmt->object);
3754 32 : const char *newTypeName = stmt->newname;
3755 : TypeName *typename;
3756 : Oid typeOid;
3757 : Relation rel;
3758 : HeapTuple tup;
3759 : Form_pg_type typTup;
3760 : ObjectAddress address;
3761 :
3762 : /* Make a TypeName so we can use standard type lookup machinery */
3763 32 : typename = makeTypeNameFromNameList(names);
3764 32 : typeOid = typenameTypeId(NULL, typename);
3765 :
3766 : /* Look up the type in the type table */
3767 32 : rel = table_open(TypeRelationId, RowExclusiveLock);
3768 :
3769 32 : tup = SearchSysCacheCopy1(TYPEOID, ObjectIdGetDatum(typeOid));
3770 32 : if (!HeapTupleIsValid(tup))
3771 0 : elog(ERROR, "cache lookup failed for type %u", typeOid);
3772 32 : typTup = (Form_pg_type) GETSTRUCT(tup);
3773 :
3774 : /* check permissions on type */
3775 32 : if (!object_ownercheck(TypeRelationId, typeOid, GetUserId()))
3776 0 : aclcheck_error_type(ACLCHECK_NOT_OWNER, typeOid);
3777 :
3778 : /* ALTER DOMAIN used on a non-domain? */
3779 32 : if (stmt->renameType == OBJECT_DOMAIN && typTup->typtype != TYPTYPE_DOMAIN)
3780 0 : ereport(ERROR,
3781 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
3782 : errmsg("%s is not a domain",
3783 : format_type_be(typeOid))));
3784 :
3785 : /*
3786 : * If it's a composite type, we need to check that it really is a
3787 : * free-standing composite type, and not a table's rowtype. We want people
3788 : * to use ALTER TABLE not ALTER TYPE for that case.
3789 : */
3790 34 : if (typTup->typtype == TYPTYPE_COMPOSITE &&
3791 2 : get_rel_relkind(typTup->typrelid) != RELKIND_COMPOSITE_TYPE)
3792 0 : ereport(ERROR,
3793 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
3794 : errmsg("%s is a table's row type",
3795 : format_type_be(typeOid)),
3796 : /* translator: %s is an SQL ALTER command */
3797 : errhint("Use %s instead.",
3798 : "ALTER TABLE")));
3799 :
3800 : /* don't allow direct alteration of array types, either */
3801 32 : if (IsTrueArrayType(typTup))
3802 0 : ereport(ERROR,
3803 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
3804 : errmsg("cannot alter array type %s",
3805 : format_type_be(typeOid)),
3806 : errhint("You can alter type %s, which will alter the array type as well.",
3807 : format_type_be(typTup->typelem))));
3808 :
3809 : /* we do allow separate renaming of multirange types, though */
3810 :
3811 : /*
3812 : * If type is composite we need to rename associated pg_class entry too.
3813 : * RenameRelationInternal will call RenameTypeInternal automatically.
3814 : */
3815 32 : if (typTup->typtype == TYPTYPE_COMPOSITE)
3816 2 : RenameRelationInternal(typTup->typrelid, newTypeName, false, false);
3817 : else
3818 30 : RenameTypeInternal(typeOid, newTypeName,
3819 : typTup->typnamespace);
3820 :
3821 32 : ObjectAddressSet(address, TypeRelationId, typeOid);
3822 : /* Clean up */
3823 32 : table_close(rel, RowExclusiveLock);
3824 :
3825 32 : return address;
3826 : }
3827 :
3828 : /*
3829 : * Change the owner of a type.
3830 : */
3831 : ObjectAddress
3832 132 : AlterTypeOwner(List *names, Oid newOwnerId, ObjectType objecttype)
3833 : {
3834 : TypeName *typename;
3835 : Oid typeOid;
3836 : Relation rel;
3837 : HeapTuple tup;
3838 : HeapTuple newtup;
3839 : Form_pg_type typTup;
3840 : AclResult aclresult;
3841 : ObjectAddress address;
3842 :
3843 132 : rel = table_open(TypeRelationId, RowExclusiveLock);
3844 :
3845 : /* Make a TypeName so we can use standard type lookup machinery */
3846 132 : typename = makeTypeNameFromNameList(names);
3847 :
3848 : /* Use LookupTypeName here so that shell types can be processed */
3849 132 : tup = LookupTypeName(NULL, typename, NULL, false);
3850 132 : if (tup == NULL)
3851 0 : ereport(ERROR,
3852 : (errcode(ERRCODE_UNDEFINED_OBJECT),
3853 : errmsg("type \"%s\" does not exist",
3854 : TypeNameToString(typename))));
3855 132 : typeOid = typeTypeId(tup);
3856 :
3857 : /* Copy the syscache entry so we can scribble on it below */
3858 132 : newtup = heap_copytuple(tup);
3859 132 : ReleaseSysCache(tup);
3860 132 : tup = newtup;
3861 132 : typTup = (Form_pg_type) GETSTRUCT(tup);
3862 :
3863 : /* Don't allow ALTER DOMAIN on a type */
3864 132 : if (objecttype == OBJECT_DOMAIN && typTup->typtype != TYPTYPE_DOMAIN)
3865 0 : ereport(ERROR,
3866 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
3867 : errmsg("%s is not a domain",
3868 : format_type_be(typeOid))));
3869 :
3870 : /*
3871 : * If it's a composite type, we need to check that it really is a
3872 : * free-standing composite type, and not a table's rowtype. We want people
3873 : * to use ALTER TABLE not ALTER TYPE for that case.
3874 : */
3875 164 : if (typTup->typtype == TYPTYPE_COMPOSITE &&
3876 32 : get_rel_relkind(typTup->typrelid) != RELKIND_COMPOSITE_TYPE)
3877 0 : ereport(ERROR,
3878 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
3879 : errmsg("%s is a table's row type",
3880 : format_type_be(typeOid)),
3881 : /* translator: %s is an SQL ALTER command */
3882 : errhint("Use %s instead.",
3883 : "ALTER TABLE")));
3884 :
3885 : /* don't allow direct alteration of array types, either */
3886 132 : if (IsTrueArrayType(typTup))
3887 0 : ereport(ERROR,
3888 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
3889 : errmsg("cannot alter array type %s",
3890 : format_type_be(typeOid)),
3891 : errhint("You can alter type %s, which will alter the array type as well.",
3892 : format_type_be(typTup->typelem))));
3893 :
3894 : /* don't allow direct alteration of multirange types, either */
3895 132 : if (typTup->typtype == TYPTYPE_MULTIRANGE)
3896 : {
3897 6 : Oid rangetype = get_multirange_range(typeOid);
3898 :
3899 : /* We don't expect get_multirange_range to fail, but cope if so */
3900 6 : ereport(ERROR,
3901 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
3902 : errmsg("cannot alter multirange type %s",
3903 : format_type_be(typeOid)),
3904 : OidIsValid(rangetype) ?
3905 : errhint("You can alter type %s, which will alter the multirange type as well.",
3906 : format_type_be(rangetype)) : 0));
3907 : }
3908 :
3909 : /*
3910 : * If the new owner is the same as the existing owner, consider the
3911 : * command to have succeeded. This is for dump restoration purposes.
3912 : */
3913 126 : if (typTup->typowner != newOwnerId)
3914 : {
3915 : /* Superusers can always do it */
3916 6 : if (!superuser())
3917 : {
3918 : /* Otherwise, must be owner of the existing object */
3919 0 : if (!object_ownercheck(TypeRelationId, typTup->oid, GetUserId()))
3920 0 : aclcheck_error_type(ACLCHECK_NOT_OWNER, typTup->oid);
3921 :
3922 : /* Must be able to become new owner */
3923 0 : check_can_set_role(GetUserId(), newOwnerId);
3924 :
3925 : /* New owner must have CREATE privilege on namespace */
3926 0 : aclresult = object_aclcheck(NamespaceRelationId, typTup->typnamespace,
3927 : newOwnerId,
3928 : ACL_CREATE);
3929 0 : if (aclresult != ACLCHECK_OK)
3930 0 : aclcheck_error(aclresult, OBJECT_SCHEMA,
3931 0 : get_namespace_name(typTup->typnamespace));
3932 : }
3933 :
3934 6 : AlterTypeOwner_oid(typeOid, newOwnerId, true);
3935 : }
3936 :
3937 126 : ObjectAddressSet(address, TypeRelationId, typeOid);
3938 :
3939 : /* Clean up */
3940 126 : table_close(rel, RowExclusiveLock);
3941 :
3942 126 : return address;
3943 : }
3944 :
3945 : /*
3946 : * AlterTypeOwner_oid - change type owner unconditionally
3947 : *
3948 : * This function recurses to handle dependent types (arrays and multiranges).
3949 : * It invokes any necessary access object hooks. If hasDependEntry is true,
3950 : * this function modifies the pg_shdepend entry appropriately (this should be
3951 : * passed as false only for table rowtypes and dependent types).
3952 : *
3953 : * This is used by ALTER TABLE/TYPE OWNER commands, as well as by REASSIGN
3954 : * OWNED BY. It assumes the caller has done all needed checks.
3955 : */
3956 : void
3957 26 : AlterTypeOwner_oid(Oid typeOid, Oid newOwnerId, bool hasDependEntry)
3958 : {
3959 : Relation rel;
3960 : HeapTuple tup;
3961 : Form_pg_type typTup;
3962 :
3963 26 : rel = table_open(TypeRelationId, RowExclusiveLock);
3964 :
3965 26 : tup = SearchSysCache1(TYPEOID, ObjectIdGetDatum(typeOid));
3966 26 : if (!HeapTupleIsValid(tup))
3967 0 : elog(ERROR, "cache lookup failed for type %u", typeOid);
3968 26 : typTup = (Form_pg_type) GETSTRUCT(tup);
3969 :
3970 : /*
3971 : * If it's a composite type, invoke ATExecChangeOwner so that we fix up
3972 : * the pg_class entry properly. That will call back to
3973 : * AlterTypeOwnerInternal to take care of the pg_type entry(s).
3974 : */
3975 26 : if (typTup->typtype == TYPTYPE_COMPOSITE)
3976 8 : ATExecChangeOwner(typTup->typrelid, newOwnerId, true, AccessExclusiveLock);
3977 : else
3978 18 : AlterTypeOwnerInternal(typeOid, newOwnerId);
3979 :
3980 : /* Update owner dependency reference */
3981 26 : if (hasDependEntry)
3982 26 : changeDependencyOnOwner(TypeRelationId, typeOid, newOwnerId);
3983 :
3984 26 : InvokeObjectPostAlterHook(TypeRelationId, typeOid, 0);
3985 :
3986 26 : ReleaseSysCache(tup);
3987 26 : table_close(rel, RowExclusiveLock);
3988 26 : }
3989 :
3990 : /*
3991 : * AlterTypeOwnerInternal - bare-bones type owner change.
3992 : *
3993 : * This routine simply modifies the owner of a pg_type entry, and recurses
3994 : * to handle any dependent types.
3995 : */
3996 : void
3997 660 : AlterTypeOwnerInternal(Oid typeOid, Oid newOwnerId)
3998 : {
3999 : Relation rel;
4000 : HeapTuple tup;
4001 : Form_pg_type typTup;
4002 : Datum repl_val[Natts_pg_type];
4003 : bool repl_null[Natts_pg_type];
4004 : bool repl_repl[Natts_pg_type];
4005 : Acl *newAcl;
4006 : Datum aclDatum;
4007 : bool isNull;
4008 :
4009 660 : rel = table_open(TypeRelationId, RowExclusiveLock);
4010 :
4011 660 : tup = SearchSysCacheCopy1(TYPEOID, ObjectIdGetDatum(typeOid));
4012 660 : if (!HeapTupleIsValid(tup))
4013 0 : elog(ERROR, "cache lookup failed for type %u", typeOid);
4014 660 : typTup = (Form_pg_type) GETSTRUCT(tup);
4015 :
4016 660 : memset(repl_null, false, sizeof(repl_null));
4017 660 : memset(repl_repl, false, sizeof(repl_repl));
4018 :
4019 660 : repl_repl[Anum_pg_type_typowner - 1] = true;
4020 660 : repl_val[Anum_pg_type_typowner - 1] = ObjectIdGetDatum(newOwnerId);
4021 :
4022 660 : aclDatum = heap_getattr(tup,
4023 : Anum_pg_type_typacl,
4024 : RelationGetDescr(rel),
4025 : &isNull);
4026 : /* Null ACLs do not require changes */
4027 660 : if (!isNull)
4028 : {
4029 2 : newAcl = aclnewowner(DatumGetAclP(aclDatum),
4030 : typTup->typowner, newOwnerId);
4031 2 : repl_repl[Anum_pg_type_typacl - 1] = true;
4032 2 : repl_val[Anum_pg_type_typacl - 1] = PointerGetDatum(newAcl);
4033 : }
4034 :
4035 660 : tup = heap_modify_tuple(tup, RelationGetDescr(rel), repl_val, repl_null,
4036 : repl_repl);
4037 :
4038 660 : CatalogTupleUpdate(rel, &tup->t_self, tup);
4039 :
4040 : /* If it has an array type, update that too */
4041 660 : if (OidIsValid(typTup->typarray))
4042 330 : AlterTypeOwnerInternal(typTup->typarray, newOwnerId);
4043 :
4044 : /* If it is a range type, update the associated multirange too */
4045 660 : if (typTup->typtype == TYPTYPE_RANGE)
4046 : {
4047 12 : Oid multirange_typeid = get_range_multirange(typeOid);
4048 :
4049 12 : if (!OidIsValid(multirange_typeid))
4050 0 : ereport(ERROR,
4051 : (errcode(ERRCODE_UNDEFINED_OBJECT),
4052 : errmsg("could not find multirange type for data type %s",
4053 : format_type_be(typeOid))));
4054 12 : AlterTypeOwnerInternal(multirange_typeid, newOwnerId);
4055 : }
4056 :
4057 : /* Clean up */
4058 660 : table_close(rel, RowExclusiveLock);
4059 660 : }
4060 :
4061 : /*
4062 : * Execute ALTER TYPE SET SCHEMA
4063 : */
4064 : ObjectAddress
4065 18 : AlterTypeNamespace(List *names, const char *newschema, ObjectType objecttype,
4066 : Oid *oldschema)
4067 : {
4068 : TypeName *typename;
4069 : Oid typeOid;
4070 : Oid nspOid;
4071 : Oid oldNspOid;
4072 : ObjectAddresses *objsMoved;
4073 : ObjectAddress myself;
4074 :
4075 : /* Make a TypeName so we can use standard type lookup machinery */
4076 18 : typename = makeTypeNameFromNameList(names);
4077 18 : typeOid = typenameTypeId(NULL, typename);
4078 :
4079 : /* Don't allow ALTER DOMAIN on a non-domain type */
4080 18 : if (objecttype == OBJECT_DOMAIN && get_typtype(typeOid) != TYPTYPE_DOMAIN)
4081 0 : ereport(ERROR,
4082 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
4083 : errmsg("%s is not a domain",
4084 : format_type_be(typeOid))));
4085 :
4086 : /* get schema OID and check its permissions */
4087 18 : nspOid = LookupCreationNamespace(newschema);
4088 :
4089 18 : objsMoved = new_object_addresses();
4090 18 : oldNspOid = AlterTypeNamespace_oid(typeOid, nspOid, false, objsMoved);
4091 18 : free_object_addresses(objsMoved);
4092 :
4093 18 : if (oldschema)
4094 18 : *oldschema = oldNspOid;
4095 :
4096 18 : ObjectAddressSet(myself, TypeRelationId, typeOid);
4097 :
4098 18 : return myself;
4099 : }
4100 :
4101 : /*
4102 : * ALTER TYPE SET SCHEMA, where the caller has already looked up the OIDs
4103 : * of the type and the target schema and checked the schema's privileges.
4104 : *
4105 : * If ignoreDependent is true, we silently ignore dependent types
4106 : * (array types and table rowtypes) rather than raising errors.
4107 : *
4108 : * This entry point is exported for use by AlterObjectNamespace_oid,
4109 : * which doesn't want errors when it passes OIDs of dependent types.
4110 : *
4111 : * Returns the type's old namespace OID, or InvalidOid if we did nothing.
4112 : */
4113 : Oid
4114 34 : AlterTypeNamespace_oid(Oid typeOid, Oid nspOid, bool ignoreDependent,
4115 : ObjectAddresses *objsMoved)
4116 : {
4117 : Oid elemOid;
4118 :
4119 : /* check permissions on type */
4120 34 : if (!object_ownercheck(TypeRelationId, typeOid, GetUserId()))
4121 0 : aclcheck_error_type(ACLCHECK_NOT_OWNER, typeOid);
4122 :
4123 : /* don't allow direct alteration of array types */
4124 34 : elemOid = get_element_type(typeOid);
4125 34 : if (OidIsValid(elemOid) && get_array_type(elemOid) == typeOid)
4126 : {
4127 8 : if (ignoreDependent)
4128 8 : return InvalidOid;
4129 0 : ereport(ERROR,
4130 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
4131 : errmsg("cannot alter array type %s",
4132 : format_type_be(typeOid)),
4133 : errhint("You can alter type %s, which will alter the array type as well.",
4134 : format_type_be(elemOid))));
4135 : }
4136 :
4137 : /* and do the work */
4138 26 : return AlterTypeNamespaceInternal(typeOid, nspOid,
4139 : false, /* isImplicitArray */
4140 : ignoreDependent, /* ignoreDependent */
4141 : true, /* errorOnTableType */
4142 : objsMoved);
4143 : }
4144 :
4145 : /*
4146 : * Move specified type to new namespace.
4147 : *
4148 : * Caller must have already checked privileges.
4149 : *
4150 : * The function automatically recurses to process the type's array type,
4151 : * if any. isImplicitArray should be true only when doing this internal
4152 : * recursion (outside callers must never try to move an array type directly).
4153 : *
4154 : * If ignoreDependent is true, we silently don't process table types.
4155 : *
4156 : * If errorOnTableType is true, the function errors out if the type is
4157 : * a table type. ALTER TABLE has to be used to move a table to a new
4158 : * namespace. (This flag is ignored if ignoreDependent is true.)
4159 : *
4160 : * We also do nothing if the type is already listed in *objsMoved.
4161 : * After a successful move, we add the type to *objsMoved.
4162 : *
4163 : * Returns the type's old namespace OID, or InvalidOid if we did nothing.
4164 : */
4165 : Oid
4166 218 : AlterTypeNamespaceInternal(Oid typeOid, Oid nspOid,
4167 : bool isImplicitArray,
4168 : bool ignoreDependent,
4169 : bool errorOnTableType,
4170 : ObjectAddresses *objsMoved)
4171 : {
4172 : Relation rel;
4173 : HeapTuple tup;
4174 : Form_pg_type typform;
4175 : Oid oldNspOid;
4176 : Oid arrayOid;
4177 : bool isCompositeType;
4178 : ObjectAddress thisobj;
4179 :
4180 : /*
4181 : * Make sure we haven't moved this object previously.
4182 : */
4183 218 : thisobj.classId = TypeRelationId;
4184 218 : thisobj.objectId = typeOid;
4185 218 : thisobj.objectSubId = 0;
4186 :
4187 218 : if (object_address_present(&thisobj, objsMoved))
4188 0 : return InvalidOid;
4189 :
4190 218 : rel = table_open(TypeRelationId, RowExclusiveLock);
4191 :
4192 218 : tup = SearchSysCacheCopy1(TYPEOID, ObjectIdGetDatum(typeOid));
4193 218 : if (!HeapTupleIsValid(tup))
4194 0 : elog(ERROR, "cache lookup failed for type %u", typeOid);
4195 218 : typform = (Form_pg_type) GETSTRUCT(tup);
4196 :
4197 218 : oldNspOid = typform->typnamespace;
4198 218 : arrayOid = typform->typarray;
4199 :
4200 : /* If the type is already there, we scan skip these next few checks. */
4201 218 : if (oldNspOid != nspOid)
4202 : {
4203 : /* common checks on switching namespaces */
4204 182 : CheckSetNamespace(oldNspOid, nspOid);
4205 :
4206 : /* check for duplicate name (more friendly than unique-index failure) */
4207 182 : if (SearchSysCacheExists2(TYPENAMENSP,
4208 : NameGetDatum(&typform->typname),
4209 : ObjectIdGetDatum(nspOid)))
4210 0 : ereport(ERROR,
4211 : (errcode(ERRCODE_DUPLICATE_OBJECT),
4212 : errmsg("type \"%s\" already exists in schema \"%s\"",
4213 : NameStr(typform->typname),
4214 : get_namespace_name(nspOid))));
4215 : }
4216 :
4217 : /* Detect whether type is a composite type (but not a table rowtype) */
4218 218 : isCompositeType =
4219 318 : (typform->typtype == TYPTYPE_COMPOSITE &&
4220 100 : get_rel_relkind(typform->typrelid) == RELKIND_COMPOSITE_TYPE);
4221 :
4222 : /* Enforce not-table-type if requested */
4223 218 : if (typform->typtype == TYPTYPE_COMPOSITE && !isCompositeType)
4224 : {
4225 86 : if (ignoreDependent)
4226 : {
4227 2 : table_close(rel, RowExclusiveLock);
4228 2 : return InvalidOid;
4229 : }
4230 84 : if (errorOnTableType)
4231 0 : ereport(ERROR,
4232 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
4233 : errmsg("%s is a table's row type",
4234 : format_type_be(typeOid)),
4235 : /* translator: %s is an SQL ALTER command */
4236 : errhint("Use %s instead.", "ALTER TABLE")));
4237 : }
4238 :
4239 216 : if (oldNspOid != nspOid)
4240 : {
4241 : /* OK, modify the pg_type row */
4242 :
4243 : /* tup is a copy, so we can scribble directly on it */
4244 180 : typform->typnamespace = nspOid;
4245 :
4246 180 : CatalogTupleUpdate(rel, &tup->t_self, tup);
4247 : }
4248 :
4249 : /*
4250 : * Composite types have pg_class entries.
4251 : *
4252 : * We need to modify the pg_class tuple as well to reflect the change of
4253 : * schema.
4254 : */
4255 216 : if (isCompositeType)
4256 : {
4257 : Relation classRel;
4258 :
4259 14 : classRel = table_open(RelationRelationId, RowExclusiveLock);
4260 :
4261 14 : AlterRelationNamespaceInternal(classRel, typform->typrelid,
4262 : oldNspOid, nspOid,
4263 : false, objsMoved);
4264 :
4265 14 : table_close(classRel, RowExclusiveLock);
4266 :
4267 : /*
4268 : * Check for constraints associated with the composite type (we don't
4269 : * currently support this, but probably will someday).
4270 : */
4271 14 : AlterConstraintNamespaces(typform->typrelid, oldNspOid,
4272 : nspOid, false, objsMoved);
4273 : }
4274 : else
4275 : {
4276 : /* If it's a domain, it might have constraints */
4277 202 : if (typform->typtype == TYPTYPE_DOMAIN)
4278 6 : AlterConstraintNamespaces(typeOid, oldNspOid, nspOid, true,
4279 : objsMoved);
4280 : }
4281 :
4282 : /*
4283 : * Update dependency on schema, if any --- a table rowtype has not got
4284 : * one, and neither does an implicit array.
4285 : */
4286 216 : if (oldNspOid != nspOid &&
4287 172 : (isCompositeType || typform->typtype != TYPTYPE_COMPOSITE) &&
4288 108 : !isImplicitArray)
4289 18 : if (changeDependencyFor(TypeRelationId, typeOid,
4290 : NamespaceRelationId, oldNspOid, nspOid) != 1)
4291 0 : elog(ERROR, "could not change schema dependency for type \"%s\"",
4292 : format_type_be(typeOid));
4293 :
4294 216 : InvokeObjectPostAlterHook(TypeRelationId, typeOid, 0);
4295 :
4296 216 : heap_freetuple(tup);
4297 :
4298 216 : table_close(rel, RowExclusiveLock);
4299 :
4300 216 : add_exact_object_address(&thisobj, objsMoved);
4301 :
4302 : /* Recursively alter the associated array type, if any */
4303 216 : if (OidIsValid(arrayOid))
4304 108 : AlterTypeNamespaceInternal(arrayOid, nspOid,
4305 : true, /* isImplicitArray */
4306 : false, /* ignoreDependent */
4307 : true, /* errorOnTableType */
4308 : objsMoved);
4309 :
4310 216 : return oldNspOid;
4311 : }
4312 :
4313 : /*
4314 : * AlterType
4315 : * ALTER TYPE <type> SET (option = ...)
4316 : *
4317 : * NOTE: the set of changes that can be allowed here is constrained by many
4318 : * non-obvious implementation restrictions. Tread carefully when considering
4319 : * adding new flexibility.
4320 : */
4321 : ObjectAddress
4322 60 : AlterType(AlterTypeStmt *stmt)
4323 : {
4324 : ObjectAddress address;
4325 : Relation catalog;
4326 : TypeName *typename;
4327 : HeapTuple tup;
4328 : Oid typeOid;
4329 : Form_pg_type typForm;
4330 60 : bool requireSuper = false;
4331 : AlterTypeRecurseParams atparams;
4332 : ListCell *pl;
4333 :
4334 60 : catalog = table_open(TypeRelationId, RowExclusiveLock);
4335 :
4336 : /* Make a TypeName so we can use standard type lookup machinery */
4337 60 : typename = makeTypeNameFromNameList(stmt->typeName);
4338 60 : tup = typenameType(NULL, typename, NULL);
4339 :
4340 54 : typeOid = typeTypeId(tup);
4341 54 : typForm = (Form_pg_type) GETSTRUCT(tup);
4342 :
4343 : /* Process options */
4344 54 : memset(&atparams, 0, sizeof(atparams));
4345 154 : foreach(pl, stmt->options)
4346 : {
4347 106 : DefElem *defel = (DefElem *) lfirst(pl);
4348 :
4349 106 : if (strcmp(defel->defname, "storage") == 0)
4350 : {
4351 12 : char *a = defGetString(defel);
4352 :
4353 12 : if (pg_strcasecmp(a, "plain") == 0)
4354 6 : atparams.storage = TYPSTORAGE_PLAIN;
4355 6 : else if (pg_strcasecmp(a, "external") == 0)
4356 0 : atparams.storage = TYPSTORAGE_EXTERNAL;
4357 6 : else if (pg_strcasecmp(a, "extended") == 0)
4358 6 : atparams.storage = TYPSTORAGE_EXTENDED;
4359 0 : else if (pg_strcasecmp(a, "main") == 0)
4360 0 : atparams.storage = TYPSTORAGE_MAIN;
4361 : else
4362 0 : ereport(ERROR,
4363 : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
4364 : errmsg("storage \"%s\" not recognized", a)));
4365 :
4366 : /*
4367 : * Validate the storage request. If the type isn't varlena, it
4368 : * certainly doesn't support non-PLAIN storage.
4369 : */
4370 12 : if (atparams.storage != TYPSTORAGE_PLAIN && typForm->typlen != -1)
4371 0 : ereport(ERROR,
4372 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
4373 : errmsg("fixed-size types must have storage PLAIN")));
4374 :
4375 : /*
4376 : * Switching from PLAIN to non-PLAIN is allowed, but it requires
4377 : * superuser, since we can't validate that the type's C functions
4378 : * will support it. Switching from non-PLAIN to PLAIN is
4379 : * disallowed outright, because it's not practical to ensure that
4380 : * no tables have toasted values of the type. Switching among
4381 : * different non-PLAIN settings is OK, since it just constitutes a
4382 : * change in the strategy requested for columns created in the
4383 : * future.
4384 : */
4385 12 : if (atparams.storage != TYPSTORAGE_PLAIN &&
4386 6 : typForm->typstorage == TYPSTORAGE_PLAIN)
4387 0 : requireSuper = true;
4388 12 : else if (atparams.storage == TYPSTORAGE_PLAIN &&
4389 6 : typForm->typstorage != TYPSTORAGE_PLAIN)
4390 6 : ereport(ERROR,
4391 : (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
4392 : errmsg("cannot change type's storage to PLAIN")));
4393 :
4394 6 : atparams.updateStorage = true;
4395 : }
4396 94 : else if (strcmp(defel->defname, "receive") == 0)
4397 : {
4398 28 : if (defel->arg != NULL)
4399 28 : atparams.receiveOid =
4400 28 : findTypeReceiveFunction(defGetQualifiedName(defel),
4401 : typeOid);
4402 : else
4403 0 : atparams.receiveOid = InvalidOid; /* NONE, remove function */
4404 28 : atparams.updateReceive = true;
4405 : /* Replacing an I/O function requires superuser. */
4406 28 : requireSuper = true;
4407 : }
4408 66 : else if (strcmp(defel->defname, "send") == 0)
4409 : {
4410 28 : if (defel->arg != NULL)
4411 28 : atparams.sendOid =
4412 28 : findTypeSendFunction(defGetQualifiedName(defel),
4413 : typeOid);
4414 : else
4415 0 : atparams.sendOid = InvalidOid; /* NONE, remove function */
4416 28 : atparams.updateSend = true;
4417 : /* Replacing an I/O function requires superuser. */
4418 28 : requireSuper = true;
4419 : }
4420 38 : else if (strcmp(defel->defname, "typmod_in") == 0)
4421 : {
4422 6 : if (defel->arg != NULL)
4423 6 : atparams.typmodinOid =
4424 6 : findTypeTypmodinFunction(defGetQualifiedName(defel));
4425 : else
4426 0 : atparams.typmodinOid = InvalidOid; /* NONE, remove function */
4427 6 : atparams.updateTypmodin = true;
4428 : /* Replacing an I/O function requires superuser. */
4429 6 : requireSuper = true;
4430 : }
4431 32 : else if (strcmp(defel->defname, "typmod_out") == 0)
4432 : {
4433 6 : if (defel->arg != NULL)
4434 6 : atparams.typmodoutOid =
4435 6 : findTypeTypmodoutFunction(defGetQualifiedName(defel));
4436 : else
4437 0 : atparams.typmodoutOid = InvalidOid; /* NONE, remove function */
4438 6 : atparams.updateTypmodout = true;
4439 : /* Replacing an I/O function requires superuser. */
4440 6 : requireSuper = true;
4441 : }
4442 26 : else if (strcmp(defel->defname, "analyze") == 0)
4443 : {
4444 6 : if (defel->arg != NULL)
4445 6 : atparams.analyzeOid =
4446 6 : findTypeAnalyzeFunction(defGetQualifiedName(defel),
4447 : typeOid);
4448 : else
4449 0 : atparams.analyzeOid = InvalidOid; /* NONE, remove function */
4450 6 : atparams.updateAnalyze = true;
4451 : /* Replacing an analyze function requires superuser. */
4452 6 : requireSuper = true;
4453 : }
4454 20 : else if (strcmp(defel->defname, "subscript") == 0)
4455 : {
4456 20 : if (defel->arg != NULL)
4457 20 : atparams.subscriptOid =
4458 20 : findTypeSubscriptingFunction(defGetQualifiedName(defel),
4459 : typeOid);
4460 : else
4461 0 : atparams.subscriptOid = InvalidOid; /* NONE, remove function */
4462 20 : atparams.updateSubscript = true;
4463 : /* Replacing a subscript function requires superuser. */
4464 20 : requireSuper = true;
4465 : }
4466 :
4467 : /*
4468 : * The rest of the options that CREATE accepts cannot be changed.
4469 : * Check for them so that we can give a meaningful error message.
4470 : */
4471 0 : else if (strcmp(defel->defname, "input") == 0 ||
4472 0 : strcmp(defel->defname, "output") == 0 ||
4473 0 : strcmp(defel->defname, "internallength") == 0 ||
4474 0 : strcmp(defel->defname, "passedbyvalue") == 0 ||
4475 0 : strcmp(defel->defname, "alignment") == 0 ||
4476 0 : strcmp(defel->defname, "like") == 0 ||
4477 0 : strcmp(defel->defname, "category") == 0 ||
4478 0 : strcmp(defel->defname, "preferred") == 0 ||
4479 0 : strcmp(defel->defname, "default") == 0 ||
4480 0 : strcmp(defel->defname, "element") == 0 ||
4481 0 : strcmp(defel->defname, "delimiter") == 0 ||
4482 0 : strcmp(defel->defname, "collatable") == 0)
4483 0 : ereport(ERROR,
4484 : (errcode(ERRCODE_SYNTAX_ERROR),
4485 : errmsg("type attribute \"%s\" cannot be changed",
4486 : defel->defname)));
4487 : else
4488 0 : ereport(ERROR,
4489 : (errcode(ERRCODE_SYNTAX_ERROR),
4490 : errmsg("type attribute \"%s\" not recognized",
4491 : defel->defname)));
4492 : }
4493 :
4494 : /*
4495 : * Permissions check. Require superuser if we decided the command
4496 : * requires that, else must own the type.
4497 : */
4498 48 : if (requireSuper)
4499 : {
4500 42 : if (!superuser())
4501 0 : ereport(ERROR,
4502 : (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
4503 : errmsg("must be superuser to alter a type")));
4504 : }
4505 : else
4506 : {
4507 6 : if (!object_ownercheck(TypeRelationId, typeOid, GetUserId()))
4508 0 : aclcheck_error_type(ACLCHECK_NOT_OWNER, typeOid);
4509 : }
4510 :
4511 : /*
4512 : * We disallow all forms of ALTER TYPE SET on types that aren't plain base
4513 : * types. It would for example be highly unsafe, not to mention
4514 : * pointless, to change the send/receive functions for a composite type.
4515 : * Moreover, pg_dump has no support for changing these properties on
4516 : * non-base types. We might weaken this someday, but not now.
4517 : *
4518 : * Note: if you weaken this enough to allow composite types, be sure to
4519 : * adjust the GenerateTypeDependencies call in AlterTypeRecurse.
4520 : */
4521 48 : if (typForm->typtype != TYPTYPE_BASE)
4522 0 : ereport(ERROR,
4523 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
4524 : errmsg("%s is not a base type",
4525 : format_type_be(typeOid))));
4526 :
4527 : /*
4528 : * For the same reasons, don't allow direct alteration of array types.
4529 : */
4530 48 : if (IsTrueArrayType(typForm))
4531 0 : ereport(ERROR,
4532 : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
4533 : errmsg("%s is not a base type",
4534 : format_type_be(typeOid))));
4535 :
4536 : /* OK, recursively update this type and any arrays/domains over it */
4537 48 : AlterTypeRecurse(typeOid, false, tup, catalog, &atparams);
4538 :
4539 : /* Clean up */
4540 48 : ReleaseSysCache(tup);
4541 :
4542 48 : table_close(catalog, RowExclusiveLock);
4543 :
4544 48 : ObjectAddressSet(address, TypeRelationId, typeOid);
4545 :
4546 48 : return address;
4547 : }
4548 :
4549 : /*
4550 : * AlterTypeRecurse: one recursion step for AlterType()
4551 : *
4552 : * Apply the changes specified by "atparams" to the type identified by
4553 : * "typeOid", whose existing pg_type tuple is "tup". If necessary,
4554 : * recursively update its array type as well. Then search for any domains
4555 : * over this type, and recursively apply (most of) the same changes to those
4556 : * domains.
4557 : *
4558 : * We need this because the system generally assumes that a domain inherits
4559 : * many properties from its base type. See DefineDomain() above for details
4560 : * of what is inherited. Arrays inherit a smaller number of properties,
4561 : * but not none.
4562 : *
4563 : * There's a race condition here, in that some other transaction could
4564 : * concurrently add another domain atop this base type; we'd miss updating
4565 : * that one. Hence, be wary of allowing ALTER TYPE to change properties for
4566 : * which it'd be really fatal for a domain to be out of sync with its base
4567 : * type (typlen, for example). In practice, races seem unlikely to be an
4568 : * issue for plausible use-cases for ALTER TYPE. If one does happen, it could
4569 : * be fixed by re-doing the same ALTER TYPE once all prior transactions have
4570 : * committed.
4571 : */
4572 : static void
4573 66 : AlterTypeRecurse(Oid typeOid, bool isImplicitArray,
4574 : HeapTuple tup, Relation catalog,
4575 : AlterTypeRecurseParams *atparams)
4576 : {
4577 : Datum values[Natts_pg_type];
4578 : bool nulls[Natts_pg_type];
4579 : bool replaces[Natts_pg_type];
4580 : HeapTuple newtup;
4581 : SysScanDesc scan;
4582 : ScanKeyData key[1];
4583 : HeapTuple domainTup;
4584 :
4585 : /* Since this function recurses, it could be driven to stack overflow */
4586 66 : check_stack_depth();
4587 :
4588 : /* Update the current type's tuple */
4589 66 : memset(values, 0, sizeof(values));
4590 66 : memset(nulls, 0, sizeof(nulls));
4591 66 : memset(replaces, 0, sizeof(replaces));
4592 :
4593 66 : if (atparams->updateStorage)
4594 : {
4595 12 : replaces[Anum_pg_type_typstorage - 1] = true;
4596 12 : values[Anum_pg_type_typstorage - 1] = CharGetDatum(atparams->storage);
4597 : }
4598 66 : if (atparams->updateReceive)
4599 : {
4600 28 : replaces[Anum_pg_type_typreceive - 1] = true;
4601 28 : values[Anum_pg_type_typreceive - 1] = ObjectIdGetDatum(atparams->receiveOid);
4602 : }
4603 66 : if (atparams->updateSend)
4604 : {
4605 34 : replaces[Anum_pg_type_typsend - 1] = true;
4606 34 : values[Anum_pg_type_typsend - 1] = ObjectIdGetDatum(atparams->sendOid);
4607 : }
4608 66 : if (atparams->updateTypmodin)
4609 : {
4610 12 : replaces[Anum_pg_type_typmodin - 1] = true;
4611 12 : values[Anum_pg_type_typmodin - 1] = ObjectIdGetDatum(atparams->typmodinOid);
4612 : }
4613 66 : if (atparams->updateTypmodout)
4614 : {
4615 12 : replaces[Anum_pg_type_typmodout - 1] = true;
4616 12 : values[Anum_pg_type_typmodout - 1] = ObjectIdGetDatum(atparams->typmodoutOid);
4617 : }
4618 66 : if (atparams->updateAnalyze)
4619 : {
4620 12 : replaces[Anum_pg_type_typanalyze - 1] = true;
4621 12 : values[Anum_pg_type_typanalyze - 1] = ObjectIdGetDatum(atparams->analyzeOid);
4622 : }
4623 66 : if (atparams->updateSubscript)
4624 : {
4625 20 : replaces[Anum_pg_type_typsubscript - 1] = true;
4626 20 : values[Anum_pg_type_typsubscript - 1] = ObjectIdGetDatum(atparams->subscriptOid);
4627 : }
4628 :
4629 66 : newtup = heap_modify_tuple(tup, RelationGetDescr(catalog),
4630 : values, nulls, replaces);
4631 :
4632 66 : CatalogTupleUpdate(catalog, &newtup->t_self, newtup);
4633 :
4634 : /* Rebuild dependencies for this type */
4635 66 : GenerateTypeDependencies(newtup,
4636 : catalog,
4637 : NULL, /* don't have defaultExpr handy */
4638 : NULL, /* don't have typacl handy */
4639 : 0, /* we rejected composite types above */
4640 : isImplicitArray, /* it might be an array */
4641 : isImplicitArray, /* dependent iff it's array */
4642 : false, /* don't touch extension membership */
4643 : true);
4644 :
4645 66 : InvokeObjectPostAlterHook(TypeRelationId, typeOid, 0);
4646 :
4647 : /*
4648 : * Arrays inherit their base type's typmodin and typmodout, but none of
4649 : * the other properties we're concerned with here. Recurse to the array
4650 : * type if needed.
4651 : */
4652 66 : if (!isImplicitArray &&
4653 60 : (atparams->updateTypmodin || atparams->updateTypmodout))
4654 : {
4655 6 : Oid arrtypoid = ((Form_pg_type) GETSTRUCT(newtup))->typarray;
4656 :
4657 6 : if (OidIsValid(arrtypoid))
4658 : {
4659 : HeapTuple arrtup;
4660 : AlterTypeRecurseParams arrparams;
4661 :
4662 6 : arrtup = SearchSysCache1(TYPEOID, ObjectIdGetDatum(arrtypoid));
4663 6 : if (!HeapTupleIsValid(arrtup))
4664 0 : elog(ERROR, "cache lookup failed for type %u", arrtypoid);
4665 :
4666 6 : memset(&arrparams, 0, sizeof(arrparams));
4667 6 : arrparams.updateTypmodin = atparams->updateTypmodin;
4668 6 : arrparams.updateTypmodout = atparams->updateTypmodout;
4669 6 : arrparams.typmodinOid = atparams->typmodinOid;
4670 6 : arrparams.typmodoutOid = atparams->typmodoutOid;
4671 :
4672 6 : AlterTypeRecurse(arrtypoid, true, arrtup, catalog, &arrparams);
4673 :
4674 6 : ReleaseSysCache(arrtup);
4675 : }
4676 : }
4677 :
4678 : /*
4679 : * Now we need to recurse to domains. However, some properties are not
4680 : * inherited by domains, so clear the update flags for those.
4681 : */
4682 66 : atparams->updateReceive = false; /* domains use F_DOMAIN_RECV */
4683 66 : atparams->updateTypmodin = false; /* domains don't have typmods */
4684 66 : atparams->updateTypmodout = false;
4685 66 : atparams->updateSubscript = false; /* domains don't have subscriptors */
4686 :
4687 : /* Skip the scan if nothing remains to be done */
4688 66 : if (!(atparams->updateStorage ||
4689 54 : atparams->updateSend ||
4690 20 : atparams->updateAnalyze))
4691 20 : return;
4692 :
4693 : /* Search pg_type for possible domains over this type */
4694 46 : ScanKeyInit(&key[0],
4695 : Anum_pg_type_typbasetype,
4696 : BTEqualStrategyNumber, F_OIDEQ,
4697 : ObjectIdGetDatum(typeOid));
4698 :
4699 46 : scan = systable_beginscan(catalog, InvalidOid, false,
4700 : NULL, 1, key);
4701 :
4702 58 : while ((domainTup = systable_getnext(scan)) != NULL)
4703 : {
4704 12 : Form_pg_type domainForm = (Form_pg_type) GETSTRUCT(domainTup);
4705 :
4706 : /*
4707 : * Shouldn't have a nonzero typbasetype in a non-domain, but let's
4708 : * check
4709 : */
4710 12 : if (domainForm->typtype != TYPTYPE_DOMAIN)
4711 0 : continue;
4712 :
4713 12 : AlterTypeRecurse(domainForm->oid, false, domainTup, catalog, atparams);
4714 : }
4715 :
4716 46 : systable_endscan(scan);
4717 : }
|
