#include "postgres.h"
#include "access/attmap.h"
#include "access/genam.h"
#include "access/gist.h"
#include "access/heapam.h"
#include "access/heapam_xlog.h"
#include "access/multixact.h"
#include "access/reloptions.h"
#include "access/relscan.h"
#include "access/sysattr.h"
#include "access/tableam.h"
#include "access/toast_compression.h"
#include "access/tupconvert.h"
#include "access/xact.h"
#include "access/xlog.h"
#include "access/xloginsert.h"
#include "catalog/catalog.h"
#include "catalog/heap.h"
#include "catalog/index.h"
#include "catalog/namespace.h"
#include "catalog/objectaccess.h"
#include "catalog/partition.h"
#include "catalog/pg_am.h"
#include "catalog/pg_attrdef.h"
#include "catalog/pg_collation.h"
#include "catalog/pg_constraint.h"
#include "catalog/pg_depend.h"
#include "catalog/pg_extension_d.h"
#include "catalog/pg_foreign_table.h"
#include "catalog/pg_inherits.h"
#include "catalog/pg_largeobject.h"
#include "catalog/pg_largeobject_metadata.h"
#include "catalog/pg_namespace.h"
#include "catalog/pg_opclass.h"
#include "catalog/pg_policy.h"
#include "catalog/pg_proc.h"
#include "catalog/pg_publication_rel.h"
#include "catalog/pg_rewrite.h"
#include "catalog/pg_statistic_ext.h"
#include "catalog/pg_tablespace.h"
#include "catalog/pg_trigger.h"
#include "catalog/pg_type.h"
#include "catalog/storage.h"
#include "catalog/storage_xlog.h"
#include "catalog/toasting.h"
#include "commands/comment.h"
#include "commands/defrem.h"
#include "commands/event_trigger.h"
#include "commands/extension.h"
#include "commands/repack.h"
#include "commands/sequence.h"
#include "commands/tablecmds.h"
#include "commands/tablespace.h"
#include "commands/trigger.h"
#include "commands/typecmds.h"
#include "commands/user.h"
#include "commands/vacuum.h"
#include "common/int.h"
#include "executor/executor.h"
#include "foreign/fdwapi.h"
#include "foreign/foreign.h"
#include "miscadmin.h"
#include "nodes/makefuncs.h"
#include "nodes/nodeFuncs.h"
#include "nodes/parsenodes.h"
#include "optimizer/optimizer.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
#include "parser/parse_relation.h"
#include "parser/parse_type.h"
#include "parser/parse_utilcmd.h"
#include "parser/parser.h"
#include "partitioning/partbounds.h"
#include "partitioning/partdesc.h"
#include "pgstat.h"
#include "rewrite/rewriteDefine.h"
#include "rewrite/rewriteHandler.h"
#include "rewrite/rewriteManip.h"
#include "storage/bufmgr.h"
#include "storage/lmgr.h"
#include "storage/lock.h"
#include "storage/predicate.h"
#include "storage/smgr.h"
#include "tcop/utility.h"
#include "utils/acl.h"
#include "utils/builtins.h"
#include "utils/fmgroids.h"
#include "utils/inval.h"
#include "utils/lsyscache.h"
#include "utils/memutils.h"
#include "utils/partcache.h"
#include "utils/relcache.h"
#include "utils/ruleutils.h"
#include "utils/snapmgr.h"
#include "utils/syscache.h"
#include "utils/timestamp.h"
#include "utils/typcache.h"
#include "utils/usercontext.h"

Include dependency graph for tablecmds.c:

Data Structures
struct	OnCommitItem

struct	AlteredTableInfo

struct	NewConstraint

struct	NewColumnValue

struct	dropmsgstrings

struct	DropRelationCallbackState

struct	ForeignTruncateInfo

struct	PartitionIndexExtDepEntry

struct	AttachIndexCallbackState

struct	SplitPartitionContext

Macros
#define	AT_NUM_PASSES (AT_PASS_MISC + 1)

#define	ATT_TABLE 0x0001

#define	ATT_VIEW 0x0002

#define	ATT_MATVIEW 0x0004

#define	ATT_INDEX 0x0008

#define	ATT_COMPOSITE_TYPE 0x0010

#define	ATT_FOREIGN_TABLE 0x0020

#define	ATT_PARTITIONED_INDEX 0x0040

#define	ATT_SEQUENCE 0x0080

#define	ATT_PARTITIONED_TABLE 0x0100

#define	child_dependency_type(child_is_partition) ((child_is_partition) ? DEPENDENCY_AUTO : DEPENDENCY_NORMAL)

Typedefs
typedef struct OnCommitItem	OnCommitItem

typedef enum AlterTablePass	AlterTablePass

typedef struct AlteredTableInfo	AlteredTableInfo

typedef struct NewConstraint	NewConstraint

typedef struct NewColumnValue	NewColumnValue

typedef struct ForeignTruncateInfo	ForeignTruncateInfo

typedef enum addFkConstraintSides	addFkConstraintSides

typedef struct PartitionIndexExtDepEntry	PartitionIndexExtDepEntry

typedef struct SplitPartitionContext	SplitPartitionContext

Enumerations
enum	AlterTablePass { AT_PASS_UNSET = -1 , AT_PASS_DROP , AT_PASS_ALTER_TYPE , AT_PASS_ADD_COL , AT_PASS_SET_EXPRESSION , AT_PASS_OLD_INDEX , AT_PASS_OLD_CONSTR , AT_PASS_ADD_CONSTR , AT_PASS_COL_ATTRS , AT_PASS_ADD_INDEXCONSTR , AT_PASS_ADD_INDEX , AT_PASS_ADD_OTHERCONSTR , AT_PASS_MISC }

enum	addFkConstraintSides { addFkReferencedSide , addFkReferencingSide , addFkBothSides }

Functions
static void	truncate_check_rel (Oid relid, Form_pg_class reltuple)

static void	truncate_check_perms (Oid relid, Form_pg_class reltuple)

static void	truncate_check_activity (Relation rel)

static void	RangeVarCallbackForTruncate (const RangeVar relation, Oid relId, Oid oldRelId, void arg)

static List *	MergeAttributes (List columns, const List supers, char relpersistence, bool is_partition, List supconstr, List supnotnulls)

static List *	MergeCheckConstraint (List constraints, const char name, Node *expr, bool is_enforced)

static void	MergeChildAttribute (List inh_columns, int exist_attno, int newcol_attno, const ColumnDef newdef)

static ColumnDef *	MergeInheritedAttribute (List inh_columns, int exist_attno, const ColumnDef newdef)

static void	MergeAttributesIntoExisting (Relation child_rel, Relation parent_rel, bool ispartition)

static void	MergeConstraintsIntoExisting (Relation child_rel, Relation parent_rel)

static void	StoreCatalogInheritance (Oid relationId, List *supers, bool child_is_partition)

static void	StoreCatalogInheritance1 (Oid relationId, Oid parentOid, int32 seqNumber, Relation inhRelation, bool child_is_partition)

static int	findAttrByName (const char attributeName, const List columns)

static void	AlterIndexNamespaces (Relation classRel, Relation rel, Oid oldNspOid, Oid newNspOid, ObjectAddresses *objsMoved)

static void	AlterSeqNamespaces (Relation classRel, Relation rel, Oid oldNspOid, Oid newNspOid, ObjectAddresses *objsMoved, LOCKMODE lockmode)

static ObjectAddress	ATExecAlterConstraint (List *wqueue, Relation rel, ATAlterConstraint cmdcon, bool recurse, LOCKMODE lockmode)

static bool	ATExecAlterConstraintInternal (List *wqueue, ATAlterConstraint cmdcon, Relation conrel, Relation tgrel, Relation rel, HeapTuple contuple, bool recurse, LOCKMODE lockmode)

static bool	ATExecAlterFKConstrEnforceability (List *wqueue, ATAlterConstraint cmdcon, Relation conrel, Relation tgrel, Oid fkrelid, Oid pkrelid, HeapTuple contuple, LOCKMODE lockmode, Oid ReferencedParentDelTrigger, Oid ReferencedParentUpdTrigger, Oid ReferencingParentInsTrigger, Oid ReferencingParentUpdTrigger)

static bool	ATExecAlterCheckConstrEnforceability (List *wqueue, ATAlterConstraint cmdcon, Relation conrel, HeapTuple contuple, bool recurse, bool recursing, List *changing_conids, LOCKMODE lockmode)

static bool	ATExecAlterConstrDeferrability (List *wqueue, ATAlterConstraint cmdcon, Relation conrel, Relation tgrel, Relation rel, HeapTuple contuple, bool recurse, List **otherrelids, LOCKMODE lockmode)

static bool	ATExecAlterConstrInheritability (List *wqueue, ATAlterConstraint cmdcon, Relation conrel, Relation rel, HeapTuple contuple, LOCKMODE lockmode)

static void	AlterConstrTriggerDeferrability (Oid conoid, Relation tgrel, Relation rel, bool deferrable, bool initdeferred, List **otherrelids)

static void	AlterFKConstrEnforceabilityRecurse (List *wqueue, ATAlterConstraint cmdcon, Relation conrel, Relation tgrel, Oid fkrelid, Oid pkrelid, HeapTuple contuple, LOCKMODE lockmode, Oid ReferencedParentDelTrigger, Oid ReferencedParentUpdTrigger, Oid ReferencingParentInsTrigger, Oid ReferencingParentUpdTrigger)

static void	AlterCheckConstrEnforceabilityRecurse (List *wqueue, ATAlterConstraint cmdcon, Relation conrel, Oid conrelid, bool recurse, bool recursing, List *changing_conids, LOCKMODE lockmode)

static void	AlterConstrDeferrabilityRecurse (List *wqueue, ATAlterConstraint cmdcon, Relation conrel, Relation tgrel, Relation rel, HeapTuple contuple, bool recurse, List **otherrelids, LOCKMODE lockmode)

static void	AlterConstrUpdateConstraintEntry (ATAlterConstraint *cmdcon, Relation conrel, HeapTuple contuple)

static bool	ATCheckCheckConstrHasEnforcedParent (Relation conrel, Relation rel, HeapTuple contuple, List changing_conids, Oid enforced_parentoid)

static ObjectAddress	ATExecValidateConstraint (List *wqueue, Relation rel, char constrName, bool recurse, bool recursing, LOCKMODE lockmode)

static void	QueueFKConstraintValidation (List **wqueue, Relation conrel, Relation fkrel, Oid pkrelid, HeapTuple contuple, LOCKMODE lockmode)

static void	QueueCheckConstraintValidation (List *wqueue, Relation conrel, Relation rel, char constrName, HeapTuple contuple, bool recurse, bool recursing, LOCKMODE lockmode)

static void	QueueNNConstraintValidation (List **wqueue, Relation conrel, Relation rel, HeapTuple contuple, bool recurse, bool recursing, LOCKMODE lockmode)

static bool	constraints_equivalent (HeapTuple a, HeapTuple b, TupleDesc tupleDesc)

static int	transformColumnNameList (Oid relId, List colList, int16 attnums, Oid atttypids, Oid attcollids)

static int	transformFkeyGetPrimaryKey (Relation pkrel, Oid indexOid, List attnamelist, int16 attnums, Oid atttypids, Oid attcollids, Oid opclasses, bool pk_has_without_overlaps)

static Oid	transformFkeyCheckAttrs (Relation pkrel, int numattrs, int16 attnums, bool with_period, Oid opclasses, bool *pk_has_without_overlaps)

static void	checkFkeyPermissions (Relation rel, int16 *attnums, int natts)

static CoercionPathType	findFkeyCast (Oid targetTypeId, Oid sourceTypeId, Oid *funcid)

static void	validateForeignKeyConstraint (char *conname, Relation rel, Relation pkrel, Oid pkindOid, Oid constraintOid, bool hasperiod)

static void	CheckAlterTableIsSafe (Relation rel)

static void	ATController (AlterTableStmt parsetree, Relation rel, List cmds, bool recurse, LOCKMODE lockmode, AlterTableUtilityContext *context)

static void	ATPrepCmd (List *wqueue, Relation rel, AlterTableCmd cmd, bool recurse, bool recursing, LOCKMODE lockmode, AlterTableUtilityContext *context)

static void	ATRewriteCatalogs (List *wqueue, LOCKMODE lockmode, AlterTableUtilityContext context)

static void	ATExecCmd (List *wqueue, AlteredTableInfo tab, AlterTableCmd cmd, LOCKMODE lockmode, AlterTablePass cur_pass, AlterTableUtilityContext context)

static AlterTableCmd *	ATParseTransformCmd (List *wqueue, AlteredTableInfo tab, Relation rel, AlterTableCmd cmd, bool recurse, LOCKMODE lockmode, AlterTablePass cur_pass, AlterTableUtilityContext context)

static void	ATRewriteTables (AlterTableStmt parsetree, List wqueue, LOCKMODE lockmode, AlterTableUtilityContext context)

static void	ATRewriteTable (AlteredTableInfo *tab, Oid OIDNewHeap)

static AlteredTableInfo *	ATGetQueueEntry (List **wqueue, Relation rel)

static void	ATSimplePermissions (AlterTableType cmdtype, Relation rel, int allowed_targets)

static void	ATSimpleRecursion (List *wqueue, Relation rel, AlterTableCmd cmd, bool recurse, LOCKMODE lockmode, AlterTableUtilityContext *context)

static void	ATCheckPartitionsNotInUse (Relation rel, LOCKMODE lockmode)

static void	ATTypedTableRecursion (List *wqueue, Relation rel, AlterTableCmd cmd, LOCKMODE lockmode, AlterTableUtilityContext *context)

static List *	find_typed_table_dependencies (Oid typeOid, const char *typeName, DropBehavior behavior)

static void	ATPrepAddColumn (List *wqueue, Relation rel, bool recurse, bool recursing, bool is_view, AlterTableCmd cmd, LOCKMODE lockmode, AlterTableUtilityContext *context)

static ObjectAddress	ATExecAddColumn (List *wqueue, AlteredTableInfo tab, Relation rel, AlterTableCmd *cmd, bool recurse, bool recursing, LOCKMODE lockmode, AlterTablePass cur_pass, AlterTableUtilityContext context)

static bool	check_for_column_name_collision (Relation rel, const char *colname, bool if_not_exists)

static void	add_column_datatype_dependency (Oid relid, int32 attnum, Oid typid)

static void	add_column_collation_dependency (Oid relid, int32 attnum, Oid collid)

static ObjectAddress	ATExecDropNotNull (Relation rel, const char *colName, bool recurse, LOCKMODE lockmode)

static void	set_attnotnull (List **wqueue, Relation rel, AttrNumber attnum, bool is_valid, bool queue_validation)

static ObjectAddress	ATExecSetNotNull (List *wqueue, Relation rel, char conName, char *colName, bool recurse, bool recursing, LOCKMODE lockmode)

static bool	NotNullImpliedByRelConstraints (Relation rel, Form_pg_attribute attr)

static bool	ConstraintImpliedByRelConstraint (Relation scanrel, List testConstraint, List provenConstraint)

static ObjectAddress	ATExecColumnDefault (Relation rel, const char colName, Node newDefault, LOCKMODE lockmode)

static ObjectAddress	ATExecCookedColumnDefault (Relation rel, AttrNumber attnum, Node *newDefault)

static ObjectAddress	ATExecAddIdentity (Relation rel, const char colName, Node def, LOCKMODE lockmode, bool recurse, bool recursing)

static ObjectAddress	ATExecSetIdentity (Relation rel, const char colName, Node def, LOCKMODE lockmode, bool recurse, bool recursing)

static ObjectAddress	ATExecDropIdentity (Relation rel, const char *colName, bool missing_ok, LOCKMODE lockmode, bool recurse, bool recursing)

static ObjectAddress	ATExecSetExpression (AlteredTableInfo tab, Relation rel, const char colName, Node *newExpr, LOCKMODE lockmode)

static void	ATPrepDropExpression (Relation rel, AlterTableCmd *cmd, bool recurse, bool recursing, LOCKMODE lockmode)

static ObjectAddress	ATExecDropExpression (Relation rel, const char *colName, bool missing_ok, LOCKMODE lockmode)

static ObjectAddress	ATExecSetStatistics (Relation rel, const char colName, int16 colNum, Node newValue, LOCKMODE lockmode)

static ObjectAddress	ATExecSetOptions (Relation rel, const char colName, Node options, bool isReset, LOCKMODE lockmode)

static ObjectAddress	ATExecSetStorage (Relation rel, const char colName, Node newValue, LOCKMODE lockmode)

static void	ATPrepDropColumn (List *wqueue, Relation rel, bool recurse, bool recursing, AlterTableCmd cmd, LOCKMODE lockmode, AlterTableUtilityContext *context)

static ObjectAddress	ATExecDropColumn (List *wqueue, Relation rel, const char colName, DropBehavior behavior, bool recurse, bool recursing, bool missing_ok, LOCKMODE lockmode, ObjectAddresses *addrs)

static void	ATPrepAddPrimaryKey (List *wqueue, Relation rel, AlterTableCmd cmd, bool recurse, LOCKMODE lockmode, AlterTableUtilityContext *context)

static void	verifyNotNullPKCompatible (HeapTuple tuple, const char *colname)

static ObjectAddress	ATExecAddIndex (AlteredTableInfo tab, Relation rel, IndexStmt stmt, bool is_rebuild, LOCKMODE lockmode)

static ObjectAddress	ATExecAddStatistics (AlteredTableInfo tab, Relation rel, CreateStatsStmt stmt, bool is_rebuild, LOCKMODE lockmode)

static ObjectAddress	ATExecAddConstraint (List *wqueue, AlteredTableInfo tab, Relation rel, Constraint *newConstraint, bool recurse, bool is_readd, LOCKMODE lockmode)

static char *	ChooseForeignKeyConstraintNameAddition (List *colnames)

static ObjectAddress	ATExecAddIndexConstraint (AlteredTableInfo tab, Relation rel, IndexStmt stmt, LOCKMODE lockmode)

static ObjectAddress	ATAddCheckNNConstraint (List *wqueue, AlteredTableInfo tab, Relation rel, Constraint *constr, bool recurse, bool recursing, bool is_readd, LOCKMODE lockmode)

static ObjectAddress	ATAddForeignKeyConstraint (List *wqueue, AlteredTableInfo tab, Relation rel, Constraint *fkconstraint, bool recurse, bool recursing, LOCKMODE lockmode)

static int	validateFkOnDeleteSetColumns (int numfks, const int16 fkattnums, int numfksetcols, int16 fksetcolsattnums, List *fksetcols)

static ObjectAddress	addFkConstraint (addFkConstraintSides fkside, char constraintname, Constraint fkconstraint, Relation rel, Relation pkrel, Oid indexOid, Oid parentConstr, int numfks, int16 pkattnum, int16 fkattnum, Oid pfeqoperators, Oid ppeqoperators, Oid ffeqoperators, int numfkdelsetcols, int16 fkdelsetcols, bool is_internal, bool with_period)

static void	addFkRecurseReferenced (Constraint fkconstraint, Relation rel, Relation pkrel, Oid indexOid, Oid parentConstr, int numfks, int16 pkattnum, int16 fkattnum, Oid pfeqoperators, Oid ppeqoperators, Oid ffeqoperators, int numfkdelsetcols, int16 *fkdelsetcols, bool old_check_ok, Oid parentDelTrigger, Oid parentUpdTrigger, bool with_period)

static void	addFkRecurseReferencing (List *wqueue, Constraint fkconstraint, Relation rel, Relation pkrel, Oid indexOid, Oid parentConstr, int numfks, int16 pkattnum, int16 fkattnum, Oid pfeqoperators, Oid ppeqoperators, Oid ffeqoperators, int numfkdelsetcols, int16 fkdelsetcols, bool old_check_ok, LOCKMODE lockmode, Oid parentInsTrigger, Oid parentUpdTrigger, bool with_period)

static void	CloneForeignKeyConstraints (List **wqueue, Relation parentRel, Relation partitionRel)

static void	CloneFkReferenced (Relation parentRel, Relation partitionRel)

static void	CloneFkReferencing (List **wqueue, Relation parentRel, Relation partRel)

static void	createForeignKeyCheckTriggers (Oid myRelOid, Oid refRelOid, Constraint fkconstraint, Oid constraintOid, Oid indexOid, Oid parentInsTrigger, Oid parentUpdTrigger, Oid insertTrigOid, Oid *updateTrigOid)

static void	createForeignKeyActionTriggers (Oid myRelOid, Oid refRelOid, Constraint fkconstraint, Oid constraintOid, Oid indexOid, Oid parentDelTrigger, Oid parentUpdTrigger, Oid deleteTrigOid, Oid *updateTrigOid)

static bool	tryAttachPartitionForeignKey (List *wqueue, ForeignKeyCacheInfo fk, Relation partition, Oid parentConstrOid, int numfks, AttrNumber mapped_conkey, AttrNumber confkey, Oid *conpfeqop, Oid parentInsTrigger, Oid parentUpdTrigger, Relation trigrel)

static void	AttachPartitionForeignKey (List **wqueue, Relation partition, Oid partConstrOid, Oid parentConstrOid, Oid parentInsTrigger, Oid parentUpdTrigger, Relation trigrel)

static void	RemoveInheritedConstraint (Relation conrel, Relation trigrel, Oid conoid, Oid conrelid)

static void	DropForeignKeyConstraintTriggers (Relation trigrel, Oid conoid, Oid confrelid, Oid conrelid)

static void	GetForeignKeyActionTriggers (Relation trigrel, Oid conoid, Oid confrelid, Oid conrelid, Oid deleteTriggerOid, Oid updateTriggerOid)

static void	GetForeignKeyCheckTriggers (Relation trigrel, Oid conoid, Oid confrelid, Oid conrelid, Oid insertTriggerOid, Oid updateTriggerOid)

static void	ATExecDropConstraint (Relation rel, const char *constrName, DropBehavior behavior, bool recurse, bool missing_ok, LOCKMODE lockmode)

static ObjectAddress	dropconstraint_internal (Relation rel, HeapTuple constraintTup, DropBehavior behavior, bool recurse, bool recursing, bool missing_ok, LOCKMODE lockmode)

static void	ATPrepAlterColumnType (List *wqueue, AlteredTableInfo tab, Relation rel, bool recurse, bool recursing, AlterTableCmd cmd, LOCKMODE lockmode, AlterTableUtilityContext context)

static bool	ATColumnChangeRequiresRewrite (Node *expr, AttrNumber varattno)

static ObjectAddress	ATExecAlterColumnType (AlteredTableInfo tab, Relation rel, AlterTableCmd cmd, LOCKMODE lockmode)

static void	RememberAllDependentForRebuilding (AlteredTableInfo tab, AlterTableType subtype, Relation rel, AttrNumber attnum, const char colName)

static void	RememberWholeRowDependentForRebuilding (AlteredTableInfo *tab, AlterTableType subtype, Relation rel)

static void	RememberConstraintForRebuilding (Oid conoid, AlteredTableInfo *tab)

static void	RememberIndexForRebuilding (Oid indoid, AlteredTableInfo *tab)

static void	RememberStatisticsForRebuilding (Oid stxoid, AlteredTableInfo *tab)

static void	ATPostAlterTypeCleanup (List *wqueue, AlteredTableInfo tab, LOCKMODE lockmode)

static void	ATPostAlterTypeParse (Oid oldId, Oid oldRelId, Oid refRelId, char cmd, List *wqueue, LOCKMODE lockmode, bool rewrite)

static void	RebuildConstraintComment (AlteredTableInfo tab, AlterTablePass pass, Oid objid, Relation rel, List domname, const char *conname)

static void	TryReuseIndex (Oid oldId, IndexStmt *stmt)

static void	TryReuseForeignKey (Oid oldId, Constraint *con)

static ObjectAddress	ATExecAlterColumnGenericOptions (Relation rel, const char colName, List options, LOCKMODE lockmode)

static void	change_owner_fix_column_acls (Oid relationOid, Oid oldOwnerId, Oid newOwnerId)

static void	change_owner_recurse_to_sequences (Oid relationOid, Oid newOwnerId, LOCKMODE lockmode)

static ObjectAddress	ATExecClusterOn (Relation rel, const char *indexName, LOCKMODE lockmode)

static void	ATExecDropCluster (Relation rel, LOCKMODE lockmode)

static void	ATPrepSetAccessMethod (AlteredTableInfo tab, Relation rel, const char amname)

static void	ATExecSetAccessMethodNoStorage (Relation rel, Oid newAccessMethodId)

static void	ATPrepChangePersistence (AlteredTableInfo *tab, Relation rel, bool toLogged)

static void	ATPrepSetTableSpace (AlteredTableInfo tab, Relation rel, const char tablespacename, LOCKMODE lockmode)

static void	ATExecSetTableSpace (Oid tableOid, Oid newTableSpace, LOCKMODE lockmode)

static void	ATExecSetTableSpaceNoStorage (Relation rel, Oid newTableSpace)

static void	ATExecSetRelOptions (Relation rel, List *defList, AlterTableType operation, LOCKMODE lockmode)

static void	ATExecEnableDisableTrigger (Relation rel, const char *trigname, char fires_when, bool skip_system, bool recurse, LOCKMODE lockmode)

static void	ATExecEnableDisableRule (Relation rel, const char *rulename, char fires_when, LOCKMODE lockmode)

static void	ATPrepChangeInherit (Relation child_rel)

static ObjectAddress	ATExecAddInherit (Relation child_rel, RangeVar *parent, LOCKMODE lockmode)

static ObjectAddress	ATExecDropInherit (Relation rel, RangeVar *parent, LOCKMODE lockmode)

static void	drop_parent_dependency (Oid relid, Oid refclassid, Oid refobjid, DependencyType deptype)

static ObjectAddress	ATExecAddOf (Relation rel, const TypeName *ofTypename, LOCKMODE lockmode)

static void	ATExecDropOf (Relation rel, LOCKMODE lockmode)

static void	ATExecReplicaIdentity (Relation rel, ReplicaIdentityStmt *stmt, LOCKMODE lockmode)

static void	ATExecGenericOptions (Relation rel, List *options)

static void	ATExecSetRowSecurity (Relation rel, bool rls)

static void	ATExecForceNoForceRowSecurity (Relation rel, bool force_rls)

static ObjectAddress	ATExecSetCompression (Relation rel, const char column, Node newValue, LOCKMODE lockmode)

static void	index_copy_data (Relation rel, RelFileLocator newrlocator)

static const char *	storage_name (char c)

static void	RangeVarCallbackForDropRelation (const RangeVar rel, Oid relOid, Oid oldRelOid, void arg)

static void	RangeVarCallbackForAlterRelation (const RangeVar rv, Oid relid, Oid oldrelid, void arg)

static PartitionSpec *	transformPartitionSpec (Relation rel, PartitionSpec *partspec)

static void	ComputePartitionAttrs (ParseState pstate, Relation rel, List partParams, AttrNumber partattrs, List partexprs, Oid partopclass, Oid *partcollation, PartitionStrategy strategy)

static void	CreateInheritance (Relation child_rel, Relation parent_rel, bool ispartition)

static void	RemoveInheritance (Relation child_rel, Relation parent_rel, bool expect_detached)

static ObjectAddress	ATExecAttachPartition (List *wqueue, Relation rel, PartitionCmd cmd, AlterTableUtilityContext *context)

static void	AttachPartitionEnsureIndexes (List **wqueue, Relation rel, Relation attachrel)

static void	QueuePartitionConstraintValidation (List *wqueue, Relation scanrel, List partConstraint, bool validate_default)

static void	CloneRowTriggersToPartition (Relation parent, Relation partition)

static void	DropClonedTriggersFromPartition (Oid partitionId)

static ObjectAddress	ATExecDetachPartition (List *wqueue, AlteredTableInfo tab, Relation rel, RangeVar *name, bool concurrent)

static void	DetachPartitionFinalize (Relation rel, Relation partRel, bool concurrent, Oid defaultPartOid)

static ObjectAddress	ATExecDetachPartitionFinalize (Relation rel, RangeVar *name)

static ObjectAddress	ATExecAttachPartitionIdx (List *wqueue, Relation parentIdx, RangeVar name)

static void	validatePartitionedIndex (Relation partedIdx, Relation partedTbl)

static void	refuseDupeIndexAttach (Relation parentIdx, Relation partIdx, Relation partitionTbl)

static void	verifyPartitionIndexNotNull (IndexInfo *iinfo, Relation partition)

static List *	GetParentedForeignKeyRefs (Relation partition)

static void	ATDetachCheckNoForeignKeyRefs (Relation partition)

static char	GetAttributeCompression (Oid atttypid, const char *compression)

static char	GetAttributeStorage (Oid atttypid, const char *storagemode)

static void	ATExecMergePartitions (List *wqueue, AlteredTableInfo tab, Relation rel, PartitionCmd cmd, AlterTableUtilityContext context)

static void	ATExecSplitPartition (List *wqueue, AlteredTableInfo tab, Relation rel, PartitionCmd cmd, AlterTableUtilityContext context)

static List *	collectPartitionIndexExtDeps (List *partitionOids)

static void	applyPartitionIndexExtDeps (Oid newPartOid, List *extDepState)

static void	freePartitionIndexExtDeps (List *extDepState)

ObjectAddress	DefineRelation (CreateStmt stmt, char relkind, Oid ownerId, ObjectAddress typaddress, const char *queryString)

TupleDesc	BuildDescForRelation (const List *columns)

static void	DropErrorMsgNonExistent (RangeVar *rel, char rightkind, bool missing_ok)

static void	DropErrorMsgWrongType (const char *relname, char wrongkind, char rightkind)

void	RemoveRelations (DropStmt *drop)

void	ExecuteTruncate (TruncateStmt *stmt)

void	ExecuteTruncateGuts (List explicit_rels, List relids, List *relids_logged, DropBehavior behavior, bool restart_seqs, bool run_as_table_owner)

void	SetRelationHasSubclass (Oid relationId, bool relhassubclass)

bool	CheckRelationTableSpaceMove (Relation rel, Oid newTableSpaceId)

void	SetRelationTableSpace (Relation rel, Oid newTableSpaceId, RelFileNumber newRelFilenumber)

static void	renameatt_check (Oid myrelid, Form_pg_class classform, bool recursing)

static AttrNumber	renameatt_internal (Oid myrelid, const char oldattname, const char newattname, bool recurse, bool recursing, int expected_parents, DropBehavior behavior)

static void	RangeVarCallbackForRenameAttribute (const RangeVar rv, Oid relid, Oid oldrelid, void arg)

ObjectAddress	renameatt (RenameStmt *stmt)

static ObjectAddress	rename_constraint_internal (Oid myrelid, Oid mytypid, const char oldconname, const char newconname, bool recurse, bool recursing, int expected_parents)

ObjectAddress	RenameConstraint (RenameStmt *stmt)

ObjectAddress	RenameRelation (RenameStmt *stmt)

void	RenameRelationInternal (Oid myrelid, const char *newrelname, bool is_internal, bool is_index)

void	ResetRelRewrite (Oid myrelid)

void	CheckTableNotInUse (Relation rel, const char *stmt)

Oid	AlterTableLookupRelation (AlterTableStmt *stmt, LOCKMODE lockmode)

void	AlterTable (AlterTableStmt stmt, LOCKMODE lockmode, AlterTableUtilityContext context)

void	AlterTableInternal (Oid relid, List *cmds, bool recurse)

LOCKMODE	AlterTableGetLockLevel (List *cmds)

static const char *	alter_table_type_to_string (AlterTableType cmdtype)

void	find_composite_type_dependencies (Oid typeOid, Relation origRelation, const char *origTypeName)

void	check_of_type (HeapTuple typetuple)

static void	SetIndexStorageProperties (Relation rel, Relation attrelation, AttrNumber attnum, bool setstorage, char newstorage, bool setcompression, char newcompression, LOCKMODE lockmode)

static Oid	CreateFKCheckTrigger (Oid myRelOid, Oid refRelOid, Constraint *fkconstraint, Oid constraintOid, Oid indexOid, Oid parentTrigOid, bool on_insert)

static void	RememberReplicaIdentityForRebuilding (Oid indoid, AlteredTableInfo *tab)

static void	RememberClusterOnForRebuilding (Oid indoid, AlteredTableInfo *tab)

void	ATExecChangeOwner (Oid relationOid, Oid newOwnerId, bool recursing, LOCKMODE lockmode)

Oid	AlterTableMoveAll (AlterTableMoveAllStmt *stmt)

static char *	decompile_conbin (HeapTuple contup, TupleDesc tupdesc)

static void	MarkInheritDetached (Relation child_rel, Relation parent_rel)

static void	relation_mark_replica_identity (Relation rel, char ri_type, Oid indexOid, bool is_internal)

ObjectAddress	AlterTableNamespace (AlterObjectSchemaStmt stmt, Oid oldschema)

void	AlterTableNamespaceInternal (Relation rel, Oid oldNspOid, Oid nspOid, ObjectAddresses *objsMoved)

void	AlterRelationNamespaceInternal (Relation classRel, Oid relOid, Oid oldNspOid, Oid newNspOid, bool hasDependEntry, ObjectAddresses *objsMoved)

void	register_on_commit_action (Oid relid, OnCommitAction action)

void	remove_on_commit_action (Oid relid)

void	PreCommit_on_commit_actions (void)

void	AtEOXact_on_commit_actions (bool isCommit)

void	AtEOSubXact_on_commit_actions (bool isCommit, SubTransactionId mySubid, SubTransactionId parentSubid)

void	RangeVarCallbackMaintainsTable (const RangeVar relation, Oid relId, Oid oldRelId, void arg)

void	RangeVarCallbackOwnsRelation (const RangeVar relation, Oid relId, Oid oldRelId, void arg)

bool	PartConstraintImpliedByRelConstraint (Relation scanrel, List *partConstraint)

static void	attachPartitionTable (List *wqueue, Relation rel, Relation attachrel, PartitionBoundSpec bound)

static void	RangeVarCallbackForAttachIndex (const RangeVar rv, Oid relOid, Oid oldRelOid, void arg)

static void	buildExpressionExecutionStates (AlteredTableInfo tab, Relation newPartRel, EState estate)

static void	evaluateGeneratedExpressionsAndCheckConstraints (AlteredTableInfo tab, Relation newPartRel, TupleTableSlot insertslot, ExprContext *econtext)

static List *	getAttributesList (Relation parent_rel)

static void	createTableConstraints (List *wqueue, AlteredTableInfo tab, Relation parent_rel, Relation newRel)

static Relation	createPartitionTable (List *wqueue, RangeVar newPartName, Relation parent_rel, Oid ownerId)

static void	MergePartitionsMoveRows (List *wqueue, List mergingPartitions, Relation newPartRel)

static void	detachPartitionTable (Relation parent_rel, Relation child_rel, Oid defaultPartOid)

static bool	equal_oid_lists (const List a, const List b)

static int	cmp_partition_index_ext_dep (const ListCell a, const ListCell b)

static SplitPartitionContext *	createSplitPartitionContext (Relation partRel)

static void	deleteSplitPartitionContext (SplitPartitionContext pc, List *wqueue, uint32 ti_options)

static void	SplitPartitionMoveRows (List *wqueue, Relation rel, Relation splitRel, List partlist, List *newPartRels)

Variables
static List *	on_commits = NIL

static const struct dropmsgstrings	dropmsgstringarray []

Macro Definition Documentation

◆ AT_NUM_PASSES

#define AT_NUM_PASSES (AT_PASS_MISC + 1)

Definition at line 170 of file tablecmds.c.

◆ ATT_COMPOSITE_TYPE

#define ATT_COMPOSITE_TYPE 0x0010

Definition at line 342 of file tablecmds.c.

◆ ATT_FOREIGN_TABLE

#define ATT_FOREIGN_TABLE 0x0020

Definition at line 343 of file tablecmds.c.

◆ ATT_INDEX

#define ATT_INDEX 0x0008

Definition at line 341 of file tablecmds.c.

◆ ATT_MATVIEW

#define ATT_MATVIEW 0x0004

Definition at line 340 of file tablecmds.c.

◆ ATT_PARTITIONED_INDEX

#define ATT_PARTITIONED_INDEX 0x0040

Definition at line 344 of file tablecmds.c.

◆ ATT_PARTITIONED_TABLE

#define ATT_PARTITIONED_TABLE 0x0100

Definition at line 346 of file tablecmds.c.

◆ ATT_SEQUENCE

#define ATT_SEQUENCE 0x0080

Definition at line 345 of file tablecmds.c.

◆ ATT_TABLE

#define ATT_TABLE 0x0001

Definition at line 338 of file tablecmds.c.

◆ ATT_VIEW

#define ATT_VIEW 0x0002

Definition at line 339 of file tablecmds.c.

◆ child_dependency_type

#define child_dependency_type ( child_is_partition ) ((child_is_partition) ? DEPENDENCY_AUTO : DEPENDENCY_NORMAL)

Definition at line 396 of file tablecmds.c.

397 : DEPENDENCY_NORMAL)

DEPENDENCY_NORMAL

@ DEPENDENCY_NORMAL

Definition dependency.h:33

Typedef Documentation

◆ addFkConstraintSides

typedef enum addFkConstraintSides addFkConstraintSides

◆ AlteredTableInfo

typedef struct AlteredTableInfo AlteredTableInfo

◆ AlterTablePass

typedef enum AlterTablePass AlterTablePass

◆ ForeignTruncateInfo

typedef struct ForeignTruncateInfo ForeignTruncateInfo

◆ NewColumnValue

typedef struct NewColumnValue NewColumnValue

◆ NewConstraint

typedef struct NewConstraint NewConstraint

◆ OnCommitItem

typedef struct OnCommitItem OnCommitItem

◆ PartitionIndexExtDepEntry

typedef struct PartitionIndexExtDepEntry PartitionIndexExtDepEntry

◆ SplitPartitionContext

typedef struct SplitPartitionContext SplitPartitionContext

Enumeration Type Documentation

◆ addFkConstraintSides

enum addFkConstraintSides

Enumerator
addFkReferencedSide
addFkReferencingSide
addFkBothSides

Definition at line 363 of file tablecmds.c.

{
    addFkReferencedSide,
    addFkReferencingSide,
    addFkBothSides,
} addFkConstraintSides;

◆ AlterTablePass

enum AlterTablePass

Enumerator
AT_PASS_UNSET
AT_PASS_DROP
AT_PASS_ALTER_TYPE
AT_PASS_ADD_COL
AT_PASS_SET_EXPRESSION
AT_PASS_OLD_INDEX
AT_PASS_OLD_CONSTR
AT_PASS_ADD_CONSTR
AT_PASS_COL_ATTRS
AT_PASS_ADD_INDEXCONSTR
AT_PASS_ADD_INDEX
AT_PASS_ADD_OTHERCONSTR
AT_PASS_MISC

Definition at line 152 of file tablecmds.c.

{
    AT_PASS_UNSET = -1,         /* UNSET will cause ERROR */
    AT_PASS_DROP,               /* DROP (all flavors) */
    AT_PASS_ALTER_TYPE,         /* ALTER COLUMN TYPE */
    AT_PASS_ADD_COL,            /* ADD COLUMN */
    AT_PASS_SET_EXPRESSION,     /* ALTER SET EXPRESSION */
    AT_PASS_OLD_INDEX,          /* re-add existing indexes */
    AT_PASS_OLD_CONSTR,         /* re-add existing constraints */
    /* We could support a RENAME COLUMN pass here, but not currently used */
    AT_PASS_ADD_CONSTR,         /* ADD constraints (initial examination) */
    AT_PASS_COL_ATTRS,          /* set column attributes, eg NOT NULL */
    AT_PASS_ADD_INDEXCONSTR,    /* ADD index-based constraints */
    AT_PASS_ADD_INDEX,          /* ADD indexes */
    AT_PASS_ADD_OTHERCONSTR,    /* ADD other constraints, defaults */
    AT_PASS_MISC,               /* other stuff */
} AlterTablePass;

Function Documentation

◆ add_column_collation_dependency()

static void add_column_collation_dependency	(	Oid	relid,
		int32	attnum,
		Oid	collid
	)

static

Definition at line 7858 of file tablecmds.c.

{
    ObjectAddress myself,
                referenced;
 
    /* We know the default collation is pinned, so don't bother recording it */
    if (OidIsValid(collid) && collid != DEFAULT_COLLATION_OID)
    {
        myself.classId = RelationRelationId;
        myself.objectId = relid;
        myself.objectSubId = attnum;
        referenced.classId = CollationRelationId;
        referenced.objectId = collid;
        referenced.objectSubId = 0;
        recordDependencyOn(&myself, &referenced, DEPENDENCY_NORMAL);
    }
}

References attnum, ObjectAddress::classId, collid, DEPENDENCY_NORMAL, fb(), OidIsValid, and recordDependencyOn().

Referenced by ATExecAddColumn(), and ATExecAlterColumnType().

◆ add_column_datatype_dependency()

static void add_column_datatype_dependency	(	Oid	relid,
		int32	attnum,
		Oid	typid
	)

static

Definition at line 7840 of file tablecmds.c.

{
    ObjectAddress myself,
                referenced;
 
    myself.classId = RelationRelationId;
    myself.objectId = relid;
    myself.objectSubId = attnum;
    referenced.classId = TypeRelationId;
    referenced.objectId = typid;
    referenced.objectSubId = 0;
    recordDependencyOn(&myself, &referenced, DEPENDENCY_NORMAL);
}

References attnum, ObjectAddress::classId, DEPENDENCY_NORMAL, fb(), and recordDependencyOn().

Referenced by ATExecAddColumn(), and ATExecAlterColumnType().

◆ addFkConstraint()

static ObjectAddress addFkConstraint	(	addFkConstraintSides	fkside,
		char *	constraintname,
		Constraint *	fkconstraint,
		Relation	rel,
		Relation	pkrel,
		Oid	indexOid,
		Oid	parentConstr,
		int	numfks,
		int16 *	pkattnum,
		int16 *	fkattnum,
		Oid *	pfeqoperators,
		Oid *	ppeqoperators,
		Oid *	ffeqoperators,
		int	numfkdelsetcols,
		int16 *	fkdelsetcols,
		bool	is_internal,
		bool	with_period
	)

static

Definition at line 10870 of file tablecmds.c.

{
    ObjectAddress address;
    Oid         constrOid;
    char       *conname;
    bool        conislocal;
    int16       coninhcount;
    bool        connoinherit;
 
    /*
     * Verify relkind for each referenced partition.  At the top level, this
     * is redundant with a previous check, but we need it when recursing.
     */
    if (pkrel->rd_rel->relkind != RELKIND_RELATION &&
        pkrel->rd_rel->relkind != RELKIND_PARTITIONED_TABLE)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("referenced relation \"%s\" is not a table",
                        RelationGetRelationName(pkrel))));
 
    /*
     * Caller supplies us with a constraint name; however, it may be used in
     * this partition, so come up with a different one in that case.  Unless
     * truncation to NAMEDATALEN dictates otherwise, the new name will be the
     * supplied name with an underscore and digit(s) appended.
     */
    if (ConstraintNameIsUsed(CONSTRAINT_RELATION,
                             RelationGetRelid(rel),
                             constraintname))
        conname = ChooseConstraintName(constraintname,
                                       NULL,
                                       "",
                                       RelationGetNamespace(rel), NIL);
    else
        conname = constraintname;
 
    if (fkconstraint->conname == NULL)
        fkconstraint->conname = pstrdup(conname);
 
    if (OidIsValid(parentConstr))
    {
        conislocal = false;
        coninhcount = 1;
        connoinherit = false;
    }
    else
    {
        conislocal = true;
        coninhcount = 0;
 
        /*
         * always inherit for partitioned tables, never for legacy inheritance
         */
        connoinherit = rel->rd_rel->relkind != RELKIND_PARTITIONED_TABLE;
    }
 
    /*
     * Record the FK constraint in pg_constraint.
     */
    constrOid = CreateConstraintEntry(conname,
                                      RelationGetNamespace(rel),
                                      CONSTRAINT_FOREIGN,
                                      fkconstraint->deferrable,
                                      fkconstraint->initdeferred,
                                      fkconstraint->is_enforced,
                                      fkconstraint->initially_valid,
                                      parentConstr,
                                      RelationGetRelid(rel),
                                      fkattnum,
                                      numfks,
                                      numfks,
                                      InvalidOid,   /* not a domain constraint */
                                      indexOid,
                                      RelationGetRelid(pkrel),
                                      pkattnum,
                                      pfeqoperators,
                                      ppeqoperators,
                                      ffeqoperators,
                                      numfks,
                                      fkconstraint->fk_upd_action,
                                      fkconstraint->fk_del_action,
                                      fkdelsetcols,
                                      numfkdelsetcols,
                                      fkconstraint->fk_matchtype,
                                      NULL, /* no exclusion constraint */
                                      NULL, /* no check constraint */
                                      NULL,
                                      conislocal,   /* islocal */
                                      coninhcount,  /* inhcount */
                                      connoinherit, /* conNoInherit */
                                      with_period,  /* conPeriod */
                                      is_internal); /* is_internal */
 
    ObjectAddressSet(address, ConstraintRelationId, constrOid);
 
    /*
     * In partitioning cases, create the dependency entries for this
     * constraint.  (For non-partitioned cases, relevant entries were created
     * by CreateConstraintEntry.)
     *
     * On the referenced side, we need the constraint to have an internal
     * dependency on its parent constraint; this means that this constraint
     * cannot be dropped on its own -- only through the parent constraint. It
     * also means the containing partition cannot be dropped on its own, but
     * it can be detached, at which point this dependency is removed (after
     * verifying that no rows are referenced via this FK.)
     *
     * When processing the referencing side, we link the constraint via the
     * special partitioning dependencies: the parent constraint is the primary
     * dependent, and the partition on which the foreign key exists is the
     * secondary dependency.  That way, this constraint is dropped if either
     * of these objects is.
     *
     * Note that this is only necessary for the subsidiary pg_constraint rows
     * in partitions; the topmost row doesn't need any of this.
     */
    if (OidIsValid(parentConstr))
    {
        ObjectAddress referenced;
 
        ObjectAddressSet(referenced, ConstraintRelationId, parentConstr);
 
        Assert(fkside != addFkBothSides);
        if (fkside == addFkReferencedSide)
            recordDependencyOn(&address, &referenced, DEPENDENCY_INTERNAL);
        else
        {
            recordDependencyOn(&address, &referenced, DEPENDENCY_PARTITION_PRI);
            ObjectAddressSet(referenced, RelationRelationId, RelationGetRelid(rel));
            recordDependencyOn(&address, &referenced, DEPENDENCY_PARTITION_SEC);
        }
    }
 
    /* make new constraint visible, in case we add more */
    CommandCounterIncrement();
 
    return address;
}

References addFkBothSides, addFkReferencedSide, Assert, ChooseConstraintName(), CommandCounterIncrement(), CONSTRAINT_RELATION, ConstraintNameIsUsed(), CreateConstraintEntry(), DEPENDENCY_INTERNAL, DEPENDENCY_PARTITION_PRI, DEPENDENCY_PARTITION_SEC, ereport, errcode(), errmsg, ERROR, fb(), InvalidOid, NIL, ObjectAddressSet, OidIsValid, pstrdup(), RelationData::rd_rel, recordDependencyOn(), RelationGetNamespace, RelationGetRelationName, and RelationGetRelid.

Referenced by addFkRecurseReferenced(), addFkRecurseReferencing(), ATAddForeignKeyConstraint(), CloneFkReferenced(), and CloneFkReferencing().

◆ addFkRecurseReferenced()

static void addFkRecurseReferenced	(	Constraint *	fkconstraint,
		Relation	rel,
		Relation	pkrel,
		Oid	indexOid,
		Oid	parentConstr,
		int	numfks,
		int16 *	pkattnum,
		int16 *	fkattnum,
		Oid *	pfeqoperators,
		Oid *	ppeqoperators,
		Oid *	ffeqoperators,
		int	numfkdelsetcols,
		int16 *	fkdelsetcols,
		bool	old_check_ok,
		Oid	parentDelTrigger,
		Oid	parentUpdTrigger,
		bool	with_period
	)

static

Definition at line 11048 of file tablecmds.c.

{
    Oid         deleteTriggerOid = InvalidOid,
                updateTriggerOid = InvalidOid;
 
    Assert(CheckRelationLockedByMe(pkrel, ShareRowExclusiveLock, true));
    Assert(CheckRelationLockedByMe(rel, ShareRowExclusiveLock, true));
 
    /*
     * Create action triggers to enforce the constraint, or skip them if the
     * constraint is NOT ENFORCED.
     */
    if (fkconstraint->is_enforced)
        createForeignKeyActionTriggers(RelationGetRelid(rel),
                                       RelationGetRelid(pkrel),
                                       fkconstraint,
                                       parentConstr, indexOid,
                                       parentDelTrigger, parentUpdTrigger,
                                       &deleteTriggerOid, &updateTriggerOid);
 
    /*
     * If the referenced table is partitioned, recurse on ourselves to handle
     * each partition.  We need one pg_constraint row created for each
     * partition in addition to the pg_constraint row for the parent table.
     */
    if (pkrel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
    {
        PartitionDesc pd = RelationGetPartitionDesc(pkrel, true);
 
        for (int i = 0; i < pd->nparts; i++)
        {
            Relation    partRel;
            AttrMap    *map;
            AttrNumber *mapped_pkattnum;
            Oid         partIndexId;
            ObjectAddress address;
 
            /* XXX would it be better to acquire these locks beforehand? */
            partRel = table_open(pd->oids[i], ShareRowExclusiveLock);
 
            /*
             * Map the attribute numbers in the referenced side of the FK
             * definition to match the partition's column layout.
             */
            map = build_attrmap_by_name_if_req(RelationGetDescr(partRel),
                                               RelationGetDescr(pkrel),
                                               false);
            if (map)
            {
                mapped_pkattnum = palloc_array(AttrNumber, numfks);
                for (int j = 0; j < numfks; j++)
                    mapped_pkattnum[j] = map->attnums[pkattnum[j] - 1];
            }
            else
                mapped_pkattnum = pkattnum;
 
            /* Determine the index to use at this level */
            partIndexId = index_get_partition(partRel, indexOid);
            if (!OidIsValid(partIndexId))
                elog(ERROR, "index for %u not found in partition %s",
                     indexOid, RelationGetRelationName(partRel));
 
            /* Create entry at this level ... */
            address = addFkConstraint(addFkReferencedSide,
                                      fkconstraint->conname, fkconstraint, rel,
                                      partRel, partIndexId, parentConstr,
                                      numfks, mapped_pkattnum,
                                      fkattnum, pfeqoperators, ppeqoperators,
                                      ffeqoperators, numfkdelsetcols,
                                      fkdelsetcols, true, with_period);
            /* ... and recurse to our children */
            addFkRecurseReferenced(fkconstraint, rel, partRel,
                                   partIndexId, address.objectId, numfks,
                                   mapped_pkattnum, fkattnum,
                                   pfeqoperators, ppeqoperators, ffeqoperators,
                                   numfkdelsetcols, fkdelsetcols,
                                   old_check_ok,
                                   deleteTriggerOid, updateTriggerOid,
                                   with_period);
 
            /* Done -- clean up (but keep the lock) */
            table_close(partRel, NoLock);
            if (map)
            {
                pfree(mapped_pkattnum);
                free_attrmap(map);
            }
        }
    }
}

References addFkConstraint(), addFkRecurseReferenced(), addFkReferencedSide, Assert, AttrMap::attnums, build_attrmap_by_name_if_req(), CheckRelationLockedByMe(), createForeignKeyActionTriggers(), elog, ERROR, fb(), free_attrmap(), i, index_get_partition(), InvalidOid, j, NoLock, PartitionDescData::nparts, ObjectAddress::objectId, OidIsValid, PartitionDescData::oids, palloc_array, pfree(), RelationGetDescr, RelationGetPartitionDesc(), RelationGetRelationName, RelationGetRelid, ShareRowExclusiveLock, table_close(), and table_open().

Referenced by addFkRecurseReferenced(), ATAddForeignKeyConstraint(), CloneFkReferenced(), and DetachPartitionFinalize().

◆ addFkRecurseReferencing()

static void addFkRecurseReferencing	(	List **	wqueue,
		Constraint *	fkconstraint,
		Relation	rel,
		Relation	pkrel,
		Oid	indexOid,
		Oid	parentConstr,
		int	numfks,
		int16 *	pkattnum,
		int16 *	fkattnum,
		Oid *	pfeqoperators,
		Oid *	ppeqoperators,
		Oid *	ffeqoperators,
		int	numfkdelsetcols,
		int16 *	fkdelsetcols,
		bool	old_check_ok,
		LOCKMODE	lockmode,
		Oid	parentInsTrigger,
		Oid	parentUpdTrigger,
		bool	with_period
	)

static

Definition at line 11186 of file tablecmds.c.

{
    Oid         insertTriggerOid = InvalidOid,
                updateTriggerOid = InvalidOid;
 
    Assert(OidIsValid(parentConstr));
    Assert(CheckRelationLockedByMe(rel, ShareRowExclusiveLock, true));
    Assert(CheckRelationLockedByMe(pkrel, ShareRowExclusiveLock, true));
 
    if (rel->rd_rel->relkind == RELKIND_FOREIGN_TABLE)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("foreign key constraints are not supported on foreign tables")));
 
    /*
     * Add check triggers if the constraint is ENFORCED, and if needed,
     * schedule them to be checked in Phase 3.
     *
     * If the relation is partitioned, drill down to do it to its partitions.
     */
    if (fkconstraint->is_enforced)
        createForeignKeyCheckTriggers(RelationGetRelid(rel),
                                      RelationGetRelid(pkrel),
                                      fkconstraint,
                                      parentConstr,
                                      indexOid,
                                      parentInsTrigger, parentUpdTrigger,
                                      &insertTriggerOid, &updateTriggerOid);
 
    if (rel->rd_rel->relkind == RELKIND_RELATION)
    {
        /*
         * Tell Phase 3 to check that the constraint is satisfied by existing
         * rows. We can skip this during table creation, when constraint is
         * specified as NOT ENFORCED, or when requested explicitly by
         * specifying NOT VALID in an ADD FOREIGN KEY command, and when we're
         * recreating a constraint following a SET DATA TYPE operation that
         * did not impugn its validity.
         */
        if (wqueue && !old_check_ok && !fkconstraint->skip_validation &&
            fkconstraint->is_enforced)
        {
            NewConstraint *newcon;
            AlteredTableInfo *tab;
 
            tab = ATGetQueueEntry(wqueue, rel);
 
            newcon = palloc0_object(NewConstraint);
            newcon->name = get_constraint_name(parentConstr);
            newcon->contype = CONSTR_FOREIGN;
            newcon->refrelid = RelationGetRelid(pkrel);
            newcon->refindid = indexOid;
            newcon->conid = parentConstr;
            newcon->conwithperiod = fkconstraint->fk_with_period;
            newcon->qual = (Node *) fkconstraint;
 
            tab->constraints = lappend(tab->constraints, newcon);
        }
    }
    else if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
    {
        PartitionDesc pd = RelationGetPartitionDesc(rel, true);
        Relation    trigrel;
 
        /*
         * Triggers of the foreign keys will be manipulated a bunch of times
         * in the loop below.  To avoid repeatedly opening/closing the trigger
         * catalog relation, we open it here and pass it to the subroutines
         * called below.
         */
        trigrel = table_open(TriggerRelationId, RowExclusiveLock);
 
        /*
         * Recurse to take appropriate action on each partition; either we
         * find an existing constraint to reparent to ours, or we create a new
         * one.
         */
        for (int i = 0; i < pd->nparts; i++)
        {
            Relation    partition = table_open(pd->oids[i], lockmode);
            List       *partFKs;
            AttrMap    *attmap;
            AttrNumber  mapped_fkattnum[INDEX_MAX_KEYS];
            bool        attached;
            ObjectAddress address;
 
            CheckAlterTableIsSafe(partition);
 
            attmap = build_attrmap_by_name(RelationGetDescr(partition),
                                           RelationGetDescr(rel),
                                           false);
            for (int j = 0; j < numfks; j++)
                mapped_fkattnum[j] = attmap->attnums[fkattnum[j] - 1];
 
            /* Check whether an existing constraint can be repurposed */
            partFKs = copyObject(RelationGetFKeyList(partition));
            attached = false;
            foreach_node(ForeignKeyCacheInfo, fk, partFKs)
            {
                if (tryAttachPartitionForeignKey(wqueue,
                                                 fk,
                                                 partition,
                                                 parentConstr,
                                                 numfks,
                                                 mapped_fkattnum,
                                                 pkattnum,
                                                 pfeqoperators,
                                                 insertTriggerOid,
                                                 updateTriggerOid,
                                                 trigrel))
                {
                    attached = true;
                    break;
                }
            }
            if (attached)
            {
                table_close(partition, NoLock);
                continue;
            }
 
            /*
             * No luck finding a good constraint to reuse; create our own.
             */
            address = addFkConstraint(addFkReferencingSide,
                                      fkconstraint->conname, fkconstraint,
                                      partition, pkrel, indexOid, parentConstr,
                                      numfks, pkattnum,
                                      mapped_fkattnum, pfeqoperators,
                                      ppeqoperators, ffeqoperators,
                                      numfkdelsetcols, fkdelsetcols, true,
                                      with_period);
 
            /* call ourselves to finalize the creation and we're done */
            addFkRecurseReferencing(wqueue, fkconstraint, partition, pkrel,
                                    indexOid,
                                    address.objectId,
                                    numfks,
                                    pkattnum,
                                    mapped_fkattnum,
                                    pfeqoperators,
                                    ppeqoperators,
                                    ffeqoperators,
                                    numfkdelsetcols,
                                    fkdelsetcols,
                                    old_check_ok,
                                    lockmode,
                                    insertTriggerOid,
                                    updateTriggerOid,
                                    with_period);
 
            table_close(partition, NoLock);
        }
 
        table_close(trigrel, RowExclusiveLock);
    }
}

References addFkConstraint(), addFkRecurseReferencing(), addFkReferencingSide, Assert, ATGetQueueEntry(), build_attrmap_by_name(), CheckAlterTableIsSafe(), CheckRelationLockedByMe(), CONSTR_FOREIGN, AlteredTableInfo::constraints, copyObject, createForeignKeyCheckTriggers(), ereport, errcode(), errmsg, ERROR, fb(), foreach_node, get_constraint_name(), i, INDEX_MAX_KEYS, InvalidOid, j, lappend(), NoLock, PartitionDescData::nparts, ObjectAddress::objectId, OidIsValid, PartitionDescData::oids, palloc0_object, RelationData::rd_rel, RelationGetDescr, RelationGetFKeyList(), RelationGetPartitionDesc(), RelationGetRelid, RowExclusiveLock, ShareRowExclusiveLock, table_close(), table_open(), and tryAttachPartitionForeignKey().

Referenced by addFkRecurseReferencing(), ATAddForeignKeyConstraint(), and CloneFkReferencing().

◆ alter_table_type_to_string()

static const char * alter_table_type_to_string ( AlterTableType cmdtype )

static

Definition at line 6717 of file tablecmds.c.

{
    switch (cmdtype)
    {
        case AT_AddColumn:
        case AT_AddColumnToView:
            return "ADD COLUMN";
        case AT_ColumnDefault:
        case AT_CookedColumnDefault:
            return "ALTER COLUMN ... SET DEFAULT";
        case AT_DropNotNull:
            return "ALTER COLUMN ... DROP NOT NULL";
        case AT_SetNotNull:
            return "ALTER COLUMN ... SET NOT NULL";
        case AT_SetExpression:
            return "ALTER COLUMN ... SET EXPRESSION";
        case AT_DropExpression:
            return "ALTER COLUMN ... DROP EXPRESSION";
        case AT_SetStatistics:
            return "ALTER COLUMN ... SET STATISTICS";
        case AT_SetOptions:
            return "ALTER COLUMN ... SET";
        case AT_ResetOptions:
            return "ALTER COLUMN ... RESET";
        case AT_SetStorage:
            return "ALTER COLUMN ... SET STORAGE";
        case AT_SetCompression:
            return "ALTER COLUMN ... SET COMPRESSION";
        case AT_DropColumn:
            return "DROP COLUMN";
        case AT_AddIndex:
        case AT_ReAddIndex:
            return NULL;        /* not real grammar */
        case AT_AddConstraint:
        case AT_ReAddConstraint:
        case AT_ReAddDomainConstraint:
        case AT_AddIndexConstraint:
            return "ADD CONSTRAINT";
        case AT_AlterConstraint:
            return "ALTER CONSTRAINT";
        case AT_ValidateConstraint:
            return "VALIDATE CONSTRAINT";
        case AT_DropConstraint:
            return "DROP CONSTRAINT";
        case AT_ReAddComment:
            return NULL;        /* not real grammar */
        case AT_AlterColumnType:
            return "ALTER COLUMN ... SET DATA TYPE";
        case AT_AlterColumnGenericOptions:
            return "ALTER COLUMN ... OPTIONS";
        case AT_ChangeOwner:
            return "OWNER TO";
        case AT_ClusterOn:
            return "CLUSTER ON";
        case AT_DropCluster:
            return "SET WITHOUT CLUSTER";
        case AT_SetAccessMethod:
            return "SET ACCESS METHOD";
        case AT_SetLogged:
            return "SET LOGGED";
        case AT_SetUnLogged:
            return "SET UNLOGGED";
        case AT_DropOids:
            return "SET WITHOUT OIDS";
        case AT_SetTableSpace:
            return "SET TABLESPACE";
        case AT_SetRelOptions:
            return "SET";
        case AT_ResetRelOptions:
            return "RESET";
        case AT_ReplaceRelOptions:
            return NULL;        /* not real grammar */
        case AT_EnableTrig:
            return "ENABLE TRIGGER";
        case AT_EnableAlwaysTrig:
            return "ENABLE ALWAYS TRIGGER";
        case AT_EnableReplicaTrig:
            return "ENABLE REPLICA TRIGGER";
        case AT_DisableTrig:
            return "DISABLE TRIGGER";
        case AT_EnableTrigAll:
            return "ENABLE TRIGGER ALL";
        case AT_DisableTrigAll:
            return "DISABLE TRIGGER ALL";
        case AT_EnableTrigUser:
            return "ENABLE TRIGGER USER";
        case AT_DisableTrigUser:
            return "DISABLE TRIGGER USER";
        case AT_EnableRule:
            return "ENABLE RULE";
        case AT_EnableAlwaysRule:
            return "ENABLE ALWAYS RULE";
        case AT_EnableReplicaRule:
            return "ENABLE REPLICA RULE";
        case AT_DisableRule:
            return "DISABLE RULE";
        case AT_AddInherit:
            return "INHERIT";
        case AT_DropInherit:
            return "NO INHERIT";
        case AT_AddOf:
            return "OF";
        case AT_DropOf:
            return "NOT OF";
        case AT_ReplicaIdentity:
            return "REPLICA IDENTITY";
        case AT_EnableRowSecurity:
            return "ENABLE ROW SECURITY";
        case AT_DisableRowSecurity:
            return "DISABLE ROW SECURITY";
        case AT_ForceRowSecurity:
            return "FORCE ROW SECURITY";
        case AT_NoForceRowSecurity:
            return "NO FORCE ROW SECURITY";
        case AT_GenericOptions:
            return "OPTIONS";
        case AT_AttachPartition:
            return "ATTACH PARTITION";
        case AT_DetachPartition:
            return "DETACH PARTITION";
        case AT_DetachPartitionFinalize:
            return "DETACH PARTITION ... FINALIZE";
        case AT_MergePartitions:
            return "MERGE PARTITIONS";
        case AT_SplitPartition:
            return "SPLIT PARTITION";
        case AT_AddIdentity:
            return "ALTER COLUMN ... ADD IDENTITY";
        case AT_SetIdentity:
            return "ALTER COLUMN ... SET";
        case AT_DropIdentity:
            return "ALTER COLUMN ... DROP IDENTITY";
        case AT_ReAddStatistics:
            return NULL;        /* not real grammar */
    }
 
    return NULL;
}

References AT_AddColumn, AT_AddColumnToView, AT_AddConstraint, AT_AddIdentity, AT_AddIndex, AT_AddIndexConstraint, AT_AddInherit, AT_AddOf, AT_AlterColumnGenericOptions, AT_AlterColumnType, AT_AlterConstraint, AT_AttachPartition, AT_ChangeOwner, AT_ClusterOn, AT_ColumnDefault, AT_CookedColumnDefault, AT_DetachPartition, AT_DetachPartitionFinalize, AT_DisableRowSecurity, AT_DisableRule, AT_DisableTrig, AT_DisableTrigAll, AT_DisableTrigUser, AT_DropCluster, AT_DropColumn, AT_DropConstraint, AT_DropExpression, AT_DropIdentity, AT_DropInherit, AT_DropNotNull, AT_DropOf, AT_DropOids, AT_EnableAlwaysRule, AT_EnableAlwaysTrig, AT_EnableReplicaRule, AT_EnableReplicaTrig, AT_EnableRowSecurity, AT_EnableRule, AT_EnableTrig, AT_EnableTrigAll, AT_EnableTrigUser, AT_ForceRowSecurity, AT_GenericOptions, AT_MergePartitions, AT_NoForceRowSecurity, AT_ReAddComment, AT_ReAddConstraint, AT_ReAddDomainConstraint, AT_ReAddIndex, AT_ReAddStatistics, AT_ReplaceRelOptions, AT_ReplicaIdentity, AT_ResetOptions, AT_ResetRelOptions, AT_SetAccessMethod, AT_SetCompression, AT_SetExpression, AT_SetIdentity, AT_SetLogged, AT_SetNotNull, AT_SetOptions, AT_SetRelOptions, AT_SetStatistics, AT_SetStorage, AT_SetTableSpace, AT_SetUnLogged, AT_SplitPartition, AT_ValidateConstraint, and fb().

Referenced by ATSimplePermissions().

◆ AlterCheckConstrEnforceabilityRecurse()

static void AlterCheckConstrEnforceabilityRecurse	(	List **	wqueue,
		ATAlterConstraint *	cmdcon,
		Relation	conrel,
		Oid	conrelid,
		bool	recurse,
		bool	recursing,
		List *	changing_conids,
		LOCKMODE	lockmode
	)

static

Definition at line 12916 of file tablecmds.c.

{
    SysScanDesc pscan;
    HeapTuple   childtup;
    ScanKeyData skey[3];
 
    ScanKeyInit(&skey[0],
                Anum_pg_constraint_conrelid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(conrelid));
    ScanKeyInit(&skey[1],
                Anum_pg_constraint_contypid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(InvalidOid));
    ScanKeyInit(&skey[2],
                Anum_pg_constraint_conname,
                BTEqualStrategyNumber, F_NAMEEQ,
                CStringGetDatum(cmdcon->conname));
 
    pscan = systable_beginscan(conrel, ConstraintRelidTypidNameIndexId, true,
                               NULL, 3, skey);
 
    if (!HeapTupleIsValid(childtup = systable_getnext(pscan)))
        ereport(ERROR,
                errcode(ERRCODE_UNDEFINED_OBJECT),
                errmsg("constraint \"%s\" of relation \"%s\" does not exist",
                       cmdcon->conname, get_rel_name(conrelid)));
 
    ATExecAlterCheckConstrEnforceability(wqueue, cmdcon, conrel, childtup,
                                         recurse, recursing, changing_conids,
                                         lockmode);
 
    systable_endscan(pscan);
}

References ATExecAlterCheckConstrEnforceability(), BTEqualStrategyNumber, CStringGetDatum(), ereport, errcode(), errmsg, ERROR, fb(), get_rel_name(), HeapTupleIsValid, InvalidOid, ObjectIdGetDatum(), ScanKeyInit(), systable_beginscan(), systable_endscan(), and systable_getnext().

Referenced by ATExecAlterCheckConstrEnforceability().

◆ AlterConstrDeferrabilityRecurse()

static void AlterConstrDeferrabilityRecurse	(	List **	wqueue,
		ATAlterConstraint *	cmdcon,
		Relation	conrel,
		Relation	tgrel,
		Relation	rel,
		HeapTuple	contuple,
		bool	recurse,
		List **	otherrelids,
		LOCKMODE	lockmode
	)

static

Definition at line 13346 of file tablecmds.c.

{
    Form_pg_constraint currcon;
    Oid         conoid;
    ScanKeyData pkey;
    SysScanDesc pscan;
    HeapTuple   childtup;
 
    currcon = (Form_pg_constraint) GETSTRUCT(contuple);
    conoid = currcon->oid;
 
    ScanKeyInit(&pkey,
                Anum_pg_constraint_conparentid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(conoid));
 
    pscan = systable_beginscan(conrel, ConstraintParentIndexId,
                               true, NULL, 1, &pkey);
 
    while (HeapTupleIsValid(childtup = systable_getnext(pscan)))
    {
        Form_pg_constraint childcon = (Form_pg_constraint) GETSTRUCT(childtup);
        Relation    childrel;
 
        childrel = table_open(childcon->conrelid, lockmode);
 
        ATExecAlterConstrDeferrability(wqueue, cmdcon, conrel, tgrel, childrel,
                                       childtup, recurse, otherrelids, lockmode);
        table_close(childrel, NoLock);
    }
 
    systable_endscan(pscan);
}

References ATExecAlterConstrDeferrability(), BTEqualStrategyNumber, fb(), Form_pg_constraint, GETSTRUCT(), HeapTupleIsValid, NoLock, ObjectIdGetDatum(), ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by ATExecAlterConstrDeferrability().

◆ AlterConstrTriggerDeferrability()

static void AlterConstrTriggerDeferrability	(	Oid	conoid,
		Relation	tgrel,
		Relation	rel,
		bool	deferrable,
		bool	initdeferred,
		List **	otherrelids
	)

static

Definition at line 13228 of file tablecmds.c.

{
    HeapTuple   tgtuple;
    ScanKeyData tgkey;
    SysScanDesc tgscan;
 
    ScanKeyInit(&tgkey,
                Anum_pg_trigger_tgconstraint,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(conoid));
    tgscan = systable_beginscan(tgrel, TriggerConstraintIndexId, true,
                                NULL, 1, &tgkey);
    while (HeapTupleIsValid(tgtuple = systable_getnext(tgscan)))
    {
        Form_pg_trigger tgform = (Form_pg_trigger) GETSTRUCT(tgtuple);
        Form_pg_trigger copy_tg;
        HeapTuple   tgCopyTuple;
 
        /*
         * Remember OIDs of other relation(s) involved in FK constraint.
         * (Note: it's likely that we could skip forcing a relcache inval for
         * other rels that don't have a trigger whose properties change, but
         * let's be conservative.)
         */
        if (tgform->tgrelid != RelationGetRelid(rel))
            *otherrelids = list_append_unique_oid(*otherrelids,
                                                  tgform->tgrelid);
 
        /*
         * Update enable status and deferrability of RI_FKey_noaction_del,
         * RI_FKey_noaction_upd, RI_FKey_check_ins and RI_FKey_check_upd
         * triggers, but not others; see createForeignKeyActionTriggers and
         * CreateFKCheckTrigger.
         */
        if (tgform->tgfoid != F_RI_FKEY_NOACTION_DEL &&
            tgform->tgfoid != F_RI_FKEY_NOACTION_UPD &&
            tgform->tgfoid != F_RI_FKEY_CHECK_INS &&
            tgform->tgfoid != F_RI_FKEY_CHECK_UPD)
            continue;
 
        tgCopyTuple = heap_copytuple(tgtuple);
        copy_tg = (Form_pg_trigger) GETSTRUCT(tgCopyTuple);
 
        copy_tg->tgdeferrable = deferrable;
        copy_tg->tginitdeferred = initdeferred;
        CatalogTupleUpdate(tgrel, &tgCopyTuple->t_self, tgCopyTuple);
 
        InvokeObjectPostAlterHook(TriggerRelationId, tgform->oid, 0);
 
        heap_freetuple(tgCopyTuple);
    }
 
    systable_endscan(tgscan);
}

References BTEqualStrategyNumber, CatalogTupleUpdate(), fb(), Form_pg_trigger, GETSTRUCT(), heap_copytuple(), heap_freetuple(), HeapTupleIsValid, InvokeObjectPostAlterHook, list_append_unique_oid(), ObjectIdGetDatum(), RelationGetRelid, ScanKeyInit(), systable_beginscan(), systable_endscan(), and systable_getnext().

Referenced by ATExecAlterConstrDeferrability().

◆ AlterConstrUpdateConstraintEntry()

static void AlterConstrUpdateConstraintEntry	(	ATAlterConstraint *	cmdcon,
		Relation	conrel,
		HeapTuple	contuple
	)

static

Definition at line 13388 of file tablecmds.c.

{
    HeapTuple   copyTuple;
    Form_pg_constraint copy_con;
 
    Assert(cmdcon->alterEnforceability || cmdcon->alterDeferrability ||
           cmdcon->alterInheritability);
 
    copyTuple = heap_copytuple(contuple);
    copy_con = (Form_pg_constraint) GETSTRUCT(copyTuple);
 
    if (cmdcon->alterEnforceability)
    {
        copy_con->conenforced = cmdcon->is_enforced;
 
        /*
         * NB: The convalidated status is irrelevant when the constraint is
         * set to NOT ENFORCED, but for consistency, it should still be set
         * appropriately. Similarly, if the constraint is later changed to
         * ENFORCED, validation will be performed during phase 3, so it makes
         * sense to mark it as valid in that case.
         */
        copy_con->convalidated = cmdcon->is_enforced;
    }
    if (cmdcon->alterDeferrability)
    {
        copy_con->condeferrable = cmdcon->deferrable;
        copy_con->condeferred = cmdcon->initdeferred;
    }
    if (cmdcon->alterInheritability)
        copy_con->connoinherit = cmdcon->noinherit;
 
    CatalogTupleUpdate(conrel, &copyTuple->t_self, copyTuple);
    InvokeObjectPostAlterHook(ConstraintRelationId, copy_con->oid, 0);
 
    /* Make new constraint flags visible to others */
    CacheInvalidateRelcacheByRelid(copy_con->conrelid);
 
    heap_freetuple(copyTuple);
}

References Assert, CacheInvalidateRelcacheByRelid(), CatalogTupleUpdate(), fb(), Form_pg_constraint, GETSTRUCT(), heap_copytuple(), heap_freetuple(), and InvokeObjectPostAlterHook.

Referenced by ATExecAlterCheckConstrEnforceability(), ATExecAlterConstrDeferrability(), ATExecAlterConstrInheritability(), and ATExecAlterFKConstrEnforceability().

◆ AlterFKConstrEnforceabilityRecurse()

static void AlterFKConstrEnforceabilityRecurse	(	List **	wqueue,
		ATAlterConstraint *	cmdcon,
		Relation	conrel,
		Relation	tgrel,
		Oid	fkrelid,
		Oid	pkrelid,
		HeapTuple	contuple,
		LOCKMODE	lockmode,
		Oid	ReferencedParentDelTrigger,
		Oid	ReferencedParentUpdTrigger,
		Oid	ReferencingParentInsTrigger,
		Oid	ReferencingParentUpdTrigger
	)

static

Definition at line 13297 of file tablecmds.c.

{
    Form_pg_constraint currcon;
    Oid         conoid;
    ScanKeyData pkey;
    SysScanDesc pscan;
    HeapTuple   childtup;
 
    currcon = (Form_pg_constraint) GETSTRUCT(contuple);
    conoid = currcon->oid;
 
    ScanKeyInit(&pkey,
                Anum_pg_constraint_conparentid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(conoid));
 
    pscan = systable_beginscan(conrel, ConstraintParentIndexId,
                               true, NULL, 1, &pkey);
 
    while (HeapTupleIsValid(childtup = systable_getnext(pscan)))
        ATExecAlterFKConstrEnforceability(wqueue, cmdcon, conrel, tgrel, fkrelid,
                                          pkrelid, childtup, lockmode,
                                          ReferencedParentDelTrigger,
                                          ReferencedParentUpdTrigger,
                                          ReferencingParentInsTrigger,
                                          ReferencingParentUpdTrigger);
 
    systable_endscan(pscan);
}

References ATExecAlterFKConstrEnforceability(), BTEqualStrategyNumber, fb(), Form_pg_constraint, GETSTRUCT(), HeapTupleIsValid, ObjectIdGetDatum(), ScanKeyInit(), systable_beginscan(), systable_endscan(), and systable_getnext().

Referenced by ATExecAlterFKConstrEnforceability().

◆ AlterIndexNamespaces()

static void AlterIndexNamespaces	(	Relation	classRel,
		Relation	rel,
		Oid	oldNspOid,
		Oid	newNspOid,
		ObjectAddresses *	objsMoved
	)

static

Definition at line 19834 of file tablecmds.c.

{
    List       *indexList;
    ListCell   *l;
 
    indexList = RelationGetIndexList(rel);
 
    foreach(l, indexList)
    {
        Oid         indexOid = lfirst_oid(l);
        ObjectAddress thisobj;
 
        thisobj.classId = RelationRelationId;
        thisobj.objectId = indexOid;
        thisobj.objectSubId = 0;
 
        /*
         * Note: currently, the index will not have its own dependency on the
         * namespace, so we don't need to do changeDependencyFor(). There's no
         * row type in pg_type, either.
         *
         * XXX this objsMoved test may be pointless -- surely we have a single
         * dependency link from a relation to each index?
         */
        if (!object_address_present(&thisobj, objsMoved))
        {
            AlterRelationNamespaceInternal(classRel, indexOid,
                                           oldNspOid, newNspOid,
                                           false, objsMoved);
            add_exact_object_address(&thisobj, objsMoved);
        }
    }
 
    list_free(indexList);
}

References add_exact_object_address(), AlterRelationNamespaceInternal(), ObjectAddress::classId, fb(), lfirst_oid, list_free(), object_address_present(), and RelationGetIndexList().

Referenced by AlterTableNamespaceInternal().

◆ AlterRelationNamespaceInternal()

void AlterRelationNamespaceInternal	(	Relation	classRel,
		Oid	relOid,
		Oid	oldNspOid,
		Oid	newNspOid,
		bool	hasDependEntry,
		ObjectAddresses *	objsMoved
	)

Definition at line 19757 of file tablecmds.c.

{
    HeapTuple   classTup;
    Form_pg_class classForm;
    ObjectAddress thisobj;
    bool        already_done = false;
 
    /* no rel lock for relkind=c so use LOCKTAG_TUPLE */
    classTup = SearchSysCacheLockedCopy1(RELOID, ObjectIdGetDatum(relOid));
    if (!HeapTupleIsValid(classTup))
        elog(ERROR, "cache lookup failed for relation %u", relOid);
    classForm = (Form_pg_class) GETSTRUCT(classTup);
 
    Assert(classForm->relnamespace == oldNspOid);
 
    thisobj.classId = RelationRelationId;
    thisobj.objectId = relOid;
    thisobj.objectSubId = 0;
 
    /*
     * If the object has already been moved, don't move it again.  If it's
     * already in the right place, don't move it, but still fire the object
     * access hook.
     */
    already_done = object_address_present(&thisobj, objsMoved);
    if (!already_done && oldNspOid != newNspOid)
    {
        ItemPointerData otid = classTup->t_self;
 
        /* check for duplicate name (more friendly than unique-index failure) */
        if (get_relname_relid(NameStr(classForm->relname),
                              newNspOid) != InvalidOid)
            ereport(ERROR,
                    (errcode(ERRCODE_DUPLICATE_TABLE),
                     errmsg("relation \"%s\" already exists in schema \"%s\"",
                            NameStr(classForm->relname),
                            get_namespace_name(newNspOid))));
 
        /* classTup is a copy, so OK to scribble on */
        classForm->relnamespace = newNspOid;
 
        CatalogTupleUpdate(classRel, &otid, classTup);
        UnlockTuple(classRel, &otid, InplaceUpdateTupleLock);
 
 
        /* Update dependency on schema if caller said so */
        if (hasDependEntry &&
            changeDependencyFor(RelationRelationId,
                                relOid,
                                NamespaceRelationId,
                                oldNspOid,
                                newNspOid) != 1)
            elog(ERROR, "could not change schema dependency for relation \"%s\"",
                 NameStr(classForm->relname));
    }
    else
        UnlockTuple(classRel, &classTup->t_self, InplaceUpdateTupleLock);
    if (!already_done)
    {
        add_exact_object_address(&thisobj, objsMoved);
 
        InvokeObjectPostAlterHook(RelationRelationId, relOid, 0);
    }
 
    heap_freetuple(classTup);
}

References add_exact_object_address(), Assert, CatalogTupleUpdate(), changeDependencyFor(), elog, ereport, errcode(), errmsg, ERROR, fb(), get_namespace_name(), get_relname_relid(), GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InplaceUpdateTupleLock, InvalidOid, InvokeObjectPostAlterHook, NameStr, object_address_present(), ObjectIdGetDatum(), SearchSysCacheLockedCopy1(), and UnlockTuple().

Referenced by AlterIndexNamespaces(), AlterSeqNamespaces(), AlterTableNamespaceInternal(), and AlterTypeNamespaceInternal().

◆ AlterSeqNamespaces()

static void AlterSeqNamespaces	(	Relation	classRel,
		Relation	rel,
		Oid	oldNspOid,
		Oid	newNspOid,
		ObjectAddresses *	objsMoved,
		LOCKMODE	lockmode
	)

static

Definition at line 19879 of file tablecmds.c.

{
    Relation    depRel;
    SysScanDesc scan;
    ScanKeyData key[2];
    HeapTuple   tup;
 
    /*
     * SERIAL sequences are those having an auto dependency on one of the
     * table's columns (we don't care *which* column, exactly).
     */
    depRel = table_open(DependRelationId, AccessShareLock);
 
    ScanKeyInit(&key[0],
                Anum_pg_depend_refclassid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationRelationId));
    ScanKeyInit(&key[1],
                Anum_pg_depend_refobjid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationGetRelid(rel)));
    /* we leave refobjsubid unspecified */
 
    scan = systable_beginscan(depRel, DependReferenceIndexId, true,
                              NULL, 2, key);
 
    while (HeapTupleIsValid(tup = systable_getnext(scan)))
    {
        Form_pg_depend depForm = (Form_pg_depend) GETSTRUCT(tup);
        Relation    seqRel;
 
        /* skip dependencies other than auto dependencies on columns */
        if (depForm->refobjsubid == 0 ||
            depForm->classid != RelationRelationId ||
            depForm->objsubid != 0 ||
            !(depForm->deptype == DEPENDENCY_AUTO || depForm->deptype == DEPENDENCY_INTERNAL))
            continue;
 
        /* Use relation_open just in case it's an index */
        seqRel = relation_open(depForm->objid, lockmode);
 
        /* skip non-sequence relations */
        if (RelationGetForm(seqRel)->relkind != RELKIND_SEQUENCE)
        {
            /* No need to keep the lock */
            relation_close(seqRel, lockmode);
            continue;
        }
 
        /* Fix the pg_class and pg_depend entries */
        AlterRelationNamespaceInternal(classRel, depForm->objid,
                                       oldNspOid, newNspOid,
                                       true, objsMoved);
 
        /*
         * Sequences used to have entries in pg_type, but no longer do.  If we
         * ever re-instate that, we'll need to move the pg_type entry to the
         * new namespace, too (using AlterTypeNamespaceInternal).
         */
        Assert(RelationGetForm(seqRel)->reltype == InvalidOid);
 
        /* Now we can close it.  Keep the lock till end of transaction. */
        relation_close(seqRel, NoLock);
    }
 
    systable_endscan(scan);
 
    relation_close(depRel, AccessShareLock);
}

References AccessShareLock, AlterRelationNamespaceInternal(), Assert, BTEqualStrategyNumber, DEPENDENCY_AUTO, DEPENDENCY_INTERNAL, fb(), Form_pg_depend, GETSTRUCT(), HeapTupleIsValid, InvalidOid, NoLock, ObjectIdGetDatum(), relation_close(), relation_open(), RelationGetForm, RelationGetRelid, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), and table_open().

Referenced by AlterTableNamespaceInternal().

◆ AlterTable()

void AlterTable	(	AlterTableStmt *	stmt,
		LOCKMODE	lockmode,
		AlterTableUtilityContext *	context
	)

Definition at line 4626 of file tablecmds.c.

{
    Relation    rel;
 
    /* Caller is required to provide an adequate lock. */
    rel = relation_open(context->relid, NoLock);
 
    CheckAlterTableIsSafe(rel);
 
    ATController(stmt, rel, stmt->cmds, stmt->relation->inh, lockmode, context);
}

References ATController(), CheckAlterTableIsSafe(), NoLock, relation_open(), AlterTableUtilityContext::relid, and stmt.

Referenced by ProcessUtilitySlow().

◆ AlterTableGetLockLevel()

LOCKMODE AlterTableGetLockLevel ( List * cmds )

Definition at line 4700 of file tablecmds.c.

{
    /*
     * This only works if we read catalog tables using MVCC snapshots.
     */
    ListCell   *lcmd;
    LOCKMODE    lockmode = ShareUpdateExclusiveLock;
 
    foreach(lcmd, cmds)
    {
        AlterTableCmd *cmd = (AlterTableCmd *) lfirst(lcmd);
        LOCKMODE    cmd_lockmode = AccessExclusiveLock; /* default for compiler */
 
        switch (cmd->subtype)
        {
                /*
                 * These subcommands rewrite the heap, so require full locks.
                 */
            case AT_AddColumn:  /* may rewrite heap, in some cases and visible
                                 * to SELECT */
            case AT_SetAccessMethod:    /* must rewrite heap */
            case AT_SetTableSpace:  /* must rewrite heap */
            case AT_AlterColumnType:    /* must rewrite heap */
                cmd_lockmode = AccessExclusiveLock;
                break;
 
                /*
                 * These subcommands may require addition of toast tables. If
                 * we add a toast table to a table currently being scanned, we
                 * might miss data added to the new toast table by concurrent
                 * insert transactions.
                 */
            case AT_SetStorage: /* may add toast tables, see
                                 * ATRewriteCatalogs() */
                cmd_lockmode = AccessExclusiveLock;
                break;
 
                /*
                 * Removing constraints can affect SELECTs that have been
                 * optimized assuming the constraint holds true. See also
                 * CloneFkReferenced.
                 */
            case AT_DropConstraint: /* as DROP INDEX */
            case AT_DropNotNull:    /* may change some SQL plans */
                cmd_lockmode = AccessExclusiveLock;
                break;
 
                /*
                 * Subcommands that may be visible to concurrent SELECTs
                 */
            case AT_DropColumn: /* change visible to SELECT */
            case AT_AddColumnToView:    /* CREATE VIEW */
            case AT_DropOids:   /* used to equiv to DropColumn */
            case AT_EnableAlwaysRule:   /* may change SELECT rules */
            case AT_EnableReplicaRule:  /* may change SELECT rules */
            case AT_EnableRule: /* may change SELECT rules */
            case AT_DisableRule:    /* may change SELECT rules */
                cmd_lockmode = AccessExclusiveLock;
                break;
 
                /*
                 * Changing owner may remove implicit SELECT privileges
                 */
            case AT_ChangeOwner:    /* change visible to SELECT */
                cmd_lockmode = AccessExclusiveLock;
                break;
 
                /*
                 * Changing foreign table options may affect optimization.
                 */
            case AT_GenericOptions:
            case AT_AlterColumnGenericOptions:
                cmd_lockmode = AccessExclusiveLock;
                break;
 
                /*
                 * These subcommands affect write operations only.
                 */
            case AT_EnableTrig:
            case AT_EnableAlwaysTrig:
            case AT_EnableReplicaTrig:
            case AT_EnableTrigAll:
            case AT_EnableTrigUser:
            case AT_DisableTrig:
            case AT_DisableTrigAll:
            case AT_DisableTrigUser:
                cmd_lockmode = ShareRowExclusiveLock;
                break;
 
                /*
                 * These subcommands affect write operations only. XXX
                 * Theoretically, these could be ShareRowExclusiveLock.
                 */
            case AT_ColumnDefault:
            case AT_CookedColumnDefault:
            case AT_AlterConstraint:
            case AT_AddIndex:   /* from ADD CONSTRAINT */
            case AT_AddIndexConstraint:
            case AT_ReplicaIdentity:
            case AT_SetNotNull:
            case AT_EnableRowSecurity:
            case AT_DisableRowSecurity:
            case AT_ForceRowSecurity:
            case AT_NoForceRowSecurity:
            case AT_AddIdentity:
            case AT_DropIdentity:
            case AT_SetIdentity:
            case AT_SetExpression:
            case AT_DropExpression:
            case AT_SetCompression:
                cmd_lockmode = AccessExclusiveLock;
                break;
 
            case AT_AddConstraint:
            case AT_ReAddConstraint:    /* becomes AT_AddConstraint */
            case AT_ReAddDomainConstraint:  /* becomes AT_AddConstraint */
                if (IsA(cmd->def, Constraint))
                {
                    Constraint *con = (Constraint *) cmd->def;
 
                    switch (con->contype)
                    {
                        case CONSTR_EXCLUSION:
                        case CONSTR_PRIMARY:
                        case CONSTR_UNIQUE:
 
                            /*
                             * Cases essentially the same as CREATE INDEX. We
                             * could reduce the lock strength to ShareLock if
                             * we can work out how to allow concurrent catalog
                             * updates. XXX Might be set down to
                             * ShareRowExclusiveLock but requires further
                             * analysis.
                             */
                            cmd_lockmode = AccessExclusiveLock;
                            break;
                        case CONSTR_FOREIGN:
 
                            /*
                             * We add triggers to both tables when we add a
                             * Foreign Key, so the lock level must be at least
                             * as strong as CREATE TRIGGER.
                             */
                            cmd_lockmode = ShareRowExclusiveLock;
                            break;
 
                        default:
                            cmd_lockmode = AccessExclusiveLock;
                    }
                }
                break;
 
                /*
                 * These subcommands affect inheritance behaviour. Queries
                 * started before us will continue to see the old inheritance
                 * behaviour, while queries started after we commit will see
                 * new behaviour. No need to prevent reads or writes to the
                 * subtable while we hook it up though. Changing the TupDesc
                 * may be a problem, so keep highest lock.
                 */
            case AT_AddInherit:
            case AT_DropInherit:
                cmd_lockmode = AccessExclusiveLock;
                break;
 
                /*
                 * These subcommands affect implicit row type conversion. They
                 * have affects similar to CREATE/DROP CAST on queries. don't
                 * provide for invalidating parse trees as a result of such
                 * changes, so we keep these at AccessExclusiveLock.
                 */
            case AT_AddOf:
            case AT_DropOf:
                cmd_lockmode = AccessExclusiveLock;
                break;
 
                /*
                 * Only used by CREATE OR REPLACE VIEW which must conflict
                 * with an SELECTs currently using the view.
                 */
            case AT_ReplaceRelOptions:
                cmd_lockmode = AccessExclusiveLock;
                break;
 
                /*
                 * These subcommands affect general strategies for performance
                 * and maintenance, though don't change the semantic results
                 * from normal data reads and writes. Delaying an ALTER TABLE
                 * behind currently active writes only delays the point where
                 * the new strategy begins to take effect, so there is no
                 * benefit in waiting. In this case the minimum restriction
                 * applies: we don't currently allow concurrent catalog
                 * updates.
                 */
            case AT_SetStatistics:  /* Uses MVCC in getTableAttrs() */
            case AT_ClusterOn:  /* Uses MVCC in getIndexes() */
            case AT_DropCluster:    /* Uses MVCC in getIndexes() */
            case AT_SetOptions: /* Uses MVCC in getTableAttrs() */
            case AT_ResetOptions:   /* Uses MVCC in getTableAttrs() */
                cmd_lockmode = ShareUpdateExclusiveLock;
                break;
 
            case AT_SetLogged:
            case AT_SetUnLogged:
                cmd_lockmode = AccessExclusiveLock;
                break;
 
            case AT_ValidateConstraint: /* Uses MVCC in getConstraints() */
                cmd_lockmode = ShareUpdateExclusiveLock;
                break;
 
                /*
                 * Rel options are more complex than first appears. Options
                 * are set here for tables, views and indexes; for historical
                 * reasons these can all be used with ALTER TABLE, so we can't
                 * decide between them using the basic grammar.
                 */
            case AT_SetRelOptions:  /* Uses MVCC in getIndexes() and
                                     * getTables() */
            case AT_ResetRelOptions:    /* Uses MVCC in getIndexes() and
                                         * getTables() */
                cmd_lockmode = AlterTableGetRelOptionsLockLevel((List *) cmd->def);
                break;
 
            case AT_AttachPartition:
                cmd_lockmode = ShareUpdateExclusiveLock;
                break;
 
            case AT_DetachPartition:
                if (((PartitionCmd *) cmd->def)->concurrent)
                    cmd_lockmode = ShareUpdateExclusiveLock;
                else
                    cmd_lockmode = AccessExclusiveLock;
                break;
 
            case AT_DetachPartitionFinalize:
                cmd_lockmode = ShareUpdateExclusiveLock;
                break;
 
            case AT_MergePartitions:
            case AT_SplitPartition:
                cmd_lockmode = AccessExclusiveLock;
                break;
 
            default:            /* oops */
                elog(ERROR, "unrecognized alter table type: %d",
                     (int) cmd->subtype);
                break;
        }
 
        /*
         * Take the greatest lockmode from any subcommand
         */
        if (cmd_lockmode > lockmode)
            lockmode = cmd_lockmode;
    }
 
    return lockmode;
}

References AccessExclusiveLock, AlterTableGetRelOptionsLockLevel(), AT_AddColumn, AT_AddColumnToView, AT_AddConstraint, AT_AddIdentity, AT_AddIndex, AT_AddIndexConstraint, AT_AddInherit, AT_AddOf, AT_AlterColumnGenericOptions, AT_AlterColumnType, AT_AlterConstraint, AT_AttachPartition, AT_ChangeOwner, AT_ClusterOn, AT_ColumnDefault, AT_CookedColumnDefault, AT_DetachPartition, AT_DetachPartitionFinalize, AT_DisableRowSecurity, AT_DisableRule, AT_DisableTrig, AT_DisableTrigAll, AT_DisableTrigUser, AT_DropCluster, AT_DropColumn, AT_DropConstraint, AT_DropExpression, AT_DropIdentity, AT_DropInherit, AT_DropNotNull, AT_DropOf, AT_DropOids, AT_EnableAlwaysRule, AT_EnableAlwaysTrig, AT_EnableReplicaRule, AT_EnableReplicaTrig, AT_EnableRowSecurity, AT_EnableRule, AT_EnableTrig, AT_EnableTrigAll, AT_EnableTrigUser, AT_ForceRowSecurity, AT_GenericOptions, AT_MergePartitions, AT_NoForceRowSecurity, AT_ReAddConstraint, AT_ReAddDomainConstraint, AT_ReplaceRelOptions, AT_ReplicaIdentity, AT_ResetOptions, AT_ResetRelOptions, AT_SetAccessMethod, AT_SetCompression, AT_SetExpression, AT_SetIdentity, AT_SetLogged, AT_SetNotNull, AT_SetOptions, AT_SetRelOptions, AT_SetStatistics, AT_SetStorage, AT_SetTableSpace, AT_SetUnLogged, AT_SplitPartition, AT_ValidateConstraint, CONSTR_EXCLUSION, CONSTR_FOREIGN, CONSTR_PRIMARY, CONSTR_UNIQUE, Constraint::contype, AlterTableCmd::def, elog, ERROR, fb(), IsA, lfirst, ShareRowExclusiveLock, ShareUpdateExclusiveLock, and AlterTableCmd::subtype.

Referenced by AlterTableInternal(), and ProcessUtilitySlow().

◆ AlterTableInternal()

void AlterTableInternal	(	Oid	relid,
		List *	cmds,
		bool	recurse
	)

Definition at line 4655 of file tablecmds.c.

{
    Relation    rel;
    LOCKMODE    lockmode = AlterTableGetLockLevel(cmds);
 
    rel = relation_open(relid, lockmode);
 
    EventTriggerAlterTableRelid(relid);
 
    ATController(NULL, rel, cmds, recurse, lockmode, NULL);
}

References AlterTableGetLockLevel(), ATController(), EventTriggerAlterTableRelid(), fb(), and relation_open().

Referenced by AlterTableMoveAll(), and DefineVirtualRelation().

◆ AlterTableLookupRelation()

Oid AlterTableLookupRelation	(	AlterTableStmt *	stmt,
		LOCKMODE	lockmode
	)

Definition at line 4567 of file tablecmds.c.

{
    return RangeVarGetRelidExtended(stmt->relation, lockmode,
                                    stmt->missing_ok ? RVR_MISSING_OK : 0,
                                    RangeVarCallbackForAlterRelation,
                                    stmt);
}

References RangeVarCallbackForAlterRelation(), RangeVarGetRelidExtended(), RVR_MISSING_OK, and stmt.

Referenced by ProcessUtilitySlow().

◆ AlterTableMoveAll()

Oid AlterTableMoveAll ( AlterTableMoveAllStmt * stmt )

Definition at line 17691 of file tablecmds.c.

{
    List       *relations = NIL;
    ListCell   *l;
    ScanKeyData key[1];
    Relation    rel;
    TableScanDesc scan;
    HeapTuple   tuple;
    Oid         orig_tablespaceoid;
    Oid         new_tablespaceoid;
    List       *role_oids = roleSpecsToIds(stmt->roles);
 
    /* Ensure we were not asked to move something we can't */
    if (stmt->objtype != OBJECT_TABLE && stmt->objtype != OBJECT_INDEX &&
        stmt->objtype != OBJECT_MATVIEW)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                 errmsg("only tables, indexes, and materialized views exist in tablespaces")));
 
    /* Get the orig and new tablespace OIDs */
    orig_tablespaceoid = get_tablespace_oid(stmt->orig_tablespacename, false);
    new_tablespaceoid = get_tablespace_oid(stmt->new_tablespacename, false);
 
    /* Can't move shared relations in to or out of pg_global */
    /* This is also checked by ATExecSetTableSpace, but nice to stop earlier */
    if (orig_tablespaceoid == GLOBALTABLESPACE_OID ||
        new_tablespaceoid == GLOBALTABLESPACE_OID)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                 errmsg("cannot move relations in to or out of pg_global tablespace")));
 
    /*
     * Must have CREATE rights on the new tablespace, unless it is the
     * database default tablespace (which all users implicitly have CREATE
     * rights on).
     */
    if (OidIsValid(new_tablespaceoid) && new_tablespaceoid != MyDatabaseTableSpace)
    {
        AclResult   aclresult;
 
        aclresult = object_aclcheck(TableSpaceRelationId, new_tablespaceoid, GetUserId(),
                                    ACL_CREATE);
        if (aclresult != ACLCHECK_OK)
            aclcheck_error(aclresult, OBJECT_TABLESPACE,
                           get_tablespace_name(new_tablespaceoid));
    }
 
    /*
     * Now that the checks are done, check if we should set either to
     * InvalidOid because it is our database's default tablespace.
     */
    if (orig_tablespaceoid == MyDatabaseTableSpace)
        orig_tablespaceoid = InvalidOid;
 
    if (new_tablespaceoid == MyDatabaseTableSpace)
        new_tablespaceoid = InvalidOid;
 
    /* no-op */
    if (orig_tablespaceoid == new_tablespaceoid)
        return new_tablespaceoid;
 
    /*
     * Walk the list of objects in the tablespace and move them. This will
     * only find objects in our database, of course.
     */
    ScanKeyInit(&key[0],
                Anum_pg_class_reltablespace,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(orig_tablespaceoid));
 
    rel = table_open(RelationRelationId, AccessShareLock);
    scan = table_beginscan_catalog(rel, 1, key);
    while ((tuple = heap_getnext(scan, ForwardScanDirection)) != NULL)
    {
        Form_pg_class relForm = (Form_pg_class) GETSTRUCT(tuple);
        Oid         relOid = relForm->oid;
 
        /*
         * Do not move objects in pg_catalog as part of this, if an admin
         * really wishes to do so, they can issue the individual ALTER
         * commands directly.
         *
         * Also, explicitly avoid any shared tables, temp tables, TOAST (TOAST
         * will be moved with the main table).
         *
         * Conflict log tables are system-managed for logical replication and
         * cannot be altered directly, so skip them as well; otherwise a
         * single such table in the source tablespace would abort the whole
         * bulk move.
         */
        if (IsCatalogNamespace(relForm->relnamespace) ||
            relForm->relisshared ||
            isAnyTempNamespace(relForm->relnamespace) ||
            IsToastNamespace(relForm->relnamespace) ||
            IsConflictLogTableNamespace(relForm->relnamespace))
            continue;
 
        /* Only move the object type requested */
        if ((stmt->objtype == OBJECT_TABLE &&
             relForm->relkind != RELKIND_RELATION &&
             relForm->relkind != RELKIND_PARTITIONED_TABLE) ||
            (stmt->objtype == OBJECT_INDEX &&
             relForm->relkind != RELKIND_INDEX &&
             relForm->relkind != RELKIND_PARTITIONED_INDEX) ||
            (stmt->objtype == OBJECT_MATVIEW &&
             relForm->relkind != RELKIND_MATVIEW))
            continue;
 
        /* Check if we are only moving objects owned by certain roles */
        if (role_oids != NIL && !list_member_oid(role_oids, relForm->relowner))
            continue;
 
        /*
         * Handle permissions-checking here since we are locking the tables
         * and also to avoid doing a bunch of work only to fail part-way. Note
         * that permissions will also be checked by AlterTableInternal().
         *
         * Caller must be considered an owner on the table to move it.
         */
        if (!object_ownercheck(RelationRelationId, relOid, GetUserId()))
            aclcheck_error(ACLCHECK_NOT_OWNER, get_relkind_objtype(get_rel_relkind(relOid)),
                           NameStr(relForm->relname));
 
        if (stmt->nowait &&
            !ConditionalLockRelationOid(relOid, AccessExclusiveLock))
            ereport(ERROR,
                    (errcode(ERRCODE_OBJECT_IN_USE),
                     errmsg("aborting because lock on relation \"%s.%s\" is not available",
                            get_namespace_name(relForm->relnamespace),
                            NameStr(relForm->relname))));
        else
            LockRelationOid(relOid, AccessExclusiveLock);
 
        /* Add to our list of objects to move */
        relations = lappend_oid(relations, relOid);
    }
 
    table_endscan(scan);
    table_close(rel, AccessShareLock);
 
    if (relations == NIL)
        ereport(NOTICE,
                (errcode(ERRCODE_NO_DATA_FOUND),
                 errmsg("no matching relations in tablespace \"%s\" found",
                        orig_tablespaceoid == InvalidOid ? "(database default)" :
                        get_tablespace_name(orig_tablespaceoid))));
 
    /* Everything is locked, loop through and move all of the relations. */
    foreach(l, relations)
    {
        List       *cmds = NIL;
        AlterTableCmd *cmd = makeNode(AlterTableCmd);
 
        cmd->subtype = AT_SetTableSpace;
        cmd->name = stmt->new_tablespacename;
 
        cmds = lappend(cmds, cmd);
 
        EventTriggerAlterTableStart((Node *) stmt);
        /* OID is set by AlterTableInternal */
        AlterTableInternal(lfirst_oid(l), cmds, false);
        EventTriggerAlterTableEnd();
    }
 
    return new_tablespaceoid;
}

References AccessExclusiveLock, AccessShareLock, ACL_CREATE, aclcheck_error(), ACLCHECK_NOT_OWNER, ACLCHECK_OK, AlterTableInternal(), AT_SetTableSpace, BTEqualStrategyNumber, ConditionalLockRelationOid(), ereport, errcode(), errmsg, ERROR, EventTriggerAlterTableEnd(), EventTriggerAlterTableStart(), fb(), ForwardScanDirection, get_namespace_name(), get_rel_relkind(), get_relkind_objtype(), get_tablespace_name(), get_tablespace_oid(), GETSTRUCT(), GetUserId(), heap_getnext(), InvalidOid, isAnyTempNamespace(), IsCatalogNamespace(), IsConflictLogTableNamespace(), IsToastNamespace(), lappend(), lappend_oid(), lfirst_oid, list_member_oid(), LockRelationOid(), makeNode, MyDatabaseTableSpace, AlterTableCmd::name, NameStr, NIL, NOTICE, object_aclcheck(), OBJECT_INDEX, OBJECT_MATVIEW, object_ownercheck(), OBJECT_TABLE, OBJECT_TABLESPACE, ObjectIdGetDatum(), OidIsValid, roleSpecsToIds(), ScanKeyInit(), stmt, AlterTableCmd::subtype, table_beginscan_catalog(), table_close(), table_endscan(), and table_open().

Referenced by ProcessUtilitySlow().

◆ AlterTableNamespace()

ObjectAddress AlterTableNamespace	(	AlterObjectSchemaStmt *	stmt,
		Oid *	oldschema
	)

Definition at line 19649 of file tablecmds.c.

{
    Relation    rel;
    Oid         relid;
    Oid         oldNspOid;
    Oid         nspOid;
    RangeVar   *newrv;
    ObjectAddresses *objsMoved;
    ObjectAddress myself;
 
    relid = RangeVarGetRelidExtended(stmt->relation, AccessExclusiveLock,
                                     stmt->missing_ok ? RVR_MISSING_OK : 0,
                                     RangeVarCallbackForAlterRelation,
                                     stmt);
 
    if (!OidIsValid(relid))
    {
        ereport(NOTICE,
                (errmsg("relation \"%s\" does not exist, skipping",
                        stmt->relation->relname)));
        return InvalidObjectAddress;
    }
 
    rel = relation_open(relid, NoLock);
 
    oldNspOid = RelationGetNamespace(rel);
 
    /* If it's an owned sequence, disallow moving it by itself. */
    if (rel->rd_rel->relkind == RELKIND_SEQUENCE)
    {
        Oid         tableId;
        int32       colId;
 
        if (sequenceIsOwned(relid, DEPENDENCY_AUTO, &tableId, &colId) ||
            sequenceIsOwned(relid, DEPENDENCY_INTERNAL, &tableId, &colId))
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("cannot move an owned sequence into another schema"),
                     errdetail("Sequence \"%s\" is linked to table \"%s\".",
                               RelationGetRelationName(rel),
                               get_rel_name(tableId))));
    }
 
    /* Get and lock schema OID and check its permissions. */
    newrv = makeRangeVar(stmt->newschema, RelationGetRelationName(rel), -1);
    nspOid = RangeVarGetAndCheckCreationNamespace(newrv, NoLock, NULL);
 
    /* common checks on switching namespaces */
    CheckSetNamespace(oldNspOid, nspOid);
 
    objsMoved = new_object_addresses();
    AlterTableNamespaceInternal(rel, oldNspOid, nspOid, objsMoved);
    free_object_addresses(objsMoved);
 
    ObjectAddressSet(myself, RelationRelationId, relid);
 
    if (oldschema)
        *oldschema = oldNspOid;
 
    /* close rel, but keep lock until commit */
    relation_close(rel, NoLock);
 
    return myself;
}

References AccessExclusiveLock, AlterTableNamespaceInternal(), CheckSetNamespace(), DEPENDENCY_AUTO, DEPENDENCY_INTERNAL, ereport, errcode(), errdetail(), errmsg, ERROR, fb(), free_object_addresses(), get_rel_name(), InvalidObjectAddress, makeRangeVar(), new_object_addresses(), NoLock, NOTICE, ObjectAddressSet, OidIsValid, RangeVarCallbackForAlterRelation(), RangeVarGetAndCheckCreationNamespace(), RangeVarGetRelidExtended(), RelationData::rd_rel, relation_close(), relation_open(), RelationGetNamespace, RelationGetRelationName, RVR_MISSING_OK, sequenceIsOwned(), and stmt.

Referenced by ExecAlterObjectSchemaStmt().

◆ AlterTableNamespaceInternal()

void AlterTableNamespaceInternal	(	Relation	rel,
		Oid	oldNspOid,
		Oid	nspOid,
		ObjectAddresses *	objsMoved
	)

Definition at line 19720 of file tablecmds.c.

{
    Relation    classRel;
 
    Assert(objsMoved != NULL);
 
    /* OK, modify the pg_class row and pg_depend entry */
    classRel = table_open(RelationRelationId, RowExclusiveLock);
 
    AlterRelationNamespaceInternal(classRel, RelationGetRelid(rel), oldNspOid,
                                   nspOid, true, objsMoved);
 
    /* Fix the table's row type too, if it has one */
    if (OidIsValid(rel->rd_rel->reltype))
        AlterTypeNamespaceInternal(rel->rd_rel->reltype, nspOid,
                                   false,   /* isImplicitArray */
                                   false,   /* ignoreDependent */
                                   false,   /* errorOnTableType */
                                   objsMoved);
 
    /* Fix other dependent stuff */
    AlterIndexNamespaces(classRel, rel, oldNspOid, nspOid, objsMoved);
    AlterSeqNamespaces(classRel, rel, oldNspOid, nspOid,
                       objsMoved, AccessExclusiveLock);
    AlterConstraintNamespaces(RelationGetRelid(rel), oldNspOid, nspOid,
                              false, objsMoved);
 
    table_close(classRel, RowExclusiveLock);
}

References AccessExclusiveLock, AlterConstraintNamespaces(), AlterIndexNamespaces(), AlterRelationNamespaceInternal(), AlterSeqNamespaces(), AlterTypeNamespaceInternal(), Assert, fb(), OidIsValid, RelationData::rd_rel, RelationGetRelid, RowExclusiveLock, table_close(), and table_open().

Referenced by AlterObjectNamespace_oid(), and AlterTableNamespace().

◆ applyPartitionIndexExtDeps()

static void applyPartitionIndexExtDeps	(	Oid	newPartOid,
		List *	extDepState
	)

static

Definition at line 23685 of file tablecmds.c.

{
    Relation    partRel;
    List       *indexList;
 
    if (extDepState == NIL)
        return;
 
    /*
     * Use NoLock since the caller already holds AccessExclusiveLock on the
     * new partition.
     */
    partRel = table_open(newPartOid, NoLock);
    indexList = RelationGetIndexList(partRel);
 
    foreach_oid(indexOid, indexList)
    {
        Oid         parentIdxOid;
 
        if (!get_rel_relispartition(indexOid))
            continue;
 
        parentIdxOid = get_partition_parent(indexOid, true);
        if (!OidIsValid(parentIdxOid))
            continue;
 
        foreach_ptr(PartitionIndexExtDepEntry, entry, extDepState)
        {
            ObjectAddress indexAddr;
 
            if (entry->parentIndexOid > parentIdxOid)
                break;
            if (entry->parentIndexOid < parentIdxOid)
                continue;
 
            ObjectAddressSet(indexAddr, RelationRelationId, indexOid);
 
            foreach_oid(extOid, entry->extensionOids)
            {
                ObjectAddress extAddr;
 
                ObjectAddressSet(extAddr, ExtensionRelationId, extOid);
                recordDependencyOn(&indexAddr, &extAddr,
                                   DEPENDENCY_AUTO_EXTENSION);
            }
            break;
        }
    }
 
    list_free(indexList);
    table_close(partRel, NoLock);
}

References DEPENDENCY_AUTO_EXTENSION, fb(), foreach_oid, foreach_ptr, get_partition_parent(), get_rel_relispartition(), list_free(), NIL, NoLock, ObjectAddressSet, OidIsValid, recordDependencyOn(), RelationGetIndexList(), table_close(), and table_open().

Referenced by ATExecMergePartitions(), and ATExecSplitPartition().

◆ ATAddCheckNNConstraint()

static ObjectAddress ATAddCheckNNConstraint	(	List **	wqueue,
		AlteredTableInfo *	tab,
		Relation	rel,
		Constraint *	constr,
		bool	recurse,
		bool	recursing,
		bool	is_readd,
		LOCKMODE	lockmode
	)

static

Definition at line 10048 of file tablecmds.c.

{
    List       *newcons;
    ListCell   *lcon;
    List       *children;
    ListCell   *child;
    ObjectAddress address = InvalidObjectAddress;
 
    /* Guard against stack overflow due to overly deep inheritance tree. */
    check_stack_depth();
 
    /* At top level, permission check was done in ATPrepCmd, else do it */
    if (recursing)
        ATSimplePermissions(AT_AddConstraint, rel,
                            ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
 
    /*
     * Call AddRelationNewConstraints to do the work, making sure it works on
     * a copy of the Constraint so transformExpr can't modify the original. It
     * returns a list of cooked constraints.
     *
     * If the constraint ends up getting merged with a pre-existing one, it's
     * omitted from the returned list, which is what we want: we do not need
     * to do any validation work.  That can only happen at child tables,
     * though, since we disallow merging at the top level.
     */
    newcons = AddRelationNewConstraints(rel, NIL,
                                        list_make1(copyObject(constr)),
                                        recursing || is_readd,  /* allow_merge */
                                        !recursing, /* is_local */
                                        is_readd,   /* is_internal */
                                        NULL);  /* queryString not available
                                                 * here */
 
    /* we don't expect more than one constraint here */
    Assert(list_length(newcons) <= 1);
 
    /* Add each to-be-validated constraint to Phase 3's queue */
    foreach(lcon, newcons)
    {
        CookedConstraint *ccon = (CookedConstraint *) lfirst(lcon);
 
        if (!ccon->skip_validation && ccon->contype != CONSTR_NOTNULL)
        {
            NewConstraint *newcon;
 
            newcon = palloc0_object(NewConstraint);
            newcon->name = ccon->name;
            newcon->contype = ccon->contype;
            newcon->qual = ccon->expr;
 
            tab->constraints = lappend(tab->constraints, newcon);
        }
 
        /* Save the actually assigned name if it was defaulted */
        if (constr->conname == NULL)
            constr->conname = ccon->name;
 
        /*
         * If adding a valid not-null constraint, set the pg_attribute flag
         * and tell phase 3 to verify existing rows, if needed.  For an
         * invalid constraint, just set attnotnull, without queueing
         * verification.
         */
        if (constr->contype == CONSTR_NOTNULL)
            set_attnotnull(wqueue, rel, ccon->attnum,
                           !constr->skip_validation,
                           !constr->skip_validation);
 
        ObjectAddressSet(address, ConstraintRelationId, ccon->conoid);
    }
 
    /* At this point we must have a locked-down name to use */
    Assert(newcons == NIL || constr->conname != NULL);
 
    /* Advance command counter in case same table is visited multiple times */
    CommandCounterIncrement();
 
    /*
     * If the constraint got merged with an existing constraint, we're done.
     * We mustn't recurse to child tables in this case, because they've
     * already got the constraint, and visiting them again would lead to an
     * incorrect value for coninhcount.
     */
    if (newcons == NIL)
        return address;
 
    /*
     * If adding a NO INHERIT constraint, no need to find our children.
     */
    if (constr->is_no_inherit)
        return address;
 
    /*
     * Propagate to children as appropriate.  Unlike most other ALTER
     * routines, we have to do this one level of recursion at a time; we can't
     * use find_all_inheritors to do it in one pass.
     */
    children =
        find_inheritance_children(RelationGetRelid(rel), lockmode);
 
    /*
     * Check if ONLY was specified with ALTER TABLE.  If so, allow the
     * constraint creation only if there are no children currently. Error out
     * otherwise.
     */
    if (!recurse && children != NIL)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("constraint must be added to child tables too")));
 
    /*
     * Recurse to create the constraint on each child.
     */
    foreach(child, children)
    {
        Oid         childrelid = lfirst_oid(child);
        Relation    childrel;
        AlteredTableInfo *childtab;
 
        /* find_inheritance_children already got lock */
        childrel = table_open(childrelid, NoLock);
        CheckAlterTableIsSafe(childrel);
 
        /* Find or create work queue entry for this table */
        childtab = ATGetQueueEntry(wqueue, childrel);
 
        /* Recurse to this child */
        ATAddCheckNNConstraint(wqueue, childtab, childrel,
                               constr, recurse, true, is_readd, lockmode);
 
        table_close(childrel, NoLock);
    }
 
    return address;
}

References AddRelationNewConstraints(), Assert, AT_AddConstraint, ATAddCheckNNConstraint(), ATGetQueueEntry(), ATSimplePermissions(), ATT_FOREIGN_TABLE, ATT_PARTITIONED_TABLE, ATT_TABLE, check_stack_depth(), CheckAlterTableIsSafe(), CommandCounterIncrement(), Constraint::conname, CONSTR_NOTNULL, AlteredTableInfo::constraints, Constraint::contype, copyObject, ereport, errcode(), errmsg, ERROR, fb(), find_inheritance_children(), InvalidObjectAddress, Constraint::is_no_inherit, lappend(), lfirst, lfirst_oid, list_length(), list_make1, NIL, NoLock, ObjectAddressSet, palloc0_object, RelationGetRelid, set_attnotnull(), Constraint::skip_validation, table_close(), and table_open().

Referenced by ATAddCheckNNConstraint(), and ATExecAddConstraint().

◆ ATAddForeignKeyConstraint()

static ObjectAddress ATAddForeignKeyConstraint	(	List **	wqueue,
		AlteredTableInfo *	tab,
		Relation	rel,
		Constraint *	fkconstraint,
		bool	recurse,
		bool	recursing,
		LOCKMODE	lockmode
	)

static

Definition at line 10203 of file tablecmds.c.

{
    Relation    pkrel;
    int16       pkattnum[INDEX_MAX_KEYS] = {0};
    int16       fkattnum[INDEX_MAX_KEYS] = {0};
    Oid         pktypoid[INDEX_MAX_KEYS] = {0};
    Oid         fktypoid[INDEX_MAX_KEYS] = {0};
    Oid         pkcolloid[INDEX_MAX_KEYS] = {0};
    Oid         fkcolloid[INDEX_MAX_KEYS] = {0};
    Oid         opclasses[INDEX_MAX_KEYS] = {0};
    Oid         pfeqoperators[INDEX_MAX_KEYS] = {0};
    Oid         ppeqoperators[INDEX_MAX_KEYS] = {0};
    Oid         ffeqoperators[INDEX_MAX_KEYS] = {0};
    int16       fkdelsetcols[INDEX_MAX_KEYS] = {0};
    bool        with_period;
    bool        pk_has_without_overlaps;
    int         i;
    int         numfks,
                numpks,
                numfkdelsetcols;
    Oid         indexOid;
    bool        old_check_ok;
    ObjectAddress address;
    ListCell   *old_pfeqop_item = list_head(fkconstraint->old_conpfeqop);
 
    /*
     * Grab ShareRowExclusiveLock on the pk table, so that someone doesn't
     * delete rows out from under us.
     */
    if (OidIsValid(fkconstraint->old_pktable_oid))
        pkrel = table_open(fkconstraint->old_pktable_oid, ShareRowExclusiveLock);
    else
        pkrel = table_openrv(fkconstraint->pktable, ShareRowExclusiveLock);
 
    /*
     * Validity checks (permission checks wait till we have the column
     * numbers)
     */
    if (!recurse && rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
        ereport(ERROR,
                errcode(ERRCODE_WRONG_OBJECT_TYPE),
                errmsg("cannot use ONLY for foreign key on partitioned table \"%s\" referencing relation \"%s\"",
                       RelationGetRelationName(rel),
                       RelationGetRelationName(pkrel)));
 
    if (pkrel->rd_rel->relkind != RELKIND_RELATION &&
        pkrel->rd_rel->relkind != RELKIND_PARTITIONED_TABLE)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("referenced relation \"%s\" is not a table",
                        RelationGetRelationName(pkrel))));
 
    /*
     * Conflict log tables are used internally for logical replication
     * conflict logging and should not be referenced by foreign keys, as it
     * could disrupt conflict logging.
     */
    if (IsConflictLogTableClass(pkrel->rd_rel))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot reference conflict log table \"%s\"",
                        RelationGetRelationName(pkrel)),
                 errdetail("Conflict log tables are system-managed tables for logical replication conflicts.")));
 
    if (!allowSystemTableMods && IsSystemRelation(pkrel))
        ereport(ERROR,
                (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
                 errmsg("permission denied: \"%s\" is a system catalog",
                        RelationGetRelationName(pkrel))));
 
    /*
     * References from permanent or unlogged tables to temp tables, and from
     * permanent tables to unlogged tables, are disallowed because the
     * referenced data can vanish out from under us.  References from temp
     * tables to any other table type are also disallowed, because other
     * backends might need to run the RI triggers on the perm table, but they
     * can't reliably see tuples in the local buffers of other backends.
     */
    switch (rel->rd_rel->relpersistence)
    {
        case RELPERSISTENCE_PERMANENT:
            if (!RelationIsPermanent(pkrel))
                ereport(ERROR,
                        (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                         errmsg("constraints on permanent tables may reference only permanent tables")));
            break;
        case RELPERSISTENCE_UNLOGGED:
            if (!RelationIsPermanent(pkrel)
                && pkrel->rd_rel->relpersistence != RELPERSISTENCE_UNLOGGED)
                ereport(ERROR,
                        (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                         errmsg("constraints on unlogged tables may reference only permanent or unlogged tables")));
            break;
        case RELPERSISTENCE_TEMP:
            if (pkrel->rd_rel->relpersistence != RELPERSISTENCE_TEMP)
                ereport(ERROR,
                        (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                         errmsg("constraints on temporary tables may reference only temporary tables")));
            if (!pkrel->rd_islocaltemp || !rel->rd_islocaltemp)
                ereport(ERROR,
                        (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                         errmsg("constraints on temporary tables must involve temporary tables of this session")));
            break;
    }
 
    /*
     * Look up the referencing attributes to make sure they exist, and record
     * their attnums and type and collation OIDs.
     */
    numfks = transformColumnNameList(RelationGetRelid(rel),
                                     fkconstraint->fk_attrs,
                                     fkattnum, fktypoid, fkcolloid);
    with_period = fkconstraint->fk_with_period || fkconstraint->pk_with_period;
    if (with_period && !fkconstraint->fk_with_period)
        ereport(ERROR,
                errcode(ERRCODE_INVALID_FOREIGN_KEY),
                errmsg("foreign key uses PERIOD on the referenced table but not the referencing table"));
 
    numfkdelsetcols = transformColumnNameList(RelationGetRelid(rel),
                                              fkconstraint->fk_del_set_cols,
                                              fkdelsetcols, NULL, NULL);
    numfkdelsetcols = validateFkOnDeleteSetColumns(numfks, fkattnum,
                                                   numfkdelsetcols,
                                                   fkdelsetcols,
                                                   fkconstraint->fk_del_set_cols);
 
    /*
     * If the attribute list for the referenced table was omitted, lookup the
     * definition of the primary key and use it.  Otherwise, validate the
     * supplied attribute list.  In either case, discover the index OID and
     * index opclasses, and the attnums and type and collation OIDs of the
     * attributes.
     */
    if (fkconstraint->pk_attrs == NIL)
    {
        numpks = transformFkeyGetPrimaryKey(pkrel, &indexOid,
                                            &fkconstraint->pk_attrs,
                                            pkattnum, pktypoid, pkcolloid,
                                            opclasses, &pk_has_without_overlaps);
 
        /* If the primary key uses WITHOUT OVERLAPS, the fk must use PERIOD */
        if (pk_has_without_overlaps && !fkconstraint->fk_with_period)
            ereport(ERROR,
                    errcode(ERRCODE_INVALID_FOREIGN_KEY),
                    errmsg("foreign key uses PERIOD on the referenced table but not the referencing table"));
    }
    else
    {
        numpks = transformColumnNameList(RelationGetRelid(pkrel),
                                         fkconstraint->pk_attrs,
                                         pkattnum, pktypoid, pkcolloid);
 
        /* Since we got pk_attrs, one should be a period. */
        if (with_period && !fkconstraint->pk_with_period)
            ereport(ERROR,
                    errcode(ERRCODE_INVALID_FOREIGN_KEY),
                    errmsg("foreign key uses PERIOD on the referencing table but not the referenced table"));
 
        /* Look for an index matching the column list */
        indexOid = transformFkeyCheckAttrs(pkrel, numpks, pkattnum,
                                           with_period, opclasses, &pk_has_without_overlaps);
    }
 
    /*
     * If the referenced primary key has WITHOUT OVERLAPS, the foreign key
     * must use PERIOD.
     */
    if (pk_has_without_overlaps && !with_period)
        ereport(ERROR,
                errcode(ERRCODE_INVALID_FOREIGN_KEY),
                errmsg("foreign key must use PERIOD when referencing a primary key using WITHOUT OVERLAPS"));
 
    /*
     * Now we can check permissions.
     */
    checkFkeyPermissions(pkrel, pkattnum, numpks);
 
    /*
     * Check some things for generated columns.
     */
    for (i = 0; i < numfks; i++)
    {
        char        attgenerated = TupleDescAttr(RelationGetDescr(rel), fkattnum[i] - 1)->attgenerated;
 
        if (attgenerated)
        {
            /*
             * Check restrictions on UPDATE/DELETE actions, per SQL standard
             */
            if (fkconstraint->fk_upd_action == FKCONSTR_ACTION_SETNULL ||
                fkconstraint->fk_upd_action == FKCONSTR_ACTION_SETDEFAULT ||
                fkconstraint->fk_upd_action == FKCONSTR_ACTION_CASCADE)
                ereport(ERROR,
                        (errcode(ERRCODE_SYNTAX_ERROR),
                         errmsg("invalid %s action for foreign key constraint containing generated column",
                                "ON UPDATE")));
            if (fkconstraint->fk_del_action == FKCONSTR_ACTION_SETNULL ||
                fkconstraint->fk_del_action == FKCONSTR_ACTION_SETDEFAULT)
                ereport(ERROR,
                        (errcode(ERRCODE_SYNTAX_ERROR),
                         errmsg("invalid %s action for foreign key constraint containing generated column",
                                "ON DELETE")));
        }
 
        /*
         * FKs on virtual columns are not supported.  This would require
         * various additional support in ri_triggers.c, including special
         * handling in ri_NullCheck(), ri_KeysEqual(),
         * RI_FKey_fk_upd_check_required() (since all virtual columns appear
         * as NULL there).  Also not really practical as long as you can't
         * index virtual columns.
         */
        if (attgenerated == ATTRIBUTE_GENERATED_VIRTUAL)
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("foreign key constraints on virtual generated columns are not supported")));
    }
 
    /*
     * Some actions are currently unsupported for foreign keys using PERIOD.
     */
    if (fkconstraint->fk_with_period)
    {
        if (fkconstraint->fk_upd_action == FKCONSTR_ACTION_RESTRICT ||
            fkconstraint->fk_upd_action == FKCONSTR_ACTION_CASCADE ||
            fkconstraint->fk_upd_action == FKCONSTR_ACTION_SETNULL ||
            fkconstraint->fk_upd_action == FKCONSTR_ACTION_SETDEFAULT)
            ereport(ERROR,
                    errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                    errmsg("unsupported %s action for foreign key constraint using PERIOD",
                           "ON UPDATE"));
 
        if (fkconstraint->fk_del_action == FKCONSTR_ACTION_RESTRICT ||
            fkconstraint->fk_del_action == FKCONSTR_ACTION_CASCADE ||
            fkconstraint->fk_del_action == FKCONSTR_ACTION_SETNULL ||
            fkconstraint->fk_del_action == FKCONSTR_ACTION_SETDEFAULT)
            ereport(ERROR,
                    errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                    errmsg("unsupported %s action for foreign key constraint using PERIOD",
                           "ON DELETE"));
    }
 
    /*
     * Look up the equality operators to use in the constraint.
     *
     * Note that we have to be careful about the difference between the actual
     * PK column type and the opclass' declared input type, which might be
     * only binary-compatible with it.  The declared opcintype is the right
     * thing to probe pg_amop with.
     */
    if (numfks != numpks)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_FOREIGN_KEY),
                 errmsg("number of referencing and referenced columns for foreign key disagree")));
 
    /*
     * On the strength of a previous constraint, we might avoid scanning
     * tables to validate this one.  See below.
     */
    old_check_ok = (fkconstraint->old_conpfeqop != NIL);
    Assert(!old_check_ok || numfks == list_length(fkconstraint->old_conpfeqop));
 
    for (i = 0; i < numpks; i++)
    {
        Oid         pktype = pktypoid[i];
        Oid         fktype = fktypoid[i];
        Oid         fktyped;
        Oid         pkcoll = pkcolloid[i];
        Oid         fkcoll = fkcolloid[i];
        HeapTuple   cla_ht;
        Form_pg_opclass cla_tup;
        Oid         amid;
        Oid         opfamily;
        Oid         opcintype;
        bool        for_overlaps;
        CompareType cmptype;
        Oid         pfeqop;
        Oid         ppeqop;
        Oid         ffeqop;
        int16       eqstrategy;
        Oid         pfeqop_right;
 
        /* We need several fields out of the pg_opclass entry */
        cla_ht = SearchSysCache1(CLAOID, ObjectIdGetDatum(opclasses[i]));
        if (!HeapTupleIsValid(cla_ht))
            elog(ERROR, "cache lookup failed for opclass %u", opclasses[i]);
        cla_tup = (Form_pg_opclass) GETSTRUCT(cla_ht);
        amid = cla_tup->opcmethod;
        opfamily = cla_tup->opcfamily;
        opcintype = cla_tup->opcintype;
        ReleaseSysCache(cla_ht);
 
        /*
         * Get strategy number from index AM.
         *
         * For a normal foreign-key constraint, this should not fail, since we
         * already checked that the index is unique and should therefore have
         * appropriate equal operators.  For a period foreign key, this could
         * fail if we selected a non-matching exclusion constraint earlier.
         * (XXX Maybe we should do these lookups earlier so we don't end up
         * doing that.)
         */
        for_overlaps = with_period && i == numpks - 1;
        cmptype = for_overlaps ? COMPARE_OVERLAP : COMPARE_EQ;
        eqstrategy = IndexAmTranslateCompareType(cmptype, amid, opfamily, true);
        if (eqstrategy == InvalidStrategy)
            ereport(ERROR,
                    errcode(ERRCODE_UNDEFINED_OBJECT),
                    for_overlaps
                    ? errmsg("could not identify an overlaps operator for foreign key")
                    : errmsg("could not identify an equality operator for foreign key"),
                    errdetail("Could not translate compare type %d for operator family \"%s\" of access method \"%s\".",
                              cmptype, get_opfamily_name(opfamily, false), get_am_name(amid)));
 
        /*
         * There had better be a primary equality operator for the index.
         * We'll use it for PK = PK comparisons.
         */
        ppeqop = get_opfamily_member(opfamily, opcintype, opcintype,
                                     eqstrategy);
 
        if (!OidIsValid(ppeqop))
            elog(ERROR, "missing operator %d(%u,%u) in opfamily %u",
                 eqstrategy, opcintype, opcintype, opfamily);
 
        /*
         * Are there equality operators that take exactly the FK type? Assume
         * we should look through any domain here.
         */
        fktyped = getBaseType(fktype);
 
        pfeqop = get_opfamily_member(opfamily, opcintype, fktyped,
                                     eqstrategy);
        if (OidIsValid(pfeqop))
        {
            pfeqop_right = fktyped;
            ffeqop = get_opfamily_member(opfamily, fktyped, fktyped,
                                         eqstrategy);
        }
        else
        {
            /* keep compiler quiet */
            pfeqop_right = InvalidOid;
            ffeqop = InvalidOid;
        }
 
        if (!(OidIsValid(pfeqop) && OidIsValid(ffeqop)))
        {
            /*
             * Otherwise, look for an implicit cast from the FK type to the
             * opcintype, and if found, use the primary equality operator.
             * This is a bit tricky because opcintype might be a polymorphic
             * type such as ANYARRAY or ANYENUM; so what we have to test is
             * whether the two actual column types can be concurrently cast to
             * that type.  (Otherwise, we'd fail to reject combinations such
             * as int[] and point[].)
             */
            Oid         input_typeids[2];
            Oid         target_typeids[2];
 
            input_typeids[0] = pktype;
            input_typeids[1] = fktype;
            target_typeids[0] = opcintype;
            target_typeids[1] = opcintype;
            if (can_coerce_type(2, input_typeids, target_typeids,
                                COERCION_IMPLICIT))
            {
                pfeqop = ffeqop = ppeqop;
                pfeqop_right = opcintype;
            }
        }
 
        if (!(OidIsValid(pfeqop) && OidIsValid(ffeqop)))
            ereport(ERROR,
                    (errcode(ERRCODE_DATATYPE_MISMATCH),
                     errmsg("foreign key constraint \"%s\" cannot be implemented",
                            fkconstraint->conname),
                     errdetail("Key columns \"%s\" of the referencing table and \"%s\" of the referenced table "
                               "are of incompatible types: %s and %s.",
                               strVal(list_nth(fkconstraint->fk_attrs, i)),
                               strVal(list_nth(fkconstraint->pk_attrs, i)),
                               format_type_be(fktype),
                               format_type_be(pktype))));
 
        /*
         * This shouldn't be possible, but better check to make sure we have a
         * consistent state for the check below.
         */
        if ((OidIsValid(pkcoll) && !OidIsValid(fkcoll)) || (!OidIsValid(pkcoll) && OidIsValid(fkcoll)))
            elog(ERROR, "key columns are not both collatable");
 
        if (OidIsValid(pkcoll) && OidIsValid(fkcoll))
        {
            bool        pkcolldet;
            bool        fkcolldet;
 
            pkcolldet = get_collation_isdeterministic(pkcoll);
            fkcolldet = get_collation_isdeterministic(fkcoll);
 
            /*
             * SQL requires that both collations are the same.  This is
             * because we need a consistent notion of equality on both
             * columns.  We relax this by allowing different collations if
             * they are both deterministic.  (This is also for backward
             * compatibility, because PostgreSQL has always allowed this.)
             */
            if ((!pkcolldet || !fkcolldet) && pkcoll != fkcoll)
                ereport(ERROR,
                        (errcode(ERRCODE_COLLATION_MISMATCH),
                         errmsg("foreign key constraint \"%s\" cannot be implemented", fkconstraint->conname),
                         errdetail("Key columns \"%s\" of the referencing table and \"%s\" of the referenced table "
                                   "have incompatible collations: \"%s\" and \"%s\".  "
                                   "If either collation is nondeterministic, then both collations have to be the same.",
                                   strVal(list_nth(fkconstraint->fk_attrs, i)),
                                   strVal(list_nth(fkconstraint->pk_attrs, i)),
                                   get_collation_name(fkcoll),
                                   get_collation_name(pkcoll))));
        }
 
        if (old_check_ok)
        {
            /*
             * When a pfeqop changes, revalidate the constraint.  We could
             * permit intra-opfamily changes, but that adds subtle complexity
             * without any concrete benefit for core types.  We need not
             * assess ppeqop or ffeqop, which RI_Initial_Check() does not use.
             */
            old_check_ok = (pfeqop == lfirst_oid(old_pfeqop_item));
            old_pfeqop_item = lnext(fkconstraint->old_conpfeqop,
                                    old_pfeqop_item);
        }
        if (old_check_ok)
        {
            Oid         old_fktype;
            Oid         new_fktype;
            CoercionPathType old_pathtype;
            CoercionPathType new_pathtype;
            Oid         old_castfunc;
            Oid         new_castfunc;
            Oid         old_fkcoll;
            Oid         new_fkcoll;
            Form_pg_attribute attr = TupleDescAttr(tab->oldDesc,
                                                   fkattnum[i] - 1);
 
            /*
             * Identify coercion pathways from each of the old and new FK-side
             * column types to the right (foreign) operand type of the pfeqop.
             * We may assume that pg_constraint.conkey is not changing.
             */
            old_fktype = attr->atttypid;
            new_fktype = fktype;
            old_pathtype = findFkeyCast(pfeqop_right, old_fktype,
                                        &old_castfunc);
            new_pathtype = findFkeyCast(pfeqop_right, new_fktype,
                                        &new_castfunc);
 
            old_fkcoll = attr->attcollation;
            new_fkcoll = fkcoll;
 
            /*
             * Upon a change to the cast from the FK column to its pfeqop
             * operand, revalidate the constraint.  For this evaluation, a
             * binary coercion cast is equivalent to no cast at all.  While
             * type implementors should design implicit casts with an eye
             * toward consistency of operations like equality, we cannot
             * assume here that they have done so.
             *
             * A function with a polymorphic argument could change behavior
             * arbitrarily in response to get_fn_expr_argtype().  Therefore,
             * when the cast destination is polymorphic, we only avoid
             * revalidation if the input type has not changed at all.  Given
             * just the core data types and operator classes, this requirement
             * prevents no would-be optimizations.
             *
             * If the cast converts from a base type to a domain thereon, then
             * that domain type must be the opcintype of the unique index.
             * Necessarily, the primary key column must then be of the domain
             * type.  Since the constraint was previously valid, all values on
             * the foreign side necessarily exist on the primary side and in
             * turn conform to the domain.  Consequently, we need not treat
             * domains specially here.
             *
             * If the collation changes, revalidation is required, unless both
             * collations are deterministic, because those share the same
             * notion of equality (because texteq reduces to bitwise
             * equality).
             *
             * We need not directly consider the PK type.  It's necessarily
             * binary coercible to the opcintype of the unique index column,
             * and ri_triggers.c will only deal with PK datums in terms of
             * that opcintype.  Changing the opcintype also changes pfeqop.
             */
            old_check_ok = (new_pathtype == old_pathtype &&
                            new_castfunc == old_castfunc &&
                            (!IsPolymorphicType(pfeqop_right) ||
                             new_fktype == old_fktype) &&
                            (new_fkcoll == old_fkcoll ||
                             (get_collation_isdeterministic(old_fkcoll) && get_collation_isdeterministic(new_fkcoll))));
        }
 
        pfeqoperators[i] = pfeqop;
        ppeqoperators[i] = ppeqop;
        ffeqoperators[i] = ffeqop;
    }
 
    /*
     * For FKs with PERIOD we need additional operators to check whether the
     * referencing row's range is contained by the aggregated ranges of the
     * referenced row(s). For rangetypes and multirangetypes this is
     * fk.periodatt <@ range_agg(pk.periodatt). Those are the only types we
     * support for now. FKs will look these up at "runtime", but we should
     * make sure the lookup works here, even if we don't use the values.
     */
    if (with_period)
    {
        Oid         periodoperoid;
        Oid         aggedperiodoperoid;
        Oid         intersectoperoid;
 
        FindFKPeriodOpers(opclasses[numpks - 1], &periodoperoid, &aggedperiodoperoid,
                          &intersectoperoid);
    }
 
    /* First, create the constraint catalog entry itself. */
    address = addFkConstraint(addFkBothSides,
                              fkconstraint->conname, fkconstraint, rel, pkrel,
                              indexOid,
                              InvalidOid,   /* no parent constraint */
                              numfks,
                              pkattnum,
                              fkattnum,
                              pfeqoperators,
                              ppeqoperators,
                              ffeqoperators,
                              numfkdelsetcols,
                              fkdelsetcols,
                              false,
                              with_period);
 
    /* Next process the action triggers at the referenced side and recurse */
    addFkRecurseReferenced(fkconstraint, rel, pkrel,
                           indexOid,
                           address.objectId,
                           numfks,
                           pkattnum,
                           fkattnum,
                           pfeqoperators,
                           ppeqoperators,
                           ffeqoperators,
                           numfkdelsetcols,
                           fkdelsetcols,
                           old_check_ok,
                           InvalidOid, InvalidOid,
                           with_period);
 
    /* Lastly create the check triggers at the referencing side and recurse */
    addFkRecurseReferencing(wqueue, fkconstraint, rel, pkrel,
                            indexOid,
                            address.objectId,
                            numfks,
                            pkattnum,
                            fkattnum,
                            pfeqoperators,
                            ppeqoperators,
                            ffeqoperators,
                            numfkdelsetcols,
                            fkdelsetcols,
                            old_check_ok,
                            lockmode,
                            InvalidOid, InvalidOid,
                            with_period);
 
    /*
     * Done.  Close pk table, but keep lock until we've committed.
     */
    table_close(pkrel, NoLock);
 
    return address;
}

References addFkBothSides, addFkConstraint(), addFkRecurseReferenced(), addFkRecurseReferencing(), allowSystemTableMods, Assert, can_coerce_type(), checkFkeyPermissions(), COERCION_IMPLICIT, COMPARE_EQ, COMPARE_OVERLAP, elog, ereport, errcode(), errdetail(), errmsg, ERROR, fb(), findFkeyCast(), FindFKPeriodOpers(), FKCONSTR_ACTION_CASCADE, FKCONSTR_ACTION_RESTRICT, FKCONSTR_ACTION_SETDEFAULT, FKCONSTR_ACTION_SETNULL, Form_pg_opclass, format_type_be(), get_am_name(), get_collation_isdeterministic(), get_collation_name(), get_opfamily_member(), get_opfamily_name(), getBaseType(), GETSTRUCT(), HeapTupleIsValid, i, INDEX_MAX_KEYS, IndexAmTranslateCompareType(), InvalidOid, InvalidStrategy, IsConflictLogTableClass(), IsSystemRelation(), lfirst_oid, list_head(), list_length(), list_nth(), lnext(), NIL, NoLock, ObjectAddress::objectId, ObjectIdGetDatum(), OidIsValid, AlteredTableInfo::oldDesc, RelationData::rd_islocaltemp, RelationData::rd_rel, RelationGetDescr, RelationGetRelationName, RelationGetRelid, RelationIsPermanent, ReleaseSysCache(), SearchSysCache1(), ShareRowExclusiveLock, strVal, table_close(), table_open(), table_openrv(), transformColumnNameList(), transformFkeyCheckAttrs(), transformFkeyGetPrimaryKey(), TupleDescAttr(), and validateFkOnDeleteSetColumns().

Referenced by ATExecAddConstraint().

◆ ATCheckCheckConstrHasEnforcedParent()

static bool ATCheckCheckConstrHasEnforcedParent	(	Relation	conrel,
		Relation	rel,
		HeapTuple	contuple,
		List *	changing_conids,
		Oid *	enforced_parentoid
	)

static

Definition at line 12961 of file tablecmds.c.

{
    Form_pg_constraint currcon;
    Relation    inhrel;
    SysScanDesc scan;
    ScanKeyData skey;
    HeapTuple   inheritsTuple;
 
    /* Since this function recurses, it could be driven to stack overflow */
    check_stack_depth();
 
    currcon = (Form_pg_constraint) GETSTRUCT(contuple);
    Assert(currcon->contype == CONSTRAINT_CHECK);
 
    if (currcon->coninhcount <= 0)
        return false;
 
    inhrel = table_open(InheritsRelationId, AccessShareLock);
 
    ScanKeyInit(&skey,
                Anum_pg_inherits_inhrelid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationGetRelid(rel)));
    scan = systable_beginscan(inhrel, InheritsRelidSeqnoIndexId,
                              true, NULL, 1, &skey);
 
    while (HeapTupleIsValid(inheritsTuple = systable_getnext(scan)))
    {
        Oid         parentoid;
        Relation    parentrel = NULL;
        SysScanDesc pscan;
        ScanKeyData pkey[3];
        HeapTuple   parenttup;
 
        parentoid = ((Form_pg_inherits) GETSTRUCT(inheritsTuple))->inhparent;
 
        ScanKeyInit(&pkey[0],
                    Anum_pg_constraint_conrelid,
                    BTEqualStrategyNumber, F_OIDEQ,
                    ObjectIdGetDatum(parentoid));
        ScanKeyInit(&pkey[1],
                    Anum_pg_constraint_contypid,
                    BTEqualStrategyNumber, F_OIDEQ,
                    ObjectIdGetDatum(InvalidOid));
        ScanKeyInit(&pkey[2],
                    Anum_pg_constraint_conname,
                    BTEqualStrategyNumber, F_NAMEEQ,
                    NameGetDatum(&currcon->conname));
 
        pscan = systable_beginscan(conrel, ConstraintRelidTypidNameIndexId,
                                   true, NULL, 3, pkey);
 
        /*
         * ConstraintRelidTypidNameIndexId is unique on (conrelid, contypid,
         * conname), so this loop body executes at most once per parent.
         */
        while (HeapTupleIsValid(parenttup = systable_getnext(pscan)))
        {
            Form_pg_constraint parentcon;
 
            parentcon = (Form_pg_constraint) GETSTRUCT(parenttup);
 
            if (parentcon->contype != CONSTRAINT_CHECK ||
                parentcon->connoinherit ||
                !parentcon->conenforced)
                continue;
 
            if (!constraints_equivalent(parenttup, contuple,
                                        RelationGetDescr(conrel)))
                elog(ERROR, "child table \"%s\" has different definition for check constraint \"%s\"",
                     RelationGetRelationName(rel),
                     NameStr(parentcon->conname));
 
            /*
             * A parent listed in changing_conids is being changed by the same
             * ALTER, but it may not have been updated yet.  For regular
             * inheritance, recurse upward to check whether an equivalent
             * enforced parent outside the ALTER will make it remain enforced.
             * Partitions cannot have multiple parents, so they do not need
             * this check.
             */
            if (!rel->rd_rel->relispartition &&
                list_member_oid(changing_conids, parentcon->oid))
            {
                Oid         parent_enforced_parentoid = InvalidOid;
 
                if (parentrel == NULL)
                    parentrel = table_open(parentoid, NoLock);
 
                if (!ATCheckCheckConstrHasEnforcedParent(conrel,
                                                         parentrel,
                                                         parenttup,
                                                         changing_conids,
                                                         &parent_enforced_parentoid))
                    continue;
            }
 
            *enforced_parentoid = parentoid;
            if (parentrel != NULL)
                table_close(parentrel, NoLock);
            systable_endscan(pscan);
            systable_endscan(scan);
            table_close(inhrel, AccessShareLock);
            return true;
        }
 
        if (parentrel != NULL)
            table_close(parentrel, NoLock);
        systable_endscan(pscan);
    }
 
    systable_endscan(scan);
    table_close(inhrel, AccessShareLock);
 
    return false;
}

References AccessShareLock, Assert, ATCheckCheckConstrHasEnforcedParent(), BTEqualStrategyNumber, check_stack_depth(), constraints_equivalent(), elog, ERROR, fb(), Form_pg_constraint, Form_pg_inherits, GETSTRUCT(), HeapTupleIsValid, InvalidOid, list_member_oid(), NameGetDatum(), NameStr, NoLock, ObjectIdGetDatum(), RelationData::rd_rel, RelationGetDescr, RelationGetRelationName, RelationGetRelid, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by ATCheckCheckConstrHasEnforcedParent(), and ATExecAlterCheckConstrEnforceability().

◆ ATCheckPartitionsNotInUse()

static void ATCheckPartitionsNotInUse	(	Relation	rel,
		LOCKMODE	lockmode
	)

static

Definition at line 7002 of file tablecmds.c.

{
    if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
    {
        List       *inh;
        ListCell   *cell;
 
        inh = find_all_inheritors(RelationGetRelid(rel), lockmode, NULL);
        /* first element is the parent rel; must ignore it */
        for_each_from(cell, inh, 1)
        {
            Relation    childrel;
 
            /* find_all_inheritors already got lock */
            childrel = table_open(lfirst_oid(cell), NoLock);
            CheckAlterTableIsSafe(childrel);
            table_close(childrel, NoLock);
        }
        list_free(inh);
    }
}

References CheckAlterTableIsSafe(), fb(), find_all_inheritors(), for_each_from, lfirst_oid, list_free(), NoLock, RelationData::rd_rel, RelationGetRelid, table_close(), and table_open().

Referenced by ATPrepCmd().

◆ ATColumnChangeRequiresRewrite()

static bool ATColumnChangeRequiresRewrite	(	Node *	expr,
		AttrNumber	varattno
	)

static

Definition at line 15216 of file tablecmds.c.

{
    Assert(expr != NULL);
 
    for (;;)
    {
        /* only one varno, so no need to check that */
        if (IsA(expr, Var) && ((Var *) expr)->varattno == varattno)
            return false;
        else if (IsA(expr, RelabelType))
            expr = (Node *) ((RelabelType *) expr)->arg;
        else if (IsA(expr, CoerceToDomain))
        {
            CoerceToDomain *d = (CoerceToDomain *) expr;
 
            if (DomainHasConstraints(d->resulttype, NULL))
                return true;
            expr = (Node *) d->arg;
        }
        else if (IsA(expr, FuncExpr))
        {
            FuncExpr   *f = (FuncExpr *) expr;
 
            switch (f->funcid)
            {
                case F_TIMESTAMPTZ_TIMESTAMP:
                case F_TIMESTAMP_TIMESTAMPTZ:
                    if (TimestampTimestampTzRequiresRewrite())
                        return true;
                    else
                        expr = linitial(f->args);
                    break;
                default:
                    return true;
            }
        }
        else
            return true;
    }
}

References arg, CoerceToDomain::arg, FuncExpr::args, Assert, DomainHasConstraints(), fb(), FuncExpr::funcid, IsA, linitial, CoerceToDomain::resulttype, and TimestampTimestampTzRequiresRewrite().

Referenced by ATPrepAlterColumnType().

◆ ATController()

static void ATController	(	AlterTableStmt *	parsetree,
		Relation	rel,
		List *	cmds,
		bool	recurse,
		LOCKMODE	lockmode,
		AlterTableUtilityContext *	context
	)

static

Definition at line 4967 of file tablecmds.c.

{
    List       *wqueue = NIL;
    ListCell   *lcmd;
 
    /* Phase 1: preliminary examination of commands, create work queue */
    foreach(lcmd, cmds)
    {
        AlterTableCmd *cmd = (AlterTableCmd *) lfirst(lcmd);
 
        ATPrepCmd(&wqueue, rel, cmd, recurse, false, lockmode, context);
    }
 
    /* Close the relation, but keep lock until commit */
    relation_close(rel, NoLock);
 
    /* Phase 2: update system catalogs */
    ATRewriteCatalogs(&wqueue, lockmode, context);
 
    /* Phase 3: scan/rewrite tables as needed, and run afterStmts */
    ATRewriteTables(parsetree, &wqueue, lockmode, context);
}

References ATPrepCmd(), ATRewriteCatalogs(), ATRewriteTables(), fb(), lfirst, NIL, NoLock, and relation_close().

Referenced by AlterTable(), and AlterTableInternal().

◆ ATDetachCheckNoForeignKeyRefs()

static void ATDetachCheckNoForeignKeyRefs ( Relation partition )

static

Definition at line 22741 of file tablecmds.c.

{
    List       *constraints;
    ListCell   *cell;
 
    constraints = GetParentedForeignKeyRefs(partition);
 
    foreach(cell, constraints)
    {
        Oid         constrOid = lfirst_oid(cell);
        HeapTuple   tuple;
        Form_pg_constraint constrForm;
        Relation    rel;
        Trigger     trig = {0};
 
        tuple = SearchSysCache1(CONSTROID, ObjectIdGetDatum(constrOid));
        if (!HeapTupleIsValid(tuple))
            elog(ERROR, "cache lookup failed for constraint %u", constrOid);
        constrForm = (Form_pg_constraint) GETSTRUCT(tuple);
 
        Assert(OidIsValid(constrForm->conparentid));
        Assert(constrForm->confrelid == RelationGetRelid(partition));
 
        /* prevent data changes into the referencing table until commit */
        rel = table_open(constrForm->conrelid, ShareLock);
 
        trig.tgoid = InvalidOid;
        trig.tgname = NameStr(constrForm->conname);
        trig.tgenabled = TRIGGER_FIRES_ON_ORIGIN;
        trig.tgisinternal = true;
        trig.tgconstrrelid = RelationGetRelid(partition);
        trig.tgconstrindid = constrForm->conindid;
        trig.tgconstraint = constrForm->oid;
        trig.tgdeferrable = false;
        trig.tginitdeferred = false;
        /* we needn't fill in remaining fields */
 
        RI_PartitionRemove_Check(&trig, rel, partition);
 
        ReleaseSysCache(tuple);
 
        table_close(rel, NoLock);
    }
}

References Assert, elog, ERROR, fb(), Form_pg_constraint, GetParentedForeignKeyRefs(), GETSTRUCT(), HeapTupleIsValid, InvalidOid, lfirst_oid, NameStr, NoLock, ObjectIdGetDatum(), OidIsValid, RelationGetRelid, ReleaseSysCache(), RI_PartitionRemove_Check(), SearchSysCache1(), ShareLock, table_close(), table_open(), and TRIGGER_FIRES_ON_ORIGIN.

Referenced by ATExecDetachPartition().

◆ AtEOSubXact_on_commit_actions()

void AtEOSubXact_on_commit_actions	(	bool	isCommit,
		SubTransactionId	mySubid,
		SubTransactionId	parentSubid
	)

Definition at line 20162 of file tablecmds.c.

{
    ListCell   *cur_item;
 
    foreach(cur_item, on_commits)
    {
        OnCommitItem *oc = (OnCommitItem *) lfirst(cur_item);
 
        if (!isCommit && oc->creating_subid == mySubid)
        {
            /* cur_item must be removed */
            on_commits = foreach_delete_current(on_commits, cur_item);
            pfree(oc);
        }
        else
        {
            /* cur_item must be preserved */
            if (oc->creating_subid == mySubid)
                oc->creating_subid = parentSubid;
            if (oc->deleting_subid == mySubid)
                oc->deleting_subid = isCommit ? parentSubid : InvalidSubTransactionId;
        }
    }
}

References fb(), foreach_delete_current, InvalidSubTransactionId, lfirst, on_commits, and pfree().

Referenced by AbortSubTransaction(), and CommitSubTransaction().

◆ AtEOXact_on_commit_actions()

void AtEOXact_on_commit_actions ( bool isCommit )

Definition at line 20130 of file tablecmds.c.

{
    ListCell   *cur_item;
 
    foreach(cur_item, on_commits)
    {
        OnCommitItem *oc = (OnCommitItem *) lfirst(cur_item);
 
        if (isCommit ? oc->deleting_subid != InvalidSubTransactionId :
            oc->creating_subid != InvalidSubTransactionId)
        {
            /* cur_item must be removed */
            on_commits = foreach_delete_current(on_commits, cur_item);
            pfree(oc);
        }
        else
        {
            /* cur_item must be preserved */
            oc->creating_subid = InvalidSubTransactionId;
            oc->deleting_subid = InvalidSubTransactionId;
        }
    }
}

References fb(), foreach_delete_current, InvalidSubTransactionId, lfirst, on_commits, and pfree().

Referenced by AbortTransaction(), CommitTransaction(), and PrepareTransaction().

◆ ATExecAddColumn()

static ObjectAddress ATExecAddColumn	(	List **	wqueue,
		AlteredTableInfo *	tab,
		Relation	rel,
		AlterTableCmd **	cmd,
		bool	recurse,
		bool	recursing,
		LOCKMODE	lockmode,
		AlterTablePass	cur_pass,
		AlterTableUtilityContext *	context
	)

static

Definition at line 7358 of file tablecmds.c.

{
    Oid         myrelid = RelationGetRelid(rel);
    ColumnDef  *colDef = castNode(ColumnDef, (*cmd)->def);
    bool        if_not_exists = (*cmd)->missing_ok;
    Relation    pgclass,
                attrdesc;
    HeapTuple   reltup;
    Form_pg_class relform;
    Form_pg_attribute attribute;
    int         newattnum;
    char        relkind;
    Expr       *defval;
    List       *children;
    ListCell   *child;
    AlterTableCmd *childcmd;
    ObjectAddress address;
    TupleDesc   tupdesc;
 
    /* since this function recurses, it could be driven to stack overflow */
    check_stack_depth();
 
    /* At top level, permission check was done in ATPrepCmd, else do it */
    if (recursing)
        ATSimplePermissions((*cmd)->subtype, rel,
                            ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
 
    if (rel->rd_rel->relispartition && !recursing)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot add column to a partition")));
 
    attrdesc = table_open(AttributeRelationId, RowExclusiveLock);
 
    /*
     * Are we adding the column to a recursion child?  If so, check whether to
     * merge with an existing definition for the column.  If we do merge, we
     * must not recurse.  Children will already have the column, and recursing
     * into them would mess up attinhcount.
     */
    if (colDef->inhcount > 0)
    {
        HeapTuple   tuple;
 
        /* Does child already have a column by this name? */
        tuple = SearchSysCacheCopyAttName(myrelid, colDef->colname);
        if (HeapTupleIsValid(tuple))
        {
            Form_pg_attribute childatt = (Form_pg_attribute) GETSTRUCT(tuple);
            Oid         ctypeId;
            int32       ctypmod;
            Oid         ccollid;
 
            /* Child column must match on type, typmod, and collation */
            typenameTypeIdAndMod(NULL, colDef->typeName, &ctypeId, &ctypmod);
            if (ctypeId != childatt->atttypid ||
                ctypmod != childatt->atttypmod)
                ereport(ERROR,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("child table \"%s\" has different type for column \"%s\"",
                                RelationGetRelationName(rel), colDef->colname)));
            ccollid = GetColumnDefCollation(NULL, colDef, ctypeId);
            if (ccollid != childatt->attcollation)
                ereport(ERROR,
                        (errcode(ERRCODE_COLLATION_MISMATCH),
                         errmsg("child table \"%s\" has different collation for column \"%s\"",
                                RelationGetRelationName(rel), colDef->colname),
                         errdetail("\"%s\" versus \"%s\"",
                                   get_collation_name(ccollid),
                                   get_collation_name(childatt->attcollation))));
 
            /* Bump the existing child att's inhcount */
            if (pg_add_s16_overflow(childatt->attinhcount, 1,
                                    &childatt->attinhcount))
                ereport(ERROR,
                        errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
                        errmsg("too many inheritance parents"));
            CatalogTupleUpdate(attrdesc, &tuple->t_self, tuple);
 
            heap_freetuple(tuple);
 
            /* Inform the user about the merge */
            ereport(NOTICE,
                    (errmsg("merging definition of column \"%s\" for child \"%s\"",
                            colDef->colname, RelationGetRelationName(rel))));
 
            table_close(attrdesc, RowExclusiveLock);
 
            /* Make the child column change visible */
            CommandCounterIncrement();
 
            return InvalidObjectAddress;
        }
    }
 
    /* skip if the name already exists and if_not_exists is true */
    if (!check_for_column_name_collision(rel, colDef->colname, if_not_exists))
    {
        table_close(attrdesc, RowExclusiveLock);
        return InvalidObjectAddress;
    }
 
    /*
     * Okay, we need to add the column, so go ahead and do parse
     * transformation.  This can result in queueing up, or even immediately
     * executing, subsidiary operations (such as creation of unique indexes);
     * so we mustn't do it until we have made the if_not_exists check.
     *
     * When recursing, the command was already transformed and we needn't do
     * so again.  Also, if context isn't given we can't transform.  (That
     * currently happens only for AT_AddColumnToView; we expect that view.c
     * passed us a ColumnDef that doesn't need work.)
     */
    if (context != NULL && !recursing)
    {
        *cmd = ATParseTransformCmd(wqueue, tab, rel, *cmd, recurse, lockmode,
                                   cur_pass, context);
        Assert(*cmd != NULL);
        colDef = castNode(ColumnDef, (*cmd)->def);
    }
 
    /*
     * Regular inheritance children are independent enough not to inherit the
     * identity column from parent hence cannot recursively add identity
     * column if the table has inheritance children.
     *
     * Partitions, on the other hand, are integral part of a partitioned table
     * and inherit identity column.  Hence propagate identity column down the
     * partition hierarchy.
     */
    if (colDef->identity &&
        recurse &&
        rel->rd_rel->relkind != RELKIND_PARTITIONED_TABLE &&
        find_inheritance_children(myrelid, NoLock) != NIL)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("cannot recursively add identity column to table that has child tables")));
 
    pgclass = table_open(RelationRelationId, RowExclusiveLock);
 
    reltup = SearchSysCacheCopy1(RELOID, ObjectIdGetDatum(myrelid));
    if (!HeapTupleIsValid(reltup))
        elog(ERROR, "cache lookup failed for relation %u", myrelid);
    relform = (Form_pg_class) GETSTRUCT(reltup);
    relkind = relform->relkind;
 
    /* Determine the new attribute's number */
    newattnum = relform->relnatts + 1;
    if (newattnum > MaxHeapAttributeNumber)
        ereport(ERROR,
                (errcode(ERRCODE_TOO_MANY_COLUMNS),
                 errmsg("tables can have at most %d columns",
                        MaxHeapAttributeNumber)));
 
    /*
     * Construct new attribute's pg_attribute entry.
     */
    tupdesc = BuildDescForRelation(list_make1(colDef));
 
    attribute = TupleDescAttr(tupdesc, 0);
 
    /* Fix up attribute number */
    attribute->attnum = newattnum;
 
    /* make sure datatype is legal for a column */
    CheckAttributeType(NameStr(attribute->attname), attribute->atttypid, attribute->attcollation,
                       list_make1_oid(rel->rd_rel->reltype),
                       (attribute->attgenerated == ATTRIBUTE_GENERATED_VIRTUAL ? CHKATYPE_IS_VIRTUAL : 0));
 
    InsertPgAttributeTuples(attrdesc, tupdesc, myrelid, NULL, NULL);
 
    table_close(attrdesc, RowExclusiveLock);
 
    /*
     * Update pg_class tuple as appropriate
     */
    relform->relnatts = newattnum;
 
    CatalogTupleUpdate(pgclass, &reltup->t_self, reltup);
 
    heap_freetuple(reltup);
 
    /* Post creation hook for new attribute */
    InvokeObjectPostCreateHook(RelationRelationId, myrelid, newattnum);
 
    table_close(pgclass, RowExclusiveLock);
 
    /* Make the attribute's catalog entry visible */
    CommandCounterIncrement();
 
    /*
     * Store the DEFAULT, if any, in the catalogs
     */
    if (colDef->raw_default)
    {
        RawColumnDefault *rawEnt;
 
        rawEnt = palloc_object(RawColumnDefault);
        rawEnt->attnum = attribute->attnum;
        rawEnt->raw_default = copyObject(colDef->raw_default);
        rawEnt->generated = colDef->generated;
 
        /*
         * This function is intended for CREATE TABLE, so it processes a
         * _list_ of defaults, but we just do one.
         */
        AddRelationNewConstraints(rel, list_make1(rawEnt), NIL,
                                  false, true, false, NULL);
 
        /* Make the additional catalog changes visible */
        CommandCounterIncrement();
    }
 
    /*
     * Tell Phase 3 to fill in the default expression, if there is one.
     *
     * If there is no default, Phase 3 doesn't have to do anything, because
     * that effectively means that the default is NULL.  The heap tuple access
     * routines always check for attnum > # of attributes in tuple, and return
     * NULL if so, so without any modification of the tuple data we will get
     * the effect of NULL values in the new column.
     *
     * An exception occurs when the new column is of a domain type: the domain
     * might have a not-null constraint, or a check constraint that indirectly
     * rejects nulls.  If there are any domain constraints then we construct
     * an explicit NULL default value that will be passed through
     * CoerceToDomain processing.  (This is a tad inefficient, since it causes
     * rewriting the table which we really wouldn't have to do; but we do it
     * to preserve the historical behavior that such a failure will be raised
     * only if the table currently contains some rows.)
     *
     * Note: we use build_column_default, and not just the cooked default
     * returned by AddRelationNewConstraints, so that the right thing happens
     * when a datatype's default applies.
     *
     * Note: it might seem that this should happen at the end of Phase 2, so
     * that the effects of subsequent subcommands can be taken into account.
     * It's intentional that we do it now, though.  The new column should be
     * filled according to what is said in the ADD COLUMN subcommand, so that
     * the effects are the same as if this subcommand had been run by itself
     * and the later subcommands had been issued in new ALTER TABLE commands.
     *
     * We can skip this entirely for relations without storage, since Phase 3
     * is certainly not going to touch them.
     */
    if (RELKIND_HAS_STORAGE(relkind))
    {
        bool        has_domain_constraints;
        bool        has_missing = false;
 
        /*
         * For an identity column, we can't use build_column_default(),
         * because the sequence ownership isn't set yet.  So do it manually.
         */
        if (colDef->identity)
        {
            NextValueExpr *nve = makeNode(NextValueExpr);
 
            nve->seqid = RangeVarGetRelid(colDef->identitySequence, NoLock, false);
            nve->typeId = attribute->atttypid;
 
            defval = (Expr *) nve;
        }
        else
            defval = (Expr *) build_column_default(rel, attribute->attnum);
 
        /* Build CoerceToDomain(NULL) expression if needed */
        has_domain_constraints = DomainHasConstraints(attribute->atttypid, NULL);
        if (!defval && has_domain_constraints)
        {
            Oid         baseTypeId;
            int32       baseTypeMod;
            Oid         baseTypeColl;
 
            baseTypeMod = attribute->atttypmod;
            baseTypeId = getBaseTypeAndTypmod(attribute->atttypid, &baseTypeMod);
            baseTypeColl = get_typcollation(baseTypeId);
            defval = (Expr *) makeNullConst(baseTypeId, baseTypeMod, baseTypeColl);
            defval = (Expr *) coerce_to_target_type(NULL,
                                                    (Node *) defval,
                                                    baseTypeId,
                                                    attribute->atttypid,
                                                    attribute->atttypmod,
                                                    COERCION_ASSIGNMENT,
                                                    COERCE_IMPLICIT_CAST,
                                                    -1);
            if (defval == NULL) /* should not happen */
                elog(ERROR, "failed to coerce base type to domain");
        }
 
        if (defval)
        {
            NewColumnValue *newval;
 
            /* Prepare defval for execution, either here or in Phase 3 */
            defval = expression_planner(defval);
 
            /* Add the new default to the newvals list */
            newval = palloc0_object(NewColumnValue);
            newval->attnum = attribute->attnum;
            newval->expr = defval;
            newval->is_generated = (colDef->generated != '\0');
 
            tab->newvals = lappend(tab->newvals, newval);
 
            /*
             * Attempt to skip a complete table rewrite by storing the
             * specified DEFAULT value outside of the heap.  This is only
             * allowed for plain relations and non-generated columns, and the
             * default expression can't be volatile (stable is OK).  Note that
             * contain_volatile_functions deems CoerceToDomain immutable, but
             * here we consider that coercion to a domain with constraints is
             * volatile; else it might fail even when the table is empty.
             */
            if (rel->rd_rel->relkind == RELKIND_RELATION &&
                !colDef->generated &&
                !has_domain_constraints &&
                !contain_volatile_functions((Node *) defval))
            {
                EState     *estate;
                ExprState  *exprState;
                Datum       missingval;
                bool        missingIsNull;
 
                /* Evaluate the default expression */
                estate = CreateExecutorState();
                exprState = ExecPrepareExpr(defval, estate);
                missingval = ExecEvalExpr(exprState,
                                          GetPerTupleExprContext(estate),
                                          &missingIsNull);
                /* If it turns out NULL, nothing to do; else store it */
                if (!missingIsNull)
                {
                    StoreAttrMissingVal(rel, attribute->attnum, missingval);
                    /* Make the additional catalog change visible */
                    CommandCounterIncrement();
                    has_missing = true;
                }
                FreeExecutorState(estate);
            }
            else
            {
                /*
                 * Failed to use missing mode.  We have to do a table rewrite
                 * to install the value --- unless it's a virtual generated
                 * column.
                 */
                if (colDef->generated != ATTRIBUTE_GENERATED_VIRTUAL)
                    tab->rewrite |= AT_REWRITE_DEFAULT_VAL;
            }
        }
 
        if (!has_missing)
        {
            /*
             * If the new column is NOT NULL, and there is no missing value,
             * tell Phase 3 it needs to check for NULLs.
             */
            tab->verify_new_notnull |= colDef->is_not_null;
        }
    }
 
    /*
     * Add needed dependency entries for the new column.
     */
    add_column_datatype_dependency(myrelid, newattnum, attribute->atttypid);
    add_column_collation_dependency(myrelid, newattnum, attribute->attcollation);
 
    /*
     * Propagate to children as appropriate.  Unlike most other ALTER
     * routines, we have to do this one level of recursion at a time; we can't
     * use find_all_inheritors to do it in one pass.
     */
    children =
        find_inheritance_children(RelationGetRelid(rel), lockmode);
 
    /*
     * If we are told not to recurse, there had better not be any child
     * tables; else the addition would put them out of step.
     */
    if (children && !recurse)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("column must be added to child tables too")));
 
    /* Children should see column as singly inherited */
    if (!recursing)
    {
        childcmd = copyObject(*cmd);
        colDef = castNode(ColumnDef, childcmd->def);
        colDef->inhcount = 1;
        colDef->is_local = false;
    }
    else
        childcmd = *cmd;        /* no need to copy again */
 
    foreach(child, children)
    {
        Oid         childrelid = lfirst_oid(child);
        Relation    childrel;
        AlteredTableInfo *childtab;
 
        /* find_inheritance_children already got lock */
        childrel = table_open(childrelid, NoLock);
        CheckAlterTableIsSafe(childrel);
 
        /* Find or create work queue entry for this table */
        childtab = ATGetQueueEntry(wqueue, childrel);
 
        /* Recurse to child; return value is ignored */
        ATExecAddColumn(wqueue, childtab, childrel,
                        &childcmd, recurse, true,
                        lockmode, cur_pass, context);
 
        table_close(childrel, NoLock);
    }
 
    ObjectAddressSubSet(address, RelationRelationId, myrelid, newattnum);
    return address;
}

References add_column_collation_dependency(), add_column_datatype_dependency(), AddRelationNewConstraints(), Assert, AT_REWRITE_DEFAULT_VAL, ATExecAddColumn(), ATGetQueueEntry(), ATParseTransformCmd(), ATSimplePermissions(), ATT_FOREIGN_TABLE, ATT_PARTITIONED_TABLE, ATT_TABLE, build_column_default(), BuildDescForRelation(), castNode, CatalogTupleUpdate(), check_for_column_name_collision(), check_stack_depth(), CheckAlterTableIsSafe(), CheckAttributeType(), CHKATYPE_IS_VIRTUAL, COERCE_IMPLICIT_CAST, coerce_to_target_type(), COERCION_ASSIGNMENT, CommandCounterIncrement(), contain_volatile_functions(), copyObject, CreateExecutorState(), DomainHasConstraints(), elog, ereport, errcode(), errdetail(), errmsg, ERROR, ExecEvalExpr(), ExecPrepareExpr(), expression_planner(), fb(), find_inheritance_children(), FreeExecutorState(), get_collation_name(), get_typcollation(), getBaseTypeAndTypmod(), GetColumnDefCollation(), GetPerTupleExprContext, GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InsertPgAttributeTuples(), InvalidObjectAddress, InvokeObjectPostCreateHook, lappend(), lfirst_oid, list_make1, list_make1_oid, makeNode, makeNullConst(), MaxHeapAttributeNumber, NameStr, newval, AlteredTableInfo::newvals, NIL, NoLock, NOTICE, ObjectAddressSubSet, ObjectIdGetDatum(), palloc0_object, palloc_object, pg_add_s16_overflow(), RangeVarGetRelid, RelationData::rd_rel, RelationGetRelationName, RelationGetRelid, AlteredTableInfo::rewrite, RowExclusiveLock, SearchSysCacheCopy1, SearchSysCacheCopyAttName(), StoreAttrMissingVal(), HeapTupleData::t_self, table_close(), table_open(), TupleDescAttr(), typenameTypeIdAndMod(), and AlteredTableInfo::verify_new_notnull.

Referenced by ATExecAddColumn(), and ATExecCmd().

◆ ATExecAddConstraint()

static ObjectAddress ATExecAddConstraint	(	List **	wqueue,
		AlteredTableInfo *	tab,
		Relation	rel,
		Constraint *	newConstraint,
		bool	recurse,
		bool	is_readd,
		LOCKMODE	lockmode
	)

static

Definition at line 9932 of file tablecmds.c.

{
    ObjectAddress address = InvalidObjectAddress;
 
    Assert(IsA(newConstraint, Constraint));
 
    /*
     * Currently, we only expect to see CONSTR_CHECK, CONSTR_NOTNULL and
     * CONSTR_FOREIGN nodes arriving here (see the preprocessing done in
     * parse_utilcmd.c).
     */
    switch (newConstraint->contype)
    {
        case CONSTR_CHECK:
        case CONSTR_NOTNULL:
            address =
                ATAddCheckNNConstraint(wqueue, tab, rel,
                                       newConstraint, recurse, false, is_readd,
                                       lockmode);
            break;
 
        case CONSTR_FOREIGN:
 
            /*
             * Assign or validate constraint name
             */
            if (newConstraint->conname)
            {
                if (ConstraintNameIsUsed(CONSTRAINT_RELATION,
                                         RelationGetRelid(rel),
                                         newConstraint->conname))
                    ereport(ERROR,
                            (errcode(ERRCODE_DUPLICATE_OBJECT),
                             errmsg("constraint \"%s\" for relation \"%s\" already exists",
                                    newConstraint->conname,
                                    RelationGetRelationName(rel))));
            }
            else
                newConstraint->conname =
                    ChooseConstraintName(RelationGetRelationName(rel),
                                         ChooseForeignKeyConstraintNameAddition(newConstraint->fk_attrs),
                                         "fkey",
                                         RelationGetNamespace(rel),
                                         NIL);
 
            address = ATAddForeignKeyConstraint(wqueue, tab, rel,
                                                newConstraint,
                                                recurse, false,
                                                lockmode);
            break;
 
        default:
            elog(ERROR, "unrecognized constraint type: %d",
                 (int) newConstraint->contype);
    }
 
    return address;
}

References Assert, ATAddCheckNNConstraint(), ATAddForeignKeyConstraint(), ChooseConstraintName(), ChooseForeignKeyConstraintNameAddition(), CONSTR_CHECK, CONSTR_FOREIGN, CONSTR_NOTNULL, CONSTRAINT_RELATION, ConstraintNameIsUsed(), elog, ereport, errcode(), ERRCODE_DUPLICATE_OBJECT, errmsg, ERROR, fb(), InvalidObjectAddress, IsA, NIL, RelationGetNamespace, RelationGetRelationName, and RelationGetRelid.

Referenced by ATExecCmd().

◆ ATExecAddIdentity()

static ObjectAddress ATExecAddIdentity	(	Relation	rel,
		const char *	colName,
		Node *	def,
		LOCKMODE	lockmode,
		bool	recurse,
		bool	recursing
	)

static

Definition at line 8381 of file tablecmds.c.

{
    Relation    attrelation;
    HeapTuple   tuple;
    Form_pg_attribute attTup;
    AttrNumber  attnum;
    ObjectAddress address;
    ColumnDef  *cdef = castNode(ColumnDef, def);
    bool        ispartitioned;
 
    ispartitioned = (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE);
    if (ispartitioned && !recurse)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("cannot add identity to a column of only the partitioned table"),
                 errhint("Do not specify the ONLY keyword.")));
 
    if (rel->rd_rel->relispartition && !recursing)
        ereport(ERROR,
                errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                errmsg("cannot add identity to a column of a partition"));
 
    attrelation = table_open(AttributeRelationId, RowExclusiveLock);
 
    tuple = SearchSysCacheCopyAttName(RelationGetRelid(rel), colName);
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel))));
    attTup = (Form_pg_attribute) GETSTRUCT(tuple);
    attnum = attTup->attnum;
 
    /* Can't alter a system attribute */
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"",
                        colName)));
 
    /*
     * Creating a column as identity implies NOT NULL, so adding the identity
     * to an existing column that is not NOT NULL would create a state that
     * cannot be reproduced without contortions.
     */
    if (!attTup->attnotnull)
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                 errmsg("column \"%s\" of relation \"%s\" must be declared NOT NULL before identity can be added",
                        colName, RelationGetRelationName(rel))));
 
    /*
     * On the other hand, if a not-null constraint exists, then verify that
     * it's compatible.
     */
    if (attTup->attnotnull)
    {
        HeapTuple   contup;
        Form_pg_constraint conForm;
 
        contup = findNotNullConstraintAttnum(RelationGetRelid(rel),
                                             attnum);
        if (!HeapTupleIsValid(contup))
            elog(ERROR, "cache lookup failed for not-null constraint on column \"%s\" of relation \"%s\"",
                 colName, RelationGetRelationName(rel));
 
        conForm = (Form_pg_constraint) GETSTRUCT(contup);
        if (!conForm->convalidated)
            ereport(ERROR,
                    errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                    errmsg("incompatible NOT VALID constraint \"%s\" on relation \"%s\"",
                           NameStr(conForm->conname), RelationGetRelationName(rel)),
                    errhint("You might need to validate it using %s.",
                            "ALTER TABLE ... VALIDATE CONSTRAINT"));
    }
 
    if (attTup->attidentity)
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                 errmsg("column \"%s\" of relation \"%s\" is already an identity column",
                        colName, RelationGetRelationName(rel))));
 
    if (attTup->atthasdef)
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                 errmsg("column \"%s\" of relation \"%s\" already has a default value",
                        colName, RelationGetRelationName(rel))));
 
    attTup->attidentity = cdef->identity;
    CatalogTupleUpdate(attrelation, &tuple->t_self, tuple);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel),
                              attTup->attnum);
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
    heap_freetuple(tuple);
 
    table_close(attrelation, RowExclusiveLock);
 
    /*
     * Recurse to propagate the identity column to partitions.  Identity is
     * not inherited in regular inheritance children.
     */
    if (recurse && ispartitioned)
    {
        List       *children;
        ListCell   *lc;
 
        children = find_inheritance_children(RelationGetRelid(rel), lockmode);
 
        foreach(lc, children)
        {
            Relation    childrel;
 
            childrel = table_open(lfirst_oid(lc), NoLock);
            ATExecAddIdentity(childrel, colName, def, lockmode, recurse, true);
            table_close(childrel, NoLock);
        }
    }
 
    return address;
}

References ATExecAddIdentity(), attnum, castNode, CatalogTupleUpdate(), elog, ereport, errcode(), errhint(), errmsg, ERROR, fb(), find_inheritance_children(), findNotNullConstraintAttnum(), Form_pg_constraint, GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InvokeObjectPostAlterHook, lfirst_oid, NameStr, NoLock, ObjectAddressSubSet, RelationData::rd_rel, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, SearchSysCacheCopyAttName(), HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecAddIdentity(), and ATExecCmd().

◆ ATExecAddIndex()

static ObjectAddress ATExecAddIndex	(	AlteredTableInfo *	tab,
		Relation	rel,
		IndexStmt *	stmt,
		bool	is_rebuild,
		LOCKMODE	lockmode
	)

static

Definition at line 9755 of file tablecmds.c.

{
    bool        check_rights;
    bool        skip_build;
    bool        quiet;
    ObjectAddress address;
 
    Assert(IsA(stmt, IndexStmt));
    Assert(!stmt->concurrent);
 
    /* The IndexStmt has already been through transformIndexStmt */
    Assert(stmt->transformed);
 
    /* suppress schema rights check when rebuilding existing index */
    check_rights = !is_rebuild;
    /* skip index build if phase 3 will do it or we're reusing an old one */
    skip_build = tab->rewrite > 0 || RelFileNumberIsValid(stmt->oldNumber);
    /* suppress notices when rebuilding existing index */
    quiet = is_rebuild;
 
    address = DefineIndex(NULL,
                          RelationGetRelid(rel),
                          stmt,
                          InvalidOid,   /* no predefined OID */
                          InvalidOid,   /* no parent index */
                          InvalidOid,   /* no parent constraint */
                          -1,   /* total_parts unknown */
                          true, /* is_alter_table */
                          check_rights,
                          false,    /* check_not_in_use - we did it already */
                          skip_build,
                          quiet);
 
    /*
     * If TryReuseIndex() stashed a relfilenumber for us, we used it for the
     * new index instead of building from scratch.  Restore associated fields.
     * This may store InvalidSubTransactionId in both fields, in which case
     * relcache.c will assume it can rebuild the relcache entry.  Hence, do
     * this after the CCI that made catalog rows visible to any rebuild.  The
     * DROP of the old edition of this index will have scheduled the storage
     * for deletion at commit, so cancel that pending deletion.
     */
    if (RelFileNumberIsValid(stmt->oldNumber))
    {
        Relation    irel = index_open(address.objectId, NoLock);
 
        irel->rd_createSubid = stmt->oldCreateSubid;
        irel->rd_firstRelfilelocatorSubid = stmt->oldFirstRelfilelocatorSubid;
        RelationPreserveStorage(irel->rd_locator, true);
        index_close(irel, NoLock);
    }
 
    return address;
}

References Assert, DefineIndex(), fb(), index_close(), index_open(), InvalidOid, IsA, NoLock, ObjectAddress::objectId, RelationData::rd_createSubid, RelationData::rd_firstRelfilelocatorSubid, RelationData::rd_locator, RelationGetRelid, RelationPreserveStorage(), RelFileNumberIsValid, AlteredTableInfo::rewrite, and stmt.

Referenced by ATExecCmd().

◆ ATExecAddIndexConstraint()

static ObjectAddress ATExecAddIndexConstraint	(	AlteredTableInfo *	tab,
		Relation	rel,
		IndexStmt *	stmt,
		LOCKMODE	lockmode
	)

static

Definition at line 9840 of file tablecmds.c.

{
    Oid         index_oid = stmt->indexOid;
    Relation    indexRel;
    char       *indexName;
    IndexInfo  *indexInfo;
    char       *constraintName;
    char        constraintType;
    ObjectAddress address;
    uint16      flags;
 
    Assert(IsA(stmt, IndexStmt));
    Assert(OidIsValid(index_oid));
    Assert(stmt->isconstraint);
 
    /*
     * Doing this on partitioned tables is not a simple feature to implement,
     * so let's punt for now.
     */
    if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("ALTER TABLE / ADD CONSTRAINT USING INDEX is not supported on partitioned tables")));
 
    indexRel = index_open(index_oid, AccessShareLock);
 
    indexName = pstrdup(RelationGetRelationName(indexRel));
 
    indexInfo = BuildIndexInfo(indexRel);
 
    /* this should have been checked at parse time */
    if (!indexInfo->ii_Unique)
        elog(ERROR, "index \"%s\" is not unique", indexName);
 
    /*
     * Determine name to assign to constraint.  We require a constraint to
     * have the same name as the underlying index; therefore, use the index's
     * existing name as the default constraint name, and if the user
     * explicitly gives some other name for the constraint, rename the index
     * to match.
     */
    constraintName = stmt->idxname;
    if (constraintName == NULL)
        constraintName = indexName;
    else if (strcmp(constraintName, indexName) != 0)
    {
        ereport(NOTICE,
                (errmsg("ALTER TABLE / ADD CONSTRAINT USING INDEX will rename index \"%s\" to \"%s\"",
                        indexName, constraintName)));
        RenameRelationInternal(index_oid, constraintName, false, true);
    }
 
    /* Extra checks needed if making primary key */
    if (stmt->primary)
        index_check_primary_key(rel, indexInfo, true, stmt);
 
    /* Note we currently don't support EXCLUSION constraints here */
    if (stmt->primary)
        constraintType = CONSTRAINT_PRIMARY;
    else
        constraintType = CONSTRAINT_UNIQUE;
 
    /* Create the catalog entries for the constraint */
    flags = INDEX_CONSTR_CREATE_UPDATE_INDEX |
        INDEX_CONSTR_CREATE_REMOVE_OLD_DEPS |
        (stmt->initdeferred ? INDEX_CONSTR_CREATE_INIT_DEFERRED : 0) |
        (stmt->deferrable ? INDEX_CONSTR_CREATE_DEFERRABLE : 0) |
        (stmt->primary ? INDEX_CONSTR_CREATE_MARK_AS_PRIMARY : 0);
 
    address = index_constraint_create(rel,
                                      index_oid,
                                      InvalidOid,
                                      indexInfo,
                                      constraintName,
                                      constraintType,
                                      flags,
                                      allowSystemTableMods,
                                      false);   /* is_internal */
 
    index_close(indexRel, NoLock);
 
    return address;
}

References AccessShareLock, allowSystemTableMods, Assert, BuildIndexInfo(), elog, ereport, errcode(), errmsg, ERROR, fb(), IndexInfo::ii_Unique, index_check_primary_key(), index_close(), INDEX_CONSTR_CREATE_DEFERRABLE, INDEX_CONSTR_CREATE_INIT_DEFERRED, INDEX_CONSTR_CREATE_MARK_AS_PRIMARY, INDEX_CONSTR_CREATE_REMOVE_OLD_DEPS, INDEX_CONSTR_CREATE_UPDATE_INDEX, index_constraint_create(), index_open(), InvalidOid, IsA, NoLock, NOTICE, OidIsValid, pstrdup(), RelationData::rd_rel, RelationGetRelationName, RenameRelationInternal(), and stmt.

Referenced by ATExecCmd().

◆ ATExecAddInherit()

static ObjectAddress ATExecAddInherit	(	Relation	child_rel,
		RangeVar *	parent,
		LOCKMODE	lockmode
	)

static

Definition at line 17968 of file tablecmds.c.

{
    Relation    parent_rel;
    List       *children;
    ObjectAddress address;
    const char *trigger_name;
 
    /*
     * A self-exclusive lock is needed here.  See the similar case in
     * MergeAttributes() for a full explanation.
     */
    parent_rel = table_openrv(parent, ShareUpdateExclusiveLock);
 
    /*
     * Must be owner of both parent and child -- child was checked by
     * ATSimplePermissions call in ATPrepCmd
     */
    ATSimplePermissions(AT_AddInherit, parent_rel,
                        ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
 
    /* Permanent rels cannot inherit from temporary ones */
    if (parent_rel->rd_rel->relpersistence == RELPERSISTENCE_TEMP &&
        child_rel->rd_rel->relpersistence != RELPERSISTENCE_TEMP)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot inherit from temporary relation \"%s\"",
                        RelationGetRelationName(parent_rel))));
 
    /* If parent rel is temp, it must belong to this session */
    if (RELATION_IS_OTHER_TEMP(parent_rel))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot inherit from temporary relation of another session")));
 
    /* Ditto for the child */
    if (RELATION_IS_OTHER_TEMP(child_rel))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot inherit to temporary relation of another session")));
 
    /* Prevent partitioned tables from becoming inheritance parents */
    if (parent_rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot inherit from partitioned table \"%s\"",
                        parent->relname)));
 
    /* Likewise for partitions */
    if (parent_rel->rd_rel->relispartition)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot inherit from a partition")));
 
    /*
     * Prevent circularity by seeing if proposed parent inherits from child.
     * (In particular, this disallows making a rel inherit from itself.)
     *
     * This is not completely bulletproof because of race conditions: in
     * multi-level inheritance trees, someone else could concurrently be
     * making another inheritance link that closes the loop but does not join
     * either of the rels we have locked.  Preventing that seems to require
     * exclusive locks on the entire inheritance tree, which is a cure worse
     * than the disease.  find_all_inheritors() will cope with circularity
     * anyway, so don't sweat it too much.
     *
     * We use weakest lock we can on child's children, namely AccessShareLock.
     */
    children = find_all_inheritors(RelationGetRelid(child_rel),
                                   AccessShareLock, NULL);
 
    if (list_member_oid(children, RelationGetRelid(parent_rel)))
        ereport(ERROR,
                (errcode(ERRCODE_DUPLICATE_TABLE),
                 errmsg("circular inheritance not allowed"),
                 errdetail("\"%s\" is already a child of \"%s\".",
                           parent->relname,
                           RelationGetRelationName(child_rel))));
 
    /*
     * If child_rel has row-level triggers with transition tables, we
     * currently don't allow it to become an inheritance child.  See also
     * prohibitions in ATExecAttachPartition() and CreateTrigger().
     */
    trigger_name = FindTriggerIncompatibleWithInheritance(child_rel->trigdesc);
    if (trigger_name != NULL)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("trigger \"%s\" prevents table \"%s\" from becoming an inheritance child",
                        trigger_name, RelationGetRelationName(child_rel)),
                 errdetail("ROW triggers with transition tables are not supported in inheritance hierarchies.")));
 
    /* OK to create inheritance */
    CreateInheritance(child_rel, parent_rel, false);
 
    ObjectAddressSet(address, RelationRelationId,
                     RelationGetRelid(parent_rel));
 
    /* keep our lock on the parent relation until commit */
    table_close(parent_rel, NoLock);
 
    return address;
}

References AccessShareLock, AT_AddInherit, ATSimplePermissions(), ATT_FOREIGN_TABLE, ATT_PARTITIONED_TABLE, ATT_TABLE, CreateInheritance(), ereport, errcode(), errdetail(), errmsg, ERROR, fb(), find_all_inheritors(), FindTriggerIncompatibleWithInheritance(), list_member_oid(), NoLock, ObjectAddressSet, RELATION_IS_OTHER_TEMP, RelationGetRelationName, RelationGetRelid, RangeVar::relname, ShareUpdateExclusiveLock, table_close(), and table_openrv().

Referenced by ATExecCmd().

◆ ATExecAddOf()

static ObjectAddress ATExecAddOf	(	Relation	rel,
		const TypeName *	ofTypename,
		LOCKMODE	lockmode
	)

static

Definition at line 18919 of file tablecmds.c.

{
    Oid         relid = RelationGetRelid(rel);
    Type        typetuple;
    Form_pg_type typeform;
    Oid         typeid;
    Relation    inheritsRelation,
                relationRelation;
    SysScanDesc scan;
    ScanKeyData key;
    AttrNumber  table_attno,
                type_attno;
    TupleDesc   typeTupleDesc,
                tableTupleDesc;
    ObjectAddress tableobj,
                typeobj;
    HeapTuple   classtuple;
 
    /* Validate the type. */
    typetuple = typenameType(NULL, ofTypename, NULL);
    check_of_type(typetuple);
    typeform = (Form_pg_type) GETSTRUCT(typetuple);
    typeid = typeform->oid;
 
    /* Fail if the table has any inheritance parents. */
    inheritsRelation = table_open(InheritsRelationId, AccessShareLock);
    ScanKeyInit(&key,
                Anum_pg_inherits_inhrelid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(relid));
    scan = systable_beginscan(inheritsRelation, InheritsRelidSeqnoIndexId,
                              true, NULL, 1, &key);
    if (HeapTupleIsValid(systable_getnext(scan)))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("typed tables cannot inherit")));
    systable_endscan(scan);
    table_close(inheritsRelation, AccessShareLock);
 
    /*
     * Check the tuple descriptors for compatibility.  Unlike inheritance, we
     * require that the order also match.  However, attnotnull need not match.
     */
    typeTupleDesc = lookup_rowtype_tupdesc(typeid, -1);
    tableTupleDesc = RelationGetDescr(rel);
    table_attno = 1;
    for (type_attno = 1; type_attno <= typeTupleDesc->natts; type_attno++)
    {
        Form_pg_attribute type_attr,
                    table_attr;
        const char *type_attname,
                   *table_attname;
 
        /* Get the next non-dropped type attribute. */
        type_attr = TupleDescAttr(typeTupleDesc, type_attno - 1);
        if (type_attr->attisdropped)
            continue;
        type_attname = NameStr(type_attr->attname);
 
        /* Get the next non-dropped table attribute. */
        do
        {
            if (table_attno > tableTupleDesc->natts)
                ereport(ERROR,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("table is missing column \"%s\"",
                                type_attname)));
            table_attr = TupleDescAttr(tableTupleDesc, table_attno - 1);
            table_attno++;
        } while (table_attr->attisdropped);
        table_attname = NameStr(table_attr->attname);
 
        /* Compare name. */
        if (strncmp(table_attname, type_attname, NAMEDATALEN) != 0)
            ereport(ERROR,
                    (errcode(ERRCODE_DATATYPE_MISMATCH),
                     errmsg("table has column \"%s\" where type requires \"%s\"",
                            table_attname, type_attname)));
 
        /* Compare type. */
        if (table_attr->atttypid != type_attr->atttypid ||
            table_attr->atttypmod != type_attr->atttypmod ||
            table_attr->attcollation != type_attr->attcollation)
            ereport(ERROR,
                    (errcode(ERRCODE_DATATYPE_MISMATCH),
                     errmsg("table \"%s\" has different type for column \"%s\"",
                            RelationGetRelationName(rel), type_attname)));
    }
    ReleaseTupleDesc(typeTupleDesc);
 
    /* Any remaining columns at the end of the table had better be dropped. */
    for (; table_attno <= tableTupleDesc->natts; table_attno++)
    {
        Form_pg_attribute table_attr = TupleDescAttr(tableTupleDesc,
                                                     table_attno - 1);
 
        if (!table_attr->attisdropped)
            ereport(ERROR,
                    (errcode(ERRCODE_DATATYPE_MISMATCH),
                     errmsg("table has extra column \"%s\"",
                            NameStr(table_attr->attname))));
    }
 
    /* If the table was already typed, drop the existing dependency. */
    if (rel->rd_rel->reloftype)
        drop_parent_dependency(relid, TypeRelationId, rel->rd_rel->reloftype,
                               DEPENDENCY_NORMAL);
 
    /* Record a dependency on the new type. */
    tableobj.classId = RelationRelationId;
    tableobj.objectId = relid;
    tableobj.objectSubId = 0;
    typeobj.classId = TypeRelationId;
    typeobj.objectId = typeid;
    typeobj.objectSubId = 0;
    recordDependencyOn(&tableobj, &typeobj, DEPENDENCY_NORMAL);
 
    /* Update pg_class.reloftype */
    relationRelation = table_open(RelationRelationId, RowExclusiveLock);
    classtuple = SearchSysCacheCopy1(RELOID, ObjectIdGetDatum(relid));
    if (!HeapTupleIsValid(classtuple))
        elog(ERROR, "cache lookup failed for relation %u", relid);
    ((Form_pg_class) GETSTRUCT(classtuple))->reloftype = typeid;
    CatalogTupleUpdate(relationRelation, &classtuple->t_self, classtuple);
 
    InvokeObjectPostAlterHook(RelationRelationId, relid, 0);
 
    heap_freetuple(classtuple);
    table_close(relationRelation, RowExclusiveLock);
 
    ReleaseSysCache(typetuple);
 
    return typeobj;
}

References AccessShareLock, BTEqualStrategyNumber, CatalogTupleUpdate(), check_of_type(), DEPENDENCY_NORMAL, drop_parent_dependency(), elog, ereport, errcode(), errmsg, ERROR, fb(), Form_pg_type, GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InvokeObjectPostAlterHook, lookup_rowtype_tupdesc(), NAMEDATALEN, NameStr, ObjectIdGetDatum(), RelationData::rd_rel, recordDependencyOn(), RelationGetDescr, RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), ReleaseTupleDesc, RowExclusiveLock, ScanKeyInit(), SearchSysCacheCopy1, systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), table_open(), TupleDescAttr(), and typenameType().

Referenced by ATExecCmd().

◆ ATExecAddStatistics()

static ObjectAddress ATExecAddStatistics	(	AlteredTableInfo *	tab,
		Relation	rel,
		CreateStatsStmt *	stmt,
		bool	is_rebuild,
		LOCKMODE	lockmode
	)

static

Definition at line 9819 of file tablecmds.c.

{
    ObjectAddress address;
 
    Assert(IsA(stmt, CreateStatsStmt));
 
    /* The CreateStatsStmt has already been through transformStatsStmt */
    Assert(stmt->transformed);
 
    address = CreateStatistics(stmt, !is_rebuild);
 
    return address;
}

References Assert, CreateStatistics(), fb(), IsA, and stmt.

Referenced by ATExecCmd().

◆ ATExecAlterCheckConstrEnforceability()

static bool ATExecAlterCheckConstrEnforceability	(	List **	wqueue,
		ATAlterConstraint *	cmdcon,
		Relation	conrel,
		HeapTuple	contuple,
		bool	recurse,
		bool	recursing,
		List *	changing_conids,
		LOCKMODE	lockmode
	)

static

Definition at line 12717 of file tablecmds.c.

{
    Form_pg_constraint currcon;
    Relation    rel;
    bool        changed = false;
    List       *children = NIL;
    bool        target_enforced = cmdcon->is_enforced;
    Oid         enforced_parentoid = InvalidOid;
 
    /* Since this function recurses, it could be driven to stack overflow */
    check_stack_depth();
 
    Assert(cmdcon->alterEnforceability);
 
    currcon = (Form_pg_constraint) GETSTRUCT(contuple);
 
    Assert(currcon->contype == CONSTRAINT_CHECK);
 
    /*
     * Parent relation already locked by caller, children will be locked by
     * find_all_inheritors. So NoLock is fine here.
     */
    rel = table_open(currcon->conrelid, NoLock);
 
    /*
     * When setting a constraint to NOT ENFORCED, check whether any matching
     * parent constraint remains ENFORCED and is not part of this ALTER.
     *
     * For a direct ALTER of an inherited constraint, reject the command,
     * because the child cannot be weakened while its parent remains enforced.
     *
     * During recursion, another parent outside this ALTER may still enforce
     * the same constraint in a regular inheritance hierarchy. In that case,
     * keep the child constraint ENFORCED so that its merged enforceability
     * still reflects the remaining enforced parent. Partitions do not need
     * this recursive parent check because a partition can have only one
     * direct parent.
     */
    if (!cmdcon->is_enforced &&
        (!recursing || !rel->rd_rel->relispartition) &&
        ATCheckCheckConstrHasEnforcedParent(conrel, rel, contuple,
                                            changing_conids,
                                            &enforced_parentoid))
    {
        if (!recursing)
            ereport(ERROR,
                    errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                    errmsg("cannot mark inherited constraint \"%s\" as %s",
                           NameStr(currcon->conname),
                           "NOT ENFORCED"),
                    errdetail("The matching constraint on parent table \"%s\" is %s.",
                              get_rel_name(enforced_parentoid), "ENFORCED"));
 
        target_enforced = true;
    }
 
    /*
     * Update to the merged enforceability if needed. This may differ from the
     * requested enforceability when another matching parent constraint
     * remains enforced.
     */
    if (currcon->conenforced != target_enforced)
    {
        ATAlterConstraint updatecon = *cmdcon;
 
        updatecon.is_enforced = target_enforced;
        AlterConstrUpdateConstraintEntry(&updatecon, conrel, contuple);
        changed = true;
    }
 
    /*
     * Note that we must recurse even when trying to change a check constraint
     * to not enforced if it is already not enforced, in case descendant
     * constraints might be enforced and need to be changed to not enforced,
     * unless they still inherit an enforced constraint from another parent.
     * Conversely, we should do nothing if a constraint is being set to
     * enforced and is already enforced, as descendant constraints cannot be
     * different in that case.
     */
    if (!cmdcon->is_enforced || changed)
    {
        /*
         * If we're recursing, the parent has already done this, so skip it.
         * Also, if the constraint is a NO INHERIT constraint, we shouldn't
         * try to look for it in the children.
         */
        if (!recursing && !currcon->connoinherit)
        {
            Assert(changing_conids == NIL);
 
            children = find_all_inheritors(RelationGetRelid(rel),
                                           lockmode, NULL);
 
            /*
             * When setting NOT ENFORCED, build the set of equivalent CHECK
             * constraints that this command will attempt to change before
             * visiting descendants. The root itself has already been checked
             * above.
             */
            if (!cmdcon->is_enforced)
                changing_conids = list_make1_oid(currcon->oid);
 
            foreach_oid(childoid, children)
            {
                if (childoid == RelationGetRelid(rel))
                    continue;
 
                /*
                 * If we are told not to recurse, there had better not be any
                 * child tables, because we can't change constraint
                 * enforceability on the parent unless we have changed
                 * enforceability for all child.
                 */
                if (!recurse)
                    ereport(ERROR,
                            errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                            errmsg("constraint must be altered on child tables too"),
                            errhint("Do not specify the ONLY keyword."));
 
                /*
                 * It is sufficient to look up the constraint by name here.
                 * Supported DDL ensures that inheritable CHECK constraints
                 * with the same name have equivalent definitions when they
                 * are propagated to children or when inheritance is
                 * established.  All descendants returned by
                 * find_all_inheritors must have this constraint: inherited
                 * CHECK constraints propagate to all children at
                 * inheritance-link creation time and cannot be dropped
                 * independently on child tables.
                 */
                if (!cmdcon->is_enforced)
                    changing_conids =
                        list_append_unique_oid(changing_conids,
                                               get_relation_constraint_oid(childoid,
                                                                           cmdcon->conname,
                                                                           false));
            }
        }
 
        foreach_oid(childoid, children)
        {
            if (childoid == RelationGetRelid(rel))
                continue;
 
            AlterCheckConstrEnforceabilityRecurse(wqueue, cmdcon, conrel,
                                                  childoid, false, true,
                                                  changing_conids,
                                                  lockmode);
        }
    }
 
    /*
     * Tell Phase 3 to check that the constraint is satisfied by existing
     * rows. We only need do this when altering the constraint from NOT
     * ENFORCED to ENFORCED.
     */
    if (rel->rd_rel->relkind == RELKIND_RELATION &&
        !currcon->conenforced &&
        target_enforced)
    {
        AlteredTableInfo *tab;
        NewConstraint *newcon;
        Datum       val;
        char       *conbin;
 
        newcon = palloc0_object(NewConstraint);
        newcon->name = pstrdup(NameStr(currcon->conname));
        newcon->contype = CONSTR_CHECK;
 
        val = SysCacheGetAttrNotNull(CONSTROID, contuple,
                                     Anum_pg_constraint_conbin);
        conbin = TextDatumGetCString(val);
        newcon->qual = expand_generated_columns_in_expr(stringToNode(conbin), rel, 1);
 
        /* Find or create work queue entry for this table */
        tab = ATGetQueueEntry(wqueue, rel);
        tab->constraints = lappend(tab->constraints, newcon);
    }
 
    table_close(rel, NoLock);
 
    return changed;
}

References AlterCheckConstrEnforceabilityRecurse(), AlterConstrUpdateConstraintEntry(), Assert, ATCheckCheckConstrHasEnforcedParent(), ATGetQueueEntry(), check_stack_depth(), CONSTR_CHECK, AlteredTableInfo::constraints, ereport, errcode(), errdetail(), errhint(), errmsg, ERROR, expand_generated_columns_in_expr(), fb(), find_all_inheritors(), foreach_oid, Form_pg_constraint, get_rel_name(), get_relation_constraint_oid(), GETSTRUCT(), InvalidOid, ATAlterConstraint::is_enforced, lappend(), list_append_unique_oid(), list_make1_oid, NameStr, NIL, NoLock, palloc0_object, pstrdup(), RelationData::rd_rel, RelationGetRelid, stringToNode(), SysCacheGetAttrNotNull(), table_close(), table_open(), TextDatumGetCString, and val.

Referenced by AlterCheckConstrEnforceabilityRecurse(), and ATExecAlterConstraintInternal().

◆ ATExecAlterColumnGenericOptions()

static ObjectAddress ATExecAlterColumnGenericOptions	(	Relation	rel,
		const char *	colName,
		List *	options,
		LOCKMODE	lockmode
	)

static

Definition at line 16660 of file tablecmds.c.

{
    Relation    ftrel;
    Relation    attrel;
    ForeignServer *server;
    ForeignDataWrapper *fdw;
    HeapTuple   tuple;
    HeapTuple   newtuple;
    bool        isnull;
    Datum       repl_val[Natts_pg_attribute];
    bool        repl_null[Natts_pg_attribute];
    bool        repl_repl[Natts_pg_attribute];
    Datum       datum;
    Form_pg_foreign_table fttableform;
    Form_pg_attribute atttableform;
    AttrNumber  attnum;
    ObjectAddress address;
 
    if (options == NIL)
        return InvalidObjectAddress;
 
    /* First, determine FDW validator associated to the foreign table. */
    ftrel = table_open(ForeignTableRelationId, AccessShareLock);
    tuple = SearchSysCache1(FOREIGNTABLEREL, ObjectIdGetDatum(rel->rd_id));
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_OBJECT),
                 errmsg("foreign table \"%s\" does not exist",
                        RelationGetRelationName(rel))));
    fttableform = (Form_pg_foreign_table) GETSTRUCT(tuple);
    server = GetForeignServer(fttableform->ftserver);
    fdw = GetForeignDataWrapper(server->fdwid);
 
    table_close(ftrel, AccessShareLock);
    ReleaseSysCache(tuple);
 
    attrel = table_open(AttributeRelationId, RowExclusiveLock);
    tuple = SearchSysCacheAttName(RelationGetRelid(rel), colName);
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel))));
 
    /* Prevent them from altering a system attribute */
    atttableform = (Form_pg_attribute) GETSTRUCT(tuple);
    attnum = atttableform->attnum;
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"", colName)));
 
 
    /* Initialize buffers for new tuple values */
    memset(repl_val, 0, sizeof(repl_val));
    memset(repl_null, false, sizeof(repl_null));
    memset(repl_repl, false, sizeof(repl_repl));
 
    /* Extract the current options */
    datum = SysCacheGetAttr(ATTNAME,
                            tuple,
                            Anum_pg_attribute_attfdwoptions,
                            &isnull);
    if (isnull)
        datum = PointerGetDatum(NULL);
 
    /* Transform the options */
    datum = transformGenericOptions(AttributeRelationId,
                                    datum,
                                    options,
                                    fdw->fdwvalidator);
 
    if (DatumGetPointer(datum) != NULL)
        repl_val[Anum_pg_attribute_attfdwoptions - 1] = datum;
    else
        repl_null[Anum_pg_attribute_attfdwoptions - 1] = true;
 
    repl_repl[Anum_pg_attribute_attfdwoptions - 1] = true;
 
    /* Everything looks good - update the tuple */
 
    newtuple = heap_modify_tuple(tuple, RelationGetDescr(attrel),
                                 repl_val, repl_null, repl_repl);
 
    CatalogTupleUpdate(attrel, &newtuple->t_self, newtuple);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel),
                              atttableform->attnum);
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
 
    ReleaseSysCache(tuple);
 
    table_close(attrel, RowExclusiveLock);
 
    heap_freetuple(newtuple);
 
    return address;
}

References AccessShareLock, attnum, CatalogTupleUpdate(), DatumGetPointer(), ereport, errcode(), errmsg, ERROR, fb(), ForeignServer::fdwid, Form_pg_foreign_table, GetForeignDataWrapper(), GetForeignServer(), GETSTRUCT(), heap_freetuple(), heap_modify_tuple(), HeapTupleIsValid, InvalidObjectAddress, InvokeObjectPostAlterHook, NIL, ObjectAddressSubSet, ObjectIdGetDatum(), PointerGetDatum, RelationData::rd_id, RelationGetDescr, RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), RowExclusiveLock, SearchSysCache1(), SearchSysCacheAttName(), SysCacheGetAttr(), HeapTupleData::t_self, table_close(), table_open(), and transformGenericOptions().

Referenced by ATExecCmd().

◆ ATExecAlterColumnType()

static ObjectAddress ATExecAlterColumnType	(	AlteredTableInfo *	tab,
		Relation	rel,
		AlterTableCmd *	cmd,
		LOCKMODE	lockmode
	)

static

Definition at line 15263 of file tablecmds.c.

{
    char       *colName = cmd->name;
    ColumnDef  *def = (ColumnDef *) cmd->def;
    TypeName   *typeName = def->typeName;
    HeapTuple   heapTup;
    Form_pg_attribute attTup,
                attOldTup;
    AttrNumber  attnum;
    HeapTuple   typeTuple;
    Form_pg_type tform;
    Oid         targettype;
    int32       targettypmod;
    Oid         targetcollid;
    Node       *defaultexpr;
    Relation    attrelation;
    Relation    depRel;
    ScanKeyData key[3];
    SysScanDesc scan;
    HeapTuple   depTup;
    ObjectAddress address;
 
    /*
     * Clear all the missing values if we're rewriting the table, since this
     * renders them pointless.
     */
    if (tab->rewrite)
    {
        Relation    newrel;
 
        newrel = table_open(RelationGetRelid(rel), NoLock);
        RelationClearMissing(newrel);
        relation_close(newrel, NoLock);
        /* make sure we don't conflict with later attribute modifications */
        CommandCounterIncrement();
    }
 
    attrelation = table_open(AttributeRelationId, RowExclusiveLock);
 
    /* Look up the target column */
    heapTup = SearchSysCacheCopyAttName(RelationGetRelid(rel), colName);
    if (!HeapTupleIsValid(heapTup)) /* shouldn't happen */
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel))));
    attTup = (Form_pg_attribute) GETSTRUCT(heapTup);
    attnum = attTup->attnum;
    attOldTup = TupleDescAttr(tab->oldDesc, attnum - 1);
 
    /* Check for multiple ALTER TYPE on same column --- can't cope */
    if (attTup->atttypid != attOldTup->atttypid ||
        attTup->atttypmod != attOldTup->atttypmod)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter type of column \"%s\" twice",
                        colName)));
 
    /* Look up the target type (should not fail, since prep found it) */
    typeTuple = typenameType(NULL, typeName, &targettypmod);
    tform = (Form_pg_type) GETSTRUCT(typeTuple);
    targettype = tform->oid;
    /* And the collation */
    targetcollid = GetColumnDefCollation(NULL, def, targettype);
 
    /*
     * If there is a default expression for the column, get it and ensure we
     * can coerce it to the new datatype.  (We must do this before changing
     * the column type, because build_column_default itself will try to
     * coerce, and will not issue the error message we want if it fails.)
     *
     * We remove any implicit coercion steps at the top level of the old
     * default expression; this has been agreed to satisfy the principle of
     * least surprise.  (The conversion to the new column type should act like
     * it started from what the user sees as the stored expression, and the
     * implicit coercions aren't going to be shown.)
     */
    if (attTup->atthasdef)
    {
        defaultexpr = build_column_default(rel, attnum);
        Assert(defaultexpr);
        defaultexpr = strip_implicit_coercions(defaultexpr);
        defaultexpr = coerce_to_target_type(NULL,   /* no UNKNOWN params */
                                            defaultexpr, exprType(defaultexpr),
                                            targettype, targettypmod,
                                            COERCION_ASSIGNMENT,
                                            COERCE_IMPLICIT_CAST,
                                            -1);
        if (defaultexpr == NULL)
        {
            if (attTup->attgenerated)
                ereport(ERROR,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("generation expression for column \"%s\" cannot be cast automatically to type %s",
                                colName, format_type_be(targettype))));
            else
                ereport(ERROR,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("default for column \"%s\" cannot be cast automatically to type %s",
                                colName, format_type_be(targettype))));
        }
    }
    else
        defaultexpr = NULL;
 
    /*
     * Find everything that depends on the column (constraints, indexes, etc),
     * and record enough information to let us recreate the objects.
     *
     * The actual recreation does not happen here, but only after we have
     * performed all the individual ALTER TYPE operations.  We have to save
     * the info before executing ALTER TYPE, though, else the deparser will
     * get confused.
     */
    RememberAllDependentForRebuilding(tab, AT_AlterColumnType, rel, attnum, colName);
 
    /*
     * Now scan for dependencies of this column on other things.  The only
     * things we should find are the dependency on the column datatype and
     * possibly a collation dependency.  Those can be removed.
     */
    depRel = table_open(DependRelationId, RowExclusiveLock);
 
    ScanKeyInit(&key[0],
                Anum_pg_depend_classid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationRelationId));
    ScanKeyInit(&key[1],
                Anum_pg_depend_objid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationGetRelid(rel)));
    ScanKeyInit(&key[2],
                Anum_pg_depend_objsubid,
                BTEqualStrategyNumber, F_INT4EQ,
                Int32GetDatum((int32) attnum));
 
    scan = systable_beginscan(depRel, DependDependerIndexId, true,
                              NULL, 3, key);
 
    while (HeapTupleIsValid(depTup = systable_getnext(scan)))
    {
        Form_pg_depend foundDep = (Form_pg_depend) GETSTRUCT(depTup);
        ObjectAddress foundObject;
 
        foundObject.classId = foundDep->refclassid;
        foundObject.objectId = foundDep->refobjid;
        foundObject.objectSubId = foundDep->refobjsubid;
 
        if (foundDep->deptype != DEPENDENCY_NORMAL)
            elog(ERROR, "found unexpected dependency type '%c'",
                 foundDep->deptype);
        if (!(foundDep->refclassid == TypeRelationId &&
              foundDep->refobjid == attTup->atttypid) &&
            !(foundDep->refclassid == CollationRelationId &&
              foundDep->refobjid == attTup->attcollation))
            elog(ERROR, "found unexpected dependency for column: %s",
                 getObjectDescription(&foundObject, false));
 
        CatalogTupleDelete(depRel, &depTup->t_self);
    }
 
    systable_endscan(scan);
 
    table_close(depRel, RowExclusiveLock);
 
    /*
     * Here we go --- change the recorded column type and collation.  (Note
     * heapTup is a copy of the syscache entry, so okay to scribble on.) First
     * fix up the missing value if any.
     */
    if (attTup->atthasmissing)
    {
        Datum       missingval;
        bool        missingNull;
 
        /* if rewrite is true the missing value should already be cleared */
        Assert(tab->rewrite == 0);
 
        /* Get the missing value datum */
        missingval = heap_getattr(heapTup,
                                  Anum_pg_attribute_attmissingval,
                                  attrelation->rd_att,
                                  &missingNull);
 
        /* if it's a null array there is nothing to do */
 
        if (!missingNull)
        {
            /*
             * Get the datum out of the array and repack it in a new array
             * built with the new type data. We assume that since the table
             * doesn't need rewriting, the actual Datum doesn't need to be
             * changed, only the array metadata.
             */
 
            int         one = 1;
            bool        isNull;
            Datum       valuesAtt[Natts_pg_attribute] = {0};
            bool        nullsAtt[Natts_pg_attribute] = {0};
            bool        replacesAtt[Natts_pg_attribute] = {0};
            HeapTuple   newTup;
 
            missingval = array_get_element(missingval,
                                           1,
                                           &one,
                                           0,
                                           attTup->attlen,
                                           attTup->attbyval,
                                           attTup->attalign,
                                           &isNull);
            missingval = PointerGetDatum(construct_array(&missingval,
                                                         1,
                                                         targettype,
                                                         tform->typlen,
                                                         tform->typbyval,
                                                         tform->typalign));
 
            valuesAtt[Anum_pg_attribute_attmissingval - 1] = missingval;
            replacesAtt[Anum_pg_attribute_attmissingval - 1] = true;
            nullsAtt[Anum_pg_attribute_attmissingval - 1] = false;
 
            newTup = heap_modify_tuple(heapTup, RelationGetDescr(attrelation),
                                       valuesAtt, nullsAtt, replacesAtt);
            heap_freetuple(heapTup);
            heapTup = newTup;
            attTup = (Form_pg_attribute) GETSTRUCT(heapTup);
        }
    }
 
    attTup->atttypid = targettype;
    attTup->atttypmod = targettypmod;
    attTup->attcollation = targetcollid;
    if (list_length(typeName->arrayBounds) > PG_INT16_MAX)
        ereport(ERROR,
                errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
                errmsg("too many array dimensions"));
    attTup->attndims = list_length(typeName->arrayBounds);
    attTup->attlen = tform->typlen;
    attTup->attbyval = tform->typbyval;
    attTup->attalign = tform->typalign;
    attTup->attstorage = tform->typstorage;
    attTup->attcompression = InvalidCompressionMethod;
 
    ReleaseSysCache(typeTuple);
 
    CatalogTupleUpdate(attrelation, &heapTup->t_self, heapTup);
 
    table_close(attrelation, RowExclusiveLock);
 
    /* Install dependencies on new datatype and collation */
    add_column_datatype_dependency(RelationGetRelid(rel), attnum, targettype);
    add_column_collation_dependency(RelationGetRelid(rel), attnum, targetcollid);
 
    /*
     * Drop any pg_statistic entry for the column, since it's now wrong type
     */
    RemoveStatistics(RelationGetRelid(rel), attnum);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel), attnum);
 
    /*
     * Update the default, if present, by brute force --- remove and re-add
     * the default.  Probably unsafe to take shortcuts, since the new version
     * may well have additional dependencies.  (It's okay to do this now,
     * rather than after other ALTER TYPE commands, since the default won't
     * depend on other column types.)
     */
    if (defaultexpr)
    {
        /*
         * If it's a GENERATED default, drop its dependency records, in
         * particular its INTERNAL dependency on the column, which would
         * otherwise cause dependency.c to refuse to perform the deletion.
         */
        if (attTup->attgenerated)
        {
            Oid         attrdefoid = GetAttrDefaultOid(RelationGetRelid(rel), attnum);
 
            if (!OidIsValid(attrdefoid))
                elog(ERROR, "could not find attrdef tuple for relation %u attnum %d",
                     RelationGetRelid(rel), attnum);
            (void) deleteDependencyRecordsFor(AttrDefaultRelationId, attrdefoid, false);
        }
 
        /*
         * Make updates-so-far visible, particularly the new pg_attribute row
         * which will be updated again.
         */
        CommandCounterIncrement();
 
        /*
         * We use RESTRICT here for safety, but at present we do not expect
         * anything to depend on the default.
         */
        RemoveAttrDefault(RelationGetRelid(rel), attnum, DROP_RESTRICT, true,
                          true);
 
        (void) StoreAttrDefault(rel, attnum, defaultexpr, true);
    }
 
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
 
    /* Cleanup */
    heap_freetuple(heapTup);
 
    return address;
}

References add_column_collation_dependency(), add_column_datatype_dependency(), array_get_element(), TypeName::arrayBounds, Assert, AT_AlterColumnType, attnum, BTEqualStrategyNumber, build_column_default(), CatalogTupleDelete(), CatalogTupleUpdate(), ObjectAddress::classId, COERCE_IMPLICIT_CAST, coerce_to_target_type(), COERCION_ASSIGNMENT, CommandCounterIncrement(), construct_array(), AlterTableCmd::def, deleteDependencyRecordsFor(), DEPENDENCY_NORMAL, DROP_RESTRICT, elog, ereport, errcode(), errmsg, ERROR, exprType(), fb(), Form_pg_depend, Form_pg_type, format_type_be(), GetAttrDefaultOid(), GetColumnDefCollation(), getObjectDescription(), GETSTRUCT(), heap_freetuple(), heap_getattr(), heap_modify_tuple(), HeapTupleIsValid, Int32GetDatum(), InvalidCompressionMethod, InvokeObjectPostAlterHook, list_length(), AlterTableCmd::name, NoLock, ObjectAddressSubSet, ObjectIdGetDatum(), OidIsValid, AlteredTableInfo::oldDesc, PG_INT16_MAX, PointerGetDatum, relation_close(), RelationClearMissing(), RelationGetDescr, RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), RememberAllDependentForRebuilding(), RemoveAttrDefault(), RemoveStatistics(), AlteredTableInfo::rewrite, RowExclusiveLock, ScanKeyInit(), SearchSysCacheCopyAttName(), StoreAttrDefault(), strip_implicit_coercions(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), table_open(), TupleDescAttr(), ColumnDef::typeName, and typenameType().

Referenced by ATExecCmd().

◆ ATExecAlterConstraint()

static ObjectAddress ATExecAlterConstraint	(	List **	wqueue,
		Relation	rel,
		ATAlterConstraint *	cmdcon,
		bool	recurse,
		LOCKMODE	lockmode
	)

static

Definition at line 12346 of file tablecmds.c.

{
    Relation    conrel;
    Relation    tgrel;
    SysScanDesc scan;
    ScanKeyData skey[3];
    HeapTuple   contuple;
    Form_pg_constraint currcon;
    ObjectAddress address;
 
    /*
     * Disallow altering ONLY a partitioned table, as it would make no sense.
     * This is okay for legacy inheritance.
     */
    if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE && !recurse)
        ereport(ERROR,
                errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                errmsg("constraint must be altered in child tables too"),
                errhint("Do not specify the ONLY keyword."));
 
 
    conrel = table_open(ConstraintRelationId, RowExclusiveLock);
    tgrel = table_open(TriggerRelationId, RowExclusiveLock);
 
    /*
     * Find and check the target constraint
     */
    ScanKeyInit(&skey[0],
                Anum_pg_constraint_conrelid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationGetRelid(rel)));
    ScanKeyInit(&skey[1],
                Anum_pg_constraint_contypid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(InvalidOid));
    ScanKeyInit(&skey[2],
                Anum_pg_constraint_conname,
                BTEqualStrategyNumber, F_NAMEEQ,
                CStringGetDatum(cmdcon->conname));
    scan = systable_beginscan(conrel, ConstraintRelidTypidNameIndexId,
                              true, NULL, 3, skey);
 
    /* There can be at most one matching row */
    if (!HeapTupleIsValid(contuple = systable_getnext(scan)))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_OBJECT),
                 errmsg("constraint \"%s\" of relation \"%s\" does not exist",
                        cmdcon->conname, RelationGetRelationName(rel))));
 
    currcon = (Form_pg_constraint) GETSTRUCT(contuple);
    if (cmdcon->alterDeferrability && currcon->contype != CONSTRAINT_FOREIGN)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("constraint \"%s\" of relation \"%s\" is not a foreign key constraint",
                        cmdcon->conname, RelationGetRelationName(rel))));
    if (cmdcon->alterEnforceability &&
        (currcon->contype != CONSTRAINT_FOREIGN && currcon->contype != CONSTRAINT_CHECK))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot alter enforceability of constraint \"%s\" of relation \"%s\"",
                        cmdcon->conname, RelationGetRelationName(rel)),
                 errhint("Only foreign key and check constraints can change enforceability.")));
    if (cmdcon->alterInheritability &&
        currcon->contype != CONSTRAINT_NOTNULL)
        ereport(ERROR,
                errcode(ERRCODE_WRONG_OBJECT_TYPE),
                errmsg("constraint \"%s\" of relation \"%s\" is not a not-null constraint",
                       cmdcon->conname, RelationGetRelationName(rel)));
    if (cmdcon->alterInheritability &&
        cmdcon->noinherit && rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
        ereport(ERROR,
                errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                errmsg("not-null constraint \"%s\" on partitioned table \"%s\" cannot be NO INHERIT",
                       cmdcon->conname, RelationGetRelationName(rel)));
 
    /* Refuse to modify inheritability of inherited constraints */
    if (cmdcon->alterInheritability &&
        cmdcon->noinherit && currcon->coninhcount > 0)
        ereport(ERROR,
                errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                errmsg("cannot alter inherited constraint \"%s\" on relation \"%s\"",
                       NameStr(currcon->conname),
                       RelationGetRelationName(rel)));
 
    /*
     * If it's not the topmost constraint, raise an error.
     *
     * Altering a non-topmost constraint leaves some triggers untouched, since
     * they are not directly connected to this constraint; also, pg_dump would
     * ignore the deferrability status of the individual constraint, since it
     * only dumps topmost constraints.  Avoid these problems by refusing this
     * operation and telling the user to alter the parent constraint instead.
     */
    if (OidIsValid(currcon->conparentid))
    {
        HeapTuple   tp;
        Oid         parent = currcon->conparentid;
        char       *ancestorname = NULL;
        char       *ancestortable = NULL;
 
        /* Loop to find the topmost constraint */
        while (HeapTupleIsValid(tp = SearchSysCache1(CONSTROID, ObjectIdGetDatum(parent))))
        {
            Form_pg_constraint contup = (Form_pg_constraint) GETSTRUCT(tp);
 
            /* If no parent, this is the constraint we want */
            if (!OidIsValid(contup->conparentid))
            {
                ancestorname = pstrdup(NameStr(contup->conname));
                ancestortable = get_rel_name(contup->conrelid);
                ReleaseSysCache(tp);
                break;
            }
 
            parent = contup->conparentid;
            ReleaseSysCache(tp);
        }
 
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                 errmsg("cannot alter constraint \"%s\" on relation \"%s\"",
                        cmdcon->conname, RelationGetRelationName(rel)),
                 ancestorname && ancestortable ?
                 errdetail("Constraint \"%s\" is derived from constraint \"%s\" of relation \"%s\".",
                           cmdcon->conname, ancestorname, ancestortable) : 0,
                 errhint("You may alter the constraint it derives from instead.")));
    }
 
    address = InvalidObjectAddress;
 
    /*
     * Do the actual catalog work, and recurse if necessary.
     */
    if (ATExecAlterConstraintInternal(wqueue, cmdcon, conrel, tgrel, rel,
                                      contuple, recurse, lockmode))
        ObjectAddressSet(address, ConstraintRelationId, currcon->oid);
 
    systable_endscan(scan);
 
    table_close(tgrel, RowExclusiveLock);
    table_close(conrel, RowExclusiveLock);
 
    return address;
}

References ATExecAlterConstraintInternal(), BTEqualStrategyNumber, CStringGetDatum(), ereport, errcode(), errdetail(), errhint(), errmsg, ERROR, fb(), Form_pg_constraint, get_rel_name(), GETSTRUCT(), HeapTupleIsValid, InvalidObjectAddress, InvalidOid, NameStr, ObjectAddressSet, ObjectIdGetDatum(), OidIsValid, pstrdup(), RelationData::rd_rel, RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), RowExclusiveLock, ScanKeyInit(), SearchSysCache1(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by ATExecCmd().

◆ ATExecAlterConstraintInternal()

static bool ATExecAlterConstraintInternal	(	List **	wqueue,
		ATAlterConstraint *	cmdcon,
		Relation	conrel,
		Relation	tgrel,
		Relation	rel,
		HeapTuple	contuple,
		bool	recurse,
		LOCKMODE	lockmode
	)

static

Definition at line 12497 of file tablecmds.c.

{
    Form_pg_constraint currcon;
    bool        changed = false;
    List       *otherrelids = NIL;
 
    currcon = (Form_pg_constraint) GETSTRUCT(contuple);
 
    /*
     * Do the catalog work for the enforceability or deferrability change,
     * recurse if necessary.
     *
     * Note that even if deferrability is requested to be altered along with
     * enforceability, we don't need to explicitly update multiple entries in
     * pg_trigger related to deferrability.
     *
     * Modifying foreign key enforceability involves either creating or
     * dropping the trigger, during which the deferrability setting will be
     * adjusted automatically.
     */
    if (cmdcon->alterEnforceability)
    {
        if (currcon->contype == CONSTRAINT_FOREIGN)
            changed = ATExecAlterFKConstrEnforceability(wqueue, cmdcon, conrel, tgrel,
                                                        currcon->conrelid,
                                                        currcon->confrelid,
                                                        contuple, lockmode,
                                                        InvalidOid, InvalidOid,
                                                        InvalidOid, InvalidOid);
        else if (currcon->contype == CONSTRAINT_CHECK)
            changed = ATExecAlterCheckConstrEnforceability(wqueue, cmdcon, conrel,
                                                           contuple, recurse, false,
                                                           NIL, lockmode);
    }
    else if (cmdcon->alterDeferrability &&
             ATExecAlterConstrDeferrability(wqueue, cmdcon, conrel, tgrel, rel,
                                            contuple, recurse, &otherrelids,
                                            lockmode))
    {
        /*
         * AlterConstrUpdateConstraintEntry already invalidated relcache for
         * the relations having the constraint itself; here we also invalidate
         * for relations that have any triggers that are part of the
         * constraint.
         */
        foreach_oid(relid, otherrelids)
            CacheInvalidateRelcacheByRelid(relid);
 
        changed = true;
    }
 
    /*
     * Do the catalog work for the inheritability change.
     */
    if (cmdcon->alterInheritability &&
        ATExecAlterConstrInheritability(wqueue, cmdcon, conrel, rel, contuple,
                                        lockmode))
        changed = true;
 
    return changed;
}

References ATExecAlterCheckConstrEnforceability(), ATExecAlterConstrDeferrability(), ATExecAlterConstrInheritability(), ATExecAlterFKConstrEnforceability(), CacheInvalidateRelcacheByRelid(), fb(), foreach_oid, Form_pg_constraint, GETSTRUCT(), InvalidOid, and NIL.

Referenced by ATExecAlterConstraint().

◆ ATExecAlterConstrDeferrability()

static bool ATExecAlterConstrDeferrability	(	List **	wqueue,
		ATAlterConstraint *	cmdcon,
		Relation	conrel,
		Relation	tgrel,
		Relation	rel,
		HeapTuple	contuple,
		bool	recurse,
		List **	otherrelids,
		LOCKMODE	lockmode
	)

static

Definition at line 13092 of file tablecmds.c.

{
    Form_pg_constraint currcon;
    Oid         refrelid;
    bool        changed = false;
 
    /* since this function recurses, it could be driven to stack overflow */
    check_stack_depth();
 
    Assert(cmdcon->alterDeferrability);
 
    currcon = (Form_pg_constraint) GETSTRUCT(contuple);
    refrelid = currcon->confrelid;
 
    /* Should be foreign key constraint */
    Assert(currcon->contype == CONSTRAINT_FOREIGN);
 
    /*
     * If called to modify a constraint that's already in the desired state,
     * silently do nothing.
     */
    if (currcon->condeferrable != cmdcon->deferrable ||
        currcon->condeferred != cmdcon->initdeferred)
    {
        AlterConstrUpdateConstraintEntry(cmdcon, conrel, contuple);
        changed = true;
 
        /*
         * Now we need to update the multiple entries in pg_trigger that
         * implement the constraint.
         */
        AlterConstrTriggerDeferrability(currcon->oid, tgrel, rel,
                                        cmdcon->deferrable,
                                        cmdcon->initdeferred, otherrelids);
    }
 
    /*
     * If the table at either end of the constraint is partitioned, we need to
     * handle every constraint that is a child of this one.
     */
    if (recurse && changed &&
        (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE ||
         get_rel_relkind(refrelid) == RELKIND_PARTITIONED_TABLE))
        AlterConstrDeferrabilityRecurse(wqueue, cmdcon, conrel, tgrel, rel,
                                        contuple, recurse, otherrelids,
                                        lockmode);
 
    return changed;
}

References AlterConstrDeferrabilityRecurse(), AlterConstrTriggerDeferrability(), AlterConstrUpdateConstraintEntry(), Assert, check_stack_depth(), fb(), Form_pg_constraint, get_rel_relkind(), GETSTRUCT(), and RelationData::rd_rel.

Referenced by AlterConstrDeferrabilityRecurse(), and ATExecAlterConstraintInternal().

◆ ATExecAlterConstrInheritability()

static bool ATExecAlterConstrInheritability	(	List **	wqueue,
		ATAlterConstraint *	cmdcon,
		Relation	conrel,
		Relation	rel,
		HeapTuple	contuple,
		LOCKMODE	lockmode
	)

static

Definition at line 13149 of file tablecmds.c.

{
    Form_pg_constraint currcon;
    AttrNumber  colNum;
    char       *colName;
    List       *children;
 
    Assert(cmdcon->alterInheritability);
 
    currcon = (Form_pg_constraint) GETSTRUCT(contuple);
 
    /* The current implementation only works for NOT NULL constraints */
    Assert(currcon->contype == CONSTRAINT_NOTNULL);
 
    /*
     * If called to modify a constraint that's already in the desired state,
     * silently do nothing.
     */
    if (cmdcon->noinherit == currcon->connoinherit)
        return false;
 
    AlterConstrUpdateConstraintEntry(cmdcon, conrel, contuple);
    CommandCounterIncrement();
 
    /* Fetch the column number and name */
    colNum = extractNotNullColumn(contuple);
    colName = get_attname(currcon->conrelid, colNum, false);
 
    /*
     * Propagate the change to children.  For this subcommand type we don't
     * recursively affect children, just the immediate level.
     */
    children = find_inheritance_children(RelationGetRelid(rel),
                                         lockmode);
    foreach_oid(childoid, children)
    {
        ObjectAddress addr;
 
        if (cmdcon->noinherit)
        {
            HeapTuple   childtup;
            Form_pg_constraint childcon;
 
            childtup = findNotNullConstraint(childoid, colName);
            if (!childtup)
                elog(ERROR, "cache lookup failed for not-null constraint on column \"%s\" of relation %u",
                     colName, childoid);
            childcon = (Form_pg_constraint) GETSTRUCT(childtup);
            Assert(childcon->coninhcount > 0);
            childcon->coninhcount--;
            childcon->conislocal = true;
            CatalogTupleUpdate(conrel, &childtup->t_self, childtup);
            heap_freetuple(childtup);
        }
        else
        {
            Relation    childrel = table_open(childoid, NoLock);
 
            addr = ATExecSetNotNull(wqueue, childrel, NameStr(currcon->conname),
                                    colName, true, true, lockmode);
            if (OidIsValid(addr.objectId))
                CommandCounterIncrement();
            table_close(childrel, NoLock);
        }
    }
 
    return true;
}

References AlterConstrUpdateConstraintEntry(), Assert, ATExecSetNotNull(), CatalogTupleUpdate(), CommandCounterIncrement(), elog, ERROR, extractNotNullColumn(), fb(), find_inheritance_children(), findNotNullConstraint(), foreach_oid, Form_pg_constraint, get_attname(), GETSTRUCT(), heap_freetuple(), NameStr, NoLock, ObjectAddress::objectId, OidIsValid, RelationGetRelid, table_close(), and table_open().

Referenced by ATExecAlterConstraintInternal().

◆ ATExecAlterFKConstrEnforceability()

static bool ATExecAlterFKConstrEnforceability	(	List **	wqueue,
		ATAlterConstraint *	cmdcon,
		Relation	conrel,
		Relation	tgrel,
		Oid	fkrelid,
		Oid	pkrelid,
		HeapTuple	contuple,
		LOCKMODE	lockmode,
		Oid	ReferencedParentDelTrigger,
		Oid	ReferencedParentUpdTrigger,
		Oid	ReferencingParentInsTrigger,
		Oid	ReferencingParentUpdTrigger
	)

static

Definition at line 12575 of file tablecmds.c.

{
    Form_pg_constraint currcon;
    Oid         conoid;
    Relation    rel;
    bool        changed = false;
 
    /* Since this function recurses, it could be driven to stack overflow */
    check_stack_depth();
 
    Assert(cmdcon->alterEnforceability);
 
    currcon = (Form_pg_constraint) GETSTRUCT(contuple);
    conoid = currcon->oid;
 
    /* Should be foreign key constraint */
    Assert(currcon->contype == CONSTRAINT_FOREIGN);
 
    rel = table_open(currcon->conrelid, lockmode);
 
    if (currcon->conenforced != cmdcon->is_enforced)
    {
        AlterConstrUpdateConstraintEntry(cmdcon, conrel, contuple);
        changed = true;
    }
 
    /* Drop triggers */
    if (!cmdcon->is_enforced)
    {
        /*
         * When setting a constraint to NOT ENFORCED, the constraint triggers
         * need to be dropped. Therefore, we must process the child relations
         * first, followed by the parent, to account for dependencies.
         */
        if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE ||
            get_rel_relkind(currcon->confrelid) == RELKIND_PARTITIONED_TABLE)
            AlterFKConstrEnforceabilityRecurse(wqueue, cmdcon, conrel, tgrel,
                                               fkrelid, pkrelid, contuple,
                                               lockmode, InvalidOid, InvalidOid,
                                               InvalidOid, InvalidOid);
 
        /* Drop all the triggers */
        DropForeignKeyConstraintTriggers(tgrel, conoid, InvalidOid, InvalidOid);
    }
    else if (changed)           /* Create triggers */
    {
        Oid         ReferencedDelTriggerOid = InvalidOid,
                    ReferencedUpdTriggerOid = InvalidOid,
                    ReferencingInsTriggerOid = InvalidOid,
                    ReferencingUpdTriggerOid = InvalidOid;
 
        /* Prepare the minimal information required for trigger creation. */
        Constraint *fkconstraint = makeNode(Constraint);
 
        fkconstraint->conname = pstrdup(NameStr(currcon->conname));
        fkconstraint->fk_matchtype = currcon->confmatchtype;
        fkconstraint->fk_upd_action = currcon->confupdtype;
        fkconstraint->fk_del_action = currcon->confdeltype;
        fkconstraint->deferrable = currcon->condeferrable;
        fkconstraint->initdeferred = currcon->condeferred;
 
        /* Create referenced triggers */
        if (currcon->conrelid == fkrelid)
            createForeignKeyActionTriggers(currcon->conrelid,
                                           currcon->confrelid,
                                           fkconstraint,
                                           conoid,
                                           currcon->conindid,
                                           ReferencedParentDelTrigger,
                                           ReferencedParentUpdTrigger,
                                           &ReferencedDelTriggerOid,
                                           &ReferencedUpdTriggerOid);
 
        /* Create referencing triggers */
        if (currcon->confrelid == pkrelid)
            createForeignKeyCheckTriggers(currcon->conrelid,
                                          pkrelid,
                                          fkconstraint,
                                          conoid,
                                          currcon->conindid,
                                          ReferencingParentInsTrigger,
                                          ReferencingParentUpdTrigger,
                                          &ReferencingInsTriggerOid,
                                          &ReferencingUpdTriggerOid);
 
        /*
         * Tell Phase 3 to check that the constraint is satisfied by existing
         * rows.  Only applies to leaf partitions, and (for constraints that
         * reference a partitioned table) only if this is not one of the
         * pg_constraint rows that exist solely to support action triggers.
         */
        if (rel->rd_rel->relkind == RELKIND_RELATION &&
            currcon->confrelid == pkrelid)
        {
            AlteredTableInfo *tab;
            NewConstraint *newcon;
 
            newcon = palloc0_object(NewConstraint);
            newcon->name = fkconstraint->conname;
            newcon->contype = CONSTR_FOREIGN;
            newcon->refrelid = currcon->confrelid;
            newcon->refindid = currcon->conindid;
            newcon->conid = currcon->oid;
            newcon->qual = (Node *) fkconstraint;
 
            /* Find or create work queue entry for this table */
            tab = ATGetQueueEntry(wqueue, rel);
            tab->constraints = lappend(tab->constraints, newcon);
        }
 
        /*
         * If the table at either end of the constraint is partitioned, we
         * need to recurse and create triggers for each constraint that is a
         * child of this one.
         */
        if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE ||
            get_rel_relkind(currcon->confrelid) == RELKIND_PARTITIONED_TABLE)
            AlterFKConstrEnforceabilityRecurse(wqueue, cmdcon, conrel, tgrel,
                                               fkrelid, pkrelid, contuple,
                                               lockmode,
                                               ReferencedDelTriggerOid,
                                               ReferencedUpdTriggerOid,
                                               ReferencingInsTriggerOid,
                                               ReferencingUpdTriggerOid);
    }
 
    table_close(rel, NoLock);
 
    return changed;
}

References AlterConstrUpdateConstraintEntry(), AlterFKConstrEnforceabilityRecurse(), Assert, ATGetQueueEntry(), check_stack_depth(), CONSTR_FOREIGN, AlteredTableInfo::constraints, createForeignKeyActionTriggers(), createForeignKeyCheckTriggers(), DropForeignKeyConstraintTriggers(), fb(), Form_pg_constraint, get_rel_relkind(), GETSTRUCT(), InvalidOid, lappend(), makeNode, NameStr, NoLock, palloc0_object, pstrdup(), RelationData::rd_rel, table_close(), and table_open().

Referenced by AlterFKConstrEnforceabilityRecurse(), and ATExecAlterConstraintInternal().

◆ ATExecAttachPartition()

static ObjectAddress ATExecAttachPartition	(	List **	wqueue,
		Relation	rel,
		PartitionCmd *	cmd,
		AlterTableUtilityContext *	context
	)

static

Definition at line 21016 of file tablecmds.c.

{
    Relation    attachrel,
                catalog;
    List       *attachrel_children;
    List       *partConstraint;
    SysScanDesc scan;
    ScanKeyData skey;
    AttrNumber  attno;
    int         natts;
    TupleDesc   tupleDesc;
    ObjectAddress address;
    const char *trigger_name;
    Oid         defaultPartOid;
    List       *partBoundConstraint;
    List       *exceptpuboids = NIL;
    ParseState *pstate = make_parsestate(NULL);
 
    pstate->p_sourcetext = context->queryString;
 
    /*
     * We must lock the default partition if one exists, because attaching a
     * new partition will change its partition constraint.
     */
    defaultPartOid =
        get_default_oid_from_partdesc(RelationGetPartitionDesc(rel, true));
    if (OidIsValid(defaultPartOid))
        LockRelationOid(defaultPartOid, AccessExclusiveLock);
 
    attachrel = table_openrv(cmd->name, AccessExclusiveLock);
 
    /*
     * XXX I think it'd be a good idea to grab locks on all tables referenced
     * by FKs at this point also.
     */
 
    /*
     * Must be owner of both parent and source table -- parent was checked by
     * ATSimplePermissions call in ATPrepCmd
     */
    ATSimplePermissions(AT_AttachPartition, attachrel,
                        ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
 
    /* A partition can only have one parent */
    if (attachrel->rd_rel->relispartition)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("\"%s\" is already a partition",
                        RelationGetRelationName(attachrel))));
 
    if (OidIsValid(attachrel->rd_rel->reloftype))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot attach a typed table as partition")));
 
    /*
     * Disallow attaching a partition if the table is referenced in a
     * publication EXCEPT clause. Changing the partition hierarchy could alter
     * the effective publication membership.
     */
    exceptpuboids = GetRelationExcludedPublications(RelationGetRelid(attachrel));
    if (exceptpuboids != NIL)
    {
        bool        first = true;
        StringInfoData pubnames;
 
        initStringInfo(&pubnames);
 
        foreach_oid(pubid, exceptpuboids)
        {
            char       *pubname = get_publication_name(pubid, false);
 
            if (first)
                appendStringInfo(&pubnames, _("\"%s\""), pubname);
            else
                appendStringInfo(&pubnames, _(", \"%s\""), pubname);
            first = false;
        }
 
        ereport(ERROR,
                errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                errmsg_plural("cannot attach table \"%s\" as partition because it is referenced in publication %s EXCEPT clause",
                              "cannot attach table \"%s\" as partition because it is referenced in publications %s EXCEPT clause",
                              list_length(exceptpuboids),
                              RelationGetRelationName(attachrel),
                              pubnames.data),
                errdetail("The publication EXCEPT clause cannot contain tables that are partitions."),
                errhint("Change the publication's EXCEPT clause using ALTER PUBLICATION ... SET ALL TABLES."));
    }
 
    list_free(exceptpuboids);
 
    /*
     * Table being attached should not already be part of inheritance; either
     * as a child table...
     */
    catalog = table_open(InheritsRelationId, AccessShareLock);
    ScanKeyInit(&skey,
                Anum_pg_inherits_inhrelid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationGetRelid(attachrel)));
    scan = systable_beginscan(catalog, InheritsRelidSeqnoIndexId, true,
                              NULL, 1, &skey);
    if (HeapTupleIsValid(systable_getnext(scan)))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot attach inheritance child as partition")));
    systable_endscan(scan);
 
    /* ...or as a parent table (except the case when it is partitioned) */
    ScanKeyInit(&skey,
                Anum_pg_inherits_inhparent,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationGetRelid(attachrel)));
    scan = systable_beginscan(catalog, InheritsParentIndexId, true, NULL,
                              1, &skey);
    if (HeapTupleIsValid(systable_getnext(scan)) &&
        attachrel->rd_rel->relkind == RELKIND_RELATION)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot attach inheritance parent as partition")));
    systable_endscan(scan);
    table_close(catalog, AccessShareLock);
 
    /*
     * Prevent circularity by seeing if rel is a partition of attachrel. (In
     * particular, this disallows making a rel a partition of itself.)
     *
     * We do that by checking if rel is a member of the list of attachrel's
     * partitions provided the latter is partitioned at all.  We want to avoid
     * having to construct this list again, so we request the strongest lock
     * on all partitions.  We need the strongest lock, because we may decide
     * to scan them if we find out that the table being attached (or its leaf
     * partitions) may contain rows that violate the partition constraint. If
     * the table has a constraint that would prevent such rows, which by
     * definition is present in all the partitions, we need not scan the
     * table, nor its partitions.  But we cannot risk a deadlock by taking a
     * weaker lock now and the stronger one only when needed.
     */
    attachrel_children = find_all_inheritors(RelationGetRelid(attachrel),
                                             AccessExclusiveLock, NULL);
    if (list_member_oid(attachrel_children, RelationGetRelid(rel)))
        ereport(ERROR,
                (errcode(ERRCODE_DUPLICATE_TABLE),
                 errmsg("circular inheritance not allowed"),
                 errdetail("\"%s\" is already a child of \"%s\".",
                           RelationGetRelationName(rel),
                           RelationGetRelationName(attachrel))));
 
    /* If the parent is permanent, so must be all of its partitions. */
    if (rel->rd_rel->relpersistence != RELPERSISTENCE_TEMP &&
        attachrel->rd_rel->relpersistence == RELPERSISTENCE_TEMP)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot attach a temporary relation as partition of permanent relation \"%s\"",
                        RelationGetRelationName(rel))));
 
    /* Temp parent cannot have a partition that is itself not a temp */
    if (rel->rd_rel->relpersistence == RELPERSISTENCE_TEMP &&
        attachrel->rd_rel->relpersistence != RELPERSISTENCE_TEMP)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot attach a permanent relation as partition of temporary relation \"%s\"",
                        RelationGetRelationName(rel))));
 
    /* If the parent is temp, it must belong to this session */
    if (RELATION_IS_OTHER_TEMP(rel))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot attach as partition of temporary relation of another session")));
 
    /* Ditto for the partition */
    if (RELATION_IS_OTHER_TEMP(attachrel))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot attach temporary relation of another session as partition")));
 
    /*
     * Check if attachrel has any identity columns or any columns that aren't
     * in the parent.
     */
    tupleDesc = RelationGetDescr(attachrel);
    natts = tupleDesc->natts;
    for (attno = 1; attno <= natts; attno++)
    {
        Form_pg_attribute attribute = TupleDescAttr(tupleDesc, attno - 1);
        char       *attributeName = NameStr(attribute->attname);
 
        /* Ignore dropped */
        if (attribute->attisdropped)
            continue;
 
        if (attribute->attidentity)
            ereport(ERROR,
                    errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                    errmsg("table \"%s\" being attached contains an identity column \"%s\"",
                           RelationGetRelationName(attachrel), attributeName),
                    errdetail("The new partition may not contain an identity column."));
 
        /* Try to find the column in parent (matching on column name) */
        if (!SearchSysCacheExists2(ATTNAME,
                                   ObjectIdGetDatum(RelationGetRelid(rel)),
                                   CStringGetDatum(attributeName)))
            ereport(ERROR,
                    (errcode(ERRCODE_DATATYPE_MISMATCH),
                     errmsg("table \"%s\" contains column \"%s\" not found in parent \"%s\"",
                            RelationGetRelationName(attachrel), attributeName,
                            RelationGetRelationName(rel)),
                     errdetail("The new partition may contain only the columns present in parent.")));
    }
 
    /*
     * If child_rel has row-level triggers with transition tables, we
     * currently don't allow it to become a partition.  See also prohibitions
     * in ATExecAddInherit() and CreateTrigger().
     */
    trigger_name = FindTriggerIncompatibleWithInheritance(attachrel->trigdesc);
    if (trigger_name != NULL)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("trigger \"%s\" prevents table \"%s\" from becoming a partition",
                        trigger_name, RelationGetRelationName(attachrel)),
                 errdetail("ROW triggers with transition tables are not supported on partitions.")));
 
    /*
     * Check that the new partition's bound is valid and does not overlap any
     * of existing partitions of the parent - note that it does not return on
     * error.
     */
    check_new_partition_bound(RelationGetRelationName(attachrel), rel,
                              cmd->bound, pstate);
 
    attachPartitionTable(wqueue, rel, attachrel, cmd->bound);
 
    /*
     * Generate a partition constraint from the partition bound specification.
     * If the parent itself is a partition, make sure to include its
     * constraint as well.
     */
    partBoundConstraint = get_qual_from_partbound(rel, cmd->bound);
 
    /*
     * Use list_concat_copy() to avoid modifying partBoundConstraint in place,
     * since it's needed later to construct the constraint expression for
     * validating against the default partition, if any.
     */
    partConstraint = list_concat_copy(partBoundConstraint,
                                      RelationGetPartitionQual(rel));
 
    /* Skip validation if there are no constraints to validate. */
    if (partConstraint)
    {
        /*
         * Run the partition quals through const-simplification similar to
         * check constraints.  We skip canonicalize_qual, though, because
         * partition quals should be in canonical form already.
         */
        partConstraint =
            (List *) eval_const_expressions(NULL,
                                            (Node *) partConstraint);
 
        /* XXX this sure looks wrong */
        partConstraint = list_make1(make_ands_explicit(partConstraint));
 
        /*
         * Adjust the generated constraint to match this partition's attribute
         * numbers.
         */
        partConstraint = map_partition_varattnos(partConstraint, 1, attachrel,
                                                 rel);
 
        /* Validate partition constraints against the table being attached. */
        QueuePartitionConstraintValidation(wqueue, attachrel, partConstraint,
                                           false);
    }
 
    /*
     * If we're attaching a partition other than the default partition and a
     * default one exists, then that partition's partition constraint changes,
     * so add an entry to the work queue to validate it, too.  (We must not do
     * this when the partition being attached is the default one; we already
     * did it above!)
     */
    if (OidIsValid(defaultPartOid))
    {
        Relation    defaultrel;
        List       *defPartConstraint;
 
        Assert(!cmd->bound->is_default);
 
        /* we already hold a lock on the default partition */
        defaultrel = table_open(defaultPartOid, NoLock);
        defPartConstraint =
            get_proposed_default_constraint(partBoundConstraint);
 
        /*
         * Map the Vars in the constraint expression from rel's attnos to
         * defaultrel's.
         */
        defPartConstraint =
            map_partition_varattnos(defPartConstraint,
                                    1, defaultrel, rel);
        QueuePartitionConstraintValidation(wqueue, defaultrel,
                                           defPartConstraint, true);
 
        /* keep our lock until commit. */
        table_close(defaultrel, NoLock);
    }
 
    ObjectAddressSet(address, RelationRelationId, RelationGetRelid(attachrel));
 
    /*
     * If the partition we just attached is partitioned itself, invalidate
     * relcache for all descendent partitions too to ensure that their
     * rd_partcheck expression trees are rebuilt; partitions already locked at
     * the beginning of this function.
     */
    if (attachrel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
    {
        ListCell   *l;
 
        foreach(l, attachrel_children)
        {
            CacheInvalidateRelcacheByRelid(lfirst_oid(l));
        }
    }
 
    /* keep our lock until commit */
    table_close(attachrel, NoLock);
 
    return address;
}

References _, AccessExclusiveLock, AccessShareLock, appendStringInfo(), Assert, AT_AttachPartition, ATSimplePermissions(), ATT_FOREIGN_TABLE, ATT_PARTITIONED_TABLE, ATT_TABLE, attachPartitionTable(), PartitionCmd::bound, BTEqualStrategyNumber, CacheInvalidateRelcacheByRelid(), check_new_partition_bound(), CStringGetDatum(), ereport, errcode(), errdetail(), errhint(), errmsg, errmsg_plural(), ERROR, eval_const_expressions(), fb(), find_all_inheritors(), FindTriggerIncompatibleWithInheritance(), foreach_oid, get_default_oid_from_partdesc(), get_proposed_default_constraint(), get_publication_name(), get_qual_from_partbound(), GetRelationExcludedPublications(), HeapTupleIsValid, initStringInfo(), PartitionBoundSpec::is_default, lfirst_oid, list_concat_copy(), list_free(), list_length(), list_make1, list_member_oid(), LockRelationOid(), make_ands_explicit(), make_parsestate(), map_partition_varattnos(), PartitionCmd::name, NameStr, NIL, NoLock, ObjectAddressSet, ObjectIdGetDatum(), OidIsValid, ParseState::p_sourcetext, AlterTableUtilityContext::queryString, QueuePartitionConstraintValidation(), RelationData::rd_rel, RELATION_IS_OTHER_TEMP, RelationGetDescr, RelationGetPartitionDesc(), RelationGetPartitionQual(), RelationGetRelationName, RelationGetRelid, ScanKeyInit(), SearchSysCacheExists2, systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), table_open(), table_openrv(), and TupleDescAttr().

Referenced by ATExecCmd().

◆ ATExecAttachPartitionIdx()

static ObjectAddress ATExecAttachPartitionIdx	(	List **	wqueue,
		Relation	parentIdx,
		RangeVar *	name
	)

static

Definition at line 22367 of file tablecmds.c.

{
    Relation    partIdx;
    Relation    partTbl;
    Relation    parentTbl;
    ObjectAddress address;
    Oid         partIdxId;
    Oid         currParent;
    struct AttachIndexCallbackState state;
 
    /*
     * We need to obtain lock on the index 'name' to modify it, but we also
     * need to read its owning table's tuple descriptor -- so we need to lock
     * both.  To avoid deadlocks, obtain lock on the table before doing so on
     * the index.  Furthermore, we need to examine the parent table of the
     * partition, so lock that one too.
     */
    state.partitionOid = InvalidOid;
    state.parentTblOid = parentIdx->rd_index->indrelid;
    state.lockedParentTbl = false;
    partIdxId =
        RangeVarGetRelidExtended(name, AccessExclusiveLock, 0,
                                 RangeVarCallbackForAttachIndex,
                                 &state);
    /* Not there? */
    if (!OidIsValid(partIdxId))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_OBJECT),
                 errmsg("index \"%s\" does not exist", name->relname)));
 
    /* no deadlock risk: RangeVarGetRelidExtended already acquired the lock */
    partIdx = relation_open(partIdxId, AccessExclusiveLock);
 
    /* we already hold locks on both tables, so this is safe: */
    parentTbl = relation_open(parentIdx->rd_index->indrelid, AccessShareLock);
    partTbl = relation_open(partIdx->rd_index->indrelid, NoLock);
 
    ObjectAddressSet(address, RelationRelationId, RelationGetRelid(partIdx));
 
    /*
     * Check if the index is already attached to the correct parent,
     * ultimately attempting one round of validation if already the case.
     */
    currParent = partIdx->rd_rel->relispartition ?
        get_partition_parent(partIdxId, false) : InvalidOid;
    if (currParent != RelationGetRelid(parentIdx))
    {
        IndexInfo  *childInfo;
        IndexInfo  *parentInfo;
        AttrMap    *attmap;
        bool        found;
        int         i;
        PartitionDesc partDesc;
        Oid         constraintOid,
                    cldConstrId = InvalidOid;
 
        /*
         * If this partition already has an index attached, refuse the
         * operation.
         */
        refuseDupeIndexAttach(parentIdx, partIdx, partTbl);
 
        if (OidIsValid(currParent))
            ereport(ERROR,
                    (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                     errmsg("cannot attach index \"%s\" as a partition of index \"%s\"",
                            RelationGetRelationName(partIdx),
                            RelationGetRelationName(parentIdx)),
                     errdetail("Index \"%s\" is already attached to another index.",
                               RelationGetRelationName(partIdx))));
 
        /* Make sure it indexes a partition of the other index's table */
        partDesc = RelationGetPartitionDesc(parentTbl, true);
        found = false;
        for (i = 0; i < partDesc->nparts; i++)
        {
            if (partDesc->oids[i] == state.partitionOid)
            {
                found = true;
                break;
            }
        }
        if (!found)
            ereport(ERROR,
                    (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                     errmsg("cannot attach index \"%s\" as a partition of index \"%s\"",
                            RelationGetRelationName(partIdx),
                            RelationGetRelationName(parentIdx)),
                     errdetail("Index \"%s\" is not an index on any partition of table \"%s\".",
                               RelationGetRelationName(partIdx),
                               RelationGetRelationName(parentTbl))));
 
        /* Ensure the indexes are compatible */
        childInfo = BuildIndexInfo(partIdx);
        parentInfo = BuildIndexInfo(parentIdx);
        attmap = build_attrmap_by_name(RelationGetDescr(partTbl),
                                       RelationGetDescr(parentTbl),
                                       false);
        if (!CompareIndexInfo(childInfo, parentInfo,
                              partIdx->rd_indcollation,
                              parentIdx->rd_indcollation,
                              partIdx->rd_opfamily,
                              parentIdx->rd_opfamily,
                              attmap))
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                     errmsg("cannot attach index \"%s\" as a partition of index \"%s\"",
                            RelationGetRelationName(partIdx),
                            RelationGetRelationName(parentIdx)),
                     errdetail("The index definitions do not match.")));
 
        /*
         * If there is a constraint in the parent, make sure there is one in
         * the child too.
         */
        constraintOid = get_relation_idx_constraint_oid(RelationGetRelid(parentTbl),
                                                        RelationGetRelid(parentIdx));
 
        if (OidIsValid(constraintOid))
        {
            cldConstrId = get_relation_idx_constraint_oid(RelationGetRelid(partTbl),
                                                          partIdxId);
            if (!OidIsValid(cldConstrId))
                ereport(ERROR,
                        (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                         errmsg("cannot attach index \"%s\" as a partition of index \"%s\"",
                                RelationGetRelationName(partIdx),
                                RelationGetRelationName(parentIdx)),
                         errdetail("The index \"%s\" belongs to a constraint in table \"%s\" but no constraint exists for index \"%s\".",
                                   RelationGetRelationName(parentIdx),
                                   RelationGetRelationName(parentTbl),
                                   RelationGetRelationName(partIdx))));
        }
 
        /*
         * If it's a primary key, make sure the columns in the partition are
         * NOT NULL.
         */
        if (parentIdx->rd_index->indisprimary)
            verifyPartitionIndexNotNull(childInfo, partTbl);
 
        /* All good -- do it */
        IndexSetParentIndex(partIdx, RelationGetRelid(parentIdx));
        if (OidIsValid(constraintOid))
            ConstraintSetParentConstraint(cldConstrId, constraintOid,
                                          RelationGetRelid(partTbl));
 
        free_attrmap(attmap);
 
        validatePartitionedIndex(parentIdx, parentTbl);
    }
    else if (!parentIdx->rd_index->indisvalid)
    {
        /*
         * The index is attached, but the parent is still invalid; see if it
         * can be validated now.
         */
        validatePartitionedIndex(parentIdx, parentTbl);
    }
 
    relation_close(parentTbl, AccessShareLock);
    /* keep these locks till commit */
    relation_close(partTbl, NoLock);
    relation_close(partIdx, NoLock);
 
    return address;
}

References AccessExclusiveLock, AccessShareLock, build_attrmap_by_name(), BuildIndexInfo(), CompareIndexInfo(), ConstraintSetParentConstraint(), ereport, errcode(), errdetail(), errmsg, ERROR, fb(), free_attrmap(), get_partition_parent(), get_relation_idx_constraint_oid(), i, IndexSetParentIndex(), InvalidOid, name, NoLock, ObjectAddressSet, OidIsValid, RangeVarCallbackForAttachIndex(), RangeVarGetRelidExtended(), RelationData::rd_indcollation, RelationData::rd_index, RelationData::rd_opfamily, refuseDupeIndexAttach(), relation_close(), relation_open(), RelationGetDescr, RelationGetPartitionDesc(), RelationGetRelationName, RelationGetRelid, validatePartitionedIndex(), and verifyPartitionIndexNotNull().

Referenced by ATExecCmd().

◆ ATExecChangeOwner()

void ATExecChangeOwner	(	Oid	relationOid,
		Oid	newOwnerId,
		bool	recursing,
		LOCKMODE	lockmode
	)

Definition at line 16777 of file tablecmds.c.

{
    Relation    target_rel;
    Relation    class_rel;
    HeapTuple   tuple;
    Form_pg_class tuple_class;
 
    /*
     * Get exclusive lock till end of transaction on the target table. Use
     * relation_open so that we can work on indexes and sequences.
     */
    target_rel = relation_open(relationOid, lockmode);
 
    /* Get its pg_class tuple, too */
    class_rel = table_open(RelationRelationId, RowExclusiveLock);
 
    tuple = SearchSysCache1(RELOID, ObjectIdGetDatum(relationOid));
    if (!HeapTupleIsValid(tuple))
        elog(ERROR, "cache lookup failed for relation %u", relationOid);
    tuple_class = (Form_pg_class) GETSTRUCT(tuple);
 
    /* Can we change the ownership of this tuple? */
    switch (tuple_class->relkind)
    {
        case RELKIND_RELATION:
        case RELKIND_VIEW:
        case RELKIND_MATVIEW:
        case RELKIND_FOREIGN_TABLE:
        case RELKIND_PARTITIONED_TABLE:
        case RELKIND_PROPGRAPH:
            /* ok to change owner */
            break;
        case RELKIND_INDEX:
            if (!recursing)
            {
                /*
                 * Because ALTER INDEX OWNER used to be allowed, and in fact
                 * is generated by old versions of pg_dump, we give a warning
                 * and do nothing rather than erroring out.  Also, to avoid
                 * unnecessary chatter while restoring those old dumps, say
                 * nothing at all if the command would be a no-op anyway.
                 */
                if (tuple_class->relowner != newOwnerId)
                    ereport(WARNING,
                            (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                             errmsg("cannot change owner of index \"%s\"",
                                    NameStr(tuple_class->relname)),
                             errhint("Change the ownership of the index's table instead.")));
                /* quick hack to exit via the no-op path */
                newOwnerId = tuple_class->relowner;
            }
            break;
        case RELKIND_PARTITIONED_INDEX:
            if (recursing)
                break;
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("cannot change owner of index \"%s\"",
                            NameStr(tuple_class->relname)),
                     errhint("Change the ownership of the index's table instead.")));
            break;
        case RELKIND_SEQUENCE:
            if (!recursing &&
                tuple_class->relowner != newOwnerId)
            {
                /* if it's an owned sequence, disallow changing it by itself */
                Oid         tableId;
                int32       colId;
 
                if (sequenceIsOwned(relationOid, DEPENDENCY_AUTO, &tableId, &colId) ||
                    sequenceIsOwned(relationOid, DEPENDENCY_INTERNAL, &tableId, &colId))
                    ereport(ERROR,
                            (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                             errmsg("cannot change owner of sequence \"%s\"",
                                    NameStr(tuple_class->relname)),
                             errdetail("Sequence \"%s\" is linked to table \"%s\".",
                                       NameStr(tuple_class->relname),
                                       get_rel_name(tableId))));
            }
            break;
        case RELKIND_COMPOSITE_TYPE:
            if (recursing)
                break;
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("\"%s\" is a composite type",
                            NameStr(tuple_class->relname)),
            /* translator: %s is an SQL ALTER command */
                     errhint("Use %s instead.",
                             "ALTER TYPE")));
            break;
        case RELKIND_TOASTVALUE:
            if (recursing)
                break;
            pg_fallthrough;
        default:
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("cannot change owner of relation \"%s\"",
                            NameStr(tuple_class->relname)),
                     errdetail_relkind_not_supported(tuple_class->relkind)));
    }
 
    /*
     * If the new owner is the same as the existing owner, consider the
     * command to have succeeded.  This is for dump restoration purposes.
     */
    if (tuple_class->relowner != newOwnerId)
    {
        Datum       repl_val[Natts_pg_class];
        bool        repl_null[Natts_pg_class];
        bool        repl_repl[Natts_pg_class];
        Acl        *newAcl;
        Datum       aclDatum;
        bool        isNull;
        HeapTuple   newtuple;
 
        /* skip permission checks when recursing to index or toast table */
        if (!recursing)
        {
            /* Superusers can always do it */
            if (!superuser())
            {
                Oid         namespaceOid = tuple_class->relnamespace;
                AclResult   aclresult;
 
                /* Otherwise, must be owner of the existing object */
                if (!object_ownercheck(RelationRelationId, relationOid, GetUserId()))
                    aclcheck_error(ACLCHECK_NOT_OWNER, get_relkind_objtype(get_rel_relkind(relationOid)),
                                   RelationGetRelationName(target_rel));
 
                /* Must be able to become new owner */
                check_can_set_role(GetUserId(), newOwnerId);
 
                /* New owner must have CREATE privilege on namespace */
                aclresult = object_aclcheck(NamespaceRelationId, namespaceOid, newOwnerId,
                                            ACL_CREATE);
                if (aclresult != ACLCHECK_OK)
                    aclcheck_error(aclresult, OBJECT_SCHEMA,
                                   get_namespace_name(namespaceOid));
            }
        }
 
        memset(repl_null, false, sizeof(repl_null));
        memset(repl_repl, false, sizeof(repl_repl));
 
        repl_repl[Anum_pg_class_relowner - 1] = true;
        repl_val[Anum_pg_class_relowner - 1] = ObjectIdGetDatum(newOwnerId);
 
        /*
         * Determine the modified ACL for the new owner.  This is only
         * necessary when the ACL is non-null.
         */
        aclDatum = SysCacheGetAttr(RELOID, tuple,
                                   Anum_pg_class_relacl,
                                   &isNull);
        if (!isNull)
        {
            newAcl = aclnewowner(DatumGetAclP(aclDatum),
                                 tuple_class->relowner, newOwnerId);
            repl_repl[Anum_pg_class_relacl - 1] = true;
            repl_val[Anum_pg_class_relacl - 1] = PointerGetDatum(newAcl);
        }
 
        newtuple = heap_modify_tuple(tuple, RelationGetDescr(class_rel), repl_val, repl_null, repl_repl);
 
        CatalogTupleUpdate(class_rel, &newtuple->t_self, newtuple);
 
        heap_freetuple(newtuple);
 
        /*
         * We must similarly update any per-column ACLs to reflect the new
         * owner; for neatness reasons that's split out as a subroutine.
         */
        change_owner_fix_column_acls(relationOid,
                                     tuple_class->relowner,
                                     newOwnerId);
 
        /*
         * Update owner dependency reference, if any.  A composite type has
         * none, because it's tracked for the pg_type entry instead of here;
         * indexes and TOAST tables don't have their own entries either.
         */
        if (tuple_class->relkind != RELKIND_COMPOSITE_TYPE &&
            tuple_class->relkind != RELKIND_INDEX &&
            tuple_class->relkind != RELKIND_PARTITIONED_INDEX &&
            tuple_class->relkind != RELKIND_TOASTVALUE)
            changeDependencyOnOwner(RelationRelationId, relationOid,
                                    newOwnerId);
 
        /*
         * Also change the ownership of the table's row type, if it has one
         */
        if (OidIsValid(tuple_class->reltype))
            AlterTypeOwnerInternal(tuple_class->reltype, newOwnerId);
 
        /*
         * If we are operating on a table or materialized view, also change
         * the ownership of any indexes and sequences that belong to the
         * relation, as well as its toast table (if it has one).
         */
        if (tuple_class->relkind == RELKIND_RELATION ||
            tuple_class->relkind == RELKIND_PARTITIONED_TABLE ||
            tuple_class->relkind == RELKIND_MATVIEW ||
            tuple_class->relkind == RELKIND_TOASTVALUE)
        {
            List       *index_oid_list;
            ListCell   *i;
 
            /* Find all the indexes belonging to this relation */
            index_oid_list = RelationGetIndexList(target_rel);
 
            /* For each index, recursively change its ownership */
            foreach(i, index_oid_list)
                ATExecChangeOwner(lfirst_oid(i), newOwnerId, true, lockmode);
 
            list_free(index_oid_list);
        }
 
        /* If it has a toast table, recurse to change its ownership */
        if (tuple_class->reltoastrelid != InvalidOid)
            ATExecChangeOwner(tuple_class->reltoastrelid, newOwnerId,
                              true, lockmode);
 
        /* If it has dependent sequences, recurse to change them too */
        change_owner_recurse_to_sequences(relationOid, newOwnerId, lockmode);
    }
 
    InvokeObjectPostAlterHook(RelationRelationId, relationOid, 0);
 
    ReleaseSysCache(tuple);
    table_close(class_rel, RowExclusiveLock);
    relation_close(target_rel, NoLock);
}

References ACL_CREATE, aclcheck_error(), ACLCHECK_NOT_OWNER, ACLCHECK_OK, aclnewowner(), AlterTypeOwnerInternal(), ATExecChangeOwner(), CatalogTupleUpdate(), change_owner_fix_column_acls(), change_owner_recurse_to_sequences(), changeDependencyOnOwner(), check_can_set_role(), DatumGetAclP, DEPENDENCY_AUTO, DEPENDENCY_INTERNAL, elog, ereport, errcode(), errdetail(), errdetail_relkind_not_supported(), errhint(), errmsg, ERROR, fb(), get_namespace_name(), get_rel_name(), get_rel_relkind(), get_relkind_objtype(), GETSTRUCT(), GetUserId(), heap_freetuple(), heap_modify_tuple(), HeapTupleIsValid, i, InvalidOid, InvokeObjectPostAlterHook, lfirst_oid, list_free(), NameStr, NoLock, object_aclcheck(), object_ownercheck(), OBJECT_SCHEMA, ObjectIdGetDatum(), OidIsValid, pg_fallthrough, PointerGetDatum, relation_close(), relation_open(), RelationGetDescr, RelationGetIndexList(), RelationGetRelationName, ReleaseSysCache(), RowExclusiveLock, SearchSysCache1(), sequenceIsOwned(), superuser(), SysCacheGetAttr(), HeapTupleData::t_self, table_close(), table_open(), and WARNING.

Referenced by AlterSubscriptionOwner_internal(), AlterTypeOwner_oid(), ATExecChangeOwner(), ATExecCmd(), change_owner_recurse_to_sequences(), and shdepReassignOwned_Owner().

◆ ATExecClusterOn()

static ObjectAddress ATExecClusterOn	(	Relation	rel,
		const char *	indexName,
		LOCKMODE	lockmode
	)

static

Definition at line 17153 of file tablecmds.c.

{
    Oid         indexOid;
    ObjectAddress address;
 
    indexOid = get_relname_relid(indexName, rel->rd_rel->relnamespace);
 
    if (!OidIsValid(indexOid))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_OBJECT),
                 errmsg("index \"%s\" for table \"%s\" does not exist",
                        indexName, RelationGetRelationName(rel))));
 
    /* Check index is valid to cluster on */
    check_index_is_clusterable(rel, indexOid, lockmode);
 
    /* And do the work */
    mark_index_clustered(rel, indexOid, false);
 
    ObjectAddressSet(address,
                     RelationRelationId, indexOid);
 
    return address;
}

References check_index_is_clusterable(), ereport, errcode(), errmsg, ERROR, fb(), get_relname_relid(), mark_index_clustered(), ObjectAddressSet, OidIsValid, RelationData::rd_rel, and RelationGetRelationName.

Referenced by ATExecCmd().

◆ ATExecCmd()

static void ATExecCmd	(	List **	wqueue,
		AlteredTableInfo *	tab,
		AlterTableCmd *	cmd,
		LOCKMODE	lockmode,
		AlterTablePass	cur_pass,
		AlterTableUtilityContext *	context
	)

static

Definition at line 5479 of file tablecmds.c.

{
    ObjectAddress address = InvalidObjectAddress;
    Relation    rel = tab->rel;
 
    switch (cmd->subtype)
    {
        case AT_AddColumn:      /* ADD COLUMN */
        case AT_AddColumnToView:    /* add column via CREATE OR REPLACE VIEW */
            address = ATExecAddColumn(wqueue, tab, rel, &cmd,
                                      cmd->recurse, false,
                                      lockmode, cur_pass, context);
            break;
        case AT_ColumnDefault:  /* ALTER COLUMN DEFAULT */
            address = ATExecColumnDefault(rel, cmd->name, cmd->def, lockmode);
            break;
        case AT_CookedColumnDefault:    /* add a pre-cooked default */
            address = ATExecCookedColumnDefault(rel, cmd->num, cmd->def);
            break;
        case AT_AddIdentity:
            cmd = ATParseTransformCmd(wqueue, tab, rel, cmd, false, lockmode,
                                      cur_pass, context);
            Assert(cmd != NULL);
            address = ATExecAddIdentity(rel, cmd->name, cmd->def, lockmode, cmd->recurse, false);
            break;
        case AT_SetIdentity:
            cmd = ATParseTransformCmd(wqueue, tab, rel, cmd, false, lockmode,
                                      cur_pass, context);
            Assert(cmd != NULL);
            address = ATExecSetIdentity(rel, cmd->name, cmd->def, lockmode, cmd->recurse, false);
            break;
        case AT_DropIdentity:
            address = ATExecDropIdentity(rel, cmd->name, cmd->missing_ok, lockmode, cmd->recurse, false);
            break;
        case AT_DropNotNull:    /* ALTER COLUMN DROP NOT NULL */
            address = ATExecDropNotNull(rel, cmd->name, cmd->recurse, lockmode);
            break;
        case AT_SetNotNull:     /* ALTER COLUMN SET NOT NULL */
            address = ATExecSetNotNull(wqueue, rel, NULL, cmd->name,
                                       cmd->recurse, false, lockmode);
            break;
        case AT_SetExpression:
            address = ATExecSetExpression(tab, rel, cmd->name, cmd->def, lockmode);
            break;
        case AT_DropExpression:
            address = ATExecDropExpression(rel, cmd->name, cmd->missing_ok, lockmode);
            break;
        case AT_SetStatistics:  /* ALTER COLUMN SET STATISTICS */
            address = ATExecSetStatistics(rel, cmd->name, cmd->num, cmd->def, lockmode);
            break;
        case AT_SetOptions:     /* ALTER COLUMN SET ( options ) */
            address = ATExecSetOptions(rel, cmd->name, cmd->def, false, lockmode);
            break;
        case AT_ResetOptions:   /* ALTER COLUMN RESET ( options ) */
            address = ATExecSetOptions(rel, cmd->name, cmd->def, true, lockmode);
            break;
        case AT_SetStorage:     /* ALTER COLUMN SET STORAGE */
            address = ATExecSetStorage(rel, cmd->name, cmd->def, lockmode);
            break;
        case AT_SetCompression: /* ALTER COLUMN SET COMPRESSION */
            address = ATExecSetCompression(rel, cmd->name, cmd->def,
                                           lockmode);
            break;
        case AT_DropColumn:     /* DROP COLUMN */
            address = ATExecDropColumn(wqueue, rel, cmd->name,
                                       cmd->behavior, cmd->recurse, false,
                                       cmd->missing_ok, lockmode,
                                       NULL);
            break;
        case AT_AddIndex:       /* ADD INDEX */
            address = ATExecAddIndex(tab, rel, (IndexStmt *) cmd->def, false,
                                     lockmode);
            break;
        case AT_ReAddIndex:     /* ADD INDEX */
            address = ATExecAddIndex(tab, rel, (IndexStmt *) cmd->def, true,
                                     lockmode);
            break;
        case AT_ReAddStatistics:    /* ADD STATISTICS */
            address = ATExecAddStatistics(tab, rel, (CreateStatsStmt *) cmd->def,
                                          true, lockmode);
            break;
        case AT_AddConstraint:  /* ADD CONSTRAINT */
            /* Transform the command only during initial examination */
            if (cur_pass == AT_PASS_ADD_CONSTR)
                cmd = ATParseTransformCmd(wqueue, tab, rel, cmd,
                                          cmd->recurse, lockmode,
                                          cur_pass, context);
            /* Depending on constraint type, might be no more work to do now */
            if (cmd != NULL)
                address =
                    ATExecAddConstraint(wqueue, tab, rel,
                                        (Constraint *) cmd->def,
                                        cmd->recurse, false, lockmode);
            break;
        case AT_ReAddConstraint:    /* Re-add pre-existing check constraint */
            address =
                ATExecAddConstraint(wqueue, tab, rel, (Constraint *) cmd->def,
                                    true, true, lockmode);
            break;
        case AT_ReAddDomainConstraint:  /* Re-add pre-existing domain check
                                         * constraint */
            address =
                AlterDomainAddConstraint(((AlterDomainStmt *) cmd->def)->typeName,
                                         ((AlterDomainStmt *) cmd->def)->def,
                                         NULL);
            break;
        case AT_ReAddComment:   /* Re-add existing comment */
            address = CommentObject((CommentStmt *) cmd->def);
            break;
        case AT_AddIndexConstraint: /* ADD CONSTRAINT USING INDEX */
            address = ATExecAddIndexConstraint(tab, rel, (IndexStmt *) cmd->def,
                                               lockmode);
            break;
        case AT_AlterConstraint:    /* ALTER CONSTRAINT */
            address = ATExecAlterConstraint(wqueue, rel,
                                            castNode(ATAlterConstraint, cmd->def),
                                            cmd->recurse, lockmode);
            break;
        case AT_ValidateConstraint: /* VALIDATE CONSTRAINT */
            address = ATExecValidateConstraint(wqueue, rel, cmd->name, cmd->recurse,
                                               false, lockmode);
            break;
        case AT_DropConstraint: /* DROP CONSTRAINT */
            ATExecDropConstraint(rel, cmd->name, cmd->behavior,
                                 cmd->recurse,
                                 cmd->missing_ok, lockmode);
            break;
        case AT_AlterColumnType:    /* ALTER COLUMN TYPE */
            /* parse transformation was done earlier */
            address = ATExecAlterColumnType(tab, rel, cmd, lockmode);
            break;
        case AT_AlterColumnGenericOptions:  /* ALTER COLUMN OPTIONS */
            address =
                ATExecAlterColumnGenericOptions(rel, cmd->name,
                                                (List *) cmd->def, lockmode);
            break;
        case AT_ChangeOwner:    /* ALTER OWNER */
            ATExecChangeOwner(RelationGetRelid(rel),
                              get_rolespec_oid(cmd->newowner, false),
                              false, lockmode);
            break;
        case AT_ClusterOn:      /* CLUSTER ON */
            address = ATExecClusterOn(rel, cmd->name, lockmode);
            break;
        case AT_DropCluster:    /* SET WITHOUT CLUSTER */
            ATExecDropCluster(rel, lockmode);
            break;
        case AT_SetLogged:      /* SET LOGGED */
        case AT_SetUnLogged:    /* SET UNLOGGED */
            break;
        case AT_DropOids:       /* SET WITHOUT OIDS */
            /* nothing to do here, oid columns don't exist anymore */
            break;
        case AT_SetAccessMethod:    /* SET ACCESS METHOD */
 
            /*
             * Only do this for partitioned tables, for which this is just a
             * catalog change.  Tables with storage are handled by Phase 3.
             */
            if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE &&
                tab->chgAccessMethod)
                ATExecSetAccessMethodNoStorage(rel, tab->newAccessMethod);
            break;
        case AT_SetTableSpace:  /* SET TABLESPACE */
 
            /*
             * Only do this for partitioned tables and indexes, for which this
             * is just a catalog change.  Other relation types which have
             * storage are handled by Phase 3.
             */
            if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE ||
                rel->rd_rel->relkind == RELKIND_PARTITIONED_INDEX)
                ATExecSetTableSpaceNoStorage(rel, tab->newTableSpace);
 
            break;
        case AT_SetRelOptions:  /* SET (...) */
        case AT_ResetRelOptions:    /* RESET (...) */
        case AT_ReplaceRelOptions:  /* replace entire option list */
            ATExecSetRelOptions(rel, (List *) cmd->def, cmd->subtype, lockmode);
            break;
        case AT_EnableTrig:     /* ENABLE TRIGGER name */
            ATExecEnableDisableTrigger(rel, cmd->name,
                                       TRIGGER_FIRES_ON_ORIGIN, false,
                                       cmd->recurse,
                                       lockmode);
            break;
        case AT_EnableAlwaysTrig:   /* ENABLE ALWAYS TRIGGER name */
            ATExecEnableDisableTrigger(rel, cmd->name,
                                       TRIGGER_FIRES_ALWAYS, false,
                                       cmd->recurse,
                                       lockmode);
            break;
        case AT_EnableReplicaTrig:  /* ENABLE REPLICA TRIGGER name */
            ATExecEnableDisableTrigger(rel, cmd->name,
                                       TRIGGER_FIRES_ON_REPLICA, false,
                                       cmd->recurse,
                                       lockmode);
            break;
        case AT_DisableTrig:    /* DISABLE TRIGGER name */
            ATExecEnableDisableTrigger(rel, cmd->name,
                                       TRIGGER_DISABLED, false,
                                       cmd->recurse,
                                       lockmode);
            break;
        case AT_EnableTrigAll:  /* ENABLE TRIGGER ALL */
            ATExecEnableDisableTrigger(rel, NULL,
                                       TRIGGER_FIRES_ON_ORIGIN, false,
                                       cmd->recurse,
                                       lockmode);
            break;
        case AT_DisableTrigAll: /* DISABLE TRIGGER ALL */
            ATExecEnableDisableTrigger(rel, NULL,
                                       TRIGGER_DISABLED, false,
                                       cmd->recurse,
                                       lockmode);
            break;
        case AT_EnableTrigUser: /* ENABLE TRIGGER USER */
            ATExecEnableDisableTrigger(rel, NULL,
                                       TRIGGER_FIRES_ON_ORIGIN, true,
                                       cmd->recurse,
                                       lockmode);
            break;
        case AT_DisableTrigUser:    /* DISABLE TRIGGER USER */
            ATExecEnableDisableTrigger(rel, NULL,
                                       TRIGGER_DISABLED, true,
                                       cmd->recurse,
                                       lockmode);
            break;
 
        case AT_EnableRule:     /* ENABLE RULE name */
            ATExecEnableDisableRule(rel, cmd->name,
                                    RULE_FIRES_ON_ORIGIN, lockmode);
            break;
        case AT_EnableAlwaysRule:   /* ENABLE ALWAYS RULE name */
            ATExecEnableDisableRule(rel, cmd->name,
                                    RULE_FIRES_ALWAYS, lockmode);
            break;
        case AT_EnableReplicaRule:  /* ENABLE REPLICA RULE name */
            ATExecEnableDisableRule(rel, cmd->name,
                                    RULE_FIRES_ON_REPLICA, lockmode);
            break;
        case AT_DisableRule:    /* DISABLE RULE name */
            ATExecEnableDisableRule(rel, cmd->name,
                                    RULE_DISABLED, lockmode);
            break;
 
        case AT_AddInherit:
            address = ATExecAddInherit(rel, (RangeVar *) cmd->def, lockmode);
            break;
        case AT_DropInherit:
            address = ATExecDropInherit(rel, (RangeVar *) cmd->def, lockmode);
            break;
        case AT_AddOf:
            address = ATExecAddOf(rel, (TypeName *) cmd->def, lockmode);
            break;
        case AT_DropOf:
            ATExecDropOf(rel, lockmode);
            break;
        case AT_ReplicaIdentity:
            ATExecReplicaIdentity(rel, (ReplicaIdentityStmt *) cmd->def, lockmode);
            break;
        case AT_EnableRowSecurity:
            ATExecSetRowSecurity(rel, true);
            break;
        case AT_DisableRowSecurity:
            ATExecSetRowSecurity(rel, false);
            break;
        case AT_ForceRowSecurity:
            ATExecForceNoForceRowSecurity(rel, true);
            break;
        case AT_NoForceRowSecurity:
            ATExecForceNoForceRowSecurity(rel, false);
            break;
        case AT_GenericOptions:
            ATExecGenericOptions(rel, (List *) cmd->def);
            break;
        case AT_AttachPartition:
            cmd = ATParseTransformCmd(wqueue, tab, rel, cmd, false, lockmode,
                                      cur_pass, context);
            Assert(cmd != NULL);
            if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
                address = ATExecAttachPartition(wqueue, rel, (PartitionCmd *) cmd->def,
                                                context);
            else
                address = ATExecAttachPartitionIdx(wqueue, rel,
                                                   ((PartitionCmd *) cmd->def)->name);
            break;
        case AT_DetachPartition:
            cmd = ATParseTransformCmd(wqueue, tab, rel, cmd, false, lockmode,
                                      cur_pass, context);
            Assert(cmd != NULL);
            /* ATPrepCmd ensures it must be a table */
            Assert(rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE);
            address = ATExecDetachPartition(wqueue, tab, rel,
                                            ((PartitionCmd *) cmd->def)->name,
                                            ((PartitionCmd *) cmd->def)->concurrent);
            break;
        case AT_DetachPartitionFinalize:
            address = ATExecDetachPartitionFinalize(rel, ((PartitionCmd *) cmd->def)->name);
            break;
        case AT_MergePartitions:
            cmd = ATParseTransformCmd(wqueue, tab, rel, cmd, false, lockmode,
                                      cur_pass, context);
            Assert(cmd != NULL);
            Assert(rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE);
            ATExecMergePartitions(wqueue, tab, rel, (PartitionCmd *) cmd->def,
                                  context);
            break;
        case AT_SplitPartition:
            cmd = ATParseTransformCmd(wqueue, tab, rel, cmd, false, lockmode,
                                      cur_pass, context);
            Assert(cmd != NULL);
            Assert(rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE);
            ATExecSplitPartition(wqueue, tab, rel, (PartitionCmd *) cmd->def,
                                 context);
            break;
        default:                /* oops */
            elog(ERROR, "unrecognized alter table type: %d",
                 (int) cmd->subtype);
            break;
    }
 
    /*
     * Report the subcommand to interested event triggers.
     */
    if (cmd)
        EventTriggerCollectAlterTableSubcmd((Node *) cmd, address);
 
    /*
     * Bump the command counter to ensure the next subcommand in the sequence
     * can see the changes so far
     */
    CommandCounterIncrement();
}

References AlterDomainAddConstraint(), Assert, AT_AddColumn, AT_AddColumnToView, AT_AddConstraint, AT_AddIdentity, AT_AddIndex, AT_AddIndexConstraint, AT_AddInherit, AT_AddOf, AT_AlterColumnGenericOptions, AT_AlterColumnType, AT_AlterConstraint, AT_AttachPartition, AT_ChangeOwner, AT_ClusterOn, AT_ColumnDefault, AT_CookedColumnDefault, AT_DetachPartition, AT_DetachPartitionFinalize, AT_DisableRowSecurity, AT_DisableRule, AT_DisableTrig, AT_DisableTrigAll, AT_DisableTrigUser, AT_DropCluster, AT_DropColumn, AT_DropConstraint, AT_DropExpression, AT_DropIdentity, AT_DropInherit, AT_DropNotNull, AT_DropOf, AT_DropOids, AT_EnableAlwaysRule, AT_EnableAlwaysTrig, AT_EnableReplicaRule, AT_EnableReplicaTrig, AT_EnableRowSecurity, AT_EnableRule, AT_EnableTrig, AT_EnableTrigAll, AT_EnableTrigUser, AT_ForceRowSecurity, AT_GenericOptions, AT_MergePartitions, AT_NoForceRowSecurity, AT_PASS_ADD_CONSTR, AT_ReAddComment, AT_ReAddConstraint, AT_ReAddDomainConstraint, AT_ReAddIndex, AT_ReAddStatistics, AT_ReplaceRelOptions, AT_ReplicaIdentity, AT_ResetOptions, AT_ResetRelOptions, AT_SetAccessMethod, AT_SetCompression, AT_SetExpression, AT_SetIdentity, AT_SetLogged, AT_SetNotNull, AT_SetOptions, AT_SetRelOptions, AT_SetStatistics, AT_SetStorage, AT_SetTableSpace, AT_SetUnLogged, AT_SplitPartition, AT_ValidateConstraint, ATExecAddColumn(), ATExecAddConstraint(), ATExecAddIdentity(), ATExecAddIndex(), ATExecAddIndexConstraint(), ATExecAddInherit(), ATExecAddOf(), ATExecAddStatistics(), ATExecAlterColumnGenericOptions(), ATExecAlterColumnType(), ATExecAlterConstraint(), ATExecAttachPartition(), ATExecAttachPartitionIdx(), ATExecChangeOwner(), ATExecClusterOn(), ATExecColumnDefault(), ATExecCookedColumnDefault(), ATExecDetachPartition(), ATExecDetachPartitionFinalize(), ATExecDropCluster(), ATExecDropColumn(), ATExecDropConstraint(), ATExecDropExpression(), ATExecDropIdentity(), ATExecDropInherit(), ATExecDropNotNull(), ATExecDropOf(), ATExecEnableDisableRule(), ATExecEnableDisableTrigger(), ATExecForceNoForceRowSecurity(), ATExecGenericOptions(), ATExecMergePartitions(), ATExecReplicaIdentity(), ATExecSetAccessMethodNoStorage(), ATExecSetCompression(), ATExecSetExpression(), ATExecSetIdentity(), ATExecSetNotNull(), ATExecSetOptions(), ATExecSetRelOptions(), ATExecSetRowSecurity(), ATExecSetStatistics(), ATExecSetStorage(), ATExecSetTableSpaceNoStorage(), ATExecSplitPartition(), ATExecValidateConstraint(), ATParseTransformCmd(), AlterTableCmd::behavior, castNode, AlteredTableInfo::chgAccessMethod, CommandCounterIncrement(), CommentObject(), AlterTableCmd::def, elog, ERROR, EventTriggerCollectAlterTableSubcmd(), fb(), get_rolespec_oid(), InvalidObjectAddress, AlterTableCmd::missing_ok, AlterTableCmd::name, AlteredTableInfo::newAccessMethod, AlterTableCmd::newowner, AlteredTableInfo::newTableSpace, AlterTableCmd::num, RelationData::rd_rel, AlterTableCmd::recurse, AlteredTableInfo::rel, RelationGetRelid, RULE_DISABLED, RULE_FIRES_ALWAYS, RULE_FIRES_ON_ORIGIN, RULE_FIRES_ON_REPLICA, AlterTableCmd::subtype, TRIGGER_DISABLED, TRIGGER_FIRES_ALWAYS, TRIGGER_FIRES_ON_ORIGIN, and TRIGGER_FIRES_ON_REPLICA.

Referenced by ATRewriteCatalogs().

◆ ATExecColumnDefault()

static ObjectAddress ATExecColumnDefault	(	Relation	rel,
		const char *	colName,
		Node *	newDefault,
		LOCKMODE	lockmode
	)

static

Definition at line 8267 of file tablecmds.c.

{
    TupleDesc   tupdesc = RelationGetDescr(rel);
    AttrNumber  attnum;
    ObjectAddress address;
 
    /*
     * get the number of the attribute
     */
    attnum = get_attnum(RelationGetRelid(rel), colName);
    if (attnum == InvalidAttrNumber)
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel))));
 
    /* Prevent them from altering a system attribute */
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"",
                        colName)));
 
    if (TupleDescAttr(tupdesc, attnum - 1)->attidentity)
        ereport(ERROR,
                (errcode(ERRCODE_SYNTAX_ERROR),
                 errmsg("column \"%s\" of relation \"%s\" is an identity column",
                        colName, RelationGetRelationName(rel)),
        /* translator: %s is an SQL ALTER command */
                 newDefault ? 0 : errhint("Use %s instead.",
                                          "ALTER TABLE ... ALTER COLUMN ... DROP IDENTITY")));
 
    if (TupleDescAttr(tupdesc, attnum - 1)->attgenerated)
        ereport(ERROR,
                (errcode(ERRCODE_SYNTAX_ERROR),
                 errmsg("column \"%s\" of relation \"%s\" is a generated column",
                        colName, RelationGetRelationName(rel)),
                 newDefault ?
        /* translator: %s is an SQL ALTER command */
                 errhint("Use %s instead.", "ALTER TABLE ... ALTER COLUMN ... SET EXPRESSION") :
                 (TupleDescAttr(tupdesc, attnum - 1)->attgenerated == ATTRIBUTE_GENERATED_STORED ?
                  errhint("Use %s instead.", "ALTER TABLE ... ALTER COLUMN ... DROP EXPRESSION") : 0)));
 
    /*
     * Remove any old default for the column.  We use RESTRICT here for
     * safety, but at present we do not expect anything to depend on the
     * default.
     *
     * We treat removing the existing default as an internal operation when it
     * is preparatory to adding a new default, but as a user-initiated
     * operation when the user asked for a drop.
     */
    RemoveAttrDefault(RelationGetRelid(rel), attnum, DROP_RESTRICT, false,
                      newDefault != NULL);
 
    if (newDefault)
    {
        /* SET DEFAULT */
        RawColumnDefault *rawEnt;
 
        rawEnt = palloc_object(RawColumnDefault);
        rawEnt->attnum = attnum;
        rawEnt->raw_default = newDefault;
        rawEnt->generated = '\0';
 
        /*
         * This function is intended for CREATE TABLE, so it processes a
         * _list_ of defaults, but we just do one.
         */
        AddRelationNewConstraints(rel, list_make1(rawEnt), NIL,
                                  false, true, false, NULL);
    }
 
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
    return address;
}

References AddRelationNewConstraints(), attnum, DROP_RESTRICT, ereport, errcode(), errhint(), errmsg, ERROR, fb(), get_attnum(), InvalidAttrNumber, list_make1, NIL, ObjectAddressSubSet, palloc_object, RelationGetDescr, RelationGetRelationName, RelationGetRelid, RemoveAttrDefault(), and TupleDescAttr().

Referenced by ATExecCmd().

◆ ATExecCookedColumnDefault()

static ObjectAddress ATExecCookedColumnDefault	(	Relation	rel,
		AttrNumber	attnum,
		Node *	newDefault
	)

static

Definition at line 8352 of file tablecmds.c.

{
    ObjectAddress address;
 
    /* We assume no checking is required */
 
    /*
     * Remove any old default for the column.  We use RESTRICT here for
     * safety, but at present we do not expect anything to depend on the
     * default.  (In ordinary cases, there could not be a default in place
     * anyway, but it's possible when combining LIKE with inheritance.)
     */
    RemoveAttrDefault(RelationGetRelid(rel), attnum, DROP_RESTRICT, false,
                      true);
 
    (void) StoreAttrDefault(rel, attnum, newDefault, true);
 
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
    return address;
}

References attnum, DROP_RESTRICT, fb(), ObjectAddressSubSet, RelationGetRelid, RemoveAttrDefault(), and StoreAttrDefault().

Referenced by ATExecCmd().

◆ ATExecDetachPartition()

static ObjectAddress ATExecDetachPartition	(	List **	wqueue,
		AlteredTableInfo *	tab,
		Relation	rel,
		RangeVar *	name,
		bool	concurrent
	)

static

Definition at line 21699 of file tablecmds.c.

{
    Relation    partRel;
    ObjectAddress address;
    Oid         defaultPartOid;
 
    /*
     * We must lock the default partition, because detaching this partition
     * will change its partition constraint.
     */
    defaultPartOid =
        get_default_oid_from_partdesc(RelationGetPartitionDesc(rel, true));
    if (OidIsValid(defaultPartOid))
    {
        /*
         * Concurrent detaching when a default partition exists is not
         * supported. The main problem is that the default partition
         * constraint would change.  And there's a definitional problem: what
         * should happen to the tuples that are being inserted that belong to
         * the partition being detached?  Putting them on the partition being
         * detached would be wrong, since they'd become "lost" after the
         * detaching completes but we cannot put them in the default partition
         * either until we alter its partition constraint.
         *
         * I think we could solve this problem if we effected the constraint
         * change before committing the first transaction.  But the lock would
         * have to remain AEL and it would cause concurrent query planning to
         * be blocked, so changing it that way would be even worse.
         */
        if (concurrent)
            ereport(ERROR,
                    (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                     errmsg("cannot detach partitions concurrently when a default partition exists")));
        LockRelationOid(defaultPartOid, AccessExclusiveLock);
    }
 
    /*
     * In concurrent mode, the partition is locked with share-update-exclusive
     * in the first transaction.  This allows concurrent transactions to be
     * doing DML to the partition.
     */
    partRel = table_openrv(name, concurrent ? ShareUpdateExclusiveLock :
                           AccessExclusiveLock);
 
    /*
     * Check inheritance conditions and either delete the pg_inherits row (in
     * non-concurrent mode) or just set the inhdetachpending flag.
     */
    if (!concurrent)
        RemoveInheritance(partRel, rel, false);
    else
        MarkInheritDetached(partRel, rel);
 
    /*
     * Ensure that foreign keys still hold after this detach.  This keeps
     * locks on the referencing tables, which prevents concurrent transactions
     * from adding rows that we wouldn't see.  For this to work in concurrent
     * mode, it is critical that the partition appears as no longer attached
     * for the RI queries as soon as the first transaction commits.
     */
    ATDetachCheckNoForeignKeyRefs(partRel);
 
    /*
     * Concurrent mode has to work harder; first we add a new constraint to
     * the partition that matches the partition constraint.  Then we close our
     * existing transaction, and in a new one wait for all processes to catch
     * up on the catalog updates we've done so far; at that point we can
     * complete the operation.
     */
    if (concurrent)
    {
        Oid         partrelid,
                    parentrelid;
        LOCKTAG     tag;
        char       *parentrelname;
        char       *partrelname;
 
        /*
         * We're almost done now; the only traces that remain are the
         * pg_inherits tuple and the partition's relpartbounds.  Before we can
         * remove those, we need to wait until all transactions that know that
         * this is a partition are gone.
         */
 
        /*
         * Remember relation OIDs to re-acquire them later; and relation names
         * too, for error messages if something is dropped in between.
         */
        partrelid = RelationGetRelid(partRel);
        parentrelid = RelationGetRelid(rel);
        parentrelname = MemoryContextStrdup(PortalContext,
                                            RelationGetRelationName(rel));
        partrelname = MemoryContextStrdup(PortalContext,
                                          RelationGetRelationName(partRel));
 
        /* Invalidate relcache entries for the parent -- must be before close */
        CacheInvalidateRelcache(rel);
 
        table_close(partRel, NoLock);
        table_close(rel, NoLock);
        tab->rel = NULL;
 
        /* Make updated catalog entry visible */
        PopActiveSnapshot();
        CommitTransactionCommand();
 
        StartTransactionCommand();
 
        /*
         * Now wait.  This ensures that all queries that were planned
         * including the partition are finished before we remove the rest of
         * catalog entries.  We don't need or indeed want to acquire this
         * lock, though -- that would block later queries.
         *
         * We don't need to concern ourselves with waiting for a lock on the
         * partition itself, since we will acquire AccessExclusiveLock below.
         */
        SET_LOCKTAG_RELATION(tag, MyDatabaseId, parentrelid);
        WaitForLockersMultiple(list_make1(&tag), AccessExclusiveLock, false);
 
        /*
         * Now acquire locks in both relations again.  Note they may have been
         * removed in the meantime, so care is required.
         */
        rel = try_relation_open(parentrelid, ShareUpdateExclusiveLock);
        partRel = try_relation_open(partrelid, AccessExclusiveLock);
 
        /* If the relations aren't there, something bad happened; bail out */
        if (rel == NULL)
        {
            if (partRel != NULL)    /* shouldn't happen */
                elog(WARNING, "dangling partition \"%s\" remains, can't fix",
                     partrelname);
            ereport(ERROR,
                    (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                     errmsg("partitioned table \"%s\" was removed concurrently",
                            parentrelname)));
        }
        if (partRel == NULL)
            ereport(ERROR,
                    (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                     errmsg("partition \"%s\" was removed concurrently", partrelname)));
 
        tab->rel = rel;
    }
 
    /*
     * Detaching the partition might involve TOAST table access, so ensure we
     * have a valid snapshot.
     */
    PushActiveSnapshot(GetTransactionSnapshot());
 
    /* Do the final part of detaching */
    DetachPartitionFinalize(rel, partRel, concurrent, defaultPartOid);
 
    PopActiveSnapshot();
 
    ObjectAddressSet(address, RelationRelationId, RelationGetRelid(partRel));
 
    /* keep our lock until commit */
    table_close(partRel, NoLock);
 
    return address;
}

References AccessExclusiveLock, ATDetachCheckNoForeignKeyRefs(), CacheInvalidateRelcache(), CommitTransactionCommand(), DetachPartitionFinalize(), elog, ereport, errcode(), errmsg, ERROR, fb(), get_default_oid_from_partdesc(), GetTransactionSnapshot(), list_make1, LockRelationOid(), MarkInheritDetached(), MemoryContextStrdup(), MyDatabaseId, name, NoLock, ObjectAddressSet, OidIsValid, PopActiveSnapshot(), PortalContext, PushActiveSnapshot(), AlteredTableInfo::rel, RelationGetPartitionDesc(), RelationGetRelationName, RelationGetRelid, RemoveInheritance(), SET_LOCKTAG_RELATION, ShareUpdateExclusiveLock, StartTransactionCommand(), table_close(), table_openrv(), try_relation_open(), WaitForLockersMultiple(), and WARNING.

Referenced by ATExecCmd().

◆ ATExecDetachPartitionFinalize()

static ObjectAddress ATExecDetachPartitionFinalize	(	Relation	rel,
		RangeVar *	name
	)

static

Definition at line 22206 of file tablecmds.c.

{
    Relation    partRel;
    ObjectAddress address;
    Snapshot    snap = GetActiveSnapshot();
 
    partRel = table_openrv(name, AccessExclusiveLock);
 
    /*
     * Wait until existing snapshots are gone.  This is important if the
     * second transaction of DETACH PARTITION CONCURRENTLY is canceled: the
     * user could immediately run DETACH FINALIZE without actually waiting for
     * existing transactions.  We must not complete the detach action until
     * all such queries are complete (otherwise we would present them with an
     * inconsistent view of catalogs).
     */
    WaitForOlderSnapshots(snap->xmin, false);
 
    DetachPartitionFinalize(rel, partRel, true, InvalidOid);
 
    ObjectAddressSet(address, RelationRelationId, RelationGetRelid(partRel));
 
    table_close(partRel, NoLock);
 
    return address;
}

References AccessExclusiveLock, DetachPartitionFinalize(), fb(), GetActiveSnapshot(), InvalidOid, name, NoLock, ObjectAddressSet, RelationGetRelid, table_close(), table_openrv(), and WaitForOlderSnapshots().

Referenced by ATExecCmd().

◆ ATExecDropCluster()

static void ATExecDropCluster	(	Relation	rel,
		LOCKMODE	lockmode
	)

static

Definition at line 17185 of file tablecmds.c.

{
    mark_index_clustered(rel, InvalidOid, false);
}

References InvalidOid, and mark_index_clustered().

Referenced by ATExecCmd().

◆ ATExecDropColumn()

static ObjectAddress ATExecDropColumn	(	List **	wqueue,
		Relation	rel,
		const char *	colName,
		DropBehavior	behavior,
		bool	recurse,
		bool	recursing,
		bool	missing_ok,
		LOCKMODE	lockmode,
		ObjectAddresses *	addrs
	)

static

Definition at line 9419 of file tablecmds.c.

{
    HeapTuple   tuple;
    Form_pg_attribute targetatt;
    AttrNumber  attnum;
    List       *children;
    ObjectAddress object;
    bool        is_expr;
 
    /* At top level, permission check was done in ATPrepCmd, else do it */
    if (recursing)
        ATSimplePermissions(AT_DropColumn, rel,
                            ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
 
    /* Initialize addrs on the first invocation */
    Assert(!recursing || addrs != NULL);
 
    /* since this function recurses, it could be driven to stack overflow */
    check_stack_depth();
 
    if (!recursing)
        addrs = new_object_addresses();
 
    /*
     * get the number of the attribute
     */
    tuple = SearchSysCacheAttName(RelationGetRelid(rel), colName);
    if (!HeapTupleIsValid(tuple))
    {
        if (!missing_ok)
        {
            ereport(ERROR,
                    (errcode(ERRCODE_UNDEFINED_COLUMN),
                     errmsg("column \"%s\" of relation \"%s\" does not exist",
                            colName, RelationGetRelationName(rel))));
        }
        else
        {
            ereport(NOTICE,
                    (errmsg("column \"%s\" of relation \"%s\" does not exist, skipping",
                            colName, RelationGetRelationName(rel))));
            return InvalidObjectAddress;
        }
    }
    targetatt = (Form_pg_attribute) GETSTRUCT(tuple);
 
    attnum = targetatt->attnum;
 
    /* Can't drop a system attribute */
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot drop system column \"%s\"",
                        colName)));
 
    /*
     * Don't drop inherited columns, unless recursing (presumably from a drop
     * of the parent column)
     */
    if (targetatt->attinhcount > 0 && !recursing)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("cannot drop inherited column \"%s\"",
                        colName)));
 
    /*
     * Don't drop columns used in the partition key, either.  (If we let this
     * go through, the key column's dependencies would cause a cascaded drop
     * of the whole table, which is surely not what the user expected.)
     */
    if (has_partition_attrs(rel,
                            bms_make_singleton(attnum - FirstLowInvalidHeapAttributeNumber),
                            &is_expr))
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("cannot drop column \"%s\" because it is part of the partition key of relation \"%s\"",
                        colName, RelationGetRelationName(rel))));
 
    ReleaseSysCache(tuple);
 
    /*
     * Propagate to children as appropriate.  Unlike most other ALTER
     * routines, we have to do this one level of recursion at a time; we can't
     * use find_all_inheritors to do it in one pass.
     */
    children =
        find_inheritance_children(RelationGetRelid(rel), lockmode);
 
    if (children)
    {
        Relation    attr_rel;
        ListCell   *child;
 
        /*
         * In case of a partitioned table, the column must be dropped from the
         * partitions as well.
         */
        if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE && !recurse)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                     errmsg("cannot drop column from only the partitioned table when partitions exist"),
                     errhint("Do not specify the ONLY keyword.")));
 
        attr_rel = table_open(AttributeRelationId, RowExclusiveLock);
        foreach(child, children)
        {
            Oid         childrelid = lfirst_oid(child);
            Relation    childrel;
            Form_pg_attribute childatt;
 
            /* find_inheritance_children already got lock */
            childrel = table_open(childrelid, NoLock);
            CheckAlterTableIsSafe(childrel);
 
            tuple = SearchSysCacheCopyAttName(childrelid, colName);
            if (!HeapTupleIsValid(tuple))   /* shouldn't happen */
                elog(ERROR, "cache lookup failed for attribute \"%s\" of relation %u",
                     colName, childrelid);
            childatt = (Form_pg_attribute) GETSTRUCT(tuple);
 
            if (childatt->attinhcount <= 0) /* shouldn't happen */
                elog(ERROR, "relation %u has non-inherited attribute \"%s\"",
                     childrelid, colName);
 
            if (recurse)
            {
                /*
                 * If the child column has other definition sources, just
                 * decrement its inheritance count; if not, recurse to delete
                 * it.
                 */
                if (childatt->attinhcount == 1 && !childatt->attislocal)
                {
                    /* Time to delete this child column, too */
                    ATExecDropColumn(wqueue, childrel, colName,
                                     behavior, true, true,
                                     false, lockmode, addrs);
                }
                else
                {
                    /* Child column must survive my deletion */
                    childatt->attinhcount--;
 
                    CatalogTupleUpdate(attr_rel, &tuple->t_self, tuple);
 
                    /* Make update visible */
                    CommandCounterIncrement();
                }
            }
            else
            {
                /*
                 * If we were told to drop ONLY in this table (no recursion),
                 * we need to mark the inheritors' attributes as locally
                 * defined rather than inherited.
                 */
                childatt->attinhcount--;
                childatt->attislocal = true;
 
                CatalogTupleUpdate(attr_rel, &tuple->t_self, tuple);
 
                /* Make update visible */
                CommandCounterIncrement();
            }
 
            heap_freetuple(tuple);
 
            table_close(childrel, NoLock);
        }
        table_close(attr_rel, RowExclusiveLock);
    }
 
    /* Add object to delete */
    object.classId = RelationRelationId;
    object.objectId = RelationGetRelid(rel);
    object.objectSubId = attnum;
    add_exact_object_address(&object, addrs);
 
    if (!recursing)
    {
        /* Recursion has ended, drop everything that was collected */
        performMultipleDeletions(addrs, behavior, 0);
        free_object_addresses(addrs);
    }
 
    return object;
}

References add_exact_object_address(), Assert, AT_DropColumn, ATExecDropColumn(), ATSimplePermissions(), ATT_FOREIGN_TABLE, ATT_PARTITIONED_TABLE, ATT_TABLE, attnum, bms_make_singleton(), CatalogTupleUpdate(), check_stack_depth(), CheckAlterTableIsSafe(), CommandCounterIncrement(), elog, ereport, errcode(), errhint(), errmsg, ERROR, fb(), find_inheritance_children(), FirstLowInvalidHeapAttributeNumber, free_object_addresses(), GETSTRUCT(), has_partition_attrs(), heap_freetuple(), HeapTupleIsValid, InvalidObjectAddress, lfirst_oid, new_object_addresses(), NoLock, NOTICE, performMultipleDeletions(), RelationData::rd_rel, RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), RowExclusiveLock, SearchSysCacheAttName(), SearchSysCacheCopyAttName(), HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecCmd(), and ATExecDropColumn().

◆ ATExecDropConstraint()

static void ATExecDropConstraint	(	Relation	rel,
		const char *	constrName,
		DropBehavior	behavior,
		bool	recurse,
		bool	missing_ok,
		LOCKMODE	lockmode
	)

static

Definition at line 14549 of file tablecmds.c.

{
    Relation    conrel;
    SysScanDesc scan;
    ScanKeyData skey[3];
    HeapTuple   tuple;
    bool        found = false;
 
    conrel = table_open(ConstraintRelationId, RowExclusiveLock);
 
    /*
     * Find and drop the target constraint
     */
    ScanKeyInit(&skey[0],
                Anum_pg_constraint_conrelid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationGetRelid(rel)));
    ScanKeyInit(&skey[1],
                Anum_pg_constraint_contypid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(InvalidOid));
    ScanKeyInit(&skey[2],
                Anum_pg_constraint_conname,
                BTEqualStrategyNumber, F_NAMEEQ,
                CStringGetDatum(constrName));
    scan = systable_beginscan(conrel, ConstraintRelidTypidNameIndexId,
                              true, NULL, 3, skey);
 
    /* There can be at most one matching row */
    if (HeapTupleIsValid(tuple = systable_getnext(scan)))
    {
        dropconstraint_internal(rel, tuple, behavior, recurse, false,
                                missing_ok, lockmode);
        found = true;
    }
 
    systable_endscan(scan);
 
    if (!found)
    {
        if (!missing_ok)
            ereport(ERROR,
                    errcode(ERRCODE_UNDEFINED_OBJECT),
                    errmsg("constraint \"%s\" of relation \"%s\" does not exist",
                           constrName, RelationGetRelationName(rel)));
        else
            ereport(NOTICE,
                    errmsg("constraint \"%s\" of relation \"%s\" does not exist, skipping",
                           constrName, RelationGetRelationName(rel)));
    }
 
    table_close(conrel, RowExclusiveLock);
}

References BTEqualStrategyNumber, CStringGetDatum(), dropconstraint_internal(), ereport, errcode(), errmsg, ERROR, fb(), HeapTupleIsValid, InvalidOid, NOTICE, ObjectIdGetDatum(), RelationGetRelationName, RelationGetRelid, RowExclusiveLock, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by ATExecCmd().

◆ ATExecDropExpression()

static ObjectAddress ATExecDropExpression	(	Relation	rel,
		const char *	colName,
		bool	missing_ok,
		LOCKMODE	lockmode
	)

static

Definition at line 8937 of file tablecmds.c.

{
    HeapTuple   tuple;
    Form_pg_attribute attTup;
    AttrNumber  attnum;
    Relation    attrelation;
    Oid         attrdefoid;
    ObjectAddress address;
 
    attrelation = table_open(AttributeRelationId, RowExclusiveLock);
    tuple = SearchSysCacheCopyAttName(RelationGetRelid(rel), colName);
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel))));
 
    attTup = (Form_pg_attribute) GETSTRUCT(tuple);
    attnum = attTup->attnum;
 
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"",
                        colName)));
 
    /*
     * TODO: This could be done, but it would need a table rewrite to
     * materialize the generated values.  Note that for the time being, we
     * still error with missing_ok, so that we don't silently leave the column
     * as generated.
     */
    if (attTup->attgenerated == ATTRIBUTE_GENERATED_VIRTUAL)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("ALTER TABLE / DROP EXPRESSION is not supported for virtual generated columns"),
                 errdetail("Column \"%s\" of relation \"%s\" is a virtual generated column.",
                           colName, RelationGetRelationName(rel))));
 
    if (!attTup->attgenerated)
    {
        if (!missing_ok)
            ereport(ERROR,
                    (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                     errmsg("column \"%s\" of relation \"%s\" is not a generated column",
                            colName, RelationGetRelationName(rel))));
        else
        {
            ereport(NOTICE,
                    (errmsg("column \"%s\" of relation \"%s\" is not a generated column, skipping",
                            colName, RelationGetRelationName(rel))));
            heap_freetuple(tuple);
            table_close(attrelation, RowExclusiveLock);
            return InvalidObjectAddress;
        }
    }
 
    /*
     * Mark the column as no longer generated.  (The atthasdef flag needs to
     * get cleared too, but RemoveAttrDefault will handle that.)
     */
    attTup->attgenerated = '\0';
    CatalogTupleUpdate(attrelation, &tuple->t_self, tuple);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel),
                              attnum);
    heap_freetuple(tuple);
 
    table_close(attrelation, RowExclusiveLock);
 
    /*
     * Drop the dependency records of the GENERATED expression, in particular
     * its INTERNAL dependency on the column, which would otherwise cause
     * dependency.c to refuse to perform the deletion.
     */
    attrdefoid = GetAttrDefaultOid(RelationGetRelid(rel), attnum);
    if (!OidIsValid(attrdefoid))
        elog(ERROR, "could not find attrdef tuple for relation %u attnum %d",
             RelationGetRelid(rel), attnum);
    (void) deleteDependencyRecordsFor(AttrDefaultRelationId, attrdefoid, false);
 
    /* Make above changes visible */
    CommandCounterIncrement();
 
    /*
     * Get rid of the GENERATED expression itself.  We use RESTRICT here for
     * safety, but at present we do not expect anything to depend on the
     * default.
     */
    RemoveAttrDefault(RelationGetRelid(rel), attnum, DROP_RESTRICT,
                      false, false);
 
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
    return address;
}

References attnum, CatalogTupleUpdate(), CommandCounterIncrement(), deleteDependencyRecordsFor(), DROP_RESTRICT, elog, ereport, errcode(), errdetail(), errmsg, ERROR, fb(), GetAttrDefaultOid(), GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InvalidObjectAddress, InvokeObjectPostAlterHook, NOTICE, ObjectAddressSubSet, OidIsValid, RelationGetRelationName, RelationGetRelid, RemoveAttrDefault(), RowExclusiveLock, SearchSysCacheCopyAttName(), HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecCmd().

◆ ATExecDropIdentity()

static ObjectAddress ATExecDropIdentity	(	Relation	rel,
		const char *	colName,
		bool	missing_ok,
		LOCKMODE	lockmode,
		bool	recurse,
		bool	recursing
	)

static

Definition at line 8629 of file tablecmds.c.

{
    HeapTuple   tuple;
    Form_pg_attribute attTup;
    AttrNumber  attnum;
    Relation    attrelation;
    ObjectAddress address;
    Oid         seqid;
    ObjectAddress seqaddress;
    bool        ispartitioned;
 
    ispartitioned = (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE);
    if (ispartitioned && !recurse)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("cannot drop identity from a column of only the partitioned table"),
                 errhint("Do not specify the ONLY keyword.")));
 
    if (rel->rd_rel->relispartition && !recursing)
        ereport(ERROR,
                errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                errmsg("cannot drop identity from a column of a partition"));
 
    attrelation = table_open(AttributeRelationId, RowExclusiveLock);
    tuple = SearchSysCacheCopyAttName(RelationGetRelid(rel), colName);
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel))));
 
    attTup = (Form_pg_attribute) GETSTRUCT(tuple);
    attnum = attTup->attnum;
 
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"",
                        colName)));
 
    if (!attTup->attidentity)
    {
        if (!missing_ok)
            ereport(ERROR,
                    (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                     errmsg("column \"%s\" of relation \"%s\" is not an identity column",
                            colName, RelationGetRelationName(rel))));
        else
        {
            ereport(NOTICE,
                    (errmsg("column \"%s\" of relation \"%s\" is not an identity column, skipping",
                            colName, RelationGetRelationName(rel))));
            heap_freetuple(tuple);
            table_close(attrelation, RowExclusiveLock);
            return InvalidObjectAddress;
        }
    }
 
    attTup->attidentity = '\0';
    CatalogTupleUpdate(attrelation, &tuple->t_self, tuple);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel),
                              attTup->attnum);
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
    heap_freetuple(tuple);
 
    table_close(attrelation, RowExclusiveLock);
 
    /*
     * Recurse to drop the identity from column in partitions.  Identity is
     * not inherited in regular inheritance children so ignore them.
     */
    if (recurse && ispartitioned)
    {
        List       *children;
        ListCell   *lc;
 
        children = find_inheritance_children(RelationGetRelid(rel), lockmode);
 
        foreach(lc, children)
        {
            Relation    childrel;
 
            childrel = table_open(lfirst_oid(lc), NoLock);
            ATExecDropIdentity(childrel, colName, false, lockmode, recurse, true);
            table_close(childrel, NoLock);
        }
    }
 
    if (!recursing)
    {
        /* drop the internal sequence */
        seqid = getIdentitySequence(rel, attnum, false);
        deleteDependencyRecordsForClass(RelationRelationId, seqid,
                                        RelationRelationId, DEPENDENCY_INTERNAL);
        CommandCounterIncrement();
        seqaddress.classId = RelationRelationId;
        seqaddress.objectId = seqid;
        seqaddress.objectSubId = 0;
        performDeletion(&seqaddress, DROP_RESTRICT, PERFORM_DELETION_INTERNAL);
    }
 
    return address;
}

References ATExecDropIdentity(), attnum, CatalogTupleUpdate(), CommandCounterIncrement(), deleteDependencyRecordsForClass(), DEPENDENCY_INTERNAL, DROP_RESTRICT, ereport, errcode(), errhint(), errmsg, ERROR, fb(), find_inheritance_children(), getIdentitySequence(), GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InvalidObjectAddress, InvokeObjectPostAlterHook, lfirst_oid, NoLock, NOTICE, ObjectAddressSubSet, PERFORM_DELETION_INTERNAL, performDeletion(), RelationData::rd_rel, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, SearchSysCacheCopyAttName(), HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecCmd(), ATExecDropIdentity(), and DetachPartitionFinalize().

◆ ATExecDropInherit()

static ObjectAddress ATExecDropInherit	(	Relation	rel,
		RangeVar *	parent,
		LOCKMODE	lockmode
	)

static

Definition at line 18533 of file tablecmds.c.

{
    ObjectAddress address;
    Relation    parent_rel;
 
    /*
     * AccessShareLock on the parent is probably enough, seeing that DROP
     * TABLE doesn't lock parent tables at all.  We need some lock since we'll
     * be inspecting the parent's schema.
     */
    parent_rel = table_openrv(parent, AccessShareLock);
 
    /*
     * We don't bother to check ownership of the parent table --- ownership of
     * the child is presumed enough rights.
     */
 
    /* Off to RemoveInheritance() where most of the work happens */
    RemoveInheritance(rel, parent_rel, false);
 
    ObjectAddressSet(address, RelationRelationId,
                     RelationGetRelid(parent_rel));
 
    /* keep our lock on the parent relation until commit */
    table_close(parent_rel, NoLock);
 
    return address;
}

References AccessShareLock, fb(), NoLock, ObjectAddressSet, RelationGetRelid, RemoveInheritance(), table_close(), and table_openrv().

Referenced by ATExecCmd().

◆ ATExecDropNotNull()

static ObjectAddress ATExecDropNotNull	(	Relation	rel,
		const char *	colName,
		bool	recurse,
		LOCKMODE	lockmode
	)

static

Definition at line 7883 of file tablecmds.c.

{
    HeapTuple   tuple;
    HeapTuple   conTup;
    Form_pg_attribute attTup;
    AttrNumber  attnum;
    Relation    attr_rel;
    ObjectAddress address;
 
    /*
     * lookup the attribute
     */
    attr_rel = table_open(AttributeRelationId, RowExclusiveLock);
 
    tuple = SearchSysCacheCopyAttName(RelationGetRelid(rel), colName);
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel))));
    attTup = (Form_pg_attribute) GETSTRUCT(tuple);
    attnum = attTup->attnum;
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
 
    /* If the column is already nullable there's nothing to do. */
    if (!attTup->attnotnull)
    {
        table_close(attr_rel, RowExclusiveLock);
        return InvalidObjectAddress;
    }
 
    /* Prevent them from altering a system attribute */
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"",
                        colName)));
 
    if (attTup->attidentity)
        ereport(ERROR,
                (errcode(ERRCODE_SYNTAX_ERROR),
                 errmsg("column \"%s\" of relation \"%s\" is an identity column",
                        colName, RelationGetRelationName(rel))));
 
    /*
     * If rel is partition, shouldn't drop NOT NULL if parent has the same.
     */
    if (rel->rd_rel->relispartition)
    {
        Oid         parentId = get_partition_parent(RelationGetRelid(rel), false);
        Relation    parent = table_open(parentId, AccessShareLock);
        TupleDesc   tupDesc = RelationGetDescr(parent);
        AttrNumber  parent_attnum;
 
        parent_attnum = get_attnum(parentId, colName);
        if (TupleDescAttr(tupDesc, parent_attnum - 1)->attnotnull)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                     errmsg("column \"%s\" is marked NOT NULL in parent table",
                            colName)));
        table_close(parent, AccessShareLock);
    }
 
    /*
     * Find the constraint that makes this column NOT NULL, and drop it.
     * dropconstraint_internal() resets attnotnull.
     */
    conTup = findNotNullConstraintAttnum(RelationGetRelid(rel), attnum);
    if (conTup == NULL)
        elog(ERROR, "cache lookup failed for not-null constraint on column \"%s\" of relation \"%s\"",
             colName, RelationGetRelationName(rel));
 
    /* The normal case: we have a pg_constraint row, remove it */
    dropconstraint_internal(rel, conTup, DROP_RESTRICT, recurse, false,
                            false, lockmode);
    heap_freetuple(conTup);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel), attnum);
 
    table_close(attr_rel, RowExclusiveLock);
 
    return address;
}

References AccessShareLock, attnotnull, attnum, DROP_RESTRICT, dropconstraint_internal(), elog, ereport, errcode(), errmsg, ERROR, fb(), findNotNullConstraintAttnum(), get_attnum(), get_partition_parent(), GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InvalidObjectAddress, InvokeObjectPostAlterHook, ObjectAddressSubSet, RelationData::rd_rel, RelationGetDescr, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, SearchSysCacheCopyAttName(), table_close(), table_open(), and TupleDescAttr().

Referenced by ATExecCmd().

◆ ATExecDropOf()

static void ATExecDropOf	(	Relation	rel,
		LOCKMODE	lockmode
	)

static

Definition at line 19061 of file tablecmds.c.

{
    Oid         relid = RelationGetRelid(rel);
    Relation    relationRelation;
    HeapTuple   tuple;
 
    if (!OidIsValid(rel->rd_rel->reloftype))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("\"%s\" is not a typed table",
                        RelationGetRelationName(rel))));
 
    /*
     * We don't bother to check ownership of the type --- ownership of the
     * table is presumed enough rights.  No lock required on the type, either.
     */
 
    drop_parent_dependency(relid, TypeRelationId, rel->rd_rel->reloftype,
                           DEPENDENCY_NORMAL);
 
    /* Clear pg_class.reloftype */
    relationRelation = table_open(RelationRelationId, RowExclusiveLock);
    tuple = SearchSysCacheCopy1(RELOID, ObjectIdGetDatum(relid));
    if (!HeapTupleIsValid(tuple))
        elog(ERROR, "cache lookup failed for relation %u", relid);
    ((Form_pg_class) GETSTRUCT(tuple))->reloftype = InvalidOid;
    CatalogTupleUpdate(relationRelation, &tuple->t_self, tuple);
 
    InvokeObjectPostAlterHook(RelationRelationId, relid, 0);
 
    heap_freetuple(tuple);
    table_close(relationRelation, RowExclusiveLock);
}

References CatalogTupleUpdate(), DEPENDENCY_NORMAL, drop_parent_dependency(), elog, ereport, errcode(), errmsg, ERROR, fb(), GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InvalidOid, InvokeObjectPostAlterHook, ObjectIdGetDatum(), OidIsValid, RelationData::rd_rel, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, SearchSysCacheCopy1, HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecCmd().

◆ ATExecEnableDisableRule()

static void ATExecEnableDisableRule	(	Relation	rel,
		const char *	rulename,
		char	fires_when,
		LOCKMODE	lockmode
	)

static

Definition at line 17934 of file tablecmds.c.

{
    EnableDisableRule(rel, rulename, fires_when);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel), 0);
}

References EnableDisableRule(), fb(), InvokeObjectPostAlterHook, and RelationGetRelid.

Referenced by ATExecCmd().

◆ ATExecEnableDisableTrigger()

static void ATExecEnableDisableTrigger	(	Relation	rel,
		const char *	trigname,
		char	fires_when,
		bool	skip_system,
		bool	recurse,
		LOCKMODE	lockmode
	)

static

Definition at line 17916 of file tablecmds.c.

{
    EnableDisableTrigger(rel, trigname, InvalidOid,
                         fires_when, skip_system, recurse,
                         lockmode);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel), 0);
}

References EnableDisableTrigger(), fb(), InvalidOid, InvokeObjectPostAlterHook, and RelationGetRelid.

Referenced by ATExecCmd().

◆ ATExecForceNoForceRowSecurity()

static void ATExecForceNoForceRowSecurity	(	Relation	rel,
		bool	force_rls
	)

static

Definition at line 19337 of file tablecmds.c.

{
    Relation    pg_class;
    Oid         relid;
    HeapTuple   tuple;
 
    relid = RelationGetRelid(rel);
 
    pg_class = table_open(RelationRelationId, RowExclusiveLock);
 
    tuple = SearchSysCacheCopy1(RELOID, ObjectIdGetDatum(relid));
 
    if (!HeapTupleIsValid(tuple))
        elog(ERROR, "cache lookup failed for relation %u", relid);
 
    ((Form_pg_class) GETSTRUCT(tuple))->relforcerowsecurity = force_rls;
    CatalogTupleUpdate(pg_class, &tuple->t_self, tuple);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel), 0);
 
    table_close(pg_class, RowExclusiveLock);
    heap_freetuple(tuple);
}

References CatalogTupleUpdate(), elog, ERROR, fb(), GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InvokeObjectPostAlterHook, ObjectIdGetDatum(), RelationGetRelid, RowExclusiveLock, SearchSysCacheCopy1, HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecCmd().

◆ ATExecGenericOptions()

static void ATExecGenericOptions	(	Relation	rel,
		List *	options
	)

static

Definition at line 19366 of file tablecmds.c.

{
    Relation    ftrel;
    ForeignServer *server;
    ForeignDataWrapper *fdw;
    HeapTuple   tuple;
    bool        isnull;
    Datum       repl_val[Natts_pg_foreign_table];
    bool        repl_null[Natts_pg_foreign_table];
    bool        repl_repl[Natts_pg_foreign_table];
    Datum       datum;
    Form_pg_foreign_table tableform;
 
    if (options == NIL)
        return;
 
    ftrel = table_open(ForeignTableRelationId, RowExclusiveLock);
 
    tuple = SearchSysCacheCopy1(FOREIGNTABLEREL,
                                ObjectIdGetDatum(rel->rd_id));
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_OBJECT),
                 errmsg("foreign table \"%s\" does not exist",
                        RelationGetRelationName(rel))));
    tableform = (Form_pg_foreign_table) GETSTRUCT(tuple);
    server = GetForeignServer(tableform->ftserver);
    fdw = GetForeignDataWrapper(server->fdwid);
 
    memset(repl_val, 0, sizeof(repl_val));
    memset(repl_null, false, sizeof(repl_null));
    memset(repl_repl, false, sizeof(repl_repl));
 
    /* Extract the current options */
    datum = SysCacheGetAttr(FOREIGNTABLEREL,
                            tuple,
                            Anum_pg_foreign_table_ftoptions,
                            &isnull);
    if (isnull)
        datum = PointerGetDatum(NULL);
 
    /* Transform the options */
    datum = transformGenericOptions(ForeignTableRelationId,
                                    datum,
                                    options,
                                    fdw->fdwvalidator);
 
    if (DatumGetPointer(datum) != NULL)
        repl_val[Anum_pg_foreign_table_ftoptions - 1] = datum;
    else
        repl_null[Anum_pg_foreign_table_ftoptions - 1] = true;
 
    repl_repl[Anum_pg_foreign_table_ftoptions - 1] = true;
 
    /* Everything looks good - update the tuple */
 
    tuple = heap_modify_tuple(tuple, RelationGetDescr(ftrel),
                              repl_val, repl_null, repl_repl);
 
    CatalogTupleUpdate(ftrel, &tuple->t_self, tuple);
 
    /*
     * Invalidate relcache so that all sessions will refresh any cached plans
     * that might depend on the old options.
     */
    CacheInvalidateRelcache(rel);
 
    InvokeObjectPostAlterHook(ForeignTableRelationId,
                              RelationGetRelid(rel), 0);
 
    table_close(ftrel, RowExclusiveLock);
 
    heap_freetuple(tuple);
}

References CacheInvalidateRelcache(), CatalogTupleUpdate(), DatumGetPointer(), ereport, errcode(), errmsg, ERROR, fb(), ForeignServer::fdwid, Form_pg_foreign_table, GetForeignDataWrapper(), GetForeignServer(), GETSTRUCT(), heap_freetuple(), heap_modify_tuple(), HeapTupleIsValid, InvokeObjectPostAlterHook, NIL, ObjectIdGetDatum(), PointerGetDatum, RelationData::rd_id, RelationGetDescr, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, SearchSysCacheCopy1, SysCacheGetAttr(), HeapTupleData::t_self, table_close(), table_open(), and transformGenericOptions().

Referenced by ATExecCmd().

◆ ATExecMergePartitions()

static void ATExecMergePartitions	(	List **	wqueue,
		AlteredTableInfo *	tab,
		Relation	rel,
		PartitionCmd *	cmd,
		AlterTableUtilityContext *	context
	)

static

Definition at line 23756 of file tablecmds.c.

{
    Relation    newPartRel;
    List       *mergingPartitions = NIL;
    List       *extDepState = NIL;
    Oid         defaultPartOid;
    Oid         existingRelid;
    Oid         ownerId = InvalidOid;
    Oid         save_userid;
    int         save_sec_context;
    int         save_nestlevel;
 
    /*
     * Check ownership of merged partitions - partitions with different owners
     * cannot be merged. Also, collect the OIDs of these partitions during the
     * check.
     */
    foreach_node(RangeVar, name, cmd->partlist)
    {
        Relation    mergingPartition;
 
        /*
         * We are going to detach and remove this partition.  We already took
         * AccessExclusiveLock lock on transformPartitionCmdForMerge, so here,
         * NoLock is fine.
         */
        mergingPartition = table_openrv_extended(name, NoLock, false);
        Assert(CheckRelationLockedByMe(mergingPartition, AccessExclusiveLock, false));
 
        if (OidIsValid(ownerId))
        {
            /* Do the partitions being merged have different owners? */
            if (ownerId != mergingPartition->rd_rel->relowner)
                ereport(ERROR,
                        errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                        errmsg("partitions being merged have different owners"));
        }
        else
            ownerId = mergingPartition->rd_rel->relowner;
 
        /* Store the next merging partition into the list. */
        mergingPartitions = lappend_oid(mergingPartitions,
                                        RelationGetRelid(mergingPartition));
 
        table_close(mergingPartition, NoLock);
    }
 
    /* Look up the existing relation by the new partition name. */
    RangeVarGetAndCheckCreationNamespace(cmd->name, NoLock, &existingRelid);
 
    /*
     * Check if this name is already taken.  This helps us to detect the
     * situation when one of the merging partitions has the same name as the
     * new partition.  Otherwise, this would fail later on anyway, but
     * catching this here allows us to emit a nicer error message.
     */
    if (OidIsValid(existingRelid))
    {
        if (list_member_oid(mergingPartitions, existingRelid))
        {
            /*
             * The new partition has the same name as one of the merging
             * partitions.
             */
            char        tmpRelName[NAMEDATALEN];
 
            /* Generate a temporary name. */
            sprintf(tmpRelName, "merge-%u-%X-tmp", RelationGetRelid(rel), MyProcPid);
 
            /*
             * Rename the existing partition with a temporary name, leaving it
             * free for the new partition.  We don't need to care about this
             * in the future because we're going to eventually drop the
             * existing partition anyway.
             */
            RenameRelationInternal(existingRelid, tmpRelName, true, false);
 
            /*
             * We must bump the command counter to make the new partition
             * tuple visible for rename.
             */
            CommandCounterIncrement();
        }
        else
        {
            ereport(ERROR,
                    errcode(ERRCODE_DUPLICATE_TABLE),
                    errmsg("relation \"%s\" already exists", cmd->name->relname));
        }
    }
 
    defaultPartOid =
        get_default_oid_from_partdesc(RelationGetPartitionDesc(rel, true));
 
    /*
     * Collect extension dependencies from indexes on the merging partitions.
     * We must do this before detaching them, so we can restore the
     * dependencies on the new partition's indexes later.
     */
    extDepState = collectPartitionIndexExtDeps(mergingPartitions);
 
    /* Detach all merging partitions. */
    foreach_oid(mergingPartitionOid, mergingPartitions)
    {
        Relation    child_rel;
 
        child_rel = table_open(mergingPartitionOid, NoLock);
 
        detachPartitionTable(rel, child_rel, defaultPartOid);
 
        table_close(child_rel, NoLock);
    }
 
    /*
     * Perform a preliminary check to determine whether it's safe to drop all
     * merging partitions before we actually do so later. After merging rows
     * into the new partitions via MergePartitionsMoveRows, all old partitions
     * need to be dropped. However, since the drop behavior is DROP_RESTRICT
     * and the merge process (MergePartitionsMoveRows) can be time-consuming,
     * performing an early check on the drop eligibility of old partitions is
     * preferable.
     */
    foreach_oid(mergingPartitionOid, mergingPartitions)
    {
        ObjectAddress object;
 
        /* Get oid of the later to be dropped relation. */
        object.objectId = mergingPartitionOid;
        object.classId = RelationRelationId;
        object.objectSubId = 0;
 
        performDeletionCheck(&object, DROP_RESTRICT, 0);
    }
 
    /*
     * Create a table for the new partition, using the partitioned table as a
     * model.
     */
    Assert(OidIsValid(ownerId));
    newPartRel = createPartitionTable(wqueue, cmd->name, rel, ownerId);
 
    /*
     * Switch to the table owner's userid, so that any index functions are run
     * as that user.  Also, lockdown security-restricted operations and
     * arrange to make GUC variable changes local to this command.
     *
     * Need to do it after determining the namespace in the
     * createPartitionTable() call.
     */
    GetUserIdAndSecContext(&save_userid, &save_sec_context);
    SetUserIdAndSecContext(ownerId,
                           save_sec_context | SECURITY_RESTRICTED_OPERATION);
    save_nestlevel = NewGUCNestLevel();
    RestrictSearchPath();
 
    /* Copy data from merged partitions to the new partition. */
    MergePartitionsMoveRows(wqueue, mergingPartitions, newPartRel);
 
    /* Drop the current partitions before attaching the new one. */
    foreach_oid(mergingPartitionOid, mergingPartitions)
    {
        ObjectAddress object;
 
        object.objectId = mergingPartitionOid;
        object.classId = RelationRelationId;
        object.objectSubId = 0;
 
        performDeletion(&object, DROP_RESTRICT, 0);
    }
 
    list_free(mergingPartitions);
 
    /*
     * Attach a new partition to the partitioned table. wqueue = NULL:
     * verification for each cloned constraint is not needed.
     */
    attachPartitionTable(NULL, rel, newPartRel, cmd->bound);
 
    /*
     * Apply extension dependencies to the new partition's indexes. This
     * preserves any "DEPENDS ON EXTENSION" settings from the merged
     * partitions.
     */
    applyPartitionIndexExtDeps(RelationGetRelid(newPartRel), extDepState);
 
    freePartitionIndexExtDeps(extDepState);
 
    /* Keep the lock until commit. */
    table_close(newPartRel, NoLock);
 
    /* Roll back any GUC changes executed by index functions. */
    AtEOXact_GUC(false, save_nestlevel);
 
    /* Restore the userid and security context. */
    SetUserIdAndSecContext(save_userid, save_sec_context);
}

References AccessExclusiveLock, applyPartitionIndexExtDeps(), Assert, AtEOXact_GUC(), attachPartitionTable(), PartitionCmd::bound, CheckRelationLockedByMe(), collectPartitionIndexExtDeps(), CommandCounterIncrement(), createPartitionTable(), detachPartitionTable(), DROP_RESTRICT, ereport, errcode(), errmsg, ERROR, fb(), foreach_node, foreach_oid, freePartitionIndexExtDeps(), get_default_oid_from_partdesc(), GetUserIdAndSecContext(), InvalidOid, lappend_oid(), list_free(), list_member_oid(), MergePartitionsMoveRows(), MyProcPid, name, PartitionCmd::name, NAMEDATALEN, NewGUCNestLevel(), NIL, NoLock, ObjectAddress::objectId, OidIsValid, PartitionCmd::partlist, performDeletion(), performDeletionCheck(), RangeVarGetAndCheckCreationNamespace(), RelationGetPartitionDesc(), RelationGetRelid, RangeVar::relname, RenameRelationInternal(), RestrictSearchPath(), SECURITY_RESTRICTED_OPERATION, SetUserIdAndSecContext(), sprintf, table_close(), table_open(), and table_openrv_extended().

Referenced by ATExecCmd().

◆ ATExecReplicaIdentity()

static void ATExecReplicaIdentity	(	Relation	rel,
		ReplicaIdentityStmt *	stmt,
		LOCKMODE	lockmode
	)

static

Definition at line 19193 of file tablecmds.c.

{
    Oid         indexOid;
    Relation    indexRel;
    int         key;
 
    if (stmt->identity_type == REPLICA_IDENTITY_DEFAULT)
    {
        relation_mark_replica_identity(rel, stmt->identity_type, InvalidOid, true);
        return;
    }
    else if (stmt->identity_type == REPLICA_IDENTITY_FULL)
    {
        relation_mark_replica_identity(rel, stmt->identity_type, InvalidOid, true);
        return;
    }
    else if (stmt->identity_type == REPLICA_IDENTITY_NOTHING)
    {
        relation_mark_replica_identity(rel, stmt->identity_type, InvalidOid, true);
        return;
    }
    else if (stmt->identity_type == REPLICA_IDENTITY_INDEX)
    {
         /* fallthrough */ ;
    }
    else
        elog(ERROR, "unexpected identity type %u", stmt->identity_type);
 
    /* Check that the index exists */
    indexOid = get_relname_relid(stmt->name, rel->rd_rel->relnamespace);
    if (!OidIsValid(indexOid))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_OBJECT),
                 errmsg("index \"%s\" for table \"%s\" does not exist",
                        stmt->name, RelationGetRelationName(rel))));
 
    indexRel = index_open(indexOid, ShareLock);
 
    /* Check that the index is on the relation we're altering. */
    if (indexRel->rd_index == NULL ||
        indexRel->rd_index->indrelid != RelationGetRelid(rel))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("\"%s\" is not an index for table \"%s\"",
                        RelationGetRelationName(indexRel),
                        RelationGetRelationName(rel))));
 
    /*
     * The AM must support uniqueness, and the index must in fact be unique.
     * If we have a WITHOUT OVERLAPS constraint (identified by uniqueness +
     * exclusion), we can use that too.
     */
    if ((!indexRel->rd_indam->amcanunique ||
         !indexRel->rd_index->indisunique) &&
        !(indexRel->rd_index->indisunique && indexRel->rd_index->indisexclusion))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot use non-unique index \"%s\" as replica identity",
                        RelationGetRelationName(indexRel))));
    /* Deferred indexes are not guaranteed to be always unique. */
    if (!indexRel->rd_index->indimmediate)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot use non-immediate index \"%s\" as replica identity",
                        RelationGetRelationName(indexRel))));
    /* Expression indexes aren't supported. */
    if (RelationGetIndexExpressions(indexRel) != NIL)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot use expression index \"%s\" as replica identity",
                        RelationGetRelationName(indexRel))));
    /* Predicate indexes aren't supported. */
    if (RelationGetIndexPredicate(indexRel) != NIL)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot use partial index \"%s\" as replica identity",
                        RelationGetRelationName(indexRel))));
 
    /* Check index for nullable columns. */
    for (key = 0; key < IndexRelationGetNumberOfKeyAttributes(indexRel); key++)
    {
        int16       attno = indexRel->rd_index->indkey.values[key];
        Form_pg_attribute attr;
 
        /*
         * Reject any other system columns.  (Going forward, we'll disallow
         * indexes containing such columns in the first place, but they might
         * exist in older branches.)
         */
        if (attno <= 0)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_COLUMN_REFERENCE),
                     errmsg("index \"%s\" cannot be used as replica identity because column %d is a system column",
                            RelationGetRelationName(indexRel), attno)));
 
        attr = TupleDescAttr(rel->rd_att, attno - 1);
        if (!attr->attnotnull)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("index \"%s\" cannot be used as replica identity because column \"%s\" is nullable",
                            RelationGetRelationName(indexRel),
                            NameStr(attr->attname))));
    }
 
    /* This index is suitable for use as a replica identity. Mark it. */
    relation_mark_replica_identity(rel, stmt->identity_type, indexOid, true);
 
    index_close(indexRel, NoLock);
}

References IndexAmRoutine::amcanunique, elog, ereport, errcode(), errmsg, ERROR, fb(), get_relname_relid(), index_close(), index_open(), IndexRelationGetNumberOfKeyAttributes, InvalidOid, NameStr, NIL, NoLock, OidIsValid, RelationData::rd_att, RelationData::rd_indam, RelationData::rd_index, RelationData::rd_rel, relation_mark_replica_identity(), RelationGetIndexExpressions(), RelationGetIndexPredicate(), RelationGetRelationName, RelationGetRelid, ShareLock, stmt, and TupleDescAttr().

Referenced by ATExecCmd().

◆ ATExecSetAccessMethodNoStorage()

static void ATExecSetAccessMethodNoStorage	(	Relation	rel,
		Oid	newAccessMethodId
	)

static

Definition at line 17231 of file tablecmds.c.

{
    Relation    pg_class;
    Oid         oldAccessMethodId;
    HeapTuple   tuple;
    Form_pg_class rd_rel;
    Oid         reloid = RelationGetRelid(rel);
 
    /*
     * Shouldn't be called on relations having storage; these are processed in
     * phase 3.
     */
    Assert(!RELKIND_HAS_STORAGE(rel->rd_rel->relkind));
 
    /* Get a modifiable copy of the relation's pg_class row. */
    pg_class = table_open(RelationRelationId, RowExclusiveLock);
 
    tuple = SearchSysCacheCopy1(RELOID, ObjectIdGetDatum(reloid));
    if (!HeapTupleIsValid(tuple))
        elog(ERROR, "cache lookup failed for relation %u", reloid);
    rd_rel = (Form_pg_class) GETSTRUCT(tuple);
 
    /* Update the pg_class row. */
    oldAccessMethodId = rd_rel->relam;
    rd_rel->relam = newAccessMethodId;
 
    /* Leave if no update required */
    if (rd_rel->relam == oldAccessMethodId)
    {
        heap_freetuple(tuple);
        table_close(pg_class, RowExclusiveLock);
        return;
    }
 
    CatalogTupleUpdate(pg_class, &tuple->t_self, tuple);
 
    /*
     * Update the dependency on the new access method.  No dependency is added
     * if the new access method is InvalidOid (default case).  Be very careful
     * that this has to compare the previous value stored in pg_class with the
     * new one.
     */
    if (!OidIsValid(oldAccessMethodId) && OidIsValid(rd_rel->relam))
    {
        ObjectAddress relobj,
                    referenced;
 
        /*
         * New access method is defined and there was no dependency
         * previously, so record a new one.
         */
        ObjectAddressSet(relobj, RelationRelationId, reloid);
        ObjectAddressSet(referenced, AccessMethodRelationId, rd_rel->relam);
        recordDependencyOn(&relobj, &referenced, DEPENDENCY_NORMAL);
    }
    else if (OidIsValid(oldAccessMethodId) &&
             !OidIsValid(rd_rel->relam))
    {
        /*
         * There was an access method defined, and no new one, so just remove
         * the existing dependency.
         */
        deleteDependencyRecordsForClass(RelationRelationId, reloid,
                                        AccessMethodRelationId,
                                        DEPENDENCY_NORMAL);
    }
    else
    {
        Assert(OidIsValid(oldAccessMethodId) &&
               OidIsValid(rd_rel->relam));
 
        /* Both are valid, so update the dependency */
        changeDependencyFor(RelationRelationId, reloid,
                            AccessMethodRelationId,
                            oldAccessMethodId, rd_rel->relam);
    }
 
    /* make the relam and dependency changes visible */
    CommandCounterIncrement();
 
    InvokeObjectPostAlterHook(RelationRelationId, RelationGetRelid(rel), 0);
 
    heap_freetuple(tuple);
    table_close(pg_class, RowExclusiveLock);
}

References Assert, CatalogTupleUpdate(), changeDependencyFor(), CommandCounterIncrement(), deleteDependencyRecordsForClass(), DEPENDENCY_NORMAL, elog, ERROR, fb(), GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InvokeObjectPostAlterHook, ObjectAddressSet, ObjectIdGetDatum(), OidIsValid, RelationData::rd_rel, recordDependencyOn(), RelationGetRelid, RowExclusiveLock, SearchSysCacheCopy1, HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecCmd().

◆ ATExecSetCompression()

static ObjectAddress ATExecSetCompression	(	Relation	rel,
		const char *	column,
		Node *	newValue,
		LOCKMODE	lockmode
	)

static

Definition at line 19447 of file tablecmds.c.

{
    Relation    attrel;
    HeapTuple   tuple;
    Form_pg_attribute atttableform;
    AttrNumber  attnum;
    char       *compression;
    char        cmethod;
    ObjectAddress address;
 
    compression = strVal(newValue);
 
    attrel = table_open(AttributeRelationId, RowExclusiveLock);
 
    /* copy the cache entry so we can scribble on it below */
    tuple = SearchSysCacheCopyAttName(RelationGetRelid(rel), column);
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        column, RelationGetRelationName(rel))));
 
    /* prevent them from altering a system attribute */
    atttableform = (Form_pg_attribute) GETSTRUCT(tuple);
    attnum = atttableform->attnum;
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"", column)));
 
    /*
     * Check that column type is compressible, then get the attribute
     * compression method code
     */
    cmethod = GetAttributeCompression(atttableform->atttypid, compression);
 
    /* update pg_attribute entry */
    atttableform->attcompression = cmethod;
    CatalogTupleUpdate(attrel, &tuple->t_self, tuple);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel),
                              attnum);
 
    /*
     * Apply the change to indexes as well (only for simple index columns,
     * matching behavior of index.c ConstructTupleDescriptor()).
     */
    SetIndexStorageProperties(rel, attrel, attnum,
                              false, 0,
                              true, cmethod,
                              lockmode);
 
    heap_freetuple(tuple);
 
    table_close(attrel, RowExclusiveLock);
 
    /* make changes visible */
    CommandCounterIncrement();
 
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
    return address;
}

References attnum, CatalogTupleUpdate(), CommandCounterIncrement(), ereport, errcode(), errmsg, ERROR, fb(), GetAttributeCompression(), GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InvokeObjectPostAlterHook, ObjectAddressSubSet, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, SearchSysCacheCopyAttName(), SetIndexStorageProperties(), strVal, HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecCmd().

◆ ATExecSetExpression()

static ObjectAddress ATExecSetExpression	(	AlteredTableInfo *	tab,
		Relation	rel,
		const char *	colName,
		Node *	newExpr,
		LOCKMODE	lockmode
	)

static

Definition at line 8743 of file tablecmds.c.

{
    HeapTuple   tuple;
    Form_pg_attribute attTup;
    AttrNumber  attnum;
    char        attgenerated;
    bool        rewrite;
    Oid         attrdefoid;
    ObjectAddress address;
    Expr       *defval;
    NewColumnValue *newval;
    RawColumnDefault *rawEnt;
 
    tuple = SearchSysCacheAttName(RelationGetRelid(rel), colName);
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel))));
 
    attTup = (Form_pg_attribute) GETSTRUCT(tuple);
 
    attnum = attTup->attnum;
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"",
                        colName)));
 
    attgenerated = attTup->attgenerated;
    if (!attgenerated)
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                 errmsg("column \"%s\" of relation \"%s\" is not a generated column",
                        colName, RelationGetRelationName(rel))));
 
    if (attgenerated == ATTRIBUTE_GENERATED_VIRTUAL && attTup->attnotnull)
        tab->verify_new_notnull = true;
 
    /*
     * We need to prevent this because a change of expression could affect a
     * row filter and inject expressions that are not permitted in a row
     * filter.  XXX We could try to have a more precise check to catch only
     * publications with row filters, or even re-verify the row filter
     * expressions.
     */
    if (attgenerated == ATTRIBUTE_GENERATED_VIRTUAL &&
        GetRelationIncludedPublications(RelationGetRelid(rel)) != NIL)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("ALTER TABLE / SET EXPRESSION is not supported for virtual generated columns in tables that are part of a publication"),
                 errdetail("Column \"%s\" of relation \"%s\" is a virtual generated column.",
                           colName, RelationGetRelationName(rel))));
 
    rewrite = (attgenerated == ATTRIBUTE_GENERATED_STORED);
 
    ReleaseSysCache(tuple);
 
    if (rewrite)
    {
        /*
         * Clear all the missing values if we're rewriting the table, since
         * this renders them pointless.
         */
        RelationClearMissing(rel);
 
        /* make sure we don't conflict with later attribute modifications */
        CommandCounterIncrement();
    }
 
    /*
     * Find everything that depends on the column (constraints, indexes, etc),
     * and record enough information to let us recreate the objects.
     */
    RememberAllDependentForRebuilding(tab, AT_SetExpression, rel, attnum, colName);
 
    /*
     * Find whole-row referenced objects that depend on the column
     * (constraints, indexes, etc.), and record enough information to let us
     * recreate the objects.
     */
    RememberWholeRowDependentForRebuilding(tab, AT_SetExpression, rel);
 
    /*
     * Drop the dependency records of the GENERATED expression, in particular
     * its INTERNAL dependency on the column, which would otherwise cause
     * dependency.c to refuse to perform the deletion.
     */
    attrdefoid = GetAttrDefaultOid(RelationGetRelid(rel), attnum);
    if (!OidIsValid(attrdefoid))
        elog(ERROR, "could not find attrdef tuple for relation %u attnum %d",
             RelationGetRelid(rel), attnum);
    (void) deleteDependencyRecordsFor(AttrDefaultRelationId, attrdefoid, false);
 
    /* Make above changes visible */
    CommandCounterIncrement();
 
    /*
     * Get rid of the GENERATED expression itself.  We use RESTRICT here for
     * safety, but at present we do not expect anything to depend on the
     * expression.
     */
    RemoveAttrDefault(RelationGetRelid(rel), attnum, DROP_RESTRICT,
                      false, false);
 
    /* Prepare to store the new expression, in the catalogs */
    rawEnt = palloc_object(RawColumnDefault);
    rawEnt->attnum = attnum;
    rawEnt->raw_default = newExpr;
    rawEnt->generated = attgenerated;
 
    /* Store the generated expression */
    AddRelationNewConstraints(rel, list_make1(rawEnt), NIL,
                              false, true, false, NULL);
 
    /* Make above new expression visible */
    CommandCounterIncrement();
 
    if (rewrite)
    {
        /* Prepare for table rewrite */
        defval = (Expr *) build_column_default(rel, attnum);
 
        newval = palloc0_object(NewColumnValue);
        newval->attnum = attnum;
        newval->expr = expression_planner(defval);
        newval->is_generated = true;
 
        tab->newvals = lappend(tab->newvals, newval);
        tab->rewrite |= AT_REWRITE_DEFAULT_VAL;
    }
 
    /* Drop any pg_statistic entry for the column */
    RemoveStatistics(RelationGetRelid(rel), attnum);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel), attnum);
 
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
    return address;
}

References AddRelationNewConstraints(), AT_REWRITE_DEFAULT_VAL, AT_SetExpression, attnum, build_column_default(), CommandCounterIncrement(), deleteDependencyRecordsFor(), DROP_RESTRICT, elog, ereport, errcode(), errdetail(), errmsg, ERROR, expression_planner(), fb(), GetAttrDefaultOid(), GetRelationIncludedPublications(), GETSTRUCT(), HeapTupleIsValid, InvokeObjectPostAlterHook, lappend(), list_make1, newval, AlteredTableInfo::newvals, NIL, ObjectAddressSubSet, OidIsValid, palloc0_object, palloc_object, RelationClearMissing(), RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), RememberAllDependentForRebuilding(), RememberWholeRowDependentForRebuilding(), RemoveAttrDefault(), RemoveStatistics(), AlteredTableInfo::rewrite, SearchSysCacheAttName(), and AlteredTableInfo::verify_new_notnull.

Referenced by ATExecCmd().

◆ ATExecSetIdentity()

static ObjectAddress ATExecSetIdentity	(	Relation	rel,
		const char *	colName,
		Node *	def,
		LOCKMODE	lockmode,
		bool	recurse,
		bool	recursing
	)

static

Definition at line 8512 of file tablecmds.c.

{
    ListCell   *option;
    DefElem    *generatedEl = NULL;
    HeapTuple   tuple;
    Form_pg_attribute attTup;
    AttrNumber  attnum;
    Relation    attrelation;
    ObjectAddress address;
    bool        ispartitioned;
 
    ispartitioned = (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE);
    if (ispartitioned && !recurse)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("cannot change identity column of only the partitioned table"),
                 errhint("Do not specify the ONLY keyword.")));
 
    if (rel->rd_rel->relispartition && !recursing)
        ereport(ERROR,
                errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                errmsg("cannot change identity column of a partition"));
 
    foreach(option, castNode(List, def))
    {
        DefElem    *defel = lfirst_node(DefElem, option);
 
        if (strcmp(defel->defname, "generated") == 0)
        {
            if (generatedEl)
                ereport(ERROR,
                        (errcode(ERRCODE_SYNTAX_ERROR),
                         errmsg("conflicting or redundant options")));
            generatedEl = defel;
        }
        else
            elog(ERROR, "option \"%s\" not recognized",
                 defel->defname);
    }
 
    /*
     * Even if there is nothing to change here, we run all the checks.  There
     * will be a subsequent ALTER SEQUENCE that relies on everything being
     * there.
     */
 
    attrelation = table_open(AttributeRelationId, RowExclusiveLock);
    tuple = SearchSysCacheCopyAttName(RelationGetRelid(rel), colName);
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel))));
 
    attTup = (Form_pg_attribute) GETSTRUCT(tuple);
    attnum = attTup->attnum;
 
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"",
                        colName)));
 
    if (!attTup->attidentity)
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                 errmsg("column \"%s\" of relation \"%s\" is not an identity column",
                        colName, RelationGetRelationName(rel))));
 
    if (generatedEl)
    {
        attTup->attidentity = defGetInt32(generatedEl);
        CatalogTupleUpdate(attrelation, &tuple->t_self, tuple);
 
        InvokeObjectPostAlterHook(RelationRelationId,
                                  RelationGetRelid(rel),
                                  attTup->attnum);
        ObjectAddressSubSet(address, RelationRelationId,
                            RelationGetRelid(rel), attnum);
    }
    else
        address = InvalidObjectAddress;
 
    heap_freetuple(tuple);
    table_close(attrelation, RowExclusiveLock);
 
    /*
     * Recurse to propagate the identity change to partitions. Identity is not
     * inherited in regular inheritance children.
     */
    if (generatedEl && recurse && ispartitioned)
    {
        List       *children;
        ListCell   *lc;
 
        children = find_inheritance_children(RelationGetRelid(rel), lockmode);
 
        foreach(lc, children)
        {
            Relation    childrel;
 
            childrel = table_open(lfirst_oid(lc), NoLock);
            ATExecSetIdentity(childrel, colName, def, lockmode, recurse, true);
            table_close(childrel, NoLock);
        }
    }
 
    return address;
}

References ATExecSetIdentity(), attnum, castNode, CatalogTupleUpdate(), defGetInt32(), elog, ereport, errcode(), errhint(), errmsg, ERROR, fb(), find_inheritance_children(), GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InvalidObjectAddress, InvokeObjectPostAlterHook, lfirst_node, lfirst_oid, NoLock, ObjectAddressSubSet, RelationData::rd_rel, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, SearchSysCacheCopyAttName(), HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecCmd(), and ATExecSetIdentity().

◆ ATExecSetNotNull()

static ObjectAddress ATExecSetNotNull	(	List **	wqueue,
		Relation	rel,
		char *	conName,
		char *	colName,
		bool	recurse,
		bool	recursing,
		LOCKMODE	lockmode
	)

static

Definition at line 8054 of file tablecmds.c.

{
    HeapTuple   tuple;
    AttrNumber  attnum;
    ObjectAddress address;
    Constraint *constraint;
    CookedConstraint *ccon;
    List       *cooked;
    bool        is_no_inherit = false;
 
    /* Guard against stack overflow due to overly deep inheritance tree. */
    check_stack_depth();
 
    /* At top level, permission check was done in ATPrepCmd, else do it */
    if (recursing)
    {
        ATSimplePermissions(AT_AddConstraint, rel,
                            ATT_PARTITIONED_TABLE | ATT_TABLE | ATT_FOREIGN_TABLE);
        Assert(conName != NULL);
    }
 
    attnum = get_attnum(RelationGetRelid(rel), colName);
    if (attnum == InvalidAttrNumber)
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel))));
 
    /* Prevent them from altering a system attribute */
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"",
                        colName)));
 
    /* See if there's already a constraint */
    tuple = findNotNullConstraintAttnum(RelationGetRelid(rel), attnum);
    if (HeapTupleIsValid(tuple))
    {
        Form_pg_constraint conForm = (Form_pg_constraint) GETSTRUCT(tuple);
        bool        changed = false;
 
        /*
         * Don't let a NO INHERIT constraint be changed into inherit.
         */
        if (conForm->connoinherit && recurse)
            ereport(ERROR,
                    errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                    errmsg("cannot change NO INHERIT status of NOT NULL constraint \"%s\" on relation \"%s\"",
                           NameStr(conForm->conname),
                           RelationGetRelationName(rel)));
 
        /*
         * If we find an appropriate constraint, we're almost done, but just
         * need to change some properties on it: if we're recursing, increment
         * coninhcount; if not, set conislocal if not already set.
         */
        if (recursing)
        {
            if (pg_add_s16_overflow(conForm->coninhcount, 1,
                                    &conForm->coninhcount))
                ereport(ERROR,
                        errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
                        errmsg("too many inheritance parents"));
            changed = true;
        }
        else if (!conForm->conislocal)
        {
            conForm->conislocal = true;
            changed = true;
        }
        else if (!conForm->convalidated)
        {
            /*
             * Flip attnotnull and convalidated, and also validate the
             * constraint.
             */
            return ATExecValidateConstraint(wqueue, rel, NameStr(conForm->conname),
                                            recurse, recursing, lockmode);
        }
 
        if (changed)
        {
            Relation    constr_rel;
 
            constr_rel = table_open(ConstraintRelationId, RowExclusiveLock);
 
            CatalogTupleUpdate(constr_rel, &tuple->t_self, tuple);
            ObjectAddressSet(address, ConstraintRelationId, conForm->oid);
            table_close(constr_rel, RowExclusiveLock);
        }
 
        if (changed)
            return address;
        else
            return InvalidObjectAddress;
    }
 
    /*
     * If we're asked not to recurse, and children exist, raise an error for
     * partitioned tables.  For inheritance, we act as if NO INHERIT had been
     * specified.
     */
    if (!recurse &&
        find_inheritance_children(RelationGetRelid(rel),
                                  NoLock) != NIL)
    {
        if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
            ereport(ERROR,
                    errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                    errmsg("constraint must be added to child tables too"),
                    errhint("Do not specify the ONLY keyword."));
        else
            is_no_inherit = true;
    }
 
    /*
     * No constraint exists; we must add one.  First determine a name to use,
     * if we haven't already.
     */
    if (!recursing)
    {
        Assert(conName == NULL);
        conName = ChooseConstraintName(RelationGetRelationName(rel),
                                       colName, "not_null",
                                       RelationGetNamespace(rel),
                                       NIL);
    }
 
    constraint = makeNotNullConstraint(makeString(colName));
    constraint->is_no_inherit = is_no_inherit;
    constraint->conname = conName;
 
    /* and do it */
    cooked = AddRelationNewConstraints(rel, NIL, list_make1(constraint),
                                       false, !recursing, false, NULL);
    ccon = linitial(cooked);
    ObjectAddressSet(address, ConstraintRelationId, ccon->conoid);
 
    /* Mark pg_attribute.attnotnull for the column and queue validation */
    set_attnotnull(wqueue, rel, attnum, true, true);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel), attnum);
 
    /*
     * Recurse to propagate the constraint to children that don't have one.
     */
    if (recurse)
    {
        List       *children;
 
        children = find_inheritance_children(RelationGetRelid(rel),
                                             lockmode);
 
        foreach_oid(childoid, children)
        {
            Relation    childrel = table_open(childoid, NoLock);
 
            CommandCounterIncrement();
 
            ATExecSetNotNull(wqueue, childrel, conName, colName,
                             recurse, true, lockmode);
            table_close(childrel, NoLock);
        }
    }
 
    return address;
}

References AddRelationNewConstraints(), Assert, AT_AddConstraint, ATExecSetNotNull(), ATExecValidateConstraint(), ATSimplePermissions(), ATT_FOREIGN_TABLE, ATT_PARTITIONED_TABLE, ATT_TABLE, attnum, CatalogTupleUpdate(), check_stack_depth(), ChooseConstraintName(), CommandCounterIncrement(), Constraint::conname, ereport, errcode(), errhint(), errmsg, ERROR, fb(), find_inheritance_children(), findNotNullConstraintAttnum(), foreach_oid, Form_pg_constraint, get_attnum(), GETSTRUCT(), HeapTupleIsValid, InvalidAttrNumber, InvalidObjectAddress, InvokeObjectPostAlterHook, Constraint::is_no_inherit, linitial, list_make1, makeNotNullConstraint(), makeString(), NameStr, NIL, NoLock, ObjectAddressSet, pg_add_s16_overflow(), RelationData::rd_rel, RelationGetNamespace, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, set_attnotnull(), HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecAlterConstrInheritability(), ATExecCmd(), and ATExecSetNotNull().

◆ ATExecSetOptions()

static ObjectAddress ATExecSetOptions	(	Relation	rel,
		const char *	colName,
		Node *	options,
		bool	isReset,
		LOCKMODE	lockmode
	)

static

Definition at line 9186 of file tablecmds.c.

{
    Relation    attrelation;
    HeapTuple   tuple,
                newtuple;
    Form_pg_attribute attrtuple;
    AttrNumber  attnum;
    Datum       datum,
                newOptions;
    bool        isnull;
    ObjectAddress address;
    Datum       repl_val[Natts_pg_attribute];
    bool        repl_null[Natts_pg_attribute];
    bool        repl_repl[Natts_pg_attribute];
 
    attrelation = table_open(AttributeRelationId, RowExclusiveLock);
 
    tuple = SearchSysCacheAttName(RelationGetRelid(rel), colName);
 
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel))));
    attrtuple = (Form_pg_attribute) GETSTRUCT(tuple);
 
    attnum = attrtuple->attnum;
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"",
                        colName)));
 
    /* Generate new proposed attoptions (text array) */
    datum = SysCacheGetAttr(ATTNAME, tuple, Anum_pg_attribute_attoptions,
                            &isnull);
    newOptions = transformRelOptions(isnull ? (Datum) 0 : datum,
                                     castNode(List, options), NULL, NULL,
                                     false, isReset);
    /* Validate new options */
    (void) attribute_reloptions(newOptions, true);
 
    /* Build new tuple. */
    memset(repl_null, false, sizeof(repl_null));
    memset(repl_repl, false, sizeof(repl_repl));
    if (newOptions != (Datum) 0)
        repl_val[Anum_pg_attribute_attoptions - 1] = newOptions;
    else
        repl_null[Anum_pg_attribute_attoptions - 1] = true;
    repl_repl[Anum_pg_attribute_attoptions - 1] = true;
    newtuple = heap_modify_tuple(tuple, RelationGetDescr(attrelation),
                                 repl_val, repl_null, repl_repl);
 
    /* Update system catalog. */
    CatalogTupleUpdate(attrelation, &newtuple->t_self, newtuple);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel),
                              attrtuple->attnum);
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
 
    heap_freetuple(newtuple);
 
    ReleaseSysCache(tuple);
 
    table_close(attrelation, RowExclusiveLock);
 
    return address;
}

References attnum, attribute_reloptions(), castNode, CatalogTupleUpdate(), ereport, errcode(), errmsg, ERROR, fb(), GETSTRUCT(), heap_freetuple(), heap_modify_tuple(), HeapTupleIsValid, InvokeObjectPostAlterHook, ObjectAddressSubSet, RelationGetDescr, RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), RowExclusiveLock, SearchSysCacheAttName(), SysCacheGetAttr(), HeapTupleData::t_self, table_close(), table_open(), and transformRelOptions().

Referenced by ATExecCmd().

◆ ATExecSetRelOptions()

static void ATExecSetRelOptions	(	Relation	rel,
		List *	defList,
		AlterTableType	operation,
		LOCKMODE	lockmode
	)

static

Definition at line 17351 of file tablecmds.c.

{
    Oid         relid;
    Relation    pgclass;
    HeapTuple   tuple;
    HeapTuple   newtuple;
    Datum       datum;
    Datum       newOptions;
    Datum       repl_val[Natts_pg_class];
    bool        repl_null[Natts_pg_class];
    bool        repl_repl[Natts_pg_class];
    const char *const validnsps[] = HEAP_RELOPT_NAMESPACES;
 
    if (defList == NIL && operation != AT_ReplaceRelOptions)
        return;                 /* nothing to do */
 
    pgclass = table_open(RelationRelationId, RowExclusiveLock);
 
    /* Fetch heap tuple */
    relid = RelationGetRelid(rel);
    tuple = SearchSysCacheLocked1(RELOID, ObjectIdGetDatum(relid));
    if (!HeapTupleIsValid(tuple))
        elog(ERROR, "cache lookup failed for relation %u", relid);
 
    if (operation == AT_ReplaceRelOptions)
    {
        /*
         * If we're supposed to replace the reloptions list, we just pretend
         * there were none before.
         */
        datum = (Datum) 0;
    }
    else
    {
        bool        isnull;
 
        /* Get the old reloptions */
        datum = SysCacheGetAttr(RELOID, tuple, Anum_pg_class_reloptions,
                                &isnull);
        if (isnull)
            datum = (Datum) 0;
    }
 
    /* Generate new proposed reloptions (text array) */
    newOptions = transformRelOptions(datum, defList, NULL, validnsps, false,
                                     operation == AT_ResetRelOptions);
 
    /* Validate */
    switch (rel->rd_rel->relkind)
    {
        case RELKIND_RELATION:
        case RELKIND_MATVIEW:
            (void) heap_reloptions(rel->rd_rel->relkind, newOptions, true);
            break;
        case RELKIND_PARTITIONED_TABLE:
            (void) partitioned_table_reloptions(newOptions, true);
            break;
        case RELKIND_VIEW:
            (void) view_reloptions(newOptions, true);
            break;
        case RELKIND_INDEX:
        case RELKIND_PARTITIONED_INDEX:
            (void) index_reloptions(rel->rd_indam->amoptions, newOptions, true);
            break;
        case RELKIND_TOASTVALUE:
            /* fall through to error -- shouldn't ever get here */
        default:
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("cannot set options for relation \"%s\"",
                            RelationGetRelationName(rel)),
                     errdetail_relkind_not_supported(rel->rd_rel->relkind)));
            break;
    }
 
    /* Special-case validation of view options */
    if (rel->rd_rel->relkind == RELKIND_VIEW)
    {
        Query      *view_query = get_view_query(rel);
        List       *view_options = untransformRelOptions(newOptions);
        ListCell   *cell;
        bool        check_option = false;
 
        foreach(cell, view_options)
        {
            DefElem    *defel = (DefElem *) lfirst(cell);
 
            if (strcmp(defel->defname, "check_option") == 0)
                check_option = true;
        }
 
        /*
         * If the check option is specified, look to see if the view is
         * actually auto-updatable or not.
         */
        if (check_option)
        {
            const char *view_updatable_error =
                view_query_is_auto_updatable(view_query, true);
 
            if (view_updatable_error)
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("WITH CHECK OPTION is supported only on automatically updatable views"),
                         errhint("%s", _(view_updatable_error))));
        }
    }
 
    /*
     * All we need do here is update the pg_class row; the new options will be
     * propagated into relcaches during post-commit cache inval.
     */
    memset(repl_val, 0, sizeof(repl_val));
    memset(repl_null, false, sizeof(repl_null));
    memset(repl_repl, false, sizeof(repl_repl));
 
    if (newOptions != (Datum) 0)
        repl_val[Anum_pg_class_reloptions - 1] = newOptions;
    else
        repl_null[Anum_pg_class_reloptions - 1] = true;
 
    repl_repl[Anum_pg_class_reloptions - 1] = true;
 
    newtuple = heap_modify_tuple(tuple, RelationGetDescr(pgclass),
                                 repl_val, repl_null, repl_repl);
 
    CatalogTupleUpdate(pgclass, &newtuple->t_self, newtuple);
    UnlockTuple(pgclass, &tuple->t_self, InplaceUpdateTupleLock);
 
    InvokeObjectPostAlterHook(RelationRelationId, RelationGetRelid(rel), 0);
 
    heap_freetuple(newtuple);
 
    ReleaseSysCache(tuple);
 
    /* repeat the whole exercise for the toast table, if there's one */
    if (OidIsValid(rel->rd_rel->reltoastrelid))
    {
        Relation    toastrel;
        Oid         toastid = rel->rd_rel->reltoastrelid;
 
        toastrel = table_open(toastid, lockmode);
 
        /* Fetch heap tuple */
        tuple = SearchSysCache1(RELOID, ObjectIdGetDatum(toastid));
        if (!HeapTupleIsValid(tuple))
            elog(ERROR, "cache lookup failed for relation %u", toastid);
 
        if (operation == AT_ReplaceRelOptions)
        {
            /*
             * If we're supposed to replace the reloptions list, we just
             * pretend there were none before.
             */
            datum = (Datum) 0;
        }
        else
        {
            bool        isnull;
 
            /* Get the old reloptions */
            datum = SysCacheGetAttr(RELOID, tuple, Anum_pg_class_reloptions,
                                    &isnull);
            if (isnull)
                datum = (Datum) 0;
        }
 
        newOptions = transformRelOptions(datum, defList, "toast", validnsps,
                                         false, operation == AT_ResetRelOptions);
 
        (void) heap_reloptions(RELKIND_TOASTVALUE, newOptions, true);
 
        memset(repl_val, 0, sizeof(repl_val));
        memset(repl_null, false, sizeof(repl_null));
        memset(repl_repl, false, sizeof(repl_repl));
 
        if (newOptions != (Datum) 0)
            repl_val[Anum_pg_class_reloptions - 1] = newOptions;
        else
            repl_null[Anum_pg_class_reloptions - 1] = true;
 
        repl_repl[Anum_pg_class_reloptions - 1] = true;
 
        newtuple = heap_modify_tuple(tuple, RelationGetDescr(pgclass),
                                     repl_val, repl_null, repl_repl);
 
        CatalogTupleUpdate(pgclass, &newtuple->t_self, newtuple);
 
        InvokeObjectPostAlterHookArg(RelationRelationId,
                                     RelationGetRelid(toastrel), 0,
                                     InvalidOid, true);
 
        heap_freetuple(newtuple);
 
        ReleaseSysCache(tuple);
 
        table_close(toastrel, NoLock);
    }
 
    table_close(pgclass, RowExclusiveLock);
}

References _, IndexAmRoutine::amoptions, AT_ReplaceRelOptions, AT_ResetRelOptions, CatalogTupleUpdate(), elog, ereport, errcode(), errdetail_relkind_not_supported(), errhint(), errmsg, ERROR, fb(), get_view_query(), heap_freetuple(), heap_modify_tuple(), HEAP_RELOPT_NAMESPACES, heap_reloptions(), HeapTupleIsValid, index_reloptions(), InplaceUpdateTupleLock, InvalidOid, InvokeObjectPostAlterHook, InvokeObjectPostAlterHookArg, lfirst, NIL, NoLock, ObjectIdGetDatum(), OidIsValid, operation, partitioned_table_reloptions(), RelationData::rd_indam, RelationData::rd_rel, RelationGetDescr, RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), RowExclusiveLock, SearchSysCache1(), SearchSysCacheLocked1(), SysCacheGetAttr(), HeapTupleData::t_self, table_close(), table_open(), transformRelOptions(), UnlockTuple(), untransformRelOptions(), view_query_is_auto_updatable(), and view_reloptions().

Referenced by ATExecCmd().

◆ ATExecSetRowSecurity()

static void ATExecSetRowSecurity	(	Relation	rel,
		bool	rls
	)

static

Definition at line 19307 of file tablecmds.c.

{
    Relation    pg_class;
    Oid         relid;
    HeapTuple   tuple;
 
    relid = RelationGetRelid(rel);
 
    /* Pull the record for this relation and update it */
    pg_class = table_open(RelationRelationId, RowExclusiveLock);
 
    tuple = SearchSysCacheCopy1(RELOID, ObjectIdGetDatum(relid));
 
    if (!HeapTupleIsValid(tuple))
        elog(ERROR, "cache lookup failed for relation %u", relid);
 
    ((Form_pg_class) GETSTRUCT(tuple))->relrowsecurity = rls;
    CatalogTupleUpdate(pg_class, &tuple->t_self, tuple);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel), 0);
 
    table_close(pg_class, RowExclusiveLock);
    heap_freetuple(tuple);
}

References CatalogTupleUpdate(), elog, ERROR, fb(), GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InvokeObjectPostAlterHook, ObjectIdGetDatum(), RelationGetRelid, RowExclusiveLock, SearchSysCacheCopy1, HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecCmd().

◆ ATExecSetStatistics()

static ObjectAddress ATExecSetStatistics	(	Relation	rel,
		const char *	colName,
		int16	colNum,
		Node *	newValue,
		LOCKMODE	lockmode
	)

static

Definition at line 9041 of file tablecmds.c.

{
    int         newtarget = 0;
    bool        newtarget_default;
    Relation    attrelation;
    HeapTuple   tuple,
                newtuple;
    Form_pg_attribute attrtuple;
    AttrNumber  attnum;
    ObjectAddress address;
    Datum       repl_val[Natts_pg_attribute];
    bool        repl_null[Natts_pg_attribute];
    bool        repl_repl[Natts_pg_attribute];
 
    /*
     * We allow referencing columns by numbers only for indexes, since table
     * column numbers could contain gaps if columns are later dropped.
     */
    if (rel->rd_rel->relkind != RELKIND_INDEX &&
        rel->rd_rel->relkind != RELKIND_PARTITIONED_INDEX &&
        !colName)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot refer to non-index column by number")));
 
    /* -1 was used in previous versions for the default setting */
    if (newValue && intVal(newValue) != -1)
    {
        newtarget = intVal(newValue);
        newtarget_default = false;
    }
    else
        newtarget_default = true;
 
    if (!newtarget_default)
    {
        /*
         * Limit target to a sane range
         */
        if (newtarget < 0)
        {
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                     errmsg("statistics target %d is too low",
                            newtarget)));
        }
        else if (newtarget > MAX_STATISTICS_TARGET)
        {
            newtarget = MAX_STATISTICS_TARGET;
            ereport(WARNING,
                    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                     errmsg("lowering statistics target to %d",
                            newtarget)));
        }
    }
 
    attrelation = table_open(AttributeRelationId, RowExclusiveLock);
 
    if (colName)
    {
        tuple = SearchSysCacheAttName(RelationGetRelid(rel), colName);
 
        if (!HeapTupleIsValid(tuple))
            ereport(ERROR,
                    (errcode(ERRCODE_UNDEFINED_COLUMN),
                     errmsg("column \"%s\" of relation \"%s\" does not exist",
                            colName, RelationGetRelationName(rel))));
    }
    else
    {
        tuple = SearchSysCacheAttNum(RelationGetRelid(rel), colNum);
 
        if (!HeapTupleIsValid(tuple))
            ereport(ERROR,
                    (errcode(ERRCODE_UNDEFINED_COLUMN),
                     errmsg("column number %d of relation \"%s\" does not exist",
                            colNum, RelationGetRelationName(rel))));
    }
 
    attrtuple = (Form_pg_attribute) GETSTRUCT(tuple);
 
    attnum = attrtuple->attnum;
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"",
                        colName)));
 
    /*
     * Prevent this as long as the ANALYZE code skips virtual generated
     * columns.
     */
    if (attrtuple->attgenerated == ATTRIBUTE_GENERATED_VIRTUAL)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter statistics on virtual generated column \"%s\"",
                        colName)));
 
    if (rel->rd_rel->relkind == RELKIND_INDEX ||
        rel->rd_rel->relkind == RELKIND_PARTITIONED_INDEX)
    {
        if (attnum > rel->rd_index->indnkeyatts)
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("cannot alter statistics on included column \"%s\" of index \"%s\"",
                            NameStr(attrtuple->attname), RelationGetRelationName(rel))));
        else if (rel->rd_index->indkey.values[attnum - 1] != 0)
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("cannot alter statistics on non-expression column \"%s\" of index \"%s\"",
                            NameStr(attrtuple->attname), RelationGetRelationName(rel)),
                     errhint("Alter statistics on table column instead.")));
    }
 
    /* Build new tuple. */
    memset(repl_null, false, sizeof(repl_null));
    memset(repl_repl, false, sizeof(repl_repl));
    if (!newtarget_default)
        repl_val[Anum_pg_attribute_attstattarget - 1] = Int16GetDatum(newtarget);
    else
        repl_null[Anum_pg_attribute_attstattarget - 1] = true;
    repl_repl[Anum_pg_attribute_attstattarget - 1] = true;
    newtuple = heap_modify_tuple(tuple, RelationGetDescr(attrelation),
                                 repl_val, repl_null, repl_repl);
    CatalogTupleUpdate(attrelation, &tuple->t_self, newtuple);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel),
                              attrtuple->attnum);
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
 
    heap_freetuple(newtuple);
 
    ReleaseSysCache(tuple);
 
    table_close(attrelation, RowExclusiveLock);
 
    return address;
}

References attnum, CatalogTupleUpdate(), ereport, errcode(), errhint(), errmsg, ERROR, fb(), GETSTRUCT(), heap_freetuple(), heap_modify_tuple(), HeapTupleIsValid, Int16GetDatum(), intVal, InvokeObjectPostAlterHook, MAX_STATISTICS_TARGET, NameStr, ObjectAddressSubSet, RelationData::rd_index, RelationData::rd_rel, RelationGetDescr, RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), RowExclusiveLock, SearchSysCacheAttName(), SearchSysCacheAttNum(), HeapTupleData::t_self, table_close(), table_open(), and WARNING.

Referenced by ATExecCmd().

◆ ATExecSetStorage()

static ObjectAddress ATExecSetStorage	(	Relation	rel,
		const char *	colName,
		Node *	newValue,
		LOCKMODE	lockmode
	)

static

Definition at line 9328 of file tablecmds.c.

{
    Relation    attrelation;
    HeapTuple   tuple;
    Form_pg_attribute attrtuple;
    AttrNumber  attnum;
    ObjectAddress address;
 
    attrelation = table_open(AttributeRelationId, RowExclusiveLock);
 
    tuple = SearchSysCacheCopyAttName(RelationGetRelid(rel), colName);
 
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel))));
    attrtuple = (Form_pg_attribute) GETSTRUCT(tuple);
 
    attnum = attrtuple->attnum;
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"",
                        colName)));
 
    attrtuple->attstorage = GetAttributeStorage(attrtuple->atttypid, strVal(newValue));
 
    CatalogTupleUpdate(attrelation, &tuple->t_self, tuple);
 
    InvokeObjectPostAlterHook(RelationRelationId,
                              RelationGetRelid(rel),
                              attrtuple->attnum);
 
    /*
     * Apply the change to indexes as well (only for simple index columns,
     * matching behavior of index.c ConstructTupleDescriptor()).
     */
    SetIndexStorageProperties(rel, attrelation, attnum,
                              true, attrtuple->attstorage,
                              false, 0,
                              lockmode);
 
    heap_freetuple(tuple);
 
    table_close(attrelation, RowExclusiveLock);
 
    ObjectAddressSubSet(address, RelationRelationId,
                        RelationGetRelid(rel), attnum);
    return address;
}

References attnum, CatalogTupleUpdate(), ereport, errcode(), errmsg, ERROR, fb(), GetAttributeStorage(), GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, InvokeObjectPostAlterHook, ObjectAddressSubSet, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, SearchSysCacheCopyAttName(), SetIndexStorageProperties(), strVal, HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecCmd().

◆ ATExecSetTableSpace()

static void ATExecSetTableSpace	(	Oid	tableOid,
		Oid	newTableSpace,
		LOCKMODE	lockmode
	)

static

Definition at line 17559 of file tablecmds.c.

{
    Relation    rel;
    Oid         reltoastrelid;
    RelFileNumber newrelfilenumber;
    RelFileLocator newrlocator;
    List       *reltoastidxids = NIL;
    ListCell   *lc;
 
    /*
     * Need lock here in case we are recursing to toast table or index
     */
    rel = relation_open(tableOid, lockmode);
 
    /* Check first if relation can be moved to new tablespace */
    if (!CheckRelationTableSpaceMove(rel, newTableSpace))
    {
        InvokeObjectPostAlterHook(RelationRelationId,
                                  RelationGetRelid(rel), 0);
        relation_close(rel, NoLock);
        return;
    }
 
    reltoastrelid = rel->rd_rel->reltoastrelid;
    /* Fetch the list of indexes on toast relation if necessary */
    if (OidIsValid(reltoastrelid))
    {
        Relation    toastRel = relation_open(reltoastrelid, lockmode);
 
        reltoastidxids = RelationGetIndexList(toastRel);
        relation_close(toastRel, lockmode);
    }
 
    /*
     * Relfilenumbers are not unique in databases across tablespaces, so we
     * need to allocate a new one in the new tablespace.
     */
    newrelfilenumber = GetNewRelFileNumber(newTableSpace, NULL,
                                           rel->rd_rel->relpersistence);
 
    /* Open old and new relation */
    newrlocator = rel->rd_locator;
    newrlocator.relNumber = newrelfilenumber;
    newrlocator.spcOid = newTableSpace;
 
    /* hand off to AM to actually create new rel storage and copy the data */
    if (rel->rd_rel->relkind == RELKIND_INDEX)
    {
        index_copy_data(rel, newrlocator);
    }
    else
    {
        Assert(RELKIND_HAS_TABLE_AM(rel->rd_rel->relkind));
        table_relation_copy_data(rel, &newrlocator);
    }
 
    /*
     * Update the pg_class row.
     *
     * NB: This wouldn't work if ATExecSetTableSpace() were allowed to be
     * executed on pg_class or its indexes (the above copy wouldn't contain
     * the updated pg_class entry), but that's forbidden with
     * CheckRelationTableSpaceMove().
     */
    SetRelationTableSpace(rel, newTableSpace, newrelfilenumber);
 
    InvokeObjectPostAlterHook(RelationRelationId, RelationGetRelid(rel), 0);
 
    RelationAssumeNewRelfilelocator(rel);
 
    relation_close(rel, NoLock);
 
    /* Make sure the reltablespace change is visible */
    CommandCounterIncrement();
 
    /* Move associated toast relation and/or indexes, too */
    if (OidIsValid(reltoastrelid))
        ATExecSetTableSpace(reltoastrelid, newTableSpace, lockmode);
    foreach(lc, reltoastidxids)
        ATExecSetTableSpace(lfirst_oid(lc), newTableSpace, lockmode);
 
    /* Clean up */
    list_free(reltoastidxids);
}

References Assert, ATExecSetTableSpace(), CheckRelationTableSpaceMove(), CommandCounterIncrement(), fb(), GetNewRelFileNumber(), index_copy_data(), InvokeObjectPostAlterHook, lfirst_oid, list_free(), NIL, NoLock, OidIsValid, RelationData::rd_locator, RelationData::rd_rel, relation_close(), relation_open(), RelationAssumeNewRelfilelocator(), RelationGetIndexList(), RelationGetRelid, RelFileLocator::relNumber, SetRelationTableSpace(), and table_relation_copy_data().

Referenced by ATExecSetTableSpace(), and ATRewriteTables().

◆ ATExecSetTableSpaceNoStorage()

static void ATExecSetTableSpaceNoStorage	(	Relation	rel,
		Oid	newTableSpace
	)

static

Definition at line 17652 of file tablecmds.c.

{
    /*
     * Shouldn't be called on relations having storage; these are processed in
     * phase 3.
     */
    Assert(!RELKIND_HAS_STORAGE(rel->rd_rel->relkind));
 
    /* check if relation can be moved to its new tablespace */
    if (!CheckRelationTableSpaceMove(rel, newTableSpace))
    {
        InvokeObjectPostAlterHook(RelationRelationId,
                                  RelationGetRelid(rel),
                                  0);
        return;
    }
 
    /* Update can be done, so change reltablespace */
    SetRelationTableSpace(rel, newTableSpace, InvalidOid);
 
    InvokeObjectPostAlterHook(RelationRelationId, RelationGetRelid(rel), 0);
 
    /* Make sure the reltablespace change is visible */
    CommandCounterIncrement();
}

References Assert, CheckRelationTableSpaceMove(), CommandCounterIncrement(), fb(), InvalidOid, InvokeObjectPostAlterHook, RelationData::rd_rel, RelationGetRelid, and SetRelationTableSpace().

Referenced by ATExecCmd().

◆ ATExecSplitPartition()

static void ATExecSplitPartition	(	List **	wqueue,
		AlteredTableInfo *	tab,
		Relation	rel,
		PartitionCmd *	cmd,
		AlterTableUtilityContext *	context
	)

static

Definition at line 24228 of file tablecmds.c.

{
    Relation    splitRel;
    Oid         splitRelOid;
    ListCell   *listptr,
               *listptr2;
    bool        isSameName = false;
    char        tmpRelName[NAMEDATALEN];
    List       *newPartRels = NIL;
    List       *extDepState = NIL;
    ObjectAddress object;
    Oid         defaultPartOid;
    Oid         save_userid;
    int         save_sec_context;
    int         save_nestlevel;
    List       *splitPartList;
 
    defaultPartOid = get_default_oid_from_partdesc(RelationGetPartitionDesc(rel, true));
 
    /*
     * Partition is already locked in the transformPartitionCmdForSplit
     * function.
     */
    splitRel = table_openrv(cmd->name, NoLock);
 
    splitRelOid = RelationGetRelid(splitRel);
 
    /* Check descriptions of new partitions. */
    foreach_node(SinglePartitionSpec, sps, cmd->partlist)
    {
        Oid         existingRelid;
 
        /* Look up the existing relation by the new partition name. */
        RangeVarGetAndCheckCreationNamespace(sps->name, NoLock, &existingRelid);
 
        /*
         * This would fail later on anyway if the relation already exists. But
         * by catching it here, we can emit a nicer error message.
         */
        if (existingRelid == splitRelOid && !isSameName)
            /* One new partition can have the same name as a split partition. */
            isSameName = true;
        else if (OidIsValid(existingRelid))
            ereport(ERROR,
                    errcode(ERRCODE_DUPLICATE_TABLE),
                    errmsg("relation \"%s\" already exists", sps->name->relname));
    }
 
    /*
     * Collect extension dependencies from indexes on the split partition. We
     * must do this before detaching it, so we can restore the dependencies on
     * the new partitions' indexes later.
     */
    splitPartList = list_make1_oid(splitRelOid);
 
    extDepState = collectPartitionIndexExtDeps(splitPartList);
    list_free(splitPartList);
 
    /* Detach the split partition. */
    detachPartitionTable(rel, splitRel, defaultPartOid);
 
    /*
     * Perform a preliminary check to determine whether it's safe to drop the
     * split partition before we actually do so later. After merging rows into
     * the new partitions via SplitPartitionMoveRows, all old partitions need
     * to be dropped. However, since the drop behavior is DROP_RESTRICT and
     * the merge process (SplitPartitionMoveRows) can be time-consuming,
     * performing an early check on the drop eligibility of old partitions is
     * preferable.
     */
    object.objectId = splitRelOid;
    object.classId = RelationRelationId;
    object.objectSubId = 0;
    performDeletionCheck(&object, DROP_RESTRICT, 0);
 
    /*
     * If a new partition has the same name as the split partition, then we
     * should rename the split partition to reuse its name.
     */
    if (isSameName)
    {
        /*
         * We must bump the command counter to make the split partition tuple
         * visible for renaming.
         */
        CommandCounterIncrement();
        /* Rename partition. */
        sprintf(tmpRelName, "split-%u-%X-tmp", RelationGetRelid(rel), MyProcPid);
        RenameRelationInternal(splitRelOid, tmpRelName, true, false);
 
        /*
         * We must bump the command counter to make the split partition tuple
         * visible after renaming.
         */
        CommandCounterIncrement();
    }
 
    /* Create new partitions (like a split partition), without indexes. */
    foreach_node(SinglePartitionSpec, sps, cmd->partlist)
    {
        Relation    newPartRel;
 
        newPartRel = createPartitionTable(wqueue, sps->name, rel,
                                          splitRel->rd_rel->relowner);
        newPartRels = lappend(newPartRels, newPartRel);
    }
 
    /*
     * Switch to the table owner's userid, so that any index functions are run
     * as that user.  Also, lockdown security-restricted operations and
     * arrange to make GUC variable changes local to this command.
     *
     * Need to do it after determining the namespace in the
     * createPartitionTable() call.
     */
    GetUserIdAndSecContext(&save_userid, &save_sec_context);
    SetUserIdAndSecContext(splitRel->rd_rel->relowner,
                           save_sec_context | SECURITY_RESTRICTED_OPERATION);
    save_nestlevel = NewGUCNestLevel();
    RestrictSearchPath();
 
    /* Copy data from the split partition to the new partitions. */
    SplitPartitionMoveRows(wqueue, rel, splitRel, cmd->partlist, newPartRels);
    /* Keep the lock until commit. */
    table_close(splitRel, NoLock);
 
    /* Attach new partitions to the partitioned table. */
    forboth(listptr, cmd->partlist, listptr2, newPartRels)
    {
        SinglePartitionSpec *sps = (SinglePartitionSpec *) lfirst(listptr);
        Relation    newPartRel = (Relation) lfirst(listptr2);
 
        /*
         * wqueue = NULL: verification for each cloned constraint is not
         * needed.
         */
        attachPartitionTable(NULL, rel, newPartRel, sps->bound);
 
        /*
         * Apply extension dependencies to the new partition's indexes. This
         * preserves any "DEPENDS ON EXTENSION" settings from the split
         * partition.
         */
        applyPartitionIndexExtDeps(RelationGetRelid(newPartRel), extDepState);
 
        /* Keep the lock until commit. */
        table_close(newPartRel, NoLock);
    }
 
    freePartitionIndexExtDeps(extDepState);
 
    /* Drop the split partition. */
    object.classId = RelationRelationId;
    object.objectId = splitRelOid;
    object.objectSubId = 0;
    /* Probably DROP_CASCADE is not needed. */
    performDeletion(&object, DROP_RESTRICT, 0);
 
    /* Roll back any GUC changes executed by index functions. */
    AtEOXact_GUC(false, save_nestlevel);
 
    /* Restore the userid and security context. */
    SetUserIdAndSecContext(save_userid, save_sec_context);
}

References applyPartitionIndexExtDeps(), AtEOXact_GUC(), attachPartitionTable(), collectPartitionIndexExtDeps(), CommandCounterIncrement(), createPartitionTable(), detachPartitionTable(), DROP_RESTRICT, ereport, errcode(), errmsg, ERROR, fb(), forboth, foreach_node, freePartitionIndexExtDeps(), get_default_oid_from_partdesc(), GetUserIdAndSecContext(), lappend(), lfirst, list_free(), list_make1_oid, MyProcPid, PartitionCmd::name, NAMEDATALEN, NewGUCNestLevel(), NIL, NoLock, OidIsValid, PartitionCmd::partlist, performDeletion(), performDeletionCheck(), RangeVarGetAndCheckCreationNamespace(), RelationData::rd_rel, RelationGetPartitionDesc(), RelationGetRelid, RenameRelationInternal(), RestrictSearchPath(), SECURITY_RESTRICTED_OPERATION, SetUserIdAndSecContext(), SplitPartitionMoveRows(), sprintf, table_close(), and table_openrv().

Referenced by ATExecCmd().

◆ ATExecValidateConstraint()

static ObjectAddress ATExecValidateConstraint	(	List **	wqueue,
		Relation	rel,
		char *	constrName,
		bool	recurse,
		bool	recursing,
		LOCKMODE	lockmode
	)

static

Definition at line 13442 of file tablecmds.c.

{
    Relation    conrel;
    SysScanDesc scan;
    ScanKeyData skey[3];
    HeapTuple   tuple;
    Form_pg_constraint con;
    ObjectAddress address;
 
    conrel = table_open(ConstraintRelationId, RowExclusiveLock);
 
    /*
     * Find and check the target constraint
     */
    ScanKeyInit(&skey[0],
                Anum_pg_constraint_conrelid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationGetRelid(rel)));
    ScanKeyInit(&skey[1],
                Anum_pg_constraint_contypid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(InvalidOid));
    ScanKeyInit(&skey[2],
                Anum_pg_constraint_conname,
                BTEqualStrategyNumber, F_NAMEEQ,
                CStringGetDatum(constrName));
    scan = systable_beginscan(conrel, ConstraintRelidTypidNameIndexId,
                              true, NULL, 3, skey);
 
    /* There can be at most one matching row */
    if (!HeapTupleIsValid(tuple = systable_getnext(scan)))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_OBJECT),
                 errmsg("constraint \"%s\" of relation \"%s\" does not exist",
                        constrName, RelationGetRelationName(rel))));
 
    con = (Form_pg_constraint) GETSTRUCT(tuple);
    if (con->contype != CONSTRAINT_FOREIGN &&
        con->contype != CONSTRAINT_CHECK &&
        con->contype != CONSTRAINT_NOTNULL)
        ereport(ERROR,
                errcode(ERRCODE_WRONG_OBJECT_TYPE),
                errmsg("cannot validate constraint \"%s\" of relation \"%s\"",
                       constrName, RelationGetRelationName(rel)),
                errdetail("This operation is not supported for this type of constraint."));
 
    if (!con->conenforced)
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                 errmsg("cannot validate NOT ENFORCED constraint")));
 
    if (!con->convalidated)
    {
        if (con->contype == CONSTRAINT_FOREIGN)
        {
            QueueFKConstraintValidation(wqueue, conrel, rel, con->confrelid,
                                        tuple, lockmode);
        }
        else if (con->contype == CONSTRAINT_CHECK)
        {
            QueueCheckConstraintValidation(wqueue, conrel, rel, constrName,
                                           tuple, recurse, recursing, lockmode);
        }
        else if (con->contype == CONSTRAINT_NOTNULL)
        {
            QueueNNConstraintValidation(wqueue, conrel, rel,
                                        tuple, recurse, recursing, lockmode);
        }
 
        ObjectAddressSet(address, ConstraintRelationId, con->oid);
    }
    else
        address = InvalidObjectAddress; /* already validated */
 
    systable_endscan(scan);
 
    table_close(conrel, RowExclusiveLock);
 
    return address;
}

References BTEqualStrategyNumber, CStringGetDatum(), ereport, errcode(), errdetail(), errmsg, ERROR, fb(), Form_pg_constraint, GETSTRUCT(), HeapTupleIsValid, InvalidObjectAddress, InvalidOid, ObjectAddressSet, ObjectIdGetDatum(), QueueCheckConstraintValidation(), QueueFKConstraintValidation(), QueueNNConstraintValidation(), RelationGetRelationName, RelationGetRelid, RowExclusiveLock, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by ATExecCmd(), ATExecSetNotNull(), QueueCheckConstraintValidation(), and QueueNNConstraintValidation().

◆ ATGetQueueEntry()

static AlteredTableInfo * ATGetQueueEntry	(	List **	wqueue,
		Relation	rel
	)

static

Definition at line 6683 of file tablecmds.c.

{
    Oid         relid = RelationGetRelid(rel);
    AlteredTableInfo *tab;
    ListCell   *ltab;
 
    foreach(ltab, *wqueue)
    {
        tab = (AlteredTableInfo *) lfirst(ltab);
        if (tab->relid == relid)
            return tab;
    }
 
    /*
     * Not there, so add it.  Note that we make a copy of the relation's
     * existing descriptor before anything interesting can happen to it.
     */
    tab = palloc0_object(AlteredTableInfo);
    tab->relid = relid;
    tab->rel = NULL;            /* set later */
    tab->relkind = rel->rd_rel->relkind;
    tab->oldDesc = CreateTupleDescCopyConstr(RelationGetDescr(rel));
    tab->newAccessMethod = InvalidOid;
    tab->chgAccessMethod = false;
    tab->newTableSpace = InvalidOid;
    tab->newrelpersistence = RELPERSISTENCE_PERMANENT;
    tab->chgPersistence = false;
 
    *wqueue = lappend(*wqueue, tab);
 
    return tab;
}

References AlteredTableInfo::chgAccessMethod, AlteredTableInfo::chgPersistence, CreateTupleDescCopyConstr(), fb(), InvalidOid, lappend(), lfirst, AlteredTableInfo::newAccessMethod, AlteredTableInfo::newrelpersistence, AlteredTableInfo::newTableSpace, AlteredTableInfo::oldDesc, palloc0_object, RelationData::rd_rel, AlteredTableInfo::rel, RelationGetDescr, RelationGetRelid, AlteredTableInfo::relid, and AlteredTableInfo::relkind.

Referenced by addFkRecurseReferencing(), ATAddCheckNNConstraint(), ATExecAddColumn(), ATExecAlterCheckConstrEnforceability(), ATExecAlterFKConstrEnforceability(), ATPostAlterTypeParse(), ATPrepCmd(), createPartitionTable(), MergePartitionsMoveRows(), QueueCheckConstraintValidation(), QueueFKConstraintValidation(), QueueNNConstraintValidation(), QueuePartitionConstraintValidation(), set_attnotnull(), and SplitPartitionMoveRows().

◆ ATParseTransformCmd()

static AlterTableCmd * ATParseTransformCmd	(	List **	wqueue,
		AlteredTableInfo *	tab,
		Relation	rel,
		AlterTableCmd *	cmd,
		bool	recurse,
		LOCKMODE	lockmode,
		AlterTablePass	cur_pass,
		AlterTableUtilityContext *	context
	)

static

Definition at line 5830 of file tablecmds.c.

{
    AlterTableCmd *newcmd = NULL;
    AlterTableStmt *atstmt = makeNode(AlterTableStmt);
    List       *beforeStmts;
    List       *afterStmts;
    ListCell   *lc;
 
    /* Gin up an AlterTableStmt with just this subcommand and this table */
    atstmt->relation =
        makeRangeVar(get_namespace_name(RelationGetNamespace(rel)),
                     pstrdup(RelationGetRelationName(rel)),
                     -1);
    atstmt->relation->inh = recurse;
    atstmt->cmds = list_make1(cmd);
    atstmt->objtype = OBJECT_TABLE; /* needn't be picky here */
    atstmt->missing_ok = false;
 
    /* Transform the AlterTableStmt */
    atstmt = transformAlterTableStmt(RelationGetRelid(rel),
                                     atstmt,
                                     context->queryString,
                                     &beforeStmts,
                                     &afterStmts);
 
    /* Execute any statements that should happen before these subcommand(s) */
    foreach(lc, beforeStmts)
    {
        Node       *stmt = (Node *) lfirst(lc);
 
        ProcessUtilityForAlterTable(stmt, context);
        CommandCounterIncrement();
    }
 
    /* Examine the transformed subcommands and schedule them appropriately */
    foreach(lc, atstmt->cmds)
    {
        AlterTableCmd *cmd2 = lfirst_node(AlterTableCmd, lc);
        AlterTablePass pass;
 
        /*
         * This switch need only cover the subcommand types that can be added
         * by parse_utilcmd.c; otherwise, we'll use the default strategy of
         * executing the subcommand immediately, as a substitute for the
         * original subcommand.  (Note, however, that this does cause
         * AT_AddConstraint subcommands to be rescheduled into later passes,
         * which is important for index and foreign key constraints.)
         *
         * We assume we needn't do any phase-1 checks for added subcommands.
         */
        switch (cmd2->subtype)
        {
            case AT_AddIndex:
                pass = AT_PASS_ADD_INDEX;
                break;
            case AT_AddIndexConstraint:
                pass = AT_PASS_ADD_INDEXCONSTR;
                break;
            case AT_AddConstraint:
                /* Recursion occurs during execution phase */
                if (recurse)
                    cmd2->recurse = true;
                switch (castNode(Constraint, cmd2->def)->contype)
                {
                    case CONSTR_NOTNULL:
                        pass = AT_PASS_COL_ATTRS;
                        break;
                    case CONSTR_PRIMARY:
                    case CONSTR_UNIQUE:
                    case CONSTR_EXCLUSION:
                        pass = AT_PASS_ADD_INDEXCONSTR;
                        break;
                    default:
                        pass = AT_PASS_ADD_OTHERCONSTR;
                        break;
                }
                break;
            case AT_AlterColumnGenericOptions:
                /* This command never recurses */
                /* No command-specific prep needed */
                pass = AT_PASS_MISC;
                break;
            default:
                pass = cur_pass;
                break;
        }
 
        if (pass < cur_pass)
        {
            /* Cannot schedule into a pass we already finished */
            elog(ERROR, "ALTER TABLE scheduling failure: too late for pass %d",
                 pass);
        }
        else if (pass > cur_pass)
        {
            /* OK, queue it up for later */
            tab->subcmds[pass] = lappend(tab->subcmds[pass], cmd2);
        }
        else
        {
            /*
             * We should see at most one subcommand for the current pass,
             * which is the transformed version of the original subcommand.
             */
            if (newcmd == NULL && cmd->subtype == cmd2->subtype)
            {
                /* Found the transformed version of our subcommand */
                newcmd = cmd2;
            }
            else
                elog(ERROR, "ALTER TABLE scheduling failure: bogus item for pass %d",
                     pass);
        }
    }
 
    /* Queue up any after-statements to happen at the end */
    tab->afterStmts = list_concat(tab->afterStmts, afterStmts);
 
    return newcmd;
}

References AlteredTableInfo::afterStmts, AT_AddConstraint, AT_AddIndex, AT_AddIndexConstraint, AT_AlterColumnGenericOptions, AT_PASS_ADD_INDEX, AT_PASS_ADD_INDEXCONSTR, AT_PASS_ADD_OTHERCONSTR, AT_PASS_COL_ATTRS, AT_PASS_MISC, castNode, CommandCounterIncrement(), CONSTR_EXCLUSION, CONSTR_NOTNULL, CONSTR_PRIMARY, CONSTR_UNIQUE, elog, ERROR, fb(), get_namespace_name(), lappend(), lfirst, lfirst_node, list_concat(), list_make1, makeNode, makeRangeVar(), OBJECT_TABLE, ProcessUtilityForAlterTable(), pstrdup(), AlterTableUtilityContext::queryString, RelationGetNamespace, RelationGetRelationName, RelationGetRelid, stmt, AlteredTableInfo::subcmds, AlterTableCmd::subtype, and transformAlterTableStmt().

Referenced by ATExecAddColumn(), ATExecCmd(), and ATPrepCmd().

◆ ATPostAlterTypeCleanup()

static void ATPostAlterTypeCleanup	(	List **	wqueue,
		AlteredTableInfo *	tab,
		LOCKMODE	lockmode
	)

static

Definition at line 16141 of file tablecmds.c.

{
    ObjectAddress obj;
    ObjectAddresses *objects;
    ListCell   *def_item;
    ListCell   *oid_item;
 
    /*
     * Collect all the constraints and indexes to drop so we can process them
     * in a single call.  That way we don't have to worry about dependencies
     * among them.
     */
    objects = new_object_addresses();
 
    /*
     * Re-parse the index and constraint definitions, and attach them to the
     * appropriate work queue entries.  We do this before dropping because in
     * the case of a constraint on another table, we might not yet have
     * exclusive lock on the table the constraint is attached to, and we need
     * to get that before reparsing/dropping.  (That's possible at least for
     * FOREIGN KEY, CHECK, and EXCLUSION constraints; in non-FK cases it
     * requires a dependency on the target table's composite type in the other
     * table's constraint expressions.)
     *
     * We can't rely on the output of deparsing to tell us which relation to
     * operate on, because concurrent activity might have made the name
     * resolve differently.  Instead, we've got to use the OID of the
     * constraint or index we're processing to figure out which relation to
     * operate on.
     */
    forboth(oid_item, tab->changedConstraintOids,
            def_item, tab->changedConstraintDefs)
    {
        Oid         oldId = lfirst_oid(oid_item);
        HeapTuple   tup;
        Form_pg_constraint con;
        Oid         relid;
        Oid         confrelid;
        bool        conislocal;
 
        tup = SearchSysCache1(CONSTROID, ObjectIdGetDatum(oldId));
        if (!HeapTupleIsValid(tup)) /* should not happen */
            elog(ERROR, "cache lookup failed for constraint %u", oldId);
        con = (Form_pg_constraint) GETSTRUCT(tup);
        if (OidIsValid(con->conrelid))
            relid = con->conrelid;
        else
        {
            /* must be a domain constraint */
            relid = get_typ_typrelid(getBaseType(con->contypid));
            if (!OidIsValid(relid))
                elog(ERROR, "could not identify relation associated with constraint %u", oldId);
        }
        confrelid = con->confrelid;
        conislocal = con->conislocal;
        ReleaseSysCache(tup);
 
        ObjectAddressSet(obj, ConstraintRelationId, oldId);
        add_exact_object_address(&obj, objects);
 
        /*
         * If the constraint is inherited (only), we don't want to inject a
         * new definition here; it'll get recreated when
         * ATAddCheckNNConstraint recurses from adding the parent table's
         * constraint.  But we had to carry the info this far so that we can
         * drop the constraint below.
         */
        if (!conislocal)
            continue;
 
        /*
         * When rebuilding another table's constraint that references the
         * table we're modifying, we might not yet have any lock on the other
         * table, so get one now.  We'll need AccessExclusiveLock for the DROP
         * CONSTRAINT step, so there's no value in asking for anything weaker.
         */
        if (relid != tab->relid)
            LockRelationOid(relid, AccessExclusiveLock);
 
        ATPostAlterTypeParse(oldId, relid, confrelid,
                             (char *) lfirst(def_item),
                             wqueue, lockmode, tab->rewrite);
    }
    forboth(oid_item, tab->changedIndexOids,
            def_item, tab->changedIndexDefs)
    {
        Oid         oldId = lfirst_oid(oid_item);
        Oid         relid;
 
        relid = IndexGetRelation(oldId, false);
 
        /*
         * As above, make sure we have lock on the index's table if it's not
         * the same table.
         */
        if (relid != tab->relid)
            LockRelationOid(relid, AccessExclusiveLock);
 
        ATPostAlterTypeParse(oldId, relid, InvalidOid,
                             (char *) lfirst(def_item),
                             wqueue, lockmode, tab->rewrite);
 
        ObjectAddressSet(obj, RelationRelationId, oldId);
        add_exact_object_address(&obj, objects);
    }
 
    /* add dependencies for new statistics */
    forboth(oid_item, tab->changedStatisticsOids,
            def_item, tab->changedStatisticsDefs)
    {
        Oid         oldId = lfirst_oid(oid_item);
        Oid         relid;
 
        relid = StatisticsGetRelation(oldId, false);
 
        /*
         * As above, make sure we have lock on the statistics object's table
         * if it's not the same table.  However, we take
         * ShareUpdateExclusiveLock here, aligning with the lock level used in
         * CreateStatistics and RemoveStatisticsById.
         *
         * CAUTION: this should be done after all cases that grab
         * AccessExclusiveLock, else we risk causing deadlock due to needing
         * to promote our table lock.
         */
        if (relid != tab->relid)
            LockRelationOid(relid, ShareUpdateExclusiveLock);
 
        ATPostAlterTypeParse(oldId, relid, InvalidOid,
                             (char *) lfirst(def_item),
                             wqueue, lockmode, tab->rewrite);
 
        ObjectAddressSet(obj, StatisticExtRelationId, oldId);
        add_exact_object_address(&obj, objects);
    }
 
    /*
     * Queue up command to restore replica identity index marking
     */
    if (tab->replicaIdentityIndex)
    {
        AlterTableCmd *cmd = makeNode(AlterTableCmd);
        ReplicaIdentityStmt *subcmd = makeNode(ReplicaIdentityStmt);
 
        subcmd->identity_type = REPLICA_IDENTITY_INDEX;
        subcmd->name = tab->replicaIdentityIndex;
        cmd->subtype = AT_ReplicaIdentity;
        cmd->def = (Node *) subcmd;
 
        /* do it after indexes and constraints */
        tab->subcmds[AT_PASS_OLD_CONSTR] =
            lappend(tab->subcmds[AT_PASS_OLD_CONSTR], cmd);
    }
 
    /*
     * Queue up command to restore marking of index used for cluster.
     */
    if (tab->clusterOnIndex)
    {
        AlterTableCmd *cmd = makeNode(AlterTableCmd);
 
        cmd->subtype = AT_ClusterOn;
        cmd->name = tab->clusterOnIndex;
 
        /* do it after indexes and constraints */
        tab->subcmds[AT_PASS_OLD_CONSTR] =
            lappend(tab->subcmds[AT_PASS_OLD_CONSTR], cmd);
    }
 
    /*
     * It should be okay to use DROP_RESTRICT here, since nothing else should
     * be depending on these objects.
     */
    performMultipleDeletions(objects, DROP_RESTRICT, PERFORM_DELETION_INTERNAL);
 
    free_object_addresses(objects);
 
    /*
     * The objects will get recreated during subsequent passes over the work
     * queue.
     */
}

References AccessExclusiveLock, add_exact_object_address(), AT_ClusterOn, AT_PASS_OLD_CONSTR, AT_ReplicaIdentity, ATPostAlterTypeParse(), AlteredTableInfo::changedConstraintDefs, AlteredTableInfo::changedConstraintOids, AlteredTableInfo::changedIndexDefs, AlteredTableInfo::changedIndexOids, AlteredTableInfo::changedStatisticsDefs, AlteredTableInfo::changedStatisticsOids, AlteredTableInfo::clusterOnIndex, AlterTableCmd::def, DROP_RESTRICT, elog, ERROR, fb(), forboth, Form_pg_constraint, free_object_addresses(), get_typ_typrelid(), getBaseType(), GETSTRUCT(), HeapTupleIsValid, IndexGetRelation(), InvalidOid, lappend(), lfirst, lfirst_oid, LockRelationOid(), makeNode, AlterTableCmd::name, new_object_addresses(), ObjectAddressSet, ObjectIdGetDatum(), OidIsValid, PERFORM_DELETION_INTERNAL, performMultipleDeletions(), ReleaseSysCache(), AlteredTableInfo::relid, AlteredTableInfo::replicaIdentityIndex, AlteredTableInfo::rewrite, SearchSysCache1(), ShareUpdateExclusiveLock, StatisticsGetRelation(), AlteredTableInfo::subcmds, and AlterTableCmd::subtype.

Referenced by ATRewriteCatalogs().

◆ ATPostAlterTypeParse()

static void ATPostAlterTypeParse	(	Oid	oldId,
		Oid	oldRelId,
		Oid	refRelId,
		char *	cmd,
		List **	wqueue,
		LOCKMODE	lockmode,
		bool	rewrite
	)

static

Definition at line 16333 of file tablecmds.c.

{
    List       *raw_parsetree_list;
    List       *querytree_list;
    ListCell   *list_item;
    Relation    rel;
 
    /*
     * We expect that we will get only ALTER TABLE and CREATE INDEX
     * statements. Hence, there is no need to pass them through
     * parse_analyze_*() or the rewriter, but instead we need to pass them
     * through parse_utilcmd.c to make them ready for execution.
     */
    raw_parsetree_list = raw_parser(cmd, RAW_PARSE_DEFAULT);
    querytree_list = NIL;
    foreach(list_item, raw_parsetree_list)
    {
        RawStmt    *rs = lfirst_node(RawStmt, list_item);
        Node       *stmt = rs->stmt;
 
        if (IsA(stmt, IndexStmt))
            querytree_list = lappend(querytree_list,
                                     transformIndexStmt(oldRelId,
                                                        (IndexStmt *) stmt,
                                                        cmd));
        else if (IsA(stmt, AlterTableStmt))
        {
            List       *beforeStmts;
            List       *afterStmts;
 
            stmt = (Node *) transformAlterTableStmt(oldRelId,
                                                    (AlterTableStmt *) stmt,
                                                    cmd,
                                                    &beforeStmts,
                                                    &afterStmts);
            querytree_list = list_concat(querytree_list, beforeStmts);
            querytree_list = lappend(querytree_list, stmt);
            querytree_list = list_concat(querytree_list, afterStmts);
        }
        else if (IsA(stmt, CreateStatsStmt))
            querytree_list = lappend(querytree_list,
                                     transformStatsStmt(oldRelId,
                                                        (CreateStatsStmt *) stmt,
                                                        cmd));
        else
            querytree_list = lappend(querytree_list, stmt);
    }
 
    /* Caller should already have acquired whatever lock we need. */
    rel = relation_open(oldRelId, NoLock);
 
    /*
     * Attach each generated command to the proper place in the work queue.
     * Note this could result in creation of entirely new work-queue entries.
     *
     * Also note that we have to tweak the command subtypes, because it turns
     * out that re-creation of indexes and constraints has to act a bit
     * differently from initial creation.
     */
    foreach(list_item, querytree_list)
    {
        Node       *stm = (Node *) lfirst(list_item);
        AlteredTableInfo *tab;
 
        tab = ATGetQueueEntry(wqueue, rel);
 
        if (IsA(stm, IndexStmt))
        {
            IndexStmt  *stmt = (IndexStmt *) stm;
            AlterTableCmd *newcmd;
 
            if (!rewrite)
                TryReuseIndex(oldId, stmt);
            stmt->reset_default_tblspc = true;
            /* keep the index's comment */
            stmt->idxcomment = GetComment(oldId, RelationRelationId, 0);
 
            newcmd = makeNode(AlterTableCmd);
            newcmd->subtype = AT_ReAddIndex;
            newcmd->def = (Node *) stmt;
            tab->subcmds[AT_PASS_OLD_INDEX] =
                lappend(tab->subcmds[AT_PASS_OLD_INDEX], newcmd);
        }
        else if (IsA(stm, AlterTableStmt))
        {
            AlterTableStmt *stmt = (AlterTableStmt *) stm;
            ListCell   *lcmd;
 
            foreach(lcmd, stmt->cmds)
            {
                AlterTableCmd *cmd = lfirst_node(AlterTableCmd, lcmd);
 
                if (cmd->subtype == AT_AddIndex)
                {
                    IndexStmt  *indstmt;
                    Oid         indoid;
 
                    indstmt = castNode(IndexStmt, cmd->def);
                    indoid = get_constraint_index(oldId);
 
                    if (!rewrite)
                        TryReuseIndex(indoid, indstmt);
                    /* keep any comment on the index */
                    indstmt->idxcomment = GetComment(indoid,
                                                     RelationRelationId, 0);
                    indstmt->reset_default_tblspc = true;
 
                    cmd->subtype = AT_ReAddIndex;
                    tab->subcmds[AT_PASS_OLD_INDEX] =
                        lappend(tab->subcmds[AT_PASS_OLD_INDEX], cmd);
 
                    /* recreate any comment on the constraint */
                    RebuildConstraintComment(tab,
                                             AT_PASS_OLD_INDEX,
                                             oldId,
                                             rel,
                                             NIL,
                                             indstmt->idxname);
                }
                else if (cmd->subtype == AT_AddConstraint)
                {
                    Constraint *con = castNode(Constraint, cmd->def);
 
                    con->old_pktable_oid = refRelId;
                    /* rewriting neither side of a FK */
                    if (con->contype == CONSTR_FOREIGN &&
                        !rewrite && tab->rewrite == 0)
                        TryReuseForeignKey(oldId, con);
                    con->reset_default_tblspc = true;
                    cmd->subtype = AT_ReAddConstraint;
                    tab->subcmds[AT_PASS_OLD_CONSTR] =
                        lappend(tab->subcmds[AT_PASS_OLD_CONSTR], cmd);
 
                    /*
                     * Recreate any comment on the constraint.  If we have
                     * recreated a primary key, then transformTableConstraint
                     * has added an unnamed not-null constraint here; skip
                     * this in that case.
                     */
                    if (con->conname)
                        RebuildConstraintComment(tab,
                                                 AT_PASS_OLD_CONSTR,
                                                 oldId,
                                                 rel,
                                                 NIL,
                                                 con->conname);
                    else
                        Assert(con->contype == CONSTR_NOTNULL);
                }
                else
                    elog(ERROR, "unexpected statement subtype: %d",
                         (int) cmd->subtype);
            }
        }
        else if (IsA(stm, AlterDomainStmt))
        {
            AlterDomainStmt *stmt = (AlterDomainStmt *) stm;
 
            if (stmt->subtype == AD_AddConstraint)
            {
                Constraint *con = castNode(Constraint, stmt->def);
                AlterTableCmd *cmd = makeNode(AlterTableCmd);
 
                cmd->subtype = AT_ReAddDomainConstraint;
                cmd->def = (Node *) stmt;
                tab->subcmds[AT_PASS_OLD_CONSTR] =
                    lappend(tab->subcmds[AT_PASS_OLD_CONSTR], cmd);
 
                /* recreate any comment on the constraint */
                RebuildConstraintComment(tab,
                                         AT_PASS_OLD_CONSTR,
                                         oldId,
                                         NULL,
                                         stmt->typeName,
                                         con->conname);
            }
            else
                elog(ERROR, "unexpected statement subtype: %d",
                     (int) stmt->subtype);
        }
        else if (IsA(stm, CreateStatsStmt))
        {
            CreateStatsStmt *stmt = (CreateStatsStmt *) stm;
            AlterTableCmd *newcmd;
 
            /* keep the statistics object's comment */
            stmt->stxcomment = GetComment(oldId, StatisticExtRelationId, 0);
 
            newcmd = makeNode(AlterTableCmd);
            newcmd->subtype = AT_ReAddStatistics;
            newcmd->def = (Node *) stmt;
            tab->subcmds[AT_PASS_MISC] =
                lappend(tab->subcmds[AT_PASS_MISC], newcmd);
        }
        else
            elog(ERROR, "unexpected statement type: %d",
                 (int) nodeTag(stm));
    }
 
    relation_close(rel, NoLock);
}

References AD_AddConstraint, Assert, AT_AddConstraint, AT_AddIndex, AT_PASS_MISC, AT_PASS_OLD_CONSTR, AT_PASS_OLD_INDEX, AT_ReAddConstraint, AT_ReAddDomainConstraint, AT_ReAddIndex, AT_ReAddStatistics, ATGetQueueEntry(), castNode, Constraint::conname, CONSTR_FOREIGN, CONSTR_NOTNULL, Constraint::contype, AlterTableCmd::def, elog, ERROR, fb(), get_constraint_index(), GetComment(), IsA, lappend(), lfirst, lfirst_node, list_concat(), makeNode, NIL, nodeTag, NoLock, Constraint::old_pktable_oid, RAW_PARSE_DEFAULT, raw_parser(), RebuildConstraintComment(), relation_close(), relation_open(), Constraint::reset_default_tblspc, AlteredTableInfo::rewrite, RawStmt::stmt, stmt, AlteredTableInfo::subcmds, AlterTableCmd::subtype, transformAlterTableStmt(), transformIndexStmt(), transformStatsStmt(), TryReuseForeignKey(), and TryReuseIndex().

Referenced by ATPostAlterTypeCleanup().

◆ ATPrepAddColumn()

static void ATPrepAddColumn	(	List **	wqueue,
		Relation	rel,
		bool	recurse,
		bool	recursing,
		bool	is_view,
		AlterTableCmd *	cmd,
		LOCKMODE	lockmode,
		AlterTableUtilityContext *	context
	)

static

Definition at line 7334 of file tablecmds.c.

{
    if (rel->rd_rel->reloftype && !recursing)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot add column to typed table")));
 
    if (rel->rd_rel->relkind == RELKIND_COMPOSITE_TYPE)
        ATTypedTableRecursion(wqueue, rel, cmd, lockmode, context);
 
    if (recurse && !is_view)
        cmd->recurse = true;
}

References ATTypedTableRecursion(), ereport, errcode(), errmsg, ERROR, fb(), RelationData::rd_rel, and AlterTableCmd::recurse.

Referenced by ATPrepCmd().

◆ ATPrepAddPrimaryKey()

static void ATPrepAddPrimaryKey	(	List **	wqueue,
		Relation	rel,
		AlterTableCmd *	cmd,
		bool	recurse,
		LOCKMODE	lockmode,
		AlterTableUtilityContext *	context
	)

static

Definition at line 9634 of file tablecmds.c.

{
    Constraint *pkconstr;
    List       *children = NIL;
    bool        got_children = false;
 
    pkconstr = castNode(Constraint, cmd->def);
    if (pkconstr->contype != CONSTR_PRIMARY)
        return;
 
    /* Verify that columns are not-null, or request that they be made so */
    foreach_node(String, column, pkconstr->keys)
    {
        AlterTableCmd *newcmd;
        Constraint *nnconstr;
        HeapTuple   tuple;
 
        /*
         * First check if a suitable constraint exists.  If it does, we don't
         * need to request another one.  We do need to bail out if it's not
         * valid, though.
         */
        tuple = findNotNullConstraint(RelationGetRelid(rel), strVal(column));
        if (tuple != NULL)
        {
            verifyNotNullPKCompatible(tuple, strVal(column));
 
            /* All good with this one; don't request another */
            heap_freetuple(tuple);
            continue;
        }
        else if (!recurse)
        {
            /*
             * No constraint on this column.  Asked not to recurse, we won't
             * create one here, but verify that all children have one.
             */
            if (!got_children)
            {
                children = find_inheritance_children(RelationGetRelid(rel),
                                                     lockmode);
                /* only search for children on the first time through */
                got_children = true;
            }
 
            foreach_oid(childrelid, children)
            {
                HeapTuple   tup;
 
                tup = findNotNullConstraint(childrelid, strVal(column));
                if (!tup)
                    ereport(ERROR,
                            errmsg("column \"%s\" of table \"%s\" is not marked NOT NULL",
                                   strVal(column), get_rel_name(childrelid)));
                /* verify it's good enough */
                verifyNotNullPKCompatible(tup, strVal(column));
            }
        }
 
        /* This column is not already not-null, so add it to the queue */
        nnconstr = makeNotNullConstraint(column);
 
        newcmd = makeNode(AlterTableCmd);
        newcmd->subtype = AT_AddConstraint;
        /* note we force recurse=true here; see above */
        newcmd->recurse = true;
        newcmd->def = (Node *) nnconstr;
 
        ATPrepCmd(wqueue, rel, newcmd, true, false, lockmode, context);
    }
}

References AT_AddConstraint, ATPrepCmd(), castNode, CONSTR_PRIMARY, AlterTableCmd::def, ereport, errmsg, ERROR, fb(), find_inheritance_children(), findNotNullConstraint(), foreach_node, foreach_oid, get_rel_name(), heap_freetuple(), makeNode, makeNotNullConstraint(), NIL, RelationGetRelid, strVal, and verifyNotNullPKCompatible().

Referenced by ATPrepCmd().

◆ ATPrepAlterColumnType()

static void ATPrepAlterColumnType	(	List **	wqueue,
		AlteredTableInfo *	tab,
		Relation	rel,
		bool	recurse,
		bool	recursing,
		AlterTableCmd *	cmd,
		LOCKMODE	lockmode,
		AlterTableUtilityContext *	context
	)

static

Definition at line 14910 of file tablecmds.c.

{
    char       *colName = cmd->name;
    ColumnDef  *def = (ColumnDef *) cmd->def;
    TypeName   *typeName = def->typeName;
    Node       *transform = def->cooked_default;
    HeapTuple   tuple;
    Form_pg_attribute attTup;
    AttrNumber  attnum;
    Oid         targettype;
    int32       targettypmod;
    Oid         targetcollid;
    NewColumnValue *newval;
    ParseState *pstate = make_parsestate(NULL);
    AclResult   aclresult;
    bool        is_expr;
 
    pstate->p_sourcetext = context->queryString;
 
    if (rel->rd_rel->reloftype && !recursing)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot alter column type of typed table"),
                 parser_errposition(pstate, def->location)));
 
    /* lookup the attribute so we can check inheritance status */
    tuple = SearchSysCacheAttName(RelationGetRelid(rel), colName);
    if (!HeapTupleIsValid(tuple))
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" does not exist",
                        colName, RelationGetRelationName(rel)),
                 parser_errposition(pstate, def->location)));
    attTup = (Form_pg_attribute) GETSTRUCT(tuple);
    attnum = attTup->attnum;
 
    /* Can't alter a system attribute */
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter system column \"%s\"", colName),
                 parser_errposition(pstate, def->location)));
 
    /*
     * Cannot specify USING when altering type of a generated column, because
     * that would violate the generation expression.
     */
    if (attTup->attgenerated && def->cooked_default)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
                 errmsg("cannot specify USING when altering type of generated column"),
                 errdetail("Column \"%s\" is a generated column.", colName),
                 parser_errposition(pstate, def->location)));
 
    /*
     * Don't alter inherited columns.  At outer level, there had better not be
     * any inherited definition; when recursing, we assume this was checked at
     * the parent level (see below).
     */
    if (attTup->attinhcount > 0 && !recursing)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("cannot alter inherited column \"%s\"", colName),
                 parser_errposition(pstate, def->location)));
 
    /* Don't alter columns used in the partition key */
    if (has_partition_attrs(rel,
                            bms_make_singleton(attnum - FirstLowInvalidHeapAttributeNumber),
                            &is_expr))
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("cannot alter column \"%s\" because it is part of the partition key of relation \"%s\"",
                        colName, RelationGetRelationName(rel)),
                 parser_errposition(pstate, def->location)));
 
    /* Look up the target type */
    typenameTypeIdAndMod(pstate, typeName, &targettype, &targettypmod);
 
    aclresult = object_aclcheck(TypeRelationId, targettype, GetUserId(), ACL_USAGE);
    if (aclresult != ACLCHECK_OK)
        aclcheck_error_type(aclresult, targettype);
 
    /* And the collation */
    targetcollid = GetColumnDefCollation(pstate, def, targettype);
 
    /* make sure datatype is legal for a column */
    CheckAttributeType(colName, targettype, targetcollid,
                       list_make1_oid(rel->rd_rel->reltype),
                       (attTup->attgenerated == ATTRIBUTE_GENERATED_VIRTUAL ? CHKATYPE_IS_VIRTUAL : 0));
 
    if (attTup->attgenerated == ATTRIBUTE_GENERATED_VIRTUAL)
    {
        /* do nothing */
    }
    else if (tab->relkind == RELKIND_RELATION ||
             tab->relkind == RELKIND_PARTITIONED_TABLE)
    {
        /*
         * Set up an expression to transform the old data value to the new
         * type. If a USING option was given, use the expression as
         * transformed by transformAlterTableStmt, else just take the old
         * value and try to coerce it.  We do this first so that type
         * incompatibility can be detected before we waste effort, and because
         * we need the expression to be parsed against the original table row
         * type.
         */
        if (!transform)
        {
            transform = (Node *) makeVar(1, attnum,
                                         attTup->atttypid, attTup->atttypmod,
                                         attTup->attcollation,
                                         0);
        }
 
        transform = coerce_to_target_type(pstate,
                                          transform, exprType(transform),
                                          targettype, targettypmod,
                                          COERCION_ASSIGNMENT,
                                          COERCE_IMPLICIT_CAST,
                                          -1);
        if (transform == NULL)
        {
            /* error text depends on whether USING was specified or not */
            if (def->cooked_default != NULL)
                ereport(ERROR,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("result of USING clause for column \"%s\""
                                " cannot be cast automatically to type %s",
                                colName, format_type_be(targettype)),
                         errhint("You might need to add an explicit cast.")));
            else
                ereport(ERROR,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("column \"%s\" cannot be cast automatically to type %s",
                                colName, format_type_be(targettype)),
                         !attTup->attgenerated ?
                /* translator: USING is SQL, don't translate it */
                         errhint("You might need to specify \"USING %s::%s\".",
                                 quote_identifier(colName),
                                 format_type_with_typemod(targettype,
                                                          targettypmod)) : 0));
        }
 
        /* Fix collations after all else */
        assign_expr_collations(pstate, transform);
 
        /* Expand virtual generated columns in the expr. */
        transform = expand_generated_columns_in_expr(transform, rel, 1);
 
        /* Plan the expr now so we can accurately assess the need to rewrite. */
        transform = (Node *) expression_planner((Expr *) transform);
 
        /*
         * Add a work queue item to make ATRewriteTable update the column
         * contents.
         */
        newval = palloc0_object(NewColumnValue);
        newval->attnum = attnum;
        newval->expr = (Expr *) transform;
        newval->is_generated = false;
 
        tab->newvals = lappend(tab->newvals, newval);
        if (ATColumnChangeRequiresRewrite(transform, attnum))
            tab->rewrite |= AT_REWRITE_COLUMN_REWRITE;
    }
    else if (transform)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("\"%s\" is not a table",
                        RelationGetRelationName(rel))));
 
    if (!RELKIND_HAS_STORAGE(tab->relkind) || attTup->attgenerated == ATTRIBUTE_GENERATED_VIRTUAL)
    {
        /*
         * For relations or columns without storage, do this check now.
         * Regular tables will check it later when the table is being
         * rewritten.
         */
        find_composite_type_dependencies(rel->rd_rel->reltype, rel, NULL);
    }
 
    ReleaseSysCache(tuple);
 
    /*
     * Recurse manually by queueing a new command for each child, if
     * necessary. We cannot apply ATSimpleRecursion here because we need to
     * remap attribute numbers in the USING expression, if any.
     *
     * If we are told not to recurse, there had better not be any child
     * tables; else the alter would put them out of step.
     */
    if (recurse)
    {
        Oid         relid = RelationGetRelid(rel);
        List       *child_oids,
                   *child_numparents;
        ListCell   *lo,
                   *li;
 
        child_oids = find_all_inheritors(relid, lockmode,
                                         &child_numparents);
 
        /*
         * find_all_inheritors does the recursive search of the inheritance
         * hierarchy, so all we have to do is process all of the relids in the
         * list that it returns.
         */
        forboth(lo, child_oids, li, child_numparents)
        {
            Oid         childrelid = lfirst_oid(lo);
            int         numparents = lfirst_int(li);
            Relation    childrel;
            HeapTuple   childtuple;
            Form_pg_attribute childattTup;
 
            if (childrelid == relid)
                continue;
 
            /* find_all_inheritors already got lock */
            childrel = relation_open(childrelid, NoLock);
            CheckAlterTableIsSafe(childrel);
 
            /*
             * Verify that the child doesn't have any inherited definitions of
             * this column that came from outside this inheritance hierarchy.
             * (renameatt makes a similar test, though in a different way
             * because of its different recursion mechanism.)
             */
            childtuple = SearchSysCacheAttName(RelationGetRelid(childrel),
                                               colName);
            if (!HeapTupleIsValid(childtuple))
                ereport(ERROR,
                        (errcode(ERRCODE_UNDEFINED_COLUMN),
                         errmsg("column \"%s\" of relation \"%s\" does not exist",
                                colName, RelationGetRelationName(childrel))));
            childattTup = (Form_pg_attribute) GETSTRUCT(childtuple);
 
            if (childattTup->attinhcount > numparents)
                ereport(ERROR,
                        (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                         errmsg("cannot alter inherited column \"%s\" of relation \"%s\"",
                                colName, RelationGetRelationName(childrel))));
 
            ReleaseSysCache(childtuple);
 
            /*
             * Remap the attribute numbers.  If no USING expression was
             * specified, there is no need for this step.
             */
            if (def->cooked_default)
            {
                AttrMap    *attmap;
                bool        found_whole_row;
 
                /* create a copy to scribble on */
                cmd = copyObject(cmd);
 
                attmap = build_attrmap_by_name(RelationGetDescr(childrel),
                                               RelationGetDescr(rel),
                                               false);
                ((ColumnDef *) cmd->def)->cooked_default =
                    map_variable_attnos(def->cooked_default,
                                        1, 0,
                                        attmap,
                                        InvalidOid, &found_whole_row);
                if (found_whole_row)
                    ereport(ERROR,
                            (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                             errmsg("cannot convert whole-row table reference"),
                             errdetail("USING expression contains a whole-row table reference.")));
                pfree(attmap);
            }
            ATPrepCmd(wqueue, childrel, cmd, false, true, lockmode, context);
            relation_close(childrel, NoLock);
        }
    }
    else if (!recursing &&
             find_inheritance_children(RelationGetRelid(rel), NoLock) != NIL)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("type of inherited column \"%s\" must be changed in child tables too",
                        colName)));
 
    if (tab->relkind == RELKIND_COMPOSITE_TYPE)
        ATTypedTableRecursion(wqueue, rel, cmd, lockmode, context);
}

References ACL_USAGE, aclcheck_error_type(), ACLCHECK_OK, assign_expr_collations(), AT_REWRITE_COLUMN_REWRITE, ATColumnChangeRequiresRewrite(), ATPrepCmd(), attnum, ATTypedTableRecursion(), bms_make_singleton(), build_attrmap_by_name(), CheckAlterTableIsSafe(), CheckAttributeType(), CHKATYPE_IS_VIRTUAL, COERCE_IMPLICIT_CAST, coerce_to_target_type(), COERCION_ASSIGNMENT, ColumnDef::cooked_default, copyObject, AlterTableCmd::def, ereport, errcode(), errdetail(), errhint(), errmsg, ERROR, expand_generated_columns_in_expr(), expression_planner(), exprType(), fb(), find_all_inheritors(), find_composite_type_dependencies(), find_inheritance_children(), FirstLowInvalidHeapAttributeNumber, forboth, format_type_be(), format_type_with_typemod(), GetColumnDefCollation(), GETSTRUCT(), GetUserId(), has_partition_attrs(), HeapTupleIsValid, InvalidOid, lappend(), lfirst_int, lfirst_oid, list_make1_oid, ColumnDef::location, make_parsestate(), makeVar(), map_variable_attnos(), AlterTableCmd::name, newval, AlteredTableInfo::newvals, NIL, NoLock, object_aclcheck(), ParseState::p_sourcetext, palloc0_object, parser_errposition(), pfree(), AlterTableUtilityContext::queryString, quote_identifier(), RelationData::rd_rel, relation_close(), relation_open(), RelationGetDescr, RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), AlteredTableInfo::relkind, AlteredTableInfo::rewrite, SearchSysCacheAttName(), ColumnDef::typeName, and typenameTypeIdAndMod().

Referenced by ATPrepCmd().

◆ ATPrepChangeInherit()

static void ATPrepChangeInherit ( Relation child_rel )

static

Definition at line 17949 of file tablecmds.c.

{
    if (child_rel->rd_rel->reloftype)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot change inheritance of typed table")));
 
    if (child_rel->rd_rel->relispartition)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot change inheritance of a partition")));
}

References ereport, errcode(), errmsg, ERROR, and fb().

Referenced by ATPrepCmd().

◆ ATPrepChangePersistence()

static void ATPrepChangePersistence	(	AlteredTableInfo *	tab,
		Relation	rel,
		bool	toLogged
	)

static

Definition at line 19524 of file tablecmds.c.

{
    Relation    pg_constraint;
    HeapTuple   tuple;
    SysScanDesc scan;
    ScanKeyData skey[1];
 
    /*
     * Disallow changing status for a temp table.  Also verify whether we can
     * get away with doing nothing; in such cases we don't need to run the
     * checks below, either.
     */
    switch (rel->rd_rel->relpersistence)
    {
        case RELPERSISTENCE_TEMP:
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                     errmsg("cannot change logged status of table \"%s\" because it is temporary",
                            RelationGetRelationName(rel)),
                     errtable(rel)));
            break;
        case RELPERSISTENCE_PERMANENT:
            if (toLogged)
                /* nothing to do */
                return;
            break;
        case RELPERSISTENCE_UNLOGGED:
            if (!toLogged)
                /* nothing to do */
                return;
            break;
    }
 
    /*
     * Check that the table is not part of any publication when changing to
     * UNLOGGED, as UNLOGGED tables can't be published.
     */
    if (!toLogged &&
        GetRelationIncludedPublications(RelationGetRelid(rel)) != NIL)
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                 errmsg("cannot change table \"%s\" to unlogged because it is part of a publication",
                        RelationGetRelationName(rel)),
                 errdetail("Unlogged relations cannot be replicated.")));
 
    /*
     * Check existing foreign key constraints to preserve the invariant that
     * permanent tables cannot reference unlogged ones.  Self-referencing
     * foreign keys can safely be ignored.
     */
    pg_constraint = table_open(ConstraintRelationId, AccessShareLock);
 
    /*
     * Scan conrelid if changing to permanent, else confrelid.  This also
     * determines whether a useful index exists.
     */
    ScanKeyInit(&skey[0],
                toLogged ? Anum_pg_constraint_conrelid :
                Anum_pg_constraint_confrelid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationGetRelid(rel)));
    scan = systable_beginscan(pg_constraint,
                              toLogged ? ConstraintRelidTypidNameIndexId : InvalidOid,
                              true, NULL, 1, skey);
 
    while (HeapTupleIsValid(tuple = systable_getnext(scan)))
    {
        Form_pg_constraint con = (Form_pg_constraint) GETSTRUCT(tuple);
 
        if (con->contype == CONSTRAINT_FOREIGN)
        {
            Oid         foreignrelid;
            Relation    foreignrel;
 
            /* the opposite end of what we used as scankey */
            foreignrelid = toLogged ? con->confrelid : con->conrelid;
 
            /* ignore if self-referencing */
            if (RelationGetRelid(rel) == foreignrelid)
                continue;
 
            foreignrel = relation_open(foreignrelid, AccessShareLock);
 
            if (toLogged)
            {
                if (!RelationIsPermanent(foreignrel))
                    ereport(ERROR,
                            (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                             errmsg("could not change table \"%s\" to logged because it references unlogged table \"%s\"",
                                    RelationGetRelationName(rel),
                                    RelationGetRelationName(foreignrel)),
                             errtableconstraint(rel, NameStr(con->conname))));
            }
            else
            {
                if (RelationIsPermanent(foreignrel))
                    ereport(ERROR,
                            (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                             errmsg("could not change table \"%s\" to unlogged because it references logged table \"%s\"",
                                    RelationGetRelationName(rel),
                                    RelationGetRelationName(foreignrel)),
                             errtableconstraint(rel, NameStr(con->conname))));
            }
 
            relation_close(foreignrel, AccessShareLock);
        }
    }
 
    systable_endscan(scan);
 
    table_close(pg_constraint, AccessShareLock);
 
    /* force rewrite if necessary; see comment in ATRewriteTables */
    tab->rewrite |= AT_REWRITE_ALTER_PERSISTENCE;
    if (toLogged)
        tab->newrelpersistence = RELPERSISTENCE_PERMANENT;
    else
        tab->newrelpersistence = RELPERSISTENCE_UNLOGGED;
    tab->chgPersistence = true;
}

References AccessShareLock, AT_REWRITE_ALTER_PERSISTENCE, BTEqualStrategyNumber, AlteredTableInfo::chgPersistence, ereport, errcode(), errdetail(), errmsg, ERROR, errtable(), errtableconstraint(), fb(), Form_pg_constraint, GetRelationIncludedPublications(), GETSTRUCT(), HeapTupleIsValid, InvalidOid, NameStr, AlteredTableInfo::newrelpersistence, NIL, ObjectIdGetDatum(), RelationData::rd_rel, relation_close(), relation_open(), RelationGetRelationName, RelationGetRelid, RelationIsPermanent, AlteredTableInfo::rewrite, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by ATPrepCmd().

◆ ATPrepCmd()

static void ATPrepCmd	(	List **	wqueue,
		Relation	rel,
		AlterTableCmd *	cmd,
		bool	recurse,
		bool	recursing,
		LOCKMODE	lockmode,
		AlterTableUtilityContext *	context
	)

static

Definition at line 5002 of file tablecmds.c.

{
    AlteredTableInfo *tab;
    AlterTablePass pass = AT_PASS_UNSET;
 
    /* Find or create work queue entry for this table */
    tab = ATGetQueueEntry(wqueue, rel);
 
    /*
     * Disallow any ALTER TABLE other than ALTER TABLE DETACH FINALIZE on
     * partitions that are pending detach.
     */
    if (rel->rd_rel->relispartition &&
        cmd->subtype != AT_DetachPartitionFinalize &&
        PartitionHasPendingDetach(RelationGetRelid(rel)))
        ereport(ERROR,
                errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                errmsg("cannot alter partition \"%s\" with an incomplete detach",
                       RelationGetRelationName(rel)),
                errhint("Use ALTER TABLE ... DETACH PARTITION ... FINALIZE to complete the pending detach operation."));
 
    /*
     * Copy the original subcommand for each table, so we can scribble on it.
     * This avoids conflicts when different child tables need to make
     * different parse transformations (for example, the same column may have
     * different column numbers in different children).
     */
    cmd = copyObject(cmd);
 
    /*
     * Do permissions and relkind checking, recursion to child tables if
     * needed, and any additional phase-1 processing needed.  (But beware of
     * adding any processing that looks at table details that another
     * subcommand could change.  In some cases we reject multiple subcommands
     * that could try to change the same state in contrary ways.)
     */
    switch (cmd->subtype)
    {
        case AT_AddColumn:      /* ADD COLUMN */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE |
                                ATT_COMPOSITE_TYPE | ATT_FOREIGN_TABLE);
            ATPrepAddColumn(wqueue, rel, recurse, recursing, false, cmd,
                            lockmode, context);
            /* Recursion occurs during execution phase */
            pass = AT_PASS_ADD_COL;
            break;
        case AT_AddColumnToView:    /* add column via CREATE OR REPLACE VIEW */
            ATSimplePermissions(cmd->subtype, rel, ATT_VIEW);
            ATPrepAddColumn(wqueue, rel, recurse, recursing, true, cmd,
                            lockmode, context);
            /* Recursion occurs during execution phase */
            pass = AT_PASS_ADD_COL;
            break;
        case AT_ColumnDefault:  /* ALTER COLUMN DEFAULT */
 
            /*
             * We allow defaults on views so that INSERT into a view can have
             * default-ish behavior.  This works because the rewriter
             * substitutes default values into INSERTs before it expands
             * rules.
             */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_VIEW |
                                ATT_FOREIGN_TABLE);
            ATSimpleRecursion(wqueue, rel, cmd, recurse, lockmode, context);
            /* No command-specific prep needed */
            pass = cmd->def ? AT_PASS_ADD_OTHERCONSTR : AT_PASS_DROP;
            break;
        case AT_CookedColumnDefault:    /* add a pre-cooked default */
            /* This is currently used only in CREATE TABLE */
            /* (so the permission check really isn't necessary) */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
            /* This command never recurses */
            pass = AT_PASS_ADD_OTHERCONSTR;
            break;
        case AT_AddIdentity:
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_VIEW |
                                ATT_FOREIGN_TABLE);
            /* Set up recursion for phase 2; no other prep needed */
            if (recurse)
                cmd->recurse = true;
            pass = AT_PASS_ADD_OTHERCONSTR;
            break;
        case AT_SetIdentity:
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_VIEW |
                                ATT_FOREIGN_TABLE);
            /* Set up recursion for phase 2; no other prep needed */
            if (recurse)
                cmd->recurse = true;
            /* This should run after AddIdentity, so do it in MISC pass */
            pass = AT_PASS_MISC;
            break;
        case AT_DropIdentity:
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_VIEW |
                                ATT_FOREIGN_TABLE);
            /* Set up recursion for phase 2; no other prep needed */
            if (recurse)
                cmd->recurse = true;
            pass = AT_PASS_DROP;
            break;
        case AT_DropNotNull:    /* ALTER COLUMN DROP NOT NULL */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
            /* Set up recursion for phase 2; no other prep needed */
            if (recurse)
                cmd->recurse = true;
            pass = AT_PASS_DROP;
            break;
        case AT_SetNotNull:     /* ALTER COLUMN SET NOT NULL */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
            /* Set up recursion for phase 2; no other prep needed */
            if (recurse)
                cmd->recurse = true;
            pass = AT_PASS_COL_ATTRS;
            break;
        case AT_SetExpression:  /* ALTER COLUMN SET EXPRESSION */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
            ATSimpleRecursion(wqueue, rel, cmd, recurse, lockmode, context);
            pass = AT_PASS_SET_EXPRESSION;
            break;
        case AT_DropExpression: /* ALTER COLUMN DROP EXPRESSION */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
            ATSimpleRecursion(wqueue, rel, cmd, recurse, lockmode, context);
            ATPrepDropExpression(rel, cmd, recurse, recursing, lockmode);
            pass = AT_PASS_DROP;
            break;
        case AT_SetStatistics:  /* ALTER COLUMN SET STATISTICS */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_MATVIEW |
                                ATT_INDEX | ATT_PARTITIONED_INDEX | ATT_FOREIGN_TABLE);
            ATSimpleRecursion(wqueue, rel, cmd, recurse, lockmode, context);
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        case AT_SetOptions:     /* ALTER COLUMN SET ( options ) */
        case AT_ResetOptions:   /* ALTER COLUMN RESET ( options ) */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE |
                                ATT_MATVIEW | ATT_FOREIGN_TABLE);
            /* This command never recurses */
            pass = AT_PASS_MISC;
            break;
        case AT_SetStorage:     /* ALTER COLUMN SET STORAGE */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE |
                                ATT_MATVIEW | ATT_FOREIGN_TABLE);
            ATSimpleRecursion(wqueue, rel, cmd, recurse, lockmode, context);
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        case AT_SetCompression: /* ALTER COLUMN SET COMPRESSION */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_MATVIEW);
            /* This command never recurses */
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        case AT_DropColumn:     /* DROP COLUMN */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE |
                                ATT_COMPOSITE_TYPE | ATT_FOREIGN_TABLE);
            ATPrepDropColumn(wqueue, rel, recurse, recursing, cmd,
                             lockmode, context);
            /* Recursion occurs during execution phase */
            pass = AT_PASS_DROP;
            break;
        case AT_AddIndex:       /* ADD INDEX */
            ATSimplePermissions(cmd->subtype, rel, ATT_TABLE | ATT_PARTITIONED_TABLE);
            /* This command never recurses */
            /* No command-specific prep needed */
            pass = AT_PASS_ADD_INDEX;
            break;
        case AT_AddConstraint:  /* ADD CONSTRAINT */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
            ATPrepAddPrimaryKey(wqueue, rel, cmd, recurse, lockmode, context);
            if (recurse)
            {
                /* recurses at exec time; lock descendants and set flag */
                (void) find_all_inheritors(RelationGetRelid(rel), lockmode, NULL);
                cmd->recurse = true;
            }
            pass = AT_PASS_ADD_CONSTR;
            break;
        case AT_AddIndexConstraint: /* ADD CONSTRAINT USING INDEX */
            ATSimplePermissions(cmd->subtype, rel, ATT_TABLE | ATT_PARTITIONED_TABLE);
            /* This command never recurses */
            /* No command-specific prep needed */
            pass = AT_PASS_ADD_INDEXCONSTR;
            break;
        case AT_DropConstraint: /* DROP CONSTRAINT */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
            ATCheckPartitionsNotInUse(rel, lockmode);
            /* Other recursion occurs during execution phase */
            /* No command-specific prep needed except saving recurse flag */
            if (recurse)
                cmd->recurse = true;
            pass = AT_PASS_DROP;
            break;
        case AT_AlterColumnType:    /* ALTER COLUMN TYPE */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE |
                                ATT_COMPOSITE_TYPE | ATT_FOREIGN_TABLE);
            /* See comments for ATPrepAlterColumnType */
            cmd = ATParseTransformCmd(wqueue, tab, rel, cmd, recurse, lockmode,
                                      AT_PASS_UNSET, context);
            Assert(cmd != NULL);
            /* Performs own recursion */
            ATPrepAlterColumnType(wqueue, tab, rel, recurse, recursing, cmd,
                                  lockmode, context);
            pass = AT_PASS_ALTER_TYPE;
            break;
        case AT_AlterColumnGenericOptions:
            ATSimplePermissions(cmd->subtype, rel, ATT_FOREIGN_TABLE);
            /* This command never recurses */
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        case AT_ChangeOwner:    /* ALTER OWNER */
            /* This command never recurses */
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        case AT_ClusterOn:      /* CLUSTER ON */
        case AT_DropCluster:    /* SET WITHOUT CLUSTER */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_MATVIEW);
            /* These commands never recurse */
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        case AT_SetLogged:      /* SET LOGGED */
        case AT_SetUnLogged:    /* SET UNLOGGED */
            ATSimplePermissions(cmd->subtype, rel, ATT_TABLE | ATT_SEQUENCE);
            if (tab->chgPersistence)
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("cannot change persistence setting twice")));
            ATPrepChangePersistence(tab, rel, cmd->subtype == AT_SetLogged);
            pass = AT_PASS_MISC;
            break;
        case AT_DropOids:       /* SET WITHOUT OIDS */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
            pass = AT_PASS_DROP;
            break;
        case AT_SetAccessMethod:    /* SET ACCESS METHOD */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_MATVIEW);
 
            /* check if another access method change was already requested */
            if (tab->chgAccessMethod)
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("cannot have multiple SET ACCESS METHOD subcommands")));
 
            ATPrepSetAccessMethod(tab, rel, cmd->name);
            pass = AT_PASS_MISC;    /* does not matter; no work in Phase 2 */
            break;
        case AT_SetTableSpace:  /* SET TABLESPACE */
            ATSimplePermissions(cmd->subtype, rel, ATT_TABLE | ATT_PARTITIONED_TABLE |
                                ATT_MATVIEW | ATT_INDEX | ATT_PARTITIONED_INDEX);
            /* This command never recurses */
            ATPrepSetTableSpace(tab, rel, cmd->name, lockmode);
            pass = AT_PASS_MISC;    /* doesn't actually matter */
            break;
        case AT_SetRelOptions:  /* SET (...) */
        case AT_ResetRelOptions:    /* RESET (...) */
        case AT_ReplaceRelOptions:  /* reset them all, then set just these */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_VIEW |
                                ATT_MATVIEW | ATT_INDEX);
            /* This command never recurses */
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        case AT_AddInherit:     /* INHERIT */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_FOREIGN_TABLE);
            /* This command never recurses */
            ATPrepChangeInherit(rel);
            pass = AT_PASS_MISC;
            break;
        case AT_DropInherit:    /* NO INHERIT */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_FOREIGN_TABLE);
            /* This command never recurses */
            ATPrepChangeInherit(rel);
            pass = AT_PASS_MISC;
            break;
        case AT_AlterConstraint:    /* ALTER CONSTRAINT */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE);
            /* Recursion occurs during execution phase */
            if (recurse)
                cmd->recurse = true;
            pass = AT_PASS_MISC;
            break;
        case AT_ValidateConstraint: /* VALIDATE CONSTRAINT */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
            /* Recursion occurs during execution phase */
            /* No command-specific prep needed except saving recurse flag */
            if (recurse)
                cmd->recurse = true;
            pass = AT_PASS_MISC;
            break;
        case AT_ReplicaIdentity:    /* REPLICA IDENTITY ... */
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_MATVIEW);
            pass = AT_PASS_MISC;
            /* This command never recurses */
            /* No command-specific prep needed */
            break;
        case AT_EnableTrig:     /* ENABLE TRIGGER variants */
        case AT_EnableAlwaysTrig:
        case AT_EnableReplicaTrig:
        case AT_EnableTrigAll:
        case AT_EnableTrigUser:
        case AT_DisableTrig:    /* DISABLE TRIGGER variants */
        case AT_DisableTrigAll:
        case AT_DisableTrigUser:
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
            /* Set up recursion for phase 2; no other prep needed */
            if (recurse)
                cmd->recurse = true;
            pass = AT_PASS_MISC;
            break;
        case AT_EnableRule:     /* ENABLE/DISABLE RULE variants */
        case AT_EnableAlwaysRule:
        case AT_EnableReplicaRule:
        case AT_DisableRule:
        case AT_AddOf:          /* OF */
        case AT_DropOf:         /* NOT OF */
        case AT_EnableRowSecurity:
        case AT_DisableRowSecurity:
        case AT_ForceRowSecurity:
        case AT_NoForceRowSecurity:
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_TABLE | ATT_PARTITIONED_TABLE);
            /* These commands never recurse */
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        case AT_GenericOptions:
            ATSimplePermissions(cmd->subtype, rel, ATT_FOREIGN_TABLE);
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        case AT_AttachPartition:
            ATSimplePermissions(cmd->subtype, rel,
                                ATT_PARTITIONED_TABLE | ATT_PARTITIONED_INDEX);
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        case AT_DetachPartition:
            ATSimplePermissions(cmd->subtype, rel, ATT_PARTITIONED_TABLE);
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        case AT_DetachPartitionFinalize:
            ATSimplePermissions(cmd->subtype, rel, ATT_PARTITIONED_TABLE);
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        case AT_MergePartitions:
        case AT_SplitPartition:
            ATSimplePermissions(cmd->subtype, rel, ATT_PARTITIONED_TABLE);
            /* No command-specific prep needed */
            pass = AT_PASS_MISC;
            break;
        default:                /* oops */
            elog(ERROR, "unrecognized alter table type: %d",
                 (int) cmd->subtype);
            pass = AT_PASS_UNSET;   /* keep compiler quiet */
            break;
    }
    Assert(pass > AT_PASS_UNSET);
 
    /* Add the subcommand to the appropriate list for phase 2 */
    tab->subcmds[pass] = lappend(tab->subcmds[pass], cmd);
}

References Assert, AT_AddColumn, AT_AddColumnToView, AT_AddConstraint, AT_AddIdentity, AT_AddIndex, AT_AddIndexConstraint, AT_AddInherit, AT_AddOf, AT_AlterColumnGenericOptions, AT_AlterColumnType, AT_AlterConstraint, AT_AttachPartition, AT_ChangeOwner, AT_ClusterOn, AT_ColumnDefault, AT_CookedColumnDefault, AT_DetachPartition, AT_DetachPartitionFinalize, AT_DisableRowSecurity, AT_DisableRule, AT_DisableTrig, AT_DisableTrigAll, AT_DisableTrigUser, AT_DropCluster, AT_DropColumn, AT_DropConstraint, AT_DropExpression, AT_DropIdentity, AT_DropInherit, AT_DropNotNull, AT_DropOf, AT_DropOids, AT_EnableAlwaysRule, AT_EnableAlwaysTrig, AT_EnableReplicaRule, AT_EnableReplicaTrig, AT_EnableRowSecurity, AT_EnableRule, AT_EnableTrig, AT_EnableTrigAll, AT_EnableTrigUser, AT_ForceRowSecurity, AT_GenericOptions, AT_MergePartitions, AT_NoForceRowSecurity, AT_PASS_ADD_COL, AT_PASS_ADD_CONSTR, AT_PASS_ADD_INDEX, AT_PASS_ADD_INDEXCONSTR, AT_PASS_ADD_OTHERCONSTR, AT_PASS_ALTER_TYPE, AT_PASS_COL_ATTRS, AT_PASS_DROP, AT_PASS_MISC, AT_PASS_SET_EXPRESSION, AT_PASS_UNSET, AT_ReplaceRelOptions, AT_ReplicaIdentity, AT_ResetOptions, AT_ResetRelOptions, AT_SetAccessMethod, AT_SetCompression, AT_SetExpression, AT_SetIdentity, AT_SetLogged, AT_SetNotNull, AT_SetOptions, AT_SetRelOptions, AT_SetStatistics, AT_SetStorage, AT_SetTableSpace, AT_SetUnLogged, AT_SplitPartition, AT_ValidateConstraint, ATCheckPartitionsNotInUse(), ATGetQueueEntry(), ATParseTransformCmd(), ATPrepAddColumn(), ATPrepAddPrimaryKey(), ATPrepAlterColumnType(), ATPrepChangeInherit(), ATPrepChangePersistence(), ATPrepDropColumn(), ATPrepDropExpression(), ATPrepSetAccessMethod(), ATPrepSetTableSpace(), ATSimplePermissions(), ATSimpleRecursion(), ATT_COMPOSITE_TYPE, ATT_FOREIGN_TABLE, ATT_INDEX, ATT_MATVIEW, ATT_PARTITIONED_INDEX, ATT_PARTITIONED_TABLE, ATT_SEQUENCE, ATT_TABLE, ATT_VIEW, AlteredTableInfo::chgAccessMethod, AlteredTableInfo::chgPersistence, copyObject, AlterTableCmd::def, elog, ereport, errcode(), errhint(), errmsg, ERROR, fb(), find_all_inheritors(), lappend(), AlterTableCmd::name, PartitionHasPendingDetach(), RelationData::rd_rel, AlterTableCmd::recurse, RelationGetRelationName, RelationGetRelid, AlteredTableInfo::subcmds, and AlterTableCmd::subtype.

Referenced by ATController(), ATPrepAddPrimaryKey(), ATPrepAlterColumnType(), ATSimpleRecursion(), and ATTypedTableRecursion().

◆ ATPrepDropColumn()

static void ATPrepDropColumn	(	List **	wqueue,
		Relation	rel,
		bool	recurse,
		bool	recursing,
		AlterTableCmd *	cmd,
		LOCKMODE	lockmode,
		AlterTableUtilityContext *	context
	)

static

Definition at line 9391 of file tablecmds.c.

{
    if (rel->rd_rel->reloftype && !recursing)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot drop column from typed table")));
 
    if (rel->rd_rel->relkind == RELKIND_COMPOSITE_TYPE)
        ATTypedTableRecursion(wqueue, rel, cmd, lockmode, context);
 
    if (recurse)
        cmd->recurse = true;
}

References ATTypedTableRecursion(), ereport, errcode(), errmsg, ERROR, fb(), RelationData::rd_rel, and AlterTableCmd::recurse.

Referenced by ATPrepCmd().

◆ ATPrepDropExpression()

static void ATPrepDropExpression	(	Relation	rel,
		AlterTableCmd *	cmd,
		bool	recurse,
		bool	recursing,
		LOCKMODE	lockmode
	)

static

Definition at line 8891 of file tablecmds.c.

{
    /*
     * Reject ONLY if there are child tables.  We could implement this, but it
     * is a bit complicated.  GENERATED clauses must be attached to the column
     * definition and cannot be added later like DEFAULT, so if a child table
     * has a generation expression that the parent does not have, the child
     * column will necessarily be an attislocal column.  So to implement ONLY
     * here, we'd need extra code to update attislocal of the direct child
     * tables, somewhat similar to how DROP COLUMN does it, so that the
     * resulting state can be properly dumped and restored.
     */
    if (!recurse &&
        find_inheritance_children(RelationGetRelid(rel), lockmode))
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("ALTER TABLE / DROP EXPRESSION must be applied to child tables too")));
 
    /*
     * Cannot drop generation expression from inherited columns.
     */
    if (!recursing)
    {
        HeapTuple   tuple;
        Form_pg_attribute attTup;
 
        tuple = SearchSysCacheCopyAttName(RelationGetRelid(rel), cmd->name);
        if (!HeapTupleIsValid(tuple))
            ereport(ERROR,
                    (errcode(ERRCODE_UNDEFINED_COLUMN),
                     errmsg("column \"%s\" of relation \"%s\" does not exist",
                            cmd->name, RelationGetRelationName(rel))));
 
        attTup = (Form_pg_attribute) GETSTRUCT(tuple);
 
        if (attTup->attinhcount > 0)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                     errmsg("cannot drop generation expression from inherited column")));
    }
}

References ereport, errcode(), errmsg, ERROR, fb(), find_inheritance_children(), GETSTRUCT(), HeapTupleIsValid, AlterTableCmd::name, RelationGetRelationName, RelationGetRelid, and SearchSysCacheCopyAttName().

Referenced by ATPrepCmd().

◆ ATPrepSetAccessMethod()

static void ATPrepSetAccessMethod	(	AlteredTableInfo *	tab,
		Relation	rel,
		const char *	amname
	)

static

Definition at line 17197 of file tablecmds.c.

{
    Oid         amoid;
 
    /*
     * Look up the access method name and check that it differs from the
     * table's current AM.  If DEFAULT was specified for a partitioned table
     * (amname is NULL), set it to InvalidOid to reset the catalogued AM.
     */
    if (amname != NULL)
        amoid = get_table_am_oid(amname, false);
    else if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
        amoid = InvalidOid;
    else
        amoid = get_table_am_oid(default_table_access_method, false);
 
    /* if it's a match, phase 3 doesn't need to do anything */
    if (rel->rd_rel->relam == amoid)
        return;
 
    /* Save info for Phase 3 to do the real work */
    tab->rewrite |= AT_REWRITE_ACCESS_METHOD;
    tab->newAccessMethod = amoid;
    tab->chgAccessMethod = true;
}

References AT_REWRITE_ACCESS_METHOD, AlteredTableInfo::chgAccessMethod, default_table_access_method, fb(), get_table_am_oid(), InvalidOid, AlteredTableInfo::newAccessMethod, RelationData::rd_rel, and AlteredTableInfo::rewrite.

Referenced by ATPrepCmd().

◆ ATPrepSetTableSpace()

static void ATPrepSetTableSpace	(	AlteredTableInfo *	tab,
		Relation	rel,
		const char *	tablespacename,
		LOCKMODE	lockmode
	)

static

Definition at line 17321 of file tablecmds.c.

{
    Oid         tablespaceId;
 
    /* Check that the tablespace exists */
    tablespaceId = get_tablespace_oid(tablespacename, false);
 
    /* Check permissions except when moving to database's default */
    if (OidIsValid(tablespaceId) && tablespaceId != MyDatabaseTableSpace)
    {
        AclResult   aclresult;
 
        aclresult = object_aclcheck(TableSpaceRelationId, tablespaceId, GetUserId(), ACL_CREATE);
        if (aclresult != ACLCHECK_OK)
            aclcheck_error(aclresult, OBJECT_TABLESPACE, tablespacename);
    }
 
    /* Save info for Phase 3 to do the real work */
    if (OidIsValid(tab->newTableSpace))
        ereport(ERROR,
                (errcode(ERRCODE_SYNTAX_ERROR),
                 errmsg("cannot have multiple SET TABLESPACE subcommands")));
 
    tab->newTableSpace = tablespaceId;
}

References ACL_CREATE, aclcheck_error(), ACLCHECK_OK, ereport, errcode(), errmsg, ERROR, fb(), get_tablespace_oid(), GetUserId(), MyDatabaseTableSpace, AlteredTableInfo::newTableSpace, object_aclcheck(), OBJECT_TABLESPACE, and OidIsValid.

Referenced by ATPrepCmd().

◆ ATRewriteCatalogs()

static void ATRewriteCatalogs	(	List **	wqueue,
		LOCKMODE	lockmode,
		AlterTableUtilityContext *	context
	)

static

Definition at line 5405 of file tablecmds.c.

{
    ListCell   *ltab;
 
    /*
     * We process all the tables "in parallel", one pass at a time.  This is
     * needed because we may have to propagate work from one table to another
     * (specifically, ALTER TYPE on a foreign key's PK has to dispatch the
     * re-adding of the foreign key constraint to the other table).  Work can
     * only be propagated into later passes, however.
     */
    for (AlterTablePass pass = 0; pass < AT_NUM_PASSES; pass++)
    {
        /* Go through each table that needs to be processed */
        foreach(ltab, *wqueue)
        {
            AlteredTableInfo *tab = (AlteredTableInfo *) lfirst(ltab);
            List       *subcmds = tab->subcmds[pass];
            ListCell   *lcmd;
 
            if (subcmds == NIL)
                continue;
 
            /*
             * Open the relation and store it in tab.  This allows subroutines
             * close and reopen, if necessary.  Appropriate lock was obtained
             * by phase 1, needn't get it again.
             */
            tab->rel = relation_open(tab->relid, NoLock);
 
            foreach(lcmd, subcmds)
                ATExecCmd(wqueue, tab,
                          lfirst_node(AlterTableCmd, lcmd),
                          lockmode, pass, context);
 
            /*
             * After the ALTER TYPE or SET EXPRESSION pass, do cleanup work
             * (this is not done in ATExecAlterColumnType since it should be
             * done only once if multiple columns of a table are altered).
             */
            if (pass == AT_PASS_ALTER_TYPE || pass == AT_PASS_SET_EXPRESSION)
                ATPostAlterTypeCleanup(wqueue, tab, lockmode);
 
            if (tab->rel)
            {
                relation_close(tab->rel, NoLock);
                tab->rel = NULL;
            }
        }
    }
 
    /* Check to see if a toast table must be added. */
    foreach(ltab, *wqueue)
    {
        AlteredTableInfo *tab = (AlteredTableInfo *) lfirst(ltab);
 
        /*
         * If the table is source table of ATTACH PARTITION command, we did
         * not modify anything about it that will change its toasting
         * requirement, so no need to check.
         */
        if (((tab->relkind == RELKIND_RELATION ||
              tab->relkind == RELKIND_PARTITIONED_TABLE) &&
             tab->partition_constraint == NULL) ||
            tab->relkind == RELKIND_MATVIEW)
            AlterTableCreateToastTable(tab->relid, (Datum) 0, lockmode);
    }
}

References AlterTableCreateToastTable(), AT_NUM_PASSES, AT_PASS_ALTER_TYPE, AT_PASS_SET_EXPRESSION, ATExecCmd(), ATPostAlterTypeCleanup(), fb(), lfirst, lfirst_node, NIL, NoLock, AlteredTableInfo::partition_constraint, AlteredTableInfo::rel, relation_close(), relation_open(), AlteredTableInfo::relid, AlteredTableInfo::relkind, and AlteredTableInfo::subcmds.

Referenced by ATController().

◆ ATRewriteTable()

static void ATRewriteTable	(	AlteredTableInfo *	tab,
		Oid	OIDNewHeap
	)

static

Definition at line 6246 of file tablecmds.c.

{
    Relation    oldrel;
    Relation    newrel;
    TupleDesc   oldTupDesc;
    TupleDesc   newTupDesc;
    bool        needscan = false;
    List       *notnull_attrs;
    List       *notnull_virtual_attrs;
    int         i;
    ListCell   *l;
    EState     *estate;
    CommandId   mycid;
    BulkInsertState bistate;
    uint32      ti_options;
    ExprState  *partqualstate = NULL;
 
    /*
     * Open the relation(s).  We have surely already locked the existing
     * table.
     */
    oldrel = table_open(tab->relid, NoLock);
    oldTupDesc = tab->oldDesc;
    newTupDesc = RelationGetDescr(oldrel);  /* includes all mods */
 
    if (OidIsValid(OIDNewHeap))
    {
        Assert(CheckRelationOidLockedByMe(OIDNewHeap, AccessExclusiveLock,
                                          false));
        newrel = table_open(OIDNewHeap, NoLock);
    }
    else
        newrel = NULL;
 
    /*
     * Prepare a BulkInsertState and options for table_tuple_insert.  The FSM
     * is empty, so don't bother using it.
     */
    if (newrel)
    {
        mycid = GetCurrentCommandId(true);
        bistate = GetBulkInsertState();
        ti_options = TABLE_INSERT_SKIP_FSM;
    }
    else
    {
        /* keep compiler quiet about using these uninitialized */
        mycid = 0;
        bistate = NULL;
        ti_options = 0;
    }
 
    /*
     * Generate the constraint and default execution states
     */
 
    estate = CreateExecutorState();
 
    /* Build the needed expression execution states */
    foreach(l, tab->constraints)
    {
        NewConstraint *con = lfirst(l);
 
        switch (con->contype)
        {
            case CONSTR_CHECK:
                needscan = true;
                con->qualstate = ExecPrepareExpr((Expr *) expand_generated_columns_in_expr(con->qual, oldrel, 1), estate);
                break;
            case CONSTR_FOREIGN:
                /* Nothing to do here */
                break;
            default:
                elog(ERROR, "unrecognized constraint type: %d",
                     (int) con->contype);
        }
    }
 
    /* Build expression execution states for partition check quals */
    if (tab->partition_constraint)
    {
        needscan = true;
        partqualstate = ExecPrepareExpr(tab->partition_constraint, estate);
    }
 
    foreach(l, tab->newvals)
    {
        NewColumnValue *ex = lfirst(l);
 
        /* expr already planned */
        ex->exprstate = ExecInitExpr(ex->expr, NULL);
    }
 
    notnull_attrs = notnull_virtual_attrs = NIL;
    if (newrel || tab->verify_new_notnull)
    {
        /*
         * If we are rebuilding the tuples OR if we added any new but not
         * verified not-null constraints, check all *valid* not-null
         * constraints. This is a bit of overkill but it minimizes risk of
         * bugs.
         *
         * notnull_attrs does *not* collect attribute numbers for valid
         * not-null constraints over virtual generated columns; instead, they
         * are collected in notnull_virtual_attrs for verification elsewhere.
         */
        for (i = 0; i < newTupDesc->natts; i++)
        {
            CompactAttribute *attr = TupleDescCompactAttr(newTupDesc, i);
 
            if (attr->attnullability == ATTNULLABLE_VALID &&
                !attr->attisdropped)
            {
                Form_pg_attribute wholeatt = TupleDescAttr(newTupDesc, i);
 
                if (wholeatt->attgenerated != ATTRIBUTE_GENERATED_VIRTUAL)
                    notnull_attrs = lappend_int(notnull_attrs, wholeatt->attnum);
                else
                    notnull_virtual_attrs = lappend_int(notnull_virtual_attrs,
                                                        wholeatt->attnum);
            }
        }
        if (notnull_attrs || notnull_virtual_attrs)
            needscan = true;
    }
 
    if (newrel || needscan)
    {
        ExprContext *econtext;
        TupleTableSlot *oldslot;
        TupleTableSlot *newslot;
        TableScanDesc scan;
        MemoryContext oldCxt;
        List       *dropped_attrs = NIL;
        ListCell   *lc;
        Snapshot    snapshot;
        ResultRelInfo *rInfo = NULL;
 
        /*
         * When adding or changing a virtual generated column with a not-null
         * constraint, we need to evaluate whether the generation expression
         * is null.  For that, we borrow ExecRelGenVirtualNotNull().  Here, we
         * prepare a dummy ResultRelInfo.
         */
        if (notnull_virtual_attrs != NIL)
        {
            MemoryContext oldcontext;
 
            Assert(newTupDesc->constr->has_generated_virtual);
            Assert(newTupDesc->constr->has_not_null);
            oldcontext = MemoryContextSwitchTo(estate->es_query_cxt);
            rInfo = makeNode(ResultRelInfo);
            InitResultRelInfo(rInfo,
                              oldrel,
                              0,    /* dummy rangetable index */
                              NULL,
                              estate->es_instrument);
            MemoryContextSwitchTo(oldcontext);
        }
 
        if (newrel)
            ereport(DEBUG1,
                    (errmsg_internal("rewriting table \"%s\"",
                                     RelationGetRelationName(oldrel))));
        else
            ereport(DEBUG1,
                    (errmsg_internal("verifying table \"%s\"",
                                     RelationGetRelationName(oldrel))));
 
        if (newrel)
        {
            /*
             * All predicate locks on the tuples or pages are about to be made
             * invalid, because we move tuples around.  Promote them to
             * relation locks.
             */
            TransferPredicateLocksToHeapRelation(oldrel);
        }
 
        econtext = GetPerTupleExprContext(estate);
 
        /*
         * Create necessary tuple slots. When rewriting, two slots are needed,
         * otherwise one suffices. In the case where one slot suffices, we
         * need to use the new tuple descriptor, otherwise some constraints
         * can't be evaluated.  Note that even when the tuple layout is the
         * same and no rewrite is required, the tupDescs might not be
         * (consider ADD COLUMN without a default).
         */
        if (tab->rewrite)
        {
            Assert(newrel != NULL);
            oldslot = MakeSingleTupleTableSlot(oldTupDesc,
                                               table_slot_callbacks(oldrel));
            newslot = MakeSingleTupleTableSlot(newTupDesc,
                                               table_slot_callbacks(newrel));
 
            /*
             * Set all columns in the new slot to NULL initially, to ensure
             * columns added as part of the rewrite are initialized to NULL.
             * That is necessary as tab->newvals will not contain an
             * expression for columns with a NULL default, e.g. when adding a
             * column without a default together with a column with a default
             * requiring an actual rewrite.
             */
            ExecStoreAllNullTuple(newslot);
        }
        else
        {
            oldslot = MakeSingleTupleTableSlot(newTupDesc,
                                               table_slot_callbacks(oldrel));
            newslot = NULL;
        }
 
        /*
         * Any attributes that are dropped according to the new tuple
         * descriptor can be set to NULL. We precompute the list of dropped
         * attributes to avoid needing to do so in the per-tuple loop.
         */
        for (i = 0; i < newTupDesc->natts; i++)
        {
            if (TupleDescAttr(newTupDesc, i)->attisdropped)
                dropped_attrs = lappend_int(dropped_attrs, i);
        }
 
        /*
         * Scan through the rows, generating a new row if needed and then
         * checking all the constraints.
         */
        snapshot = RegisterSnapshot(GetLatestSnapshot());
        scan = table_beginscan(oldrel, snapshot, 0, NULL,
                               SO_NONE);
 
        /*
         * Switch to per-tuple memory context and reset it for each tuple
         * produced, so we don't leak memory.
         */
        oldCxt = MemoryContextSwitchTo(GetPerTupleMemoryContext(estate));
 
        while (table_scan_getnextslot(scan, ForwardScanDirection, oldslot))
        {
            TupleTableSlot *insertslot;
 
            if (tab->rewrite > 0)
            {
                /* Extract data from old tuple */
                slot_getallattrs(oldslot);
                ExecClearTuple(newslot);
 
                /* copy attributes */
                memcpy(newslot->tts_values, oldslot->tts_values,
                       sizeof(Datum) * oldslot->tts_nvalid);
                memcpy(newslot->tts_isnull, oldslot->tts_isnull,
                       sizeof(bool) * oldslot->tts_nvalid);
 
                /* Set dropped attributes to null in new tuple */
                foreach(lc, dropped_attrs)
                    newslot->tts_isnull[lfirst_int(lc)] = true;
 
                /*
                 * Constraints and GENERATED expressions might reference the
                 * tableoid column, so fill tts_tableOid with the desired
                 * value.  (We must do this each time, because it gets
                 * overwritten with newrel's OID during storing.)
                 */
                newslot->tts_tableOid = RelationGetRelid(oldrel);
 
                /*
                 * Process supplied expressions to replace selected columns.
                 *
                 * First, evaluate expressions whose inputs come from the old
                 * tuple.
                 */
                econtext->ecxt_scantuple = oldslot;
 
                foreach(l, tab->newvals)
                {
                    NewColumnValue *ex = lfirst(l);
 
                    if (ex->is_generated)
                        continue;
 
                    newslot->tts_values[ex->attnum - 1]
                        = ExecEvalExpr(ex->exprstate,
                                       econtext,
                                       &newslot->tts_isnull[ex->attnum - 1]);
                }
 
                ExecStoreVirtualTuple(newslot);
 
                /*
                 * Now, evaluate any expressions whose inputs come from the
                 * new tuple.  We assume these columns won't reference each
                 * other, so that there's no ordering dependency.
                 */
                econtext->ecxt_scantuple = newslot;
 
                foreach(l, tab->newvals)
                {
                    NewColumnValue *ex = lfirst(l);
 
                    if (!ex->is_generated)
                        continue;
 
                    newslot->tts_values[ex->attnum - 1]
                        = ExecEvalExpr(ex->exprstate,
                                       econtext,
                                       &newslot->tts_isnull[ex->attnum - 1]);
                }
 
                insertslot = newslot;
            }
            else
            {
                /*
                 * If there's no rewrite, old and new table are guaranteed to
                 * have the same AM, so we can just use the old slot to verify
                 * new constraints etc.
                 */
                insertslot = oldslot;
            }
 
            /* Now check any constraints on the possibly-changed tuple */
            econtext->ecxt_scantuple = insertslot;
 
            foreach_int(attn, notnull_attrs)
            {
                if (slot_attisnull(insertslot, attn))
                {
                    Form_pg_attribute attr = TupleDescAttr(newTupDesc, attn - 1);
 
                    ereport(ERROR,
                            (errcode(ERRCODE_NOT_NULL_VIOLATION),
                             errmsg("column \"%s\" of relation \"%s\" contains null values",
                                    NameStr(attr->attname),
                                    RelationGetRelationName(oldrel)),
                             errtablecol(oldrel, attn)));
                }
            }
 
            if (notnull_virtual_attrs != NIL)
            {
                AttrNumber  attnum;
 
                attnum = ExecRelGenVirtualNotNull(rInfo, insertslot,
                                                  estate,
                                                  notnull_virtual_attrs);
                if (attnum != InvalidAttrNumber)
                {
                    Form_pg_attribute attr = TupleDescAttr(newTupDesc, attnum - 1);
 
                    ereport(ERROR,
                            errcode(ERRCODE_NOT_NULL_VIOLATION),
                            errmsg("column \"%s\" of relation \"%s\" contains null values",
                                   NameStr(attr->attname),
                                   RelationGetRelationName(oldrel)),
                            errtablecol(oldrel, attnum));
                }
            }
 
            foreach(l, tab->constraints)
            {
                NewConstraint *con = lfirst(l);
 
                switch (con->contype)
                {
                    case CONSTR_CHECK:
                        if (!ExecCheck(con->qualstate, econtext))
                            ereport(ERROR,
                                    (errcode(ERRCODE_CHECK_VIOLATION),
                                     errmsg("check constraint \"%s\" of relation \"%s\" is violated by some row",
                                            con->name,
                                            RelationGetRelationName(oldrel)),
                                     errtableconstraint(oldrel, con->name)));
                        break;
                    case CONSTR_NOTNULL:
                    case CONSTR_FOREIGN:
                        /* Nothing to do here */
                        break;
                    default:
                        elog(ERROR, "unrecognized constraint type: %d",
                             (int) con->contype);
                }
            }
 
            if (partqualstate && !ExecCheck(partqualstate, econtext))
            {
                if (tab->validate_default)
                    ereport(ERROR,
                            (errcode(ERRCODE_CHECK_VIOLATION),
                             errmsg("updated partition constraint for default partition \"%s\" would be violated by some row",
                                    RelationGetRelationName(oldrel)),
                             errtable(oldrel)));
                else
                    ereport(ERROR,
                            (errcode(ERRCODE_CHECK_VIOLATION),
                             errmsg("partition constraint of relation \"%s\" is violated by some row",
                                    RelationGetRelationName(oldrel)),
                             errtable(oldrel)));
            }
 
            /* Write the tuple out to the new relation */
            if (newrel)
                table_tuple_insert(newrel, insertslot, mycid,
                                   ti_options, bistate);
 
            ResetExprContext(econtext);
 
            CHECK_FOR_INTERRUPTS();
        }
 
        MemoryContextSwitchTo(oldCxt);
        table_endscan(scan);
        UnregisterSnapshot(snapshot);
 
        ExecDropSingleTupleTableSlot(oldslot);
        if (newslot)
            ExecDropSingleTupleTableSlot(newslot);
    }
 
    FreeExecutorState(estate);
 
    table_close(oldrel, NoLock);
    if (newrel)
    {
        FreeBulkInsertState(bistate);
 
        table_finish_bulk_insert(newrel, ti_options);
 
        table_close(newrel, NoLock);
    }
}

References AccessExclusiveLock, Assert, CompactAttribute::attisdropped, CompactAttribute::attnullability, ATTNULLABLE_VALID, attnum, CHECK_FOR_INTERRUPTS, CheckRelationOidLockedByMe(), CONSTR_CHECK, CONSTR_FOREIGN, CONSTR_NOTNULL, AlteredTableInfo::constraints, NewConstraint::contype, CreateExecutorState(), DEBUG1, ExprContext::ecxt_scantuple, elog, ereport, errcode(), errmsg, errmsg_internal(), ERROR, errtable(), errtablecol(), errtableconstraint(), EState::es_instrument, EState::es_query_cxt, ExecCheck(), ExecClearTuple(), ExecDropSingleTupleTableSlot(), ExecEvalExpr(), ExecInitExpr(), ExecPrepareExpr(), ExecRelGenVirtualNotNull(), ExecStoreAllNullTuple(), ExecStoreVirtualTuple(), expand_generated_columns_in_expr(), fb(), foreach_int, ForwardScanDirection, FreeBulkInsertState(), FreeExecutorState(), GetBulkInsertState(), GetCurrentCommandId(), GetLatestSnapshot(), GetPerTupleExprContext, GetPerTupleMemoryContext, i, InitResultRelInfo(), InvalidAttrNumber, lappend_int(), lfirst, lfirst_int, makeNode, MakeSingleTupleTableSlot(), memcpy(), MemoryContextSwitchTo(), NewConstraint::name, NameStr, AlteredTableInfo::newvals, NIL, NoLock, OidIsValid, AlteredTableInfo::oldDesc, AlteredTableInfo::partition_constraint, NewConstraint::qual, NewConstraint::qualstate, RegisterSnapshot(), RelationGetDescr, RelationGetRelationName, RelationGetRelid, AlteredTableInfo::relid, ResetExprContext, AlteredTableInfo::rewrite, slot_attisnull(), slot_getallattrs(), SO_NONE, table_beginscan(), table_close(), table_endscan(), table_finish_bulk_insert(), TABLE_INSERT_SKIP_FSM, table_open(), table_scan_getnextslot(), table_slot_callbacks(), table_tuple_insert(), TransferPredicateLocksToHeapRelation(), TupleDescAttr(), TupleDescCompactAttr(), UnregisterSnapshot(), AlteredTableInfo::validate_default, and AlteredTableInfo::verify_new_notnull.

Referenced by ATRewriteTables().

◆ ATRewriteTables()

static void ATRewriteTables	(	AlterTableStmt *	parsetree,
		List **	wqueue,
		LOCKMODE	lockmode,
		AlterTableUtilityContext *	context
	)

static

Definition at line 5957 of file tablecmds.c.

{
    ListCell   *ltab;
 
    /* Go through each table that needs to be checked or rewritten */
    foreach(ltab, *wqueue)
    {
        AlteredTableInfo *tab = (AlteredTableInfo *) lfirst(ltab);
 
        /* Relations without storage may be ignored here */
        if (!RELKIND_HAS_STORAGE(tab->relkind))
            continue;
 
        /*
         * If we change column data types, the operation has to be propagated
         * to tables that use this table's rowtype as a column type.
         * tab->newvals will also be non-NULL in the case where we're adding a
         * column with a default.  We choose to forbid that case as well,
         * since composite types might eventually support defaults.
         *
         * (Eventually we'll probably need to check for composite type
         * dependencies even when we're just scanning the table without a
         * rewrite, but at the moment a composite type does not enforce any
         * constraints, so it's not necessary/appropriate to enforce them just
         * during ALTER.)
         */
        if (tab->newvals != NIL || tab->rewrite > 0)
        {
            Relation    rel;
 
            rel = table_open(tab->relid, NoLock);
            find_composite_type_dependencies(rel->rd_rel->reltype, rel, NULL);
            table_close(rel, NoLock);
        }
 
        /*
         * We only need to rewrite the table if at least one column needs to
         * be recomputed, or we are changing its persistence or access method.
         *
         * There are two reasons for requiring a rewrite when changing
         * persistence: on one hand, we need to ensure that the buffers
         * belonging to each of the two relations are marked with or without
         * BM_PERMANENT properly.  On the other hand, since rewriting creates
         * and assigns a new relfilenumber, we automatically create or drop an
         * init fork for the relation as appropriate.
         */
        if (tab->rewrite > 0 && tab->relkind != RELKIND_SEQUENCE)
        {
            /* Build a temporary relation and copy data */
            Relation    OldHeap;
            Oid         OIDNewHeap;
            Oid         NewAccessMethod;
            Oid         NewTableSpace;
            char        persistence;
 
            OldHeap = table_open(tab->relid, NoLock);
 
            /*
             * We don't support rewriting of system catalogs; there are too
             * many corner cases and too little benefit.  In particular this
             * is certainly not going to work for mapped catalogs.
             */
            if (IsSystemRelation(OldHeap))
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("cannot rewrite system relation \"%s\"",
                                RelationGetRelationName(OldHeap))));
 
            if (RelationIsUsedAsCatalogTable(OldHeap))
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("cannot rewrite table \"%s\" used as a catalog table",
                                RelationGetRelationName(OldHeap))));
 
            /*
             * Don't allow rewrite on temp tables of other backends ... their
             * local buffer manager is not going to cope.  (This is redundant
             * with the check in CheckAlterTableIsSafe, but for safety we'll
             * check here too.)
             */
            if (RELATION_IS_OTHER_TEMP(OldHeap))
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("cannot rewrite temporary tables of other sessions")));
 
            /*
             * Select destination tablespace (same as original unless user
             * requested a change)
             */
            if (tab->newTableSpace)
                NewTableSpace = tab->newTableSpace;
            else
                NewTableSpace = OldHeap->rd_rel->reltablespace;
 
            /*
             * Select destination access method (same as original unless user
             * requested a change)
             */
            if (tab->chgAccessMethod)
                NewAccessMethod = tab->newAccessMethod;
            else
                NewAccessMethod = OldHeap->rd_rel->relam;
 
            /*
             * Select persistence of transient table (same as original unless
             * user requested a change)
             */
            persistence = tab->chgPersistence ?
                tab->newrelpersistence : OldHeap->rd_rel->relpersistence;
 
            table_close(OldHeap, NoLock);
 
            /*
             * Fire off an Event Trigger now, before actually rewriting the
             * table.
             *
             * We don't support Event Trigger for nested commands anywhere,
             * here included, and parsetree is given NULL when coming from
             * AlterTableInternal.
             *
             * And fire it only once.
             */
            if (parsetree)
                EventTriggerTableRewrite((Node *) parsetree,
                                         tab->relid,
                                         tab->rewrite);
 
            /*
             * Create transient table that will receive the modified data.
             *
             * Ensure it is marked correctly as logged or unlogged.  We have
             * to do this here so that buffers for the new relfilenumber will
             * have the right persistence set, and at the same time ensure
             * that the original filenumbers's buffers will get read in with
             * the correct setting (i.e. the original one).  Otherwise a
             * rollback after the rewrite would possibly result with buffers
             * for the original filenumbers having the wrong persistence
             * setting.
             *
             * NB: This relies on swap_relation_files() also swapping the
             * persistence. That wouldn't work for pg_class, but that can't be
             * unlogged anyway.
             */
            OIDNewHeap = make_new_heap(tab->relid, NewTableSpace, NewAccessMethod,
                                       persistence, lockmode);
 
            /*
             * Copy the heap data into the new table with the desired
             * modifications, and test the current data within the table
             * against new constraints generated by ALTER TABLE commands.
             */
            ATRewriteTable(tab, OIDNewHeap);
 
            /*
             * Swap the physical files of the old and new heaps, then rebuild
             * indexes and discard the old heap.  We can use RecentXmin for
             * the table's new relfrozenxid because we rewrote all the tuples
             * in ATRewriteTable, so no older Xid remains in the table.  Also,
             * we never try to swap toast tables by content, since we have no
             * interest in letting this code work on system catalogs.
             */
            finish_heap_swap(tab->relid, OIDNewHeap,
                             false, false, true,
                             !OidIsValid(tab->newTableSpace),
                             true,  /* reindex */
                             RecentXmin,
                             ReadNextMultiXactId(),
                             persistence);
 
            InvokeObjectPostAlterHook(RelationRelationId, tab->relid, 0);
        }
        else if (tab->rewrite > 0 && tab->relkind == RELKIND_SEQUENCE)
        {
            if (tab->chgPersistence)
                SequenceChangePersistence(tab->relid, tab->newrelpersistence);
        }
        else
        {
            /*
             * If required, test the current data within the table against new
             * constraints generated by ALTER TABLE commands, but don't
             * rebuild data.
             */
            if (tab->constraints != NIL || tab->verify_new_notnull ||
                tab->partition_constraint != NULL)
                ATRewriteTable(tab, InvalidOid);
 
            /*
             * If we had SET TABLESPACE but no reason to reconstruct tuples,
             * just do a block-by-block copy.
             */
            if (tab->newTableSpace)
                ATExecSetTableSpace(tab->relid, tab->newTableSpace, lockmode);
        }
 
        /*
         * Also change persistence of owned sequences, so that it matches the
         * table persistence.
         */
        if (tab->chgPersistence)
        {
            List       *seqlist = getOwnedSequences(tab->relid);
            ListCell   *lc;
 
            foreach(lc, seqlist)
            {
                Oid         seq_relid = lfirst_oid(lc);
 
                SequenceChangePersistence(seq_relid, tab->newrelpersistence);
            }
        }
    }
 
    /*
     * Foreign key constraints are checked in a final pass, since (a) it's
     * generally best to examine each one separately, and (b) it's at least
     * theoretically possible that we have changed both relations of the
     * foreign key, and we'd better have finished both rewrites before we try
     * to read the tables.
     */
    foreach(ltab, *wqueue)
    {
        AlteredTableInfo *tab = (AlteredTableInfo *) lfirst(ltab);
        Relation    rel = NULL;
        ListCell   *lcon;
 
        /* Relations without storage may be ignored here too */
        if (!RELKIND_HAS_STORAGE(tab->relkind))
            continue;
 
        foreach(lcon, tab->constraints)
        {
            NewConstraint *con = lfirst(lcon);
 
            if (con->contype == CONSTR_FOREIGN)
            {
                Constraint *fkconstraint = (Constraint *) con->qual;
                Relation    refrel;
 
                if (rel == NULL)
                {
                    /* Long since locked, no need for another */
                    rel = table_open(tab->relid, NoLock);
                }
 
                refrel = table_open(con->refrelid, RowShareLock);
 
                validateForeignKeyConstraint(fkconstraint->conname, rel, refrel,
                                             con->refindid,
                                             con->conid,
                                             con->conwithperiod);
 
                /*
                 * No need to mark the constraint row as validated, we did
                 * that when we inserted the row earlier.
                 */
 
                table_close(refrel, NoLock);
            }
        }
 
        if (rel)
            table_close(rel, NoLock);
    }
 
    /* Finally, run any afterStmts that were queued up */
    foreach(ltab, *wqueue)
    {
        AlteredTableInfo *tab = (AlteredTableInfo *) lfirst(ltab);
        ListCell   *lc;
 
        foreach(lc, tab->afterStmts)
        {
            Node       *stmt = (Node *) lfirst(lc);
 
            ProcessUtilityForAlterTable(stmt, context);
            CommandCounterIncrement();
        }
    }
}

Referenced by ATController().

◆ ATSimplePermissions()

static void ATSimplePermissions	(	AlterTableType	cmdtype,
		Relation	rel,
		int	allowed_targets
	)

static

Definition at line 6864 of file tablecmds.c.

{
    int         actual_target;
 
    switch (rel->rd_rel->relkind)
    {
        case RELKIND_RELATION:
            actual_target = ATT_TABLE;
            break;
        case RELKIND_PARTITIONED_TABLE:
            actual_target = ATT_PARTITIONED_TABLE;
            break;
        case RELKIND_VIEW:
            actual_target = ATT_VIEW;
            break;
        case RELKIND_MATVIEW:
            actual_target = ATT_MATVIEW;
            break;
        case RELKIND_INDEX:
            actual_target = ATT_INDEX;
            break;
        case RELKIND_PARTITIONED_INDEX:
            actual_target = ATT_PARTITIONED_INDEX;
            break;
        case RELKIND_COMPOSITE_TYPE:
            actual_target = ATT_COMPOSITE_TYPE;
            break;
        case RELKIND_FOREIGN_TABLE:
            actual_target = ATT_FOREIGN_TABLE;
            break;
        case RELKIND_SEQUENCE:
            actual_target = ATT_SEQUENCE;
            break;
        default:
            actual_target = 0;
            break;
    }
 
    /* Wrong target type? */
    if ((actual_target & allowed_targets) == 0)
    {
        const char *action_str = alter_table_type_to_string(cmdtype);
 
        if (action_str)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
            /* translator: %s is a group of some SQL keywords */
                     errmsg("ALTER action %s cannot be performed on relation \"%s\"",
                            action_str, RelationGetRelationName(rel)),
                     errdetail_relkind_not_supported(rel->rd_rel->relkind)));
        else
            /* internal error? */
            elog(ERROR, "invalid ALTER action attempted on relation \"%s\"",
                 RelationGetRelationName(rel));
    }
 
    /* Permissions checks */
    if (!object_ownercheck(RelationRelationId, RelationGetRelid(rel), GetUserId()))
        aclcheck_error(ACLCHECK_NOT_OWNER, get_relkind_objtype(rel->rd_rel->relkind),
                       RelationGetRelationName(rel));
 
    /*
     * Conflict log tables are used internally for logical replication
     * conflict logging and should not be altered directly, as it could
     * disrupt conflict logging. Direct ALTER commands are already rejected
     * during relation lookup in RangeVarCallbackForAlterRelation(), and
     * AlterTableMoveAll() skips these tables, so a conflict log table does
     * not normally reach here; this check guards any internal caller that
     * arrives via AlterTableInternal().
     */
    if (IsConflictLogTableClass(rel->rd_rel))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot alter conflict log table \"%s\"",
                        RelationGetRelationName(rel)),
                 errdetail("Conflict log tables are system-managed tables for logical replication conflicts.")));
 
    if (!allowSystemTableMods && IsSystemRelation(rel))
        ereport(ERROR,
                (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
                 errmsg("permission denied: \"%s\" is a system catalog",
                        RelationGetRelationName(rel))));
}

References aclcheck_error(), ACLCHECK_NOT_OWNER, allowSystemTableMods, alter_table_type_to_string(), ATT_COMPOSITE_TYPE, ATT_FOREIGN_TABLE, ATT_INDEX, ATT_MATVIEW, ATT_PARTITIONED_INDEX, ATT_PARTITIONED_TABLE, ATT_SEQUENCE, ATT_TABLE, ATT_VIEW, elog, ereport, errcode(), errdetail(), errdetail_relkind_not_supported(), errmsg, ERROR, fb(), get_relkind_objtype(), GetUserId(), IsConflictLogTableClass(), IsSystemRelation(), object_ownercheck(), RelationData::rd_rel, RelationGetRelationName, and RelationGetRelid.

Referenced by ATAddCheckNNConstraint(), ATExecAddColumn(), ATExecAddInherit(), ATExecAttachPartition(), ATExecDropColumn(), ATExecSetNotNull(), ATPrepCmd(), and dropconstraint_internal().

◆ ATSimpleRecursion()

static void ATSimpleRecursion	(	List **	wqueue,
		Relation	rel,
		AlterTableCmd *	cmd,
		bool	recurse,
		LOCKMODE	lockmode,
		AlterTableUtilityContext *	context
	)

static

Definition at line 6957 of file tablecmds.c.

{
    /*
     * Propagate to children, if desired and if there are (or might be) any
     * children.
     */
    if (recurse && rel->rd_rel->relhassubclass)
    {
        Oid         relid = RelationGetRelid(rel);
        ListCell   *child;
        List       *children;
 
        children = find_all_inheritors(relid, lockmode, NULL);
 
        /*
         * find_all_inheritors does the recursive search of the inheritance
         * hierarchy, so all we have to do is process all of the relids in the
         * list that it returns.
         */
        foreach(child, children)
        {
            Oid         childrelid = lfirst_oid(child);
            Relation    childrel;
 
            if (childrelid == relid)
                continue;
            /* find_all_inheritors already got lock */
            childrel = relation_open(childrelid, NoLock);
            CheckAlterTableIsSafe(childrel);
            ATPrepCmd(wqueue, childrel, cmd, false, true, lockmode, context);
            relation_close(childrel, NoLock);
        }
    }
}

References ATPrepCmd(), CheckAlterTableIsSafe(), fb(), find_all_inheritors(), lfirst_oid, NoLock, RelationData::rd_rel, relation_close(), relation_open(), and RelationGetRelid.

Referenced by ATPrepCmd().

◆ AttachPartitionEnsureIndexes()

static void AttachPartitionEnsureIndexes	(	List **	wqueue,
		Relation	rel,
		Relation	attachrel
	)

static

Definition at line 21359 of file tablecmds.c.

{
    List       *idxes;
    List       *attachRelIdxs;
    Relation   *attachrelIdxRels;
    IndexInfo **attachInfos;
    ListCell   *cell;
    MemoryContext cxt;
    MemoryContext oldcxt;
 
    cxt = AllocSetContextCreate(CurrentMemoryContext,
                                "AttachPartitionEnsureIndexes",
                                ALLOCSET_DEFAULT_SIZES);
    oldcxt = MemoryContextSwitchTo(cxt);
 
    idxes = RelationGetIndexList(rel);
    attachRelIdxs = RelationGetIndexList(attachrel);
    attachrelIdxRels = palloc_array(Relation, list_length(attachRelIdxs));
    attachInfos = palloc_array(IndexInfo *, list_length(attachRelIdxs));
 
    /* Build arrays of all existing indexes and their IndexInfos */
    foreach_oid(cldIdxId, attachRelIdxs)
    {
        int         i = foreach_current_index(cldIdxId);
 
        attachrelIdxRels[i] = index_open(cldIdxId, AccessShareLock);
        attachInfos[i] = BuildIndexInfo(attachrelIdxRels[i]);
    }
 
    /*
     * If we're attaching a foreign table, we must fail if any of the indexes
     * is a constraint index; otherwise, there's nothing to do here.  Do this
     * before starting work, to avoid wasting the effort of building a few
     * non-unique indexes before coming across a unique one.
     */
    if (attachrel->rd_rel->relkind == RELKIND_FOREIGN_TABLE)
    {
        foreach(cell, idxes)
        {
            Oid         idx = lfirst_oid(cell);
            Relation    idxRel = index_open(idx, AccessShareLock);
 
            if (idxRel->rd_index->indisunique ||
                idxRel->rd_index->indisprimary)
                ereport(ERROR,
                        (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                         errmsg("cannot attach foreign table \"%s\" as partition of partitioned table \"%s\"",
                                RelationGetRelationName(attachrel),
                                RelationGetRelationName(rel)),
                         errdetail("Partitioned table \"%s\" contains unique indexes.",
                                   RelationGetRelationName(rel))));
            index_close(idxRel, AccessShareLock);
        }
 
        goto out;
    }
 
    /*
     * For each index on the partitioned table, find a matching one in the
     * partition-to-be; if one is not found, create one.
     */
    foreach(cell, idxes)
    {
        Oid         idx = lfirst_oid(cell);
        Relation    idxRel = index_open(idx, AccessShareLock);
        IndexInfo  *info;
        AttrMap    *attmap;
        bool        found = false;
        Oid         constraintOid;
 
        /*
         * Ignore indexes in the partitioned table other than partitioned
         * indexes.
         */
        if (idxRel->rd_rel->relkind != RELKIND_PARTITIONED_INDEX)
        {
            index_close(idxRel, AccessShareLock);
            continue;
        }
 
        /* construct an indexinfo to compare existing indexes against */
        info = BuildIndexInfo(idxRel);
        attmap = build_attrmap_by_name(RelationGetDescr(attachrel),
                                       RelationGetDescr(rel),
                                       false);
        constraintOid = get_relation_idx_constraint_oid(RelationGetRelid(rel), idx);
 
        /*
         * Scan the list of existing indexes in the partition-to-be, and mark
         * the first matching, valid, unattached one we find, if any, as
         * partition of the parent index.  If we find one, we're done.
         */
        for (int i = 0; i < list_length(attachRelIdxs); i++)
        {
            Oid         cldIdxId = RelationGetRelid(attachrelIdxRels[i]);
            Oid         cldConstrOid = InvalidOid;
 
            /* does this index have a parent?  if so, can't use it */
            if (attachrelIdxRels[i]->rd_rel->relispartition)
                continue;
 
            /* If this index is invalid, can't use it */
            if (!attachrelIdxRels[i]->rd_index->indisvalid)
                continue;
 
            if (CompareIndexInfo(attachInfos[i], info,
                                 attachrelIdxRels[i]->rd_indcollation,
                                 idxRel->rd_indcollation,
                                 attachrelIdxRels[i]->rd_opfamily,
                                 idxRel->rd_opfamily,
                                 attmap))
            {
                /*
                 * If this index is being created in the parent because of a
                 * constraint, then the child needs to have a constraint also,
                 * so look for one.  If there is no such constraint, this
                 * index is no good, so keep looking.
                 */
                if (OidIsValid(constraintOid))
                {
                    cldConstrOid =
                        get_relation_idx_constraint_oid(RelationGetRelid(attachrel),
                                                        cldIdxId);
                    /* no dice */
                    if (!OidIsValid(cldConstrOid))
                        continue;
 
                    /* Ensure they're both the same type of constraint */
                    if (get_constraint_type(constraintOid) !=
                        get_constraint_type(cldConstrOid))
                        continue;
                }
 
                /* bingo. */
                IndexSetParentIndex(attachrelIdxRels[i], idx);
                if (OidIsValid(constraintOid))
                    ConstraintSetParentConstraint(cldConstrOid, constraintOid,
                                                  RelationGetRelid(attachrel));
                found = true;
 
                CommandCounterIncrement();
                break;
            }
        }
 
        /*
         * If no suitable index was found in the partition-to-be, create one
         * now.  Note that if this is a PK, not-null constraints must already
         * exist.
         */
        if (!found)
        {
            IndexStmt  *stmt;
            Oid         conOid;
 
            stmt = generateClonedIndexStmt(NULL,
                                           idxRel, attmap,
                                           &conOid);
            DefineIndex(NULL,
                        RelationGetRelid(attachrel), stmt, InvalidOid,
                        RelationGetRelid(idxRel),
                        conOid,
                        -1,
                        true, false, false, false, false);
        }
 
        index_close(idxRel, AccessShareLock);
    }
 
out:
    /* Clean up. */
    for (int i = 0; i < list_length(attachRelIdxs); i++)
        index_close(attachrelIdxRels[i], AccessShareLock);
    MemoryContextSwitchTo(oldcxt);
    MemoryContextDelete(cxt);
}

References AccessShareLock, ALLOCSET_DEFAULT_SIZES, AllocSetContextCreate, build_attrmap_by_name(), BuildIndexInfo(), CommandCounterIncrement(), CompareIndexInfo(), ConstraintSetParentConstraint(), CurrentMemoryContext, DefineIndex(), ereport, errcode(), errdetail(), errmsg, ERROR, fb(), foreach_current_index, foreach_oid, generateClonedIndexStmt(), get_constraint_type(), get_relation_idx_constraint_oid(), i, idx(), index_close(), index_open(), IndexSetParentIndex(), InvalidOid, lfirst_oid, list_length(), MemoryContextDelete(), MemoryContextSwitchTo(), OidIsValid, palloc_array, RelationGetDescr, RelationGetIndexList(), RelationGetRelationName, RelationGetRelid, and stmt.

Referenced by attachPartitionTable().

◆ AttachPartitionForeignKey()

static void AttachPartitionForeignKey	(	List **	wqueue,
		Relation	partition,
		Oid	partConstrOid,
		Oid	parentConstrOid,
		Oid	parentInsTrigger,
		Oid	parentUpdTrigger,
		Relation	trigrel
	)

static

Definition at line 11938 of file tablecmds.c.

{
    HeapTuple   parentConstrTup;
    Form_pg_constraint parentConstr;
    HeapTuple   partcontup;
    Form_pg_constraint partConstr;
    bool        queueValidation;
    Oid         partConstrFrelid;
    Oid         partConstrRelid;
    bool        parentConstrIsEnforced;
 
    /* Fetch the parent constraint tuple */
    parentConstrTup = SearchSysCache1(CONSTROID,
                                      ObjectIdGetDatum(parentConstrOid));
    if (!HeapTupleIsValid(parentConstrTup))
        elog(ERROR, "cache lookup failed for constraint %u", parentConstrOid);
    parentConstr = (Form_pg_constraint) GETSTRUCT(parentConstrTup);
    parentConstrIsEnforced = parentConstr->conenforced;
 
    /* Fetch the child constraint tuple */
    partcontup = SearchSysCache1(CONSTROID,
                                 ObjectIdGetDatum(partConstrOid));
    if (!HeapTupleIsValid(partcontup))
        elog(ERROR, "cache lookup failed for constraint %u", partConstrOid);
    partConstr = (Form_pg_constraint) GETSTRUCT(partcontup);
    partConstrFrelid = partConstr->confrelid;
    partConstrRelid = partConstr->conrelid;
 
    /*
     * If the referenced table is partitioned, then the partition we're
     * attaching now has extra pg_constraint rows and action triggers that are
     * no longer needed.  Remove those.
     */
    if (get_rel_relkind(partConstrFrelid) == RELKIND_PARTITIONED_TABLE)
    {
        Relation    pg_constraint = table_open(ConstraintRelationId, RowShareLock);
 
        RemoveInheritedConstraint(pg_constraint, trigrel, partConstrOid,
                                  partConstrRelid);
 
        table_close(pg_constraint, RowShareLock);
    }
 
    /*
     * Will we need to validate this constraint?   A valid parent constraint
     * implies that all child constraints have been validated, so if this one
     * isn't, we must trigger phase 3 validation.
     */
    queueValidation = parentConstr->convalidated && !partConstr->convalidated;
 
    ReleaseSysCache(partcontup);
    ReleaseSysCache(parentConstrTup);
 
    /*
     * The action triggers in the new partition become redundant -- the parent
     * table already has equivalent ones, and those will be able to reach the
     * partition.  Remove the ones in the partition.  We identify them because
     * they have our constraint OID, as well as being on the referenced rel.
     */
    DropForeignKeyConstraintTriggers(trigrel, partConstrOid, partConstrFrelid,
                                     partConstrRelid);
 
    ConstraintSetParentConstraint(partConstrOid, parentConstrOid,
                                  RelationGetRelid(partition));
 
    /*
     * Like the constraint, attach partition's "check" triggers to the
     * corresponding parent triggers if the constraint is ENFORCED. NOT
     * ENFORCED constraints do not have these triggers.
     */
    if (parentConstrIsEnforced)
    {
        Oid         insertTriggerOid,
                    updateTriggerOid;
 
        GetForeignKeyCheckTriggers(trigrel,
                                   partConstrOid, partConstrFrelid, partConstrRelid,
                                   &insertTriggerOid, &updateTriggerOid);
        Assert(OidIsValid(insertTriggerOid) && OidIsValid(parentInsTrigger));
        TriggerSetParentTrigger(trigrel, insertTriggerOid, parentInsTrigger,
                                RelationGetRelid(partition));
        Assert(OidIsValid(updateTriggerOid) && OidIsValid(parentUpdTrigger));
        TriggerSetParentTrigger(trigrel, updateTriggerOid, parentUpdTrigger,
                                RelationGetRelid(partition));
    }
 
    /*
     * We updated this pg_constraint row above to set its parent; validating
     * it will cause its convalidated flag to change, so we need CCI here.  In
     * addition, we need it unconditionally for the rare case where the parent
     * table has *two* identical constraints; when reaching this function for
     * the second one, we must have made our changes visible, otherwise we
     * would try to attach both to this one.
     */
    CommandCounterIncrement();
 
    /* If validation is needed, put it in the queue now. */
    if (queueValidation)
    {
        Relation    conrel;
        Oid         confrelid;
 
        conrel = table_open(ConstraintRelationId, RowExclusiveLock);
 
        partcontup = SearchSysCache1(CONSTROID, ObjectIdGetDatum(partConstrOid));
        if (!HeapTupleIsValid(partcontup))
            elog(ERROR, "cache lookup failed for constraint %u", partConstrOid);
 
        confrelid = ((Form_pg_constraint) GETSTRUCT(partcontup))->confrelid;
 
        /* Use the same lock as for AT_ValidateConstraint */
        QueueFKConstraintValidation(wqueue, conrel, partition, confrelid,
                                    partcontup, ShareUpdateExclusiveLock);
        ReleaseSysCache(partcontup);
        table_close(conrel, RowExclusiveLock);
    }
}

References Assert, CommandCounterIncrement(), ConstraintSetParentConstraint(), DropForeignKeyConstraintTriggers(), elog, ERROR, fb(), Form_pg_constraint, get_rel_relkind(), GetForeignKeyCheckTriggers(), GETSTRUCT(), HeapTupleIsValid, ObjectIdGetDatum(), OidIsValid, QueueFKConstraintValidation(), RelationGetRelid, ReleaseSysCache(), RemoveInheritedConstraint(), RowExclusiveLock, RowShareLock, SearchSysCache1(), ShareUpdateExclusiveLock, table_close(), table_open(), and TriggerSetParentTrigger().

Referenced by tryAttachPartitionForeignKey().

◆ attachPartitionTable()

static void attachPartitionTable	(	List **	wqueue,
		Relation	rel,
		Relation	attachrel,
		PartitionBoundSpec *	bound
	)

static

Definition at line 20986 of file tablecmds.c.

{
    /*
     * Create an inheritance; the relevant checks are performed inside the
     * function.
     */
    CreateInheritance(attachrel, rel, true);
 
    /* Update the pg_class entry. */
    StorePartitionBound(attachrel, rel, bound);
 
    /* Ensure there exists a correct set of indexes in the partition. */
    AttachPartitionEnsureIndexes(wqueue, rel, attachrel);
 
    /* and triggers */
    CloneRowTriggersToPartition(rel, attachrel);
 
    /*
     * Clone foreign key constraints.  Callee is responsible for setting up
     * for phase 3 constraint verification.
     */
    CloneForeignKeyConstraints(wqueue, rel, attachrel);
}

References AttachPartitionEnsureIndexes(), CloneForeignKeyConstraints(), CloneRowTriggersToPartition(), CreateInheritance(), fb(), and StorePartitionBound().

Referenced by ATExecAttachPartition(), ATExecMergePartitions(), and ATExecSplitPartition().

◆ ATTypedTableRecursion()

static void ATTypedTableRecursion	(	List **	wqueue,
		Relation	rel,
		AlterTableCmd *	cmd,
		LOCKMODE	lockmode,
		AlterTableUtilityContext *	context
	)

static

Definition at line 7032 of file tablecmds.c.

{
    ListCell   *child;
    List       *children;
 
    Assert(rel->rd_rel->relkind == RELKIND_COMPOSITE_TYPE);
 
    children = find_typed_table_dependencies(rel->rd_rel->reltype,
                                             RelationGetRelationName(rel),
                                             cmd->behavior);
 
    foreach(child, children)
    {
        Oid         childrelid = lfirst_oid(child);
        Relation    childrel;
 
        childrel = relation_open(childrelid, lockmode);
        CheckAlterTableIsSafe(childrel);
        ATPrepCmd(wqueue, childrel, cmd, true, true, lockmode, context);
        relation_close(childrel, NoLock);
    }
}

References Assert, ATPrepCmd(), AlterTableCmd::behavior, CheckAlterTableIsSafe(), fb(), find_typed_table_dependencies(), lfirst_oid, NoLock, RelationData::rd_rel, relation_close(), relation_open(), and RelationGetRelationName.

Referenced by ATPrepAddColumn(), ATPrepAlterColumnType(), and ATPrepDropColumn().

◆ BuildDescForRelation()

TupleDesc BuildDescForRelation ( const List * columns )

Definition at line 1437 of file tablecmds.c.

{
    int         natts;
    AttrNumber  attnum;
    ListCell   *l;
    TupleDesc   desc;
    char       *attname;
    Oid         atttypid;
    int32       atttypmod;
    Oid         attcollation;
    int         attdim;
 
    /*
     * allocate a new tuple descriptor
     */
    natts = list_length(columns);
    desc = CreateTemplateTupleDesc(natts);
 
    attnum = 0;
 
    foreach(l, columns)
    {
        ColumnDef  *entry = lfirst(l);
        AclResult   aclresult;
        Form_pg_attribute att;
 
        /*
         * for each entry in the list, get the name and type information from
         * the list and have TupleDescInitEntry fill in the attribute
         * information we need.
         */
        attnum++;
 
        attname = entry->colname;
        typenameTypeIdAndMod(NULL, entry->typeName, &atttypid, &atttypmod);
 
        aclresult = object_aclcheck(TypeRelationId, atttypid, GetUserId(), ACL_USAGE);
        if (aclresult != ACLCHECK_OK)
            aclcheck_error_type(aclresult, atttypid);
 
        attcollation = GetColumnDefCollation(NULL, entry, atttypid);
        attdim = list_length(entry->typeName->arrayBounds);
        if (attdim > PG_INT16_MAX)
            ereport(ERROR,
                    errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
                    errmsg("too many array dimensions"));
 
        if (entry->typeName->setof)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                     errmsg("column \"%s\" cannot be declared SETOF",
                            attname)));
 
        TupleDescInitEntry(desc, attnum, attname,
                           atttypid, atttypmod, attdim);
        att = TupleDescAttr(desc, attnum - 1);
 
        /* Override TupleDescInitEntry's settings as requested */
        TupleDescInitEntryCollation(desc, attnum, attcollation);
 
        /* Fill in additional stuff not handled by TupleDescInitEntry */
        att->attnotnull = entry->is_not_null;
        att->attislocal = entry->is_local;
        att->attinhcount = entry->inhcount;
        att->attidentity = entry->identity;
        att->attgenerated = entry->generated;
        att->attcompression = GetAttributeCompression(att->atttypid, entry->compression);
        if (entry->storage)
            att->attstorage = entry->storage;
        else if (entry->storage_name)
            att->attstorage = GetAttributeStorage(att->atttypid, entry->storage_name);
 
        populate_compact_attribute(desc, attnum - 1);
    }
 
    TupleDescFinalize(desc);
 
    return desc;
}

References ACL_USAGE, aclcheck_error_type(), ACLCHECK_OK, TypeName::arrayBounds, attname, attnum, ColumnDef::colname, ColumnDef::compression, CreateTemplateTupleDesc(), ereport, errcode(), errmsg, ERROR, fb(), ColumnDef::generated, GetAttributeCompression(), GetAttributeStorage(), GetColumnDefCollation(), GetUserId(), ColumnDef::identity, ColumnDef::inhcount, ColumnDef::is_local, ColumnDef::is_not_null, lfirst, list_length(), object_aclcheck(), PG_INT16_MAX, populate_compact_attribute(), TypeName::setof, ColumnDef::storage, ColumnDef::storage_name, TupleDescAttr(), TupleDescFinalize(), TupleDescInitEntry(), TupleDescInitEntryCollation(), ColumnDef::typeName, and typenameTypeIdAndMod().

Referenced by ATExecAddColumn(), createPartitionTable(), DefineRelation(), and DefineVirtualRelation().

◆ buildExpressionExecutionStates()

static void buildExpressionExecutionStates	(	AlteredTableInfo *	tab,
		Relation	newPartRel,
		EState *	estate
	)

static

Definition at line 22868 of file tablecmds.c.

{
    /*
     * Build the needed expression execution states. Here, we expect only NOT
     * NULL and CHECK constraint.
     */
    foreach_ptr(NewConstraint, con, tab->constraints)
    {
        switch (con->contype)
        {
            case CONSTR_CHECK:
 
                /*
                 * We already expanded virtual expression in
                 * createTableConstraints.
                 */
                con->qualstate = ExecPrepareExpr((Expr *) con->qual, estate);
                break;
            case CONSTR_NOTNULL:
                /* Nothing to do here. */
                break;
            default:
                elog(ERROR, "unrecognized constraint type: %d",
                     (int) con->contype);
        }
    }
 
    /* Expression already planned in createTableConstraints */
    foreach_ptr(NewColumnValue, ex, tab->newvals)
        ex->exprstate = ExecInitExpr((Expr *) ex->expr, NULL);
}

References CONSTR_CHECK, CONSTR_NOTNULL, AlteredTableInfo::constraints, elog, ERROR, ExecInitExpr(), ExecPrepareExpr(), fb(), foreach_ptr, and AlteredTableInfo::newvals.

Referenced by MergePartitionsMoveRows(), and SplitPartitionMoveRows().

◆ change_owner_fix_column_acls()

static void change_owner_fix_column_acls	(	Oid	relationOid,
		Oid	oldOwnerId,
		Oid	newOwnerId
	)

static

Definition at line 17019 of file tablecmds.c.

{
    Relation    attRelation;
    SysScanDesc scan;
    ScanKeyData key[1];
    HeapTuple   attributeTuple;
 
    attRelation = table_open(AttributeRelationId, RowExclusiveLock);
    ScanKeyInit(&key[0],
                Anum_pg_attribute_attrelid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(relationOid));
    scan = systable_beginscan(attRelation, AttributeRelidNumIndexId,
                              true, NULL, 1, key);
    while (HeapTupleIsValid(attributeTuple = systable_getnext(scan)))
    {
        Form_pg_attribute att = (Form_pg_attribute) GETSTRUCT(attributeTuple);
        Datum       repl_val[Natts_pg_attribute];
        bool        repl_null[Natts_pg_attribute];
        bool        repl_repl[Natts_pg_attribute];
        Acl        *newAcl;
        Datum       aclDatum;
        bool        isNull;
        HeapTuple   newtuple;
 
        /* Ignore dropped columns */
        if (att->attisdropped)
            continue;
 
        aclDatum = heap_getattr(attributeTuple,
                                Anum_pg_attribute_attacl,
                                RelationGetDescr(attRelation),
                                &isNull);
        /* Null ACLs do not require changes */
        if (isNull)
            continue;
 
        memset(repl_null, false, sizeof(repl_null));
        memset(repl_repl, false, sizeof(repl_repl));
 
        newAcl = aclnewowner(DatumGetAclP(aclDatum),
                             oldOwnerId, newOwnerId);
        repl_repl[Anum_pg_attribute_attacl - 1] = true;
        repl_val[Anum_pg_attribute_attacl - 1] = PointerGetDatum(newAcl);
 
        newtuple = heap_modify_tuple(attributeTuple,
                                     RelationGetDescr(attRelation),
                                     repl_val, repl_null, repl_repl);
 
        CatalogTupleUpdate(attRelation, &newtuple->t_self, newtuple);
 
        heap_freetuple(newtuple);
    }
    systable_endscan(scan);
    table_close(attRelation, RowExclusiveLock);
}

References aclnewowner(), BTEqualStrategyNumber, CatalogTupleUpdate(), DatumGetAclP, fb(), GETSTRUCT(), heap_freetuple(), heap_getattr(), heap_modify_tuple(), HeapTupleIsValid, ObjectIdGetDatum(), PointerGetDatum, RelationGetDescr, RowExclusiveLock, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecChangeOwner().

◆ change_owner_recurse_to_sequences()

static void change_owner_recurse_to_sequences	(	Oid	relationOid,
		Oid	newOwnerId,
		LOCKMODE	lockmode
	)

static

Definition at line 17084 of file tablecmds.c.

{
    Relation    depRel;
    SysScanDesc scan;
    ScanKeyData key[2];
    HeapTuple   tup;
 
    /*
     * SERIAL sequences are those having an auto dependency on one of the
     * table's columns (we don't care *which* column, exactly).
     */
    depRel = table_open(DependRelationId, AccessShareLock);
 
    ScanKeyInit(&key[0],
                Anum_pg_depend_refclassid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationRelationId));
    ScanKeyInit(&key[1],
                Anum_pg_depend_refobjid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(relationOid));
    /* we leave refobjsubid unspecified */
 
    scan = systable_beginscan(depRel, DependReferenceIndexId, true,
                              NULL, 2, key);
 
    while (HeapTupleIsValid(tup = systable_getnext(scan)))
    {
        Form_pg_depend depForm = (Form_pg_depend) GETSTRUCT(tup);
        Relation    seqRel;
 
        /* skip dependencies other than auto dependencies on columns */
        if (depForm->refobjsubid == 0 ||
            depForm->classid != RelationRelationId ||
            depForm->objsubid != 0 ||
            !(depForm->deptype == DEPENDENCY_AUTO || depForm->deptype == DEPENDENCY_INTERNAL))
            continue;
 
        /* Use relation_open just in case it's an index */
        seqRel = relation_open(depForm->objid, lockmode);
 
        /* skip non-sequence relations */
        if (RelationGetForm(seqRel)->relkind != RELKIND_SEQUENCE)
        {
            /* No need to keep the lock */
            relation_close(seqRel, lockmode);
            continue;
        }
 
        /* We don't need to close the sequence while we alter it. */
        ATExecChangeOwner(depForm->objid, newOwnerId, true, lockmode);
 
        /* Now we can close it.  Keep the lock till end of transaction. */
        relation_close(seqRel, NoLock);
    }
 
    systable_endscan(scan);
 
    relation_close(depRel, AccessShareLock);
}

References AccessShareLock, ATExecChangeOwner(), BTEqualStrategyNumber, DEPENDENCY_AUTO, DEPENDENCY_INTERNAL, fb(), Form_pg_depend, GETSTRUCT(), HeapTupleIsValid, NoLock, ObjectIdGetDatum(), relation_close(), relation_open(), RelationGetForm, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), and table_open().

Referenced by ATExecChangeOwner().

◆ check_for_column_name_collision()

static bool check_for_column_name_collision	(	Relation	rel,
		const char *	colname,
		bool	if_not_exists
	)

static

Definition at line 7787 of file tablecmds.c.

{
    HeapTuple   attTuple;
    int         attnum;
 
    /*
     * this test is deliberately not attisdropped-aware, since if one tries to
     * add a column matching a dropped column name, it's gonna fail anyway.
     */
    attTuple = SearchSysCache2(ATTNAME,
                               ObjectIdGetDatum(RelationGetRelid(rel)),
                               PointerGetDatum(colname));
    if (!HeapTupleIsValid(attTuple))
        return true;
 
    attnum = ((Form_pg_attribute) GETSTRUCT(attTuple))->attnum;
    ReleaseSysCache(attTuple);
 
    /*
     * We throw a different error message for conflicts with system column
     * names, since they are normally not shown and the user might otherwise
     * be confused about the reason for the conflict.
     */
    if (attnum <= 0)
        ereport(ERROR,
                (errcode(ERRCODE_DUPLICATE_COLUMN),
                 errmsg("column name \"%s\" conflicts with a system column name",
                        colname)));
    else
    {
        if (if_not_exists)
        {
            ereport(NOTICE,
                    (errcode(ERRCODE_DUPLICATE_COLUMN),
                     errmsg("column \"%s\" of relation \"%s\" already exists, skipping",
                            colname, RelationGetRelationName(rel))));
            return false;
        }
 
        ereport(ERROR,
                (errcode(ERRCODE_DUPLICATE_COLUMN),
                 errmsg("column \"%s\" of relation \"%s\" already exists",
                        colname, RelationGetRelationName(rel))));
    }
 
    return true;
}

References attnum, ereport, errcode(), errmsg, ERROR, fb(), GETSTRUCT(), HeapTupleIsValid, NOTICE, ObjectIdGetDatum(), PointerGetDatum, RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), and SearchSysCache2().

Referenced by ATExecAddColumn(), and renameatt_internal().

◆ check_of_type()

void check_of_type ( HeapTuple typetuple )

Definition at line 7284 of file tablecmds.c.

{
    Form_pg_type typ = (Form_pg_type) GETSTRUCT(typetuple);
    bool        typeOk = false;
 
    if (typ->typtype == TYPTYPE_COMPOSITE)
    {
        Relation    typeRelation;
 
        Assert(OidIsValid(typ->typrelid));
        typeRelation = relation_open(typ->typrelid, AccessShareLock);
        typeOk = (typeRelation->rd_rel->relkind == RELKIND_COMPOSITE_TYPE);
 
        /*
         * Close the parent rel, but keep our AccessShareLock on it until xact
         * commit.  That will prevent someone else from deleting or ALTERing
         * the type before the typed table creation/conversion commits.
         */
        relation_close(typeRelation, NoLock);
 
        if (!typeOk)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("type %s is the row type of another table",
                            format_type_be(typ->oid)),
                     errdetail("A typed table must use a stand-alone composite type created with CREATE TYPE.")));
    }
    else
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("type %s is not a composite type",
                        format_type_be(typ->oid))));
}

References AccessShareLock, Assert, ereport, errcode(), errdetail(), errmsg, ERROR, fb(), Form_pg_type, format_type_be(), GETSTRUCT(), NoLock, OidIsValid, relation_close(), and relation_open().

Referenced by ATExecAddOf(), and transformOfType().

◆ CheckAlterTableIsSafe()

static void CheckAlterTableIsSafe ( Relation rel )

static

Definition at line 4541 of file tablecmds.c.

{
    /*
     * Don't allow ALTER on temp tables of other backends.  Their local buffer
     * manager is not going to cope if we need to change the table's contents.
     * Even if we don't, there may be optimizations that assume temp tables
     * aren't subject to such interference.
     */
    if (RELATION_IS_OTHER_TEMP(rel))
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot alter temporary tables of other sessions")));
 
    /*
     * Also check for active uses of the relation in the current transaction,
     * including open scans and pending AFTER trigger events.
     */
    CheckTableNotInUse(rel, "ALTER TABLE");
}

References CheckTableNotInUse(), ereport, errcode(), errmsg, ERROR, fb(), and RELATION_IS_OTHER_TEMP.

Referenced by addFkRecurseReferencing(), AlterTable(), ATAddCheckNNConstraint(), ATCheckPartitionsNotInUse(), ATExecAddColumn(), ATExecDropColumn(), ATPrepAlterColumnType(), ATSimpleRecursion(), ATTypedTableRecursion(), dropconstraint_internal(), and set_attnotnull().

◆ checkFkeyPermissions()

static void checkFkeyPermissions	(	Relation	rel,
		int16 *	attnums,
		int	natts
	)

static

Definition at line 14202 of file tablecmds.c.

{
    Oid         roleid = GetUserId();
    AclResult   aclresult;
    int         i;
 
    /* Okay if we have relation-level REFERENCES permission */
    aclresult = pg_class_aclcheck(RelationGetRelid(rel), roleid,
                                  ACL_REFERENCES);
    if (aclresult == ACLCHECK_OK)
        return;
    /* Else we must have REFERENCES on each column */
    for (i = 0; i < natts; i++)
    {
        aclresult = pg_attribute_aclcheck(RelationGetRelid(rel), attnums[i],
                                          roleid, ACL_REFERENCES);
        if (aclresult != ACLCHECK_OK)
            aclcheck_error(aclresult, get_relkind_objtype(rel->rd_rel->relkind),
                           RelationGetRelationName(rel));
    }
}

References ACL_REFERENCES, aclcheck_error(), ACLCHECK_OK, fb(), get_relkind_objtype(), GetUserId(), i, pg_attribute_aclcheck(), pg_class_aclcheck(), RelationData::rd_rel, RelationGetRelationName, and RelationGetRelid.

Referenced by ATAddForeignKeyConstraint().

◆ CheckRelationTableSpaceMove()

bool CheckRelationTableSpaceMove	(	Relation	rel,
		Oid	newTableSpaceId
	)

Definition at line 3772 of file tablecmds.c.

{
    Oid         oldTableSpaceId;
 
    /*
     * No work if no change in tablespace.  Note that MyDatabaseTableSpace is
     * stored as 0.
     */
    oldTableSpaceId = rel->rd_rel->reltablespace;
    if (newTableSpaceId == oldTableSpaceId ||
        (newTableSpaceId == MyDatabaseTableSpace && oldTableSpaceId == 0))
        return false;
 
    /*
     * We cannot support moving mapped relations into different tablespaces.
     * (In particular this eliminates all shared catalogs.)
     */
    if (RelationIsMapped(rel))
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot move system relation \"%s\"",
                        RelationGetRelationName(rel))));
 
    /* Cannot move a non-shared relation into pg_global */
    if (newTableSpaceId == GLOBALTABLESPACE_OID)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                 errmsg("only shared relations can be placed in pg_global tablespace")));
 
    /*
     * Do not allow moving temp tables of other backends ... their local
     * buffer manager is not going to cope.
     */
    if (RELATION_IS_OTHER_TEMP(rel))
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("cannot move temporary tables of other sessions")));
 
    return true;
}

References ereport, errcode(), errmsg, ERROR, fb(), MyDatabaseTableSpace, RelationData::rd_rel, RELATION_IS_OTHER_TEMP, RelationGetRelationName, and RelationIsMapped.

Referenced by ATExecSetTableSpace(), ATExecSetTableSpaceNoStorage(), reindex_index(), and SetRelationTableSpace().

◆ CheckTableNotInUse()

void CheckTableNotInUse	(	Relation	rel,
		const char *	stmt
	)

Definition at line 4508 of file tablecmds.c.

{
    int         expected_refcnt;
 
    expected_refcnt = rel->rd_isnailed ? 2 : 1;
    if (rel->rd_refcnt != expected_refcnt)
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_IN_USE),
        /* translator: first %s is a SQL command, eg ALTER TABLE */
                 errmsg("cannot %s \"%s\" because it is being used by active queries in this session",
                        stmt, RelationGetRelationName(rel))));
 
    if (rel->rd_rel->relkind != RELKIND_INDEX &&
        rel->rd_rel->relkind != RELKIND_PARTITIONED_INDEX &&
        AfterTriggerPendingOnRel(RelationGetRelid(rel)))
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_IN_USE),
        /* translator: first %s is a SQL command, eg ALTER TABLE */
                 errmsg("cannot %s \"%s\" because it has pending trigger events",
                        stmt, RelationGetRelationName(rel))));
}

References AfterTriggerPendingOnRel(), ereport, errcode(), errmsg, ERROR, fb(), RelationData::rd_isnailed, RelationData::rd_refcnt, RelationData::rd_rel, RelationGetRelationName, RelationGetRelid, and stmt.

Referenced by CheckAlterTableIsSafe(), cluster_rel(), DefineIndex(), DefineVirtualRelation(), heap_drop_with_catalog(), index_drop(), MergeAttributes(), RefreshMatViewByOid(), reindex_index(), and truncate_check_activity().

◆ ChooseForeignKeyConstraintNameAddition()

static char * ChooseForeignKeyConstraintNameAddition ( List * colnames )

static

Definition at line 10006 of file tablecmds.c.

{
    char        buf[NAMEDATALEN * 2];
    int         buflen = 0;
    ListCell   *lc;
 
    buf[0] = '\0';
    foreach(lc, colnames)
    {
        const char *name = strVal(lfirst(lc));
 
        if (buflen > 0)
            buf[buflen++] = '_';    /* insert _ between names */
 
        /*
         * At this point we have buflen <= NAMEDATALEN.  name should be less
         * than NAMEDATALEN already, but use strlcpy for paranoia.
         */
        strlcpy(buf + buflen, name, NAMEDATALEN);
        buflen += strlen(buf + buflen);
        if (buflen >= NAMEDATALEN)
            break;
    }
    return pstrdup(buf);
}

References buf, fb(), lfirst, name, NAMEDATALEN, pstrdup(), strlcpy(), and strVal.

Referenced by ATExecAddConstraint().

◆ CloneFkReferenced()

static void CloneFkReferenced	(	Relation	parentRel,
		Relation	partitionRel
	)

static

Definition at line 11394 of file tablecmds.c.

{
    Relation    pg_constraint;
    AttrMap    *attmap;
    ListCell   *cell;
    SysScanDesc scan;
    ScanKeyData key[2];
    HeapTuple   tuple;
    List       *clone = NIL;
    Relation    trigrel;
 
    /*
     * Search for any constraints where this partition's parent is in the
     * referenced side.  However, we must not clone any constraint whose
     * parent constraint is also going to be cloned, to avoid duplicates.  So
     * do it in two steps: first construct the list of constraints to clone,
     * then go over that list cloning those whose parents are not in the list.
     * (We must not rely on the parent being seen first, since the catalog
     * scan could return children first.)
     */
    pg_constraint = table_open(ConstraintRelationId, RowShareLock);
    ScanKeyInit(&key[0],
                Anum_pg_constraint_confrelid, BTEqualStrategyNumber,
                F_OIDEQ, ObjectIdGetDatum(RelationGetRelid(parentRel)));
    ScanKeyInit(&key[1],
                Anum_pg_constraint_contype, BTEqualStrategyNumber,
                F_CHAREQ, CharGetDatum(CONSTRAINT_FOREIGN));
    /* This is a seqscan, as we don't have a usable index ... */
    scan = systable_beginscan(pg_constraint, InvalidOid, true,
                              NULL, 2, key);
    while ((tuple = systable_getnext(scan)) != NULL)
    {
        Form_pg_constraint constrForm = (Form_pg_constraint) GETSTRUCT(tuple);
 
        clone = lappend_oid(clone, constrForm->oid);
    }
    systable_endscan(scan);
    table_close(pg_constraint, RowShareLock);
 
    /*
     * Triggers of the foreign keys will be manipulated a bunch of times in
     * the loop below.  To avoid repeatedly opening/closing the trigger
     * catalog relation, we open it here and pass it to the subroutines called
     * below.
     */
    trigrel = table_open(TriggerRelationId, RowExclusiveLock);
 
    attmap = build_attrmap_by_name(RelationGetDescr(partitionRel),
                                   RelationGetDescr(parentRel),
                                   false);
    foreach(cell, clone)
    {
        Oid         constrOid = lfirst_oid(cell);
        Form_pg_constraint constrForm;
        Relation    fkRel;
        Oid         indexOid;
        Oid         partIndexId;
        int         numfks;
        AttrNumber  conkey[INDEX_MAX_KEYS];
        AttrNumber  mapped_confkey[INDEX_MAX_KEYS];
        AttrNumber  confkey[INDEX_MAX_KEYS];
        Oid         conpfeqop[INDEX_MAX_KEYS];
        Oid         conppeqop[INDEX_MAX_KEYS];
        Oid         conffeqop[INDEX_MAX_KEYS];
        int         numfkdelsetcols;
        AttrNumber  confdelsetcols[INDEX_MAX_KEYS];
        Constraint *fkconstraint;
        ObjectAddress address;
        Oid         deleteTriggerOid = InvalidOid,
                    updateTriggerOid = InvalidOid;
 
        tuple = SearchSysCache1(CONSTROID, ObjectIdGetDatum(constrOid));
        if (!HeapTupleIsValid(tuple))
            elog(ERROR, "cache lookup failed for constraint %u", constrOid);
        constrForm = (Form_pg_constraint) GETSTRUCT(tuple);
 
        /*
         * As explained above: don't try to clone a constraint for which we're
         * going to clone the parent.
         */
        if (list_member_oid(clone, constrForm->conparentid))
        {
            ReleaseSysCache(tuple);
            continue;
        }
 
        /* We need the same lock level that CreateTrigger will acquire */
        fkRel = table_open(constrForm->conrelid, ShareRowExclusiveLock);
 
        indexOid = constrForm->conindid;
        DeconstructFkConstraintRow(tuple,
                                   &numfks,
                                   conkey,
                                   confkey,
                                   conpfeqop,
                                   conppeqop,
                                   conffeqop,
                                   &numfkdelsetcols,
                                   confdelsetcols);
 
        for (int i = 0; i < numfks; i++)
            mapped_confkey[i] = attmap->attnums[confkey[i] - 1];
 
        fkconstraint = makeNode(Constraint);
        fkconstraint->contype = CONSTRAINT_FOREIGN;
        fkconstraint->conname = NameStr(constrForm->conname);
        fkconstraint->deferrable = constrForm->condeferrable;
        fkconstraint->initdeferred = constrForm->condeferred;
        fkconstraint->location = -1;
        fkconstraint->pktable = NULL;
        /* ->fk_attrs determined below */
        fkconstraint->pk_attrs = NIL;
        fkconstraint->fk_matchtype = constrForm->confmatchtype;
        fkconstraint->fk_upd_action = constrForm->confupdtype;
        fkconstraint->fk_del_action = constrForm->confdeltype;
        fkconstraint->fk_del_set_cols = NIL;
        fkconstraint->old_conpfeqop = NIL;
        fkconstraint->old_pktable_oid = InvalidOid;
        fkconstraint->is_enforced = constrForm->conenforced;
        fkconstraint->skip_validation = false;
        fkconstraint->initially_valid = constrForm->convalidated;
 
        /* set up colnames that are used to generate the constraint name */
        for (int i = 0; i < numfks; i++)
        {
            Form_pg_attribute att;
 
            att = TupleDescAttr(RelationGetDescr(fkRel),
                                conkey[i] - 1);
            fkconstraint->fk_attrs = lappend(fkconstraint->fk_attrs,
                                             makeString(NameStr(att->attname)));
        }
 
        /*
         * Add the new foreign key constraint pointing to the new partition.
         * Because this new partition appears in the referenced side of the
         * constraint, we don't need to set up for Phase 3 check.
         */
        partIndexId = index_get_partition(partitionRel, indexOid);
        if (!OidIsValid(partIndexId))
            elog(ERROR, "index for %u not found in partition %s",
                 indexOid, RelationGetRelationName(partitionRel));
 
        /*
         * Get the "action" triggers belonging to the constraint to pass as
         * parent OIDs for similar triggers that will be created on the
         * partition in addFkRecurseReferenced().
         */
        if (constrForm->conenforced)
            GetForeignKeyActionTriggers(trigrel, constrOid,
                                        constrForm->confrelid, constrForm->conrelid,
                                        &deleteTriggerOid, &updateTriggerOid);
 
        /* Add this constraint ... */
        address = addFkConstraint(addFkReferencedSide,
                                  fkconstraint->conname, fkconstraint, fkRel,
                                  partitionRel, partIndexId, constrOid,
                                  numfks, mapped_confkey,
                                  conkey, conpfeqop, conppeqop, conffeqop,
                                  numfkdelsetcols, confdelsetcols, false,
                                  constrForm->conperiod);
        /* ... and recurse */
        addFkRecurseReferenced(fkconstraint,
                               fkRel,
                               partitionRel,
                               partIndexId,
                               address.objectId,
                               numfks,
                               mapped_confkey,
                               conkey,
                               conpfeqop,
                               conppeqop,
                               conffeqop,
                               numfkdelsetcols,
                               confdelsetcols,
                               true,
                               deleteTriggerOid,
                               updateTriggerOid,
                               constrForm->conperiod);
 
        table_close(fkRel, NoLock);
        ReleaseSysCache(tuple);
    }
 
    table_close(trigrel, RowExclusiveLock);
}

References addFkConstraint(), addFkRecurseReferenced(), addFkReferencedSide, BTEqualStrategyNumber, build_attrmap_by_name(), CharGetDatum(), DeconstructFkConstraintRow(), elog, ERROR, fb(), Form_pg_constraint, GetForeignKeyActionTriggers(), GETSTRUCT(), HeapTupleIsValid, i, index_get_partition(), INDEX_MAX_KEYS, InvalidOid, lappend(), lappend_oid(), lfirst_oid, list_member_oid(), makeNode, makeString(), NameStr, NIL, NoLock, ObjectAddress::objectId, ObjectIdGetDatum(), OidIsValid, RelationGetDescr, RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), RowExclusiveLock, RowShareLock, ScanKeyInit(), SearchSysCache1(), ShareRowExclusiveLock, systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), table_open(), and TupleDescAttr().

Referenced by CloneForeignKeyConstraints().

◆ CloneFkReferencing()

static void CloneFkReferencing	(	List **	wqueue,
		Relation	parentRel,
		Relation	partRel
	)

static

Definition at line 11595 of file tablecmds.c.

{
    AttrMap    *attmap;
    List       *partFKs;
    List       *clone = NIL;
    ListCell   *cell;
    Relation    trigrel;
 
    /* obtain a list of constraints that we need to clone */
    foreach(cell, RelationGetFKeyList(parentRel))
    {
        ForeignKeyCacheInfo *fk = lfirst(cell);
 
        /*
         * Refuse to attach a table as partition that this partitioned table
         * already has a foreign key to.  This isn't useful schema, which is
         * proven by the fact that there have been no user complaints that
         * it's already impossible to achieve this in the opposite direction,
         * i.e., creating a foreign key that references a partition.  This
         * restriction allows us to dodge some complexities around
         * pg_constraint and pg_trigger row creations that would be needed
         * during ATTACH/DETACH for this kind of relationship.
         */
        if (fk->confrelid == RelationGetRelid(partRel))
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("cannot attach table \"%s\" as a partition because it is referenced by foreign key \"%s\"",
                            RelationGetRelationName(partRel),
                            get_constraint_name(fk->conoid))));
 
        clone = lappend_oid(clone, fk->conoid);
    }
 
    /*
     * Silently do nothing if there's nothing to do.  In particular, this
     * avoids throwing a spurious error for foreign tables.
     */
    if (clone == NIL)
        return;
 
    if (partRel->rd_rel->relkind == RELKIND_FOREIGN_TABLE)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("foreign key constraints are not supported on foreign tables")));
 
    /*
     * Triggers of the foreign keys will be manipulated a bunch of times in
     * the loop below.  To avoid repeatedly opening/closing the trigger
     * catalog relation, we open it here and pass it to the subroutines called
     * below.
     */
    trigrel = table_open(TriggerRelationId, RowExclusiveLock);
 
    /*
     * The constraint key may differ, if the columns in the partition are
     * different.  This map is used to convert them.
     */
    attmap = build_attrmap_by_name(RelationGetDescr(partRel),
                                   RelationGetDescr(parentRel),
                                   false);
 
    partFKs = copyObject(RelationGetFKeyList(partRel));
 
    foreach(cell, clone)
    {
        Oid         parentConstrOid = lfirst_oid(cell);
        Form_pg_constraint constrForm;
        Relation    pkrel;
        HeapTuple   tuple;
        int         numfks;
        AttrNumber  conkey[INDEX_MAX_KEYS];
        AttrNumber  mapped_conkey[INDEX_MAX_KEYS];
        AttrNumber  confkey[INDEX_MAX_KEYS];
        Oid         conpfeqop[INDEX_MAX_KEYS];
        Oid         conppeqop[INDEX_MAX_KEYS];
        Oid         conffeqop[INDEX_MAX_KEYS];
        int         numfkdelsetcols;
        AttrNumber  confdelsetcols[INDEX_MAX_KEYS];
        Constraint *fkconstraint;
        bool        attached;
        Oid         indexOid;
        ObjectAddress address;
        ListCell   *lc;
        Oid         insertTriggerOid = InvalidOid,
                    updateTriggerOid = InvalidOid;
        bool        with_period;
 
        tuple = SearchSysCache1(CONSTROID, ObjectIdGetDatum(parentConstrOid));
        if (!HeapTupleIsValid(tuple))
            elog(ERROR, "cache lookup failed for constraint %u",
                 parentConstrOid);
        constrForm = (Form_pg_constraint) GETSTRUCT(tuple);
 
        /* Don't clone constraints whose parents are being cloned */
        if (list_member_oid(clone, constrForm->conparentid))
        {
            ReleaseSysCache(tuple);
            continue;
        }
 
        /*
         * Need to prevent concurrent deletions.  If pkrel is a partitioned
         * relation, that means to lock all partitions.
         */
        pkrel = table_open(constrForm->confrelid, ShareRowExclusiveLock);
        if (pkrel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
            (void) find_all_inheritors(RelationGetRelid(pkrel),
                                       ShareRowExclusiveLock, NULL);
 
        DeconstructFkConstraintRow(tuple, &numfks, conkey, confkey,
                                   conpfeqop, conppeqop, conffeqop,
                                   &numfkdelsetcols, confdelsetcols);
        for (int i = 0; i < numfks; i++)
            mapped_conkey[i] = attmap->attnums[conkey[i] - 1];
 
        /*
         * Get the "check" triggers belonging to the constraint, if it is
         * ENFORCED, to pass as parent OIDs for similar triggers that will be
         * created on the partition in addFkRecurseReferencing().  They are
         * also passed to tryAttachPartitionForeignKey() below to simply
         * assign as parents to the partition's existing "check" triggers,
         * that is, if the corresponding constraints is deemed attachable to
         * the parent constraint.
         */
        if (constrForm->conenforced)
            GetForeignKeyCheckTriggers(trigrel, constrForm->oid,
                                       constrForm->confrelid, constrForm->conrelid,
                                       &insertTriggerOid, &updateTriggerOid);
 
        /*
         * Before creating a new constraint, see whether any existing FKs are
         * fit for the purpose.  If one is, attach the parent constraint to
         * it, and don't clone anything.  This way we avoid the expensive
         * verification step and don't end up with a duplicate FK, and we
         * don't need to recurse to partitions for this constraint.
         */
        attached = false;
        foreach(lc, partFKs)
        {
            ForeignKeyCacheInfo *fk = lfirst_node(ForeignKeyCacheInfo, lc);
 
            if (tryAttachPartitionForeignKey(wqueue,
                                             fk,
                                             partRel,
                                             parentConstrOid,
                                             numfks,
                                             mapped_conkey,
                                             confkey,
                                             conpfeqop,
                                             insertTriggerOid,
                                             updateTriggerOid,
                                             trigrel))
            {
                attached = true;
                table_close(pkrel, NoLock);
                break;
            }
        }
        if (attached)
        {
            ReleaseSysCache(tuple);
            continue;
        }
 
        /* No dice.  Set up to create our own constraint */
        fkconstraint = makeNode(Constraint);
        fkconstraint->contype = CONSTRAINT_FOREIGN;
        /* ->conname determined below */
        fkconstraint->deferrable = constrForm->condeferrable;
        fkconstraint->initdeferred = constrForm->condeferred;
        fkconstraint->location = -1;
        fkconstraint->pktable = NULL;
        /* ->fk_attrs determined below */
        fkconstraint->pk_attrs = NIL;
        fkconstraint->fk_matchtype = constrForm->confmatchtype;
        fkconstraint->fk_upd_action = constrForm->confupdtype;
        fkconstraint->fk_del_action = constrForm->confdeltype;
        fkconstraint->fk_del_set_cols = NIL;
        fkconstraint->old_conpfeqop = NIL;
        fkconstraint->old_pktable_oid = InvalidOid;
        fkconstraint->is_enforced = constrForm->conenforced;
        fkconstraint->skip_validation = false;
        fkconstraint->initially_valid = constrForm->convalidated;
        for (int i = 0; i < numfks; i++)
        {
            Form_pg_attribute att;
 
            att = TupleDescAttr(RelationGetDescr(partRel),
                                mapped_conkey[i] - 1);
            fkconstraint->fk_attrs = lappend(fkconstraint->fk_attrs,
                                             makeString(NameStr(att->attname)));
        }
 
        indexOid = constrForm->conindid;
        with_period = constrForm->conperiod;
 
        /* Create the pg_constraint entry at this level */
        address = addFkConstraint(addFkReferencingSide,
                                  NameStr(constrForm->conname), fkconstraint,
                                  partRel, pkrel, indexOid, parentConstrOid,
                                  numfks, confkey,
                                  mapped_conkey, conpfeqop,
                                  conppeqop, conffeqop,
                                  numfkdelsetcols, confdelsetcols,
                                  false, with_period);
 
        /* Done with the cloned constraint's tuple */
        ReleaseSysCache(tuple);
 
        /* Create the check triggers, and recurse to partitions, if any */
        addFkRecurseReferencing(wqueue,
                                fkconstraint,
                                partRel,
                                pkrel,
                                indexOid,
                                address.objectId,
                                numfks,
                                confkey,
                                mapped_conkey,
                                conpfeqop,
                                conppeqop,
                                conffeqop,
                                numfkdelsetcols,
                                confdelsetcols,
                                false,  /* no old check exists */
                                AccessExclusiveLock,
                                insertTriggerOid,
                                updateTriggerOid,
                                with_period);
        table_close(pkrel, NoLock);
    }
 
    table_close(trigrel, RowExclusiveLock);
}

References AccessExclusiveLock, addFkConstraint(), addFkRecurseReferencing(), addFkReferencingSide, build_attrmap_by_name(), copyObject, DeconstructFkConstraintRow(), elog, ereport, errcode(), errmsg, ERROR, fb(), find_all_inheritors(), Form_pg_constraint, get_constraint_name(), GetForeignKeyCheckTriggers(), GETSTRUCT(), HeapTupleIsValid, i, INDEX_MAX_KEYS, InvalidOid, lappend(), lappend_oid(), lfirst, lfirst_node, lfirst_oid, list_member_oid(), makeNode, makeString(), NameStr, NIL, NoLock, ObjectAddress::objectId, ObjectIdGetDatum(), RelationData::rd_rel, RelationGetDescr, RelationGetFKeyList(), RelationGetRelationName, RelationGetRelid, ReleaseSysCache(), RowExclusiveLock, SearchSysCache1(), ShareRowExclusiveLock, table_close(), table_open(), tryAttachPartitionForeignKey(), and TupleDescAttr().

Referenced by CloneForeignKeyConstraints().

◆ CloneForeignKeyConstraints()

static void CloneForeignKeyConstraints	(	List **	wqueue,
		Relation	parentRel,
		Relation	partitionRel
	)

static

Definition at line 11365 of file tablecmds.c.

{
    /* This only works for declarative partitioning */
    Assert(parentRel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE);
 
    /*
     * First, clone constraints where the parent is on the referencing side.
     */
    CloneFkReferencing(wqueue, parentRel, partitionRel);
 
    /*
     * Clone constraints for which the parent is on the referenced side.
     */
    CloneFkReferenced(parentRel, partitionRel);
}

References Assert, CloneFkReferenced(), CloneFkReferencing(), and fb().

Referenced by attachPartitionTable(), and DefineRelation().

◆ CloneRowTriggersToPartition()

static void CloneRowTriggersToPartition	(	Relation	parent,
		Relation	partition
	)

static

Definition at line 21542 of file tablecmds.c.

{
    Relation    pg_trigger;
    ScanKeyData key;
    SysScanDesc scan;
    HeapTuple   tuple;
    MemoryContext perTupCxt;
 
    ScanKeyInit(&key, Anum_pg_trigger_tgrelid, BTEqualStrategyNumber,
                F_OIDEQ, ObjectIdGetDatum(RelationGetRelid(parent)));
    pg_trigger = table_open(TriggerRelationId, RowExclusiveLock);
    scan = systable_beginscan(pg_trigger, TriggerRelidNameIndexId,
                              true, NULL, 1, &key);
 
    perTupCxt = AllocSetContextCreate(CurrentMemoryContext,
                                      "clone trig", ALLOCSET_SMALL_SIZES);
 
    while (HeapTupleIsValid(tuple = systable_getnext(scan)))
    {
        Form_pg_trigger trigForm = (Form_pg_trigger) GETSTRUCT(tuple);
        CreateTrigStmt *trigStmt;
        Node       *qual = NULL;
        Datum       value;
        bool        isnull;
        List       *cols = NIL;
        List       *trigargs = NIL;
        MemoryContext oldcxt;
 
        /*
         * Ignore statement-level triggers; those are not cloned.
         */
        if (!TRIGGER_FOR_ROW(trigForm->tgtype))
            continue;
 
        /*
         * Don't clone internal triggers, because the constraint cloning code
         * will.
         */
        if (trigForm->tgisinternal)
            continue;
 
        /*
         * Complain if we find an unexpected trigger type.
         */
        if (!TRIGGER_FOR_BEFORE(trigForm->tgtype) &&
            !TRIGGER_FOR_AFTER(trigForm->tgtype))
            elog(ERROR, "unexpected trigger \"%s\" found",
                 NameStr(trigForm->tgname));
 
        /* Use short-lived context for CREATE TRIGGER */
        oldcxt = MemoryContextSwitchTo(perTupCxt);
 
        /*
         * If there is a WHEN clause, generate a 'cooked' version of it that's
         * appropriate for the partition.
         */
        value = heap_getattr(tuple, Anum_pg_trigger_tgqual,
                             RelationGetDescr(pg_trigger), &isnull);
        if (!isnull)
        {
            qual = stringToNode(TextDatumGetCString(value));
            qual = (Node *) map_partition_varattnos((List *) qual, PRS2_OLD_VARNO,
                                                    partition, parent);
            qual = (Node *) map_partition_varattnos((List *) qual, PRS2_NEW_VARNO,
                                                    partition, parent);
        }
 
        /*
         * If there is a column list, transform it to a list of column names.
         * Note we don't need to map this list in any way ...
         */
        if (trigForm->tgattr.dim1 > 0)
        {
            int         i;
 
            for (i = 0; i < trigForm->tgattr.dim1; i++)
            {
                Form_pg_attribute col;
 
                col = TupleDescAttr(parent->rd_att,
                                    trigForm->tgattr.values[i] - 1);
                cols = lappend(cols,
                               makeString(pstrdup(NameStr(col->attname))));
            }
        }
 
        /* Reconstruct trigger arguments list. */
        if (trigForm->tgnargs > 0)
        {
            char       *p;
 
            value = heap_getattr(tuple, Anum_pg_trigger_tgargs,
                                 RelationGetDescr(pg_trigger), &isnull);
            if (isnull)
                elog(ERROR, "tgargs is null for trigger \"%s\" in partition \"%s\"",
                     NameStr(trigForm->tgname), RelationGetRelationName(partition));
 
            p = (char *) VARDATA_ANY(DatumGetByteaPP(value));
 
            for (int i = 0; i < trigForm->tgnargs; i++)
            {
                trigargs = lappend(trigargs, makeString(pstrdup(p)));
                p += strlen(p) + 1;
            }
        }
 
        trigStmt = makeNode(CreateTrigStmt);
        trigStmt->replace = false;
        trigStmt->isconstraint = OidIsValid(trigForm->tgconstraint);
        trigStmt->trigname = NameStr(trigForm->tgname);
        trigStmt->relation = NULL;
        trigStmt->funcname = NULL;  /* passed separately */
        trigStmt->args = trigargs;
        trigStmt->row = true;
        trigStmt->timing = trigForm->tgtype & TRIGGER_TYPE_TIMING_MASK;
        trigStmt->events = trigForm->tgtype & TRIGGER_TYPE_EVENT_MASK;
        trigStmt->columns = cols;
        trigStmt->whenClause = NULL;    /* passed separately */
        trigStmt->transitionRels = NIL; /* not supported at present */
        trigStmt->deferrable = trigForm->tgdeferrable;
        trigStmt->initdeferred = trigForm->tginitdeferred;
        trigStmt->constrrel = NULL; /* passed separately */
 
        CreateTriggerFiringOn(trigStmt, NULL, RelationGetRelid(partition),
                              trigForm->tgconstrrelid, InvalidOid, InvalidOid,
                              trigForm->tgfoid, trigForm->oid, qual,
                              false, true, trigForm->tgenabled);
 
        MemoryContextSwitchTo(oldcxt);
        MemoryContextReset(perTupCxt);
    }
 
    MemoryContextDelete(perTupCxt);
 
    systable_endscan(scan);
    table_close(pg_trigger, RowExclusiveLock);
}

References ALLOCSET_SMALL_SIZES, AllocSetContextCreate, BTEqualStrategyNumber, CreateTriggerFiringOn(), CurrentMemoryContext, DatumGetByteaPP, elog, ERROR, fb(), Form_pg_trigger, GETSTRUCT(), heap_getattr(), HeapTupleIsValid, i, InvalidOid, lappend(), makeNode, makeString(), map_partition_varattnos(), MemoryContextDelete(), MemoryContextReset(), MemoryContextSwitchTo(), NameStr, NIL, ObjectIdGetDatum(), OidIsValid, PRS2_NEW_VARNO, PRS2_OLD_VARNO, pstrdup(), RelationData::rd_att, RelationGetDescr, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, ScanKeyInit(), stringToNode(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), table_open(), TextDatumGetCString, TupleDescAttr(), value, and VARDATA_ANY().

Referenced by attachPartitionTable(), and DefineRelation().

◆ cmp_partition_index_ext_dep()

static int cmp_partition_index_ext_dep	(	const ListCell *	a,
		const ListCell *	b
	)

static

Definition at line 23561 of file tablecmds.c.

{
    const PartitionIndexExtDepEntry *ea = lfirst(a);
    const PartitionIndexExtDepEntry *eb = lfirst(b);
 
    if (ea->parentIndexOid != eb->parentIndexOid)
        return pg_cmp_u32(ea->parentIndexOid, eb->parentIndexOid);
    return pg_cmp_u32(ea->indexOid, eb->indexOid);
}

References a, b, fb(), lfirst, and pg_cmp_u32().

Referenced by collectPartitionIndexExtDeps().

◆ collectPartitionIndexExtDeps()

static List * collectPartitionIndexExtDeps ( List * partitionOids )

static

Definition at line 23589 of file tablecmds.c.

{
    List       *collected = NIL;
    List       *result = NIL;
    PartitionIndexExtDepEntry *prev = NULL;
 
    /*
     * Phase 1: collect one entry per (partition index -> parent index) pair,
     * with its extension dependency OIDs sorted ascending.
     */
    foreach_oid(partOid, partitionOids)
    {
        Relation    partRel;
        List       *indexList;
 
        /*
         * Use NoLock since the caller already holds AccessExclusiveLock on
         * these partitions.
         */
        partRel = table_open(partOid, NoLock);
        indexList = RelationGetIndexList(partRel);
 
        foreach_oid(indexOid, indexList)
        {
            Oid         parentIndexOid;
            PartitionIndexExtDepEntry *entry;
 
            if (!get_rel_relispartition(indexOid))
                continue;
 
            parentIndexOid = get_partition_parent(indexOid, true);
            if (!OidIsValid(parentIndexOid))
                continue;
 
            entry = palloc(sizeof(PartitionIndexExtDepEntry));
            entry->parentIndexOid = parentIndexOid;
            entry->indexOid = indexOid;
            entry->extensionOids = getAutoExtensionsOfObject(RelationRelationId,
                                                             indexOid);
            list_sort(entry->extensionOids, list_oid_cmp);
 
            collected = lappend(collected, entry);
        }
 
        list_free(indexList);
        table_close(partRel, NoLock);
    }
 
    /*
     * Phase 2: sort by parentIndexOid so entries sharing a parent index sit
     * adjacent.
     */
    list_sort(collected, cmp_partition_index_ext_dep);
 
    /*
     * Phase 3: single linear pass verifying that adjacent entries sharing a
     * parent index have identical extension dependencies, and keeping one
     * representative entry per parent index.
     */
    foreach_ptr(PartitionIndexExtDepEntry, entry, collected)
    {
        if (prev != NULL && prev->parentIndexOid == entry->parentIndexOid)
        {
            if (!equal_oid_lists(prev->extensionOids, entry->extensionOids))
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("cannot merge partitions with conflicting extension dependencies"),
                         errdetail("Partition indexes \"%s\" and \"%s\" depend on different extensions.",
                                   get_rel_name(prev->indexOid),
                                   get_rel_name(entry->indexOid))));
 
            /* Duplicate entry for the same parent index; discard. */
            list_free(entry->extensionOids);
            pfree(entry);
            continue;
        }
 
        result = lappend(result, entry);
        prev = entry;
    }
 
    list_free(collected);
 
    return result;
}

References cmp_partition_index_ext_dep(), equal_oid_lists(), ereport, errcode(), errdetail(), errmsg, ERROR, PartitionIndexExtDepEntry::extensionOids, fb(), foreach_oid, foreach_ptr, get_partition_parent(), get_rel_name(), get_rel_relispartition(), getAutoExtensionsOfObject(), PartitionIndexExtDepEntry::indexOid, lappend(), list_free(), list_oid_cmp(), list_sort(), NIL, NoLock, OidIsValid, palloc(), PartitionIndexExtDepEntry::parentIndexOid, pfree(), RelationGetIndexList(), result, table_close(), and table_open().

Referenced by ATExecMergePartitions(), and ATExecSplitPartition().

◆ ComputePartitionAttrs()

static void ComputePartitionAttrs	(	ParseState *	pstate,
		Relation	rel,
		List *	partParams,
		AttrNumber *	partattrs,
		List **	partexprs,
		Oid *	partopclass,
		Oid *	partcollation,
		PartitionStrategy	strategy
	)

static

Definition at line 20517 of file tablecmds.c.

{
    int         attn;
    ListCell   *lc;
    Oid         am_oid;
 
    attn = 0;
    foreach(lc, partParams)
    {
        PartitionElem *pelem = lfirst_node(PartitionElem, lc);
        Oid         atttype;
        Oid         attcollation;
 
        if (pelem->name != NULL)
        {
            /* Simple attribute reference */
            HeapTuple   atttuple;
            Form_pg_attribute attform;
 
            atttuple = SearchSysCacheAttName(RelationGetRelid(rel),
                                             pelem->name);
            if (!HeapTupleIsValid(atttuple))
                ereport(ERROR,
                        (errcode(ERRCODE_UNDEFINED_COLUMN),
                         errmsg("column \"%s\" named in partition key does not exist",
                                pelem->name),
                         parser_errposition(pstate, pelem->location)));
            attform = (Form_pg_attribute) GETSTRUCT(atttuple);
 
            if (attform->attnum <= 0)
                ereport(ERROR,
                        (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                         errmsg("cannot use system column \"%s\" in partition key",
                                pelem->name),
                         parser_errposition(pstate, pelem->location)));
 
            /*
             * Stored generated columns cannot work: They are computed after
             * BEFORE triggers, but partition routing is done before all
             * triggers.  Maybe virtual generated columns could be made to
             * work, but then they would need to be handled as an expression
             * below.
             */
            if (attform->attgenerated)
                ereport(ERROR,
                        (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                         errmsg("cannot use generated column in partition key"),
                         errdetail("Column \"%s\" is a generated column.",
                                   pelem->name),
                         parser_errposition(pstate, pelem->location)));
 
            partattrs[attn] = attform->attnum;
            atttype = attform->atttypid;
            attcollation = attform->attcollation;
            ReleaseSysCache(atttuple);
        }
        else
        {
            /* Expression */
            Node       *expr = pelem->expr;
            char        partattname[16];
            Bitmapset  *expr_attrs = NULL;
            int         i;
 
            Assert(expr != NULL);
            atttype = exprType(expr);
            attcollation = exprCollation(expr);
 
            /*
             * The expression must be of a storable type (e.g., not RECORD).
             * The test is the same as for whether a table column is of a safe
             * type (which is why we needn't check for the non-expression
             * case).
             */
            snprintf(partattname, sizeof(partattname), "%d", attn + 1);
            CheckAttributeType(partattname,
                               atttype, attcollation,
                               NIL, CHKATYPE_IS_PARTKEY);
 
            /*
             * Strip any top-level COLLATE clause.  This ensures that we treat
             * "x COLLATE y" and "(x COLLATE y)" alike.
             */
            while (IsA(expr, CollateExpr))
                expr = (Node *) ((CollateExpr *) expr)->arg;
 
            /*
             * Examine all the columns in the partition key expression. When
             * the whole-row reference is present, examine all the columns of
             * the partitioned table.
             */
            pull_varattnos(expr, 1, &expr_attrs);
            if (bms_is_member(0 - FirstLowInvalidHeapAttributeNumber, expr_attrs))
            {
                expr_attrs = bms_add_range(expr_attrs,
                                           1 - FirstLowInvalidHeapAttributeNumber,
                                           RelationGetNumberOfAttributes(rel) - FirstLowInvalidHeapAttributeNumber);
                expr_attrs = bms_del_member(expr_attrs, 0 - FirstLowInvalidHeapAttributeNumber);
            }
 
            i = -1;
            while ((i = bms_next_member(expr_attrs, i)) >= 0)
            {
                AttrNumber  attno = i + FirstLowInvalidHeapAttributeNumber;
 
                Assert(attno != 0);
 
                /*
                 * Cannot allow system column references, since that would
                 * make partition routing impossible: their values won't be
                 * known yet when we need to do that.
                 */
                if (attno < 0)
                    ereport(ERROR,
                            (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                             errmsg("partition key expressions cannot contain system column references")));
 
                /*
                 * Stored generated columns cannot work: They are computed
                 * after BEFORE triggers, but partition routing is done before
                 * all triggers.  Virtual generated columns could probably
                 * work, but it would require more work elsewhere (for example
                 * SET EXPRESSION would need to check whether the column is
                 * used in partition keys).  Seems safer to prohibit for now.
                 */
                if (TupleDescAttr(RelationGetDescr(rel), attno - 1)->attgenerated)
                    ereport(ERROR,
                            (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                             errmsg("cannot use generated column in partition key"),
                             errdetail("Column \"%s\" is a generated column.",
                                       get_attname(RelationGetRelid(rel), attno, false)),
                             parser_errposition(pstate, pelem->location)));
            }
 
            if (IsA(expr, Var) &&
                ((Var *) expr)->varattno > 0)
            {
 
                /*
                 * User wrote "(column)" or "(column COLLATE something)".
                 * Treat it like simple attribute anyway.
                 */
                partattrs[attn] = ((Var *) expr)->varattno;
            }
            else
            {
                partattrs[attn] = 0;    /* marks the column as expression */
                *partexprs = lappend(*partexprs, expr);
 
                /*
                 * transformPartitionSpec() should have already rejected
                 * subqueries, aggregates, window functions, and SRFs, based
                 * on the EXPR_KIND_ for partition expressions.
                 */
 
                /*
                 * Preprocess the expression before checking for mutability.
                 * This is essential for the reasons described in
                 * contain_mutable_functions_after_planning.  However, we call
                 * expression_planner for ourselves rather than using that
                 * function, because if constant-folding reduces the
                 * expression to a constant, we'd like to know that so we can
                 * complain below.
                 *
                 * Like contain_mutable_functions_after_planning, assume that
                 * expression_planner won't scribble on its input, so this
                 * won't affect the partexprs entry we saved above.
                 */
                expr = (Node *) expression_planner((Expr *) expr);
 
                /*
                 * Partition expressions cannot contain mutable functions,
                 * because a given row must always map to the same partition
                 * as long as there is no change in the partition boundary
                 * structure.
                 */
                if (contain_mutable_functions(expr))
                    ereport(ERROR,
                            (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                             errmsg("functions in partition key expression must be marked IMMUTABLE")));
 
                /*
                 * While it is not exactly *wrong* for a partition expression
                 * to be a constant, it seems better to reject such keys.
                 */
                if (IsA(expr, Const))
                    ereport(ERROR,
                            (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                             errmsg("cannot use constant expression as partition key")));
            }
        }
 
        /*
         * Apply collation override if any
         */
        if (pelem->collation)
            attcollation = get_collation_oid(pelem->collation, false);
 
        /*
         * Check we have a collation iff it's a collatable type.  The only
         * expected failures here are (1) COLLATE applied to a noncollatable
         * type, or (2) partition expression had an unresolved collation. But
         * we might as well code this to be a complete consistency check.
         */
        if (type_is_collatable(atttype))
        {
            if (!OidIsValid(attcollation))
                ereport(ERROR,
                        (errcode(ERRCODE_INDETERMINATE_COLLATION),
                         errmsg("could not determine which collation to use for partition expression"),
                         errhint("Use the COLLATE clause to set the collation explicitly.")));
        }
        else
        {
            if (OidIsValid(attcollation))
                ereport(ERROR,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("collations are not supported by type %s",
                                format_type_be(atttype))));
        }
 
        partcollation[attn] = attcollation;
 
        /*
         * Identify the appropriate operator class.  For list and range
         * partitioning, we use a btree operator class; hash partitioning uses
         * a hash operator class.
         */
        if (strategy == PARTITION_STRATEGY_HASH)
            am_oid = HASH_AM_OID;
        else
            am_oid = BTREE_AM_OID;
 
        if (!pelem->opclass)
        {
            partopclass[attn] = GetDefaultOpClass(atttype, am_oid);
 
            if (!OidIsValid(partopclass[attn]))
            {
                if (strategy == PARTITION_STRATEGY_HASH)
                    ereport(ERROR,
                            (errcode(ERRCODE_UNDEFINED_OBJECT),
                             errmsg("data type %s has no default operator class for access method \"%s\"",
                                    format_type_be(atttype), "hash"),
                             errhint("You must specify a hash operator class or define a default hash operator class for the data type.")));
                else
                    ereport(ERROR,
                            (errcode(ERRCODE_UNDEFINED_OBJECT),
                             errmsg("data type %s has no default operator class for access method \"%s\"",
                                    format_type_be(atttype), "btree"),
                             errhint("You must specify a btree operator class or define a default btree operator class for the data type.")));
            }
        }
        else
            partopclass[attn] = ResolveOpClass(pelem->opclass,
                                               atttype,
                                               am_oid == HASH_AM_OID ? "hash" : "btree",
                                               am_oid);
 
        attn++;
    }
}

References arg, Assert, bms_add_range(), bms_del_member(), bms_is_member(), bms_next_member(), CheckAttributeType(), CHKATYPE_IS_PARTKEY, contain_mutable_functions(), ereport, errcode(), errdetail(), errhint(), errmsg, ERROR, exprCollation(), expression_planner(), exprType(), fb(), FirstLowInvalidHeapAttributeNumber, format_type_be(), get_attname(), get_collation_oid(), GetDefaultOpClass(), GETSTRUCT(), HeapTupleIsValid, i, IsA, lappend(), lfirst_node, NIL, OidIsValid, parser_errposition(), PARTITION_STRATEGY_HASH, pull_varattnos(), RelationGetDescr, RelationGetNumberOfAttributes, RelationGetRelid, ReleaseSysCache(), ResolveOpClass(), SearchSysCacheAttName(), snprintf, TupleDescAttr(), and type_is_collatable().

Referenced by DefineRelation().

◆ ConstraintImpliedByRelConstraint()

bool ConstraintImpliedByRelConstraint	(	Relation	scanrel,
		List *	testConstraint,
		List *	provenConstraint
	)

static

Definition at line 20846 of file tablecmds.c.

{
    List       *existConstraint = list_copy(provenConstraint);
    TupleConstr *constr = RelationGetDescr(scanrel)->constr;
    int         num_check,
                i;
 
    num_check = (constr != NULL) ? constr->num_check : 0;
    for (i = 0; i < num_check; i++)
    {
        Node       *cexpr;
 
        /*
         * If this constraint hasn't been fully validated yet, we must ignore
         * it here.
         */
        if (!constr->check[i].ccvalid)
            continue;
 
        /*
         * NOT ENFORCED constraints are always marked as invalid, which should
         * have been ignored.
         */
        Assert(constr->check[i].ccenforced);
 
        cexpr = stringToNode(constr->check[i].ccbin);
 
        /*
         * Run each expression through const-simplification and
         * canonicalization.  It is necessary, because we will be comparing it
         * to similarly-processed partition constraint expressions, and may
         * fail to detect valid matches without this.
         */
        cexpr = eval_const_expressions(NULL, cexpr);
        cexpr = (Node *) canonicalize_qual((Expr *) cexpr, true);
 
        existConstraint = list_concat(existConstraint,
                                      make_ands_implicit((Expr *) cexpr));
    }
 
    /*
     * Try to make the proof.  Since we are comparing CHECK constraints, we
     * need to use weak implication, i.e., we assume existConstraint is
     * not-false and try to prove the same for testConstraint.
     *
     * Note that predicate_implied_by assumes its first argument is known
     * immutable.  That should always be true for both NOT NULL and partition
     * constraints, so we don't test it here.
     */
    return predicate_implied_by(testConstraint, existConstraint, true);
}

References Assert, canonicalize_qual(), ConstrCheck::ccbin, ConstrCheck::ccenforced, ConstrCheck::ccvalid, TupleConstr::check, eval_const_expressions(), fb(), i, list_concat(), list_copy(), make_ands_implicit(), TupleConstr::num_check, predicate_implied_by(), RelationGetDescr, and stringToNode().

Referenced by NotNullImpliedByRelConstraints(), and PartConstraintImpliedByRelConstraint().

◆ constraints_equivalent()

static bool constraints_equivalent	(	HeapTuple	a,
		HeapTuple	b,
		TupleDesc	tupleDesc
	)

static

Definition at line 18179 of file tablecmds.c.

{
    Form_pg_constraint acon = (Form_pg_constraint) GETSTRUCT(a);
    Form_pg_constraint bcon = (Form_pg_constraint) GETSTRUCT(b);
 
    if (acon->condeferrable != bcon->condeferrable ||
        acon->condeferred != bcon->condeferred ||
        strcmp(decompile_conbin(a, tupleDesc),
               decompile_conbin(b, tupleDesc)) != 0)
        return false;
    else
        return true;
}

References a, b, decompile_conbin(), fb(), Form_pg_constraint, and GETSTRUCT().

Referenced by ATCheckCheckConstrHasEnforcedParent(), and MergeConstraintsIntoExisting().

◆ CreateFKCheckTrigger()

static Oid CreateFKCheckTrigger	(	Oid	myRelOid,
		Oid	refRelOid,
		Constraint *	fkconstraint,
		Oid	constraintOid,
		Oid	indexOid,
		Oid	parentTrigOid,
		bool	on_insert
	)

static

Definition at line 14331 of file tablecmds.c.

{
    ObjectAddress trigAddress;
    CreateTrigStmt *fk_trigger;
 
    /*
     * Note: for a self-referential FK (referencing and referenced tables are
     * the same), it is important that the ON UPDATE action fires before the
     * CHECK action, since both triggers will fire on the same row during an
     * UPDATE event; otherwise the CHECK trigger will be checking a non-final
     * state of the row.  Triggers fire in name order, so we ensure this by
     * using names like "RI_ConstraintTrigger_a_NNNN" for the action triggers
     * and "RI_ConstraintTrigger_c_NNNN" for the check triggers.
     */
    fk_trigger = makeNode(CreateTrigStmt);
    fk_trigger->replace = false;
    fk_trigger->isconstraint = true;
    fk_trigger->trigname = "RI_ConstraintTrigger_c";
    fk_trigger->relation = NULL;
 
    /* Either ON INSERT or ON UPDATE */
    if (on_insert)
    {
        fk_trigger->funcname = SystemFuncName("RI_FKey_check_ins");
        fk_trigger->events = TRIGGER_TYPE_INSERT;
    }
    else
    {
        fk_trigger->funcname = SystemFuncName("RI_FKey_check_upd");
        fk_trigger->events = TRIGGER_TYPE_UPDATE;
    }
 
    fk_trigger->args = NIL;
    fk_trigger->row = true;
    fk_trigger->timing = TRIGGER_TYPE_AFTER;
    fk_trigger->columns = NIL;
    fk_trigger->whenClause = NULL;
    fk_trigger->transitionRels = NIL;
    fk_trigger->deferrable = fkconstraint->deferrable;
    fk_trigger->initdeferred = fkconstraint->initdeferred;
    fk_trigger->constrrel = NULL;
 
    trigAddress = CreateTrigger(fk_trigger, NULL, myRelOid, refRelOid,
                                constraintOid, indexOid, InvalidOid,
                                parentTrigOid, NULL, true, false);
 
    /* Make changes-so-far visible */
    CommandCounterIncrement();
 
    return trigAddress.objectId;
}

References CommandCounterIncrement(), CreateTrigger(), fb(), InvalidOid, makeNode, NIL, and SystemFuncName().

Referenced by createForeignKeyCheckTriggers().

◆ createForeignKeyActionTriggers()

static void createForeignKeyActionTriggers	(	Oid	myRelOid,
		Oid	refRelOid,
		Constraint *	fkconstraint,
		Oid	constraintOid,
		Oid	indexOid,
		Oid	parentDelTrigger,
		Oid	parentUpdTrigger,
		Oid *	deleteTrigOid,
		Oid *	updateTrigOid
	)

static

Definition at line 14394 of file tablecmds.c.

{
    CreateTrigStmt *fk_trigger;
    ObjectAddress trigAddress;
 
    /*
     * Build and execute a CREATE CONSTRAINT TRIGGER statement for the ON
     * DELETE action on the referenced table.
     */
    fk_trigger = makeNode(CreateTrigStmt);
    fk_trigger->replace = false;
    fk_trigger->isconstraint = true;
    fk_trigger->trigname = "RI_ConstraintTrigger_a";
    fk_trigger->relation = NULL;
    fk_trigger->args = NIL;
    fk_trigger->row = true;
    fk_trigger->timing = TRIGGER_TYPE_AFTER;
    fk_trigger->events = TRIGGER_TYPE_DELETE;
    fk_trigger->columns = NIL;
    fk_trigger->whenClause = NULL;
    fk_trigger->transitionRels = NIL;
    fk_trigger->constrrel = NULL;
 
    switch (fkconstraint->fk_del_action)
    {
        case FKCONSTR_ACTION_NOACTION:
            fk_trigger->deferrable = fkconstraint->deferrable;
            fk_trigger->initdeferred = fkconstraint->initdeferred;
            fk_trigger->funcname = SystemFuncName("RI_FKey_noaction_del");
            break;
        case FKCONSTR_ACTION_RESTRICT:
            fk_trigger->deferrable = false;
            fk_trigger->initdeferred = false;
            fk_trigger->funcname = SystemFuncName("RI_FKey_restrict_del");
            break;
        case FKCONSTR_ACTION_CASCADE:
            fk_trigger->deferrable = false;
            fk_trigger->initdeferred = false;
            fk_trigger->funcname = SystemFuncName("RI_FKey_cascade_del");
            break;
        case FKCONSTR_ACTION_SETNULL:
            fk_trigger->deferrable = false;
            fk_trigger->initdeferred = false;
            fk_trigger->funcname = SystemFuncName("RI_FKey_setnull_del");
            break;
        case FKCONSTR_ACTION_SETDEFAULT:
            fk_trigger->deferrable = false;
            fk_trigger->initdeferred = false;
            fk_trigger->funcname = SystemFuncName("RI_FKey_setdefault_del");
            break;
        default:
            elog(ERROR, "unrecognized FK action type: %d",
                 (int) fkconstraint->fk_del_action);
            break;
    }
 
    trigAddress = CreateTrigger(fk_trigger, NULL, refRelOid, myRelOid,
                                constraintOid, indexOid, InvalidOid,
                                parentDelTrigger, NULL, true, false);
    if (deleteTrigOid)
        *deleteTrigOid = trigAddress.objectId;
 
    /* Make changes-so-far visible */
    CommandCounterIncrement();
 
    /*
     * Build and execute a CREATE CONSTRAINT TRIGGER statement for the ON
     * UPDATE action on the referenced table.
     */
    fk_trigger = makeNode(CreateTrigStmt);
    fk_trigger->replace = false;
    fk_trigger->isconstraint = true;
    fk_trigger->trigname = "RI_ConstraintTrigger_a";
    fk_trigger->relation = NULL;
    fk_trigger->args = NIL;
    fk_trigger->row = true;
    fk_trigger->timing = TRIGGER_TYPE_AFTER;
    fk_trigger->events = TRIGGER_TYPE_UPDATE;
    fk_trigger->columns = NIL;
    fk_trigger->whenClause = NULL;
    fk_trigger->transitionRels = NIL;
    fk_trigger->constrrel = NULL;
 
    switch (fkconstraint->fk_upd_action)
    {
        case FKCONSTR_ACTION_NOACTION:
            fk_trigger->deferrable = fkconstraint->deferrable;
            fk_trigger->initdeferred = fkconstraint->initdeferred;
            fk_trigger->funcname = SystemFuncName("RI_FKey_noaction_upd");
            break;
        case FKCONSTR_ACTION_RESTRICT:
            fk_trigger->deferrable = false;
            fk_trigger->initdeferred = false;
            fk_trigger->funcname = SystemFuncName("RI_FKey_restrict_upd");
            break;
        case FKCONSTR_ACTION_CASCADE:
            fk_trigger->deferrable = false;
            fk_trigger->initdeferred = false;
            fk_trigger->funcname = SystemFuncName("RI_FKey_cascade_upd");
            break;
        case FKCONSTR_ACTION_SETNULL:
            fk_trigger->deferrable = false;
            fk_trigger->initdeferred = false;
            fk_trigger->funcname = SystemFuncName("RI_FKey_setnull_upd");
            break;
        case FKCONSTR_ACTION_SETDEFAULT:
            fk_trigger->deferrable = false;
            fk_trigger->initdeferred = false;
            fk_trigger->funcname = SystemFuncName("RI_FKey_setdefault_upd");
            break;
        default:
            elog(ERROR, "unrecognized FK action type: %d",
                 (int) fkconstraint->fk_upd_action);
            break;
    }
 
    trigAddress = CreateTrigger(fk_trigger, NULL, refRelOid, myRelOid,
                                constraintOid, indexOid, InvalidOid,
                                parentUpdTrigger, NULL, true, false);
    if (updateTrigOid)
        *updateTrigOid = trigAddress.objectId;
}

References CommandCounterIncrement(), CreateTrigger(), elog, ERROR, fb(), FKCONSTR_ACTION_CASCADE, FKCONSTR_ACTION_NOACTION, FKCONSTR_ACTION_RESTRICT, FKCONSTR_ACTION_SETDEFAULT, FKCONSTR_ACTION_SETNULL, InvalidOid, makeNode, NIL, and SystemFuncName().

Referenced by addFkRecurseReferenced(), and ATExecAlterFKConstrEnforceability().

◆ createForeignKeyCheckTriggers()

static void createForeignKeyCheckTriggers	(	Oid	myRelOid,
		Oid	refRelOid,
		Constraint *	fkconstraint,
		Oid	constraintOid,
		Oid	indexOid,
		Oid	parentInsTrigger,
		Oid	parentUpdTrigger,
		Oid *	insertTrigOid,
		Oid *	updateTrigOid
	)

static

Definition at line 14529 of file tablecmds.c.

{
    *insertTrigOid = CreateFKCheckTrigger(myRelOid, refRelOid, fkconstraint,
                                          constraintOid, indexOid,
                                          parentInsTrigger, true);
    *updateTrigOid = CreateFKCheckTrigger(myRelOid, refRelOid, fkconstraint,
                                          constraintOid, indexOid,
                                          parentUpdTrigger, false);
}

References CreateFKCheckTrigger(), and fb().

Referenced by addFkRecurseReferencing(), and ATExecAlterFKConstrEnforceability().

◆ CreateInheritance()

static void CreateInheritance	(	Relation	child_rel,
		Relation	parent_rel,
		bool	ispartition
	)

static

Definition at line 18082 of file tablecmds.c.

{
    Relation    catalogRelation;
    SysScanDesc scan;
    ScanKeyData key;
    HeapTuple   inheritsTuple;
    int32       inhseqno;
 
    /* Note: get RowExclusiveLock because we will write pg_inherits below. */
    catalogRelation = table_open(InheritsRelationId, RowExclusiveLock);
 
    /*
     * Check for duplicates in the list of parents, and determine the highest
     * inhseqno already present; we'll use the next one for the new parent.
     * Also, if proposed child is a partition, it cannot already be
     * inheriting.
     *
     * Note: we do not reject the case where the child already inherits from
     * the parent indirectly; CREATE TABLE doesn't reject comparable cases.
     */
    ScanKeyInit(&key,
                Anum_pg_inherits_inhrelid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationGetRelid(child_rel)));
    scan = systable_beginscan(catalogRelation, InheritsRelidSeqnoIndexId,
                              true, NULL, 1, &key);
 
    /* inhseqno sequences start at 1 */
    inhseqno = 0;
    while (HeapTupleIsValid(inheritsTuple = systable_getnext(scan)))
    {
        Form_pg_inherits inh = (Form_pg_inherits) GETSTRUCT(inheritsTuple);
 
        if (inh->inhparent == RelationGetRelid(parent_rel))
            ereport(ERROR,
                    (errcode(ERRCODE_DUPLICATE_TABLE),
                     errmsg("relation \"%s\" would be inherited from more than once",
                            RelationGetRelationName(parent_rel))));
 
        if (inh->inhseqno > inhseqno)
            inhseqno = inh->inhseqno;
    }
    systable_endscan(scan);
 
    /* Match up the columns and bump attinhcount as needed */
    MergeAttributesIntoExisting(child_rel, parent_rel, ispartition);
 
    /* Match up the constraints and bump coninhcount as needed */
    MergeConstraintsIntoExisting(child_rel, parent_rel);
 
    /*
     * OK, it looks valid.  Make the catalog entries that show inheritance.
     */
    StoreCatalogInheritance1(RelationGetRelid(child_rel),
                             RelationGetRelid(parent_rel),
                             inhseqno + 1,
                             catalogRelation,
                             parent_rel->rd_rel->relkind ==
                             RELKIND_PARTITIONED_TABLE);
 
    /* Now we're done with pg_inherits */
    table_close(catalogRelation, RowExclusiveLock);
}

References BTEqualStrategyNumber, ereport, errcode(), errmsg, ERROR, fb(), Form_pg_inherits, GETSTRUCT(), HeapTupleIsValid, MergeAttributesIntoExisting(), MergeConstraintsIntoExisting(), ObjectIdGetDatum(), RelationGetRelationName, RelationGetRelid, RowExclusiveLock, ScanKeyInit(), StoreCatalogInheritance1(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by ATExecAddInherit(), and attachPartitionTable().

◆ createPartitionTable()

static Relation createPartitionTable	(	List **	wqueue,
		RangeVar *	newPartName,
		Relation	parent_rel,
		Oid	ownerId
	)

static

Definition at line 23211 of file tablecmds.c.

{
    Relation    newRel;
    Oid         newRelId;
    Oid         existingRelid;
    Oid         tablespaceId;
    TupleDesc   descriptor;
    List       *colList = NIL;
    Oid         relamId;
    Oid         namespaceId;
    AlteredTableInfo *new_partrel_tab;
    Form_pg_class parent_relform = parent_rel->rd_rel;
 
    /* If the existing rel is temp, it must belong to this session. */
    if (RELATION_IS_OTHER_TEMP(parent_rel))
        ereport(ERROR,
                errcode(ERRCODE_WRONG_OBJECT_TYPE),
                errmsg("cannot create as partition of temporary relation of another session"));
 
    /* Look up inheritance ancestors and generate the relation schema. */
    colList = getAttributesList(parent_rel);
 
    /* Create a tuple descriptor from the relation schema. */
    descriptor = BuildDescForRelation(colList);
 
    /* Look up the access method for the new relation. */
    relamId = (parent_relform->relam != InvalidOid) ? parent_relform->relam : HEAP_TABLE_AM_OID;
 
    /* Look up the namespace in which we are supposed to create the relation. */
    namespaceId =
        RangeVarGetAndCheckCreationNamespace(newPartName, NoLock, &existingRelid);
    if (OidIsValid(existingRelid))
        ereport(ERROR,
                errcode(ERRCODE_DUPLICATE_TABLE),
                errmsg("relation \"%s\" already exists", newPartName->relname));
 
    /*
     * We intended to create the partition with the same persistence as the
     * parent table, but we still need to recheck because that might be
     * affected by the search_path.  If the parent is permanent, so must be
     * all of its partitions.
     */
    if (parent_relform->relpersistence != RELPERSISTENCE_TEMP &&
        newPartName->relpersistence == RELPERSISTENCE_TEMP)
        ereport(ERROR,
                errcode(ERRCODE_WRONG_OBJECT_TYPE),
                errmsg("cannot create a temporary relation as partition of permanent relation \"%s\"",
                       RelationGetRelationName(parent_rel)));
 
    /* Permanent rels cannot be partitions belonging to a temporary parent. */
    if (newPartName->relpersistence != RELPERSISTENCE_TEMP &&
        parent_relform->relpersistence == RELPERSISTENCE_TEMP)
        ereport(ERROR,
                errcode(ERRCODE_WRONG_OBJECT_TYPE),
                errmsg("cannot create a permanent relation as partition of temporary relation \"%s\"",
                       RelationGetRelationName(parent_rel)));
 
    /*
     * Select the tablespace for the new partition.  Mirror the logic that
     * CREATE TABLE foo PARTITION OF ... uses in DefineRelation: take the
     * partitioned parent's explicit tablespace if it has one, otherwise take
     * default_tablespace into account, and finally use the database default.
     */
    tablespaceId = parent_relform->reltablespace;
    if (!OidIsValid(tablespaceId))
        tablespaceId = GetDefaultTablespace(newPartName->relpersistence, false);
 
    /* Check permissions except when using database's default */
    if (OidIsValid(tablespaceId) && tablespaceId != MyDatabaseTableSpace)
    {
        AclResult   aclresult;
 
        aclresult = object_aclcheck(TableSpaceRelationId, tablespaceId,
                                    GetUserId(), ACL_CREATE);
        if (aclresult != ACLCHECK_OK)
            aclcheck_error(aclresult, OBJECT_TABLESPACE,
                           get_tablespace_name(tablespaceId));
    }
 
    /* In all cases disallow placing user relations in pg_global */
    if (tablespaceId == GLOBALTABLESPACE_OID)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                 errmsg("only shared relations can be placed in pg_global tablespace")));
 
    /* Create the relation. */
    newRelId = heap_create_with_catalog(newPartName->relname,
                                        namespaceId,
                                        tablespaceId,
                                        InvalidOid,
                                        InvalidOid,
                                        InvalidOid,
                                        ownerId,
                                        relamId,
                                        descriptor,
                                        NIL,
                                        RELKIND_RELATION,
                                        newPartName->relpersistence,
                                        false,
                                        false,
                                        ONCOMMIT_NOOP,
                                        (Datum) 0,
                                        true,
                                        allowSystemTableMods,
                                        false,  /* is_internal */
                                        InvalidOid,
                                        NULL);
 
    /*
     * We must bump the command counter to make the newly-created relation
     * tuple visible for opening.
     */
    CommandCounterIncrement();
 
    /*
     * Create a TOAST table if the table needs one.  MERGE/SPLIT PARTITION
     * moves rows from existing partition(s) into new partition(s), which may
     * carry out-of-line varlena values that the new relation must be able to
     * store.  Also, the new partition must be able to receive out-of-line
     * varlena values after the DDL operation is complete.
     */
    NewRelationCreateToastTable(newRelId, (Datum) 0);
 
    /*
     * Open the new partition with no lock, because we already have an
     * AccessExclusiveLock placed there after creation.
     */
    newRel = table_open(newRelId, NoLock);
 
    /* Find or create a work queue entry for the newly created table. */
    new_partrel_tab = ATGetQueueEntry(wqueue, newRel);
 
    /* Create constraints, default values, and generated values. */
    createTableConstraints(wqueue, new_partrel_tab, parent_rel, newRel);
 
    /*
     * Need to call CommandCounterIncrement, so a fresh relcache entry has
     * newly installed constraint info.
     */
    CommandCounterIncrement();
 
    return newRel;
}

References ACL_CREATE, aclcheck_error(), ACLCHECK_OK, allowSystemTableMods, ATGetQueueEntry(), BuildDescForRelation(), CommandCounterIncrement(), createTableConstraints(), ereport, errcode(), errmsg, ERROR, fb(), get_tablespace_name(), getAttributesList(), GetDefaultTablespace(), GetUserId(), heap_create_with_catalog(), InvalidOid, MyDatabaseTableSpace, NewRelationCreateToastTable(), NIL, NoLock, object_aclcheck(), OBJECT_TABLESPACE, OidIsValid, ONCOMMIT_NOOP, RangeVarGetAndCheckCreationNamespace(), RELATION_IS_OTHER_TEMP, RelationGetRelationName, and table_open().

Referenced by ATExecMergePartitions(), and ATExecSplitPartition().

◆ createSplitPartitionContext()

static SplitPartitionContext * createSplitPartitionContext ( Relation partRel )

static

Definition at line 23973 of file tablecmds.c.

{
    SplitPartitionContext *pc;
 
    pc = palloc0_object(SplitPartitionContext);
    pc->partRel = partRel;
 
    /*
     * Prepare a BulkInsertState for table_tuple_insert. The FSM is empty, so
     * don't bother using it.
     */
    pc->bistate = GetBulkInsertState();
 
    /* Create a destination tuple slot for the new partition. */
    pc->dstslot = table_slot_create(pc->partRel, NULL);
 
    return pc;
}

References fb(), GetBulkInsertState(), palloc0_object, and table_slot_create().

Referenced by SplitPartitionMoveRows().

◆ createTableConstraints()

static void createTableConstraints	(	List **	wqueue,
		AlteredTableInfo *	tab,
		Relation	parent_rel,
		Relation	newRel
	)

static

Definition at line 23004 of file tablecmds.c.

{
    TupleDesc   tupleDesc;
    TupleConstr *constr;
    AttrMap    *attmap;
    AttrNumber  parent_attno;
    int         ccnum;
    List       *constraints = NIL;
    List       *cookedConstraints = NIL;
 
    tupleDesc = RelationGetDescr(parent_rel);
    constr = tupleDesc->constr;
 
    if (!constr)
        return;
 
    /*
     * Construct a map from the parent relation's attnos to the child rel's.
     * This re-checks type match, etc, although it shouldn't be possible to
     * have a failure since both tables are locked.
     */
    attmap = build_attrmap_by_name(RelationGetDescr(newRel),
                                   tupleDesc,
                                   false);
 
    /* Cycle for default values. */
    for (parent_attno = 1; parent_attno <= tupleDesc->natts; parent_attno++)
    {
        Form_pg_attribute attribute = TupleDescAttr(tupleDesc,
                                                    parent_attno - 1);
 
        /* Ignore dropped columns in the parent. */
        if (attribute->attisdropped)
            continue;
 
        /* Copy the default, if present, and it should be copied. */
        if (attribute->atthasdef)
        {
            Node       *this_default = NULL;
            bool        found_whole_row;
            AttrNumber  num;
            Node       *def;
            NewColumnValue *newval;
 
            if (attribute->attgenerated == ATTRIBUTE_GENERATED_VIRTUAL)
                this_default = build_generation_expression(parent_rel, attribute->attnum);
            else
            {
                this_default = TupleDescGetDefault(tupleDesc, attribute->attnum);
                if (this_default == NULL)
                    elog(ERROR, "default expression not found for attribute %d of relation \"%s\"",
                         attribute->attnum, RelationGetRelationName(parent_rel));
            }
 
            num = attmap->attnums[parent_attno - 1];
            def = map_variable_attnos(this_default, 1, 0, attmap, InvalidOid, &found_whole_row);
 
            if (found_whole_row && attribute->attgenerated != '\0')
                elog(ERROR, "cannot convert whole-row table reference");
 
            /* Add a pre-cooked default expression. */
            StoreAttrDefault(newRel, num, def, false);
 
            /*
             * Stored generated column expressions in parent_rel might
             * reference the tableoid.  newRel, parent_rel tableoid clear is
             * not the same. If so, these stored generated columns require
             * recomputation for newRel within MergePartitionsMoveRows.
             */
            if (attribute->attgenerated == ATTRIBUTE_GENERATED_STORED)
            {
                newval = palloc0_object(NewColumnValue);
                newval->attnum = num;
                newval->expr = expression_planner((Expr *) def);
                newval->is_generated = (attribute->attgenerated != '\0');
                tab->newvals = lappend(tab->newvals, newval);
            }
        }
    }
 
    /* Cycle for CHECK constraints. */
    for (ccnum = 0; ccnum < constr->num_check; ccnum++)
    {
        char       *ccname = constr->check[ccnum].ccname;
        char       *ccbin = constr->check[ccnum].ccbin;
        bool        ccenforced = constr->check[ccnum].ccenforced;
        bool        ccnoinherit = constr->check[ccnum].ccnoinherit;
        bool        ccvalid = constr->check[ccnum].ccvalid;
        Node       *ccbin_node;
        bool        found_whole_row;
        Constraint *con;
 
        /*
         * The partitioned table can not have a NO INHERIT check constraint
         * (see StoreRelCheck function for details).
         */
        Assert(!ccnoinherit);
 
        ccbin_node = map_variable_attnos(stringToNode(ccbin),
                                         1, 0,
                                         attmap,
                                         InvalidOid, &found_whole_row);
 
        /*
         * For the moment we have to reject whole-row variables (as for CREATE
         * TABLE LIKE and inheritances).
         */
        if (found_whole_row)
            elog(ERROR, "Constraint \"%s\" contains a whole-row reference to table \"%s\".",
                 ccname,
                 RelationGetRelationName(parent_rel));
 
        con = makeNode(Constraint);
        con->contype = CONSTR_CHECK;
        con->conname = pstrdup(ccname);
        con->deferrable = false;
        con->initdeferred = false;
        con->is_enforced = ccenforced;
        con->skip_validation = !ccvalid;
        con->initially_valid = ccvalid;
        con->is_no_inherit = ccnoinherit;
        con->raw_expr = NULL;
        con->cooked_expr = nodeToString(ccbin_node);
        con->location = -1;
        constraints = lappend(constraints, con);
    }
 
    /* Install all CHECK constraints. */
    cookedConstraints = AddRelationNewConstraints(newRel, NIL, constraints,
                                                  false, true, false, NULL);
 
    /* Make the additional catalog changes visible. */
    CommandCounterIncrement();
 
    /*
     * parent_rel check constraint expression may reference tableoid, so later
     * in MergePartitionsMoveRows, we need to evaluate the check constraint
     * again for the newRel. We can check whether the check constraint
     * contains a tableoid reference via pull_varattnos.
     */
    foreach_ptr(CookedConstraint, ccon, cookedConstraints)
    {
        if (!ccon->skip_validation)
        {
            Node       *qual;
            Bitmapset  *attnums = NULL;
 
            Assert(ccon->contype == CONSTR_CHECK);
            qual = expand_generated_columns_in_expr(ccon->expr, newRel, 1);
            pull_varattnos(qual, 1, &attnums);
 
            /*
             * Add a check only if it contains a tableoid
             * (TableOidAttributeNumber).
             */
            if (bms_is_member(TableOidAttributeNumber - FirstLowInvalidHeapAttributeNumber,
                              attnums))
            {
                NewConstraint *newcon;
 
                newcon = palloc0_object(NewConstraint);
                newcon->name = ccon->name;
                newcon->contype = CONSTR_CHECK;
                newcon->qual = qual;
 
                tab->constraints = lappend(tab->constraints, newcon);
            }
        }
    }
 
    /* Don't need the cookedConstraints anymore. */
    list_free_deep(cookedConstraints);
 
    /* Reproduce not-null constraints. */
    if (constr->has_not_null)
    {
        List       *nnconstraints;
 
        /*
         * The "include_noinh" argument is false because a partitioned table
         * can't have NO INHERIT constraint.
         */
        nnconstraints = RelationGetNotNullConstraints(RelationGetRelid(parent_rel),
                                                      false, false);
 
        Assert(list_length(nnconstraints) > 0);
 
        /*
         * We already set pg_attribute.attnotnull in createPartitionTable. No
         * need call set_attnotnull again.
         */
        AddRelationNewConstraints(newRel, NIL, nnconstraints, false, true, false, NULL);
    }
}

References AddRelationNewConstraints(), Assert, bms_is_member(), build_attrmap_by_name(), build_generation_expression(), ConstrCheck::ccbin, ConstrCheck::ccenforced, ConstrCheck::ccname, ConstrCheck::ccnoinherit, ConstrCheck::ccvalid, TupleConstr::check, CommandCounterIncrement(), Constraint::conname, CONSTR_CHECK, AlteredTableInfo::constraints, Constraint::contype, Constraint::cooked_expr, Constraint::deferrable, elog, ERROR, expand_generated_columns_in_expr(), expression_planner(), fb(), FirstLowInvalidHeapAttributeNumber, foreach_ptr, TupleConstr::has_not_null, Constraint::initdeferred, Constraint::initially_valid, InvalidOid, Constraint::is_enforced, Constraint::is_no_inherit, lappend(), list_free_deep(), list_length(), Constraint::location, makeNode, map_variable_attnos(), newval, AlteredTableInfo::newvals, NIL, nodeToString(), TupleConstr::num_check, palloc0_object, pstrdup(), pull_varattnos(), Constraint::raw_expr, RelationGetDescr, RelationGetNotNullConstraints(), RelationGetRelationName, RelationGetRelid, Constraint::skip_validation, StoreAttrDefault(), stringToNode(), TableOidAttributeNumber, TupleDescAttr(), and TupleDescGetDefault().

Referenced by createPartitionTable().

◆ decompile_conbin()

static char * decompile_conbin	(	HeapTuple	contup,
		TupleDesc	tupdesc
	)

static

Definition at line 18151 of file tablecmds.c.

{
    Form_pg_constraint con;
    bool        isnull;
    Datum       attr;
    Datum       expr;
 
    con = (Form_pg_constraint) GETSTRUCT(contup);
    attr = heap_getattr(contup, Anum_pg_constraint_conbin, tupdesc, &isnull);
    if (isnull)
        elog(ERROR, "null conbin for constraint %u", con->oid);
 
    expr = DirectFunctionCall2(pg_get_expr, attr,
                               ObjectIdGetDatum(con->conrelid));
    return TextDatumGetCString(expr);
}

References DirectFunctionCall2, elog, ERROR, fb(), Form_pg_constraint, GETSTRUCT(), heap_getattr(), ObjectIdGetDatum(), pg_get_expr(), and TextDatumGetCString.

Referenced by constraints_equivalent().

◆ DefineRelation()

ObjectAddress DefineRelation	(	CreateStmt *	stmt,
		char	relkind,
		Oid	ownerId,
		ObjectAddress *	typaddress,
		const char *	queryString
	)

Definition at line 818 of file tablecmds.c.

{
    char        relname[NAMEDATALEN];
    Oid         namespaceId;
    Oid         relationId;
    Oid         tablespaceId;
    Relation    rel;
    TupleDesc   descriptor;
    List       *inheritOids;
    List       *old_constraints;
    List       *old_notnulls;
    List       *rawDefaults;
    List       *cookedDefaults;
    List       *nncols;
    List       *connames = NIL;
    Datum       reloptions;
    ListCell   *listptr;
    AttrNumber  attnum;
    bool        partitioned;
    const char *const validnsps[] = HEAP_RELOPT_NAMESPACES;
    Oid         ofTypeId;
    ObjectAddress address;
    LOCKMODE    parentLockmode;
    Oid         accessMethodId = InvalidOid;
 
    /*
     * Truncate relname to appropriate length (probably a waste of time, as
     * parser should have done this already).
     */
    strlcpy(relname, stmt->relation->relname, NAMEDATALEN);
 
    /*
     * Check consistency of arguments
     */
    if (stmt->oncommit != ONCOMMIT_NOOP
        && stmt->relation->relpersistence != RELPERSISTENCE_TEMP)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("ON COMMIT can only be used on temporary tables")));
 
    if (stmt->partspec != NULL)
    {
        if (relkind != RELKIND_RELATION)
            elog(ERROR, "unexpected relkind: %d", (int) relkind);
 
        relkind = RELKIND_PARTITIONED_TABLE;
        partitioned = true;
    }
    else
        partitioned = false;
 
    if (relkind == RELKIND_PARTITIONED_TABLE &&
        stmt->relation->relpersistence == RELPERSISTENCE_UNLOGGED)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("partitioned tables cannot be unlogged")));
 
    /*
     * Look up the namespace in which we are supposed to create the relation,
     * check we have permission to create there, lock it against concurrent
     * drop, and mark stmt->relation as RELPERSISTENCE_TEMP if a temporary
     * namespace is selected.
     */
    namespaceId =
        RangeVarGetAndCheckCreationNamespace(stmt->relation, NoLock, NULL);
 
    /*
     * Security check: disallow creating temp tables from security-restricted
     * code.  This is needed because calling code might not expect untrusted
     * tables to appear in pg_temp at the front of its search path.
     */
    if (stmt->relation->relpersistence == RELPERSISTENCE_TEMP
        && InSecurityRestrictedOperation())
        ereport(ERROR,
                (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
                 errmsg("cannot create temporary table within security-restricted operation")));
 
    /*
     * Determine the lockmode to use when scanning parents.  A self-exclusive
     * lock is needed here.
     *
     * For regular inheritance, if two backends attempt to add children to the
     * same parent simultaneously, and that parent has no pre-existing
     * children, then both will attempt to update the parent's relhassubclass
     * field, leading to a "tuple concurrently updated" error.  Also, this
     * interlocks against a concurrent ANALYZE on the parent table, which
     * might otherwise be attempting to clear the parent's relhassubclass
     * field, if its previous children were recently dropped.
     *
     * If the child table is a partition, then we instead grab an exclusive
     * lock on the parent because its partition descriptor will be changed by
     * addition of the new partition.
     */
    parentLockmode = (stmt->partbound != NULL ? AccessExclusiveLock :
                      ShareUpdateExclusiveLock);
 
    /* Determine the list of OIDs of the parents. */
    inheritOids = NIL;
    foreach(listptr, stmt->inhRelations)
    {
        RangeVar   *rv = (RangeVar *) lfirst(listptr);
        Oid         parentOid;
 
        parentOid = RangeVarGetRelid(rv, parentLockmode, false);
 
        /*
         * Reject duplications in the list of parents.
         */
        if (list_member_oid(inheritOids, parentOid))
            ereport(ERROR,
                    (errcode(ERRCODE_DUPLICATE_TABLE),
                     errmsg("relation \"%s\" would be inherited from more than once",
                            get_rel_name(parentOid))));
 
        inheritOids = lappend_oid(inheritOids, parentOid);
    }
 
    /*
     * Select tablespace to use: an explicitly indicated one, or (in the case
     * of a partitioned table) the parent's, if it has one.
     */
    if (stmt->tablespacename)
    {
        tablespaceId = get_tablespace_oid(stmt->tablespacename, false);
 
        if (partitioned && tablespaceId == MyDatabaseTableSpace)
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("cannot specify default tablespace for partitioned relations")));
    }
    else if (stmt->partbound)
    {
        Assert(list_length(inheritOids) == 1);
        tablespaceId = get_rel_tablespace(linitial_oid(inheritOids));
    }
    else
        tablespaceId = InvalidOid;
 
    /* still nothing? use the default */
    if (!OidIsValid(tablespaceId))
        tablespaceId = GetDefaultTablespace(stmt->relation->relpersistence,
                                            partitioned);
 
    /* Check permissions except when using database's default */
    if (OidIsValid(tablespaceId) && tablespaceId != MyDatabaseTableSpace)
    {
        AclResult   aclresult;
 
        aclresult = object_aclcheck(TableSpaceRelationId, tablespaceId, GetUserId(),
                                    ACL_CREATE);
        if (aclresult != ACLCHECK_OK)
            aclcheck_error(aclresult, OBJECT_TABLESPACE,
                           get_tablespace_name(tablespaceId));
    }
 
    /* In all cases disallow placing user relations in pg_global */
    if (tablespaceId == GLOBALTABLESPACE_OID)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                 errmsg("only shared relations can be placed in pg_global tablespace")));
 
    /* Identify user ID that will own the table */
    if (!OidIsValid(ownerId))
        ownerId = GetUserId();
 
    /*
     * Parse and validate reloptions, if any.
     */
    reloptions = transformRelOptions((Datum) 0, stmt->options, NULL, validnsps,
                                     true, false);
 
    switch (relkind)
    {
        case RELKIND_VIEW:
            (void) view_reloptions(reloptions, true);
            break;
        case RELKIND_PARTITIONED_TABLE:
            (void) partitioned_table_reloptions(reloptions, true);
            break;
        default:
            (void) heap_reloptions(relkind, reloptions, true);
    }
 
    if (stmt->ofTypename)
    {
        AclResult   aclresult;
 
        ofTypeId = typenameTypeId(NULL, stmt->ofTypename);
 
        aclresult = object_aclcheck(TypeRelationId, ofTypeId, GetUserId(), ACL_USAGE);
        if (aclresult != ACLCHECK_OK)
            aclcheck_error_type(aclresult, ofTypeId);
    }
    else
        ofTypeId = InvalidOid;
 
    /*
     * Look up inheritance ancestors and generate relation schema, including
     * inherited attributes.  (Note that stmt->tableElts is destructively
     * modified by MergeAttributes.)
     */
    stmt->tableElts =
        MergeAttributes(stmt->tableElts, inheritOids,
                        stmt->relation->relpersistence,
                        stmt->partbound != NULL,
                        &old_constraints, &old_notnulls);
 
    /*
     * Create a tuple descriptor from the relation schema.  Note that this
     * deals with column names, types, and in-descriptor NOT NULL flags, but
     * not default values, NOT NULL or CHECK constraints; we handle those
     * below.
     */
    descriptor = BuildDescForRelation(stmt->tableElts);
 
    /*
     * Find columns with default values and prepare for insertion of the
     * defaults.  Pre-cooked (that is, inherited) defaults go into a list of
     * CookedConstraint structs that we'll pass to heap_create_with_catalog,
     * while raw defaults go into a list of RawColumnDefault structs that will
     * be processed by AddRelationNewConstraints.  (We can't deal with raw
     * expressions until we can do transformExpr.)
     */
    rawDefaults = NIL;
    cookedDefaults = NIL;
    attnum = 0;
 
    foreach(listptr, stmt->tableElts)
    {
        ColumnDef  *colDef = lfirst(listptr);
 
        attnum++;
        if (colDef->raw_default != NULL)
        {
            RawColumnDefault *rawEnt;
 
            Assert(colDef->cooked_default == NULL);
 
            rawEnt = palloc_object(RawColumnDefault);
            rawEnt->attnum = attnum;
            rawEnt->raw_default = colDef->raw_default;
            rawEnt->generated = colDef->generated;
            rawDefaults = lappend(rawDefaults, rawEnt);
        }
        else if (colDef->cooked_default != NULL)
        {
            CookedConstraint *cooked;
 
            cooked = palloc_object(CookedConstraint);
            cooked->contype = CONSTR_DEFAULT;
            cooked->conoid = InvalidOid;    /* until created */
            cooked->name = NULL;
            cooked->attnum = attnum;
            cooked->expr = colDef->cooked_default;
            cooked->is_enforced = true;
            cooked->skip_validation = false;
            cooked->is_local = true;    /* not used for defaults */
            cooked->inhcount = 0;   /* ditto */
            cooked->is_no_inherit = false;
            cookedDefaults = lappend(cookedDefaults, cooked);
        }
    }
 
    TupleDescFinalize(descriptor);
 
    /*
     * For relations with table AM and partitioned tables, select access
     * method to use: an explicitly indicated one, or (in the case of a
     * partitioned table) the parent's, if it has one.
     */
    if (stmt->accessMethod != NULL)
    {
        Assert(RELKIND_HAS_TABLE_AM(relkind) || relkind == RELKIND_PARTITIONED_TABLE);
        accessMethodId = get_table_am_oid(stmt->accessMethod, false);
    }
    else if (RELKIND_HAS_TABLE_AM(relkind) || relkind == RELKIND_PARTITIONED_TABLE)
    {
        if (stmt->partbound)
        {
            Assert(list_length(inheritOids) == 1);
            accessMethodId = get_rel_relam(linitial_oid(inheritOids));
        }
 
        if (RELKIND_HAS_TABLE_AM(relkind) && !OidIsValid(accessMethodId))
            accessMethodId = get_table_am_oid(default_table_access_method, false);
    }
 
    /*
     * Create the relation.  Inherited defaults and CHECK constraints are
     * passed in for immediate handling --- since they don't need parsing,
     * they can be stored immediately.
     */
    relationId = heap_create_with_catalog(relname,
                                          namespaceId,
                                          tablespaceId,
                                          InvalidOid,
                                          InvalidOid,
                                          ofTypeId,
                                          ownerId,
                                          accessMethodId,
                                          descriptor,
                                          list_concat(cookedDefaults,
                                                      old_constraints),
                                          relkind,
                                          stmt->relation->relpersistence,
                                          false,
                                          false,
                                          stmt->oncommit,
                                          reloptions,
                                          true,
                                          allowSystemTableMods,
                                          false,
                                          InvalidOid,
                                          typaddress);
 
    /*
     * We must bump the command counter to make the newly-created relation
     * tuple visible for opening.
     */
    CommandCounterIncrement();
 
    /*
     * Open the new relation and acquire exclusive lock on it.  This isn't
     * really necessary for locking out other backends (since they can't see
     * the new rel anyway until we commit), but it keeps the lock manager from
     * complaining about deadlock risks.
     */
    rel = relation_open(relationId, AccessExclusiveLock);
 
    /*
     * Now add any newly specified column default and generation expressions
     * to the new relation.  These are passed to us in the form of raw
     * parsetrees; we need to transform them to executable expression trees
     * before they can be added. The most convenient way to do that is to
     * apply the parser's transformExpr routine, but transformExpr doesn't
     * work unless we have a pre-existing relation. So, the transformation has
     * to be postponed to this final step of CREATE TABLE.
     *
     * This needs to be before processing the partitioning clauses because
     * those could refer to generated columns.
     */
    if (rawDefaults)
        AddRelationNewConstraints(rel, rawDefaults, NIL,
                                  true, true, false, queryString);
 
    /*
     * Make column generation expressions visible for use by partitioning.
     */
    CommandCounterIncrement();
 
    /* Process and store partition bound, if any. */
    if (stmt->partbound)
    {
        PartitionBoundSpec *bound;
        ParseState *pstate;
        Oid         parentId = linitial_oid(inheritOids),
                    defaultPartOid;
        Relation    parent,
                    defaultRel = NULL;
        ParseNamespaceItem *nsitem;
 
        /* Already have strong enough lock on the parent */
        parent = table_open(parentId, NoLock);
 
        /*
         * We are going to try to validate the partition bound specification
         * against the partition key of parentRel, so it better have one.
         */
        if (parent->rd_rel->relkind != RELKIND_PARTITIONED_TABLE)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                     errmsg("\"%s\" is not partitioned",
                            RelationGetRelationName(parent))));
 
        /*
         * The partition constraint of the default partition depends on the
         * partition bounds of every other partition. It is possible that
         * another backend might be about to execute a query on the default
         * partition table, and that the query relies on previously cached
         * default partition constraints. We must therefore take a table lock
         * strong enough to prevent all queries on the default partition from
         * proceeding until we commit and send out a shared-cache-inval notice
         * that will make them update their index lists.
         *
         * Order of locking: The relation being added won't be visible to
         * other backends until it is committed, hence here in
         * DefineRelation() the order of locking the default partition and the
         * relation being added does not matter. But at all other places we
         * need to lock the default relation before we lock the relation being
         * added or removed i.e. we should take the lock in same order at all
         * the places such that lock parent, lock default partition and then
         * lock the partition so as to avoid a deadlock.
         */
        defaultPartOid =
            get_default_oid_from_partdesc(RelationGetPartitionDesc(parent,
                                                                   true));
        if (OidIsValid(defaultPartOid))
            defaultRel = table_open(defaultPartOid, AccessExclusiveLock);
 
        /* Transform the bound values */
        pstate = make_parsestate(NULL);
        pstate->p_sourcetext = queryString;
 
        /*
         * Add an nsitem containing this relation, so that transformExpr
         * called on partition bound expressions is able to report errors
         * using a proper context.
         */
        nsitem = addRangeTableEntryForRelation(pstate, rel, AccessShareLock,
                                               NULL, false, false);
        addNSItemToQuery(pstate, nsitem, false, true, true);
 
        bound = transformPartitionBound(pstate, parent, stmt->partbound);
 
        /*
         * Check first that the new partition's bound is valid and does not
         * overlap with any of existing partitions of the parent.
         */
        check_new_partition_bound(relname, parent, bound, pstate);
 
        /*
         * If the default partition exists, its partition constraints will
         * change after the addition of this new partition such that it won't
         * allow any row that qualifies for this new partition. So, check that
         * the existing data in the default partition satisfies the constraint
         * as it will exist after adding this partition.
         */
        if (OidIsValid(defaultPartOid))
        {
            check_default_partition_contents(parent, defaultRel, bound);
            /* Keep the lock until commit. */
            table_close(defaultRel, NoLock);
        }
 
        /* Update the pg_class entry. */
        StorePartitionBound(rel, parent, bound);
 
        table_close(parent, NoLock);
    }
 
    /* Store inheritance information for new rel. */
    StoreCatalogInheritance(relationId, inheritOids, stmt->partbound != NULL);
 
    /*
     * Process the partitioning specification (if any) and store the partition
     * key information into the catalog.
     */
    if (partitioned)
    {
        ParseState *pstate;
        int         partnatts;
        AttrNumber  partattrs[PARTITION_MAX_KEYS];
        Oid         partopclass[PARTITION_MAX_KEYS];
        Oid         partcollation[PARTITION_MAX_KEYS];
        List       *partexprs = NIL;
 
        pstate = make_parsestate(NULL);
        pstate->p_sourcetext = queryString;
 
        partnatts = list_length(stmt->partspec->partParams);
 
        /* Protect fixed-size arrays here and in executor */
        if (partnatts > PARTITION_MAX_KEYS)
            ereport(ERROR,
                    (errcode(ERRCODE_TOO_MANY_COLUMNS),
                     errmsg("cannot partition using more than %d columns",
                            PARTITION_MAX_KEYS)));
 
        /*
         * We need to transform the raw parsetrees corresponding to partition
         * expressions into executable expression trees.  Like column defaults
         * and CHECK constraints, we could not have done the transformation
         * earlier.
         */
        stmt->partspec = transformPartitionSpec(rel, stmt->partspec);
 
        ComputePartitionAttrs(pstate, rel, stmt->partspec->partParams,
                              partattrs, &partexprs, partopclass,
                              partcollation, stmt->partspec->strategy);
 
        StorePartitionKey(rel, stmt->partspec->strategy, partnatts, partattrs,
                          partexprs,
                          partopclass, partcollation);
 
        /* make it all visible */
        CommandCounterIncrement();
    }
 
    /*
     * If we're creating a partition, create now all the indexes, triggers,
     * FKs defined in the parent.
     *
     * We can't do it earlier, because DefineIndex wants to know the partition
     * key which we just stored.
     */
    if (stmt->partbound)
    {
        Oid         parentId = linitial_oid(inheritOids);
        Relation    parent;
        List       *idxlist;
        ListCell   *cell;
 
        /* Already have strong enough lock on the parent */
        parent = table_open(parentId, NoLock);
        idxlist = RelationGetIndexList(parent);
 
        /*
         * For each index in the parent table, create one in the partition
         */
        foreach(cell, idxlist)
        {
            Relation    idxRel = index_open(lfirst_oid(cell), AccessShareLock);
            AttrMap    *attmap;
            IndexStmt  *idxstmt;
            Oid         constraintOid;
 
            if (rel->rd_rel->relkind == RELKIND_FOREIGN_TABLE)
            {
                if (idxRel->rd_index->indisunique)
                    ereport(ERROR,
                            (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                             errmsg("cannot create foreign partition of partitioned table \"%s\"",
                                    RelationGetRelationName(parent)),
                             errdetail("Table \"%s\" contains indexes that are unique.",
                                       RelationGetRelationName(parent))));
                else
                {
                    index_close(idxRel, AccessShareLock);
                    continue;
                }
            }
 
            attmap = build_attrmap_by_name(RelationGetDescr(rel),
                                           RelationGetDescr(parent),
                                           false);
            idxstmt =
                generateClonedIndexStmt(NULL, idxRel,
                                        attmap, &constraintOid);
            DefineIndex(NULL,
                        RelationGetRelid(rel),
                        idxstmt,
                        InvalidOid,
                        RelationGetRelid(idxRel),
                        constraintOid,
                        -1,
                        false, false, false, false, false);
 
            index_close(idxRel, AccessShareLock);
        }
 
        list_free(idxlist);
 
        /*
         * If there are any row-level triggers, clone them to the new
         * partition.
         */
        if (parent->trigdesc != NULL)
            CloneRowTriggersToPartition(parent, rel);
 
        /*
         * And foreign keys too.  Note that because we're freshly creating the
         * table, there is no need to verify these new constraints.
         */
        CloneForeignKeyConstraints(NULL, parent, rel);
 
        table_close(parent, NoLock);
    }
 
    /*
     * Now add any newly specified CHECK constraints to the new relation. Same
     * as for defaults above, but these need to come after partitioning is set
     * up.  We save the constraint names that were used, to avoid dupes below.
     */
    if (stmt->constraints)
    {
        List       *conlist;
 
        conlist = AddRelationNewConstraints(rel, NIL, stmt->constraints,
                                            true, true, false, queryString);
        foreach_ptr(CookedConstraint, cons, conlist)
        {
            if (cons->name != NULL)
                connames = lappend(connames, cons->name);
        }
    }
 
    /*
     * Finally, merge the not-null constraints that are declared directly with
     * those that come from parent relations (making sure to count inheritance
     * appropriately for each), create them, and set the attnotnull flag on
     * columns that don't yet have it.
     */
    nncols = AddRelationNotNullConstraints(rel, stmt->nnconstraints,
                                           old_notnulls, connames);
    foreach_int(attrnum, nncols)
        set_attnotnull(NULL, rel, attrnum, true, false);
 
    ObjectAddressSet(address, RelationRelationId, relationId);
 
    /*
     * Clean up.  We keep lock on new relation (although it shouldn't be
     * visible to anyone else anyway, until commit).
     */
    relation_close(rel, NoLock);
 
    return address;
}

References AccessExclusiveLock, AccessShareLock, ACL_CREATE, ACL_USAGE, aclcheck_error(), aclcheck_error_type(), ACLCHECK_OK, addNSItemToQuery(), addRangeTableEntryForRelation(), AddRelationNewConstraints(), AddRelationNotNullConstraints(), allowSystemTableMods, Assert, attnum, build_attrmap_by_name(), BuildDescForRelation(), check_default_partition_contents(), check_new_partition_bound(), CloneForeignKeyConstraints(), CloneRowTriggersToPartition(), CommandCounterIncrement(), ComputePartitionAttrs(), CONSTR_DEFAULT, default_table_access_method, DefineIndex(), elog, ereport, errcode(), errdetail(), errmsg, ERROR, fb(), foreach_int, foreach_ptr, generateClonedIndexStmt(), get_default_oid_from_partdesc(), get_rel_name(), get_rel_relam(), get_rel_tablespace(), get_table_am_oid(), get_tablespace_name(), get_tablespace_oid(), GetDefaultTablespace(), GetUserId(), heap_create_with_catalog(), HEAP_RELOPT_NAMESPACES, heap_reloptions(), index_close(), index_open(), InSecurityRestrictedOperation(), InvalidOid, lappend(), lappend_oid(), lfirst, lfirst_oid, linitial_oid, list_concat(), list_free(), list_length(), list_member_oid(), make_parsestate(), MergeAttributes(), MyDatabaseTableSpace, NAMEDATALEN, NIL, NoLock, object_aclcheck(), OBJECT_TABLESPACE, ObjectAddressSet, OidIsValid, ONCOMMIT_NOOP, ParseState::p_sourcetext, palloc_object, PARTITION_MAX_KEYS, partitioned_table_reloptions(), RangeVarGetAndCheckCreationNamespace(), RangeVarGetRelid, RelationData::rd_rel, relation_close(), relation_open(), RelationGetDescr, RelationGetIndexList(), RelationGetPartitionDesc(), RelationGetRelationName, RelationGetRelid, relname, set_attnotnull(), ShareUpdateExclusiveLock, stmt, StoreCatalogInheritance(), StorePartitionBound(), StorePartitionKey(), strlcpy(), table_close(), table_open(), transformPartitionBound(), transformPartitionSpec(), transformRelOptions(), RelationData::trigdesc, TupleDescFinalize(), typenameTypeId(), and view_reloptions().

Referenced by create_ctas_internal(), CreatePropGraph(), DefineCompositeType(), DefineSequence(), DefineVirtualRelation(), and ProcessUtilitySlow().

◆ deleteSplitPartitionContext()

static void deleteSplitPartitionContext	(	SplitPartitionContext *	pc,
		List **	wqueue,
		uint32	ti_options
	)

static

Definition at line 23996 of file tablecmds.c.

{
    ListCell   *ltab;
 
    ExecDropSingleTupleTableSlot(pc->dstslot);
    FreeBulkInsertState(pc->bistate);
 
    table_finish_bulk_insert(pc->partRel, ti_options);
 
    /*
     * We don't need to process this pc->partRel so delete the ALTER TABLE
     * queue of it.
     */
    foreach(ltab, *wqueue)
    {
        AlteredTableInfo *tab = (AlteredTableInfo *) lfirst(ltab);
 
        if (tab->relid == RelationGetRelid(pc->partRel))
        {
            *wqueue = list_delete_cell(*wqueue, ltab);
            break;
        }
    }
 
    pfree(pc);
}

References ExecDropSingleTupleTableSlot(), fb(), FreeBulkInsertState(), lfirst, list_delete_cell(), pfree(), RelationGetRelid, AlteredTableInfo::relid, and table_finish_bulk_insert().

Referenced by SplitPartitionMoveRows().

◆ DetachPartitionFinalize()

static void DetachPartitionFinalize	(	Relation	rel,
		Relation	partRel,
		bool	concurrent,
		Oid	defaultPartOid
	)

static

Definition at line 21872 of file tablecmds.c.

{
    Relation    classRel;
    List       *fks;
    ListCell   *cell;
    List       *indexes;
    Datum       new_val[Natts_pg_class];
    bool        new_null[Natts_pg_class],
                new_repl[Natts_pg_class];
    HeapTuple   tuple,
                newtuple;
    Relation    trigrel = NULL;
    List       *fkoids = NIL;
 
    if (concurrent)
    {
        /*
         * We can remove the pg_inherits row now. (In the non-concurrent case,
         * this was already done).
         */
        RemoveInheritance(partRel, rel, true);
    }
 
    /* Drop any triggers that were cloned on creation/attach. */
    DropClonedTriggersFromPartition(RelationGetRelid(partRel));
 
    /*
     * Detach any foreign keys that are inherited.  This includes creating
     * additional action triggers.
     */
    fks = copyObject(RelationGetFKeyList(partRel));
    if (fks != NIL)
        trigrel = table_open(TriggerRelationId, RowExclusiveLock);
 
    /*
     * It's possible that the partition being detached has a foreign key that
     * references a partitioned table.  When that happens, there are multiple
     * pg_constraint rows for the partition: one points to the partitioned
     * table itself, while the others point to each of its partitions.  Only
     * the topmost one is to be considered here; the child constraints must be
     * left alone, because conceptually those aren't coming from our parent
     * partitioned table, but from this partition itself.
     *
     * We implement this by collecting all the constraint OIDs in a first scan
     * of the FK array, and skipping in the loop below those constraints whose
     * parents are listed here.
     */
    foreach_node(ForeignKeyCacheInfo, fk, fks)
        fkoids = lappend_oid(fkoids, fk->conoid);
 
    foreach(cell, fks)
    {
        ForeignKeyCacheInfo *fk = lfirst(cell);
        HeapTuple   contup;
        Form_pg_constraint conform;
 
        contup = SearchSysCache1(CONSTROID, ObjectIdGetDatum(fk->conoid));
        if (!HeapTupleIsValid(contup))
            elog(ERROR, "cache lookup failed for constraint %u", fk->conoid);
        conform = (Form_pg_constraint) GETSTRUCT(contup);
 
        /*
         * Consider only inherited foreign keys, and only if their parents
         * aren't in the list.
         */
        if (conform->contype != CONSTRAINT_FOREIGN ||
            !OidIsValid(conform->conparentid) ||
            list_member_oid(fkoids, conform->conparentid))
        {
            ReleaseSysCache(contup);
            continue;
        }
 
        /*
         * The constraint on this table must be marked no longer a child of
         * the parent's constraint, as do its check triggers.
         */
        ConstraintSetParentConstraint(fk->conoid, InvalidOid, InvalidOid);
 
        /*
         * Also, look up the partition's "check" triggers corresponding to the
         * ENFORCED constraint being detached and detach them from the parent
         * triggers. NOT ENFORCED constraints do not have these triggers;
         * therefore, this step is not needed.
         */
        if (fk->conenforced)
        {
            Oid         insertTriggerOid,
                        updateTriggerOid;
 
            GetForeignKeyCheckTriggers(trigrel,
                                       fk->conoid, fk->confrelid, fk->conrelid,
                                       &insertTriggerOid, &updateTriggerOid);
            Assert(OidIsValid(insertTriggerOid));
            TriggerSetParentTrigger(trigrel, insertTriggerOid, InvalidOid,
                                    RelationGetRelid(partRel));
            Assert(OidIsValid(updateTriggerOid));
            TriggerSetParentTrigger(trigrel, updateTriggerOid, InvalidOid,
                                    RelationGetRelid(partRel));
        }
 
        /*
         * Lastly, create the action triggers on the referenced table, using
         * addFkRecurseReferenced, which requires some elaborate setup (so put
         * it in a separate block).  While at it, if the table is partitioned,
         * that function will recurse to create the pg_constraint rows and
         * action triggers for each partition.
         *
         * Note there's no need to do addFkConstraint() here, because the
         * pg_constraint row already exists.
         */
        {
            Constraint *fkconstraint;
            int         numfks;
            AttrNumber  conkey[INDEX_MAX_KEYS];
            AttrNumber  confkey[INDEX_MAX_KEYS];
            Oid         conpfeqop[INDEX_MAX_KEYS];
            Oid         conppeqop[INDEX_MAX_KEYS];
            Oid         conffeqop[INDEX_MAX_KEYS];
            int         numfkdelsetcols;
            AttrNumber  confdelsetcols[INDEX_MAX_KEYS];
            Relation    refdRel;
 
            DeconstructFkConstraintRow(contup,
                                       &numfks,
                                       conkey,
                                       confkey,
                                       conpfeqop,
                                       conppeqop,
                                       conffeqop,
                                       &numfkdelsetcols,
                                       confdelsetcols);
 
            /* Create a synthetic node we'll use throughout */
            fkconstraint = makeNode(Constraint);
            fkconstraint->contype = CONSTRAINT_FOREIGN;
            fkconstraint->conname = pstrdup(NameStr(conform->conname));
            fkconstraint->deferrable = conform->condeferrable;
            fkconstraint->initdeferred = conform->condeferred;
            fkconstraint->is_enforced = conform->conenforced;
            fkconstraint->skip_validation = true;
            fkconstraint->initially_valid = conform->convalidated;
            /* a few irrelevant fields omitted here */
            fkconstraint->pktable = NULL;
            fkconstraint->fk_attrs = NIL;
            fkconstraint->pk_attrs = NIL;
            fkconstraint->fk_matchtype = conform->confmatchtype;
            fkconstraint->fk_upd_action = conform->confupdtype;
            fkconstraint->fk_del_action = conform->confdeltype;
            fkconstraint->fk_del_set_cols = NIL;
            fkconstraint->old_conpfeqop = NIL;
            fkconstraint->old_pktable_oid = InvalidOid;
            fkconstraint->location = -1;
 
            /* set up colnames, used to generate the constraint name */
            for (int i = 0; i < numfks; i++)
            {
                Form_pg_attribute att;
 
                att = TupleDescAttr(RelationGetDescr(partRel),
                                    conkey[i] - 1);
 
                fkconstraint->fk_attrs = lappend(fkconstraint->fk_attrs,
                                                 makeString(NameStr(att->attname)));
            }
 
            refdRel = table_open(fk->confrelid, ShareRowExclusiveLock);
 
            addFkRecurseReferenced(fkconstraint, partRel,
                                   refdRel,
                                   conform->conindid,
                                   fk->conoid,
                                   numfks,
                                   confkey,
                                   conkey,
                                   conpfeqop,
                                   conppeqop,
                                   conffeqop,
                                   numfkdelsetcols,
                                   confdelsetcols,
                                   true,
                                   InvalidOid, InvalidOid,
                                   conform->conperiod);
            table_close(refdRel, NoLock);   /* keep lock till end of xact */
        }
 
        ReleaseSysCache(contup);
    }
    list_free_deep(fks);
    if (trigrel)
        table_close(trigrel, RowExclusiveLock);
 
    /*
     * Any sub-constraints that are in the referenced-side of a larger
     * constraint have to be removed.  This partition is no longer part of the
     * key space of the constraint.
     */
    foreach(cell, GetParentedForeignKeyRefs(partRel))
    {
        Oid         constrOid = lfirst_oid(cell);
        ObjectAddress constraint;
 
        ConstraintSetParentConstraint(constrOid, InvalidOid, InvalidOid);
        deleteDependencyRecordsForClass(ConstraintRelationId,
                                        constrOid,
                                        ConstraintRelationId,
                                        DEPENDENCY_INTERNAL);
        CommandCounterIncrement();
 
        ObjectAddressSet(constraint, ConstraintRelationId, constrOid);
        performDeletion(&constraint, DROP_RESTRICT, 0);
    }
 
    /* Now we can detach indexes */
    indexes = RelationGetIndexList(partRel);
    foreach(cell, indexes)
    {
        Oid         idxid = lfirst_oid(cell);
        Oid         parentidx;
        Relation    idx;
        Oid         constrOid;
        Oid         parentConstrOid;
 
        if (!has_superclass(idxid))
            continue;
 
        parentidx = get_partition_parent(idxid, false);
        Assert((IndexGetRelation(parentidx, false) == RelationGetRelid(rel)));
 
        idx = index_open(idxid, AccessExclusiveLock);
        IndexSetParentIndex(idx, InvalidOid);
 
        /*
         * If there's a constraint associated with the index, detach it too.
         * Careful: it is possible for a constraint index in a partition to be
         * the child of a non-constraint index, so verify whether the parent
         * index does actually have a constraint.
         */
        constrOid = get_relation_idx_constraint_oid(RelationGetRelid(partRel),
                                                    idxid);
        parentConstrOid = get_relation_idx_constraint_oid(RelationGetRelid(rel),
                                                          parentidx);
        if (OidIsValid(parentConstrOid) && OidIsValid(constrOid))
            ConstraintSetParentConstraint(constrOid, InvalidOid, InvalidOid);
 
        index_close(idx, NoLock);
    }
 
    /* Update pg_class tuple */
    classRel = table_open(RelationRelationId, RowExclusiveLock);
    tuple = SearchSysCacheCopy1(RELOID,
                                ObjectIdGetDatum(RelationGetRelid(partRel)));
    if (!HeapTupleIsValid(tuple))
        elog(ERROR, "cache lookup failed for relation %u",
             RelationGetRelid(partRel));
    Assert(((Form_pg_class) GETSTRUCT(tuple))->relispartition);
 
    /* Clear relpartbound and reset relispartition */
    memset(new_val, 0, sizeof(new_val));
    memset(new_null, false, sizeof(new_null));
    memset(new_repl, false, sizeof(new_repl));
    new_val[Anum_pg_class_relpartbound - 1] = (Datum) 0;
    new_null[Anum_pg_class_relpartbound - 1] = true;
    new_repl[Anum_pg_class_relpartbound - 1] = true;
    newtuple = heap_modify_tuple(tuple, RelationGetDescr(classRel),
                                 new_val, new_null, new_repl);
 
    ((Form_pg_class) GETSTRUCT(newtuple))->relispartition = false;
    CatalogTupleUpdate(classRel, &newtuple->t_self, newtuple);
    heap_freetuple(newtuple);
    table_close(classRel, RowExclusiveLock);
 
    /*
     * Drop identity property from all identity columns of partition.
     */
    for (int attno = 0; attno < RelationGetNumberOfAttributes(partRel); attno++)
    {
        Form_pg_attribute attr = TupleDescAttr(partRel->rd_att, attno);
 
        if (!attr->attisdropped && attr->attidentity)
            ATExecDropIdentity(partRel, NameStr(attr->attname), false,
                               AccessExclusiveLock, true, true);
    }
 
    if (OidIsValid(defaultPartOid))
    {
        /*
         * If the relation being detached is the default partition itself,
         * remove it from the parent's pg_partitioned_table entry.
         *
         * If not, we must invalidate default partition's relcache entry, as
         * in StorePartitionBound: its partition constraint depends on every
         * other partition's partition constraint.
         */
        if (RelationGetRelid(partRel) == defaultPartOid)
            update_default_partition_oid(RelationGetRelid(rel), InvalidOid);
        else
            CacheInvalidateRelcacheByRelid(defaultPartOid);
    }
 
    /*
     * Invalidate the parent's relcache so that the partition is no longer
     * included in its partition descriptor.
     */
    CacheInvalidateRelcache(rel);
 
    /*
     * If the partition we just detached is partitioned itself, invalidate
     * relcache for all descendent partitions too to ensure that their
     * rd_partcheck expression trees are rebuilt; must lock partitions before
     * doing so, using the same lockmode as what partRel has been locked with
     * by the caller.
     */
    if (partRel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
    {
        List       *children;
 
        children = find_all_inheritors(RelationGetRelid(partRel),
                                       AccessExclusiveLock, NULL);
        foreach(cell, children)
        {
            CacheInvalidateRelcacheByRelid(lfirst_oid(cell));
        }
    }
}

References AccessExclusiveLock, addFkRecurseReferenced(), Assert, ATExecDropIdentity(), CacheInvalidateRelcache(), CacheInvalidateRelcacheByRelid(), CatalogTupleUpdate(), CommandCounterIncrement(), ConstraintSetParentConstraint(), copyObject, DeconstructFkConstraintRow(), deleteDependencyRecordsForClass(), DEPENDENCY_INTERNAL, DROP_RESTRICT, DropClonedTriggersFromPartition(), elog, ERROR, fb(), find_all_inheritors(), foreach_node, Form_pg_constraint, get_partition_parent(), get_relation_idx_constraint_oid(), GetForeignKeyCheckTriggers(), GetParentedForeignKeyRefs(), GETSTRUCT(), has_superclass(), heap_freetuple(), heap_modify_tuple(), HeapTupleIsValid, i, idx(), index_close(), INDEX_MAX_KEYS, index_open(), IndexGetRelation(), IndexSetParentIndex(), InvalidOid, lappend(), lappend_oid(), lfirst, lfirst_oid, list_free_deep(), list_member_oid(), makeNode, makeString(), NameStr, NIL, NoLock, ObjectAddressSet, ObjectIdGetDatum(), OidIsValid, performDeletion(), pstrdup(), RelationData::rd_att, RelationData::rd_rel, RelationGetDescr, RelationGetFKeyList(), RelationGetIndexList(), RelationGetNumberOfAttributes, RelationGetRelid, ReleaseSysCache(), RemoveInheritance(), RowExclusiveLock, SearchSysCache1(), SearchSysCacheCopy1, ShareRowExclusiveLock, HeapTupleData::t_self, table_close(), table_open(), TriggerSetParentTrigger(), TupleDescAttr(), and update_default_partition_oid().

Referenced by ATExecDetachPartition(), ATExecDetachPartitionFinalize(), and detachPartitionTable().

◆ detachPartitionTable()

static void detachPartitionTable	(	Relation	parent_rel,
		Relation	child_rel,
		Oid	defaultPartOid
	)

static

Definition at line 23517 of file tablecmds.c.

{
    /* Remove the pg_inherits row first. */
    RemoveInheritance(child_rel, parent_rel, false);
 
    /*
     * Detaching the partition might involve TOAST table access, so ensure we
     * have a valid snapshot.
     */
    PushActiveSnapshot(GetTransactionSnapshot());
 
    /* Do the final part of detaching. */
    DetachPartitionFinalize(parent_rel, child_rel, false, defaultPartOid);
 
    PopActiveSnapshot();
}

References DetachPartitionFinalize(), fb(), GetTransactionSnapshot(), PopActiveSnapshot(), PushActiveSnapshot(), and RemoveInheritance().

Referenced by ATExecMergePartitions(), and ATExecSplitPartition().

◆ drop_parent_dependency()

static void drop_parent_dependency	(	Oid	relid,
		Oid	refclassid,
		Oid	refobjid,
		DependencyType	deptype
	)

static

Definition at line 18867 of file tablecmds.c.

{
    Relation    catalogRelation;
    SysScanDesc scan;
    ScanKeyData key[3];
    HeapTuple   depTuple;
 
    catalogRelation = table_open(DependRelationId, RowExclusiveLock);
 
    ScanKeyInit(&key[0],
                Anum_pg_depend_classid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationRelationId));
    ScanKeyInit(&key[1],
                Anum_pg_depend_objid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(relid));
    ScanKeyInit(&key[2],
                Anum_pg_depend_objsubid,
                BTEqualStrategyNumber, F_INT4EQ,
                Int32GetDatum(0));
 
    scan = systable_beginscan(catalogRelation, DependDependerIndexId, true,
                              NULL, 3, key);
 
    while (HeapTupleIsValid(depTuple = systable_getnext(scan)))
    {
        Form_pg_depend dep = (Form_pg_depend) GETSTRUCT(depTuple);
 
        if (dep->refclassid == refclassid &&
            dep->refobjid == refobjid &&
            dep->refobjsubid == 0 &&
            dep->deptype == deptype)
            CatalogTupleDelete(catalogRelation, &depTuple->t_self);
    }
 
    systable_endscan(scan);
    table_close(catalogRelation, RowExclusiveLock);
}

References BTEqualStrategyNumber, CatalogTupleDelete(), fb(), Form_pg_depend, GETSTRUCT(), HeapTupleIsValid, Int32GetDatum(), ObjectIdGetDatum(), RowExclusiveLock, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by ATExecAddOf(), ATExecDropOf(), and RemoveInheritance().

◆ DropClonedTriggersFromPartition()

static void DropClonedTriggersFromPartition ( Oid partitionId )

static

Definition at line 22240 of file tablecmds.c.

{
    ScanKeyData skey;
    SysScanDesc scan;
    HeapTuple   trigtup;
    Relation    tgrel;
    ObjectAddresses *objects;
 
    objects = new_object_addresses();
 
    /*
     * Scan pg_trigger to search for all triggers on this rel.
     */
    ScanKeyInit(&skey, Anum_pg_trigger_tgrelid, BTEqualStrategyNumber,
                F_OIDEQ, ObjectIdGetDatum(partitionId));
    tgrel = table_open(TriggerRelationId, RowExclusiveLock);
    scan = systable_beginscan(tgrel, TriggerRelidNameIndexId,
                              true, NULL, 1, &skey);
    while (HeapTupleIsValid(trigtup = systable_getnext(scan)))
    {
        Form_pg_trigger pg_trigger = (Form_pg_trigger) GETSTRUCT(trigtup);
        ObjectAddress trig;
 
        /* Ignore triggers that weren't cloned */
        if (!OidIsValid(pg_trigger->tgparentid))
            continue;
 
        /*
         * Ignore internal triggers that are implementation objects of foreign
         * keys, because these will be detached when the foreign keys
         * themselves are.
         */
        if (OidIsValid(pg_trigger->tgconstrrelid))
            continue;
 
        /*
         * This is ugly, but necessary: remove the dependency markings on the
         * trigger so that it can be removed.
         */
        deleteDependencyRecordsForClass(TriggerRelationId, pg_trigger->oid,
                                        TriggerRelationId,
                                        DEPENDENCY_PARTITION_PRI);
        deleteDependencyRecordsForClass(TriggerRelationId, pg_trigger->oid,
                                        RelationRelationId,
                                        DEPENDENCY_PARTITION_SEC);
 
        /* remember this trigger to remove it below */
        ObjectAddressSet(trig, TriggerRelationId, pg_trigger->oid);
        add_exact_object_address(&trig, objects);
    }
 
    /* make the dependency removal visible to the deletion below */
    CommandCounterIncrement();
    performMultipleDeletions(objects, DROP_RESTRICT, PERFORM_DELETION_INTERNAL);
 
    /* done */
    free_object_addresses(objects);
    systable_endscan(scan);
    table_close(tgrel, RowExclusiveLock);
}

References add_exact_object_address(), BTEqualStrategyNumber, CommandCounterIncrement(), deleteDependencyRecordsForClass(), DEPENDENCY_PARTITION_PRI, DEPENDENCY_PARTITION_SEC, DROP_RESTRICT, fb(), Form_pg_trigger, free_object_addresses(), GETSTRUCT(), HeapTupleIsValid, new_object_addresses(), ObjectAddressSet, ObjectIdGetDatum(), OidIsValid, PERFORM_DELETION_INTERNAL, performMultipleDeletions(), RowExclusiveLock, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by DetachPartitionFinalize().

◆ dropconstraint_internal()

static ObjectAddress dropconstraint_internal	(	Relation	rel,
		HeapTuple	constraintTup,
		DropBehavior	behavior,
		bool	recurse,
		bool	recursing,
		bool	missing_ok,
		LOCKMODE	lockmode
	)

static

Definition at line 14614 of file tablecmds.c.

{
    Relation    conrel;
    Form_pg_constraint con;
    ObjectAddress conobj;
    List       *children;
    bool        is_no_inherit_constraint = false;
    char       *constrName;
    char       *colname = NULL;
 
    /* Guard against stack overflow due to overly deep inheritance tree. */
    check_stack_depth();
 
    /* At top level, permission check was done in ATPrepCmd, else do it */
    if (recursing)
        ATSimplePermissions(AT_DropConstraint, rel,
                            ATT_TABLE | ATT_PARTITIONED_TABLE | ATT_FOREIGN_TABLE);
 
    conrel = table_open(ConstraintRelationId, RowExclusiveLock);
 
    con = (Form_pg_constraint) GETSTRUCT(constraintTup);
    constrName = NameStr(con->conname);
 
    /* Don't allow drop of inherited constraints */
    if (con->coninhcount > 0 && !recursing)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                 errmsg("cannot drop inherited constraint \"%s\" of relation \"%s\"",
                        constrName, RelationGetRelationName(rel))));
 
    /*
     * Reset pg_constraint.attnotnull, if this is a not-null constraint.
     *
     * While doing that, we're in a good position to disallow dropping a not-
     * null constraint underneath a primary key, a replica identity index, or
     * a generated identity column.
     */
    if (con->contype == CONSTRAINT_NOTNULL)
    {
        Relation    attrel = table_open(AttributeRelationId, RowExclusiveLock);
        AttrNumber  attnum = extractNotNullColumn(constraintTup);
        Bitmapset  *pkattrs;
        Bitmapset  *irattrs;
        HeapTuple   atttup;
        Form_pg_attribute attForm;
 
        /* save column name for recursion step */
        colname = get_attname(RelationGetRelid(rel), attnum, false);
 
        /*
         * Disallow if it's in the primary key.  For partitioned tables we
         * cannot rely solely on RelationGetIndexAttrBitmap, because it'll
         * return NULL if the primary key is invalid; but we still need to
         * protect not-null constraints under such a constraint, so check the
         * slow way.
         */
        pkattrs = RelationGetIndexAttrBitmap(rel, INDEX_ATTR_BITMAP_PRIMARY_KEY);
 
        if (pkattrs == NULL &&
            rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
        {
            Oid         pkindex = RelationGetPrimaryKeyIndex(rel, true);
 
            if (OidIsValid(pkindex))
            {
                Relation    pk = relation_open(pkindex, AccessShareLock);
 
                pkattrs = NULL;
                for (int i = 0; i < pk->rd_index->indnkeyatts; i++)
                    pkattrs = bms_add_member(pkattrs, pk->rd_index->indkey.values[i]);
 
                relation_close(pk, AccessShareLock);
            }
        }
 
        if (pkattrs &&
            bms_is_member(attnum - FirstLowInvalidHeapAttributeNumber, pkattrs))
            ereport(ERROR,
                    errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                    errmsg("column \"%s\" is in a primary key",
                           get_attname(RelationGetRelid(rel), attnum, false)));
 
        /* Disallow if it's in the replica identity */
        irattrs = RelationGetIndexAttrBitmap(rel, INDEX_ATTR_BITMAP_IDENTITY_KEY);
        if (bms_is_member(attnum - FirstLowInvalidHeapAttributeNumber, irattrs))
            ereport(ERROR,
                    errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                    errmsg("column \"%s\" is in index used as replica identity",
                           get_attname(RelationGetRelid(rel), attnum, false)));
 
        /* Disallow if it's a GENERATED AS IDENTITY column */
        atttup = SearchSysCacheCopyAttNum(RelationGetRelid(rel), attnum);
        if (!HeapTupleIsValid(atttup))
            elog(ERROR, "cache lookup failed for attribute %d of relation %u",
                 attnum, RelationGetRelid(rel));
        attForm = (Form_pg_attribute) GETSTRUCT(atttup);
        if (attForm->attidentity != '\0')
            ereport(ERROR,
                    errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                    errmsg("column \"%s\" of relation \"%s\" is an identity column",
                           get_attname(RelationGetRelid(rel), attnum,
                                       false),
                           RelationGetRelationName(rel)));
 
        /* All good -- reset attnotnull if needed */
        if (attForm->attnotnull)
        {
            attForm->attnotnull = false;
            CatalogTupleUpdate(attrel, &atttup->t_self, atttup);
        }
 
        table_close(attrel, RowExclusiveLock);
    }
 
    is_no_inherit_constraint = con->connoinherit;
 
    /*
     * If it's a foreign-key constraint, we'd better lock the referenced table
     * and check that that's not in use, just as we've already done for the
     * constrained table (else we might, eg, be dropping a trigger that has
     * unfired events).  But we can/must skip that in the self-referential
     * case.
     */
    if (con->contype == CONSTRAINT_FOREIGN &&
        con->confrelid != RelationGetRelid(rel))
    {
        Relation    frel;
 
        /* Must match lock taken by RemoveTriggerById: */
        frel = table_open(con->confrelid, AccessExclusiveLock);
        CheckAlterTableIsSafe(frel);
        table_close(frel, NoLock);
    }
 
    /*
     * Perform the actual constraint deletion
     */
    ObjectAddressSet(conobj, ConstraintRelationId, con->oid);
    performDeletion(&conobj, behavior, 0);
 
    /*
     * For partitioned tables, non-CHECK, non-NOT-NULL inherited constraints
     * are dropped via the dependency mechanism, so we're done here.
     */
    if (con->contype != CONSTRAINT_CHECK &&
        con->contype != CONSTRAINT_NOTNULL &&
        rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
    {
        table_close(conrel, RowExclusiveLock);
        return conobj;
    }
 
    /*
     * Propagate to children as appropriate.  Unlike most other ALTER
     * routines, we have to do this one level of recursion at a time; we can't
     * use find_all_inheritors to do it in one pass.
     */
    if (!is_no_inherit_constraint)
        children = find_inheritance_children(RelationGetRelid(rel), lockmode);
    else
        children = NIL;
 
    foreach_oid(childrelid, children)
    {
        Relation    childrel;
        HeapTuple   tuple;
        Form_pg_constraint childcon;
 
        /* find_inheritance_children already got lock */
        childrel = table_open(childrelid, NoLock);
        CheckAlterTableIsSafe(childrel);
 
        /*
         * We search for not-null constraints by column name, and others by
         * constraint name.
         */
        if (con->contype == CONSTRAINT_NOTNULL)
        {
            tuple = findNotNullConstraint(childrelid, colname);
            if (!HeapTupleIsValid(tuple))
                elog(ERROR, "cache lookup failed for not-null constraint on column \"%s\" of relation %u",
                     colname, RelationGetRelid(childrel));
        }
        else
        {
            SysScanDesc scan;
            ScanKeyData skey[3];
 
            ScanKeyInit(&skey[0],
                        Anum_pg_constraint_conrelid,
                        BTEqualStrategyNumber, F_OIDEQ,
                        ObjectIdGetDatum(childrelid));
            ScanKeyInit(&skey[1],
                        Anum_pg_constraint_contypid,
                        BTEqualStrategyNumber, F_OIDEQ,
                        ObjectIdGetDatum(InvalidOid));
            ScanKeyInit(&skey[2],
                        Anum_pg_constraint_conname,
                        BTEqualStrategyNumber, F_NAMEEQ,
                        CStringGetDatum(constrName));
            scan = systable_beginscan(conrel, ConstraintRelidTypidNameIndexId,
                                      true, NULL, 3, skey);
            /* There can only be one, so no need to loop */
            tuple = systable_getnext(scan);
            if (!HeapTupleIsValid(tuple))
                ereport(ERROR,
                        (errcode(ERRCODE_UNDEFINED_OBJECT),
                         errmsg("constraint \"%s\" of relation \"%s\" does not exist",
                                constrName,
                                RelationGetRelationName(childrel))));
            tuple = heap_copytuple(tuple);
            systable_endscan(scan);
        }
 
        childcon = (Form_pg_constraint) GETSTRUCT(tuple);
 
        /* Right now only CHECK and not-null constraints can be inherited */
        if (childcon->contype != CONSTRAINT_CHECK &&
            childcon->contype != CONSTRAINT_NOTNULL)
            elog(ERROR, "inherited constraint is not a CHECK or not-null constraint");
 
        if (childcon->coninhcount <= 0) /* shouldn't happen */
            elog(ERROR, "relation %u has non-inherited constraint \"%s\"",
                 childrelid, NameStr(childcon->conname));
 
        if (recurse)
        {
            /*
             * If the child constraint has other definition sources, just
             * decrement its inheritance count; if not, recurse to delete it.
             */
            if (childcon->coninhcount == 1 && !childcon->conislocal)
            {
                /* Time to delete this child constraint, too */
                dropconstraint_internal(childrel, tuple, behavior,
                                        recurse, true, missing_ok,
                                        lockmode);
            }
            else
            {
                /* Child constraint must survive my deletion */
                childcon->coninhcount--;
                CatalogTupleUpdate(conrel, &tuple->t_self, tuple);
 
                /* Make update visible */
                CommandCounterIncrement();
            }
        }
        else
        {
            /*
             * If we were told to drop ONLY in this table (no recursion) and
             * there are no further parents for this constraint, we need to
             * mark the inheritors' constraints as locally defined rather than
             * inherited.
             */
            childcon->coninhcount--;
            if (childcon->coninhcount == 0)
                childcon->conislocal = true;
 
            CatalogTupleUpdate(conrel, &tuple->t_self, tuple);
 
            /* Make update visible */
            CommandCounterIncrement();
        }
 
        heap_freetuple(tuple);
 
        table_close(childrel, NoLock);
    }
 
    table_close(conrel, RowExclusiveLock);
 
    return conobj;
}

References AccessExclusiveLock, AccessShareLock, AT_DropConstraint, ATSimplePermissions(), ATT_FOREIGN_TABLE, ATT_PARTITIONED_TABLE, ATT_TABLE, attnum, bms_add_member(), bms_is_member(), BTEqualStrategyNumber, CatalogTupleUpdate(), check_stack_depth(), CheckAlterTableIsSafe(), CommandCounterIncrement(), CStringGetDatum(), dropconstraint_internal(), elog, ereport, errcode(), errmsg, ERROR, extractNotNullColumn(), fb(), find_inheritance_children(), findNotNullConstraint(), FirstLowInvalidHeapAttributeNumber, foreach_oid, Form_pg_constraint, get_attname(), GETSTRUCT(), heap_copytuple(), heap_freetuple(), HeapTupleIsValid, i, INDEX_ATTR_BITMAP_IDENTITY_KEY, INDEX_ATTR_BITMAP_PRIMARY_KEY, InvalidOid, NameStr, NIL, NoLock, ObjectAddressSet, ObjectIdGetDatum(), OidIsValid, performDeletion(), RelationData::rd_rel, relation_close(), relation_open(), RelationGetIndexAttrBitmap(), RelationGetPrimaryKeyIndex(), RelationGetRelationName, RelationGetRelid, RowExclusiveLock, ScanKeyInit(), SearchSysCacheCopyAttNum(), systable_beginscan(), systable_endscan(), systable_getnext(), HeapTupleData::t_self, table_close(), and table_open().

Referenced by ATExecDropConstraint(), ATExecDropNotNull(), and dropconstraint_internal().

◆ DropErrorMsgNonExistent()

static void DropErrorMsgNonExistent	(	RangeVar *	rel,
		char	rightkind,
		bool	missing_ok
	)

static

Definition at line 1522 of file tablecmds.c.

{
    const struct dropmsgstrings *rentry;
 
    if (rel->schemaname != NULL &&
        !OidIsValid(LookupNamespaceNoError(rel->schemaname)))
    {
        if (!missing_ok)
        {
            ereport(ERROR,
                    (errcode(ERRCODE_UNDEFINED_SCHEMA),
                     errmsg("schema \"%s\" does not exist", rel->schemaname)));
        }
        else
        {
            ereport(NOTICE,
                    (errmsg("schema \"%s\" does not exist, skipping",
                            rel->schemaname)));
        }
        return;
    }
 
    for (rentry = dropmsgstringarray; rentry->kind != '\0'; rentry++)
    {
        if (rentry->kind == rightkind)
        {
            if (!missing_ok)
            {
                ereport(ERROR,
                        (errcode(rentry->nonexistent_code),
                         errmsg(rentry->nonexistent_msg, rel->relname)));
            }
            else
            {
                ereport(NOTICE, (errmsg(rentry->skipping_msg, rel->relname)));
                break;
            }
        }
    }
 
    Assert(rentry->kind != '\0');   /* Should be impossible */
}

References Assert, dropmsgstringarray, ereport, errcode(), errmsg, ERROR, fb(), dropmsgstrings::kind, LookupNamespaceNoError(), NOTICE, OidIsValid, RangeVar::relname, and RangeVar::schemaname.

Referenced by RemoveRelations().

◆ DropErrorMsgWrongType()

static void DropErrorMsgWrongType	(	const char *	relname,
		char	wrongkind,
		char	rightkind
	)

static

Definition at line 1570 of file tablecmds.c.

{
    const struct dropmsgstrings *rentry;
    const struct dropmsgstrings *wentry;
 
    for (rentry = dropmsgstringarray; rentry->kind != '\0'; rentry++)
        if (rentry->kind == rightkind)
            break;
    Assert(rentry->kind != '\0');
 
    for (wentry = dropmsgstringarray; wentry->kind != '\0'; wentry++)
        if (wentry->kind == wrongkind)
            break;
    /* wrongkind could be something we don't have in our table... */
 
    ereport(ERROR,
            (errcode(ERRCODE_WRONG_OBJECT_TYPE),
             errmsg(rentry->nota_msg, relname),
             (wentry->kind != '\0') ? errhint("%s", _(wentry->drophint_msg)) : 0));
}

References _, Assert, dropmsgstringarray, ereport, errcode(), errhint(), errmsg, ERROR, fb(), dropmsgstrings::kind, and relname.

Referenced by RangeVarCallbackForDropRelation().

◆ DropForeignKeyConstraintTriggers()

static void DropForeignKeyConstraintTriggers	(	Relation	trigrel,
		Oid	conoid,
		Oid	confrelid,
		Oid	conrelid
	)

static

Definition at line 12151 of file tablecmds.c.

{
    ScanKeyData key;
    SysScanDesc scan;
    HeapTuple   trigtup;
 
    ScanKeyInit(&key,
                Anum_pg_trigger_tgconstraint,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(conoid));
    scan = systable_beginscan(trigrel, TriggerConstraintIndexId, true,
                              NULL, 1, &key);
    while ((trigtup = systable_getnext(scan)) != NULL)
    {
        Form_pg_trigger trgform = (Form_pg_trigger) GETSTRUCT(trigtup);
        ObjectAddress trigger;
 
        /* Invalid if trigger is not for a referential integrity constraint */
        if (!OidIsValid(trgform->tgconstrrelid))
            continue;
        if (OidIsValid(conrelid) && trgform->tgconstrrelid != conrelid)
            continue;
        if (OidIsValid(confrelid) && trgform->tgrelid != confrelid)
            continue;
 
        /* We should be dropping trigger related to foreign key constraint */
        Assert(trgform->tgfoid == F_RI_FKEY_CHECK_INS ||
               trgform->tgfoid == F_RI_FKEY_CHECK_UPD ||
               trgform->tgfoid == F_RI_FKEY_CASCADE_DEL ||
               trgform->tgfoid == F_RI_FKEY_CASCADE_UPD ||
               trgform->tgfoid == F_RI_FKEY_RESTRICT_DEL ||
               trgform->tgfoid == F_RI_FKEY_RESTRICT_UPD ||
               trgform->tgfoid == F_RI_FKEY_SETNULL_DEL ||
               trgform->tgfoid == F_RI_FKEY_SETNULL_UPD ||
               trgform->tgfoid == F_RI_FKEY_SETDEFAULT_DEL ||
               trgform->tgfoid == F_RI_FKEY_SETDEFAULT_UPD ||
               trgform->tgfoid == F_RI_FKEY_NOACTION_DEL ||
               trgform->tgfoid == F_RI_FKEY_NOACTION_UPD);
 
        /*
         * The constraint is originally set up to contain this trigger as an
         * implementation object, so there's a dependency record that links
         * the two; however, since the trigger is no longer needed, we remove
         * the dependency link in order to be able to drop the trigger while
         * keeping the constraint intact.
         */
        deleteDependencyRecordsFor(TriggerRelationId,
                                   trgform->oid,
                                   false);
        /* make dependency deletion visible to performDeletion */
        CommandCounterIncrement();
        ObjectAddressSet(trigger, TriggerRelationId,
                         trgform->oid);
        performDeletion(&trigger, DROP_RESTRICT, 0);
        /* make trigger drop visible, in case the loop iterates */
        CommandCounterIncrement();
    }
 
    systable_endscan(scan);
}

References Assert, BTEqualStrategyNumber, CommandCounterIncrement(), deleteDependencyRecordsFor(), DROP_RESTRICT, fb(), Form_pg_trigger, GETSTRUCT(), ObjectAddressSet, ObjectIdGetDatum(), OidIsValid, performDeletion(), ScanKeyInit(), systable_beginscan(), systable_endscan(), and systable_getnext().

Referenced by ATExecAlterFKConstrEnforceability(), and AttachPartitionForeignKey().

◆ equal_oid_lists()

static bool equal_oid_lists	(	const List *	a,
		const List *	b
	)

static

Definition at line 23539 of file tablecmds.c.

{
    ListCell   *la,
               *lb;
 
    if (list_length(a) != list_length(b))
        return false;
 
    forboth(la, a, lb, b)
    {
        if (lfirst_oid(la) != lfirst_oid(lb))
            return false;
    }
    return true;
}

References a, b, fb(), forboth, lfirst_oid, and list_length().

Referenced by collectPartitionIndexExtDeps().

◆ evaluateGeneratedExpressionsAndCheckConstraints()

static void evaluateGeneratedExpressionsAndCheckConstraints	(	AlteredTableInfo *	tab,
		Relation	newPartRel,
		TupleTableSlot *	insertslot,
		ExprContext *	econtext
	)

static

Definition at line 22906 of file tablecmds.c.

{
    econtext->ecxt_scantuple = insertslot;
 
    foreach_ptr(NewColumnValue, ex, tab->newvals)
    {
        if (!ex->is_generated)
            continue;
 
        insertslot->tts_values[ex->attnum - 1]
            = ExecEvalExpr(ex->exprstate,
                           econtext,
                           &insertslot->tts_isnull[ex->attnum - 1]);
    }
 
    foreach_ptr(NewConstraint, con, tab->constraints)
    {
        switch (con->contype)
        {
            case CONSTR_CHECK:
                if (!ExecCheck(con->qualstate, econtext))
                    ereport(ERROR,
                            errcode(ERRCODE_CHECK_VIOLATION),
                            errmsg("check constraint \"%s\" of relation \"%s\" is violated by some row",
                                   con->name, RelationGetRelationName(newPartRel)),
                            errtableconstraint(newPartRel, con->name));
                break;
            case CONSTR_NOTNULL:
            case CONSTR_FOREIGN:
                /* Nothing to do here */
                break;
            default:
                elog(ERROR, "unrecognized constraint type: %d",
                     (int) con->contype);
        }
    }
}

References CONSTR_CHECK, CONSTR_FOREIGN, CONSTR_NOTNULL, AlteredTableInfo::constraints, ExprContext::ecxt_scantuple, elog, ereport, errcode(), errmsg, ERROR, errtableconstraint(), ExecCheck(), ExecEvalExpr(), fb(), foreach_ptr, AlteredTableInfo::newvals, RelationGetRelationName, and TupleTableSlot::tts_values.

Referenced by MergePartitionsMoveRows(), and SplitPartitionMoveRows().

◆ ExecuteTruncate()

void ExecuteTruncate ( TruncateStmt * stmt )

Definition at line 1924 of file tablecmds.c.

{
    List       *rels = NIL;
    List       *relids = NIL;
    List       *relids_logged = NIL;
    ListCell   *cell;
 
    /*
     * Open, exclusive-lock, and check all the explicitly-specified relations
     */
    foreach(cell, stmt->relations)
    {
        RangeVar   *rv = lfirst(cell);
        Relation    rel;
        bool        recurse = rv->inh;
        Oid         myrelid;
        LOCKMODE    lockmode = AccessExclusiveLock;
 
        myrelid = RangeVarGetRelidExtended(rv, lockmode,
                                           0, RangeVarCallbackForTruncate,
                                           NULL);
 
        /* don't throw error for "TRUNCATE foo, foo" */
        if (list_member_oid(relids, myrelid))
            continue;
 
        /* open the relation, we already hold a lock on it */
        rel = table_open(myrelid, NoLock);
 
        /*
         * RangeVarGetRelidExtended() has done most checks with its callback,
         * but other checks with the now-opened Relation remain.
         */
        truncate_check_activity(rel);
 
        rels = lappend(rels, rel);
        relids = lappend_oid(relids, myrelid);
 
        /* Log this relation only if needed for logical decoding */
        if (RelationIsLogicallyLogged(rel))
            relids_logged = lappend_oid(relids_logged, myrelid);
 
        if (recurse)
        {
            ListCell   *child;
            List       *children;
 
            children = find_all_inheritors(myrelid, lockmode, NULL);
 
            foreach(child, children)
            {
                Oid         childrelid = lfirst_oid(child);
 
                if (list_member_oid(relids, childrelid))
                    continue;
 
                /* find_all_inheritors already got lock */
                rel = table_open(childrelid, NoLock);
 
                /*
                 * It is possible that the parent table has children that are
                 * temp tables of other backends.  We cannot safely access
                 * such tables (because of buffering issues), and the best
                 * thing to do is to silently ignore them.  Note that this
                 * check is the same as one of the checks done in
                 * truncate_check_activity() called below, still it is kept
                 * here for simplicity.
                 */
                if (RELATION_IS_OTHER_TEMP(rel))
                {
                    table_close(rel, lockmode);
                    continue;
                }
 
                /*
                 * Inherited TRUNCATE commands perform access permission
                 * checks on the parent table only. So we skip checking the
                 * children's permissions and don't call
                 * truncate_check_perms() here.
                 */
                truncate_check_rel(RelationGetRelid(rel), rel->rd_rel);
                truncate_check_activity(rel);
 
                rels = lappend(rels, rel);
                relids = lappend_oid(relids, childrelid);
 
                /* Log this relation only if needed for logical decoding */
                if (RelationIsLogicallyLogged(rel))
                    relids_logged = lappend_oid(relids_logged, childrelid);
            }
        }
        else if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("cannot truncate only a partitioned table"),
                     errhint("Do not specify the ONLY keyword, or use TRUNCATE ONLY on the partitions directly.")));
    }
 
    ExecuteTruncateGuts(rels, relids, relids_logged,
                        stmt->behavior, stmt->restart_seqs, false);
 
    /* And close the rels */
    foreach(cell, rels)
    {
        Relation    rel = (Relation) lfirst(cell);
 
        table_close(rel, NoLock);
    }
}

References AccessExclusiveLock, ereport, errcode(), errhint(), errmsg, ERROR, ExecuteTruncateGuts(), fb(), find_all_inheritors(), RangeVar::inh, lappend(), lappend_oid(), lfirst, lfirst_oid, list_member_oid(), NIL, NoLock, RangeVarCallbackForTruncate(), RangeVarGetRelidExtended(), RelationData::rd_rel, RELATION_IS_OTHER_TEMP, RelationGetRelid, RelationIsLogicallyLogged, stmt, table_close(), table_open(), truncate_check_activity(), and truncate_check_rel().

Referenced by standard_ProcessUtility().

◆ ExecuteTruncateGuts()

void ExecuteTruncateGuts	(	List *	explicit_rels,
		List *	relids,
		List *	relids_logged,
		DropBehavior	behavior,
		bool	restart_seqs,
		bool	run_as_table_owner
	)

Definition at line 2048 of file tablecmds.c.

{
    List       *rels;
    List       *seq_relids = NIL;
    HTAB       *ft_htab = NULL;
    EState     *estate;
    ResultRelInfo *resultRelInfos;
    ResultRelInfo *resultRelInfo;
    SubTransactionId mySubid;
    ListCell   *cell;
    Oid        *logrelids;
 
    /*
     * Check the explicitly-specified relations.
     *
     * In CASCADE mode, suck in all referencing relations as well.  This
     * requires multiple iterations to find indirectly-dependent relations. At
     * each phase, we need to exclusive-lock new rels before looking for their
     * dependencies, else we might miss something.  Also, we check each rel as
     * soon as we open it, to avoid a faux pas such as holding lock for a long
     * time on a rel we have no permissions for.
     */
    rels = list_copy(explicit_rels);
    if (behavior == DROP_CASCADE)
    {
        for (;;)
        {
            List       *newrelids;
 
            newrelids = heap_truncate_find_FKs(relids);
            if (newrelids == NIL)
                break;          /* nothing else to add */
 
            foreach(cell, newrelids)
            {
                Oid         relid = lfirst_oid(cell);
                Relation    rel;
 
                rel = table_open(relid, AccessExclusiveLock);
                ereport(NOTICE,
                        (errmsg("truncate cascades to table \"%s\"",
                                RelationGetRelationName(rel))));
                truncate_check_rel(relid, rel->rd_rel);
                truncate_check_perms(relid, rel->rd_rel);
                truncate_check_activity(rel);
                rels = lappend(rels, rel);
                relids = lappend_oid(relids, relid);
 
                /* Log this relation only if needed for logical decoding */
                if (RelationIsLogicallyLogged(rel))
                    relids_logged = lappend_oid(relids_logged, relid);
            }
        }
    }
 
    /*
     * Check foreign key references.  In CASCADE mode, this should be
     * unnecessary since we just pulled in all the references; but as a
     * cross-check, do it anyway if in an Assert-enabled build.
     */
#ifdef USE_ASSERT_CHECKING
    heap_truncate_check_FKs(rels, false);
#else
    if (behavior == DROP_RESTRICT)
        heap_truncate_check_FKs(rels, false);
#endif
 
    /*
     * If we are asked to restart sequences, find all the sequences, lock them
     * (we need AccessExclusiveLock for ResetSequence), and check permissions.
     * We want to do this early since it's pointless to do all the truncation
     * work only to fail on sequence permissions.
     */
    if (restart_seqs)
    {
        foreach(cell, rels)
        {
            Relation    rel = (Relation) lfirst(cell);
            List       *seqlist = getOwnedSequences(RelationGetRelid(rel));
            ListCell   *seqcell;
 
            foreach(seqcell, seqlist)
            {
                Oid         seq_relid = lfirst_oid(seqcell);
                Relation    seq_rel;
 
                seq_rel = relation_open(seq_relid, AccessExclusiveLock);
 
                /* This check must match AlterSequence! */
                if (!object_ownercheck(RelationRelationId, seq_relid, GetUserId()))
                    aclcheck_error(ACLCHECK_NOT_OWNER, OBJECT_SEQUENCE,
                                   RelationGetRelationName(seq_rel));
 
                seq_relids = lappend_oid(seq_relids, seq_relid);
 
                relation_close(seq_rel, NoLock);
            }
        }
    }
 
    /* Prepare to catch AFTER triggers. */
    AfterTriggerBeginQuery();
 
    /*
     * To fire triggers, we'll need an EState as well as a ResultRelInfo for
     * each relation.  We don't need to call ExecOpenIndices, though.
     *
     * We put the ResultRelInfos in the es_opened_result_relations list, even
     * though we don't have a range table and don't populate the
     * es_result_relations array.  That's a bit bogus, but it's enough to make
     * ExecGetTriggerResultRel() find them.
     */
    estate = CreateExecutorState();
    resultRelInfos = (ResultRelInfo *)
        palloc(list_length(rels) * sizeof(ResultRelInfo));
    resultRelInfo = resultRelInfos;
    foreach(cell, rels)
    {
        Relation    rel = (Relation) lfirst(cell);
 
        InitResultRelInfo(resultRelInfo,
                          rel,
                          0,    /* dummy rangetable index */
                          NULL,
                          0);
        estate->es_opened_result_relations =
            lappend(estate->es_opened_result_relations, resultRelInfo);
        resultRelInfo++;
    }
 
    /*
     * Process all BEFORE STATEMENT TRUNCATE triggers before we begin
     * truncating (this is because one of them might throw an error). Also, if
     * we were to allow them to prevent statement execution, that would need
     * to be handled here.
     */
    resultRelInfo = resultRelInfos;
    foreach(cell, rels)
    {
        UserContext ucxt;
 
        if (run_as_table_owner)
            SwitchToUntrustedUser(resultRelInfo->ri_RelationDesc->rd_rel->relowner,
                                  &ucxt);
        ExecBSTruncateTriggers(estate, resultRelInfo);
        if (run_as_table_owner)
            RestoreUserContext(&ucxt);
        resultRelInfo++;
    }
 
    /*
     * OK, truncate each table.
     */
    mySubid = GetCurrentSubTransactionId();
 
    foreach(cell, rels)
    {
        Relation    rel = (Relation) lfirst(cell);
 
        /* Skip partitioned tables as there is nothing to do */
        if (rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
            continue;
 
        /*
         * Build the lists of foreign tables belonging to each foreign server
         * and pass each list to the foreign data wrapper's callback function,
         * so that each server can truncate its all foreign tables in bulk.
         * Each list is saved as a single entry in a hash table that uses the
         * server OID as lookup key.
         */
        if (rel->rd_rel->relkind == RELKIND_FOREIGN_TABLE)
        {
            Oid         serverid = GetForeignServerIdByRelId(RelationGetRelid(rel));
            bool        found;
            ForeignTruncateInfo *ft_info;
 
            /* First time through, initialize hashtable for foreign tables */
            if (!ft_htab)
            {
                HASHCTL     hctl;
 
                memset(&hctl, 0, sizeof(HASHCTL));
                hctl.keysize = sizeof(Oid);
                hctl.entrysize = sizeof(ForeignTruncateInfo);
                hctl.hcxt = CurrentMemoryContext;
 
                ft_htab = hash_create("TRUNCATE for Foreign Tables",
                                      32,   /* start small and extend */
                                      &hctl,
                                      HASH_ELEM | HASH_BLOBS | HASH_CONTEXT);
            }
 
            /* Find or create cached entry for the foreign table */
            ft_info = hash_search(ft_htab, &serverid, HASH_ENTER, &found);
            if (!found)
                ft_info->rels = NIL;
 
            /*
             * Save the foreign table in the entry of the server that the
             * foreign table belongs to.
             */
            ft_info->rels = lappend(ft_info->rels, rel);
            continue;
        }
 
        /*
         * Normally, we need a transaction-safe truncation here.  However, if
         * the table was either created in the current (sub)transaction or has
         * a new relfilenumber in the current (sub)transaction, then we can
         * just truncate it in-place, because a rollback would cause the whole
         * table or the current physical file to be thrown away anyway.
         */
        if (rel->rd_createSubid == mySubid ||
            rel->rd_newRelfilelocatorSubid == mySubid)
        {
            /* Immediate, non-rollbackable truncation is OK */
            heap_truncate_one_rel(rel);
        }
        else
        {
            Oid         heap_relid;
            Oid         toast_relid;
            ReindexParams reindex_params = {0};
 
            /*
             * This effectively deletes all rows in the table, and may be done
             * in a serializable transaction.  In that case we must record a
             * rw-conflict in to this transaction from each transaction
             * holding a predicate lock on the table.
             */
            CheckTableForSerializableConflictIn(rel);
 
            /*
             * Need the full transaction-safe pushups.
             *
             * Create a new empty storage file for the relation, and assign it
             * as the relfilenumber value. The old storage file is scheduled
             * for deletion at commit.
             */
            RelationSetNewRelfilenumber(rel, rel->rd_rel->relpersistence);
 
            heap_relid = RelationGetRelid(rel);
 
            /*
             * The same for the toast table, if any.
             */
            toast_relid = rel->rd_rel->reltoastrelid;
            if (OidIsValid(toast_relid))
            {
                Relation    toastrel = relation_open(toast_relid,
                                                     AccessExclusiveLock);
 
                RelationSetNewRelfilenumber(toastrel,
                                            toastrel->rd_rel->relpersistence);
                table_close(toastrel, NoLock);
            }
 
            /*
             * Reconstruct the indexes to match, and we're done.
             */
            reindex_relation(NULL, heap_relid, REINDEX_REL_PROCESS_TOAST,
                             &reindex_params);
        }
 
        pgstat_count_truncate(rel);
    }
 
    /* Now go through the hash table, and truncate foreign tables */
    if (ft_htab)
    {
        ForeignTruncateInfo *ft_info;
        HASH_SEQ_STATUS seq;
 
        hash_seq_init(&seq, ft_htab);
 
        PG_TRY();
        {
            while ((ft_info = hash_seq_search(&seq)) != NULL)
            {
                FdwRoutine *routine = GetFdwRoutineByServerId(ft_info->serverid);
 
                /* truncate_check_rel() has checked that already */
                Assert(routine->ExecForeignTruncate != NULL);
 
                routine->ExecForeignTruncate(ft_info->rels,
                                             behavior,
                                             restart_seqs);
            }
        }
        PG_FINALLY();
        {
            hash_destroy(ft_htab);
        }
        PG_END_TRY();
    }
 
    /*
     * Restart owned sequences if we were asked to.
     */
    foreach(cell, seq_relids)
    {
        Oid         seq_relid = lfirst_oid(cell);
 
        ResetSequence(seq_relid);
    }
 
    /*
     * Write a WAL record to allow this set of actions to be logically
     * decoded.
     *
     * Assemble an array of relids so we can write a single WAL record for the
     * whole action.
     */
    if (relids_logged != NIL)
    {
        xl_heap_truncate xlrec;
        int         i = 0;
 
        /* should only get here if effective_wal_level is 'logical' */
        Assert(XLogLogicalInfoActive());
 
        logrelids = palloc(list_length(relids_logged) * sizeof(Oid));
        foreach(cell, relids_logged)
            logrelids[i++] = lfirst_oid(cell);
 
        xlrec.dbId = MyDatabaseId;
        xlrec.nrelids = list_length(relids_logged);
        xlrec.flags = 0;
        if (behavior == DROP_CASCADE)
            xlrec.flags |= XLH_TRUNCATE_CASCADE;
        if (restart_seqs)
            xlrec.flags |= XLH_TRUNCATE_RESTART_SEQS;
 
        XLogBeginInsert();
        XLogRegisterData(&xlrec, SizeOfHeapTruncate);
        XLogRegisterData(logrelids, list_length(relids_logged) * sizeof(Oid));
 
        XLogSetRecordFlags(XLOG_INCLUDE_ORIGIN);
 
        (void) XLogInsert(RM_HEAP_ID, XLOG_HEAP_TRUNCATE);
    }
 
    /*
     * Process all AFTER STATEMENT TRUNCATE triggers.
     */
    resultRelInfo = resultRelInfos;
    foreach(cell, rels)
    {
        UserContext ucxt;
 
        if (run_as_table_owner)
            SwitchToUntrustedUser(resultRelInfo->ri_RelationDesc->rd_rel->relowner,
                                  &ucxt);
        ExecASTruncateTriggers(estate, resultRelInfo);
        if (run_as_table_owner)
            RestoreUserContext(&ucxt);
        resultRelInfo++;
    }
 
    /* Handle queued AFTER triggers */
    AfterTriggerEndQuery(estate);
 
    /* We can clean up the EState now */
    FreeExecutorState(estate);
 
    /*
     * Close any rels opened by CASCADE (can't do this while EState still
     * holds refs)
     */
    rels = list_difference_ptr(rels, explicit_rels);
    foreach(cell, rels)
    {
        Relation    rel = (Relation) lfirst(cell);
 
        table_close(rel, NoLock);
    }
}

References AccessExclusiveLock, aclcheck_error(), ACLCHECK_NOT_OWNER, AfterTriggerBeginQuery(), AfterTriggerEndQuery(), Assert, CheckTableForSerializableConflictIn(), CreateExecutorState(), CurrentMemoryContext, DROP_CASCADE, DROP_RESTRICT, HASHCTL::entrysize, ereport, errmsg, EState::es_opened_result_relations, ExecASTruncateTriggers(), ExecBSTruncateTriggers(), FdwRoutine::ExecForeignTruncate, fb(), FreeExecutorState(), GetCurrentSubTransactionId(), GetFdwRoutineByServerId(), GetForeignServerIdByRelId(), getOwnedSequences(), GetUserId(), HASH_BLOBS, HASH_CONTEXT, hash_create(), hash_destroy(), HASH_ELEM, HASH_ENTER, hash_search(), hash_seq_init(), hash_seq_search(), HASHCTL::hcxt, heap_truncate_check_FKs(), heap_truncate_find_FKs(), heap_truncate_one_rel(), i, InitResultRelInfo(), HASHCTL::keysize, lappend(), lappend_oid(), lfirst, lfirst_oid, list_copy(), list_difference_ptr(), list_length(), MyDatabaseId, NIL, NoLock, NOTICE, object_ownercheck(), OBJECT_SEQUENCE, OidIsValid, palloc(), PG_END_TRY, PG_FINALLY, PG_TRY, pgstat_count_truncate(), RelationData::rd_createSubid, RelationData::rd_newRelfilelocatorSubid, RelationData::rd_rel, REINDEX_REL_PROCESS_TOAST, reindex_relation(), relation_close(), relation_open(), RelationGetRelationName, RelationGetRelid, RelationIsLogicallyLogged, RelationSetNewRelfilenumber(), ResetSequence(), RestoreUserContext(), ResultRelInfo::ri_RelationDesc, SizeOfHeapTruncate, SwitchToUntrustedUser(), table_close(), table_open(), truncate_check_activity(), truncate_check_perms(), truncate_check_rel(), XLH_TRUNCATE_CASCADE, XLH_TRUNCATE_RESTART_SEQS, XLOG_HEAP_TRUNCATE, XLOG_INCLUDE_ORIGIN, XLogBeginInsert(), XLogInsert(), XLogLogicalInfoActive, XLogRegisterData(), and XLogSetRecordFlags().

Referenced by apply_handle_truncate(), and ExecuteTruncate().

◆ find_composite_type_dependencies()

void find_composite_type_dependencies	(	Oid	typeOid,
		Relation	origRelation,
		const char *	origTypeName
	)

Definition at line 7077 of file tablecmds.c.

{
    Relation    depRel;
    ScanKeyData key[2];
    SysScanDesc depScan;
    HeapTuple   depTup;
 
    /* since this function recurses, it could be driven to stack overflow */
    check_stack_depth();
 
    /*
     * We scan pg_depend to find those things that depend on the given type.
     * (We assume we can ignore refobjsubid for a type.)
     */
    depRel = table_open(DependRelationId, AccessShareLock);
 
    ScanKeyInit(&key[0],
                Anum_pg_depend_refclassid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(TypeRelationId));
    ScanKeyInit(&key[1],
                Anum_pg_depend_refobjid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(typeOid));
 
    depScan = systable_beginscan(depRel, DependReferenceIndexId, true,
                                 NULL, 2, key);
 
    while (HeapTupleIsValid(depTup = systable_getnext(depScan)))
    {
        Form_pg_depend pg_depend = (Form_pg_depend) GETSTRUCT(depTup);
        Relation    rel;
        TupleDesc   tupleDesc;
        Form_pg_attribute att;
 
        /* Check for directly dependent types */
        if (pg_depend->classid == TypeRelationId)
        {
            /*
             * This must be an array, domain, or range containing the given
             * type, so recursively check for uses of this type.  Note that
             * any error message will mention the original type not the
             * container; this is intentional.
             */
            find_composite_type_dependencies(pg_depend->objid,
                                             origRelation, origTypeName);
            continue;
        }
 
        /* Else, ignore dependees that aren't relations */
        if (pg_depend->classid != RelationRelationId)
            continue;
 
        rel = relation_open(pg_depend->objid, AccessShareLock);
        tupleDesc = RelationGetDescr(rel);
 
        /*
         * If objsubid identifies a specific column, refer to that in error
         * messages.  Otherwise, search to see if there's a user column of the
         * type.  (We assume system columns are never of interesting types.)
         * The search is needed because an index containing an expression
         * column of the target type will just be recorded as a whole-relation
         * dependency.  If we do not find a column of the type, the dependency
         * must indicate that the type is transiently referenced in an index
         * expression but not stored on disk, which we assume is OK, just as
         * we do for references in views.  (It could also be that the target
         * type is embedded in some container type that is stored in an index
         * column, but the previous recursion should catch such cases.)
         */
        if (pg_depend->objsubid > 0 && pg_depend->objsubid <= tupleDesc->natts)
            att = TupleDescAttr(tupleDesc, pg_depend->objsubid - 1);
        else
        {
            att = NULL;
            for (int attno = 1; attno <= tupleDesc->natts; attno++)
            {
                att = TupleDescAttr(tupleDesc, attno - 1);
                if (att->atttypid == typeOid && !att->attisdropped)
                    break;
                att = NULL;
            }
            if (att == NULL)
            {
                /* No such column, so assume OK */
                relation_close(rel, AccessShareLock);
                continue;
            }
        }
 
        /*
         * We definitely should reject if the relation has storage.  If it's
         * partitioned, then perhaps we don't have to reject: if there are
         * partitions then we'll fail when we find one, else there is no
         * stored data to worry about.  However, it's possible that the type
         * change would affect conclusions about whether the type is sortable
         * or hashable and thus (if it's a partitioning column) break the
         * partitioning rule.  For now, reject for partitioned rels too.
         */
        if (RELKIND_HAS_STORAGE(rel->rd_rel->relkind) ||
            RELKIND_HAS_PARTITIONS(rel->rd_rel->relkind))
        {
            if (origTypeName)
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("cannot alter type \"%s\" because column \"%s.%s\" uses it",
                                origTypeName,
                                RelationGetRelationName(rel),
                                NameStr(att->attname))));
            else if (origRelation->rd_rel->relkind == RELKIND_COMPOSITE_TYPE)
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("cannot alter type \"%s\" because column \"%s.%s\" uses it",
                                RelationGetRelationName(origRelation),
                                RelationGetRelationName(rel),
                                NameStr(att->attname))));
            else if (origRelation->rd_rel->relkind == RELKIND_FOREIGN_TABLE)
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("cannot alter foreign table \"%s\" because column \"%s.%s\" uses its row type",
                                RelationGetRelationName(origRelation),
                                RelationGetRelationName(rel),
                                NameStr(att->attname))));
            else
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("cannot alter table \"%s\" because column \"%s.%s\" uses its row type",
                                RelationGetRelationName(origRelation),
                                RelationGetRelationName(rel),
                                NameStr(att->attname))));
        }
        else if (OidIsValid(rel->rd_rel->reltype))
        {
            /*
             * A view or composite type itself isn't a problem, but we must
             * recursively check for indirect dependencies via its rowtype.
             */
            find_composite_type_dependencies(rel->rd_rel->reltype,
                                             origRelation, origTypeName);
        }
 
        relation_close(rel, AccessShareLock);
    }
 
    systable_endscan(depScan);
 
    relation_close(depRel, AccessShareLock);
}

References AccessShareLock, BTEqualStrategyNumber, check_stack_depth(), ereport, errcode(), errmsg, ERROR, fb(), find_composite_type_dependencies(), Form_pg_depend, GETSTRUCT(), HeapTupleIsValid, NameStr, ObjectIdGetDatum(), OidIsValid, RelationData::rd_rel, relation_close(), relation_open(), RelationGetDescr, RelationGetRelationName, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_open(), and TupleDescAttr().

Referenced by ATPrepAlterColumnType(), ATRewriteTables(), find_composite_type_dependencies(), and get_rels_with_domain().

◆ find_typed_table_dependencies()

static List * find_typed_table_dependencies	(	Oid	typeOid,
		const char *	typeName,
		DropBehavior	behavior
	)

static

Definition at line 7235 of file tablecmds.c.

{
    Relation    classRel;
    ScanKeyData key[1];
    TableScanDesc scan;
    HeapTuple   tuple;
    List       *result = NIL;
 
    classRel = table_open(RelationRelationId, AccessShareLock);
 
    ScanKeyInit(&key[0],
                Anum_pg_class_reloftype,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(typeOid));
 
    scan = table_beginscan_catalog(classRel, 1, key);
 
    while ((tuple = heap_getnext(scan, ForwardScanDirection)) != NULL)
    {
        Form_pg_class classform = (Form_pg_class) GETSTRUCT(tuple);
 
        if (behavior == DROP_RESTRICT)
            ereport(ERROR,
                    (errcode(ERRCODE_DEPENDENT_OBJECTS_STILL_EXIST),
                     errmsg("cannot alter type \"%s\" because it is the type of a typed table",
                            typeName),
                     errhint("Use ALTER ... CASCADE to alter the typed tables too.")));
        else
            result = lappend_oid(result, classform->oid);
    }
 
    table_endscan(scan);
    table_close(classRel, AccessShareLock);
 
    return result;
}

References AccessShareLock, BTEqualStrategyNumber, DROP_RESTRICT, ereport, errcode(), errhint(), errmsg, ERROR, fb(), ForwardScanDirection, GETSTRUCT(), heap_getnext(), lappend_oid(), NIL, ObjectIdGetDatum(), result, ScanKeyInit(), table_beginscan_catalog(), table_close(), table_endscan(), and table_open().

Referenced by ATTypedTableRecursion(), and renameatt_internal().

◆ findAttrByName()

static int findAttrByName	(	const char *	attributeName,
		const List *	columns
	)

static

Definition at line 3689 of file tablecmds.c.

{
    ListCell   *lc;
    int         i = 1;
 
    foreach(lc, columns)
    {
        if (strcmp(attributeName, lfirst_node(ColumnDef, lc)->colname) == 0)
            return i;
 
        i++;
    }
    return 0;
}

References fb(), i, and lfirst_node.

Referenced by MergeAttributes().

◆ findFkeyCast()

static CoercionPathType findFkeyCast	(	Oid	targetTypeId,
		Oid	sourceTypeId,
		Oid *	funcid
	)

static

Definition at line 14173 of file tablecmds.c.

{
    CoercionPathType ret;
 
    if (targetTypeId == sourceTypeId)
    {
        ret = COERCION_PATH_RELABELTYPE;
        *funcid = InvalidOid;
    }
    else
    {
        ret = find_coercion_pathway(targetTypeId, sourceTypeId,
                                    COERCION_IMPLICIT, funcid);
        if (ret == COERCION_PATH_NONE)
            /* A previously-relied-upon cast is now gone. */
            elog(ERROR, "could not find cast from %u to %u",
                 sourceTypeId, targetTypeId);
    }
 
    return ret;
}

References COERCION_IMPLICIT, COERCION_PATH_NONE, COERCION_PATH_RELABELTYPE, elog, ERROR, fb(), find_coercion_pathway(), and InvalidOid.

Referenced by ATAddForeignKeyConstraint().

◆ freePartitionIndexExtDeps()

static void freePartitionIndexExtDeps ( List * extDepState )

static

Definition at line 23742 of file tablecmds.c.

{
    foreach_ptr(PartitionIndexExtDepEntry, entry, extDepState)
    {
        list_free(entry->extensionOids);
        pfree(entry);
    }
    list_free(extDepState);
}

References fb(), foreach_ptr, list_free(), and pfree().

Referenced by ATExecMergePartitions(), and ATExecSplitPartition().

◆ GetAttributeCompression()

static char GetAttributeCompression	(	Oid	atttypid,
		const char *	compression
	)

static

Definition at line 22790 of file tablecmds.c.

{
    char        cmethod;
 
    if (compression == NULL || strcmp(compression, "default") == 0)
        return InvalidCompressionMethod;
 
    /*
     * To specify a nondefault method, the column data type must be toastable.
     * Note this says nothing about whether the column's attstorage setting
     * permits compression; we intentionally allow attstorage and
     * attcompression to be independent.  But with a non-toastable type,
     * attstorage could not be set to a value that would permit compression.
     *
     * We don't actually need to enforce this, since nothing bad would happen
     * if attcompression were non-default; it would never be consulted.  But
     * it seems more user-friendly to complain about a certainly-useless
     * attempt to set the property.
     */
    if (!TypeIsToastable(atttypid))
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("column data type %s does not support compression",
                        format_type_be(atttypid))));
 
    cmethod = CompressionNameToMethod(compression);
    if (!CompressionMethodIsValid(cmethod))
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                 errmsg("invalid compression method \"%s\"", compression)));
 
    return cmethod;
}

References CompressionMethodIsValid, CompressionNameToMethod(), ereport, errcode(), errmsg, ERROR, fb(), format_type_be(), InvalidCompressionMethod, and TypeIsToastable.

Referenced by ATExecSetCompression(), and BuildDescForRelation().

◆ getAttributesList()

static List * getAttributesList ( Relation parent_rel )

static

Definition at line 22951 of file tablecmds.c.

{
    AttrNumber  parent_attno;
    TupleDesc   modelDesc;
    List       *colList = NIL;
 
    modelDesc = RelationGetDescr(parent_rel);
 
    for (parent_attno = 1; parent_attno <= modelDesc->natts;
         parent_attno++)
    {
        Form_pg_attribute attribute = TupleDescAttr(modelDesc,
                                                    parent_attno - 1);
        ColumnDef  *def;
 
        /* Ignore dropped columns in the parent. */
        if (attribute->attisdropped)
            continue;
 
        def = makeColumnDef(NameStr(attribute->attname), attribute->atttypid,
                            attribute->atttypmod, attribute->attcollation);
 
        def->is_not_null = attribute->attnotnull;
 
        /* Copy identity. */
        def->identity = attribute->attidentity;
 
        /* Copy attgenerated. */
        def->generated = attribute->attgenerated;
 
        def->storage = attribute->attstorage;
 
        /* Likewise, copy compression. */
        if (CompressionMethodIsValid(attribute->attcompression))
            def->compression =
                pstrdup(GetCompressionMethodName(attribute->attcompression));
        else
            def->compression = NULL;
 
        /* Add to column list. */
        colList = lappend(colList, def);
    }
 
    return colList;
}

References ColumnDef::compression, CompressionMethodIsValid, fb(), ColumnDef::generated, GetCompressionMethodName(), ColumnDef::identity, ColumnDef::is_not_null, lappend(), makeColumnDef(), NameStr, NIL, pstrdup(), RelationGetDescr, ColumnDef::storage, and TupleDescAttr().

Referenced by createPartitionTable().

◆ GetAttributeStorage()

static char GetAttributeStorage	(	Oid	atttypid,
		const char *	storagemode
	)

static

Definition at line 22828 of file tablecmds.c.

{
    char        cstorage = 0;
 
    if (pg_strcasecmp(storagemode, "plain") == 0)
        cstorage = TYPSTORAGE_PLAIN;
    else if (pg_strcasecmp(storagemode, "external") == 0)
        cstorage = TYPSTORAGE_EXTERNAL;
    else if (pg_strcasecmp(storagemode, "extended") == 0)
        cstorage = TYPSTORAGE_EXTENDED;
    else if (pg_strcasecmp(storagemode, "main") == 0)
        cstorage = TYPSTORAGE_MAIN;
    else if (pg_strcasecmp(storagemode, "default") == 0)
        cstorage = get_typstorage(atttypid);
    else
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                 errmsg("invalid storage type \"%s\"",
                        storagemode)));
 
    /*
     * safety check: do not allow toasted storage modes unless column datatype
     * is TOAST-aware.
     */
    if (!(cstorage == TYPSTORAGE_PLAIN || TypeIsToastable(atttypid)))
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("column data type %s can only have storage PLAIN",
                        format_type_be(atttypid))));
 
    return cstorage;
}

References ereport, errcode(), errmsg, ERROR, fb(), format_type_be(), get_typstorage(), pg_strcasecmp(), and TypeIsToastable.

Referenced by ATExecSetStorage(), and BuildDescForRelation().

◆ GetForeignKeyActionTriggers()

static void GetForeignKeyActionTriggers	(	Relation	trigrel,
		Oid	conoid,
		Oid	confrelid,
		Oid	conrelid,
		Oid *	deleteTriggerOid,
		Oid *	updateTriggerOid
	)

static

Definition at line 12219 of file tablecmds.c.

{
    ScanKeyData key;
    SysScanDesc scan;
    HeapTuple   trigtup;
 
    *deleteTriggerOid = *updateTriggerOid = InvalidOid;
    ScanKeyInit(&key,
                Anum_pg_trigger_tgconstraint,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(conoid));
 
    scan = systable_beginscan(trigrel, TriggerConstraintIndexId, true,
                              NULL, 1, &key);
    while ((trigtup = systable_getnext(scan)) != NULL)
    {
        Form_pg_trigger trgform = (Form_pg_trigger) GETSTRUCT(trigtup);
 
        if (trgform->tgconstrrelid != conrelid)
            continue;
        if (trgform->tgrelid != confrelid)
            continue;
        /* Only ever look at "action" triggers on the PK side. */
        if (RI_FKey_trigger_type(trgform->tgfoid) != RI_TRIGGER_PK)
            continue;
        if (TRIGGER_FOR_DELETE(trgform->tgtype))
        {
            Assert(*deleteTriggerOid == InvalidOid);
            *deleteTriggerOid = trgform->oid;
        }
        else if (TRIGGER_FOR_UPDATE(trgform->tgtype))
        {
            Assert(*updateTriggerOid == InvalidOid);
            *updateTriggerOid = trgform->oid;
        }
#ifndef USE_ASSERT_CHECKING
        /* In an assert-enabled build, continue looking to find duplicates */
        if (OidIsValid(*deleteTriggerOid) && OidIsValid(*updateTriggerOid))
            break;
#endif
    }
 
    if (!OidIsValid(*deleteTriggerOid))
        elog(ERROR, "could not find ON DELETE action trigger of foreign key constraint %u",
             conoid);
    if (!OidIsValid(*updateTriggerOid))
        elog(ERROR, "could not find ON UPDATE action trigger of foreign key constraint %u",
             conoid);
 
    systable_endscan(scan);
}

References Assert, BTEqualStrategyNumber, elog, ERROR, fb(), Form_pg_trigger, GETSTRUCT(), InvalidOid, ObjectIdGetDatum(), OidIsValid, RI_FKey_trigger_type(), RI_TRIGGER_PK, ScanKeyInit(), systable_beginscan(), systable_endscan(), and systable_getnext().

Referenced by CloneFkReferenced().

◆ GetForeignKeyCheckTriggers()

static void GetForeignKeyCheckTriggers	(	Relation	trigrel,
		Oid	conoid,
		Oid	confrelid,
		Oid	conrelid,
		Oid *	insertTriggerOid,
		Oid *	updateTriggerOid
	)

static

Definition at line 12280 of file tablecmds.c.

{
    ScanKeyData key;
    SysScanDesc scan;
    HeapTuple   trigtup;
 
    *insertTriggerOid = *updateTriggerOid = InvalidOid;
    ScanKeyInit(&key,
                Anum_pg_trigger_tgconstraint,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(conoid));
 
    scan = systable_beginscan(trigrel, TriggerConstraintIndexId, true,
                              NULL, 1, &key);
    while ((trigtup = systable_getnext(scan)) != NULL)
    {
        Form_pg_trigger trgform = (Form_pg_trigger) GETSTRUCT(trigtup);
 
        if (trgform->tgconstrrelid != confrelid)
            continue;
        if (trgform->tgrelid != conrelid)
            continue;
        /* Only ever look at "check" triggers on the FK side. */
        if (RI_FKey_trigger_type(trgform->tgfoid) != RI_TRIGGER_FK)
            continue;
        if (TRIGGER_FOR_INSERT(trgform->tgtype))
        {
            Assert(*insertTriggerOid == InvalidOid);
            *insertTriggerOid = trgform->oid;
        }
        else if (TRIGGER_FOR_UPDATE(trgform->tgtype))
        {
            Assert(*updateTriggerOid == InvalidOid);
            *updateTriggerOid = trgform->oid;
        }
#ifndef USE_ASSERT_CHECKING
        /* In an assert-enabled build, continue looking to find duplicates. */
        if (OidIsValid(*insertTriggerOid) && OidIsValid(*updateTriggerOid))
            break;
#endif
    }
 
    if (!OidIsValid(*insertTriggerOid))
        elog(ERROR, "could not find ON INSERT check triggers of foreign key constraint %u",
             conoid);
    if (!OidIsValid(*updateTriggerOid))
        elog(ERROR, "could not find ON UPDATE check triggers of foreign key constraint %u",
             conoid);
 
    systable_endscan(scan);
}

References Assert, BTEqualStrategyNumber, elog, ERROR, fb(), Form_pg_trigger, GETSTRUCT(), InvalidOid, ObjectIdGetDatum(), OidIsValid, RI_FKey_trigger_type(), RI_TRIGGER_FK, ScanKeyInit(), systable_beginscan(), systable_endscan(), and systable_getnext().

Referenced by AttachPartitionForeignKey(), CloneFkReferencing(), and DetachPartitionFinalize().

◆ GetParentedForeignKeyRefs()

static List * GetParentedForeignKeyRefs ( Relation partition )

static

Definition at line 22688 of file tablecmds.c.

{
    Relation    pg_constraint;
    HeapTuple   tuple;
    SysScanDesc scan;
    ScanKeyData key[2];
    List       *constraints = NIL;
 
    /*
     * If no indexes, or no columns are referenceable by FKs, we can avoid the
     * scan.
     */
    if (RelationGetIndexList(partition) == NIL ||
        bms_is_empty(RelationGetIndexAttrBitmap(partition,
                                                INDEX_ATTR_BITMAP_KEY)))
        return NIL;
 
    /* Search for constraints referencing this table */
    pg_constraint = table_open(ConstraintRelationId, AccessShareLock);
    ScanKeyInit(&key[0],
                Anum_pg_constraint_confrelid, BTEqualStrategyNumber,
                F_OIDEQ, ObjectIdGetDatum(RelationGetRelid(partition)));
    ScanKeyInit(&key[1],
                Anum_pg_constraint_contype, BTEqualStrategyNumber,
                F_CHAREQ, CharGetDatum(CONSTRAINT_FOREIGN));
 
    /* XXX This is a seqscan, as we don't have a usable index */
    scan = systable_beginscan(pg_constraint, InvalidOid, true, NULL, 2, key);
    while ((tuple = systable_getnext(scan)) != NULL)
    {
        Form_pg_constraint constrForm = (Form_pg_constraint) GETSTRUCT(tuple);
 
        /*
         * We only need to process constraints that are part of larger ones.
         */
        if (!OidIsValid(constrForm->conparentid))
            continue;
 
        constraints = lappend_oid(constraints, constrForm->oid);
    }
 
    systable_endscan(scan);
    table_close(pg_constraint, AccessShareLock);
 
    return constraints;
}

References AccessShareLock, bms_is_empty, BTEqualStrategyNumber, CharGetDatum(), fb(), Form_pg_constraint, GETSTRUCT(), INDEX_ATTR_BITMAP_KEY, InvalidOid, lappend_oid(), NIL, ObjectIdGetDatum(), OidIsValid, RelationGetIndexAttrBitmap(), RelationGetIndexList(), RelationGetRelid, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by ATDetachCheckNoForeignKeyRefs(), and DetachPartitionFinalize().

◆ index_copy_data()

static void index_copy_data	(	Relation	rel,
		RelFileLocator	newrlocator
	)

static

Definition at line 17859 of file tablecmds.c.

{
    SMgrRelation dstrel;
 
    /*
     * Since we copy the file directly without looking at the shared buffers,
     * we'd better first flush out any pages of the source relation that are
     * in shared buffers.  We assume no new changes will be made while we are
     * holding exclusive lock on the rel.
     */
    FlushRelationBuffers(rel);
 
    /*
     * Create and copy all forks of the relation, and schedule unlinking of
     * old physical files.
     *
     * NOTE: any conflict in relfilenumber value will be caught in
     * RelationCreateStorage().
     */
    dstrel = RelationCreateStorage(newrlocator, rel->rd_rel->relpersistence, true);
 
    /* copy main fork */
    RelationCopyStorage(RelationGetSmgr(rel), dstrel, MAIN_FORKNUM,
                        rel->rd_rel->relpersistence);
 
    /* copy those extra forks that exist */
    for (ForkNumber forkNum = MAIN_FORKNUM + 1;
         forkNum <= MAX_FORKNUM; forkNum++)
    {
        if (smgrexists(RelationGetSmgr(rel), forkNum))
        {
            smgrcreate(dstrel, forkNum, false);
 
            /*
             * WAL log creation if the relation is persistent, or this is the
             * init fork of an unlogged relation.
             */
            if (RelationIsPermanent(rel) ||
                (rel->rd_rel->relpersistence == RELPERSISTENCE_UNLOGGED &&
                 forkNum == INIT_FORKNUM))
                log_smgrcreate(&newrlocator, forkNum);
            RelationCopyStorage(RelationGetSmgr(rel), dstrel, forkNum,
                                rel->rd_rel->relpersistence);
        }
    }
 
    /* drop old relation, and close new one */
    RelationDropStorage(rel);
    smgrclose(dstrel);
}

References fb(), FlushRelationBuffers(), INIT_FORKNUM, log_smgrcreate(), MAIN_FORKNUM, MAX_FORKNUM, RelationData::rd_rel, RelationCopyStorage(), RelationCreateStorage(), RelationDropStorage(), RelationGetSmgr(), RelationIsPermanent, smgrclose(), smgrcreate(), and smgrexists().

Referenced by ATExecSetTableSpace().

◆ MarkInheritDetached()

static void MarkInheritDetached	(	Relation	child_rel,
		Relation	parent_rel
	)

static

Definition at line 18570 of file tablecmds.c.

{
    Relation    catalogRelation;
    SysScanDesc scan;
    ScanKeyData key;
    HeapTuple   inheritsTuple;
    bool        found = false;
 
    Assert(parent_rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE);
 
    /*
     * Find pg_inherits entries by inhparent.  (We need to scan them all in
     * order to verify that no other partition is pending detach.)
     */
    catalogRelation = table_open(InheritsRelationId, RowExclusiveLock);
    ScanKeyInit(&key,
                Anum_pg_inherits_inhparent,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(RelationGetRelid(parent_rel)));
    scan = systable_beginscan(catalogRelation, InheritsParentIndexId,
                              true, NULL, 1, &key);
 
    while (HeapTupleIsValid(inheritsTuple = systable_getnext(scan)))
    {
        Form_pg_inherits inhForm;
 
        inhForm = (Form_pg_inherits) GETSTRUCT(inheritsTuple);
        if (inhForm->inhdetachpending)
            ereport(ERROR,
                    errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                    errmsg("partition \"%s\" already pending detach in partitioned table \"%s.%s\"",
                           get_rel_name(inhForm->inhrelid),
                           get_namespace_name(parent_rel->rd_rel->relnamespace),
                           RelationGetRelationName(parent_rel)),
                    errhint("Use ALTER TABLE ... DETACH PARTITION ... FINALIZE to complete the pending detach operation."));
 
        if (inhForm->inhrelid == RelationGetRelid(child_rel))
        {
            HeapTuple   newtup;
 
            newtup = heap_copytuple(inheritsTuple);
            ((Form_pg_inherits) GETSTRUCT(newtup))->inhdetachpending = true;
 
            CatalogTupleUpdate(catalogRelation,
                               &inheritsTuple->t_self,
                               newtup);
            found = true;
            heap_freetuple(newtup);
            /* keep looking, to ensure we catch others pending detach */
        }
    }
 
    /* Done */
    systable_endscan(scan);
    table_close(catalogRelation, RowExclusiveLock);
 
    if (!found)
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_TABLE),
                 errmsg("relation \"%s\" is not a partition of relation \"%s\"",
                        RelationGetRelationName(child_rel),
                        RelationGetRelationName(parent_rel))));
}

References Assert, BTEqualStrategyNumber, CatalogTupleUpdate(), ereport, errcode(), ERRCODE_UNDEFINED_TABLE, errhint(), errmsg, ERROR, fb(), Form_pg_inherits, get_namespace_name(), get_rel_name(), GETSTRUCT(), heap_copytuple(), heap_freetuple(), HeapTupleIsValid, ObjectIdGetDatum(), RelationGetRelationName, RelationGetRelid, RowExclusiveLock, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by ATExecDetachPartition().

◆ MergeAttributes()

static List * MergeAttributes	(	List *	columns,
		const List *	supers,
		char	relpersistence,
		bool	is_partition,
		List **	supconstr,
		List **	supnotnulls
	)

static

Definition at line 2614 of file tablecmds.c.

{
    List       *inh_columns = NIL;
    List       *constraints = NIL;
    List       *nnconstraints = NIL;
    bool        have_bogus_defaults = false;
    int         child_attno;
    static Node bogus_marker = {0}; /* marks conflicting defaults */
    List       *saved_columns = NIL;
    ListCell   *lc;
 
    /*
     * Check for and reject tables with too many columns. We perform this
     * check relatively early for two reasons: (a) we don't run the risk of
     * overflowing an AttrNumber in subsequent code (b) an O(n^2) algorithm is
     * okay if we're processing <= 1600 columns, but could take minutes to
     * execute if the user attempts to create a table with hundreds of
     * thousands of columns.
     *
     * Note that we also need to check that we do not exceed this figure after
     * including columns from inherited relations.
     */
    if (list_length(columns) > MaxHeapAttributeNumber)
        ereport(ERROR,
                (errcode(ERRCODE_TOO_MANY_COLUMNS),
                 errmsg("tables can have at most %d columns",
                        MaxHeapAttributeNumber)));
 
    /*
     * Check for duplicate names in the explicit list of attributes.
     *
     * Although we might consider merging such entries in the same way that we
     * handle name conflicts for inherited attributes, it seems to make more
     * sense to assume such conflicts are errors.
     *
     * We don't use foreach() here because we have two nested loops over the
     * columns list, with possible element deletions in the inner one.  If we
     * used foreach_delete_current() it could only fix up the state of one of
     * the loops, so it seems cleaner to use looping over list indexes for
     * both loops.  Note that any deletion will happen beyond where the outer
     * loop is, so its index never needs adjustment.
     */
    for (int coldefpos = 0; coldefpos < list_length(columns); coldefpos++)
    {
        ColumnDef  *coldef = list_nth_node(ColumnDef, columns, coldefpos);
 
        if (!is_partition && coldef->typeName == NULL)
        {
            /*
             * Typed table column option that does not belong to a column from
             * the type.  This works because the columns from the type come
             * first in the list.  (We omit this check for partition column
             * lists; those are processed separately below.)
             */
            ereport(ERROR,
                    (errcode(ERRCODE_UNDEFINED_COLUMN),
                     errmsg("column \"%s\" does not exist",
                            coldef->colname)));
        }
 
        /* restpos scans all entries beyond coldef; incr is in loop body */
        for (int restpos = coldefpos + 1; restpos < list_length(columns);)
        {
            ColumnDef  *restdef = list_nth_node(ColumnDef, columns, restpos);
 
            if (strcmp(coldef->colname, restdef->colname) == 0)
            {
                if (coldef->is_from_type)
                {
                    /*
                     * merge the column options into the column from the type
                     */
                    coldef->is_not_null = restdef->is_not_null;
                    coldef->raw_default = restdef->raw_default;
                    coldef->cooked_default = restdef->cooked_default;
                    coldef->constraints = restdef->constraints;
                    coldef->is_from_type = false;
                    columns = list_delete_nth_cell(columns, restpos);
                }
                else
                    ereport(ERROR,
                            (errcode(ERRCODE_DUPLICATE_COLUMN),
                             errmsg("column \"%s\" specified more than once",
                                    coldef->colname)));
            }
            else
                restpos++;
        }
    }
 
    /*
     * In case of a partition, there are no new column definitions, only dummy
     * ColumnDefs created for column constraints.  Set them aside for now and
     * process them at the end.
     */
    if (is_partition)
    {
        saved_columns = columns;
        columns = NIL;
    }
 
    /*
     * Scan the parents left-to-right, and merge their attributes to form a
     * list of inherited columns (inh_columns).
     */
    child_attno = 0;
    foreach(lc, supers)
    {
        Oid         parent = lfirst_oid(lc);
        Relation    relation;
        TupleDesc   tupleDesc;
        TupleConstr *constr;
        AttrMap    *newattmap;
        List       *inherited_defaults;
        List       *cols_with_defaults;
        List       *nnconstrs;
        ListCell   *lc1;
        ListCell   *lc2;
        Bitmapset  *nncols = NULL;
 
        /* caller already got lock */
        relation = table_open(parent, NoLock);
 
        /*
         * Check for active uses of the parent partitioned table in the
         * current transaction, such as being used in some manner by an
         * enclosing command.
         */
        if (is_partition)
            CheckTableNotInUse(relation, "CREATE TABLE .. PARTITION OF");
 
        /*
         * We do not allow partitioned tables and partitions to participate in
         * regular inheritance.
         */
        if (relation->rd_rel->relkind == RELKIND_PARTITIONED_TABLE && !is_partition)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("cannot inherit from partitioned table \"%s\"",
                            RelationGetRelationName(relation))));
        if (relation->rd_rel->relispartition && !is_partition)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("cannot inherit from partition \"%s\"",
                            RelationGetRelationName(relation))));
 
        /*
         * Conflict log tables are managed by the system for logical
         * replication and should not be used as parent tables, as inheritance
         * could interfere with the logging behavior.
         */
        if (IsConflictLogTableNamespace(relation->rd_rel->relnamespace))
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("cannot inherit from conflict log table \"%s\"",
                            RelationGetRelationName(relation)),
                     errdetail("Conflict log tables are system-managed tables for logical replication conflicts.")));
 
        if (relation->rd_rel->relkind != RELKIND_RELATION &&
            relation->rd_rel->relkind != RELKIND_FOREIGN_TABLE &&
            relation->rd_rel->relkind != RELKIND_PARTITIONED_TABLE)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("inherited relation \"%s\" is not a table or foreign table",
                            RelationGetRelationName(relation))));
 
        /*
         * If the parent is permanent, so must be all of its partitions.  Note
         * that inheritance allows that case.
         */
        if (is_partition &&
            relation->rd_rel->relpersistence != RELPERSISTENCE_TEMP &&
            relpersistence == RELPERSISTENCE_TEMP)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg("cannot create a temporary relation as partition of permanent relation \"%s\"",
                            RelationGetRelationName(relation))));
 
        /* Permanent rels cannot inherit from temporary ones */
        if (relpersistence != RELPERSISTENCE_TEMP &&
            relation->rd_rel->relpersistence == RELPERSISTENCE_TEMP)
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg(!is_partition
                            ? "cannot inherit from temporary relation \"%s\""
                            : "cannot create a permanent relation as partition of temporary relation \"%s\"",
                            RelationGetRelationName(relation))));
 
        /* If existing rel is temp, it must belong to this session */
        if (RELATION_IS_OTHER_TEMP(relation))
            ereport(ERROR,
                    (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                     errmsg(!is_partition
                            ? "cannot inherit from temporary relation of another session"
                            : "cannot create as partition of temporary relation of another session")));
 
        /*
         * We should have an UNDER permission flag for this, but for now,
         * demand that creator of a child table own the parent.
         */
        if (!object_ownercheck(RelationRelationId, RelationGetRelid(relation), GetUserId()))
            aclcheck_error(ACLCHECK_NOT_OWNER, get_relkind_objtype(relation->rd_rel->relkind),
                           RelationGetRelationName(relation));
 
        tupleDesc = RelationGetDescr(relation);
        constr = tupleDesc->constr;
 
        /*
         * newattmap->attnums[] will contain the child-table attribute numbers
         * for the attributes of this parent table.  (They are not the same
         * for parents after the first one, nor if we have dropped columns.)
         */
        newattmap = make_attrmap(tupleDesc->natts);
 
        /* We can't process inherited defaults until newattmap is complete. */
        inherited_defaults = cols_with_defaults = NIL;
 
        /*
         * Request attnotnull on columns that have a not-null constraint
         * that's not marked NO INHERIT (even if not valid).
         */
        nnconstrs = RelationGetNotNullConstraints(RelationGetRelid(relation),
                                                  true, false);
        foreach_ptr(CookedConstraint, cc, nnconstrs)
            nncols = bms_add_member(nncols, cc->attnum);
 
        for (AttrNumber parent_attno = 1; parent_attno <= tupleDesc->natts;
             parent_attno++)
        {
            Form_pg_attribute attribute = TupleDescAttr(tupleDesc,
                                                        parent_attno - 1);
            char       *attributeName = NameStr(attribute->attname);
            int         exist_attno;
            ColumnDef  *newdef;
            ColumnDef  *mergeddef;
 
            /*
             * Ignore dropped columns in the parent.
             */
            if (attribute->attisdropped)
                continue;       /* leave newattmap->attnums entry as zero */
 
            /*
             * Create new column definition
             */
            newdef = makeColumnDef(attributeName, attribute->atttypid,
                                   attribute->atttypmod, attribute->attcollation);
            newdef->storage = attribute->attstorage;
            newdef->generated = attribute->attgenerated;
            if (CompressionMethodIsValid(attribute->attcompression))
                newdef->compression =
                    pstrdup(GetCompressionMethodName(attribute->attcompression));
 
            /*
             * Regular inheritance children are independent enough not to
             * inherit identity columns.  But partitions are integral part of
             * a partitioned table and inherit identity column.
             */
            if (is_partition)
                newdef->identity = attribute->attidentity;
 
            /*
             * Does it match some previously considered column from another
             * parent?
             */
            exist_attno = findAttrByName(attributeName, inh_columns);
            if (exist_attno > 0)
            {
                /*
                 * Yes, try to merge the two column definitions.
                 */
                mergeddef = MergeInheritedAttribute(inh_columns, exist_attno, newdef);
 
                newattmap->attnums[parent_attno - 1] = exist_attno;
 
                /*
                 * Partitions have only one parent, so conflict should never
                 * occur.
                 */
                Assert(!is_partition);
            }
            else
            {
                /*
                 * No, create a new inherited column
                 */
                newdef->inhcount = 1;
                newdef->is_local = false;
                inh_columns = lappend(inh_columns, newdef);
 
                newattmap->attnums[parent_attno - 1] = ++child_attno;
                mergeddef = newdef;
            }
 
            /*
             * mark attnotnull if parent has it
             */
            if (bms_is_member(parent_attno, nncols))
                mergeddef->is_not_null = true;
 
            /*
             * Locate default/generation expression if any
             */
            if (attribute->atthasdef)
            {
                Node       *this_default;
 
                this_default = TupleDescGetDefault(tupleDesc, parent_attno);
                if (this_default == NULL)
                    elog(ERROR, "default expression not found for attribute %d of relation \"%s\"",
                         parent_attno, RelationGetRelationName(relation));
 
                /*
                 * If it's a GENERATED default, it might contain Vars that
                 * need to be mapped to the inherited column(s)' new numbers.
                 * We can't do that till newattmap is ready, so just remember
                 * all the inherited default expressions for the moment.
                 */
                inherited_defaults = lappend(inherited_defaults, this_default);
                cols_with_defaults = lappend(cols_with_defaults, mergeddef);
            }
        }
 
        /*
         * Now process any inherited default expressions, adjusting attnos
         * using the completed newattmap map.
         */
        forboth(lc1, inherited_defaults, lc2, cols_with_defaults)
        {
            Node       *this_default = (Node *) lfirst(lc1);
            ColumnDef  *def = (ColumnDef *) lfirst(lc2);
            bool        found_whole_row;
 
            /* Adjust Vars to match new table's column numbering */
            this_default = map_variable_attnos(this_default,
                                               1, 0,
                                               newattmap,
                                               InvalidOid, &found_whole_row);
 
            /*
             * For the moment we have to reject whole-row variables.  We could
             * convert them, if we knew the new table's rowtype OID, but that
             * hasn't been assigned yet.  (A variable could only appear in a
             * generation expression, so the error message is correct.)
             */
            if (found_whole_row)
                ereport(ERROR,
                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                         errmsg("cannot convert whole-row table reference"),
                         errdetail("Generation expression for column \"%s\" contains a whole-row reference to table \"%s\".",
                                   def->colname,
                                   RelationGetRelationName(relation))));
 
            /*
             * If we already had a default from some prior parent, check to
             * see if they are the same.  If so, no problem; if not, mark the
             * column as having a bogus default.  Below, we will complain if
             * the bogus default isn't overridden by the child columns.
             */
            Assert(def->raw_default == NULL);
            if (def->cooked_default == NULL)
                def->cooked_default = this_default;
            else if (!equal(def->cooked_default, this_default))
            {
                def->cooked_default = &bogus_marker;
                have_bogus_defaults = true;
            }
        }
 
        /*
         * Now copy the CHECK constraints of this parent, adjusting attnos
         * using the completed newattmap map.  Identically named constraints
         * are merged if possible, else we throw error.
         */
        if (constr && constr->num_check > 0)
        {
            ConstrCheck *check = constr->check;
 
            for (int i = 0; i < constr->num_check; i++)
            {
                char       *name = check[i].ccname;
                Node       *expr;
                bool        found_whole_row;
 
                /* ignore if the constraint is non-inheritable */
                if (check[i].ccnoinherit)
                    continue;
 
                /* Adjust Vars to match new table's column numbering */
                expr = map_variable_attnos(stringToNode(check[i].ccbin),
                                           1, 0,
                                           newattmap,
                                           InvalidOid, &found_whole_row);
 
                /*
                 * For the moment we have to reject whole-row variables. We
                 * could convert them, if we knew the new table's rowtype OID,
                 * but that hasn't been assigned yet.
                 */
                if (found_whole_row)
                    ereport(ERROR,
                            (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                             errmsg("cannot convert whole-row table reference"),
                             errdetail("Constraint \"%s\" contains a whole-row reference to table \"%s\".",
                                       name,
                                       RelationGetRelationName(relation))));
 
                constraints = MergeCheckConstraint(constraints, name, expr,
                                                   check[i].ccenforced);
            }
        }
 
        /*
         * Also copy the not-null constraints from this parent.  The
         * attnotnull markings were already installed above.
         */
        foreach_ptr(CookedConstraint, nn, nnconstrs)
        {
            Assert(nn->contype == CONSTR_NOTNULL);
 
            nn->attnum = newattmap->attnums[nn->attnum - 1];
 
            nnconstraints = lappend(nnconstraints, nn);
        }
 
        free_attrmap(newattmap);
 
        /*
         * Close the parent rel, but keep our lock on it until xact commit.
         * That will prevent someone else from deleting or ALTERing the parent
         * before the child is committed.
         */
        table_close(relation, NoLock);
    }
 
    /*
     * If we had no inherited attributes, the result columns are just the
     * explicitly declared columns.  Otherwise, we need to merge the declared
     * columns into the inherited column list.  Although, we never have any
     * explicitly declared columns if the table is a partition.
     */
    if (inh_columns != NIL)
    {
        int         newcol_attno = 0;
 
        foreach(lc, columns)
        {
            ColumnDef  *newdef = lfirst_node(ColumnDef, lc);
            char       *attributeName = newdef->colname;
            int         exist_attno;
 
            /*
             * Partitions have only one parent and have no column definitions
             * of their own, so conflict should never occur.
             */
            Assert(!is_partition);
 
            newcol_attno++;
 
            /*
             * Does it match some inherited column?
             */
            exist_attno = findAttrByName(attributeName, inh_columns);
            if (exist_attno > 0)
            {
                /*
                 * Yes, try to merge the two column definitions.
                 */
                MergeChildAttribute(inh_columns, exist_attno, newcol_attno, newdef);
            }
            else
            {
                /*
                 * No, attach new column unchanged to result columns.
                 */
                inh_columns = lappend(inh_columns, newdef);
            }
        }
 
        columns = inh_columns;
 
        /*
         * Check that we haven't exceeded the legal # of columns after merging
         * in inherited columns.
         */
        if (list_length(columns) > MaxHeapAttributeNumber)
            ereport(ERROR,
                    (errcode(ERRCODE_TOO_MANY_COLUMNS),
                     errmsg("tables can have at most %d columns",
                            MaxHeapAttributeNumber)));
    }
 
    /*
     * Now that we have the column definition list for a partition, we can
     * check whether the columns referenced in the column constraint specs
     * actually exist.  Also, merge column defaults.
     */
    if (is_partition)
    {
        foreach(lc, saved_columns)
        {
            ColumnDef  *restdef = lfirst(lc);
            bool        found = false;
            ListCell   *l;
 
            foreach(l, columns)
            {
                ColumnDef  *coldef = lfirst(l);
 
                if (strcmp(coldef->colname, restdef->colname) == 0)
                {
                    found = true;
 
                    /*
                     * Check for conflicts related to generated columns.
                     *
                     * Same rules as above: generated-ness has to match the
                     * parent, but the contents of the generation expression
                     * can be different.
                     */
                    if (coldef->generated)
                    {
                        if (restdef->raw_default && !restdef->generated)
                            ereport(ERROR,
                                    (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
                                     errmsg("column \"%s\" inherits from generated column but specifies default",
                                            restdef->colname)));
                        if (restdef->identity)
                            ereport(ERROR,
                                    (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
                                     errmsg("column \"%s\" inherits from generated column but specifies identity",
                                            restdef->colname)));
                    }
                    else
                    {
                        if (restdef->generated)
                            ereport(ERROR,
                                    (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
                                     errmsg("child column \"%s\" specifies generation expression",
                                            restdef->colname),
                                     errhint("A child table column cannot be generated unless its parent column is.")));
                    }
 
                    if (coldef->generated && restdef->generated && coldef->generated != restdef->generated)
                        ereport(ERROR,
                                (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
                                 errmsg("column \"%s\" inherits from generated column of different kind",
                                        restdef->colname),
                                 errdetail("Parent column is %s, child column is %s.",
                                           coldef->generated == ATTRIBUTE_GENERATED_STORED ? "STORED" : "VIRTUAL",
                                           restdef->generated == ATTRIBUTE_GENERATED_STORED ? "STORED" : "VIRTUAL")));
 
                    /*
                     * Override the parent's default value for this column
                     * (coldef->cooked_default) with the partition's local
                     * definition (restdef->raw_default), if there's one. It
                     * should be physically impossible to get a cooked default
                     * in the local definition or a raw default in the
                     * inherited definition, but make sure they're nulls, for
                     * future-proofing.
                     */
                    Assert(restdef->cooked_default == NULL);
                    Assert(coldef->raw_default == NULL);
                    if (restdef->raw_default)
                    {
                        coldef->raw_default = restdef->raw_default;
                        coldef->cooked_default = NULL;
                    }
                }
            }
 
            /* complain for constraints on columns not in parent */
            if (!found)
                ereport(ERROR,
                        (errcode(ERRCODE_UNDEFINED_COLUMN),
                         errmsg("column \"%s\" does not exist",
                                restdef->colname)));
        }
    }
 
    /*
     * If we found any conflicting parent default values, check to make sure
     * they were overridden by the child.
     */
    if (have_bogus_defaults)
    {
        foreach(lc, columns)
        {
            ColumnDef  *def = lfirst(lc);
 
            if (def->cooked_default == &bogus_marker)
            {
                if (def->generated)
                    ereport(ERROR,
                            (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
                             errmsg("column \"%s\" inherits conflicting generation expressions",
                                    def->colname),
                             errhint("To resolve the conflict, specify a generation expression explicitly.")));
                else
                    ereport(ERROR,
                            (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
                             errmsg("column \"%s\" inherits conflicting default values",
                                    def->colname),
                             errhint("To resolve the conflict, specify a default explicitly.")));
            }
        }
    }
 
    *supconstr = constraints;
    *supnotnulls = nnconstraints;
 
    return columns;
}

References aclcheck_error(), ACLCHECK_NOT_OWNER, Assert, bms_add_member(), bms_is_member(), ConstrCheck::ccname, TupleConstr::check, CheckTableNotInUse(), ColumnDef::colname, CompressionMethodIsValid, CONSTR_NOTNULL, ColumnDef::cooked_default, elog, equal(), ereport, errcode(), errdetail(), errhint(), errmsg, ERROR, fb(), findAttrByName(), forboth, foreach_ptr, free_attrmap(), ColumnDef::generated, get_relkind_objtype(), GetCompressionMethodName(), GetUserId(), i, InvalidOid, IsConflictLogTableNamespace(), lappend(), lfirst, lfirst_node, lfirst_oid, list_delete_nth_cell(), list_length(), list_nth_node, make_attrmap(), makeColumnDef(), map_variable_attnos(), MaxHeapAttributeNumber, MergeCheckConstraint(), MergeChildAttribute(), MergeInheritedAttribute(), name, NameStr, NIL, NoLock, TupleConstr::num_check, object_ownercheck(), pstrdup(), ColumnDef::raw_default, RelationData::rd_rel, RELATION_IS_OTHER_TEMP, RelationGetDescr, RelationGetNotNullConstraints(), RelationGetRelationName, RelationGetRelid, stringToNode(), table_close(), table_open(), TupleDescAttr(), and TupleDescGetDefault().

Referenced by DefineRelation().

◆ MergeAttributesIntoExisting()

static void MergeAttributesIntoExisting	(	Relation	child_rel,
		Relation	parent_rel,
		bool	ispartition
	)

static

Definition at line 18208 of file tablecmds.c.

{
    Relation    attrrel;
    TupleDesc   parent_desc;
 
    attrrel = table_open(AttributeRelationId, RowExclusiveLock);
    parent_desc = RelationGetDescr(parent_rel);
 
    for (AttrNumber parent_attno = 1; parent_attno <= parent_desc->natts; parent_attno++)
    {
        Form_pg_attribute parent_att = TupleDescAttr(parent_desc, parent_attno - 1);
        char       *parent_attname = NameStr(parent_att->attname);
        HeapTuple   tuple;
 
        /* Ignore dropped columns in the parent. */
        if (parent_att->attisdropped)
            continue;
 
        /* Find same column in child (matching on column name). */
        tuple = SearchSysCacheCopyAttName(RelationGetRelid(child_rel), parent_attname);
        if (HeapTupleIsValid(tuple))
        {
            Form_pg_attribute child_att = (Form_pg_attribute) GETSTRUCT(tuple);
 
            if (parent_att->atttypid != child_att->atttypid ||
                parent_att->atttypmod != child_att->atttypmod)
                ereport(ERROR,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("child table \"%s\" has different type for column \"%s\"",
                                RelationGetRelationName(child_rel), parent_attname)));
 
            if (parent_att->attcollation != child_att->attcollation)
                ereport(ERROR,
                        (errcode(ERRCODE_COLLATION_MISMATCH),
                         errmsg("child table \"%s\" has different collation for column \"%s\"",
                                RelationGetRelationName(child_rel), parent_attname)));
 
            /*
             * If the parent has a not-null constraint that's not NO INHERIT,
             * make sure the child has one too.
             *
             * Other constraints are checked elsewhere.
             */
            if (parent_att->attnotnull && !child_att->attnotnull)
            {
                HeapTuple   contup;
 
                contup = findNotNullConstraintAttnum(RelationGetRelid(parent_rel),
                                                     parent_att->attnum);
                if (HeapTupleIsValid(contup) &&
                    !((Form_pg_constraint) GETSTRUCT(contup))->connoinherit)
                    ereport(ERROR,
                            errcode(ERRCODE_DATATYPE_MISMATCH),
                            errmsg("column \"%s\" in child table \"%s\" must be marked NOT NULL",
                                   parent_attname, RelationGetRelationName(child_rel)));
            }
 
            /*
             * Child column must be generated if and only if parent column is.
             */
            if (parent_att->attgenerated && !child_att->attgenerated)
                ereport(ERROR,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("column \"%s\" in child table must be a generated column", parent_attname)));
            if (child_att->attgenerated && !parent_att->attgenerated)
                ereport(ERROR,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("column \"%s\" in child table must not be a generated column", parent_attname)));
 
            if (parent_att->attgenerated && child_att->attgenerated && child_att->attgenerated != parent_att->attgenerated)
                ereport(ERROR,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("column \"%s\" inherits from generated column of different kind", parent_attname),
                         errdetail("Parent column is %s, child column is %s.",
                                   parent_att->attgenerated == ATTRIBUTE_GENERATED_STORED ? "STORED" : "VIRTUAL",
                                   child_att->attgenerated == ATTRIBUTE_GENERATED_STORED ? "STORED" : "VIRTUAL")));
 
            /*
             * Regular inheritance children are independent enough not to
             * inherit identity columns.  But partitions are integral part of
             * a partitioned table and inherit identity column.
             */
            if (ispartition)
                child_att->attidentity = parent_att->attidentity;
 
            /*
             * OK, bump the child column's inheritance count.  (If we fail
             * later on, this change will just roll back.)
             */
            if (pg_add_s16_overflow(child_att->attinhcount, 1,
                                    &child_att->attinhcount))
                ereport(ERROR,
                        errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
                        errmsg("too many inheritance parents"));
 
            /*
             * In case of partitions, we must enforce that value of attislocal
             * is same in all partitions. (Note: there are only inherited
             * attributes in partitions)
             */
            if (parent_rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
            {
                Assert(child_att->attinhcount == 1);
                child_att->attislocal = false;
            }
 
            CatalogTupleUpdate(attrrel, &tuple->t_self, tuple);
            heap_freetuple(tuple);
        }
        else
        {
            ereport(ERROR,
                    (errcode(ERRCODE_DATATYPE_MISMATCH),
                     errmsg("child table is missing column \"%s\"", parent_attname)));
        }
    }
 
    table_close(attrrel, RowExclusiveLock);
}

References Assert, CatalogTupleUpdate(), ereport, errcode(), errdetail(), errmsg, ERROR, fb(), findNotNullConstraintAttnum(), Form_pg_constraint, GETSTRUCT(), heap_freetuple(), HeapTupleIsValid, NameStr, pg_add_s16_overflow(), RelationGetDescr, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, SearchSysCacheCopyAttName(), HeapTupleData::t_self, table_close(), table_open(), and TupleDescAttr().

Referenced by CreateInheritance().

◆ MergeCheckConstraint()

static List * MergeCheckConstraint	(	List *	constraints,
		const char *	name,
		Node *	expr,
		bool	is_enforced
	)

static

Definition at line 3246 of file tablecmds.c.

{
    ListCell   *lc;
    CookedConstraint *newcon;
 
    foreach(lc, constraints)
    {
        CookedConstraint *ccon = (CookedConstraint *) lfirst(lc);
 
        Assert(ccon->contype == CONSTR_CHECK);
 
        /* Non-matching names never conflict */
        if (strcmp(ccon->name, name) != 0)
            continue;
 
        if (equal(expr, ccon->expr))
        {
            /* OK to merge constraint with existing */
            if (pg_add_s16_overflow(ccon->inhcount, 1,
                                    &ccon->inhcount))
                ereport(ERROR,
                        errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
                        errmsg("too many inheritance parents"));
 
            /*
             * When enforceability differs, the merged constraint should be
             * marked as ENFORCED because one of the parents is ENFORCED.
             */
            if (!ccon->is_enforced && is_enforced)
            {
                ccon->is_enforced = true;
                ccon->skip_validation = false;
            }
 
            return constraints;
        }
 
        ereport(ERROR,
                (errcode(ERRCODE_DUPLICATE_OBJECT),
                 errmsg("check constraint name \"%s\" appears multiple times but with different expressions",
                        name)));
    }
 
    /*
     * Constraint couldn't be merged with an existing one and also didn't
     * conflict with an existing one, so add it as a new one to the list.
     */
    newcon = palloc0_object(CookedConstraint);
    newcon->contype = CONSTR_CHECK;
    newcon->name = pstrdup(name);
    newcon->expr = expr;
    newcon->inhcount = 1;
    newcon->is_enforced = is_enforced;
    newcon->skip_validation = !is_enforced;
    return lappend(constraints, newcon);
}

References Assert, CONSTR_CHECK, equal(), ereport, errcode(), ERRCODE_DUPLICATE_OBJECT, errmsg, ERROR, fb(), lappend(), lfirst, name, palloc0_object, pg_add_s16_overflow(), and pstrdup().

Referenced by MergeAttributes().

◆ MergeChildAttribute()

static void MergeChildAttribute	(	List *	inh_columns,
		int	exist_attno,
		int	newcol_attno,
		const ColumnDef *	newdef
	)

static

Definition at line 3325 of file tablecmds.c.

{
    char       *attributeName = newdef->colname;
    ColumnDef  *inhdef;
    Oid         inhtypeid,
                newtypeid;
    int32       inhtypmod,
                newtypmod;
    Oid         inhcollid,
                newcollid;
 
    if (exist_attno == newcol_attno)
        ereport(NOTICE,
                (errmsg("merging column \"%s\" with inherited definition",
                        attributeName)));
    else
        ereport(NOTICE,
                (errmsg("moving and merging column \"%s\" with inherited definition", attributeName),
                 errdetail("User-specified column moved to the position of the inherited column.")));
 
    inhdef = list_nth_node(ColumnDef, inh_columns, exist_attno - 1);
 
    /*
     * Must have the same type and typmod
     */
    typenameTypeIdAndMod(NULL, inhdef->typeName, &inhtypeid, &inhtypmod);
    typenameTypeIdAndMod(NULL, newdef->typeName, &newtypeid, &newtypmod);
    if (inhtypeid != newtypeid || inhtypmod != newtypmod)
        ereport(ERROR,
                (errcode(ERRCODE_DATATYPE_MISMATCH),
                 errmsg("column \"%s\" has a type conflict",
                        attributeName),
                 errdetail("%s versus %s",
                           format_type_with_typemod(inhtypeid, inhtypmod),
                           format_type_with_typemod(newtypeid, newtypmod))));
 
    /*
     * Must have the same collation
     */
    inhcollid = GetColumnDefCollation(NULL, inhdef, inhtypeid);
    newcollid = GetColumnDefCollation(NULL, newdef, newtypeid);
    if (inhcollid != newcollid)
        ereport(ERROR,
                (errcode(ERRCODE_COLLATION_MISMATCH),
                 errmsg("column \"%s\" has a collation conflict",
                        attributeName),
                 errdetail("\"%s\" versus \"%s\"",
                           get_collation_name(inhcollid),
                           get_collation_name(newcollid))));
 
    /*
     * Identity is never inherited by a regular inheritance child. Pick
     * child's identity definition if there's one.
     */
    inhdef->identity = newdef->identity;
 
    /*
     * Copy storage parameter
     */
    if (inhdef->storage == 0)
        inhdef->storage = newdef->storage;
    else if (newdef->storage != 0 && inhdef->storage != newdef->storage)
        ereport(ERROR,
                (errcode(ERRCODE_DATATYPE_MISMATCH),
                 errmsg("column \"%s\" has a storage parameter conflict",
                        attributeName),
                 errdetail("%s versus %s",
                           storage_name(inhdef->storage),
                           storage_name(newdef->storage))));
 
    /*
     * Copy compression parameter
     */
    if (inhdef->compression == NULL)
        inhdef->compression = newdef->compression;
    else if (newdef->compression != NULL)
    {
        if (strcmp(inhdef->compression, newdef->compression) != 0)
            ereport(ERROR,
                    (errcode(ERRCODE_DATATYPE_MISMATCH),
                     errmsg("column \"%s\" has a compression method conflict",
                            attributeName),
                     errdetail("%s versus %s", inhdef->compression, newdef->compression)));
    }
 
    /*
     * Merge of not-null constraints = OR 'em together
     */
    inhdef->is_not_null |= newdef->is_not_null;
 
    /*
     * Check for conflicts related to generated columns.
     *
     * If the parent column is generated, the child column will be made a
     * generated column if it isn't already.  If it is a generated column,
     * we'll take its generation expression in preference to the parent's.  We
     * must check that the child column doesn't specify a default value or
     * identity, which matches the rules for a single column in
     * parse_utilcmd.c.
     *
     * Conversely, if the parent column is not generated, the child column
     * can't be either.  (We used to allow that, but it results in being able
     * to override the generation expression via UPDATEs through the parent.)
     */
    if (inhdef->generated)
    {
        if (newdef->raw_default && !newdef->generated)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
                     errmsg("column \"%s\" inherits from generated column but specifies default",
                            inhdef->colname)));
        if (newdef->identity)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
                     errmsg("column \"%s\" inherits from generated column but specifies identity",
                            inhdef->colname)));
    }
    else
    {
        if (newdef->generated)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
                     errmsg("child column \"%s\" specifies generation expression",
                            inhdef->colname),
                     errhint("A child table column cannot be generated unless its parent column is.")));
    }
 
    if (inhdef->generated && newdef->generated && newdef->generated != inhdef->generated)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_COLUMN_DEFINITION),
                 errmsg("column \"%s\" inherits from generated column of different kind",
                        inhdef->colname),
                 errdetail("Parent column is %s, child column is %s.",
                           inhdef->generated == ATTRIBUTE_GENERATED_STORED ? "STORED" : "VIRTUAL",
                           newdef->generated == ATTRIBUTE_GENERATED_STORED ? "STORED" : "VIRTUAL")));
 
    /*
     * If new def has a default, override previous default
     */
    if (newdef->raw_default != NULL)
    {
        inhdef->raw_default = newdef->raw_default;
        inhdef->cooked_default = newdef->cooked_default;
    }
 
    /* Mark the column as locally defined */
    inhdef->is_local = true;
}

References ereport, errcode(), errdetail(), errhint(), errmsg, ERROR, fb(), format_type_with_typemod(), get_collation_name(), GetColumnDefCollation(), list_nth_node, NOTICE, storage_name(), and typenameTypeIdAndMod().

Referenced by MergeAttributes().

◆ MergeConstraintsIntoExisting()

static void MergeConstraintsIntoExisting	(	Relation	child_rel,
		Relation	parent_rel
	)

static

Definition at line 18346 of file tablecmds.c.

{
    Relation    constraintrel;
    SysScanDesc parent_scan;
    ScanKeyData parent_key;
    HeapTuple   parent_tuple;
    Oid         parent_relid = RelationGetRelid(parent_rel);
    AttrMap    *attmap;
 
    constraintrel = table_open(ConstraintRelationId, RowExclusiveLock);
 
    /* Outer loop scans through the parent's constraint definitions */
    ScanKeyInit(&parent_key,
                Anum_pg_constraint_conrelid,
                BTEqualStrategyNumber, F_OIDEQ,
                ObjectIdGetDatum(parent_relid));
    parent_scan = systable_beginscan(constraintrel, ConstraintRelidTypidNameIndexId,
                                     true, NULL, 1, &parent_key);
 
    attmap = build_attrmap_by_name(RelationGetDescr(parent_rel),
                                   RelationGetDescr(child_rel),
                                   true);
 
    while (HeapTupleIsValid(parent_tuple = systable_getnext(parent_scan)))
    {
        Form_pg_constraint parent_con = (Form_pg_constraint) GETSTRUCT(parent_tuple);
        SysScanDesc child_scan;
        ScanKeyData child_key;
        HeapTuple   child_tuple;
        AttrNumber  parent_attno;
        bool        found = false;
 
        if (parent_con->contype != CONSTRAINT_CHECK &&
            parent_con->contype != CONSTRAINT_NOTNULL)
            continue;
 
        /* if the parent's constraint is marked NO INHERIT, it's not inherited */
        if (parent_con->connoinherit)
            continue;
 
        if (parent_con->contype == CONSTRAINT_NOTNULL)
            parent_attno = extractNotNullColumn(parent_tuple);
        else
            parent_attno = InvalidAttrNumber;
 
        /* Search for a child constraint matching this one */
        ScanKeyInit(&child_key,
                    Anum_pg_constraint_conrelid,
                    BTEqualStrategyNumber, F_OIDEQ,
                    ObjectIdGetDatum(RelationGetRelid(child_rel)));
        child_scan = systable_beginscan(constraintrel, ConstraintRelidTypidNameIndexId,
                                        true, NULL, 1, &child_key);
 
        while (HeapTupleIsValid(child_tuple = systable_getnext(child_scan)))
        {
            Form_pg_constraint child_con = (Form_pg_constraint) GETSTRUCT(child_tuple);
            HeapTuple   child_copy;
 
            if (child_con->contype != parent_con->contype)
                continue;
 
            /*
             * CHECK constraint are matched by constraint name, NOT NULL ones
             * by attribute number.
             */
            if (child_con->contype == CONSTRAINT_CHECK)
            {
                if (strcmp(NameStr(parent_con->conname),
                           NameStr(child_con->conname)) != 0)
                    continue;
            }
            else if (child_con->contype == CONSTRAINT_NOTNULL)
            {
                Form_pg_attribute parent_attr;
                Form_pg_attribute child_attr;
                AttrNumber  child_attno;
 
                parent_attr = TupleDescAttr(parent_rel->rd_att, parent_attno - 1);
                child_attno = extractNotNullColumn(child_tuple);
                if (parent_attno != attmap->attnums[child_attno - 1])
                    continue;
 
                child_attr = TupleDescAttr(child_rel->rd_att, child_attno - 1);
                /* there shouldn't be constraints on dropped columns */
                if (parent_attr->attisdropped || child_attr->attisdropped)
                    elog(ERROR, "found not-null constraint on dropped columns");
            }
 
            if (child_con->contype == CONSTRAINT_CHECK &&
                !constraints_equivalent(parent_tuple, child_tuple, RelationGetDescr(constraintrel)))
                ereport(ERROR,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("child table \"%s\" has different definition for check constraint \"%s\"",
                                RelationGetRelationName(child_rel), NameStr(parent_con->conname))));
 
            /*
             * If the child constraint is "no inherit" then cannot merge
             */
            if (child_con->connoinherit)
                ereport(ERROR,
                        (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                         errmsg("constraint \"%s\" conflicts with non-inherited constraint on child table \"%s\"",
                                NameStr(child_con->conname), RelationGetRelationName(child_rel))));
 
            /*
             * If the child constraint is "not valid" then cannot merge with a
             * valid parent constraint
             */
            if (parent_con->convalidated && child_con->conenforced &&
                !child_con->convalidated)
                ereport(ERROR,
                        (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                         errmsg("constraint \"%s\" conflicts with NOT VALID constraint on child table \"%s\"",
                                NameStr(child_con->conname), RelationGetRelationName(child_rel))));
 
            /*
             * A NOT ENFORCED child constraint cannot be merged with an
             * ENFORCED parent constraint. However, the reverse is allowed,
             * where the child constraint is ENFORCED.
             */
            if (parent_con->conenforced && !child_con->conenforced)
                ereport(ERROR,
                        (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
                         errmsg("constraint \"%s\" conflicts with NOT ENFORCED constraint on child table \"%s\"",
                                NameStr(child_con->conname), RelationGetRelationName(child_rel))));
 
            /*
             * OK, bump the child constraint's inheritance count.  (If we fail
             * later on, this change will just roll back.)
             */
            child_copy = heap_copytuple(child_tuple);
            child_con = (Form_pg_constraint) GETSTRUCT(child_copy);
 
            if (pg_add_s16_overflow(child_con->coninhcount, 1,
                                    &child_con->coninhcount))
                ereport(ERROR,
                        errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
                        errmsg("too many inheritance parents"));
 
            /*
             * In case of partitions, an inherited constraint must be
             * inherited only once since it cannot have multiple parents and
             * it is never considered local.
             */
            if (parent_rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
            {
                Assert(child_con->coninhcount == 1);
                child_con->conislocal = false;
            }
 
            CatalogTupleUpdate(constraintrel, &child_copy->t_self, child_copy);
            heap_freetuple(child_copy);
 
            found = true;
            break;
        }
 
        systable_endscan(child_scan);
 
        if (!found)
        {
            if (parent_con->contype == CONSTRAINT_NOTNULL)
                ereport(ERROR,
                        errcode(ERRCODE_DATATYPE_MISMATCH),
                        errmsg("column \"%s\" in child table \"%s\" must be marked NOT NULL",
                               get_attname(parent_relid,
                                           extractNotNullColumn(parent_tuple),
                                           false),
                               RelationGetRelationName(child_rel)));
 
            ereport(ERROR,
                    (errcode(ERRCODE_DATATYPE_MISMATCH),
                     errmsg("child table is missing constraint \"%s\"",
                            NameStr(parent_con->conname))));
        }
    }
 
    systable_endscan(parent_scan);
    table_close(constraintrel, RowExclusiveLock);
}

References Assert, BTEqualStrategyNumber, build_attrmap_by_name(), CatalogTupleUpdate(), constraints_equivalent(), elog, ereport, errcode(), errmsg, ERROR, extractNotNullColumn(), fb(), Form_pg_constraint, get_attname(), GETSTRUCT(), heap_copytuple(), heap_freetuple(), HeapTupleIsValid, InvalidAttrNumber, NameStr, ObjectIdGetDatum(), pg_add_s16_overflow(), RelationGetDescr, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), table_open(), and TupleDescAttr().

Referenced by CreateInheritance().

◆ MergeInheritedAttribute()

static ColumnDef * MergeInheritedAttribute	(	List *	inh_columns,
		int	exist_attno,
		const ColumnDef *	newdef
	)

static

Definition at line 3495 of file tablecmds.c.

{
    char       *attributeName = newdef->colname;
    ColumnDef  *prevdef;
    Oid         prevtypeid,
                newtypeid;
    int32       prevtypmod,
                newtypmod;
    Oid         prevcollid,
                newcollid;
 
    ereport(NOTICE,
            (errmsg("merging multiple inherited definitions of column \"%s\"",
                    attributeName)));
    prevdef = list_nth_node(ColumnDef, inh_columns, exist_attno - 1);
 
    /*
     * Must have the same type and typmod
     */
    typenameTypeIdAndMod(NULL, prevdef->typeName, &prevtypeid, &prevtypmod);
    typenameTypeIdAndMod(NULL, newdef->typeName, &newtypeid, &newtypmod);
    if (prevtypeid != newtypeid || prevtypmod != newtypmod)
        ereport(ERROR,
                (errcode(ERRCODE_DATATYPE_MISMATCH),
                 errmsg("inherited column \"%s\" has a type conflict",
                        attributeName),
                 errdetail("%s versus %s",
                           format_type_with_typemod(prevtypeid, prevtypmod),
                           format_type_with_typemod(newtypeid, newtypmod))));
 
    /*
     * Must have the same collation
     */
    prevcollid = GetColumnDefCollation(NULL, prevdef, prevtypeid);
    newcollid = GetColumnDefCollation(NULL, newdef, newtypeid);
    if (prevcollid != newcollid)
        ereport(ERROR,
                (errcode(ERRCODE_COLLATION_MISMATCH),
                 errmsg("inherited column \"%s\" has a collation conflict",
                        attributeName),
                 errdetail("\"%s\" versus \"%s\"",
                           get_collation_name(prevcollid),
                           get_collation_name(newcollid))));
 
    /*
     * Copy/check storage parameter
     */
    if (prevdef->storage == 0)
        prevdef->storage = newdef->storage;
    else if (prevdef->storage != newdef->storage)
        ereport(ERROR,
                (errcode(ERRCODE_DATATYPE_MISMATCH),
                 errmsg("inherited column \"%s\" has a storage parameter conflict",
                        attributeName),
                 errdetail("%s versus %s",
                           storage_name(prevdef->storage),
                           storage_name(newdef->storage))));
 
    /*
     * Copy/check compression parameter
     */
    if (prevdef->compression == NULL)
        prevdef->compression = newdef->compression;
    else if (newdef->compression != NULL)
    {
        if (strcmp(prevdef->compression, newdef->compression) != 0)
            ereport(ERROR,
                    (errcode(ERRCODE_DATATYPE_MISMATCH),
                     errmsg("column \"%s\" has a compression method conflict",
                            attributeName),
                     errdetail("%s versus %s",
                               prevdef->compression, newdef->compression)));
    }
 
    /*
     * Check for GENERATED conflicts
     */
    if (prevdef->generated != newdef->generated)
        ereport(ERROR,
                (errcode(ERRCODE_DATATYPE_MISMATCH),
                 errmsg("inherited column \"%s\" has a generation conflict",
                        attributeName)));
 
    /*
     * Default and other constraints are handled by the caller.
     */
 
    if (pg_add_s16_overflow(prevdef->inhcount, 1,
                            &prevdef->inhcount))
        ereport(ERROR,
                errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
                errmsg("too many inheritance parents"));
 
    return prevdef;
}

References ColumnDef::colname, ereport, errcode(), errdetail(), errmsg, ERROR, fb(), format_type_with_typemod(), get_collation_name(), GetColumnDefCollation(), list_nth_node, NOTICE, pg_add_s16_overflow(), storage_name(), and typenameTypeIdAndMod().

Referenced by MergeAttributes().

◆ MergePartitionsMoveRows()

static void MergePartitionsMoveRows	(	List **	wqueue,
		List *	mergingPartitions,
		Relation	newPartRel
	)

static

Definition at line 23362 of file tablecmds.c.

{
    CommandId   mycid;
    EState     *estate;
    AlteredTableInfo *tab;
    ListCell   *ltab;
 
    /* The FSM is empty, so don't bother using it. */
    uint32      ti_options = TABLE_INSERT_SKIP_FSM;
    BulkInsertState bistate;    /* state of bulk inserts for partition */
    TupleTableSlot *dstslot;
 
    /* Find the work queue entry for the new partition table: newPartRel. */
    tab = ATGetQueueEntry(wqueue, newPartRel);
 
    /* Generate the constraint and default execution states. */
    estate = CreateExecutorState();
 
    buildExpressionExecutionStates(tab, newPartRel, estate);
 
    mycid = GetCurrentCommandId(true);
 
    /* Prepare a BulkInsertState for table_tuple_insert. */
    bistate = GetBulkInsertState();
 
    /* Create the necessary tuple slot. */
    dstslot = table_slot_create(newPartRel, NULL);
 
    foreach_oid(merging_oid, mergingPartitions)
    {
        ExprContext *econtext;
        TupleTableSlot *srcslot;
        TupleConversionMap *tuple_map;
        TableScanDesc scan;
        MemoryContext oldCxt;
        Snapshot    snapshot;
        Relation    mergingPartition;
 
        econtext = GetPerTupleExprContext(estate);
 
        /*
         * Partition is already locked in the transformPartitionCmdForMerge
         * function.
         */
        mergingPartition = table_open(merging_oid, NoLock);
 
        /* Create a source tuple slot for the partition being merged. */
        srcslot = table_slot_create(mergingPartition, NULL);
 
        /*
         * Map computing for moving attributes of the merged partition to the
         * new partition.
         */
        tuple_map = convert_tuples_by_name(RelationGetDescr(mergingPartition),
                                           RelationGetDescr(newPartRel));
 
        /* Scan through the rows. */
        snapshot = RegisterSnapshot(GetLatestSnapshot());
        scan = table_beginscan(mergingPartition, snapshot, 0, NULL,
                               SO_NONE);
 
        /*
         * Switch to per-tuple memory context and reset it for each tuple
         * produced, so we don't leak memory.
         */
        oldCxt = MemoryContextSwitchTo(GetPerTupleMemoryContext(estate));
 
        while (table_scan_getnextslot(scan, ForwardScanDirection, srcslot))
        {
            TupleTableSlot *insertslot;
 
            CHECK_FOR_INTERRUPTS();
 
            if (tuple_map)
            {
                /* Need to use a map to copy attributes. */
                insertslot = execute_attr_map_slot(tuple_map->attrMap, srcslot, dstslot);
            }
            else
            {
                slot_getallattrs(srcslot);
 
                /* Copy attributes directly. */
                insertslot = dstslot;
 
                ExecClearTuple(insertslot);
 
                memcpy(insertslot->tts_values, srcslot->tts_values,
                       sizeof(Datum) * srcslot->tts_nvalid);
                memcpy(insertslot->tts_isnull, srcslot->tts_isnull,
                       sizeof(bool) * srcslot->tts_nvalid);
 
                ExecStoreVirtualTuple(insertslot);
            }
 
            /*
             * Constraints and GENERATED expressions might reference the
             * tableoid column, so fill tts_tableOid with the desired value.
             * (We must do this each time, because it gets overwritten with
             * newrel's OID during storing.)
             */
            insertslot->tts_tableOid = RelationGetRelid(newPartRel);
 
            /*
             * Now, evaluate any generated expressions whose inputs come from
             * the new tuple.  We assume these columns won't reference each
             * other, so that there's no ordering dependency.
             */
            evaluateGeneratedExpressionsAndCheckConstraints(tab, newPartRel,
                                                            insertslot, econtext);
 
            /* Write the tuple out to the new relation. */
            table_tuple_insert(newPartRel, insertslot, mycid,
                               ti_options, bistate);
 
            ResetExprContext(econtext);
        }
 
        MemoryContextSwitchTo(oldCxt);
        table_endscan(scan);
        UnregisterSnapshot(snapshot);
 
        if (tuple_map)
            free_conversion_map(tuple_map);
 
        ExecDropSingleTupleTableSlot(srcslot);
        table_close(mergingPartition, NoLock);
    }
 
    FreeExecutorState(estate);
    ExecDropSingleTupleTableSlot(dstslot);
    FreeBulkInsertState(bistate);
 
    table_finish_bulk_insert(newPartRel, ti_options);
 
    /*
     * We don't need to process this newPartRel since we already processed it
     * here, so delete the ALTER TABLE queue for it.
     */
    foreach(ltab, *wqueue)
    {
        tab = (AlteredTableInfo *) lfirst(ltab);
        if (tab->relid == RelationGetRelid(newPartRel))
        {
            *wqueue = list_delete_cell(*wqueue, ltab);
            break;
        }
    }
}

References ATGetQueueEntry(), buildExpressionExecutionStates(), CHECK_FOR_INTERRUPTS, convert_tuples_by_name(), CreateExecutorState(), evaluateGeneratedExpressionsAndCheckConstraints(), ExecClearTuple(), ExecDropSingleTupleTableSlot(), ExecStoreVirtualTuple(), execute_attr_map_slot(), fb(), foreach_oid, ForwardScanDirection, free_conversion_map(), FreeBulkInsertState(), FreeExecutorState(), GetBulkInsertState(), GetCurrentCommandId(), GetLatestSnapshot(), GetPerTupleExprContext, GetPerTupleMemoryContext, lfirst, list_delete_cell(), memcpy(), MemoryContextSwitchTo(), NoLock, RegisterSnapshot(), RelationGetDescr, RelationGetRelid, AlteredTableInfo::relid, ResetExprContext, slot_getallattrs(), SO_NONE, table_beginscan(), table_close(), table_endscan(), table_finish_bulk_insert(), TABLE_INSERT_SKIP_FSM, table_open(), table_scan_getnextslot(), table_slot_create(), table_tuple_insert(), and UnregisterSnapshot().

Referenced by ATExecMergePartitions().

◆ NotNullImpliedByRelConstraints()

static bool NotNullImpliedByRelConstraints	(	Relation	rel,
		Form_pg_attribute	attr
	)

static

Definition at line 8230 of file tablecmds.c.

{
    NullTest   *nnulltest = makeNode(NullTest);
 
    nnulltest->arg = (Expr *) makeVar(1,
                                      attr->attnum,
                                      attr->atttypid,
                                      attr->atttypmod,
                                      attr->attcollation,
                                      0);
    nnulltest->nulltesttype = IS_NOT_NULL;
 
    /*
     * argisrow = false is correct even for a composite column, because
     * attnotnull does not represent a SQL-spec IS NOT NULL test in such a
     * case, just IS DISTINCT FROM NULL.
     */
    nnulltest->argisrow = false;
    nnulltest->location = -1;
 
    if (ConstraintImpliedByRelConstraint(rel, list_make1(nnulltest), NIL))
    {
        ereport(DEBUG1,
                (errmsg_internal("existing constraints on column \"%s.%s\" are sufficient to prove that it does not contain nulls",
                                 RelationGetRelationName(rel), NameStr(attr->attname))));
        return true;
    }
 
    return false;
}

References ConstraintImpliedByRelConstraint(), DEBUG1, ereport, errmsg_internal(), fb(), IS_NOT_NULL, list_make1, Var::location, makeNode, makeVar(), NameStr, NIL, and RelationGetRelationName.

Referenced by set_attnotnull().

◆ PartConstraintImpliedByRelConstraint()

bool PartConstraintImpliedByRelConstraint	(	Relation	scanrel,
		List *	partConstraint
	)

Definition at line 20791 of file tablecmds.c.

{
    List       *existConstraint = NIL;
    TupleConstr *constr = RelationGetDescr(scanrel)->constr;
    int         i;
 
    if (constr && constr->has_not_null)
    {
        int         natts = scanrel->rd_att->natts;
 
        for (i = 1; i <= natts; i++)
        {
            CompactAttribute *att = TupleDescCompactAttr(scanrel->rd_att, i - 1);
 
            /* invalid not-null constraint must be ignored here */
            if (att->attnullability == ATTNULLABLE_VALID && !att->attisdropped)
            {
                Form_pg_attribute wholeatt = TupleDescAttr(scanrel->rd_att, i - 1);
                NullTest   *ntest = makeNode(NullTest);
 
                ntest->arg = (Expr *) makeVar(1,
                                              i,
                                              wholeatt->atttypid,
                                              wholeatt->atttypmod,
                                              wholeatt->attcollation,
                                              0);
                ntest->nulltesttype = IS_NOT_NULL;
 
                /*
                 * argisrow=false is correct even for a composite column,
                 * because attnotnull does not represent a SQL-spec IS NOT
                 * NULL test in such a case, just IS DISTINCT FROM NULL.
                 */
                ntest->argisrow = false;
                ntest->location = -1;
                existConstraint = lappend(existConstraint, ntest);
            }
        }
    }
 
    return ConstraintImpliedByRelConstraint(scanrel, partConstraint, existConstraint);
}

References CompactAttribute::attisdropped, CompactAttribute::attnullability, ATTNULLABLE_VALID, ConstraintImpliedByRelConstraint(), fb(), TupleConstr::has_not_null, i, IS_NOT_NULL, lappend(), Var::location, makeNode, makeVar(), TupleDescData::natts, NIL, RelationData::rd_att, RelationGetDescr, TupleDescAttr(), and TupleDescCompactAttr().

Referenced by check_default_partition_contents(), and QueuePartitionConstraintValidation().

◆ PreCommit_on_commit_actions()

void PreCommit_on_commit_actions ( void )

Definition at line 20023 of file tablecmds.c.

{
    ListCell   *l;
    List       *oids_to_truncate = NIL;
    List       *oids_to_drop = NIL;
 
    foreach(l, on_commits)
    {
        OnCommitItem *oc = (OnCommitItem *) lfirst(l);
 
        /* Ignore entry if already dropped in this xact */
        if (oc->deleting_subid != InvalidSubTransactionId)
            continue;
 
        switch (oc->oncommit)
        {
            case ONCOMMIT_NOOP:
            case ONCOMMIT_PRESERVE_ROWS:
                /* Do nothing (there shouldn't be such entries, actually) */
                break;
            case ONCOMMIT_DELETE_ROWS:
 
                /*
                 * If this transaction hasn't accessed any temporary
                 * relations, we can skip truncating ON COMMIT DELETE ROWS
                 * tables, as they must still be empty.
                 */
                if ((MyXactFlags & XACT_FLAGS_ACCESSEDTEMPNAMESPACE))
                    oids_to_truncate = lappend_oid(oids_to_truncate, oc->relid);
                break;
            case ONCOMMIT_DROP:
                oids_to_drop = lappend_oid(oids_to_drop, oc->relid);
                break;
        }
    }
 
    /*
     * Truncate relations before dropping so that all dependencies between
     * relations are removed after they are worked on.  Doing it like this
     * might be a waste as it is possible that a relation being truncated will
     * be dropped anyway due to its parent being dropped, but this makes the
     * code more robust because of not having to re-check that the relation
     * exists at truncation time.
     */
    if (oids_to_truncate != NIL)
        heap_truncate(oids_to_truncate);
 
    if (oids_to_drop != NIL)
    {
        ObjectAddresses *targetObjects = new_object_addresses();
 
        foreach(l, oids_to_drop)
        {
            ObjectAddress object;
 
            object.classId = RelationRelationId;
            object.objectId = lfirst_oid(l);
            object.objectSubId = 0;
 
            Assert(!object_address_present(&object, targetObjects));
 
            add_exact_object_address(&object, targetObjects);
        }
 
        /*
         * Object deletion might involve toast table access (to clean up
         * toasted catalog entries), so ensure we have a valid snapshot.
         */
        PushActiveSnapshot(GetTransactionSnapshot());
 
        /*
         * Since this is an automatic drop, rather than one directly initiated
         * by the user, we pass the PERFORM_DELETION_INTERNAL flag.
         */
        performMultipleDeletions(targetObjects, DROP_CASCADE,
                                 PERFORM_DELETION_INTERNAL | PERFORM_DELETION_QUIETLY);
 
        PopActiveSnapshot();
 
#ifdef USE_ASSERT_CHECKING
 
        /*
         * Note that table deletion will call remove_on_commit_action, so the
         * entry should get marked as deleted.
         */
        foreach(l, on_commits)
        {
            OnCommitItem *oc = (OnCommitItem *) lfirst(l);
 
            if (oc->oncommit != ONCOMMIT_DROP)
                continue;
 
            Assert(oc->deleting_subid != InvalidSubTransactionId);
        }
#endif
    }
}

References add_exact_object_address(), Assert, ObjectAddress::classId, DROP_CASCADE, fb(), GetTransactionSnapshot(), heap_truncate(), InvalidSubTransactionId, lappend_oid(), lfirst, lfirst_oid, MyXactFlags, new_object_addresses(), NIL, object_address_present(), on_commits, ONCOMMIT_DELETE_ROWS, ONCOMMIT_DROP, ONCOMMIT_NOOP, ONCOMMIT_PRESERVE_ROWS, PERFORM_DELETION_INTERNAL, PERFORM_DELETION_QUIETLY, performMultipleDeletions(), PopActiveSnapshot(), PushActiveSnapshot(), and XACT_FLAGS_ACCESSEDTEMPNAMESPACE.

Referenced by CommitTransaction(), and PrepareTransaction().

◆ QueueCheckConstraintValidation()

static void QueueCheckConstraintValidation	(	List **	wqueue,
		Relation	conrel,
		Relation	rel,
		char *	constrName,
		HeapTuple	contuple,
		bool	recurse,
		bool	recursing,
		LOCKMODE	lockmode
	)

static

Definition at line 13654 of file tablecmds.c.

{
    Form_pg_constraint con;
    AlteredTableInfo *tab;
    HeapTuple   copyTuple;
    Form_pg_constraint copy_con;
 
    List       *children = NIL;
    ListCell   *child;
    NewConstraint *newcon;
    Datum       val;
    char       *conbin;
 
    con = (Form_pg_constraint) GETSTRUCT(contuple);
    Assert(con->contype == CONSTRAINT_CHECK);
 
    /*
     * If we're recursing, the parent has already done this, so skip it. Also,
     * if the constraint is a NO INHERIT constraint, we shouldn't try to look
     * for it in the children.
     */
    if (!recursing && !con->connoinherit)
        children = find_all_inheritors(RelationGetRelid(rel),
                                       lockmode, NULL);
 
    /*
     * For CHECK constraints, we must ensure that we only mark the constraint
     * as validated on the parent if it's already validated on the children.
     *
     * We recurse before validating on the parent, to reduce risk of
     * deadlocks.
     */
    foreach(child, children)
    {
        Oid         childoid = lfirst_oid(child);
        Relation    childrel;
 
        if (childoid == RelationGetRelid(rel))
            continue;
 
        /*
         * If we are told not to recurse, there had better not be any child
         * tables, because we can't mark the constraint on the parent valid
         * unless it is valid for all child tables.
         */
        if (!recurse)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                     errmsg("constraint must be validated on child tables too")));
 
        /* find_all_inheritors already got lock */
        childrel = table_open(childoid, NoLock);
 
        ATExecValidateConstraint(wqueue, childrel, constrName, false,
                                 true, lockmode);
        table_close(childrel, NoLock);
    }
 
    /* Queue validation for phase 3 */
    newcon = palloc0_object(NewConstraint);
    newcon->name = constrName;
    newcon->contype = CONSTR_CHECK;
    newcon->refrelid = InvalidOid;
    newcon->refindid = InvalidOid;
    newcon->conid = con->oid;
 
    val = SysCacheGetAttrNotNull(CONSTROID, contuple,
                                 Anum_pg_constraint_conbin);
    conbin = TextDatumGetCString(val);
    newcon->qual = expand_generated_columns_in_expr(stringToNode(conbin), rel, 1);
 
    /* Find or create work queue entry for this table */
    tab = ATGetQueueEntry(wqueue, rel);
    tab->constraints = lappend(tab->constraints, newcon);
 
    /*
     * Invalidate relcache so that others see the new validated constraint.
     */
    CacheInvalidateRelcache(rel);
 
    /*
     * Now update the catalog, while we have the door open.
     */
    copyTuple = heap_copytuple(contuple);
    copy_con = (Form_pg_constraint) GETSTRUCT(copyTuple);
    copy_con->convalidated = true;
    CatalogTupleUpdate(conrel, &copyTuple->t_self, copyTuple);
 
    InvokeObjectPostAlterHook(ConstraintRelationId, con->oid, 0);
 
    heap_freetuple(copyTuple);
}

References Assert, ATExecValidateConstraint(), ATGetQueueEntry(), CacheInvalidateRelcache(), CatalogTupleUpdate(), CONSTR_CHECK, AlteredTableInfo::constraints, ereport, errcode(), errmsg, ERROR, expand_generated_columns_in_expr(), fb(), find_all_inheritors(), Form_pg_constraint, GETSTRUCT(), heap_copytuple(), heap_freetuple(), InvalidOid, InvokeObjectPostAlterHook, lappend(), lfirst_oid, NIL, NoLock, palloc0_object, RelationGetRelid, stringToNode(), SysCacheGetAttrNotNull(), table_close(), table_open(), TextDatumGetCString, and val.

Referenced by ATExecValidateConstraint().

◆ QueueFKConstraintValidation()

static void QueueFKConstraintValidation	(	List **	wqueue,
		Relation	conrel,
		Relation	fkrel,
		Oid	pkrelid,
		HeapTuple	contuple,
		LOCKMODE	lockmode
	)

static

Definition at line 13532 of file tablecmds.c.

{
    Form_pg_constraint con;
    AlteredTableInfo *tab;
    HeapTuple   copyTuple;
    Form_pg_constraint copy_con;
 
    /* since this function recurses, it could be driven to stack overflow */
    check_stack_depth();
 
    con = (Form_pg_constraint) GETSTRUCT(contuple);
    Assert(con->contype == CONSTRAINT_FOREIGN);
    Assert(!con->convalidated);
 
    /*
     * Add the validation to phase 3's queue; not needed for partitioned
     * tables themselves, only for their partitions.
     *
     * When the referenced table (pkrelid) is partitioned, the referencing
     * table (fkrel) has one pg_constraint row pointing to each partition
     * thereof.  These rows are there only to support action triggers and no
     * table scan is needed, therefore skip this for them as well.
     */
    if (fkrel->rd_rel->relkind == RELKIND_RELATION &&
        con->confrelid == pkrelid)
    {
        NewConstraint *newcon;
        Constraint *fkconstraint;
 
        /* Queue validation for phase 3 */
        fkconstraint = makeNode(Constraint);
        /* for now this is all we need */
        fkconstraint->conname = pstrdup(NameStr(con->conname));
 
        newcon = palloc0_object(NewConstraint);
        newcon->name = fkconstraint->conname;
        newcon->contype = CONSTR_FOREIGN;
        newcon->refrelid = con->confrelid;
        newcon->refindid = con->conindid;
        newcon->conid = con->oid;
        newcon->qual = (Node *) fkconstraint;
 
        /* Find or create work queue entry for this table */
        tab = ATGetQueueEntry(wqueue, fkrel);
        tab->constraints = lappend(tab->constraints, newcon);
    }
 
    /*
     * If the table at either end of the constraint is partitioned, we need to
     * recurse and handle every unvalidated constraint that is a child of this
     * constraint.
     */
    if (fkrel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE ||
        get_rel_relkind(con->confrelid) == RELKIND_PARTITIONED_TABLE)
    {
        ScanKeyData pkey;
        SysScanDesc pscan;
        HeapTuple   childtup;
 
        ScanKeyInit(&pkey,
                    Anum_pg_constraint_conparentid,
                    BTEqualStrategyNumber, F_OIDEQ,
                    ObjectIdGetDatum(con->oid));
 
        pscan = systable_beginscan(conrel, ConstraintParentIndexId,
                                   true, NULL, 1, &pkey);
 
        while (HeapTupleIsValid(childtup = systable_getnext(pscan)))
        {
            Form_pg_constraint childcon;
            Relation    childrel;
 
            childcon = (Form_pg_constraint) GETSTRUCT(childtup);
 
            /*
             * If the child constraint has already been validated, no further
             * action is required for it or its descendants, as they are all
             * valid.
             */
            if (childcon->convalidated)
                continue;
 
            childrel = table_open(childcon->conrelid, lockmode);
 
            /*
             * NB: Note that pkrelid should be passed as-is during recursion,
             * as it is required to identify the root referenced table.
             */
            QueueFKConstraintValidation(wqueue, conrel, childrel, pkrelid,
                                        childtup, lockmode);
            table_close(childrel, NoLock);
        }
 
        systable_endscan(pscan);
    }
 
    /*
     * Now mark the pg_constraint row as validated (even if we didn't check,
     * notably the ones for partitions on the referenced side).
     *
     * We rely on transaction abort to roll back this change if phase 3
     * ultimately finds violating rows.  This is a bit ugly.
     */
    copyTuple = heap_copytuple(contuple);
    copy_con = (Form_pg_constraint) GETSTRUCT(copyTuple);
    copy_con->convalidated = true;
    CatalogTupleUpdate(conrel, &copyTuple->t_self, copyTuple);
 
    InvokeObjectPostAlterHook(ConstraintRelationId, con->oid, 0);
 
    heap_freetuple(copyTuple);
}

References Assert, ATGetQueueEntry(), BTEqualStrategyNumber, CatalogTupleUpdate(), check_stack_depth(), CONSTR_FOREIGN, AlteredTableInfo::constraints, fb(), Form_pg_constraint, get_rel_relkind(), GETSTRUCT(), heap_copytuple(), heap_freetuple(), HeapTupleIsValid, InvokeObjectPostAlterHook, lappend(), makeNode, NameStr, NoLock, ObjectIdGetDatum(), palloc0_object, pstrdup(), QueueFKConstraintValidation(), ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by ATExecValidateConstraint(), AttachPartitionForeignKey(), and QueueFKConstraintValidation().

◆ QueueNNConstraintValidation()

static void QueueNNConstraintValidation	(	List **	wqueue,
		Relation	conrel,
		Relation	rel,
		HeapTuple	contuple,
		bool	recurse,
		bool	recursing,
		LOCKMODE	lockmode
	)

static

Definition at line 13757 of file tablecmds.c.

{
    Form_pg_constraint con;
    AlteredTableInfo *tab;
    HeapTuple   copyTuple;
    Form_pg_constraint copy_con;
    List       *children = NIL;
    AttrNumber  attnum;
    char       *colname;
 
    con = (Form_pg_constraint) GETSTRUCT(contuple);
    Assert(con->contype == CONSTRAINT_NOTNULL);
 
    attnum = extractNotNullColumn(contuple);
 
    /*
     * If we're recursing, we've already done this for parent, so skip it.
     * Also, if the constraint is a NO INHERIT constraint, we shouldn't try to
     * look for it in the children.
     *
     * We recurse before validating on the parent, to reduce risk of
     * deadlocks.
     */
    if (!recursing && !con->connoinherit)
        children = find_all_inheritors(RelationGetRelid(rel), lockmode, NULL);
 
    colname = get_attname(RelationGetRelid(rel), attnum, false);
    foreach_oid(childoid, children)
    {
        Relation    childrel;
        HeapTuple   contup;
        Form_pg_constraint childcon;
        char       *conname;
 
        if (childoid == RelationGetRelid(rel))
            continue;
 
        /*
         * If we are told not to recurse, there had better not be any child
         * tables, because we can't mark the constraint on the parent valid
         * unless it is valid for all child tables.
         */
        if (!recurse)
            ereport(ERROR,
                    errcode(ERRCODE_INVALID_TABLE_DEFINITION),
                    errmsg("constraint must be validated on child tables too"));
 
        /*
         * The column on child might have a different attnum, so search by
         * column name.
         */
        contup = findNotNullConstraint(childoid, colname);
        if (!contup)
            elog(ERROR, "cache lookup failed for not-null constraint on column \"%s\" of relation \"%s\"",
                 colname, get_rel_name(childoid));
        childcon = (Form_pg_constraint) GETSTRUCT(contup);
        if (childcon->convalidated)
            continue;
 
        /* find_all_inheritors already got lock */
        childrel = table_open(childoid, NoLock);
        conname = pstrdup(NameStr(childcon->conname));
 
        /* XXX improve ATExecValidateConstraint API to avoid double search */
        ATExecValidateConstraint(wqueue, childrel, conname,
                                 false, true, lockmode);
        table_close(childrel, NoLock);
    }
 
    /* Set attnotnull appropriately without queueing another validation */
    set_attnotnull(NULL, rel, attnum, true, false);
 
    tab = ATGetQueueEntry(wqueue, rel);
    tab->verify_new_notnull = true;
 
    /*
     * Invalidate relcache so that others see the new validated constraint.
     */
    CacheInvalidateRelcache(rel);
 
    /*
     * Now update the catalogs, while we have the door open.
     */
    copyTuple = heap_copytuple(contuple);
    copy_con = (Form_pg_constraint) GETSTRUCT(copyTuple);
    copy_con->convalidated = true;
    CatalogTupleUpdate(conrel, &copyTuple->t_self, copyTuple);
 
    InvokeObjectPostAlterHook(ConstraintRelationId, con->oid, 0);
 
    heap_freetuple(copyTuple);
}

References Assert, ATExecValidateConstraint(), ATGetQueueEntry(), attnum, CacheInvalidateRelcache(), CatalogTupleUpdate(), elog, ereport, errcode(), errmsg, ERROR, extractNotNullColumn(), fb(), find_all_inheritors(), findNotNullConstraint(), foreach_oid, Form_pg_constraint, get_attname(), get_rel_name(), GETSTRUCT(), heap_copytuple(), heap_freetuple(), InvokeObjectPostAlterHook, NameStr, NIL, NoLock, pstrdup(), RelationGetRelid, set_attnotnull(), table_close(), table_open(), and AlteredTableInfo::verify_new_notnull.

Referenced by ATExecValidateConstraint().

◆ QueuePartitionConstraintValidation()

static void QueuePartitionConstraintValidation	(	List **	wqueue,
		Relation	scanrel,
		List *	partConstraint,
		bool	validate_default
	)

static

Definition at line 20909 of file tablecmds.c.

{
    /*
     * Based on the table's existing constraints, determine whether or not we
     * may skip scanning the table.
     */
    if (PartConstraintImpliedByRelConstraint(scanrel, partConstraint))
    {
        if (!validate_default)
            ereport(DEBUG1,
                    (errmsg_internal("partition constraint for table \"%s\" is implied by existing constraints",
                                     RelationGetRelationName(scanrel))));
        else
            ereport(DEBUG1,
                    (errmsg_internal("updated partition constraint for default partition \"%s\" is implied by existing constraints",
                                     RelationGetRelationName(scanrel))));
        return;
    }
 
    /*
     * Constraints proved insufficient. For plain relations, queue a
     * validation item now; for partitioned tables, recurse to process each
     * partition.
     */
    if (scanrel->rd_rel->relkind == RELKIND_RELATION)
    {
        AlteredTableInfo *tab;
 
        /* Grab a work queue entry. */
        tab = ATGetQueueEntry(wqueue, scanrel);
        Assert(tab->partition_constraint == NULL);
        tab->partition_constraint = (Expr *) linitial(partConstraint);
        tab->validate_default = validate_default;
    }
    else if (scanrel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
    {
        PartitionDesc partdesc = RelationGetPartitionDesc(scanrel, true);
        int         i;
 
        for (i = 0; i < partdesc->nparts; i++)
        {
            Relation    part_rel;
            List       *thisPartConstraint;
 
            /*
             * This is the minimum lock we need to prevent deadlocks.
             */
            part_rel = table_open(partdesc->oids[i], AccessExclusiveLock);
 
            /*
             * Adjust the constraint for scanrel so that it matches this
             * partition's attribute numbers.
             */
            thisPartConstraint =
                map_partition_varattnos(partConstraint, 1,
                                        part_rel, scanrel);
 
            QueuePartitionConstraintValidation(wqueue, part_rel,
                                               thisPartConstraint,
                                               validate_default);
            table_close(part_rel, NoLock);  /* keep lock till commit */
        }
    }
}

References AccessExclusiveLock, Assert, ATGetQueueEntry(), DEBUG1, ereport, errmsg_internal(), fb(), i, linitial, map_partition_varattnos(), NoLock, PartitionDescData::nparts, PartitionDescData::oids, PartConstraintImpliedByRelConstraint(), AlteredTableInfo::partition_constraint, QueuePartitionConstraintValidation(), RelationData::rd_rel, RelationGetPartitionDesc(), RelationGetRelationName, table_close(), table_open(), and AlteredTableInfo::validate_default.

Referenced by ATExecAttachPartition(), and QueuePartitionConstraintValidation().

◆ RangeVarCallbackForAlterRelation()

static void RangeVarCallbackForAlterRelation	(	const RangeVar *	rv,
		Oid	relid,
		Oid	oldrelid,
		void *	arg
	)

static

Definition at line 20301 of file tablecmds.c.

{
    Node       *stmt = (Node *) arg;
    ObjectType  reltype;
    HeapTuple   tuple;
    Form_pg_class classform;
    AclResult   aclresult;
    char        relkind;
 
    tuple = SearchSysCache1(RELOID, ObjectIdGetDatum(relid));
    if (!HeapTupleIsValid(tuple))
        return;                 /* concurrently dropped */
    classform = (Form_pg_class) GETSTRUCT(tuple);
    relkind = classform->relkind;
 
    /* Must own relation. */
    if (!object_ownercheck(RelationRelationId, relid, GetUserId()))
        aclcheck_error(ACLCHECK_NOT_OWNER, get_relkind_objtype(get_rel_relkind(relid)), rv->relname);
 
    /*
     * Conflict log tables are used internally for logical replication
     * conflict logging and should not be altered directly, as it could
     * disrupt conflict logging.
     */
    if (IsConflictLogTableClass(classform))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("cannot alter conflict log table \"%s\"",
                        rv->relname),
                 errdetail("Conflict log tables are system-managed tables for logical replication conflicts.")));
 
    /* No system table modifications unless explicitly allowed. */
    if (!allowSystemTableMods && IsSystemClass(relid, classform))
        ereport(ERROR,
                (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
                 errmsg("permission denied: \"%s\" is a system catalog",
                        rv->relname)));
 
    /*
     * Extract the specified relation type from the statement parse tree.
     *
     * Also, for ALTER .. RENAME, check permissions: the user must (still)
     * have CREATE rights on the containing namespace.
     */
    if (IsA(stmt, RenameStmt))
    {
        aclresult = object_aclcheck(NamespaceRelationId, classform->relnamespace,
                                    GetUserId(), ACL_CREATE);
        if (aclresult != ACLCHECK_OK)
            aclcheck_error(aclresult, OBJECT_SCHEMA,
                           get_namespace_name(classform->relnamespace));
        reltype = ((RenameStmt *) stmt)->renameType;
    }
    else if (IsA(stmt, AlterObjectSchemaStmt))
        reltype = ((AlterObjectSchemaStmt *) stmt)->objectType;
 
    else if (IsA(stmt, AlterTableStmt))
        reltype = ((AlterTableStmt *) stmt)->objtype;
    else
    {
        elog(ERROR, "unrecognized node type: %d", (int) nodeTag(stmt));
        reltype = OBJECT_TABLE; /* placate compiler */
    }
 
    /*
     * For compatibility with prior releases, we allow ALTER TABLE to be used
     * with most other types of relations (but not composite types). We allow
     * similar flexibility for ALTER INDEX in the case of RENAME, but not
     * otherwise.  Otherwise, the user must select the correct form of the
     * command for the relation at issue.
     */
    if (reltype == OBJECT_SEQUENCE && relkind != RELKIND_SEQUENCE)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("\"%s\" is not a sequence", rv->relname)));
 
    if (reltype == OBJECT_VIEW && relkind != RELKIND_VIEW)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("\"%s\" is not a view", rv->relname)));
 
    if (reltype == OBJECT_MATVIEW && relkind != RELKIND_MATVIEW)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("\"%s\" is not a materialized view", rv->relname)));
 
    if (reltype == OBJECT_FOREIGN_TABLE && relkind != RELKIND_FOREIGN_TABLE)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("\"%s\" is not a foreign table", rv->relname)));
 
    if (reltype == OBJECT_TYPE && relkind != RELKIND_COMPOSITE_TYPE)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("\"%s\" is not a composite type", rv->relname)));
 
    if (reltype == OBJECT_PROPGRAPH && relkind != RELKIND_PROPGRAPH)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("\"%s\" is not a property graph", rv->relname)));
 
    if (reltype == OBJECT_INDEX && relkind != RELKIND_INDEX &&
        relkind != RELKIND_PARTITIONED_INDEX
        && !IsA(stmt, RenameStmt))
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("\"%s\" is not an index", rv->relname)));
 
    /*
     * Don't allow ALTER TABLE on composite types. We want people to use ALTER
     * TYPE for that.
     */
    if (reltype != OBJECT_TYPE && relkind == RELKIND_COMPOSITE_TYPE)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("\"%s\" is a composite type", rv->relname),
        /* translator: %s is an SQL ALTER command */
                 errhint("Use %s instead.",
                         "ALTER TYPE")));
 
    /*
     * Don't allow ALTER TABLE .. SET SCHEMA on relations that can't be moved
     * to a different schema, such as indexes and TOAST tables.

Data Structures

Macros

Typedefs

Enumerations

Functions

Variables

Macro Definition Documentation

◆ AT_NUM_PASSES

◆ ATT_COMPOSITE_TYPE

◆ ATT_FOREIGN_TABLE

◆ ATT_INDEX

◆ ATT_MATVIEW

◆ ATT_PARTITIONED_INDEX

◆ ATT_PARTITIONED_TABLE

◆ ATT_SEQUENCE

◆ ATT_TABLE

◆ ATT_VIEW

◆ child_dependency_type

Typedef Documentation

◆ addFkConstraintSides

◆ AlteredTableInfo

◆ AlterTablePass

◆ ForeignTruncateInfo

◆ NewColumnValue

◆ NewConstraint

◆ OnCommitItem

◆ PartitionIndexExtDepEntry

◆ SplitPartitionContext

Enumeration Type Documentation

◆ addFkConstraintSides

◆ AlterTablePass

Function Documentation

◆ add_column_collation_dependency()

◆ add_column_datatype_dependency()

◆ addFkConstraint()

◆ addFkRecurseReferenced()

◆ addFkRecurseReferencing()

◆ alter_table_type_to_string()

◆ AlterCheckConstrEnforceabilityRecurse()

◆ AlterConstrDeferrabilityRecurse()

◆ AlterConstrTriggerDeferrability()

◆ AlterConstrUpdateConstraintEntry()

◆ AlterFKConstrEnforceabilityRecurse()

◆ AlterIndexNamespaces()

◆ AlterRelationNamespaceInternal()

◆ AlterSeqNamespaces()

◆ AlterTable()

◆ AlterTableGetLockLevel()

◆ AlterTableInternal()

◆ AlterTableLookupRelation()

◆ AlterTableMoveAll()

◆ AlterTableNamespace()

◆ AlterTableNamespaceInternal()

◆ applyPartitionIndexExtDeps()

◆ ATAddCheckNNConstraint()

◆ ATAddForeignKeyConstraint()

◆ ATCheckCheckConstrHasEnforcedParent()

◆ ATCheckPartitionsNotInUse()

◆ ATColumnChangeRequiresRewrite()

◆ ATController()

◆ ATDetachCheckNoForeignKeyRefs()

◆ AtEOSubXact_on_commit_actions()

◆ AtEOXact_on_commit_actions()

◆ ATExecAddColumn()

◆ ATExecAddConstraint()

◆ ATExecAddIdentity()

◆ ATExecAddIndex()

◆ ATExecAddIndexConstraint()

◆ ATExecAddInherit()

◆ ATExecAddOf()

◆ ATExecAddStatistics()

◆ ATExecAlterCheckConstrEnforceability()

◆ ATExecAlterColumnGenericOptions()

◆ ATExecAlterColumnType()

◆ ATExecAlterConstraint()

◆ ATExecAlterConstraintInternal()

◆ ATExecAlterConstrDeferrability()

◆ ATExecAlterConstrInheritability()

◆ ATExecAlterFKConstrEnforceability()

◆ ATExecAttachPartition()