mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-09-27 21:52:02 +02:00
69946411d3
coding by Mark Cave-Ayland, some kibitzing by Tom Lane. initdb forced due to new column in pg_type.
2098 lines
57 KiB
C
2098 lines
57 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* typecmds.c
|
|
* Routines for SQL commands that manipulate types (and domains).
|
|
*
|
|
* Portions Copyright (c) 1996-2003, PostgreSQL Global Development Group
|
|
* Portions Copyright (c) 1994, Regents of the University of California
|
|
*
|
|
*
|
|
* IDENTIFICATION
|
|
* $PostgreSQL: pgsql/src/backend/commands/typecmds.c,v 1.53 2004/02/12 23:41:02 tgl Exp $
|
|
*
|
|
* DESCRIPTION
|
|
* The "DefineFoo" routines take the parse tree and pick out the
|
|
* appropriate arguments/flags, passing the results to the
|
|
* corresponding "FooDefine" routines (in src/catalog) that do
|
|
* the actual catalog-munging. These routines also verify permission
|
|
* of the user to execute the command.
|
|
*
|
|
* NOTES
|
|
* These things must be defined and committed in the following order:
|
|
* "create function":
|
|
* input/output, recv/send functions
|
|
* "create type":
|
|
* type
|
|
* "create operator":
|
|
* operators
|
|
*
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
#include "postgres.h"
|
|
|
|
#include "access/heapam.h"
|
|
#include "access/genam.h"
|
|
#include "catalog/catname.h"
|
|
#include "catalog/dependency.h"
|
|
#include "catalog/heap.h"
|
|
#include "catalog/indexing.h"
|
|
#include "catalog/namespace.h"
|
|
#include "catalog/pg_constraint.h"
|
|
#include "catalog/pg_depend.h"
|
|
#include "catalog/pg_type.h"
|
|
#include "commands/defrem.h"
|
|
#include "commands/tablecmds.h"
|
|
#include "commands/typecmds.h"
|
|
#include "executor/executor.h"
|
|
#include "miscadmin.h"
|
|
#include "nodes/execnodes.h"
|
|
#include "nodes/nodes.h"
|
|
#include "optimizer/clauses.h"
|
|
#include "optimizer/planmain.h"
|
|
#include "optimizer/var.h"
|
|
#include "parser/parse_coerce.h"
|
|
#include "parser/parse_expr.h"
|
|
#include "parser/parse_func.h"
|
|
#include "parser/parse_relation.h"
|
|
#include "parser/parse_type.h"
|
|
#include "utils/acl.h"
|
|
#include "utils/builtins.h"
|
|
#include "utils/fmgroids.h"
|
|
#include "utils/lsyscache.h"
|
|
#include "utils/syscache.h"
|
|
|
|
|
|
/* result structure for get_rels_with_domain() */
|
|
typedef struct
|
|
{
|
|
Relation rel; /* opened and locked relation */
|
|
int natts; /* number of attributes of interest */
|
|
int *atts; /* attribute numbers */
|
|
/* atts[] is of allocated length RelationGetNumberOfAttributes(rel) */
|
|
} RelToCheck;
|
|
|
|
|
|
static Oid findTypeInputFunction(List *procname, Oid typeOid);
|
|
static Oid findTypeOutputFunction(List *procname, Oid typeOid);
|
|
static Oid findTypeReceiveFunction(List *procname, Oid typeOid);
|
|
static Oid findTypeSendFunction(List *procname, Oid typeOid);
|
|
static Oid findTypeAnalyzeFunction(List *procname, Oid typeOid);
|
|
static List *get_rels_with_domain(Oid domainOid, LOCKMODE lockmode);
|
|
static void domainOwnerCheck(HeapTuple tup, TypeName *typename);
|
|
static char *domainAddConstraint(Oid domainOid, Oid domainNamespace,
|
|
Oid baseTypeOid,
|
|
int typMod, Constraint *constr,
|
|
int *counter, char *domainName);
|
|
|
|
|
|
/*
|
|
* DefineType
|
|
* Registers a new type.
|
|
*/
|
|
void
|
|
DefineType(List *names, List *parameters)
|
|
{
|
|
char *typeName;
|
|
Oid typeNamespace;
|
|
AclResult aclresult;
|
|
int16 internalLength = -1; /* default: variable-length */
|
|
Oid elemType = InvalidOid;
|
|
List *inputName = NIL;
|
|
List *outputName = NIL;
|
|
List *receiveName = NIL;
|
|
List *sendName = NIL;
|
|
List *analyzeName = NIL;
|
|
char *defaultValue = NULL;
|
|
bool byValue = false;
|
|
char delimiter = DEFAULT_TYPDELIM;
|
|
char alignment = 'i'; /* default alignment */
|
|
char storage = 'p'; /* default TOAST storage method */
|
|
Oid inputOid;
|
|
Oid outputOid;
|
|
Oid receiveOid = InvalidOid;
|
|
Oid sendOid = InvalidOid;
|
|
Oid analyzeOid = InvalidOid;
|
|
char *shadow_type;
|
|
List *pl;
|
|
Oid typoid;
|
|
Oid resulttype;
|
|
|
|
/* Convert list of names to a name and namespace */
|
|
typeNamespace = QualifiedNameGetCreationNamespace(names, &typeName);
|
|
|
|
/* Check we have creation rights in target namespace */
|
|
aclresult = pg_namespace_aclcheck(typeNamespace, GetUserId(), ACL_CREATE);
|
|
if (aclresult != ACLCHECK_OK)
|
|
aclcheck_error(aclresult, ACL_KIND_NAMESPACE,
|
|
get_namespace_name(typeNamespace));
|
|
|
|
/*
|
|
* Type names must be one character shorter than other names, allowing
|
|
* room to create the corresponding array type name with prepended
|
|
* "_".
|
|
*/
|
|
if (strlen(typeName) > (NAMEDATALEN - 2))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_NAME),
|
|
errmsg("type names must be %d characters or less",
|
|
NAMEDATALEN - 2)));
|
|
|
|
foreach(pl, parameters)
|
|
{
|
|
DefElem *defel = (DefElem *) lfirst(pl);
|
|
|
|
if (strcasecmp(defel->defname, "internallength") == 0)
|
|
internalLength = defGetTypeLength(defel);
|
|
else if (strcasecmp(defel->defname, "externallength") == 0)
|
|
; /* ignored -- remove after 7.3 */
|
|
else if (strcasecmp(defel->defname, "input") == 0)
|
|
inputName = defGetQualifiedName(defel);
|
|
else if (strcasecmp(defel->defname, "output") == 0)
|
|
outputName = defGetQualifiedName(defel);
|
|
else if (strcasecmp(defel->defname, "receive") == 0)
|
|
receiveName = defGetQualifiedName(defel);
|
|
else if (strcasecmp(defel->defname, "send") == 0)
|
|
sendName = defGetQualifiedName(defel);
|
|
else if (strcasecmp(defel->defname, "analyze") == 0 ||
|
|
strcasecmp(defel->defname, "analyse") == 0)
|
|
analyzeName = defGetQualifiedName(defel);
|
|
else if (strcasecmp(defel->defname, "delimiter") == 0)
|
|
{
|
|
char *p = defGetString(defel);
|
|
|
|
delimiter = p[0];
|
|
}
|
|
else if (strcasecmp(defel->defname, "element") == 0)
|
|
{
|
|
elemType = typenameTypeId(defGetTypeName(defel));
|
|
/* disallow arrays of pseudotypes */
|
|
if (get_typtype(elemType) == 'p')
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_DATATYPE_MISMATCH),
|
|
errmsg("array element type cannot be %s",
|
|
format_type_be(elemType))));
|
|
}
|
|
else if (strcasecmp(defel->defname, "default") == 0)
|
|
defaultValue = defGetString(defel);
|
|
else if (strcasecmp(defel->defname, "passedbyvalue") == 0)
|
|
byValue = true;
|
|
else if (strcasecmp(defel->defname, "alignment") == 0)
|
|
{
|
|
char *a = defGetString(defel);
|
|
|
|
/*
|
|
* Note: if argument was an unquoted identifier, parser will
|
|
* have applied translations to it, so be prepared to
|
|
* recognize translated type names as well as the nominal
|
|
* form.
|
|
*/
|
|
if (strcasecmp(a, "double") == 0 ||
|
|
strcasecmp(a, "float8") == 0 ||
|
|
strcasecmp(a, "pg_catalog.float8") == 0)
|
|
alignment = 'd';
|
|
else if (strcasecmp(a, "int4") == 0 ||
|
|
strcasecmp(a, "pg_catalog.int4") == 0)
|
|
alignment = 'i';
|
|
else if (strcasecmp(a, "int2") == 0 ||
|
|
strcasecmp(a, "pg_catalog.int2") == 0)
|
|
alignment = 's';
|
|
else if (strcasecmp(a, "char") == 0 ||
|
|
strcasecmp(a, "pg_catalog.bpchar") == 0)
|
|
alignment = 'c';
|
|
else
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_PARAMETER_VALUE),
|
|
errmsg("alignment \"%s\" not recognized", a)));
|
|
}
|
|
else if (strcasecmp(defel->defname, "storage") == 0)
|
|
{
|
|
char *a = defGetString(defel);
|
|
|
|
if (strcasecmp(a, "plain") == 0)
|
|
storage = 'p';
|
|
else if (strcasecmp(a, "external") == 0)
|
|
storage = 'e';
|
|
else if (strcasecmp(a, "extended") == 0)
|
|
storage = 'x';
|
|
else if (strcasecmp(a, "main") == 0)
|
|
storage = 'm';
|
|
else
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_PARAMETER_VALUE),
|
|
errmsg("storage \"%s\" not recognized", a)));
|
|
}
|
|
else
|
|
ereport(WARNING,
|
|
(errcode(ERRCODE_SYNTAX_ERROR),
|
|
errmsg("type attribute \"%s\" not recognized",
|
|
defel->defname)));
|
|
}
|
|
|
|
/*
|
|
* make sure we have our required definitions
|
|
*/
|
|
if (inputName == NIL)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
|
|
errmsg("type input function must be specified")));
|
|
if (outputName == NIL)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
|
|
errmsg("type output function must be specified")));
|
|
|
|
/*
|
|
* Look to see if type already exists (presumably as a shell; if not,
|
|
* TypeCreate will complain). If it doesn't, create it as a shell, so
|
|
* that the OID is known for use in the I/O function definitions.
|
|
*/
|
|
typoid = GetSysCacheOid(TYPENAMENSP,
|
|
CStringGetDatum(typeName),
|
|
ObjectIdGetDatum(typeNamespace),
|
|
0, 0);
|
|
if (!OidIsValid(typoid))
|
|
{
|
|
typoid = TypeShellMake(typeName, typeNamespace);
|
|
/* Make new shell type visible for modification below */
|
|
CommandCounterIncrement();
|
|
}
|
|
|
|
/*
|
|
* Convert I/O proc names to OIDs
|
|
*/
|
|
inputOid = findTypeInputFunction(inputName, typoid);
|
|
outputOid = findTypeOutputFunction(outputName, typoid);
|
|
if (receiveName)
|
|
receiveOid = findTypeReceiveFunction(receiveName, typoid);
|
|
if (sendName)
|
|
sendOid = findTypeSendFunction(sendName, typoid);
|
|
|
|
/*
|
|
* Verify that I/O procs return the expected thing. If we see OPAQUE,
|
|
* complain and change it to the correct type-safe choice.
|
|
*/
|
|
resulttype = get_func_rettype(inputOid);
|
|
if (resulttype != typoid)
|
|
{
|
|
if (resulttype == OPAQUEOID)
|
|
{
|
|
/* backwards-compatibility hack */
|
|
ereport(WARNING,
|
|
(errmsg("changing return type of function %s from \"opaque\" to %s",
|
|
NameListToString(inputName), typeName)));
|
|
SetFunctionReturnType(inputOid, typoid);
|
|
}
|
|
else
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
|
|
errmsg("type input function %s must return type %s",
|
|
NameListToString(inputName), typeName)));
|
|
}
|
|
resulttype = get_func_rettype(outputOid);
|
|
if (resulttype != CSTRINGOID)
|
|
{
|
|
if (resulttype == OPAQUEOID)
|
|
{
|
|
/* backwards-compatibility hack */
|
|
ereport(WARNING,
|
|
(errmsg("changing return type of function %s from \"opaque\" to \"cstring\"",
|
|
NameListToString(outputName))));
|
|
SetFunctionReturnType(outputOid, CSTRINGOID);
|
|
}
|
|
else
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
|
|
errmsg("type output function %s must return type \"cstring\"",
|
|
NameListToString(outputName))));
|
|
}
|
|
if (receiveOid)
|
|
{
|
|
resulttype = get_func_rettype(receiveOid);
|
|
if (resulttype != typoid)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
|
|
errmsg("type receive function %s must return type %s",
|
|
NameListToString(receiveName), typeName)));
|
|
}
|
|
if (sendOid)
|
|
{
|
|
resulttype = get_func_rettype(sendOid);
|
|
if (resulttype != BYTEAOID)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
|
|
errmsg("type send function %s must return type \"bytea\"",
|
|
NameListToString(sendName))));
|
|
}
|
|
|
|
/*
|
|
* Convert analysis function proc name to an OID. If no analysis function
|
|
* is specified, we'll use zero to select the built-in default algorithm.
|
|
*/
|
|
if (analyzeName)
|
|
analyzeOid = findTypeAnalyzeFunction(analyzeName, typoid);
|
|
|
|
/*
|
|
* now have TypeCreate do all the real work.
|
|
*/
|
|
typoid =
|
|
TypeCreate(typeName, /* type name */
|
|
typeNamespace, /* namespace */
|
|
InvalidOid, /* preassigned type oid (not done here) */
|
|
InvalidOid, /* relation oid (n/a here) */
|
|
0, /* relation kind (ditto) */
|
|
internalLength, /* internal size */
|
|
'b', /* type-type (base type) */
|
|
delimiter, /* array element delimiter */
|
|
inputOid, /* input procedure */
|
|
outputOid, /* output procedure */
|
|
receiveOid, /* receive procedure */
|
|
sendOid, /* send procedure */
|
|
analyzeOid, /* analyze procedure */
|
|
elemType, /* element type ID */
|
|
InvalidOid, /* base type ID (only for domains) */
|
|
defaultValue, /* default type value */
|
|
NULL, /* no binary form available */
|
|
byValue, /* passed by value */
|
|
alignment, /* required alignment */
|
|
storage, /* TOAST strategy */
|
|
-1, /* typMod (Domains only) */
|
|
0, /* Array Dimensions of typbasetype */
|
|
false); /* Type NOT NULL */
|
|
|
|
/*
|
|
* When we create a base type (as opposed to a complex type) we need
|
|
* to have an array entry for it in pg_type as well.
|
|
*/
|
|
shadow_type = makeArrayTypeName(typeName);
|
|
|
|
/* alignment must be 'i' or 'd' for arrays */
|
|
alignment = (alignment == 'd') ? 'd' : 'i';
|
|
|
|
TypeCreate(shadow_type, /* type name */
|
|
typeNamespace, /* namespace */
|
|
InvalidOid, /* preassigned type oid (not done here) */
|
|
InvalidOid, /* relation oid (n/a here) */
|
|
0, /* relation kind (ditto) */
|
|
-1, /* internal size */
|
|
'b', /* type-type (base type) */
|
|
DEFAULT_TYPDELIM, /* array element delimiter */
|
|
F_ARRAY_IN, /* input procedure */
|
|
F_ARRAY_OUT, /* output procedure */
|
|
F_ARRAY_RECV, /* receive procedure */
|
|
F_ARRAY_SEND, /* send procedure */
|
|
InvalidOid, /* analyze procedure - default */
|
|
typoid, /* element type ID */
|
|
InvalidOid, /* base type ID */
|
|
NULL, /* never a default type value */
|
|
NULL, /* binary default isn't sent either */
|
|
false, /* never passed by value */
|
|
alignment, /* see above */
|
|
'x', /* ARRAY is always toastable */
|
|
-1, /* typMod (Domains only) */
|
|
0, /* Array dimensions of typbasetype */
|
|
false); /* Type NOT NULL */
|
|
|
|
pfree(shadow_type);
|
|
}
|
|
|
|
|
|
/*
|
|
* RemoveType
|
|
* Removes a datatype.
|
|
*/
|
|
void
|
|
RemoveType(List *names, DropBehavior behavior)
|
|
{
|
|
TypeName *typename;
|
|
Oid typeoid;
|
|
HeapTuple tup;
|
|
ObjectAddress object;
|
|
|
|
/* Make a TypeName so we can use standard type lookup machinery */
|
|
typename = makeNode(TypeName);
|
|
typename->names = names;
|
|
typename->typmod = -1;
|
|
typename->arrayBounds = NIL;
|
|
|
|
/* Use LookupTypeName here so that shell types can be removed. */
|
|
typeoid = LookupTypeName(typename);
|
|
if (!OidIsValid(typeoid))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_UNDEFINED_OBJECT),
|
|
errmsg("type \"%s\" does not exist",
|
|
TypeNameToString(typename))));
|
|
|
|
tup = SearchSysCache(TYPEOID,
|
|
ObjectIdGetDatum(typeoid),
|
|
0, 0, 0);
|
|
if (!HeapTupleIsValid(tup))
|
|
elog(ERROR, "cache lookup failed for type %u", typeoid);
|
|
|
|
/* Permission check: must own type or its namespace */
|
|
if (!pg_type_ownercheck(typeoid, GetUserId()) &&
|
|
!pg_namespace_ownercheck(((Form_pg_type) GETSTRUCT(tup))->typnamespace,
|
|
GetUserId()))
|
|
aclcheck_error(ACLCHECK_NOT_OWNER, ACL_KIND_TYPE,
|
|
TypeNameToString(typename));
|
|
|
|
ReleaseSysCache(tup);
|
|
|
|
/*
|
|
* Do the deletion
|
|
*/
|
|
object.classId = RelOid_pg_type;
|
|
object.objectId = typeoid;
|
|
object.objectSubId = 0;
|
|
|
|
performDeletion(&object, behavior);
|
|
}
|
|
|
|
|
|
/*
|
|
* Guts of type deletion.
|
|
*/
|
|
void
|
|
RemoveTypeById(Oid typeOid)
|
|
{
|
|
Relation relation;
|
|
HeapTuple tup;
|
|
|
|
relation = heap_openr(TypeRelationName, RowExclusiveLock);
|
|
|
|
tup = SearchSysCache(TYPEOID,
|
|
ObjectIdGetDatum(typeOid),
|
|
0, 0, 0);
|
|
if (!HeapTupleIsValid(tup))
|
|
elog(ERROR, "cache lookup failed for type %u", typeOid);
|
|
|
|
simple_heap_delete(relation, &tup->t_self);
|
|
|
|
ReleaseSysCache(tup);
|
|
|
|
heap_close(relation, RowExclusiveLock);
|
|
}
|
|
|
|
|
|
/*
|
|
* DefineDomain
|
|
* Registers a new domain.
|
|
*/
|
|
void
|
|
DefineDomain(CreateDomainStmt *stmt)
|
|
{
|
|
char *domainName;
|
|
Oid domainNamespace;
|
|
AclResult aclresult;
|
|
int16 internalLength;
|
|
Oid inputProcedure;
|
|
Oid outputProcedure;
|
|
Oid receiveProcedure;
|
|
Oid sendProcedure;
|
|
Oid analyzeProcedure;
|
|
bool byValue;
|
|
char delimiter;
|
|
char alignment;
|
|
char storage;
|
|
char typtype;
|
|
Datum datum;
|
|
bool isnull;
|
|
Node *defaultExpr = NULL;
|
|
char *defaultValue = NULL;
|
|
char *defaultValueBin = NULL;
|
|
bool typNotNull = false;
|
|
bool nullDefined = false;
|
|
Oid basetypelem;
|
|
int32 typNDims = length(stmt->typename->arrayBounds);
|
|
HeapTuple typeTup;
|
|
List *schema = stmt->constraints;
|
|
List *listptr;
|
|
Oid basetypeoid;
|
|
Oid domainoid;
|
|
Form_pg_type baseType;
|
|
int counter = 0;
|
|
|
|
/* Convert list of names to a name and namespace */
|
|
domainNamespace = QualifiedNameGetCreationNamespace(stmt->domainname,
|
|
&domainName);
|
|
|
|
/* Check we have creation rights in target namespace */
|
|
aclresult = pg_namespace_aclcheck(domainNamespace, GetUserId(),
|
|
ACL_CREATE);
|
|
if (aclresult != ACLCHECK_OK)
|
|
aclcheck_error(aclresult, ACL_KIND_NAMESPACE,
|
|
get_namespace_name(domainNamespace));
|
|
|
|
/*
|
|
* Domainnames, unlike typenames don't need to account for the '_'
|
|
* prefix. So they can be one character longer. (This test is
|
|
* presently useless since the parser will have truncated the name to
|
|
* fit. But leave it here since we may someday support arrays of
|
|
* domains, in which case we'll be back to needing to enforce
|
|
* NAMEDATALEN-2.)
|
|
*/
|
|
if (strlen(domainName) > (NAMEDATALEN - 1))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_NAME),
|
|
errmsg("domain names must be %d characters or less",
|
|
NAMEDATALEN - 1)));
|
|
|
|
/*
|
|
* Look up the base type.
|
|
*/
|
|
typeTup = typenameType(stmt->typename);
|
|
|
|
baseType = (Form_pg_type) GETSTRUCT(typeTup);
|
|
basetypeoid = HeapTupleGetOid(typeTup);
|
|
|
|
/*
|
|
* Base type must be a plain base type. Domains over pseudo types
|
|
* would create a security hole. Domains of domains might be made to
|
|
* work in the future, but not today. Ditto for domains over complex
|
|
* types.
|
|
*/
|
|
typtype = baseType->typtype;
|
|
if (typtype != 'b')
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_DATATYPE_MISMATCH),
|
|
errmsg("\"%s\" is not a valid base type for a domain",
|
|
TypeNameToString(stmt->typename))));
|
|
|
|
/* passed by value */
|
|
byValue = baseType->typbyval;
|
|
|
|
/* Required Alignment */
|
|
alignment = baseType->typalign;
|
|
|
|
/* TOAST Strategy */
|
|
storage = baseType->typstorage;
|
|
|
|
/* Storage Length */
|
|
internalLength = baseType->typlen;
|
|
|
|
/* Array element Delimiter */
|
|
delimiter = baseType->typdelim;
|
|
|
|
/* I/O Functions */
|
|
inputProcedure = baseType->typinput;
|
|
outputProcedure = baseType->typoutput;
|
|
receiveProcedure = baseType->typreceive;
|
|
sendProcedure = baseType->typsend;
|
|
|
|
/* Analysis function */
|
|
analyzeProcedure = baseType->typanalyze;
|
|
|
|
/* Inherited default value */
|
|
datum = SysCacheGetAttr(TYPEOID, typeTup,
|
|
Anum_pg_type_typdefault, &isnull);
|
|
if (!isnull)
|
|
defaultValue = DatumGetCString(DirectFunctionCall1(textout, datum));
|
|
|
|
/* Inherited default binary value */
|
|
datum = SysCacheGetAttr(TYPEOID, typeTup,
|
|
Anum_pg_type_typdefaultbin, &isnull);
|
|
if (!isnull)
|
|
defaultValueBin = DatumGetCString(DirectFunctionCall1(textout, datum));
|
|
|
|
/*
|
|
* Pull out the typelem name of the parent OID.
|
|
*
|
|
* This is what enables us to make a domain of an array
|
|
*/
|
|
basetypelem = baseType->typelem;
|
|
|
|
/*
|
|
* Run through constraints manually to avoid the additional processing
|
|
* conducted by DefineRelation() and friends.
|
|
*/
|
|
foreach(listptr, schema)
|
|
{
|
|
Node *newConstraint = lfirst(listptr);
|
|
Constraint *constr;
|
|
ParseState *pstate;
|
|
|
|
/* Check for unsupported constraint types */
|
|
if (IsA(newConstraint, FkConstraint))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_SYNTAX_ERROR),
|
|
errmsg("foreign key constraints not possible for domains")));
|
|
|
|
/* otherwise it should be a plain Constraint */
|
|
if (!IsA(newConstraint, Constraint))
|
|
elog(ERROR, "unrecognized node type: %d",
|
|
(int) nodeTag(newConstraint));
|
|
|
|
constr = (Constraint *) newConstraint;
|
|
|
|
switch (constr->contype)
|
|
{
|
|
case CONSTR_DEFAULT:
|
|
|
|
/*
|
|
* The inherited default value may be overridden by the
|
|
* user with the DEFAULT <expr> statement.
|
|
*/
|
|
if (defaultExpr)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_SYNTAX_ERROR),
|
|
errmsg("multiple default expressions")));
|
|
|
|
/* Create a dummy ParseState for transformExpr */
|
|
pstate = make_parsestate(NULL);
|
|
|
|
/*
|
|
* Cook the constr->raw_expr into an expression. Note:
|
|
* Name is strictly for error message
|
|
*/
|
|
defaultExpr = cookDefault(pstate, constr->raw_expr,
|
|
basetypeoid,
|
|
stmt->typename->typmod,
|
|
domainName);
|
|
|
|
/*
|
|
* Expression must be stored as a nodeToString result, but
|
|
* we also require a valid textual representation (mainly
|
|
* to make life easier for pg_dump).
|
|
*/
|
|
defaultValue = deparse_expression(defaultExpr,
|
|
deparse_context_for(domainName,
|
|
InvalidOid),
|
|
false, false);
|
|
defaultValueBin = nodeToString(defaultExpr);
|
|
break;
|
|
|
|
case CONSTR_NOTNULL:
|
|
if (nullDefined && !typNotNull)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_SYNTAX_ERROR),
|
|
errmsg("conflicting NULL/NOT NULL constraints")));
|
|
typNotNull = true;
|
|
nullDefined = true;
|
|
break;
|
|
|
|
case CONSTR_NULL:
|
|
if (nullDefined && typNotNull)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_SYNTAX_ERROR),
|
|
errmsg("conflicting NULL/NOT NULL constraints")));
|
|
typNotNull = false;
|
|
nullDefined = true;
|
|
break;
|
|
|
|
case CONSTR_CHECK:
|
|
|
|
/*
|
|
* Check constraints are handled after domain creation, as
|
|
* they require the Oid of the domain
|
|
*/
|
|
break;
|
|
|
|
/*
|
|
* All else are error cases
|
|
*/
|
|
case CONSTR_UNIQUE:
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_SYNTAX_ERROR),
|
|
errmsg("unique constraints not possible for domains")));
|
|
break;
|
|
|
|
case CONSTR_PRIMARY:
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_SYNTAX_ERROR),
|
|
errmsg("primary key constraints not possible for domains")));
|
|
break;
|
|
|
|
case CONSTR_ATTR_DEFERRABLE:
|
|
case CONSTR_ATTR_NOT_DEFERRABLE:
|
|
case CONSTR_ATTR_DEFERRED:
|
|
case CONSTR_ATTR_IMMEDIATE:
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
|
|
errmsg("specifying constraint deferrability not supported for domains")));
|
|
break;
|
|
|
|
default:
|
|
elog(ERROR, "unrecognized constraint subtype: %d",
|
|
(int) constr->contype);
|
|
break;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Have TypeCreate do all the real work.
|
|
*/
|
|
domainoid =
|
|
TypeCreate(domainName, /* type name */
|
|
domainNamespace, /* namespace */
|
|
InvalidOid, /* preassigned type oid (none here) */
|
|
InvalidOid, /* relation oid (n/a here) */
|
|
0, /* relation kind (ditto) */
|
|
internalLength, /* internal size */
|
|
'd', /* type-type (domain type) */
|
|
delimiter, /* array element delimiter */
|
|
inputProcedure, /* input procedure */
|
|
outputProcedure, /* output procedure */
|
|
receiveProcedure, /* receive procedure */
|
|
sendProcedure, /* send procedure */
|
|
analyzeProcedure, /* analyze procedure */
|
|
basetypelem, /* element type ID */
|
|
basetypeoid, /* base type ID */
|
|
defaultValue, /* default type value (text) */
|
|
defaultValueBin, /* default type value (binary) */
|
|
byValue, /* passed by value */
|
|
alignment, /* required alignment */
|
|
storage, /* TOAST strategy */
|
|
stmt->typename->typmod, /* typeMod value */
|
|
typNDims, /* Array dimensions for base type */
|
|
typNotNull); /* Type NOT NULL */
|
|
|
|
/*
|
|
* Process constraints which refer to the domain ID returned by
|
|
* TypeCreate
|
|
*/
|
|
foreach(listptr, schema)
|
|
{
|
|
Constraint *constr = lfirst(listptr);
|
|
|
|
/* it must be a Constraint, per check above */
|
|
|
|
switch (constr->contype)
|
|
{
|
|
case CONSTR_CHECK:
|
|
domainAddConstraint(domainoid, domainNamespace,
|
|
basetypeoid, stmt->typename->typmod,
|
|
constr, &counter, domainName);
|
|
break;
|
|
|
|
/* Other constraint types were fully processed above */
|
|
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Now we can clean up.
|
|
*/
|
|
ReleaseSysCache(typeTup);
|
|
}
|
|
|
|
|
|
/*
|
|
* RemoveDomain
|
|
* Removes a domain.
|
|
*
|
|
* This is identical to RemoveType except we insist it be a domain.
|
|
*/
|
|
void
|
|
RemoveDomain(List *names, DropBehavior behavior)
|
|
{
|
|
TypeName *typename;
|
|
Oid typeoid;
|
|
HeapTuple tup;
|
|
char typtype;
|
|
ObjectAddress object;
|
|
|
|
/* Make a TypeName so we can use standard type lookup machinery */
|
|
typename = makeNode(TypeName);
|
|
typename->names = names;
|
|
typename->typmod = -1;
|
|
typename->arrayBounds = NIL;
|
|
|
|
/* Use LookupTypeName here so that shell types can be removed. */
|
|
typeoid = LookupTypeName(typename);
|
|
if (!OidIsValid(typeoid))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_UNDEFINED_OBJECT),
|
|
errmsg("type \"%s\" does not exist",
|
|
TypeNameToString(typename))));
|
|
|
|
tup = SearchSysCache(TYPEOID,
|
|
ObjectIdGetDatum(typeoid),
|
|
0, 0, 0);
|
|
if (!HeapTupleIsValid(tup))
|
|
elog(ERROR, "cache lookup failed for type %u", typeoid);
|
|
|
|
/* Permission check: must own type or its namespace */
|
|
if (!pg_type_ownercheck(typeoid, GetUserId()) &&
|
|
!pg_namespace_ownercheck(((Form_pg_type) GETSTRUCT(tup))->typnamespace,
|
|
GetUserId()))
|
|
aclcheck_error(ACLCHECK_NOT_OWNER, ACL_KIND_TYPE,
|
|
TypeNameToString(typename));
|
|
|
|
/* Check that this is actually a domain */
|
|
typtype = ((Form_pg_type) GETSTRUCT(tup))->typtype;
|
|
|
|
if (typtype != 'd')
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_WRONG_OBJECT_TYPE),
|
|
errmsg("\"%s\" is not a domain",
|
|
TypeNameToString(typename))));
|
|
|
|
ReleaseSysCache(tup);
|
|
|
|
/*
|
|
* Do the deletion
|
|
*/
|
|
object.classId = RelOid_pg_type;
|
|
object.objectId = typeoid;
|
|
object.objectSubId = 0;
|
|
|
|
performDeletion(&object, behavior);
|
|
}
|
|
|
|
|
|
/*
|
|
* Find suitable I/O functions for a type.
|
|
*
|
|
* typeOid is the type's OID (which will already exist, if only as a shell
|
|
* type).
|
|
*/
|
|
|
|
static Oid
|
|
findTypeInputFunction(List *procname, Oid typeOid)
|
|
{
|
|
Oid argList[FUNC_MAX_ARGS];
|
|
Oid procOid;
|
|
|
|
/*
|
|
* Input functions can take a single argument of type CSTRING, or
|
|
* three arguments (string, element OID, typmod).
|
|
*
|
|
* For backwards compatibility we allow OPAQUE in place of CSTRING; if we
|
|
* see this, we issue a warning and fix up the pg_proc entry.
|
|
*/
|
|
MemSet(argList, 0, FUNC_MAX_ARGS * sizeof(Oid));
|
|
|
|
argList[0] = CSTRINGOID;
|
|
|
|
procOid = LookupFuncName(procname, 1, argList, true);
|
|
if (OidIsValid(procOid))
|
|
return procOid;
|
|
|
|
argList[1] = OIDOID;
|
|
argList[2] = INT4OID;
|
|
|
|
procOid = LookupFuncName(procname, 3, argList, true);
|
|
if (OidIsValid(procOid))
|
|
return procOid;
|
|
|
|
/* No luck, try it with OPAQUE */
|
|
MemSet(argList, 0, FUNC_MAX_ARGS * sizeof(Oid));
|
|
|
|
argList[0] = OPAQUEOID;
|
|
|
|
procOid = LookupFuncName(procname, 1, argList, true);
|
|
|
|
if (!OidIsValid(procOid))
|
|
{
|
|
argList[1] = OIDOID;
|
|
argList[2] = INT4OID;
|
|
|
|
procOid = LookupFuncName(procname, 3, argList, true);
|
|
}
|
|
|
|
if (OidIsValid(procOid))
|
|
{
|
|
/* Found, but must complain and fix the pg_proc entry */
|
|
ereport(WARNING,
|
|
(errmsg("changing argument type of function %s from \"opaque\" to \"cstring\"",
|
|
NameListToString(procname))));
|
|
SetFunctionArgType(procOid, 0, CSTRINGOID);
|
|
|
|
/*
|
|
* Need CommandCounterIncrement since DefineType will likely try
|
|
* to alter the pg_proc tuple again.
|
|
*/
|
|
CommandCounterIncrement();
|
|
|
|
return procOid;
|
|
}
|
|
|
|
/* Use CSTRING (preferred) in the error message */
|
|
argList[0] = CSTRINGOID;
|
|
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_UNDEFINED_FUNCTION),
|
|
errmsg("function %s does not exist",
|
|
func_signature_string(procname, 1, argList))));
|
|
|
|
return InvalidOid; /* keep compiler quiet */
|
|
}
|
|
|
|
static Oid
|
|
findTypeOutputFunction(List *procname, Oid typeOid)
|
|
{
|
|
Oid argList[FUNC_MAX_ARGS];
|
|
Oid procOid;
|
|
|
|
/*
|
|
* Output functions can take a single argument of the type, or two
|
|
* arguments (data value, element OID).
|
|
*
|
|
* For backwards compatibility we allow OPAQUE in place of the actual
|
|
* type name; if we see this, we issue a warning and fix up the pg_proc
|
|
* entry.
|
|
*/
|
|
MemSet(argList, 0, FUNC_MAX_ARGS * sizeof(Oid));
|
|
|
|
argList[0] = typeOid;
|
|
|
|
procOid = LookupFuncName(procname, 1, argList, true);
|
|
if (OidIsValid(procOid))
|
|
return procOid;
|
|
|
|
argList[1] = OIDOID;
|
|
|
|
procOid = LookupFuncName(procname, 2, argList, true);
|
|
if (OidIsValid(procOid))
|
|
return procOid;
|
|
|
|
/* No luck, try it with OPAQUE */
|
|
MemSet(argList, 0, FUNC_MAX_ARGS * sizeof(Oid));
|
|
|
|
argList[0] = OPAQUEOID;
|
|
|
|
procOid = LookupFuncName(procname, 1, argList, true);
|
|
|
|
if (!OidIsValid(procOid))
|
|
{
|
|
argList[1] = OIDOID;
|
|
|
|
procOid = LookupFuncName(procname, 2, argList, true);
|
|
}
|
|
|
|
if (OidIsValid(procOid))
|
|
{
|
|
/* Found, but must complain and fix the pg_proc entry */
|
|
ereport(WARNING,
|
|
(errmsg("changing argument type of function %s from \"opaque\" to %s",
|
|
NameListToString(procname), format_type_be(typeOid))));
|
|
SetFunctionArgType(procOid, 0, typeOid);
|
|
|
|
/*
|
|
* Need CommandCounterIncrement since DefineType will likely try
|
|
* to alter the pg_proc tuple again.
|
|
*/
|
|
CommandCounterIncrement();
|
|
|
|
return procOid;
|
|
}
|
|
|
|
/* Use type name, not OPAQUE, in the failure message. */
|
|
argList[0] = typeOid;
|
|
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_UNDEFINED_FUNCTION),
|
|
errmsg("function %s does not exist",
|
|
func_signature_string(procname, 1, argList))));
|
|
|
|
return InvalidOid; /* keep compiler quiet */
|
|
}
|
|
|
|
static Oid
|
|
findTypeReceiveFunction(List *procname, Oid typeOid)
|
|
{
|
|
Oid argList[FUNC_MAX_ARGS];
|
|
Oid procOid;
|
|
|
|
/*
|
|
* Receive functions can take a single argument of type INTERNAL, or
|
|
* two arguments (internal, oid).
|
|
*/
|
|
MemSet(argList, 0, FUNC_MAX_ARGS * sizeof(Oid));
|
|
|
|
argList[0] = INTERNALOID;
|
|
|
|
procOid = LookupFuncName(procname, 1, argList, true);
|
|
if (OidIsValid(procOid))
|
|
return procOid;
|
|
|
|
argList[1] = OIDOID;
|
|
|
|
procOid = LookupFuncName(procname, 2, argList, true);
|
|
if (OidIsValid(procOid))
|
|
return procOid;
|
|
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_UNDEFINED_FUNCTION),
|
|
errmsg("function %s does not exist",
|
|
func_signature_string(procname, 1, argList))));
|
|
|
|
return InvalidOid; /* keep compiler quiet */
|
|
}
|
|
|
|
static Oid
|
|
findTypeSendFunction(List *procname, Oid typeOid)
|
|
{
|
|
Oid argList[FUNC_MAX_ARGS];
|
|
Oid procOid;
|
|
|
|
/*
|
|
* Send functions can take a single argument of the type, or two
|
|
* arguments (data value, element OID).
|
|
*/
|
|
MemSet(argList, 0, FUNC_MAX_ARGS * sizeof(Oid));
|
|
|
|
argList[0] = typeOid;
|
|
|
|
procOid = LookupFuncName(procname, 1, argList, true);
|
|
if (OidIsValid(procOid))
|
|
return procOid;
|
|
|
|
argList[1] = OIDOID;
|
|
|
|
procOid = LookupFuncName(procname, 2, argList, true);
|
|
if (OidIsValid(procOid))
|
|
return procOid;
|
|
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_UNDEFINED_FUNCTION),
|
|
errmsg("function %s does not exist",
|
|
func_signature_string(procname, 1, argList))));
|
|
|
|
return InvalidOid; /* keep compiler quiet */
|
|
}
|
|
|
|
static Oid
|
|
findTypeAnalyzeFunction(List *procname, Oid typeOid)
|
|
{
|
|
Oid argList[FUNC_MAX_ARGS];
|
|
Oid procOid;
|
|
|
|
/*
|
|
* Analyze functions always take one INTERNAL argument and return bool.
|
|
*/
|
|
MemSet(argList, 0, FUNC_MAX_ARGS * sizeof(Oid));
|
|
|
|
argList[0] = INTERNALOID;
|
|
|
|
procOid = LookupFuncName(procname, 1, argList, true);
|
|
if (!OidIsValid(procOid))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_UNDEFINED_FUNCTION),
|
|
errmsg("function %s does not exist",
|
|
func_signature_string(procname, 1, argList))));
|
|
|
|
if (get_func_rettype(procOid) != BOOLOID)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
|
|
errmsg("type analyze function %s must return type \"boolean\"",
|
|
NameListToString(procname))));
|
|
|
|
return procOid;
|
|
}
|
|
|
|
|
|
/*-------------------------------------------------------------------
|
|
* DefineCompositeType
|
|
*
|
|
* Create a Composite Type relation.
|
|
* `DefineRelation' does all the work, we just provide the correct
|
|
* arguments!
|
|
*
|
|
* If the relation already exists, then 'DefineRelation' will abort
|
|
* the xact...
|
|
*
|
|
* DefineCompositeType returns relid for use when creating
|
|
* an implicit composite type during function creation
|
|
*-------------------------------------------------------------------
|
|
*/
|
|
Oid
|
|
DefineCompositeType(const RangeVar *typevar, List *coldeflist)
|
|
{
|
|
CreateStmt *createStmt = makeNode(CreateStmt);
|
|
|
|
if (coldeflist == NIL)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
|
|
errmsg("composite type must have at least one attribute")));
|
|
|
|
/*
|
|
* now create the parameters for keys/inheritance etc. All of them are
|
|
* nil...
|
|
*/
|
|
createStmt->relation = (RangeVar *) typevar;
|
|
createStmt->tableElts = coldeflist;
|
|
createStmt->inhRelations = NIL;
|
|
createStmt->constraints = NIL;
|
|
createStmt->hasoids = MUST_NOT_HAVE_OIDS;
|
|
createStmt->oncommit = ONCOMMIT_NOOP;
|
|
|
|
/*
|
|
* finally create the relation...
|
|
*/
|
|
return DefineRelation(createStmt, RELKIND_COMPOSITE_TYPE);
|
|
}
|
|
|
|
/*
|
|
* AlterDomainDefault
|
|
*
|
|
* Routine implementing ALTER DOMAIN SET/DROP DEFAULT statements.
|
|
*/
|
|
void
|
|
AlterDomainDefault(List *names, Node *defaultRaw)
|
|
{
|
|
TypeName *typename;
|
|
Oid domainoid;
|
|
HeapTuple tup;
|
|
ParseState *pstate;
|
|
Relation rel;
|
|
char *defaultValue;
|
|
Node *defaultExpr = NULL; /* NULL if no default specified */
|
|
Datum new_record[Natts_pg_type];
|
|
char new_record_nulls[Natts_pg_type];
|
|
char new_record_repl[Natts_pg_type];
|
|
HeapTuple newtuple;
|
|
Form_pg_type typTup;
|
|
|
|
/* Make a TypeName so we can use standard type lookup machinery */
|
|
typename = makeNode(TypeName);
|
|
typename->names = names;
|
|
typename->typmod = -1;
|
|
typename->arrayBounds = NIL;
|
|
|
|
/* Lock the domain in the type table */
|
|
rel = heap_openr(TypeRelationName, RowExclusiveLock);
|
|
|
|
/* Use LookupTypeName here so that shell types can be removed. */
|
|
domainoid = LookupTypeName(typename);
|
|
if (!OidIsValid(domainoid))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_UNDEFINED_OBJECT),
|
|
errmsg("type \"%s\" does not exist",
|
|
TypeNameToString(typename))));
|
|
|
|
tup = SearchSysCacheCopy(TYPEOID,
|
|
ObjectIdGetDatum(domainoid),
|
|
0, 0, 0);
|
|
if (!HeapTupleIsValid(tup))
|
|
elog(ERROR, "cache lookup failed for type %u", domainoid);
|
|
|
|
/* Doesn't return if user isn't allowed to alter the domain */
|
|
domainOwnerCheck(tup, typename);
|
|
|
|
/* Setup new tuple */
|
|
MemSet(new_record, (Datum) 0, sizeof(new_record));
|
|
MemSet(new_record_nulls, ' ', sizeof(new_record_nulls));
|
|
MemSet(new_record_repl, ' ', sizeof(new_record_repl));
|
|
|
|
/* Useful later */
|
|
typTup = (Form_pg_type) GETSTRUCT(tup);
|
|
|
|
/* Store the new default, if null then skip this step */
|
|
if (defaultRaw)
|
|
{
|
|
/* Create a dummy ParseState for transformExpr */
|
|
pstate = make_parsestate(NULL);
|
|
|
|
/*
|
|
* Cook the colDef->raw_expr into an expression. Note: Name is
|
|
* strictly for error message
|
|
*/
|
|
defaultExpr = cookDefault(pstate, defaultRaw,
|
|
typTup->typbasetype,
|
|
typTup->typtypmod,
|
|
NameStr(typTup->typname));
|
|
|
|
/*
|
|
* Expression must be stored as a nodeToString result, but we also
|
|
* require a valid textual representation (mainly to make life
|
|
* easier for pg_dump).
|
|
*/
|
|
defaultValue = deparse_expression(defaultExpr,
|
|
deparse_context_for(NameStr(typTup->typname),
|
|
InvalidOid),
|
|
false, false);
|
|
|
|
/*
|
|
* Form an updated tuple with the new default and write it back.
|
|
*/
|
|
new_record[Anum_pg_type_typdefaultbin - 1] = DirectFunctionCall1(textin,
|
|
CStringGetDatum(
|
|
nodeToString(defaultExpr)));
|
|
|
|
new_record_repl[Anum_pg_type_typdefaultbin - 1] = 'r';
|
|
new_record[Anum_pg_type_typdefault - 1] = DirectFunctionCall1(textin,
|
|
CStringGetDatum(defaultValue));
|
|
new_record_repl[Anum_pg_type_typdefault - 1] = 'r';
|
|
}
|
|
else
|
|
/* Default is NULL, drop it */
|
|
{
|
|
new_record_nulls[Anum_pg_type_typdefaultbin - 1] = 'n';
|
|
new_record_repl[Anum_pg_type_typdefaultbin - 1] = 'r';
|
|
new_record_nulls[Anum_pg_type_typdefault - 1] = 'n';
|
|
new_record_repl[Anum_pg_type_typdefault - 1] = 'r';
|
|
}
|
|
|
|
newtuple = heap_modifytuple(tup, rel,
|
|
new_record, new_record_nulls, new_record_repl);
|
|
|
|
simple_heap_update(rel, &tup->t_self, newtuple);
|
|
|
|
CatalogUpdateIndexes(rel, newtuple);
|
|
|
|
/* Rebuild dependencies */
|
|
GenerateTypeDependencies(typTup->typnamespace,
|
|
domainoid,
|
|
typTup->typrelid,
|
|
0, /* relation kind is n/a */
|
|
typTup->typinput,
|
|
typTup->typoutput,
|
|
typTup->typreceive,
|
|
typTup->typsend,
|
|
typTup->typanalyze,
|
|
typTup->typelem,
|
|
typTup->typbasetype,
|
|
defaultExpr,
|
|
true); /* Rebuild is true */
|
|
|
|
/* Clean up */
|
|
heap_close(rel, NoLock);
|
|
heap_freetuple(newtuple);
|
|
}
|
|
|
|
/*
|
|
* AlterDomainNotNull
|
|
*
|
|
* Routine implementing ALTER DOMAIN SET/DROP NOT NULL statements.
|
|
*/
|
|
void
|
|
AlterDomainNotNull(List *names, bool notNull)
|
|
{
|
|
TypeName *typename;
|
|
Oid domainoid;
|
|
Relation typrel;
|
|
HeapTuple tup;
|
|
Form_pg_type typTup;
|
|
|
|
/* Make a TypeName so we can use standard type lookup machinery */
|
|
typename = makeNode(TypeName);
|
|
typename->names = names;
|
|
typename->typmod = -1;
|
|
typename->arrayBounds = NIL;
|
|
|
|
/* Lock the type table */
|
|
typrel = heap_openr(TypeRelationName, RowExclusiveLock);
|
|
|
|
/* Use LookupTypeName here so that shell types can be found (why?). */
|
|
domainoid = LookupTypeName(typename);
|
|
if (!OidIsValid(domainoid))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_UNDEFINED_OBJECT),
|
|
errmsg("type \"%s\" does not exist",
|
|
TypeNameToString(typename))));
|
|
|
|
tup = SearchSysCacheCopy(TYPEOID,
|
|
ObjectIdGetDatum(domainoid),
|
|
0, 0, 0);
|
|
if (!HeapTupleIsValid(tup))
|
|
elog(ERROR, "cache lookup failed for type %u", domainoid);
|
|
typTup = (Form_pg_type) GETSTRUCT(tup);
|
|
|
|
/* Doesn't return if user isn't allowed to alter the domain */
|
|
domainOwnerCheck(tup, typename);
|
|
|
|
/* Is the domain already set to the desired constraint? */
|
|
if (typTup->typnotnull == notNull)
|
|
{
|
|
heap_close(typrel, RowExclusiveLock);
|
|
return;
|
|
}
|
|
|
|
/* Adding a NOT NULL constraint requires checking existing columns */
|
|
if (notNull)
|
|
{
|
|
List *rels;
|
|
List *rt;
|
|
|
|
/* Fetch relation list with attributes based on this domain */
|
|
/* ShareLock is sufficient to prevent concurrent data changes */
|
|
|
|
rels = get_rels_with_domain(domainoid, ShareLock);
|
|
|
|
foreach(rt, rels)
|
|
{
|
|
RelToCheck *rtc = (RelToCheck *) lfirst(rt);
|
|
Relation testrel = rtc->rel;
|
|
TupleDesc tupdesc = RelationGetDescr(testrel);
|
|
HeapScanDesc scan;
|
|
HeapTuple tuple;
|
|
|
|
/* Scan all tuples in this relation */
|
|
scan = heap_beginscan(testrel, SnapshotNow, 0, NULL);
|
|
while ((tuple = heap_getnext(scan, ForwardScanDirection)) != NULL)
|
|
{
|
|
int i;
|
|
|
|
/* Test attributes that are of the domain */
|
|
for (i = 0; i < rtc->natts; i++)
|
|
{
|
|
int attnum = rtc->atts[i];
|
|
Datum d;
|
|
bool isNull;
|
|
|
|
d = heap_getattr(tuple, attnum, tupdesc, &isNull);
|
|
|
|
if (isNull)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_NOT_NULL_VIOLATION),
|
|
errmsg("column \"%s\" of table \"%s\" contains null values",
|
|
NameStr(tupdesc->attrs[attnum - 1]->attname),
|
|
RelationGetRelationName(testrel))));
|
|
}
|
|
}
|
|
heap_endscan(scan);
|
|
|
|
/* Close each rel after processing, but keep lock */
|
|
heap_close(testrel, NoLock);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Okay to update pg_type row. We can scribble on typTup because it's
|
|
* a copy.
|
|
*/
|
|
typTup->typnotnull = notNull;
|
|
|
|
simple_heap_update(typrel, &tup->t_self, tup);
|
|
|
|
CatalogUpdateIndexes(typrel, tup);
|
|
|
|
/* Clean up */
|
|
heap_freetuple(tup);
|
|
heap_close(typrel, RowExclusiveLock);
|
|
}
|
|
|
|
/*
|
|
* AlterDomainDropConstraint
|
|
*
|
|
* Implements the ALTER DOMAIN DROP CONSTRAINT statement
|
|
*/
|
|
void
|
|
AlterDomainDropConstraint(List *names, const char *constrName, DropBehavior behavior)
|
|
{
|
|
TypeName *typename;
|
|
Oid domainoid;
|
|
HeapTuple tup;
|
|
Relation rel;
|
|
Form_pg_type typTup;
|
|
Relation conrel;
|
|
SysScanDesc conscan;
|
|
ScanKeyData key[1];
|
|
HeapTuple contup;
|
|
|
|
/* Make a TypeName so we can use standard type lookup machinery */
|
|
typename = makeNode(TypeName);
|
|
typename->names = names;
|
|
typename->typmod = -1;
|
|
typename->arrayBounds = NIL;
|
|
|
|
/* Lock the type table */
|
|
rel = heap_openr(TypeRelationName, RowExclusiveLock);
|
|
|
|
/* Use LookupTypeName here so that shell types can be removed. */
|
|
domainoid = LookupTypeName(typename);
|
|
if (!OidIsValid(domainoid))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_UNDEFINED_OBJECT),
|
|
errmsg("type \"%s\" does not exist",
|
|
TypeNameToString(typename))));
|
|
|
|
tup = SearchSysCacheCopy(TYPEOID,
|
|
ObjectIdGetDatum(domainoid),
|
|
0, 0, 0);
|
|
if (!HeapTupleIsValid(tup))
|
|
elog(ERROR, "cache lookup failed for type %u", domainoid);
|
|
|
|
/* Doesn't return if user isn't allowed to alter the domain */
|
|
domainOwnerCheck(tup, typename);
|
|
|
|
/* Grab an appropriate lock on the pg_constraint relation */
|
|
conrel = heap_openr(ConstraintRelationName, RowExclusiveLock);
|
|
|
|
/* Use the index to scan only constraints of the target relation */
|
|
ScanKeyInit(&key[0],
|
|
Anum_pg_constraint_contypid,
|
|
BTEqualStrategyNumber, F_OIDEQ,
|
|
ObjectIdGetDatum(HeapTupleGetOid(tup)));
|
|
|
|
conscan = systable_beginscan(conrel, ConstraintTypidIndex, true,
|
|
SnapshotNow, 1, key);
|
|
|
|
typTup = (Form_pg_type) GETSTRUCT(tup);
|
|
|
|
/*
|
|
* Scan over the result set, removing any matching entries.
|
|
*/
|
|
while ((contup = systable_getnext(conscan)) != NULL)
|
|
{
|
|
Form_pg_constraint con = (Form_pg_constraint) GETSTRUCT(contup);
|
|
|
|
if (strcmp(NameStr(con->conname), constrName) == 0)
|
|
{
|
|
ObjectAddress conobj;
|
|
|
|
conobj.classId = RelationGetRelid(conrel);
|
|
conobj.objectId = HeapTupleGetOid(contup);
|
|
conobj.objectSubId = 0;
|
|
|
|
performDeletion(&conobj, behavior);
|
|
}
|
|
}
|
|
/* Clean up after the scan */
|
|
systable_endscan(conscan);
|
|
heap_close(conrel, RowExclusiveLock);
|
|
|
|
heap_close(rel, NoLock);
|
|
}
|
|
|
|
/*
|
|
* AlterDomainAddConstraint
|
|
*
|
|
* Implements the ALTER DOMAIN .. ADD CONSTRAINT statement.
|
|
*/
|
|
void
|
|
AlterDomainAddConstraint(List *names, Node *newConstraint)
|
|
{
|
|
TypeName *typename;
|
|
Oid domainoid;
|
|
Relation typrel;
|
|
HeapTuple tup;
|
|
Form_pg_type typTup;
|
|
List *rels;
|
|
List *rt;
|
|
EState *estate;
|
|
ExprContext *econtext;
|
|
char *ccbin;
|
|
Expr *expr;
|
|
ExprState *exprstate;
|
|
int counter = 0;
|
|
Constraint *constr;
|
|
|
|
/* Make a TypeName so we can use standard type lookup machinery */
|
|
typename = makeNode(TypeName);
|
|
typename->names = names;
|
|
typename->typmod = -1;
|
|
typename->arrayBounds = NIL;
|
|
|
|
/* Lock the type table */
|
|
typrel = heap_openr(TypeRelationName, RowExclusiveLock);
|
|
|
|
/* Use LookupTypeName here so that shell types can be found (why?). */
|
|
domainoid = LookupTypeName(typename);
|
|
if (!OidIsValid(domainoid))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_UNDEFINED_OBJECT),
|
|
errmsg("type \"%s\" does not exist",
|
|
TypeNameToString(typename))));
|
|
|
|
tup = SearchSysCacheCopy(TYPEOID,
|
|
ObjectIdGetDatum(domainoid),
|
|
0, 0, 0);
|
|
if (!HeapTupleIsValid(tup))
|
|
elog(ERROR, "cache lookup failed for type %u", domainoid);
|
|
typTup = (Form_pg_type) GETSTRUCT(tup);
|
|
|
|
/* Doesn't return if user isn't allowed to alter the domain */
|
|
domainOwnerCheck(tup, typename);
|
|
|
|
/* Check for unsupported constraint types */
|
|
if (IsA(newConstraint, FkConstraint))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_SYNTAX_ERROR),
|
|
errmsg("foreign key constraints not possible for domains")));
|
|
|
|
/* otherwise it should be a plain Constraint */
|
|
if (!IsA(newConstraint, Constraint))
|
|
elog(ERROR, "unrecognized node type: %d",
|
|
(int) nodeTag(newConstraint));
|
|
|
|
constr = (Constraint *) newConstraint;
|
|
|
|
switch (constr->contype)
|
|
{
|
|
case CONSTR_CHECK:
|
|
/* processed below */
|
|
break;
|
|
|
|
case CONSTR_UNIQUE:
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_SYNTAX_ERROR),
|
|
errmsg("unique constraints not possible for domains")));
|
|
break;
|
|
|
|
case CONSTR_PRIMARY:
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_SYNTAX_ERROR),
|
|
errmsg("primary key constraints not possible for domains")));
|
|
break;
|
|
|
|
case CONSTR_ATTR_DEFERRABLE:
|
|
case CONSTR_ATTR_NOT_DEFERRABLE:
|
|
case CONSTR_ATTR_DEFERRED:
|
|
case CONSTR_ATTR_IMMEDIATE:
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
|
|
errmsg("specifying constraint deferrability not supported for domains")));
|
|
break;
|
|
|
|
default:
|
|
elog(ERROR, "unrecognized constraint subtype: %d",
|
|
(int) constr->contype);
|
|
break;
|
|
}
|
|
|
|
/*
|
|
* Since all other constraint types throw errors, this must be a check
|
|
* constraint. First, process the constraint expression and add an
|
|
* entry to pg_constraint.
|
|
*/
|
|
|
|
ccbin = domainAddConstraint(HeapTupleGetOid(tup), typTup->typnamespace,
|
|
typTup->typbasetype, typTup->typtypmod,
|
|
constr, &counter, NameStr(typTup->typname));
|
|
|
|
/*
|
|
* Test all values stored in the attributes based on the domain the
|
|
* constraint is being added to.
|
|
*/
|
|
expr = (Expr *) stringToNode(ccbin);
|
|
|
|
/* Need an EState to run ExecEvalExpr */
|
|
estate = CreateExecutorState();
|
|
econtext = GetPerTupleExprContext(estate);
|
|
|
|
/* build execution state for expr */
|
|
exprstate = ExecPrepareExpr(expr, estate);
|
|
|
|
/* Fetch relation list with attributes based on this domain */
|
|
/* ShareLock is sufficient to prevent concurrent data changes */
|
|
|
|
rels = get_rels_with_domain(domainoid, ShareLock);
|
|
|
|
foreach(rt, rels)
|
|
{
|
|
RelToCheck *rtc = (RelToCheck *) lfirst(rt);
|
|
Relation testrel = rtc->rel;
|
|
TupleDesc tupdesc = RelationGetDescr(testrel);
|
|
HeapScanDesc scan;
|
|
HeapTuple tuple;
|
|
|
|
/* Scan all tuples in this relation */
|
|
scan = heap_beginscan(testrel, SnapshotNow, 0, NULL);
|
|
while ((tuple = heap_getnext(scan, ForwardScanDirection)) != NULL)
|
|
{
|
|
int i;
|
|
|
|
/* Test attributes that are of the domain */
|
|
for (i = 0; i < rtc->natts; i++)
|
|
{
|
|
int attnum = rtc->atts[i];
|
|
Datum d;
|
|
bool isNull;
|
|
Datum conResult;
|
|
|
|
d = heap_getattr(tuple, attnum, tupdesc, &isNull);
|
|
|
|
econtext->domainValue_datum = d;
|
|
econtext->domainValue_isNull = isNull;
|
|
|
|
conResult = ExecEvalExprSwitchContext(exprstate,
|
|
econtext,
|
|
&isNull, NULL);
|
|
|
|
if (!isNull && !DatumGetBool(conResult))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_CHECK_VIOLATION),
|
|
errmsg("column \"%s\" of table \"%s\" contains values that violate the new constraint",
|
|
NameStr(tupdesc->attrs[attnum - 1]->attname),
|
|
RelationGetRelationName(testrel))));
|
|
}
|
|
|
|
ResetExprContext(econtext);
|
|
}
|
|
heap_endscan(scan);
|
|
|
|
/* Hold relation lock till commit (XXX bad for concurrency) */
|
|
heap_close(testrel, NoLock);
|
|
}
|
|
|
|
FreeExecutorState(estate);
|
|
|
|
/* Clean up */
|
|
heap_close(typrel, RowExclusiveLock);
|
|
}
|
|
|
|
/*
|
|
* get_rels_with_domain
|
|
*
|
|
* Fetch all relations / attributes which are using the domain
|
|
*
|
|
* The result is a list of RelToCheck structs, one for each distinct
|
|
* relation, each containing one or more attribute numbers that are of
|
|
* the domain type. We have opened each rel and acquired the specified lock
|
|
* type on it.
|
|
*
|
|
* XXX this is completely broken because there is no way to lock the domain
|
|
* to prevent columns from being added or dropped while our command runs.
|
|
* We can partially protect against column drops by locking relations as we
|
|
* come across them, but there is still a race condition (the window between
|
|
* seeing a pg_depend entry and acquiring lock on the relation it references).
|
|
* Also, holding locks on all these relations simultaneously creates a non-
|
|
* trivial risk of deadlock. We can minimize but not eliminate the deadlock
|
|
* risk by using the weakest suitable lock (ShareLock for most callers).
|
|
*
|
|
* XXX to support domains over domains, we'd need to make this smarter,
|
|
* or make its callers smarter, so that we could find columns of derived
|
|
* domains. Arrays of domains would be a problem too.
|
|
*
|
|
* Generally used for retrieving a list of tests when adding
|
|
* new constraints to a domain.
|
|
*/
|
|
static List *
|
|
get_rels_with_domain(Oid domainOid, LOCKMODE lockmode)
|
|
{
|
|
List *result = NIL;
|
|
Relation depRel;
|
|
ScanKeyData key[2];
|
|
SysScanDesc depScan;
|
|
HeapTuple depTup;
|
|
|
|
/*
|
|
* We scan pg_depend to find those things that depend on the domain.
|
|
* (We assume we can ignore refobjsubid for a domain.)
|
|
*/
|
|
depRel = relation_openr(DependRelationName, AccessShareLock);
|
|
|
|
ScanKeyInit(&key[0],
|
|
Anum_pg_depend_refclassid,
|
|
BTEqualStrategyNumber, F_OIDEQ,
|
|
ObjectIdGetDatum(RelOid_pg_type));
|
|
ScanKeyInit(&key[1],
|
|
Anum_pg_depend_refobjid,
|
|
BTEqualStrategyNumber, F_OIDEQ,
|
|
ObjectIdGetDatum(domainOid));
|
|
|
|
depScan = systable_beginscan(depRel, DependReferenceIndex, true,
|
|
SnapshotNow, 2, key);
|
|
|
|
while (HeapTupleIsValid(depTup = systable_getnext(depScan)))
|
|
{
|
|
Form_pg_depend pg_depend = (Form_pg_depend) GETSTRUCT(depTup);
|
|
RelToCheck *rtc = NULL;
|
|
List *rellist;
|
|
Form_pg_attribute pg_att;
|
|
int ptr;
|
|
|
|
/* Ignore dependees that aren't user columns of tables */
|
|
/* (we assume system columns are never of domain types) */
|
|
if (pg_depend->classid != RelOid_pg_class ||
|
|
pg_depend->objsubid <= 0)
|
|
continue;
|
|
|
|
/* See if we already have an entry for this relation */
|
|
foreach(rellist, result)
|
|
{
|
|
RelToCheck *rt = (RelToCheck *) lfirst(rellist);
|
|
|
|
if (RelationGetRelid(rt->rel) == pg_depend->objid)
|
|
{
|
|
rtc = rt;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (rtc == NULL)
|
|
{
|
|
/* First attribute found for this relation */
|
|
Relation rel;
|
|
|
|
/* Acquire requested lock on relation */
|
|
rel = heap_open(pg_depend->objid, lockmode);
|
|
|
|
/* Build the RelToCheck entry with enough space for all atts */
|
|
rtc = (RelToCheck *) palloc(sizeof(RelToCheck));
|
|
rtc->rel = rel;
|
|
rtc->natts = 0;
|
|
rtc->atts = (int *) palloc(sizeof(int) * RelationGetNumberOfAttributes(rel));
|
|
result = lcons(rtc, result);
|
|
}
|
|
|
|
/*
|
|
* Confirm column has not been dropped, and is of the expected
|
|
* type. This defends against an ALTER DROP COLUMN occuring just
|
|
* before we acquired lock ... but if the whole table were
|
|
* dropped, we'd still have a problem.
|
|
*/
|
|
if (pg_depend->objsubid > RelationGetNumberOfAttributes(rtc->rel))
|
|
continue;
|
|
pg_att = rtc->rel->rd_att->attrs[pg_depend->objsubid - 1];
|
|
if (pg_att->attisdropped || pg_att->atttypid != domainOid)
|
|
continue;
|
|
|
|
/*
|
|
* Okay, add column to result. We store the columns in
|
|
* column-number order; this is just a hack to improve
|
|
* predictability of regression test output ...
|
|
*/
|
|
Assert(rtc->natts < RelationGetNumberOfAttributes(rtc->rel));
|
|
|
|
ptr = rtc->natts++;
|
|
while (ptr > 0 && rtc->atts[ptr - 1] > pg_depend->objsubid)
|
|
{
|
|
rtc->atts[ptr] = rtc->atts[ptr - 1];
|
|
ptr--;
|
|
}
|
|
rtc->atts[ptr] = pg_depend->objsubid;
|
|
}
|
|
|
|
systable_endscan(depScan);
|
|
|
|
relation_close(depRel, AccessShareLock);
|
|
|
|
return result;
|
|
}
|
|
|
|
/*
|
|
* domainOwnerCheck
|
|
*
|
|
* Throw an error if the current user doesn't have permission to modify
|
|
* the domain in an ALTER DOMAIN statement, or if the type isn't actually
|
|
* a domain.
|
|
*/
|
|
static void
|
|
domainOwnerCheck(HeapTuple tup, TypeName *typename)
|
|
{
|
|
Form_pg_type typTup = (Form_pg_type) GETSTRUCT(tup);
|
|
|
|
/* Check that this is actually a domain */
|
|
if (typTup->typtype != 'd')
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_WRONG_OBJECT_TYPE),
|
|
errmsg("\"%s\" is not a domain",
|
|
TypeNameToString(typename))));
|
|
|
|
/* Permission check: must own type */
|
|
if (!pg_type_ownercheck(HeapTupleGetOid(tup), GetUserId()))
|
|
aclcheck_error(ACLCHECK_NOT_OWNER, ACL_KIND_TYPE,
|
|
TypeNameToString(typename));
|
|
}
|
|
|
|
/*
|
|
* domainAddConstraint - code shared between CREATE and ALTER DOMAIN
|
|
*/
|
|
static char *
|
|
domainAddConstraint(Oid domainOid, Oid domainNamespace, Oid baseTypeOid,
|
|
int typMod, Constraint *constr,
|
|
int *counter, char *domainName)
|
|
{
|
|
Node *expr;
|
|
char *ccsrc;
|
|
char *ccbin;
|
|
ParseState *pstate;
|
|
CoerceToDomainValue *domVal;
|
|
|
|
/*
|
|
* Assign or validate constraint name
|
|
*/
|
|
if (constr->name)
|
|
{
|
|
if (ConstraintNameIsUsed(CONSTRAINT_DOMAIN,
|
|
domainOid,
|
|
domainNamespace,
|
|
constr->name))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_DUPLICATE_OBJECT),
|
|
errmsg("constraint \"%s\" for domain \"%s\" already exists",
|
|
constr->name, domainName)));
|
|
}
|
|
else
|
|
constr->name = GenerateConstraintName(CONSTRAINT_DOMAIN,
|
|
domainOid,
|
|
domainNamespace,
|
|
counter);
|
|
|
|
/*
|
|
* Convert the A_EXPR in raw_expr into an EXPR
|
|
*/
|
|
pstate = make_parsestate(NULL);
|
|
|
|
/*
|
|
* Set up a CoerceToDomainValue to represent the occurrence of VALUE
|
|
* in the expression. Note that it will appear to have the type of
|
|
* the base type, not the domain. This seems correct since within the
|
|
* check expression, we should not assume the input value can be
|
|
* considered a member of the domain.
|
|
*/
|
|
domVal = makeNode(CoerceToDomainValue);
|
|
domVal->typeId = baseTypeOid;
|
|
domVal->typeMod = typMod;
|
|
|
|
pstate->p_value_substitute = (Node *) domVal;
|
|
|
|
expr = transformExpr(pstate, constr->raw_expr);
|
|
|
|
/*
|
|
* Make sure it yields a boolean result.
|
|
*/
|
|
expr = coerce_to_boolean(pstate, expr, "CHECK");
|
|
|
|
/*
|
|
* Make sure no outside relations are referred to.
|
|
*/
|
|
if (length(pstate->p_rtable) != 0)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_COLUMN_REFERENCE),
|
|
errmsg("cannot use table references in domain check constraint")));
|
|
|
|
/*
|
|
* Domains don't allow var clauses (this should be redundant with the
|
|
* above check, but make it anyway)
|
|
*/
|
|
if (contain_var_clause(expr))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_INVALID_COLUMN_REFERENCE),
|
|
errmsg("cannot use table references in domain check constraint")));
|
|
|
|
/*
|
|
* No subplans or aggregates, either...
|
|
*/
|
|
if (pstate->p_hasSubLinks)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
|
|
errmsg("cannot use subquery in check constraint")));
|
|
if (pstate->p_hasAggs)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_GROUPING_ERROR),
|
|
errmsg("cannot use aggregate in check constraint")));
|
|
|
|
/*
|
|
* Convert to string form for storage.
|
|
*/
|
|
ccbin = nodeToString(expr);
|
|
|
|
/*
|
|
* Deparse it to produce text for consrc.
|
|
*
|
|
* Since VARNOs aren't allowed in domain constraints, relation context
|
|
* isn't required as anything other than a shell.
|
|
*/
|
|
ccsrc = deparse_expression(expr,
|
|
deparse_context_for(domainName,
|
|
InvalidOid),
|
|
false, false);
|
|
|
|
/*
|
|
* Store the constraint in pg_constraint
|
|
*/
|
|
CreateConstraintEntry(constr->name, /* Constraint Name */
|
|
domainNamespace, /* namespace */
|
|
CONSTRAINT_CHECK, /* Constraint Type */
|
|
false, /* Is Deferrable */
|
|
false, /* Is Deferred */
|
|
InvalidOid, /* not a relation constraint */
|
|
NULL,
|
|
0,
|
|
domainOid, /* domain constraint */
|
|
InvalidOid, /* Foreign key fields */
|
|
NULL,
|
|
0,
|
|
' ',
|
|
' ',
|
|
' ',
|
|
InvalidOid,
|
|
expr, /* Tree form check constraint */
|
|
ccbin, /* Binary form check constraint */
|
|
ccsrc); /* Source form check constraint */
|
|
|
|
/*
|
|
* Return the compiled constraint expression so the calling routine
|
|
* can perform any additional required tests.
|
|
*/
|
|
return ccbin;
|
|
}
|
|
|
|
/*
|
|
* GetDomainConstraints - get a list of the current constraints of domain
|
|
*
|
|
* Returns a possibly-empty list of DomainConstraintState nodes.
|
|
*
|
|
* This is called by the executor during plan startup for a CoerceToDomain
|
|
* expression node. The given constraints will be checked for each value
|
|
* passed through the node.
|
|
*/
|
|
List *
|
|
GetDomainConstraints(Oid typeOid)
|
|
{
|
|
List *result = NIL;
|
|
bool notNull = false;
|
|
Relation conRel;
|
|
|
|
conRel = heap_openr(ConstraintRelationName, AccessShareLock);
|
|
|
|
for (;;)
|
|
{
|
|
HeapTuple tup;
|
|
HeapTuple conTup;
|
|
Form_pg_type typTup;
|
|
ScanKeyData key[1];
|
|
SysScanDesc scan;
|
|
|
|
tup = SearchSysCache(TYPEOID,
|
|
ObjectIdGetDatum(typeOid),
|
|
0, 0, 0);
|
|
if (!HeapTupleIsValid(tup))
|
|
elog(ERROR, "cache lookup failed for type %u", typeOid);
|
|
typTup = (Form_pg_type) GETSTRUCT(tup);
|
|
|
|
/* Test for NOT NULL Constraint */
|
|
if (typTup->typnotnull)
|
|
notNull = true;
|
|
|
|
/* Look for CHECK Constraints on this domain */
|
|
ScanKeyInit(&key[0],
|
|
Anum_pg_constraint_contypid,
|
|
BTEqualStrategyNumber, F_OIDEQ,
|
|
ObjectIdGetDatum(typeOid));
|
|
|
|
scan = systable_beginscan(conRel, ConstraintTypidIndex, true,
|
|
SnapshotNow, 1, key);
|
|
|
|
while (HeapTupleIsValid(conTup = systable_getnext(scan)))
|
|
{
|
|
Form_pg_constraint c = (Form_pg_constraint) GETSTRUCT(conTup);
|
|
Datum val;
|
|
bool isNull;
|
|
Expr *check_expr;
|
|
DomainConstraintState *r;
|
|
|
|
/* Ignore non-CHECK constraints (presently, shouldn't be any) */
|
|
if (c->contype != CONSTRAINT_CHECK)
|
|
continue;
|
|
|
|
/*
|
|
* Not expecting conbin to be NULL, but we'll test for it
|
|
* anyway
|
|
*/
|
|
val = fastgetattr(conTup, Anum_pg_constraint_conbin,
|
|
conRel->rd_att, &isNull);
|
|
if (isNull)
|
|
elog(ERROR, "domain \"%s\" constraint \"%s\" has NULL conbin",
|
|
NameStr(typTup->typname), NameStr(c->conname));
|
|
|
|
check_expr = (Expr *)
|
|
stringToNode(DatumGetCString(DirectFunctionCall1(textout,
|
|
val)));
|
|
|
|
/* ExecInitExpr assumes we already fixed opfuncids */
|
|
fix_opfuncids((Node *) check_expr);
|
|
|
|
r = makeNode(DomainConstraintState);
|
|
r->constrainttype = DOM_CONSTRAINT_CHECK;
|
|
r->name = pstrdup(NameStr(c->conname));
|
|
r->check_expr = ExecInitExpr(check_expr, NULL);
|
|
|
|
/*
|
|
* use lcons() here because constraints of lower domains
|
|
* should be applied earlier.
|
|
*/
|
|
result = lcons(r, result);
|
|
}
|
|
|
|
systable_endscan(scan);
|
|
|
|
if (typTup->typtype != 'd')
|
|
{
|
|
/* Not a domain, so done */
|
|
ReleaseSysCache(tup);
|
|
break;
|
|
}
|
|
|
|
/* else loop to next domain in stack */
|
|
typeOid = typTup->typbasetype;
|
|
ReleaseSysCache(tup);
|
|
}
|
|
|
|
heap_close(conRel, AccessShareLock);
|
|
|
|
/*
|
|
* Only need to add one NOT NULL check regardless of how many domains
|
|
* in the stack request it.
|
|
*/
|
|
if (notNull)
|
|
{
|
|
DomainConstraintState *r = makeNode(DomainConstraintState);
|
|
|
|
r->constrainttype = DOM_CONSTRAINT_NOTNULL;
|
|
r->name = pstrdup("NOT NULL");
|
|
r->check_expr = NULL;
|
|
|
|
/* lcons to apply the nullness check FIRST */
|
|
result = lcons(r, result);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
/*
|
|
* ALTER DOMAIN .. OWNER TO
|
|
*
|
|
* Eventually this should allow changing ownership of other kinds of types,
|
|
* but some thought must be given to handling complex types. (A table's
|
|
* rowtype probably shouldn't be allowed as target, but what of a standalone
|
|
* composite type?)
|
|
*
|
|
* Assumes that permission checks have been completed earlier.
|
|
*/
|
|
void
|
|
AlterTypeOwner(List *names, AclId newOwnerSysId)
|
|
{
|
|
TypeName *typename;
|
|
Oid typeOid;
|
|
Relation rel;
|
|
HeapTuple tup;
|
|
Form_pg_type typTup;
|
|
|
|
/* Make a TypeName so we can use standard type lookup machinery */
|
|
typename = makeNode(TypeName);
|
|
typename->names = names;
|
|
typename->typmod = -1;
|
|
typename->arrayBounds = NIL;
|
|
|
|
/* Lock the type table */
|
|
rel = heap_openr(TypeRelationName, RowExclusiveLock);
|
|
|
|
/* Use LookupTypeName here so that shell types can be processed (why?) */
|
|
typeOid = LookupTypeName(typename);
|
|
if (!OidIsValid(typeOid))
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_UNDEFINED_OBJECT),
|
|
errmsg("type \"%s\" does not exist",
|
|
TypeNameToString(typename))));
|
|
|
|
tup = SearchSysCacheCopy(TYPEOID,
|
|
ObjectIdGetDatum(typeOid),
|
|
0, 0, 0);
|
|
if (!HeapTupleIsValid(tup))
|
|
elog(ERROR, "cache lookup failed for type %u", typeOid);
|
|
typTup = (Form_pg_type) GETSTRUCT(tup);
|
|
|
|
/* Check that this is actually a domain */
|
|
if (typTup->typtype != 'd')
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_WRONG_OBJECT_TYPE),
|
|
errmsg("\"%s\" is not a domain",
|
|
TypeNameToString(typename))));
|
|
|
|
/* Modify the owner --- okay to scribble on typTup because it's a copy */
|
|
typTup->typowner = newOwnerSysId;
|
|
|
|
simple_heap_update(rel, &tup->t_self, tup);
|
|
|
|
CatalogUpdateIndexes(rel, tup);
|
|
|
|
/* Clean up */
|
|
heap_close(rel, RowExclusiveLock);
|
|
}
|