PostgreSQL Source Code  git master
amapi.c
Go to the documentation of this file.
1 /*-------------------------------------------------------------------------
2  *
3  * amapi.c
4  * Support routines for API for Postgres index access methods.
5  *
6  * Copyright (c) 2015-2024, PostgreSQL Global Development Group
7  *
8  *
9  * IDENTIFICATION
10  * src/backend/access/index/amapi.c
11  *
12  *-------------------------------------------------------------------------
13  */
14 #include "postgres.h"
15 
16 #include "access/amapi.h"
17 #include "access/htup_details.h"
18 #include "catalog/pg_am.h"
19 #include "catalog/pg_opclass.h"
20 #include "utils/fmgrprotos.h"
21 #include "utils/syscache.h"
22 
23 
24 /*
25  * GetIndexAmRoutine - call the specified access method handler routine to get
26  * its IndexAmRoutine struct, which will be palloc'd in the caller's context.
27  *
28  * Note that if the amhandler function is built-in, this will not involve
29  * any catalog access. It's therefore safe to use this while bootstrapping
30  * indexes for the system catalogs. relcache.c relies on that.
31  */
34 {
35  Datum datum;
36  IndexAmRoutine *routine;
37 
38  datum = OidFunctionCall0(amhandler);
39  routine = (IndexAmRoutine *) DatumGetPointer(datum);
40 
41  if (routine == NULL || !IsA(routine, IndexAmRoutine))
42  elog(ERROR, "index access method handler function %u did not return an IndexAmRoutine struct",
43  amhandler);
44 
45  return routine;
46 }
47 
48 /*
49  * GetIndexAmRoutineByAmId - look up the handler of the index access method
50  * with the given OID, and get its IndexAmRoutine struct.
51  *
52  * If the given OID isn't a valid index access method, returns NULL if
53  * noerror is true, else throws error.
54  */
56 GetIndexAmRoutineByAmId(Oid amoid, bool noerror)
57 {
58  HeapTuple tuple;
59  Form_pg_am amform;
60  regproc amhandler;
61 
62  /* Get handler function OID for the access method */
63  tuple = SearchSysCache1(AMOID, ObjectIdGetDatum(amoid));
64  if (!HeapTupleIsValid(tuple))
65  {
66  if (noerror)
67  return NULL;
68  elog(ERROR, "cache lookup failed for access method %u",
69  amoid);
70  }
71  amform = (Form_pg_am) GETSTRUCT(tuple);
72 
73  /* Check if it's an index access method as opposed to some other AM */
74  if (amform->amtype != AMTYPE_INDEX)
75  {
76  if (noerror)
77  {
78  ReleaseSysCache(tuple);
79  return NULL;
80  }
81  ereport(ERROR,
82  (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
83  errmsg("access method \"%s\" is not of type %s",
84  NameStr(amform->amname), "INDEX")));
85  }
86 
87  amhandler = amform->amhandler;
88 
89  /* Complain if handler OID is invalid */
90  if (!RegProcedureIsValid(amhandler))
91  {
92  if (noerror)
93  {
94  ReleaseSysCache(tuple);
95  return NULL;
96  }
97  ereport(ERROR,
98  (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
99  errmsg("index access method \"%s\" does not have a handler",
100  NameStr(amform->amname))));
101  }
102 
103  ReleaseSysCache(tuple);
104 
105  /* And finally, call the handler function to get the API struct. */
106  return GetIndexAmRoutine(amhandler);
107 }
108 
109 
110 /*
111  * Ask appropriate access method to validate the specified opclass.
112  */
113 Datum
115 {
116  Oid opclassoid = PG_GETARG_OID(0);
117  bool result;
118  HeapTuple classtup;
119  Form_pg_opclass classform;
120  Oid amoid;
121  IndexAmRoutine *amroutine;
122 
123  classtup = SearchSysCache1(CLAOID, ObjectIdGetDatum(opclassoid));
124  if (!HeapTupleIsValid(classtup))
125  elog(ERROR, "cache lookup failed for operator class %u", opclassoid);
126  classform = (Form_pg_opclass) GETSTRUCT(classtup);
127 
128  amoid = classform->opcmethod;
129 
130  ReleaseSysCache(classtup);
131 
132  amroutine = GetIndexAmRoutineByAmId(amoid, false);
133 
134  if (amroutine->amvalidate == NULL)
135  elog(ERROR, "function amvalidate is not defined for index access method %u",
136  amoid);
137 
138  result = amroutine->amvalidate(opclassoid);
139 
140  pfree(amroutine);
141 
142  PG_RETURN_BOOL(result);
143 }
IndexAmRoutine * GetIndexAmRoutineByAmId(Oid amoid, bool noerror)
Definition: amapi.c:56
IndexAmRoutine * GetIndexAmRoutine(Oid amhandler)
Definition: amapi.c:33
Datum amvalidate(PG_FUNCTION_ARGS)
Definition: amapi.c:114
#define NameStr(name)
Definition: c.h:733
#define RegProcedureIsValid(p)
Definition: c.h:764
Oid regproc
Definition: c.h:636
int errcode(int sqlerrcode)
Definition: elog.c:859
int errmsg(const char *fmt,...)
Definition: elog.c:1072
#define ERROR
Definition: elog.h:39
#define elog(elevel,...)
Definition: elog.h:224
#define ereport(elevel,...)
Definition: elog.h:149
#define PG_GETARG_OID(n)
Definition: fmgr.h:275
#define OidFunctionCall0(functionId)
Definition: fmgr.h:678
#define PG_FUNCTION_ARGS
Definition: fmgr.h:193
#define PG_RETURN_BOOL(x)
Definition: fmgr.h:359
#define HeapTupleIsValid(tuple)
Definition: htup.h:78
#define GETSTRUCT(TUP)
Definition: htup_details.h:653
void pfree(void *pointer)
Definition: mcxt.c:1508
#define IsA(nodeptr, _type_)
Definition: nodes.h:158
FormData_pg_am * Form_pg_am
Definition: pg_am.h:48
FormData_pg_opclass * Form_pg_opclass
Definition: pg_opclass.h:83
uintptr_t Datum
Definition: postgres.h:64
static Datum ObjectIdGetDatum(Oid X)
Definition: postgres.h:252
static Pointer DatumGetPointer(Datum X)
Definition: postgres.h:312
unsigned int Oid
Definition: postgres_ext.h:31
amvalidate_function amvalidate
Definition: amapi.h:277
void ReleaseSysCache(HeapTuple tuple)
Definition: syscache.c:266
HeapTuple SearchSysCache1(int cacheId, Datum key1)
Definition: syscache.c:218