amcmds.c File Reference

#include "postgres.h"
#include "access/htup_details.h"
#include "access/table.h"
#include "catalog/catalog.h"
#include "catalog/dependency.h"
#include "catalog/indexing.h"
#include "catalog/objectaccess.h"
#include "catalog/pg_am.h"
#include "catalog/pg_proc.h"
#include "catalog/pg_type.h"
#include "commands/defrem.h"
#include "miscadmin.h"
#include "parser/parse_func.h"
#include "utils/builtins.h"
#include "utils/lsyscache.h"
#include "utils/rel.h"
#include "utils/syscache.h"

Include dependency graph for amcmds.c:

Go to the source code of this file.

Functions
static Oid	lookup_am_handler_func (List *handler_name, char amtype)
static const char *	get_am_type_string (char amtype)
ObjectAddress	CreateAccessMethod (CreateAmStmt *stmt)
static Oid	get_am_type_oid (const char *amname, char amtype, bool missing_ok)
Oid	get_index_am_oid (const char *amname, bool missing_ok)
Oid	get_table_am_oid (const char *amname, bool missing_ok)
Oid	get_am_oid (const char *amname, bool missing_ok)
char *	get_am_name (Oid amOid)

Function Documentation

CreateAccessMethod()

ObjectAddress CreateAccessMethod

(

CreateAmStmt *

stmt

)

Definition at line 43 of file amcmds.c.

{
    Relation    rel;
    ObjectAddress myself;
    ObjectAddress referenced;
    Oid         amoid;
    Oid         amhandler;
    bool        nulls[Natts_pg_am];
    Datum       values[Natts_pg_am];
    HeapTuple   tup;
 
    rel = table_open(AccessMethodRelationId, RowExclusiveLock);
 
    /* Must be superuser */
    if (!superuser())
        ereport(ERROR,
                (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
                 errmsg("permission denied to create access method \"%s\"",
                        stmt->amname),
                 errhint("Must be superuser to create an access method.")));
 
    /* Check if name is used */
    amoid = GetSysCacheOid1(AMNAME, Anum_pg_am_oid,
                            CStringGetDatum(stmt->amname));
    if (OidIsValid(amoid))
    {
        ereport(ERROR,
                (errcode(ERRCODE_DUPLICATE_OBJECT),
                 errmsg("access method \"%s\" already exists",
                        stmt->amname)));
    }
 
    /*
     * Get the handler function oid, verifying the AM type while at it.
     */
    amhandler = lookup_am_handler_func(stmt->handler_name, stmt->amtype);
 
    /*
     * Insert tuple into pg_am.
     */
    memset(values, 0, sizeof(values));
    memset(nulls, false, sizeof(nulls));
 
    amoid = GetNewOidWithIndex(rel, AmOidIndexId, Anum_pg_am_oid);
    values[Anum_pg_am_oid - 1] = ObjectIdGetDatum(amoid);
    values[Anum_pg_am_amname - 1] =
        DirectFunctionCall1(namein, CStringGetDatum(stmt->amname));
    values[Anum_pg_am_amhandler - 1] = ObjectIdGetDatum(amhandler);
    values[Anum_pg_am_amtype - 1] = CharGetDatum(stmt->amtype);
 
    tup = heap_form_tuple(RelationGetDescr(rel), values, nulls);
 
    CatalogTupleInsert(rel, tup);
    heap_freetuple(tup);
 
    myself.classId = AccessMethodRelationId;
    myself.objectId = amoid;
    myself.objectSubId = 0;
 
    /* Record dependency on handler function */
    referenced.classId = ProcedureRelationId;
    referenced.objectId = amhandler;
    referenced.objectSubId = 0;
 
    recordDependencyOn(&myself, &referenced, DEPENDENCY_NORMAL);
 
    recordDependencyOnCurrentExtension(&myself, false);
 
    InvokeObjectPostCreateHook(AccessMethodRelationId, amoid, 0);
 
    table_close(rel, RowExclusiveLock);
 
    return myself;
}

References CatalogTupleInsert(), CharGetDatum(), ObjectAddress::classId, CStringGetDatum(), DEPENDENCY_NORMAL, DirectFunctionCall1, ereport, errcode(), ERRCODE_DUPLICATE_OBJECT, errhint(), errmsg(), ERROR, GetNewOidWithIndex(), GetSysCacheOid1, heap_form_tuple(), heap_freetuple(), InvokeObjectPostCreateHook, lookup_am_handler_func(), namein(), ObjectAddress::objectId, ObjectIdGetDatum(), ObjectAddress::objectSubId, OidIsValid, recordDependencyOn(), recordDependencyOnCurrentExtension(), RelationGetDescr, RowExclusiveLock, stmt, superuser(), table_close(), table_open(), and values.

Referenced by ProcessUtilitySlow().

get_am_name()

char * get_am_name

(

Oid

amOid

)

Definition at line 192 of file amcmds.c.

