/src/postgres/src/include/funcapi.h
Line | Count | Source (jump to first uncovered line) |
1 | | /*------------------------------------------------------------------------- |
2 | | * |
3 | | * funcapi.h |
4 | | * Definitions for functions which return composite type and/or sets |
5 | | * or work on VARIADIC inputs. |
6 | | * |
7 | | * This file must be included by all Postgres modules that either define |
8 | | * or call FUNCAPI-callable functions or macros. |
9 | | * |
10 | | * |
11 | | * Copyright (c) 2002-2025, PostgreSQL Global Development Group |
12 | | * |
13 | | * src/include/funcapi.h |
14 | | * |
15 | | *------------------------------------------------------------------------- |
16 | | */ |
17 | | #ifndef FUNCAPI_H |
18 | | #define FUNCAPI_H |
19 | | |
20 | | #include "access/tupdesc.h" |
21 | | #include "executor/executor.h" |
22 | | #include "executor/tuptable.h" |
23 | | #include "fmgr.h" |
24 | | |
25 | | /*------------------------------------------------------------------------- |
26 | | * Support to ease writing Functions returning composite types |
27 | | *------------------------------------------------------------------------- |
28 | | * |
29 | | * This struct holds arrays of individual attribute information |
30 | | * needed to create a tuple from raw C strings. It also requires |
31 | | * a copy of the TupleDesc. The information carried here |
32 | | * is derived from the TupleDesc, but it is stored here to |
33 | | * avoid redundant cpu cycles on each call to an SRF. |
34 | | */ |
35 | | typedef struct AttInMetadata |
36 | | { |
37 | | /* full TupleDesc */ |
38 | | TupleDesc tupdesc; |
39 | | |
40 | | /* array of attribute type input function finfo */ |
41 | | FmgrInfo *attinfuncs; |
42 | | |
43 | | /* array of attribute type i/o parameter OIDs */ |
44 | | Oid *attioparams; |
45 | | |
46 | | /* array of attribute typmod */ |
47 | | int32 *atttypmods; |
48 | | } AttInMetadata; |
49 | | |
50 | | /*------------------------------------------------------------------------- |
51 | | * Support struct to ease writing Set Returning Functions (SRFs) |
52 | | *------------------------------------------------------------------------- |
53 | | * |
54 | | * This struct holds function context for Set Returning Functions. |
55 | | * Use fn_extra to hold a pointer to it across calls |
56 | | */ |
57 | | typedef struct FuncCallContext |
58 | | { |
59 | | /* |
60 | | * Number of times we've been called before |
61 | | * |
62 | | * call_cntr is initialized to 0 for you by SRF_FIRSTCALL_INIT(), and |
63 | | * incremented for you every time SRF_RETURN_NEXT() is called. |
64 | | */ |
65 | | uint64 call_cntr; |
66 | | |
67 | | /* |
68 | | * OPTIONAL maximum number of calls |
69 | | * |
70 | | * max_calls is here for convenience only and setting it is optional. If |
71 | | * not set, you must provide alternative means to know when the function |
72 | | * is done. |
73 | | */ |
74 | | uint64 max_calls; |
75 | | |
76 | | /* |
77 | | * OPTIONAL pointer to miscellaneous user-provided context information |
78 | | * |
79 | | * user_fctx is for use as a pointer to your own struct to retain |
80 | | * arbitrary context information between calls of your function. |
81 | | */ |
82 | | void *user_fctx; |
83 | | |
84 | | /* |
85 | | * OPTIONAL pointer to struct containing attribute type input metadata |
86 | | * |
87 | | * attinmeta is for use when returning tuples (i.e. composite data types) |
88 | | * and is not used when returning base data types. It is only needed if |
89 | | * you intend to use BuildTupleFromCStrings() to create the return tuple. |
90 | | */ |
91 | | AttInMetadata *attinmeta; |
92 | | |
93 | | /* |
94 | | * memory context used for structures that must live for multiple calls |
95 | | * |
96 | | * multi_call_memory_ctx is set by SRF_FIRSTCALL_INIT() for you, and used |
97 | | * by SRF_RETURN_DONE() for cleanup. It is the most appropriate memory |
98 | | * context for any memory that is to be reused across multiple calls of |
99 | | * the SRF. |
100 | | */ |
101 | | MemoryContext multi_call_memory_ctx; |
102 | | |
103 | | /* |
104 | | * OPTIONAL pointer to struct containing tuple description |
105 | | * |
106 | | * tuple_desc is for use when returning tuples (i.e. composite data types) |
107 | | * and is only needed if you are going to build the tuples with |
108 | | * heap_form_tuple() rather than with BuildTupleFromCStrings(). Note that |
109 | | * the TupleDesc pointer stored here should usually have been run through |
110 | | * BlessTupleDesc() first. |
111 | | */ |
112 | | TupleDesc tuple_desc; |
113 | | |
114 | | } FuncCallContext; |
115 | | |
116 | | /*---------- |
117 | | * Support to ease writing functions returning composite types |
118 | | * |
119 | | * External declarations: |
120 | | * get_call_result_type: |
121 | | * Given a function's call info record, determine the kind of datatype |
122 | | * it is supposed to return. If resultTypeId isn't NULL, *resultTypeId |
123 | | * receives the actual datatype OID (this is mainly useful for scalar |
124 | | * result types). If resultTupleDesc isn't NULL, *resultTupleDesc |
125 | | * receives a pointer to a TupleDesc when the result is of a composite |
126 | | * type, or NULL when it's a scalar result or the rowtype could not be |
127 | | * determined. NB: the tupledesc should be copied if it is to be |
128 | | * accessed over a long period. |
129 | | * get_expr_result_type: |
130 | | * Given an expression node, return the same info as for |
131 | | * get_call_result_type. Note: the cases in which rowtypes cannot be |
132 | | * determined are different from the cases for get_call_result_type. |
133 | | * get_func_result_type: |
134 | | * Given only a function's OID, return the same info as for |
135 | | * get_call_result_type. Note: the cases in which rowtypes cannot be |
136 | | * determined are different from the cases for get_call_result_type. |
137 | | * Do *not* use this if you can use one of the others. |
138 | | * |
139 | | * See also get_expr_result_tupdesc(), which is a convenient wrapper around |
140 | | * get_expr_result_type() for use when the caller only cares about |
141 | | * determinable-rowtype cases. |
142 | | *---------- |
143 | | */ |
144 | | |
145 | | /* Type categories for get_call_result_type and siblings */ |
146 | | typedef enum TypeFuncClass |
147 | | { |
148 | | TYPEFUNC_SCALAR, /* scalar result type */ |
149 | | TYPEFUNC_COMPOSITE, /* determinable rowtype result */ |
150 | | TYPEFUNC_COMPOSITE_DOMAIN, /* domain over determinable rowtype result */ |
151 | | TYPEFUNC_RECORD, /* indeterminate rowtype result */ |
152 | | TYPEFUNC_OTHER, /* bogus type, eg pseudotype */ |
153 | | } TypeFuncClass; |
154 | | |
155 | | extern TypeFuncClass get_call_result_type(FunctionCallInfo fcinfo, |
156 | | Oid *resultTypeId, |
157 | | TupleDesc *resultTupleDesc); |
158 | | extern TypeFuncClass get_expr_result_type(Node *expr, |
159 | | Oid *resultTypeId, |
160 | | TupleDesc *resultTupleDesc); |
161 | | extern TypeFuncClass get_func_result_type(Oid functionId, |
162 | | Oid *resultTypeId, |
163 | | TupleDesc *resultTupleDesc); |
164 | | |
165 | | extern TupleDesc get_expr_result_tupdesc(Node *expr, bool noError); |
166 | | |
167 | | extern bool resolve_polymorphic_argtypes(int numargs, Oid *argtypes, |
168 | | char *argmodes, |
169 | | Node *call_expr); |
170 | | |
171 | | extern int get_func_arg_info(HeapTuple procTup, |
172 | | Oid **p_argtypes, char ***p_argnames, |
173 | | char **p_argmodes); |
174 | | |
175 | | extern int get_func_input_arg_names(Datum proargnames, Datum proargmodes, |
176 | | char ***arg_names); |
177 | | |
178 | | extern int get_func_trftypes(HeapTuple procTup, Oid **p_trftypes); |
179 | | extern char *get_func_result_name(Oid functionId); |
180 | | |
181 | | extern TupleDesc build_function_result_tupdesc_d(char prokind, |
182 | | Datum proallargtypes, |
183 | | Datum proargmodes, |
184 | | Datum proargnames); |
185 | | extern TupleDesc build_function_result_tupdesc_t(HeapTuple procTuple); |
186 | | |
187 | | |
188 | | /*---------- |
189 | | * Support to ease writing functions returning composite types |
190 | | * |
191 | | * External declarations: |
192 | | * TupleDesc BlessTupleDesc(TupleDesc tupdesc) - "Bless" a completed tuple |
193 | | * descriptor so that it can be used to return properly labeled tuples. |
194 | | * You need to call this if you are going to use heap_form_tuple directly. |
195 | | * TupleDescGetAttInMetadata does it for you, however, so no need to call |
196 | | * it if you call TupleDescGetAttInMetadata. |
197 | | * AttInMetadata *TupleDescGetAttInMetadata(TupleDesc tupdesc) - Build an |
198 | | * AttInMetadata struct based on the given TupleDesc. AttInMetadata can |
199 | | * be used in conjunction with C strings to produce a properly formed |
200 | | * tuple. |
201 | | * HeapTuple BuildTupleFromCStrings(AttInMetadata *attinmeta, char **values) - |
202 | | * build a HeapTuple given user data in C string form. values is an array |
203 | | * of C strings, one for each attribute of the return tuple. |
204 | | * Datum HeapTupleHeaderGetDatum(HeapTupleHeader tuple) - convert a |
205 | | * HeapTupleHeader to a Datum. |
206 | | * |
207 | | * Inline declarations: |
208 | | * HeapTupleGetDatum(HeapTuple tuple) - convert a HeapTuple to a Datum. |
209 | | * |
210 | | * Obsolete routines and macros: |
211 | | * TupleDesc RelationNameGetTupleDesc(const char *relname) - Use to get a |
212 | | * TupleDesc based on a named relation. |
213 | | * TupleDesc TypeGetTupleDesc(Oid typeoid, List *colaliases) - Use to get a |
214 | | * TupleDesc based on a type OID. |
215 | | * TupleGetDatum(TupleTableSlot *slot, HeapTuple tuple) - get a Datum |
216 | | * given a tuple and a slot. |
217 | | *---------- |
218 | | */ |
219 | | |
220 | | extern TupleDesc RelationNameGetTupleDesc(const char *relname); |
221 | | extern TupleDesc TypeGetTupleDesc(Oid typeoid, List *colaliases); |
222 | | |
223 | | /* from execTuples.c */ |
224 | | extern TupleDesc BlessTupleDesc(TupleDesc tupdesc); |
225 | | extern AttInMetadata *TupleDescGetAttInMetadata(TupleDesc tupdesc); |
226 | | extern HeapTuple BuildTupleFromCStrings(AttInMetadata *attinmeta, char **values); |
227 | | extern Datum HeapTupleHeaderGetDatum(HeapTupleHeader tuple); |
228 | | |
229 | | static inline Datum |
230 | | HeapTupleGetDatum(const HeapTupleData *tuple) |
231 | 0 | { |
232 | 0 | return HeapTupleHeaderGetDatum(tuple->t_data); |
233 | 0 | } Unexecuted instantiation: commit_ts.c:HeapTupleGetDatum Unexecuted instantiation: multixact.c:HeapTupleGetDatum Unexecuted instantiation: rmgr.c:HeapTupleGetDatum Unexecuted instantiation: twophase.c:HeapTupleGetDatum Unexecuted instantiation: xlogfuncs.c:HeapTupleGetDatum Unexecuted instantiation: xlogprefetcher.c:HeapTupleGetDatum Unexecuted instantiation: dependency.c:HeapTupleGetDatum Unexecuted instantiation: namespace.c:HeapTupleGetDatum Unexecuted instantiation: objectaddress.c:HeapTupleGetDatum Unexecuted instantiation: pg_proc.c:HeapTupleGetDatum Unexecuted instantiation: pg_publication.c:HeapTupleGetDatum Unexecuted instantiation: parse_func.c:HeapTupleGetDatum Unexecuted instantiation: parse_relation.c:HeapTupleGetDatum Unexecuted instantiation: parse_target.c:HeapTupleGetDatum Unexecuted instantiation: async.c:HeapTupleGetDatum Unexecuted instantiation: event_trigger.c:HeapTupleGetDatum Unexecuted instantiation: extension.c:HeapTupleGetDatum Unexecuted instantiation: functioncmds.c:HeapTupleGetDatum Unexecuted instantiation: prepare.c:HeapTupleGetDatum Unexecuted instantiation: sequence.c:HeapTupleGetDatum Unexecuted instantiation: execExpr.c:HeapTupleGetDatum Unexecuted instantiation: execExprInterp.c:HeapTupleGetDatum Unexecuted instantiation: execSRF.c:HeapTupleGetDatum Unexecuted instantiation: execTuples.c:HeapTupleGetDatum Unexecuted instantiation: functions.c:HeapTupleGetDatum Unexecuted instantiation: nodeFunctionscan.c:HeapTupleGetDatum Unexecuted instantiation: foreign.c:HeapTupleGetDatum Unexecuted instantiation: prepjointree.c:HeapTupleGetDatum Unexecuted instantiation: clauses.c:HeapTupleGetDatum Unexecuted instantiation: launcher.c:HeapTupleGetDatum Unexecuted instantiation: logicalfuncs.c:HeapTupleGetDatum Unexecuted instantiation: origin.c:HeapTupleGetDatum Unexecuted instantiation: slotfuncs.c:HeapTupleGetDatum Unexecuted instantiation: walreceiver.c:HeapTupleGetDatum Unexecuted instantiation: walsender.c:HeapTupleGetDatum Unexecuted instantiation: walsummaryfuncs.c:HeapTupleGetDatum Unexecuted instantiation: mcv.c:HeapTupleGetDatum Unexecuted instantiation: stat_utils.c:HeapTupleGetDatum Unexecuted instantiation: aio_funcs.c:HeapTupleGetDatum Unexecuted instantiation: dsm_registry.c:HeapTupleGetDatum Unexecuted instantiation: shmem.c:HeapTupleGetDatum Unexecuted instantiation: wparser.c:HeapTupleGetDatum Unexecuted instantiation: wait_event_funcs.c:HeapTupleGetDatum Unexecuted instantiation: acl.c:HeapTupleGetDatum Unexecuted instantiation: arrayfuncs.c:HeapTupleGetDatum Unexecuted instantiation: datetime.c:HeapTupleGetDatum Unexecuted instantiation: genfile.c:HeapTupleGetDatum Unexecuted instantiation: hbafuncs.c:HeapTupleGetDatum Unexecuted instantiation: int.c:HeapTupleGetDatum Unexecuted instantiation: int8.c:HeapTupleGetDatum Unexecuted instantiation: json.c:HeapTupleGetDatum Unexecuted instantiation: jsonb.c:HeapTupleGetDatum Unexecuted instantiation: jsonfuncs.c:HeapTupleGetDatum Unexecuted instantiation: jsonpath_exec.c:HeapTupleGetDatum Unexecuted instantiation: lockfuncs.c:HeapTupleGetDatum Unexecuted instantiation: mcxtfuncs.c:HeapTupleGetDatum Unexecuted instantiation: misc.c:HeapTupleGetDatum Unexecuted instantiation: multirangetypes.c:HeapTupleGetDatum Unexecuted instantiation: numeric.c:HeapTupleGetDatum Unexecuted instantiation: partitionfuncs.c:HeapTupleGetDatum Unexecuted instantiation: pgstatfuncs.c:HeapTupleGetDatum Unexecuted instantiation: regexp.c:HeapTupleGetDatum Unexecuted instantiation: rowtypes.c:HeapTupleGetDatum Unexecuted instantiation: ruleutils.c:HeapTupleGetDatum Unexecuted instantiation: timestamp.c:HeapTupleGetDatum Unexecuted instantiation: tsvector_op.c:HeapTupleGetDatum Unexecuted instantiation: varlena.c:HeapTupleGetDatum Unexecuted instantiation: xid8funcs.c:HeapTupleGetDatum Unexecuted instantiation: funccache.c:HeapTupleGetDatum Unexecuted instantiation: funcapi.c:HeapTupleGetDatum Unexecuted instantiation: guc_funcs.c:HeapTupleGetDatum Unexecuted instantiation: pg_config.c:HeapTupleGetDatum Unexecuted instantiation: pg_controldata.c:HeapTupleGetDatum Unexecuted instantiation: portalmem.c:HeapTupleGetDatum |
234 | | |
235 | | /* obsolete version of above */ |
236 | | #define TupleGetDatum(_slot, _tuple) HeapTupleGetDatum(_tuple) |
237 | | |
238 | | |
239 | | /*---------- |
240 | | * Support for Set Returning Functions (SRFs) |
241 | | * |
242 | | * The basic API for SRFs using ValuePerCall mode looks something like this: |
243 | | * |
244 | | * Datum |
245 | | * my_Set_Returning_Function(PG_FUNCTION_ARGS) |
246 | | * { |
247 | | * FuncCallContext *funcctx; |
248 | | * Datum result; |
249 | | * MemoryContext oldcontext; |
250 | | * <user defined declarations> |
251 | | * |
252 | | * if (SRF_IS_FIRSTCALL()) |
253 | | * { |
254 | | * funcctx = SRF_FIRSTCALL_INIT(); |
255 | | * // switch context when allocating stuff to be used in later calls |
256 | | * oldcontext = MemoryContextSwitchTo(funcctx->multi_call_memory_ctx); |
257 | | * <user defined code> |
258 | | * <if returning composite> |
259 | | * <build TupleDesc, and perhaps AttInMetadata> |
260 | | * <endif returning composite> |
261 | | * <user defined code> |
262 | | * // return to original context when allocating transient memory |
263 | | * MemoryContextSwitchTo(oldcontext); |
264 | | * } |
265 | | * <user defined code> |
266 | | * funcctx = SRF_PERCALL_SETUP(); |
267 | | * <user defined code> |
268 | | * |
269 | | * if (funcctx->call_cntr < funcctx->max_calls) |
270 | | * { |
271 | | * <user defined code> |
272 | | * <obtain result Datum> |
273 | | * SRF_RETURN_NEXT(funcctx, result); |
274 | | * } |
275 | | * else |
276 | | * SRF_RETURN_DONE(funcctx); |
277 | | * } |
278 | | * |
279 | | * NOTE: there is no guarantee that a SRF using ValuePerCall mode will be |
280 | | * run to completion; for example, a query with LIMIT might stop short of |
281 | | * fetching all the rows. Therefore, do not expect that you can do resource |
282 | | * cleanup just before SRF_RETURN_DONE(). You need not worry about releasing |
283 | | * memory allocated in multi_call_memory_ctx, but holding file descriptors or |
284 | | * other non-memory resources open across calls is a bug. SRFs that need |
285 | | * such resources should not use these macros, but instead populate a |
286 | | * tuplestore during a single call, as set up by InitMaterializedSRF() (see |
287 | | * fmgr/README). Alternatively, set up a callback to release resources |
288 | | * at query shutdown, using RegisterExprContextCallback(). |
289 | | * |
290 | | *---------- |
291 | | */ |
292 | | |
293 | | /* from funcapi.c */ |
294 | | |
295 | | /* flag bits for InitMaterializedSRF() */ |
296 | 0 | #define MAT_SRF_USE_EXPECTED_DESC 0x01 /* use expectedDesc as tupdesc. */ |
297 | 0 | #define MAT_SRF_BLESS 0x02 /* "Bless" a tuple descriptor with |
298 | | * BlessTupleDesc(). */ |
299 | | extern void InitMaterializedSRF(FunctionCallInfo fcinfo, bits32 flags); |
300 | | |
301 | | extern FuncCallContext *init_MultiFuncCall(PG_FUNCTION_ARGS); |
302 | | extern FuncCallContext *per_MultiFuncCall(PG_FUNCTION_ARGS); |
303 | | extern void end_MultiFuncCall(PG_FUNCTION_ARGS, FuncCallContext *funcctx); |
304 | | |
305 | 0 | #define SRF_IS_FIRSTCALL() (fcinfo->flinfo->fn_extra == NULL) |
306 | | |
307 | 0 | #define SRF_FIRSTCALL_INIT() init_MultiFuncCall(fcinfo) |
308 | | |
309 | 0 | #define SRF_PERCALL_SETUP() per_MultiFuncCall(fcinfo) |
310 | | |
311 | | #define SRF_RETURN_NEXT(_funcctx, _result) \ |
312 | 0 | do { \ |
313 | 0 | ReturnSetInfo *rsi; \ |
314 | 0 | (_funcctx)->call_cntr++; \ |
315 | 0 | rsi = (ReturnSetInfo *) fcinfo->resultinfo; \ |
316 | 0 | rsi->isDone = ExprMultipleResult; \ |
317 | 0 | PG_RETURN_DATUM(_result); \ |
318 | 0 | } while (0) |
319 | | |
320 | | #define SRF_RETURN_NEXT_NULL(_funcctx) \ |
321 | | do { \ |
322 | | ReturnSetInfo *rsi; \ |
323 | | (_funcctx)->call_cntr++; \ |
324 | | rsi = (ReturnSetInfo *) fcinfo->resultinfo; \ |
325 | | rsi->isDone = ExprMultipleResult; \ |
326 | | PG_RETURN_NULL(); \ |
327 | | } while (0) |
328 | | |
329 | | #define SRF_RETURN_DONE(_funcctx) \ |
330 | 0 | do { \ |
331 | 0 | ReturnSetInfo *rsi; \ |
332 | 0 | end_MultiFuncCall(fcinfo, _funcctx); \ |
333 | 0 | rsi = (ReturnSetInfo *) fcinfo->resultinfo; \ |
334 | 0 | rsi->isDone = ExprEndResult; \ |
335 | 0 | PG_RETURN_NULL(); \ |
336 | 0 | } while (0) |
337 | | |
338 | | /*---------- |
339 | | * Support to ease writing of functions dealing with VARIADIC inputs |
340 | | *---------- |
341 | | * |
342 | | * This function extracts a set of argument values, types and NULL markers |
343 | | * for a given input function. This returns a set of data: |
344 | | * - **values includes the set of Datum values extracted. |
345 | | * - **types the data type OID for each element. |
346 | | * - **nulls tracks if an element is NULL. |
347 | | * |
348 | | * variadic_start indicates the argument number where the VARIADIC argument |
349 | | * starts. |
350 | | * convert_unknown set to true will enforce the conversion of arguments |
351 | | * with unknown data type to text. |
352 | | * |
353 | | * The return result is the number of elements stored, or -1 in the case of |
354 | | * "VARIADIC NULL". |
355 | | */ |
356 | | extern int extract_variadic_args(FunctionCallInfo fcinfo, int variadic_start, |
357 | | bool convert_unknown, Datum **args, |
358 | | Oid **types, bool **nulls); |
359 | | |
360 | | #endif /* FUNCAPI_H */ |