{
    HeapTuple   tup;
    char       *result = NULL;
 
    tup = SearchSysCache1(AMOID, ObjectIdGetDatum(amOid));
    if (HeapTupleIsValid(tup))
    {
        Form_pg_am  amform = (Form_pg_am) GETSTRUCT(tup);
 
        result = pstrdup(NameStr(amform->amname));
        ReleaseSysCache(tup);
    }
    return result;
}

References GETSTRUCT(), HeapTupleIsValid, NameStr, ObjectIdGetDatum(), pstrdup(), ReleaseSysCache(), and SearchSysCache1().

Referenced by assignOperTypes(), ATAddForeignKeyConstraint(), DefineIndex(), getObjectIdentityParts(), GetOperatorFromCompareType(), IsThereOpClassInNamespace(), and IsThereOpFamilyInNamespace().

get_am_oid()

Oid get_am_oid	(	const char *	amname,
		bool	missing_ok
	)

Definition at line 183 of file amcmds.c.

{
    return get_am_type_oid(amname, '\0', missing_ok);
}

References get_am_type_oid().

Referenced by get_object_address_unqualified().

get_am_type_oid()

static Oid get_am_type_oid ( const char *  amname, char  amtype, bool  missing_ok  )

static

Definition at line 129 of file amcmds.c.

{
    HeapTuple   tup;
    Oid         oid = InvalidOid;
 
    tup = SearchSysCache1(AMNAME, CStringGetDatum(amname));
    if (HeapTupleIsValid(tup))
    {
        Form_pg_am  amform = (Form_pg_am) GETSTRUCT(tup);
 
        if (amtype != '\0' &&
            amform->amtype != amtype)
            ereport(ERROR,
                    (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                     errmsg("access method \"%s\" is not of type %s",
                            NameStr(amform->amname),
                            get_am_type_string(amtype))));
 
        oid = amform->oid;
        ReleaseSysCache(tup);
    }
 
    if (!OidIsValid(oid) && !missing_ok)
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_OBJECT),
                 errmsg("access method \"%s\" does not exist", amname)));
    return oid;
}

References CStringGetDatum(), ereport, errcode(), errmsg(), ERROR, get_am_type_string(), GETSTRUCT(), HeapTupleIsValid, InvalidOid, NameStr, OidIsValid, ReleaseSysCache(), and SearchSysCache1().

Referenced by get_am_oid(), get_index_am_oid(), and get_table_am_oid().

get_am_type_string()

static const char * get_am_type_string ( char  amtype )

static

Definition at line 212 of file amcmds.c.

{
    switch (amtype)
    {
        case AMTYPE_INDEX:
            return "INDEX";
        case AMTYPE_TABLE:
            return "TABLE";
        default:
            /* shouldn't happen */
            elog(ERROR, "invalid access method type '%c'", amtype);
            return NULL;        /* keep compiler quiet */
    }
}

References elog, and ERROR.

Referenced by get_am_type_oid().

get_index_am_oid()

Oid get_index_am_oid	(	const char *	amname,
		bool	missing_ok
	)

Definition at line 163 of file amcmds.c.

{
    return get_am_type_oid(amname, AMTYPE_INDEX, missing_ok);
}

References get_am_type_oid().

Referenced by DefineOpFamily(), get_object_address_opcf(), and transformIndexConstraint().

get_table_am_oid()

Oid get_table_am_oid	(	const char *	amname,
		bool	missing_ok
	)

Definition at line 173 of file amcmds.c.

{
    return get_am_type_oid(amname, AMTYPE_TABLE, missing_ok);
}

References get_am_type_oid().

Referenced by ATPrepSetAccessMethod(), check_default_table_access_method(), and DefineRelation().

lookup_am_handler_func()

static Oid lookup_am_handler_func ( List *  handler_name, char  amtype  )

static

Definition at line 234 of file amcmds.c.

{
    Oid         handlerOid;
    Oid         funcargtypes[1] = {INTERNALOID};
    Oid         expectedType = InvalidOid;
 
    if (handler_name == NIL)
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_FUNCTION),
                 errmsg("handler function is not specified")));
 
    /* handlers have one argument of type internal */
    handlerOid = LookupFuncName(handler_name, 1, funcargtypes, false);
 
    /* check that handler has the correct return type */
    switch (amtype)
    {
        case AMTYPE_INDEX:
            expectedType = INDEX_AM_HANDLEROID;
            break;
        case AMTYPE_TABLE:
            expectedType = TABLE_AM_HANDLEROID;
            break;
        default:
            elog(ERROR, "unrecognized access method type \"%c\"", amtype);
    }
 
    if (get_func_rettype(handlerOid) != expectedType)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("function %s must return type %s",
                        get_func_name(handlerOid),
                        format_type_extended(expectedType, -1, 0))));
 
    return handlerOid;
}

References elog, ereport, errcode(), errmsg(), ERROR, format_type_extended(), get_func_name(), get_func_rettype(), InvalidOid, LookupFuncName(), and NIL.

Referenced by CreateAccessMethod().