PostgreSQL Source Code  git master
funcapi.c
Go to the documentation of this file.
1 /*-------------------------------------------------------------------------
2  *
3  * funcapi.c
4  * Utility and convenience functions for fmgr functions that return
5  * sets and/or composite types, or deal with VARIADIC inputs.
6  *
7  * Copyright (c) 2002-2022, PostgreSQL Global Development Group
8  *
9  * IDENTIFICATION
10  * src/backend/utils/fmgr/funcapi.c
11  *
12  *-------------------------------------------------------------------------
13  */
14 #include "postgres.h"
15 
16 #include "access/htup_details.h"
17 #include "access/relation.h"
18 #include "catalog/namespace.h"
19 #include "catalog/pg_proc.h"
20 #include "catalog/pg_type.h"
21 #include "funcapi.h"
22 #include "miscadmin.h"
23 #include "nodes/nodeFuncs.h"
24 #include "utils/array.h"
25 #include "utils/builtins.h"
26 #include "utils/lsyscache.h"
27 #include "utils/memutils.h"
28 #include "utils/regproc.h"
29 #include "utils/rel.h"
30 #include "utils/syscache.h"
31 #include "utils/tuplestore.h"
32 #include "utils/typcache.h"
33 
34 
35 typedef struct polymorphic_actuals
36 {
37  Oid anyelement_type; /* anyelement mapping, if known */
38  Oid anyarray_type; /* anyarray mapping, if known */
39  Oid anyrange_type; /* anyrange mapping, if known */
40  Oid anymultirange_type; /* anymultirange mapping, if known */
42 
43 static void shutdown_MultiFuncCall(Datum arg);
45  Node *call_expr,
46  ReturnSetInfo *rsinfo,
47  Oid *resultTypeId,
48  TupleDesc *resultTupleDesc);
53 static bool resolve_polymorphic_tupdesc(TupleDesc tupdesc,
54  oidvector *declared_args,
55  Node *call_expr);
56 static TypeFuncClass get_type_func_class(Oid typid, Oid *base_typeid);
57 
58 
59 /*
60  * InitMaterializedSRF
61  *
62  * Helper function to build the state of a set-returning function used
63  * in the context of a single call with materialize mode. This code
64  * includes sanity checks on ReturnSetInfo, creates the Tuplestore and
65  * the TupleDesc used with the function and stores them into the
66  * function's ReturnSetInfo.
67  *
68  * "flags" can be set to MAT_SRF_USE_EXPECTED_DESC, to use the tuple
69  * descriptor coming from expectedDesc, which is the tuple descriptor
70  * expected by the caller. MAT_SRF_BLESS can be set to complete the
71  * information associated to the tuple descriptor, which is necessary
72  * in some cases where the tuple descriptor comes from a transient
73  * RECORD datatype.
74  */
75 void
77 {
78  bool random_access;
79  ReturnSetInfo *rsinfo = (ReturnSetInfo *) fcinfo->resultinfo;
80  Tuplestorestate *tupstore;
81  MemoryContext old_context,
82  per_query_ctx;
83  TupleDesc stored_tupdesc;
84 
85  /* check to see if caller supports returning a tuplestore */
86  if (rsinfo == NULL || !IsA(rsinfo, ReturnSetInfo))
87  ereport(ERROR,
88  (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
89  errmsg("set-valued function called in context that cannot accept a set")));
90  if (!(rsinfo->allowedModes & SFRM_Materialize) ||
91  ((flags & MAT_SRF_USE_EXPECTED_DESC) != 0 && rsinfo->expectedDesc == NULL))
92  ereport(ERROR,
93  (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
94  errmsg("materialize mode required, but it is not allowed in this context")));
95 
96  /*
97  * Store the tuplestore and the tuple descriptor in ReturnSetInfo. This
98  * must be done in the per-query memory context.
99  */
100  per_query_ctx = rsinfo->econtext->ecxt_per_query_memory;
101  old_context = MemoryContextSwitchTo(per_query_ctx);
102 
103  /* build a tuple descriptor for our result type */
104  if ((flags & MAT_SRF_USE_EXPECTED_DESC) != 0)
105  stored_tupdesc = CreateTupleDescCopy(rsinfo->expectedDesc);
106  else
107  {
108  if (get_call_result_type(fcinfo, NULL, &stored_tupdesc) != TYPEFUNC_COMPOSITE)
109  elog(ERROR, "return type must be a row type");
110  }
111 
112  /* If requested, bless the tuple descriptor */
113  if ((flags & MAT_SRF_BLESS) != 0)
114  BlessTupleDesc(stored_tupdesc);
115 
116  random_access = (rsinfo->allowedModes & SFRM_Materialize_Random) != 0;
117 
118  tupstore = tuplestore_begin_heap(random_access, false, work_mem);
119  rsinfo->returnMode = SFRM_Materialize;
120  rsinfo->setResult = tupstore;
121  rsinfo->setDesc = stored_tupdesc;
122  MemoryContextSwitchTo(old_context);
123 }
124 
125 
126 /*
127  * init_MultiFuncCall
128  * Create an empty FuncCallContext data structure
129  * and do some other basic Multi-function call setup
130  * and error checking
131  */
134 {
135  FuncCallContext *retval;
136 
137  /*
138  * Bail if we're called in the wrong context
139  */
140  if (fcinfo->resultinfo == NULL || !IsA(fcinfo->resultinfo, ReturnSetInfo))
141  ereport(ERROR,
142  (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
143  errmsg("set-valued function called in context that cannot accept a set")));
144 
145  if (fcinfo->flinfo->fn_extra == NULL)
146  {
147  /*
148  * First call
149  */
150  ReturnSetInfo *rsi = (ReturnSetInfo *) fcinfo->resultinfo;
151  MemoryContext multi_call_ctx;
152 
153  /*
154  * Create a suitably long-lived context to hold cross-call data
155  */
156  multi_call_ctx = AllocSetContextCreate(fcinfo->flinfo->fn_mcxt,
157  "SRF multi-call context",
159 
160  /*
161  * Allocate suitably long-lived space and zero it
162  */
163  retval = (FuncCallContext *)
164  MemoryContextAllocZero(multi_call_ctx,
165  sizeof(FuncCallContext));
166 
167  /*
168  * initialize the elements
169  */
170  retval->call_cntr = 0;
171  retval->max_calls = 0;
172  retval->user_fctx = NULL;
173  retval->attinmeta = NULL;
174  retval->tuple_desc = NULL;
175  retval->multi_call_memory_ctx = multi_call_ctx;
176 
177  /*
178  * save the pointer for cross-call use
179  */
180  fcinfo->flinfo->fn_extra = retval;
181 
182  /*
183  * Ensure we will get shut down cleanly if the exprcontext is not run
184  * to completion.
185  */
188  PointerGetDatum(fcinfo->flinfo));
189  }
190  else
191  {
192  /* second and subsequent calls */
193  elog(ERROR, "init_MultiFuncCall cannot be called more than once");
194 
195  /* never reached, but keep compiler happy */
196  retval = NULL;
197  }
198 
199  return retval;
200 }
201 
202 /*
203  * per_MultiFuncCall
204  *
205  * Do Multi-function per-call setup
206  */
209 {
210  FuncCallContext *retval = (FuncCallContext *) fcinfo->flinfo->fn_extra;
211 
212  return retval;
213 }
214 
215 /*
216  * end_MultiFuncCall
217  * Clean up after init_MultiFuncCall
218  */
219 void
221 {
222  ReturnSetInfo *rsi = (ReturnSetInfo *) fcinfo->resultinfo;
223 
224  /* Deregister the shutdown callback */
227  PointerGetDatum(fcinfo->flinfo));
228 
229  /* But use it to do the real work */
230  shutdown_MultiFuncCall(PointerGetDatum(fcinfo->flinfo));
231 }
232 
233 /*
234  * shutdown_MultiFuncCall
235  * Shutdown function to clean up after init_MultiFuncCall
236  */
237 static void
239 {
240  FmgrInfo *flinfo = (FmgrInfo *) DatumGetPointer(arg);
241  FuncCallContext *funcctx = (FuncCallContext *) flinfo->fn_extra;
242 
243  /* unbind from flinfo */
244  flinfo->fn_extra = NULL;
245 
246  /*
247  * Delete context that holds all multi-call data, including the
248  * FuncCallContext itself
249  */
251 }
252 
253 
254 /*
255  * get_call_result_type
256  * Given a function's call info record, determine the kind of datatype
257  * it is supposed to return. If resultTypeId isn't NULL, *resultTypeId
258  * receives the actual datatype OID (this is mainly useful for scalar
259  * result types). If resultTupleDesc isn't NULL, *resultTupleDesc
260  * receives a pointer to a TupleDesc when the result is of a composite
261  * type, or NULL when it's a scalar result.
262  *
263  * One hard case that this handles is resolution of actual rowtypes for
264  * functions returning RECORD (from either the function's OUT parameter
265  * list, or a ReturnSetInfo context node). TYPEFUNC_RECORD is returned
266  * only when we couldn't resolve the actual rowtype for lack of information.
267  *
268  * The other hard case that this handles is resolution of polymorphism.
269  * We will never return polymorphic pseudotypes (ANYELEMENT etc), either
270  * as a scalar result type or as a component of a rowtype.
271  *
272  * This function is relatively expensive --- in a function returning set,
273  * try to call it only the first time through.
274  */
277  Oid *resultTypeId,
278  TupleDesc *resultTupleDesc)
279 {
280  return internal_get_result_type(fcinfo->flinfo->fn_oid,
281  fcinfo->flinfo->fn_expr,
282  (ReturnSetInfo *) fcinfo->resultinfo,
283  resultTypeId,
284  resultTupleDesc);
285 }
286 
287 /*
288  * get_expr_result_type
289  * As above, but work from a calling expression node tree
290  */
293  Oid *resultTypeId,
294  TupleDesc *resultTupleDesc)
295 {
296  TypeFuncClass result;
297 
298  if (expr && IsA(expr, FuncExpr))
299  result = internal_get_result_type(((FuncExpr *) expr)->funcid,
300  expr,
301  NULL,
302  resultTypeId,
303  resultTupleDesc);
304  else if (expr && IsA(expr, OpExpr))
305  result = internal_get_result_type(get_opcode(((OpExpr *) expr)->opno),
306  expr,
307  NULL,
308  resultTypeId,
309  resultTupleDesc);
310  else if (expr && IsA(expr, RowExpr) &&
311  ((RowExpr *) expr)->row_typeid == RECORDOID)
312  {
313  /* We can resolve the record type by generating the tupdesc directly */
314  RowExpr *rexpr = (RowExpr *) expr;
315  TupleDesc tupdesc;
316  AttrNumber i = 1;
317  ListCell *lcc,
318  *lcn;
319 
320  tupdesc = CreateTemplateTupleDesc(list_length(rexpr->args));
321  Assert(list_length(rexpr->args) == list_length(rexpr->colnames));
322  forboth(lcc, rexpr->args, lcn, rexpr->colnames)
323  {
324  Node *col = (Node *) lfirst(lcc);
325  char *colname = strVal(lfirst(lcn));
326 
327  TupleDescInitEntry(tupdesc, i,
328  colname,
329  exprType(col),
330  exprTypmod(col),
331  0);
333  exprCollation(col));
334  i++;
335  }
336  if (resultTypeId)
337  *resultTypeId = rexpr->row_typeid;
338  if (resultTupleDesc)
339  *resultTupleDesc = BlessTupleDesc(tupdesc);
340  return TYPEFUNC_COMPOSITE;
341  }
342  else if (expr && IsA(expr, Const) &&
343  ((Const *) expr)->consttype == RECORDOID &&
344  !((Const *) expr)->constisnull)
345  {
346  /*
347  * When EXPLAIN'ing some queries with SEARCH/CYCLE clauses, we may
348  * need to resolve field names of a RECORD-type Const. The datum
349  * should contain a typmod that will tell us that.
350  */
351  HeapTupleHeader rec;
352  Oid tupType;
353  int32 tupTypmod;
354 
355  rec = DatumGetHeapTupleHeader(((Const *) expr)->constvalue);
356  tupType = HeapTupleHeaderGetTypeId(rec);
357  tupTypmod = HeapTupleHeaderGetTypMod(rec);
358  if (resultTypeId)
359  *resultTypeId = tupType;
360  if (tupType != RECORDOID || tupTypmod >= 0)
361  {
362  /* Should be able to look it up */
363  if (resultTupleDesc)
364  *resultTupleDesc = lookup_rowtype_tupdesc_copy(tupType,
365  tupTypmod);
366  return TYPEFUNC_COMPOSITE;
367  }
368  else
369  {
370  /* This shouldn't really happen ... */
371  if (resultTupleDesc)
372  *resultTupleDesc = NULL;
373  return TYPEFUNC_RECORD;
374  }
375  }
376  else
377  {
378  /* handle as a generic expression; no chance to resolve RECORD */
379  Oid typid = exprType(expr);
380  Oid base_typid;
381 
382  if (resultTypeId)
383  *resultTypeId = typid;
384  if (resultTupleDesc)
385  *resultTupleDesc = NULL;
386  result = get_type_func_class(typid, &base_typid);
387  if ((result == TYPEFUNC_COMPOSITE ||
388  result == TYPEFUNC_COMPOSITE_DOMAIN) &&
389  resultTupleDesc)
390  *resultTupleDesc = lookup_rowtype_tupdesc_copy(base_typid, -1);
391  }
392 
393  return result;
394 }
395 
396 /*
397  * get_func_result_type
398  * As above, but work from a function's OID only
399  *
400  * This will not be able to resolve pure-RECORD results nor polymorphism.
401  */
404  Oid *resultTypeId,
405  TupleDesc *resultTupleDesc)
406 {
407  return internal_get_result_type(functionId,
408  NULL,
409  NULL,
410  resultTypeId,
411  resultTupleDesc);
412 }
413 
414 /*
415  * internal_get_result_type -- workhorse code implementing all the above
416  *
417  * funcid must always be supplied. call_expr and rsinfo can be NULL if not
418  * available. We will return TYPEFUNC_RECORD, and store NULL into
419  * *resultTupleDesc, if we cannot deduce the complete result rowtype from
420  * the available information.
421  */
422 static TypeFuncClass
424  Node *call_expr,
425  ReturnSetInfo *rsinfo,
426  Oid *resultTypeId,
427  TupleDesc *resultTupleDesc)
428 {
429  TypeFuncClass result;
430  HeapTuple tp;
431  Form_pg_proc procform;
432  Oid rettype;
433  Oid base_rettype;
434  TupleDesc tupdesc;
435 
436  /* First fetch the function's pg_proc row to inspect its rettype */
438  if (!HeapTupleIsValid(tp))
439  elog(ERROR, "cache lookup failed for function %u", funcid);
440  procform = (Form_pg_proc) GETSTRUCT(tp);
441 
442  rettype = procform->prorettype;
443 
444  /* Check for OUT parameters defining a RECORD result */
445  tupdesc = build_function_result_tupdesc_t(tp);
446  if (tupdesc)
447  {
448  /*
449  * It has OUT parameters, so it's basically like a regular composite
450  * type, except we have to be able to resolve any polymorphic OUT
451  * parameters.
452  */
453  if (resultTypeId)
454  *resultTypeId = rettype;
455 
456  if (resolve_polymorphic_tupdesc(tupdesc,
457  &procform->proargtypes,
458  call_expr))
459  {
460  if (tupdesc->tdtypeid == RECORDOID &&
461  tupdesc->tdtypmod < 0)
462  assign_record_type_typmod(tupdesc);
463  if (resultTupleDesc)
464  *resultTupleDesc = tupdesc;
465  result = TYPEFUNC_COMPOSITE;
466  }
467  else
468  {
469  if (resultTupleDesc)
470  *resultTupleDesc = NULL;
471  result = TYPEFUNC_RECORD;
472  }
473 
474  ReleaseSysCache(tp);
475 
476  return result;
477  }
478 
479  /*
480  * If scalar polymorphic result, try to resolve it.
481  */
482  if (IsPolymorphicType(rettype))
483  {
484  Oid newrettype = exprType(call_expr);
485 
486  if (newrettype == InvalidOid) /* this probably should not happen */
487  ereport(ERROR,
488  (errcode(ERRCODE_DATATYPE_MISMATCH),
489  errmsg("could not determine actual result type for function \"%s\" declared to return type %s",
490  NameStr(procform->proname),
491  format_type_be(rettype))));
492  rettype = newrettype;
493  }
494 
495  if (resultTypeId)
496  *resultTypeId = rettype;
497  if (resultTupleDesc)
498  *resultTupleDesc = NULL; /* default result */
499 
500  /* Classify the result type */
501  result = get_type_func_class(rettype, &base_rettype);
502  switch (result)
503  {
504  case TYPEFUNC_COMPOSITE:
506  if (resultTupleDesc)
507  *resultTupleDesc = lookup_rowtype_tupdesc_copy(base_rettype, -1);
508  /* Named composite types can't have any polymorphic columns */
509  break;
510  case TYPEFUNC_SCALAR:
511  break;
512  case TYPEFUNC_RECORD:
513  /* We must get the tupledesc from call context */
514  if (rsinfo && IsA(rsinfo, ReturnSetInfo) &&
515  rsinfo->expectedDesc != NULL)
516  {
517  result = TYPEFUNC_COMPOSITE;
518  if (resultTupleDesc)
519  *resultTupleDesc = rsinfo->expectedDesc;
520  /* Assume no polymorphic columns here, either */
521  }
522  break;
523  default:
524  break;
525  }
526 
527  ReleaseSysCache(tp);
528 
529  return result;
530 }
531 
532 /*
533  * get_expr_result_tupdesc
534  * Get a tupdesc describing the result of a composite-valued expression
535  *
536  * If expression is not composite or rowtype can't be determined, returns NULL
537  * if noError is true, else throws error.
538  *
539  * This is a simpler version of get_expr_result_type() for use when the caller
540  * is only interested in determinate rowtype results.
541  */
542 TupleDesc
543 get_expr_result_tupdesc(Node *expr, bool noError)
544 {
545  TupleDesc tupleDesc;
546  TypeFuncClass functypclass;
547 
548  functypclass = get_expr_result_type(expr, NULL, &tupleDesc);
549 
550  if (functypclass == TYPEFUNC_COMPOSITE ||
551  functypclass == TYPEFUNC_COMPOSITE_DOMAIN)
552  return tupleDesc;
553 
554  if (!noError)
555  {
556  Oid exprTypeId = exprType(expr);
557 
558  if (exprTypeId != RECORDOID)
559  ereport(ERROR,
560  (errcode(ERRCODE_WRONG_OBJECT_TYPE),
561  errmsg("type %s is not composite",
562  format_type_be(exprTypeId))));
563  else
564  ereport(ERROR,
565  (errcode(ERRCODE_WRONG_OBJECT_TYPE),
566  errmsg("record type has not been registered")));
567  }
568 
569  return NULL;
570 }
571 
572 /*
573  * Resolve actual type of ANYELEMENT from other polymorphic inputs
574  *
575  * Note: the error cases here and in the sibling functions below are not
576  * really user-facing; they could only occur if the function signature is
577  * incorrect or the parser failed to enforce consistency of the actual
578  * argument types. Hence, we don't sweat too much over the error messages.
579  */
580 static void
582 {
583  if (OidIsValid(actuals->anyarray_type))
584  {
585  /* Use the element type corresponding to actual type */
586  Oid array_base_type = getBaseType(actuals->anyarray_type);
587  Oid array_typelem = get_element_type(array_base_type);
588 
589  if (!OidIsValid(array_typelem))
590  ereport(ERROR,
591  (errcode(ERRCODE_DATATYPE_MISMATCH),
592  errmsg("argument declared %s is not an array but type %s",
593  "anyarray",
594  format_type_be(array_base_type))));
595  actuals->anyelement_type = array_typelem;
596  }
597  else if (OidIsValid(actuals->anyrange_type))
598  {
599  /* Use the element type corresponding to actual type */
600  Oid range_base_type = getBaseType(actuals->anyrange_type);
601  Oid range_typelem = get_range_subtype(range_base_type);
602 
603  if (!OidIsValid(range_typelem))
604  ereport(ERROR,
605  (errcode(ERRCODE_DATATYPE_MISMATCH),
606  errmsg("argument declared %s is not a range type but type %s",
607  "anyrange",
608  format_type_be(range_base_type))));
609  actuals->anyelement_type = range_typelem;
610  }
611  else if (OidIsValid(actuals->anymultirange_type))
612  {
613  /* Use the element type based on the multirange type */
614  Oid multirange_base_type;
615  Oid multirange_typelem;
616  Oid range_base_type;
617  Oid range_typelem;
618 
619  multirange_base_type = getBaseType(actuals->anymultirange_type);
620  multirange_typelem = get_multirange_range(multirange_base_type);
621  if (!OidIsValid(multirange_typelem))
622  ereport(ERROR,
623  (errcode(ERRCODE_DATATYPE_MISMATCH),
624  errmsg("argument declared %s is not a multirange type but type %s",
625  "anymultirange",
626  format_type_be(multirange_base_type))));
627 
628  range_base_type = getBaseType(multirange_typelem);
629  range_typelem = get_range_subtype(range_base_type);
630 
631  if (!OidIsValid(range_typelem))
632  ereport(ERROR,
633  (errcode(ERRCODE_DATATYPE_MISMATCH),
634  errmsg("argument declared %s does not contain a range type but type %s",
635  "anymultirange",
636  format_type_be(range_base_type))));
637  actuals->anyelement_type = range_typelem;
638  }
639  else
640  elog(ERROR, "could not determine polymorphic type");
641 }
642 
643 /*
644  * Resolve actual type of ANYARRAY from other polymorphic inputs
645  */
646 static void
648 {
649  /* If we don't know ANYELEMENT, resolve that first */
650  if (!OidIsValid(actuals->anyelement_type))
652 
653  if (OidIsValid(actuals->anyelement_type))
654  {
655  /* Use the array type corresponding to actual type */
656  Oid array_typeid = get_array_type(actuals->anyelement_type);
657 
658  if (!OidIsValid(array_typeid))
659  ereport(ERROR,
660  (errcode(ERRCODE_UNDEFINED_OBJECT),
661  errmsg("could not find array type for data type %s",
662  format_type_be(actuals->anyelement_type))));
663  actuals->anyarray_type = array_typeid;
664  }
665  else
666  elog(ERROR, "could not determine polymorphic type");
667 }
668 
669 /*
670  * Resolve actual type of ANYRANGE from other polymorphic inputs
671  */
672 static void
674 {
675  /*
676  * We can't deduce a range type from other polymorphic array or base
677  * types, because there may be multiple range types with the same subtype,
678  * but we can deduce it from a polymorphic multirange type.
679  */
680  if (OidIsValid(actuals->anymultirange_type))
681  {
682  /* Use the element type based on the multirange type */
683  Oid multirange_base_type = getBaseType(actuals->anymultirange_type);
684  Oid multirange_typelem = get_multirange_range(multirange_base_type);
685 
686  if (!OidIsValid(multirange_typelem))
687  ereport(ERROR,
688  (errcode(ERRCODE_DATATYPE_MISMATCH),
689  errmsg("argument declared %s is not a multirange type but type %s",
690  "anymultirange",
691  format_type_be(multirange_base_type))));
692  actuals->anyrange_type = multirange_typelem;
693  }
694  else
695  elog(ERROR, "could not determine polymorphic type");
696 }
697 
698 /*
699  * Resolve actual type of ANYMULTIRANGE from other polymorphic inputs
700  */
701 static void
703 {
704  /*
705  * We can't deduce a multirange type from polymorphic array or base types,
706  * because there may be multiple range types with the same subtype, but we
707  * can deduce it from a polymorphic range type.
708  */
709  if (OidIsValid(actuals->anyrange_type))
710  {
711  Oid range_base_type = getBaseType(actuals->anyrange_type);
712  Oid multirange_typeid = get_range_multirange(range_base_type);
713 
714  if (!OidIsValid(multirange_typeid))
715  ereport(ERROR,
716  (errcode(ERRCODE_UNDEFINED_OBJECT),
717  errmsg("could not find multirange type for data type %s",
718  format_type_be(actuals->anyrange_type))));
719  actuals->anymultirange_type = multirange_typeid;
720  }
721  else
722  elog(ERROR, "could not determine polymorphic type");
723 }
724 
725 /*
726  * Given the result tuple descriptor for a function with OUT parameters,
727  * replace any polymorphic column types (ANYELEMENT etc) in the tupdesc
728  * with concrete data types deduced from the input arguments.
729  * declared_args is an oidvector of the function's declared input arg types
730  * (showing which are polymorphic), and call_expr is the call expression.
731  *
732  * Returns true if able to deduce all types, false if necessary information
733  * is not provided (call_expr is NULL or arg types aren't identifiable).
734  */
735 static bool
737  Node *call_expr)
738 {
739  int natts = tupdesc->natts;
740  int nargs = declared_args->dim1;
741  bool have_polymorphic_result = false;
742  bool have_anyelement_result = false;
743  bool have_anyarray_result = false;
744  bool have_anyrange_result = false;
745  bool have_anymultirange_result = false;
746  bool have_anycompatible_result = false;
747  bool have_anycompatible_array_result = false;
748  bool have_anycompatible_range_result = false;
749  bool have_anycompatible_multirange_result = false;
750  polymorphic_actuals poly_actuals;
751  polymorphic_actuals anyc_actuals;
752  Oid anycollation = InvalidOid;
753  Oid anycompatcollation = InvalidOid;
754  int i;
755 
756  /* See if there are any polymorphic outputs; quick out if not */
757  for (i = 0; i < natts; i++)
758  {
759  switch (TupleDescAttr(tupdesc, i)->atttypid)
760  {
761  case ANYELEMENTOID:
762  case ANYNONARRAYOID:
763  case ANYENUMOID:
764  have_polymorphic_result = true;
765  have_anyelement_result = true;
766  break;
767  case ANYARRAYOID:
768  have_polymorphic_result = true;
769  have_anyarray_result = true;
770  break;
771  case ANYRANGEOID:
772  have_polymorphic_result = true;
773  have_anyrange_result = true;
774  break;
775  case ANYMULTIRANGEOID:
776  have_polymorphic_result = true;
777  have_anymultirange_result = true;
778  break;
779  case ANYCOMPATIBLEOID:
780  case ANYCOMPATIBLENONARRAYOID:
781  have_polymorphic_result = true;
782  have_anycompatible_result = true;
783  break;
784  case ANYCOMPATIBLEARRAYOID:
785  have_polymorphic_result = true;
786  have_anycompatible_array_result = true;
787  break;
788  case ANYCOMPATIBLERANGEOID:
789  have_polymorphic_result = true;
790  have_anycompatible_range_result = true;
791  break;
792  case ANYCOMPATIBLEMULTIRANGEOID:
793  have_polymorphic_result = true;
794  have_anycompatible_multirange_result = true;
795  break;
796  default:
797  break;
798  }
799  }
800  if (!have_polymorphic_result)
801  return true;
802 
803  /*
804  * Otherwise, extract actual datatype(s) from input arguments. (We assume
805  * the parser already validated consistency of the arguments. Also, for
806  * the ANYCOMPATIBLE pseudotype family, we expect that all matching
807  * arguments were coerced to the selected common supertype, so that it
808  * doesn't matter which one's exposed type we look at.)
809  */
810  if (!call_expr)
811  return false; /* no hope */
812 
813  memset(&poly_actuals, 0, sizeof(poly_actuals));
814  memset(&anyc_actuals, 0, sizeof(anyc_actuals));
815 
816  for (i = 0; i < nargs; i++)
817  {
818  switch (declared_args->values[i])
819  {
820  case ANYELEMENTOID:
821  case ANYNONARRAYOID:
822  case ANYENUMOID:
823  if (!OidIsValid(poly_actuals.anyelement_type))
824  {
825  poly_actuals.anyelement_type =
826  get_call_expr_argtype(call_expr, i);
827  if (!OidIsValid(poly_actuals.anyelement_type))
828  return false;
829  }
830  break;
831  case ANYARRAYOID:
832  if (!OidIsValid(poly_actuals.anyarray_type))
833  {
834  poly_actuals.anyarray_type =
835  get_call_expr_argtype(call_expr, i);
836  if (!OidIsValid(poly_actuals.anyarray_type))
837  return false;
838  }
839  break;
840  case ANYRANGEOID:
841  if (!OidIsValid(poly_actuals.anyrange_type))
842  {
843  poly_actuals.anyrange_type =
844  get_call_expr_argtype(call_expr, i);
845  if (!OidIsValid(poly_actuals.anyrange_type))
846  return false;
847  }
848  break;
849  case ANYMULTIRANGEOID:
850  if (!OidIsValid(poly_actuals.anymultirange_type))
851  {
852  poly_actuals.anymultirange_type =
853  get_call_expr_argtype(call_expr, i);
854  if (!OidIsValid(poly_actuals.anymultirange_type))
855  return false;
856  }
857  break;
858  case ANYCOMPATIBLEOID:
859  case ANYCOMPATIBLENONARRAYOID:
860  if (!OidIsValid(anyc_actuals.anyelement_type))
861  {
862  anyc_actuals.anyelement_type =
863  get_call_expr_argtype(call_expr, i);
864  if (!OidIsValid(anyc_actuals.anyelement_type))
865  return false;
866  }
867  break;
868  case ANYCOMPATIBLEARRAYOID:
869  if (!OidIsValid(anyc_actuals.anyarray_type))
870  {
871  anyc_actuals.anyarray_type =
872  get_call_expr_argtype(call_expr, i);
873  if (!OidIsValid(anyc_actuals.anyarray_type))
874  return false;
875  }
876  break;
877  case ANYCOMPATIBLERANGEOID:
878  if (!OidIsValid(anyc_actuals.anyrange_type))
879  {
880  anyc_actuals.anyrange_type =
881  get_call_expr_argtype(call_expr, i);
882  if (!OidIsValid(anyc_actuals.anyrange_type))
883  return false;
884  }
885  break;
886  case ANYCOMPATIBLEMULTIRANGEOID:
887  if (!OidIsValid(anyc_actuals.anymultirange_type))
888  {
889  anyc_actuals.anymultirange_type =
890  get_call_expr_argtype(call_expr, i);
891  if (!OidIsValid(anyc_actuals.anymultirange_type))
892  return false;
893  }
894  break;
895  default:
896  break;
897  }
898  }
899 
900  /* If needed, deduce one polymorphic type from others */
901  if (have_anyelement_result && !OidIsValid(poly_actuals.anyelement_type))
902  resolve_anyelement_from_others(&poly_actuals);
903 
904  if (have_anyarray_result && !OidIsValid(poly_actuals.anyarray_type))
905  resolve_anyarray_from_others(&poly_actuals);
906 
907  if (have_anyrange_result && !OidIsValid(poly_actuals.anyrange_type))
908  resolve_anyrange_from_others(&poly_actuals);
909 
910  if (have_anymultirange_result && !OidIsValid(poly_actuals.anymultirange_type))
911  resolve_anymultirange_from_others(&poly_actuals);
912 
913  if (have_anycompatible_result && !OidIsValid(anyc_actuals.anyelement_type))
914  resolve_anyelement_from_others(&anyc_actuals);
915 
916  if (have_anycompatible_array_result && !OidIsValid(anyc_actuals.anyarray_type))
917  resolve_anyarray_from_others(&anyc_actuals);
918 
919  if (have_anycompatible_range_result && !OidIsValid(anyc_actuals.anyrange_type))
920  resolve_anyrange_from_others(&anyc_actuals);
921 
922  if (have_anycompatible_multirange_result && !OidIsValid(anyc_actuals.anymultirange_type))
923  resolve_anymultirange_from_others(&anyc_actuals);
924 
925  /*
926  * Identify the collation to use for polymorphic OUT parameters. (It'll
927  * necessarily be the same for both anyelement and anyarray, likewise for
928  * anycompatible and anycompatiblearray.) Note that range types are not
929  * collatable, so any possible internal collation of a range type is not
930  * considered here.
931  */
932  if (OidIsValid(poly_actuals.anyelement_type))
933  anycollation = get_typcollation(poly_actuals.anyelement_type);
934  else if (OidIsValid(poly_actuals.anyarray_type))
935  anycollation = get_typcollation(poly_actuals.anyarray_type);
936 
937  if (OidIsValid(anyc_actuals.anyelement_type))
938  anycompatcollation = get_typcollation(anyc_actuals.anyelement_type);
939  else if (OidIsValid(anyc_actuals.anyarray_type))
940  anycompatcollation = get_typcollation(anyc_actuals.anyarray_type);
941 
942  if (OidIsValid(anycollation) || OidIsValid(anycompatcollation))
943  {
944  /*
945  * The types are collatable, so consider whether to use a nondefault
946  * collation. We do so if we can identify the input collation used
947  * for the function.
948  */
949  Oid inputcollation = exprInputCollation(call_expr);
950 
951  if (OidIsValid(inputcollation))
952  {
953  if (OidIsValid(anycollation))
954  anycollation = inputcollation;
955  if (OidIsValid(anycompatcollation))
956  anycompatcollation = inputcollation;
957  }
958  }
959 
960  /* And finally replace the tuple column types as needed */
961  for (i = 0; i < natts; i++)
962  {
963  Form_pg_attribute att = TupleDescAttr(tupdesc, i);
964 
965  switch (att->atttypid)
966  {
967  case ANYELEMENTOID:
968  case ANYNONARRAYOID:
969  case ANYENUMOID:
970  TupleDescInitEntry(tupdesc, i + 1,
971  NameStr(att->attname),
972  poly_actuals.anyelement_type,
973  -1,
974  0);
975  TupleDescInitEntryCollation(tupdesc, i + 1, anycollation);
976  break;
977  case ANYARRAYOID:
978  TupleDescInitEntry(tupdesc, i + 1,
979  NameStr(att->attname),
980  poly_actuals.anyarray_type,
981  -1,
982  0);
983  TupleDescInitEntryCollation(tupdesc, i + 1, anycollation);
984  break;
985  case ANYRANGEOID:
986  TupleDescInitEntry(tupdesc, i + 1,
987  NameStr(att->attname),
988  poly_actuals.anyrange_type,
989  -1,
990  0);
991  /* no collation should be attached to a range type */
992  break;
993  case ANYMULTIRANGEOID:
994  TupleDescInitEntry(tupdesc, i + 1,
995  NameStr(att->attname),
996  poly_actuals.anymultirange_type,
997  -1,
998  0);
999  /* no collation should be attached to a multirange type */
1000  break;
1001  case ANYCOMPATIBLEOID:
1002  case ANYCOMPATIBLENONARRAYOID:
1003  TupleDescInitEntry(tupdesc, i + 1,
1004  NameStr(att->attname),
1005  anyc_actuals.anyelement_type,
1006  -1,
1007  0);
1008  TupleDescInitEntryCollation(tupdesc, i + 1, anycompatcollation);
1009  break;
1010  case ANYCOMPATIBLEARRAYOID:
1011  TupleDescInitEntry(tupdesc, i + 1,
1012  NameStr(att->attname),
1013  anyc_actuals.anyarray_type,
1014  -1,
1015  0);
1016  TupleDescInitEntryCollation(tupdesc, i + 1, anycompatcollation);
1017  break;
1018  case ANYCOMPATIBLERANGEOID:
1019  TupleDescInitEntry(tupdesc, i + 1,
1020  NameStr(att->attname),
1021  anyc_actuals.anyrange_type,
1022  -1,
1023  0);
1024  /* no collation should be attached to a range type */
1025  break;
1026  case ANYCOMPATIBLEMULTIRANGEOID:
1027  TupleDescInitEntry(tupdesc, i + 1,
1028  NameStr(att->attname),
1029  anyc_actuals.anymultirange_type,
1030  -1,
1031  0);
1032  /* no collation should be attached to a multirange type */
1033  break;
1034  default:
1035  break;
1036  }
1037  }
1038 
1039  return true;
1040 }
1041 
1042 /*
1043  * Given the declared argument types and modes for a function, replace any
1044  * polymorphic types (ANYELEMENT etc) in argtypes[] with concrete data types
1045  * deduced from the input arguments found in call_expr.
1046  *
1047  * Returns true if able to deduce all types, false if necessary information
1048  * is not provided (call_expr is NULL or arg types aren't identifiable).
1049  *
1050  * This is the same logic as resolve_polymorphic_tupdesc, but with a different
1051  * argument representation, and slightly different output responsibilities.
1052  *
1053  * argmodes may be NULL, in which case all arguments are assumed to be IN mode.
1054  */
1055 bool
1056 resolve_polymorphic_argtypes(int numargs, Oid *argtypes, char *argmodes,
1057  Node *call_expr)
1058 {
1059  bool have_polymorphic_result = false;
1060  bool have_anyelement_result = false;
1061  bool have_anyarray_result = false;
1062  bool have_anyrange_result = false;
1063  bool have_anymultirange_result = false;
1064  bool have_anycompatible_result = false;
1065  bool have_anycompatible_array_result = false;
1066  bool have_anycompatible_range_result = false;
1067  bool have_anycompatible_multirange_result = false;
1068  polymorphic_actuals poly_actuals;
1069  polymorphic_actuals anyc_actuals;
1070  int inargno;
1071  int i;
1072 
1073  /*
1074  * First pass: resolve polymorphic inputs, check for outputs. As in
1075  * resolve_polymorphic_tupdesc, we rely on the parser to have enforced
1076  * type consistency and coerced ANYCOMPATIBLE args to a common supertype.
1077  */
1078  memset(&poly_actuals, 0, sizeof(poly_actuals));
1079  memset(&anyc_actuals, 0, sizeof(anyc_actuals));
1080  inargno = 0;
1081  for (i = 0; i < numargs; i++)
1082  {
1083  char argmode = argmodes ? argmodes[i] : PROARGMODE_IN;
1084 
1085  switch (argtypes[i])
1086  {
1087  case ANYELEMENTOID:
1088  case ANYNONARRAYOID:
1089  case ANYENUMOID:
1090  if (argmode == PROARGMODE_OUT || argmode == PROARGMODE_TABLE)
1091  {
1092  have_polymorphic_result = true;
1093  have_anyelement_result = true;
1094  }
1095  else
1096  {
1097  if (!OidIsValid(poly_actuals.anyelement_type))
1098  {
1099  poly_actuals.anyelement_type =
1100  get_call_expr_argtype(call_expr, inargno);
1101  if (!OidIsValid(poly_actuals.anyelement_type))
1102  return false;
1103  }
1104  argtypes[i] = poly_actuals.anyelement_type;
1105  }
1106  break;
1107  case ANYARRAYOID:
1108  if (argmode == PROARGMODE_OUT || argmode == PROARGMODE_TABLE)
1109  {
1110  have_polymorphic_result = true;
1111  have_anyarray_result = true;
1112  }
1113  else
1114  {
1115  if (!OidIsValid(poly_actuals.anyarray_type))
1116  {
1117  poly_actuals.anyarray_type =
1118  get_call_expr_argtype(call_expr, inargno);
1119  if (!OidIsValid(poly_actuals.anyarray_type))
1120  return false;
1121  }
1122  argtypes[i] = poly_actuals.anyarray_type;
1123  }
1124  break;
1125  case ANYRANGEOID:
1126  if (argmode == PROARGMODE_OUT || argmode == PROARGMODE_TABLE)
1127  {
1128  have_polymorphic_result = true;
1129  have_anyrange_result = true;
1130  }
1131  else
1132  {
1133  if (!OidIsValid(poly_actuals.anyrange_type))
1134  {
1135  poly_actuals.anyrange_type =
1136  get_call_expr_argtype(call_expr, inargno);
1137  if (!OidIsValid(poly_actuals.anyrange_type))
1138  return false;
1139  }
1140  argtypes[i] = poly_actuals.anyrange_type;
1141  }
1142  break;
1143  case ANYMULTIRANGEOID:
1144  if (argmode == PROARGMODE_OUT || argmode == PROARGMODE_TABLE)
1145  {
1146  have_polymorphic_result = true;
1147  have_anymultirange_result = true;
1148  }
1149  else
1150  {
1151  if (!OidIsValid(poly_actuals.anymultirange_type))
1152  {
1153  poly_actuals.anymultirange_type =
1154  get_call_expr_argtype(call_expr, inargno);
1155  if (!OidIsValid(poly_actuals.anymultirange_type))
1156  return false;
1157  }
1158  argtypes[i] = poly_actuals.anymultirange_type;
1159  }
1160  break;
1161  case ANYCOMPATIBLEOID:
1162  case ANYCOMPATIBLENONARRAYOID:
1163  if (argmode == PROARGMODE_OUT || argmode == PROARGMODE_TABLE)
1164  {
1165  have_polymorphic_result = true;
1166  have_anycompatible_result = true;
1167  }
1168  else
1169  {
1170  if (!OidIsValid(anyc_actuals.anyelement_type))
1171  {
1172  anyc_actuals.anyelement_type =
1173  get_call_expr_argtype(call_expr, inargno);
1174  if (!OidIsValid(anyc_actuals.anyelement_type))
1175  return false;
1176  }
1177  argtypes[i] = anyc_actuals.anyelement_type;
1178  }
1179  break;
1180  case ANYCOMPATIBLEARRAYOID:
1181  if (argmode == PROARGMODE_OUT || argmode == PROARGMODE_TABLE)
1182  {
1183  have_polymorphic_result = true;
1184  have_anycompatible_array_result = true;
1185  }
1186  else
1187  {
1188  if (!OidIsValid(anyc_actuals.anyarray_type))
1189  {
1190  anyc_actuals.anyarray_type =
1191  get_call_expr_argtype(call_expr, inargno);
1192  if (!OidIsValid(anyc_actuals.anyarray_type))
1193  return false;
1194  }
1195  argtypes[i] = anyc_actuals.anyarray_type;
1196  }
1197  break;
1198  case ANYCOMPATIBLERANGEOID:
1199  if (argmode == PROARGMODE_OUT || argmode == PROARGMODE_TABLE)
1200  {
1201  have_polymorphic_result = true;
1202  have_anycompatible_range_result = true;
1203  }
1204  else
1205  {
1206  if (!OidIsValid(anyc_actuals.anyrange_type))
1207  {
1208  anyc_actuals.anyrange_type =
1209  get_call_expr_argtype(call_expr, inargno);
1210  if (!OidIsValid(anyc_actuals.anyrange_type))
1211  return false;
1212  }
1213  argtypes[i] = anyc_actuals.anyrange_type;
1214  }
1215  break;
1216  case ANYCOMPATIBLEMULTIRANGEOID:
1217  if (argmode == PROARGMODE_OUT || argmode == PROARGMODE_TABLE)
1218  {
1219  have_polymorphic_result = true;
1220  have_anycompatible_multirange_result = true;
1221  }
1222  else
1223  {
1224  if (!OidIsValid(anyc_actuals.anymultirange_type))
1225  {
1226  anyc_actuals.anymultirange_type =
1227  get_call_expr_argtype(call_expr, inargno);
1228  if (!OidIsValid(anyc_actuals.anymultirange_type))
1229  return false;
1230  }
1231  argtypes[i] = anyc_actuals.anymultirange_type;
1232  }
1233  break;
1234  default:
1235  break;
1236  }
1237  if (argmode != PROARGMODE_OUT && argmode != PROARGMODE_TABLE)
1238  inargno++;
1239  }
1240 
1241  /* Done? */
1242  if (!have_polymorphic_result)
1243  return true;
1244 
1245  /* If needed, deduce one polymorphic type from others */
1246  if (have_anyelement_result && !OidIsValid(poly_actuals.anyelement_type))
1247  resolve_anyelement_from_others(&poly_actuals);
1248 
1249  if (have_anyarray_result && !OidIsValid(poly_actuals.anyarray_type))
1250  resolve_anyarray_from_others(&poly_actuals);
1251 
1252  if (have_anyrange_result && !OidIsValid(poly_actuals.anyrange_type))
1253  resolve_anyrange_from_others(&poly_actuals);
1254 
1255  if (have_anymultirange_result && !OidIsValid(poly_actuals.anymultirange_type))
1256  resolve_anymultirange_from_others(&poly_actuals);
1257 
1258  if (have_anycompatible_result && !OidIsValid(anyc_actuals.anyelement_type))
1259  resolve_anyelement_from_others(&anyc_actuals);
1260 
1261  if (have_anycompatible_array_result && !OidIsValid(anyc_actuals.anyarray_type))
1262  resolve_anyarray_from_others(&anyc_actuals);
1263 
1264  if (have_anycompatible_range_result && !OidIsValid(anyc_actuals.anyrange_type))
1265  resolve_anyrange_from_others(&anyc_actuals);
1266 
1267  if (have_anycompatible_multirange_result && !OidIsValid(anyc_actuals.anymultirange_type))
1268  resolve_anymultirange_from_others(&anyc_actuals);
1269 
1270  /* And finally replace the output column types as needed */
1271  for (i = 0; i < numargs; i++)
1272  {
1273  switch (argtypes[i])
1274  {
1275  case ANYELEMENTOID:
1276  case ANYNONARRAYOID:
1277  case ANYENUMOID:
1278  argtypes[i] = poly_actuals.anyelement_type;
1279  break;
1280  case ANYARRAYOID:
1281  argtypes[i] = poly_actuals.anyarray_type;
1282  break;
1283  case ANYRANGEOID:
1284  argtypes[i] = poly_actuals.anyrange_type;
1285  break;
1286  case ANYMULTIRANGEOID:
1287  argtypes[i] = poly_actuals.anymultirange_type;
1288  break;
1289  case ANYCOMPATIBLEOID:
1290  case ANYCOMPATIBLENONARRAYOID:
1291  argtypes[i] = anyc_actuals.anyelement_type;
1292  break;
1293  case ANYCOMPATIBLEARRAYOID:
1294  argtypes[i] = anyc_actuals.anyarray_type;
1295  break;
1296  case ANYCOMPATIBLERANGEOID:
1297  argtypes[i] = anyc_actuals.anyrange_type;
1298  break;
1299  case ANYCOMPATIBLEMULTIRANGEOID:
1300  argtypes[i] = anyc_actuals.anymultirange_type;
1301  break;
1302  default:
1303  break;
1304  }
1305  }
1306 
1307  return true;
1308 }
1309 
1310 /*
1311  * get_type_func_class
1312  * Given the type OID, obtain its TYPEFUNC classification.
1313  * Also, if it's a domain, return the base type OID.
1314  *
1315  * This is intended to centralize a bunch of formerly ad-hoc code for
1316  * classifying types. The categories used here are useful for deciding
1317  * how to handle functions returning the datatype.
1318  */
1319 static TypeFuncClass
1320 get_type_func_class(Oid typid, Oid *base_typeid)
1321 {
1322  *base_typeid = typid;
1323 
1324  switch (get_typtype(typid))
1325  {
1326  case TYPTYPE_COMPOSITE:
1327  return TYPEFUNC_COMPOSITE;
1328  case TYPTYPE_BASE:
1329  case TYPTYPE_ENUM:
1330  case TYPTYPE_RANGE:
1331  case TYPTYPE_MULTIRANGE:
1332  return TYPEFUNC_SCALAR;
1333  case TYPTYPE_DOMAIN:
1334  *base_typeid = typid = getBaseType(typid);
1335  if (get_typtype(typid) == TYPTYPE_COMPOSITE)
1337  else /* domain base type can't be a pseudotype */
1338  return TYPEFUNC_SCALAR;
1339  case TYPTYPE_PSEUDO:
1340  if (typid == RECORDOID)
1341  return TYPEFUNC_RECORD;
1342 
1343  /*
1344  * We treat VOID and CSTRING as legitimate scalar datatypes,
1345  * mostly for the convenience of the JDBC driver (which wants to
1346  * be able to do "SELECT * FROM foo()" for all legitimately
1347  * user-callable functions).
1348  */
1349  if (typid == VOIDOID || typid == CSTRINGOID)
1350  return TYPEFUNC_SCALAR;
1351  return TYPEFUNC_OTHER;
1352  }
1353  /* shouldn't get here, probably */
1354  return TYPEFUNC_OTHER;
1355 }
1356 
1357 
1358 /*
1359  * get_func_arg_info
1360  *
1361  * Fetch info about the argument types, names, and IN/OUT modes from the
1362  * pg_proc tuple. Return value is the total number of arguments.
1363  * Other results are palloc'd. *p_argtypes is always filled in, but
1364  * *p_argnames and *p_argmodes will be set NULL in the default cases
1365  * (no names, and all IN arguments, respectively).
1366  *
1367  * Note that this function simply fetches what is in the pg_proc tuple;
1368  * it doesn't do any interpretation of polymorphic types.
1369  */
1370 int
1372  Oid **p_argtypes, char ***p_argnames, char **p_argmodes)
1373 {
1374  Form_pg_proc procStruct = (Form_pg_proc) GETSTRUCT(procTup);
1375  Datum proallargtypes;
1376  Datum proargmodes;
1377  Datum proargnames;
1378  bool isNull;
1379  ArrayType *arr;
1380  int numargs;
1381  Datum *elems;
1382  int nelems;
1383  int i;
1384 
1385  /* First discover the total number of parameters and get their types */
1386  proallargtypes = SysCacheGetAttr(PROCOID, procTup,
1387  Anum_pg_proc_proallargtypes,
1388  &isNull);
1389  if (!isNull)
1390  {
1391  /*
1392  * We expect the arrays to be 1-D arrays of the right types; verify
1393  * that. For the OID and char arrays, we don't need to use
1394  * deconstruct_array() since the array data is just going to look like
1395  * a C array of values.
1396  */
1397  arr = DatumGetArrayTypeP(proallargtypes); /* ensure not toasted */
1398  numargs = ARR_DIMS(arr)[0];
1399  if (ARR_NDIM(arr) != 1 ||
1400  numargs < 0 ||
1401  ARR_HASNULL(arr) ||
1402  ARR_ELEMTYPE(arr) != OIDOID)
1403  elog(ERROR, "proallargtypes is not a 1-D Oid array or it contains nulls");
1404  Assert(numargs >= procStruct->pronargs);
1405  *p_argtypes = (Oid *) palloc(numargs * sizeof(Oid));
1406  memcpy(*p_argtypes, ARR_DATA_PTR(arr),
1407  numargs * sizeof(Oid));
1408  }
1409  else
1410  {
1411  /* If no proallargtypes, use proargtypes */
1412  numargs = procStruct->proargtypes.dim1;
1413  Assert(numargs == procStruct->pronargs);
1414  *p_argtypes = (Oid *) palloc(numargs * sizeof(Oid));
1415  memcpy(*p_argtypes, procStruct->proargtypes.values,
1416  numargs * sizeof(Oid));
1417  }
1418 
1419  /* Get argument names, if available */
1420  proargnames = SysCacheGetAttr(PROCOID, procTup,
1421  Anum_pg_proc_proargnames,
1422  &isNull);
1423  if (isNull)
1424  *p_argnames = NULL;
1425  else
1426  {
1427  deconstruct_array_builtin(DatumGetArrayTypeP(proargnames), TEXTOID,
1428  &elems, NULL, &nelems);
1429  if (nelems != numargs) /* should not happen */
1430  elog(ERROR, "proargnames must have the same number of elements as the function has arguments");
1431  *p_argnames = (char **) palloc(sizeof(char *) * numargs);
1432  for (i = 0; i < numargs; i++)
1433  (*p_argnames)[i] = TextDatumGetCString(elems[i]);
1434  }
1435 
1436  /* Get argument modes, if available */
1437  proargmodes = SysCacheGetAttr(PROCOID, procTup,
1438  Anum_pg_proc_proargmodes,
1439  &isNull);
1440  if (isNull)
1441  *p_argmodes = NULL;
1442  else
1443  {
1444  arr = DatumGetArrayTypeP(proargmodes); /* ensure not toasted */
1445  if (ARR_NDIM(arr) != 1 ||
1446  ARR_DIMS(arr)[0] != numargs ||
1447  ARR_HASNULL(arr) ||
1448  ARR_ELEMTYPE(arr) != CHAROID)
1449  elog(ERROR, "proargmodes is not a 1-D char array of length %d or it contains nulls",
1450  numargs);
1451  *p_argmodes = (char *) palloc(numargs * sizeof(char));
1452  memcpy(*p_argmodes, ARR_DATA_PTR(arr),
1453  numargs * sizeof(char));
1454  }
1455 
1456  return numargs;
1457 }
1458 
1459 /*
1460  * get_func_trftypes
1461  *
1462  * Returns the number of transformed types used by the function.
1463  * If there are any, a palloc'd array of the type OIDs is returned
1464  * into *p_trftypes.
1465  */
1466 int
1468  Oid **p_trftypes)
1469 {
1470  Datum protrftypes;
1471  ArrayType *arr;
1472  int nelems;
1473  bool isNull;
1474 
1475  protrftypes = SysCacheGetAttr(PROCOID, procTup,
1476  Anum_pg_proc_protrftypes,
1477  &isNull);
1478  if (!isNull)
1479  {
1480  /*
1481  * We expect the arrays to be 1-D arrays of the right types; verify
1482  * that. For the OID and char arrays, we don't need to use
1483  * deconstruct_array() since the array data is just going to look like
1484  * a C array of values.
1485  */
1486  arr = DatumGetArrayTypeP(protrftypes); /* ensure not toasted */
1487  nelems = ARR_DIMS(arr)[0];
1488  if (ARR_NDIM(arr) != 1 ||
1489  nelems < 0 ||
1490  ARR_HASNULL(arr) ||
1491  ARR_ELEMTYPE(arr) != OIDOID)
1492  elog(ERROR, "protrftypes is not a 1-D Oid array or it contains nulls");
1493  *p_trftypes = (Oid *) palloc(nelems * sizeof(Oid));
1494  memcpy(*p_trftypes, ARR_DATA_PTR(arr),
1495  nelems * sizeof(Oid));
1496 
1497  return nelems;
1498  }
1499  else
1500  return 0;
1501 }
1502 
1503 /*
1504  * get_func_input_arg_names
1505  *
1506  * Extract the names of input arguments only, given a function's
1507  * proargnames and proargmodes entries in Datum form.
1508  *
1509  * Returns the number of input arguments, which is the length of the
1510  * palloc'd array returned to *arg_names. Entries for unnamed args
1511  * are set to NULL. You don't get anything if proargnames is NULL.
1512  */
1513 int
1514 get_func_input_arg_names(Datum proargnames, Datum proargmodes,
1515  char ***arg_names)
1516 {
1517  ArrayType *arr;
1518  int numargs;
1519  Datum *argnames;
1520  char *argmodes;
1521  char **inargnames;
1522  int numinargs;
1523  int i;
1524 
1525  /* Do nothing if null proargnames */
1526  if (proargnames == PointerGetDatum(NULL))
1527  {
1528  *arg_names = NULL;
1529  return 0;
1530  }
1531 
1532  /*
1533  * We expect the arrays to be 1-D arrays of the right types; verify that.
1534  * For proargmodes, we don't need to use deconstruct_array() since the
1535  * array data is just going to look like a C array of values.
1536  */
1537  arr = DatumGetArrayTypeP(proargnames); /* ensure not toasted */
1538  if (ARR_NDIM(arr) != 1 ||
1539  ARR_HASNULL(arr) ||
1540  ARR_ELEMTYPE(arr) != TEXTOID)
1541  elog(ERROR, "proargnames is not a 1-D text array or it contains nulls");
1542  deconstruct_array_builtin(arr, TEXTOID, &argnames, NULL, &numargs);
1543  if (proargmodes != PointerGetDatum(NULL))
1544  {
1545  arr = DatumGetArrayTypeP(proargmodes); /* ensure not toasted */
1546  if (ARR_NDIM(arr) != 1 ||
1547  ARR_DIMS(arr)[0] != numargs ||
1548  ARR_HASNULL(arr) ||
1549  ARR_ELEMTYPE(arr) != CHAROID)
1550  elog(ERROR, "proargmodes is not a 1-D char array of length %d or it contains nulls",
1551  numargs);
1552  argmodes = (char *) ARR_DATA_PTR(arr);
1553  }
1554  else
1555  argmodes = NULL;
1556 
1557  /* zero elements probably shouldn't happen, but handle it gracefully */
1558  if (numargs <= 0)
1559  {
1560  *arg_names = NULL;
1561  return 0;
1562  }
1563 
1564  /* extract input-argument names */
1565  inargnames = (char **) palloc(numargs * sizeof(char *));
1566  numinargs = 0;
1567  for (i = 0; i < numargs; i++)
1568  {
1569  if (argmodes == NULL ||
1570  argmodes[i] == PROARGMODE_IN ||
1571  argmodes[i] == PROARGMODE_INOUT ||
1572  argmodes[i] == PROARGMODE_VARIADIC)
1573  {
1574  char *pname = TextDatumGetCString(argnames[i]);
1575 
1576  if (pname[0] != '\0')
1577  inargnames[numinargs] = pname;
1578  else
1579  inargnames[numinargs] = NULL;
1580  numinargs++;
1581  }
1582  }
1583 
1584  *arg_names = inargnames;
1585  return numinargs;
1586 }
1587 
1588 
1589 /*
1590  * get_func_result_name
1591  *
1592  * If the function has exactly one output parameter, and that parameter
1593  * is named, return the name (as a palloc'd string). Else return NULL.
1594  *
1595  * This is used to determine the default output column name for functions
1596  * returning scalar types.
1597  */
1598 char *
1600 {
1601  char *result;
1602  HeapTuple procTuple;
1603  Datum proargmodes;
1604  Datum proargnames;
1605  bool isnull;
1606  ArrayType *arr;
1607  int numargs;
1608  char *argmodes;
1609  Datum *argnames;
1610  int numoutargs;
1611  int nargnames;
1612  int i;
1613 
1614  /* First fetch the function's pg_proc row */
1615  procTuple = SearchSysCache1(PROCOID, ObjectIdGetDatum(functionId));
1616  if (!HeapTupleIsValid(procTuple))
1617  elog(ERROR, "cache lookup failed for function %u", functionId);
1618 
1619  /* If there are no named OUT parameters, return NULL */
1620  if (heap_attisnull(procTuple, Anum_pg_proc_proargmodes, NULL) ||
1621  heap_attisnull(procTuple, Anum_pg_proc_proargnames, NULL))
1622  result = NULL;
1623  else
1624  {
1625  /* Get the data out of the tuple */
1626  proargmodes = SysCacheGetAttr(PROCOID, procTuple,
1627  Anum_pg_proc_proargmodes,
1628  &isnull);
1629  Assert(!isnull);
1630  proargnames = SysCacheGetAttr(PROCOID, procTuple,
1631  Anum_pg_proc_proargnames,
1632  &isnull);
1633  Assert(!isnull);
1634 
1635  /*
1636  * We expect the arrays to be 1-D arrays of the right types; verify
1637  * that. For the char array, we don't need to use deconstruct_array()
1638  * since the array data is just going to look like a C array of
1639  * values.
1640  */
1641  arr = DatumGetArrayTypeP(proargmodes); /* ensure not toasted */
1642  numargs = ARR_DIMS(arr)[0];
1643  if (ARR_NDIM(arr) != 1 ||
1644  numargs < 0 ||
1645  ARR_HASNULL(arr) ||
1646  ARR_ELEMTYPE(arr) != CHAROID)
1647  elog(ERROR, "proargmodes is not a 1-D char array or it contains nulls");
1648  argmodes = (char *) ARR_DATA_PTR(arr);
1649  arr = DatumGetArrayTypeP(proargnames); /* ensure not toasted */
1650  if (ARR_NDIM(arr) != 1 ||
1651  ARR_DIMS(arr)[0] != numargs ||
1652  ARR_HASNULL(arr) ||
1653  ARR_ELEMTYPE(arr) != TEXTOID)
1654  elog(ERROR, "proargnames is not a 1-D text array of length %d or it contains nulls",
1655  numargs);
1656  deconstruct_array_builtin(arr, TEXTOID, &argnames, NULL, &nargnames);
1657  Assert(nargnames == numargs);
1658 
1659  /* scan for output argument(s) */
1660  result = NULL;
1661  numoutargs = 0;
1662  for (i = 0; i < numargs; i++)
1663  {
1664  if (argmodes[i] == PROARGMODE_IN ||
1665  argmodes[i] == PROARGMODE_VARIADIC)
1666  continue;
1667  Assert(argmodes[i] == PROARGMODE_OUT ||
1668  argmodes[i] == PROARGMODE_INOUT ||
1669  argmodes[i] == PROARGMODE_TABLE);
1670  if (++numoutargs > 1)
1671  {
1672  /* multiple out args, so forget it */
1673  result = NULL;
1674  break;
1675  }
1676  result = TextDatumGetCString(argnames[i]);
1677  if (result == NULL || result[0] == '\0')
1678  {
1679  /* Parameter is not named, so forget it */
1680  result = NULL;
1681  break;
1682  }
1683  }
1684  }
1685 
1686  ReleaseSysCache(procTuple);
1687 
1688  return result;
1689 }
1690 
1691 
1692 /*
1693  * build_function_result_tupdesc_t
1694  *
1695  * Given a pg_proc row for a function, return a tuple descriptor for the
1696  * result rowtype, or NULL if the function does not have OUT parameters.
1697  *
1698  * Note that this does not handle resolution of polymorphic types;
1699  * that is deliberate.
1700  */
1701 TupleDesc
1703 {
1704  Form_pg_proc procform = (Form_pg_proc) GETSTRUCT(procTuple);
1705  Datum proallargtypes;
1706  Datum proargmodes;
1707  Datum proargnames;
1708  bool isnull;
1709 
1710  /* Return NULL if the function isn't declared to return RECORD */
1711  if (procform->prorettype != RECORDOID)
1712  return NULL;
1713 
1714  /* If there are no OUT parameters, return NULL */
1715  if (heap_attisnull(procTuple, Anum_pg_proc_proallargtypes, NULL) ||
1716  heap_attisnull(procTuple, Anum_pg_proc_proargmodes, NULL))
1717  return NULL;
1718 
1719  /* Get the data out of the tuple */
1720  proallargtypes = SysCacheGetAttr(PROCOID, procTuple,
1721  Anum_pg_proc_proallargtypes,
1722  &isnull);
1723  Assert(!isnull);
1724  proargmodes = SysCacheGetAttr(PROCOID, procTuple,
1725  Anum_pg_proc_proargmodes,
1726  &isnull);
1727  Assert(!isnull);
1728  proargnames = SysCacheGetAttr(PROCOID, procTuple,
1729  Anum_pg_proc_proargnames,
1730  &isnull);
1731  if (isnull)
1732  proargnames = PointerGetDatum(NULL); /* just to be sure */
1733 
1734  return build_function_result_tupdesc_d(procform->prokind,
1735  proallargtypes,
1736  proargmodes,
1737  proargnames);
1738 }
1739 
1740 /*
1741  * build_function_result_tupdesc_d
1742  *
1743  * Build a RECORD function's tupledesc from the pg_proc proallargtypes,
1744  * proargmodes, and proargnames arrays. This is split out for the
1745  * convenience of ProcedureCreate, which needs to be able to compute the
1746  * tupledesc before actually creating the function.
1747  *
1748  * For functions (but not for procedures), returns NULL if there are not at
1749  * least two OUT or INOUT arguments.
1750  */
1751 TupleDesc
1753  Datum proallargtypes,
1754  Datum proargmodes,
1755  Datum proargnames)
1756 {
1757  TupleDesc desc;
1758  ArrayType *arr;
1759  int numargs;
1760  Oid *argtypes;
1761  char *argmodes;
1762  Datum *argnames = NULL;
1763  Oid *outargtypes;
1764  char **outargnames;
1765  int numoutargs;
1766  int nargnames;
1767  int i;
1768 
1769  /* Can't have output args if columns are null */
1770  if (proallargtypes == PointerGetDatum(NULL) ||
1771  proargmodes == PointerGetDatum(NULL))
1772  return NULL;
1773 
1774  /*
1775  * We expect the arrays to be 1-D arrays of the right types; verify that.
1776  * For the OID and char arrays, we don't need to use deconstruct_array()
1777  * since the array data is just going to look like a C array of values.
1778  */
1779  arr = DatumGetArrayTypeP(proallargtypes); /* ensure not toasted */
1780  numargs = ARR_DIMS(arr)[0];
1781  if (ARR_NDIM(arr) != 1 ||
1782  numargs < 0 ||
1783  ARR_HASNULL(arr) ||
1784  ARR_ELEMTYPE(arr) != OIDOID)
1785  elog(ERROR, "proallargtypes is not a 1-D Oid array or it contains nulls");
1786  argtypes = (Oid *) ARR_DATA_PTR(arr);
1787  arr = DatumGetArrayTypeP(proargmodes); /* ensure not toasted */
1788  if (ARR_NDIM(arr) != 1 ||
1789  ARR_DIMS(arr)[0] != numargs ||
1790  ARR_HASNULL(arr) ||
1791  ARR_ELEMTYPE(arr) != CHAROID)
1792  elog(ERROR, "proargmodes is not a 1-D char array of length %d or it contains nulls",
1793  numargs);
1794  argmodes = (char *) ARR_DATA_PTR(arr);
1795  if (proargnames != PointerGetDatum(NULL))
1796  {
1797  arr = DatumGetArrayTypeP(proargnames); /* ensure not toasted */
1798  if (ARR_NDIM(arr) != 1 ||
1799  ARR_DIMS(arr)[0] != numargs ||
1800  ARR_HASNULL(arr) ||
1801  ARR_ELEMTYPE(arr) != TEXTOID)
1802  elog(ERROR, "proargnames is not a 1-D text array of length %d or it contains nulls",
1803  numargs);
1804  deconstruct_array_builtin(arr, TEXTOID, &argnames, NULL, &nargnames);
1805  Assert(nargnames == numargs);
1806  }
1807 
1808  /* zero elements probably shouldn't happen, but handle it gracefully */
1809  if (numargs <= 0)
1810  return NULL;
1811 
1812  /* extract output-argument types and names */
1813  outargtypes = (Oid *) palloc(numargs * sizeof(Oid));
1814  outargnames = (char **) palloc(numargs * sizeof(char *));
1815  numoutargs = 0;
1816  for (i = 0; i < numargs; i++)
1817  {
1818  char *pname;
1819 
1820  if (argmodes[i] == PROARGMODE_IN ||
1821  argmodes[i] == PROARGMODE_VARIADIC)
1822  continue;
1823  Assert(argmodes[i] == PROARGMODE_OUT ||
1824  argmodes[i] == PROARGMODE_INOUT ||
1825  argmodes[i] == PROARGMODE_TABLE);
1826  outargtypes[numoutargs] = argtypes[i];
1827  if (argnames)
1828  pname = TextDatumGetCString(argnames[i]);
1829  else
1830  pname = NULL;
1831  if (pname == NULL || pname[0] == '\0')
1832  {
1833  /* Parameter is not named, so gin up a column name */
1834  pname = psprintf("column%d", numoutargs + 1);
1835  }
1836  outargnames[numoutargs] = pname;
1837  numoutargs++;
1838  }
1839 
1840  /*
1841  * If there is no output argument, or only one, the function does not
1842  * return tuples.
1843  */
1844  if (numoutargs < 2 && prokind != PROKIND_PROCEDURE)
1845  return NULL;
1846 
1847  desc = CreateTemplateTupleDesc(numoutargs);
1848  for (i = 0; i < numoutargs; i++)
1849  {
1850  TupleDescInitEntry(desc, i + 1,
1851  outargnames[i],
1852  outargtypes[i],
1853  -1,
1854  0);
1855  }
1856 
1857  return desc;
1858 }
1859 
1860 
1861 /*
1862  * RelationNameGetTupleDesc
1863  *
1864  * Given a (possibly qualified) relation name, build a TupleDesc.
1865  *
1866  * Note: while this works as advertised, it's seldom the best way to
1867  * build a tupdesc for a function's result type. It's kept around
1868  * only for backwards compatibility with existing user-written code.
1869  */
1870 TupleDesc
1872 {
1873  RangeVar *relvar;
1874  Relation rel;
1875  TupleDesc tupdesc;
1876  List *relname_list;
1877 
1878  /* Open relation and copy the tuple description */
1879  relname_list = stringToQualifiedNameList(relname);
1880  relvar = makeRangeVarFromNameList(relname_list);
1881  rel = relation_openrv(relvar, AccessShareLock);
1882  tupdesc = CreateTupleDescCopy(RelationGetDescr(rel));
1884 
1885  return tupdesc;
1886 }
1887 
1888 /*
1889  * TypeGetTupleDesc
1890  *
1891  * Given a type Oid, build a TupleDesc. (In most cases you should be
1892  * using get_call_result_type or one of its siblings instead of this
1893  * routine, so that you can handle OUT parameters, RECORD result type,
1894  * and polymorphic results.)
1895  *
1896  * If the type is composite, *and* a colaliases List is provided, *and*
1897  * the List is of natts length, use the aliases instead of the relation
1898  * attnames. (NB: this usage is deprecated since it may result in
1899  * creation of unnecessary transient record types.)
1900  *
1901  * If the type is a base type, a single item alias List is required.
1902  */
1903 TupleDesc
1904 TypeGetTupleDesc(Oid typeoid, List *colaliases)
1905 {
1906  Oid base_typeoid;
1907  TypeFuncClass functypclass = get_type_func_class(typeoid, &base_typeoid);
1908  TupleDesc tupdesc = NULL;
1909 
1910  /*
1911  * Build a suitable tupledesc representing the output rows. We
1912  * intentionally do not support TYPEFUNC_COMPOSITE_DOMAIN here, as it's
1913  * unlikely that legacy callers of this obsolete function would be
1914  * prepared to apply domain constraints.
1915  */
1916  if (functypclass == TYPEFUNC_COMPOSITE)
1917  {
1918  /* Composite data type, e.g. a table's row type */
1919  tupdesc = lookup_rowtype_tupdesc_copy(base_typeoid, -1);
1920 
1921  if (colaliases != NIL)
1922  {
1923  int natts = tupdesc->natts;
1924  int varattno;
1925 
1926  /* does the list length match the number of attributes? */
1927  if (list_length(colaliases) != natts)
1928  ereport(ERROR,
1929  (errcode(ERRCODE_DATATYPE_MISMATCH),
1930  errmsg("number of aliases does not match number of columns")));
1931 
1932  /* OK, use the aliases instead */
1933  for (varattno = 0; varattno < natts; varattno++)
1934  {
1935  char *label = strVal(list_nth(colaliases, varattno));
1936  Form_pg_attribute attr = TupleDescAttr(tupdesc, varattno);
1937 
1938  if (label != NULL)
1939  namestrcpy(&(attr->attname), label);
1940  }
1941 
1942  /* The tuple type is now an anonymous record type */
1943  tupdesc->tdtypeid = RECORDOID;
1944  tupdesc->tdtypmod = -1;
1945  }
1946  }
1947  else if (functypclass == TYPEFUNC_SCALAR)
1948  {
1949  /* Base data type, i.e. scalar */
1950  char *attname;
1951 
1952  /* the alias list is required for base types */
1953  if (colaliases == NIL)
1954  ereport(ERROR,
1955  (errcode(ERRCODE_DATATYPE_MISMATCH),
1956  errmsg("no column alias was provided")));
1957 
1958  /* the alias list length must be 1 */
1959  if (list_length(colaliases) != 1)
1960  ereport(ERROR,
1961  (errcode(ERRCODE_DATATYPE_MISMATCH),
1962  errmsg("number of aliases does not match number of columns")));
1963 
1964  /* OK, get the column alias */
1965  attname = strVal(linitial(colaliases));
1966 
1967  tupdesc = CreateTemplateTupleDesc(1);
1968  TupleDescInitEntry(tupdesc,
1969  (AttrNumber) 1,
1970  attname,
1971  typeoid,
1972  -1,
1973  0);
1974  }
1975  else if (functypclass == TYPEFUNC_RECORD)
1976  {
1977  /* XXX can't support this because typmod wasn't passed in ... */
1978  ereport(ERROR,
1979  (errcode(ERRCODE_DATATYPE_MISMATCH),
1980  errmsg("could not determine row description for function returning record")));
1981  }
1982  else
1983  {
1984  /* crummy error message, but parser should have caught this */
1985  elog(ERROR, "function in FROM has unsupported return type");
1986  }
1987 
1988  return tupdesc;
1989 }
1990 
1991 /*
1992  * extract_variadic_args
1993  *
1994  * Extract a set of argument values, types and NULL markers for a given
1995  * input function which makes use of a VARIADIC input whose argument list
1996  * depends on the caller context. When doing a VARIADIC call, the caller
1997  * has provided one argument made of an array of values, so deconstruct the
1998  * array data before using it for the next processing. If no VARIADIC call
1999  * is used, just fill in the status data based on all the arguments given
2000  * by the caller.
2001  *
2002  * This function returns the number of arguments generated, or -1 in the
2003  * case of "VARIADIC NULL".
2004  */
2005 int
2006 extract_variadic_args(FunctionCallInfo fcinfo, int variadic_start,
2007  bool convert_unknown, Datum **args, Oid **types,
2008  bool **nulls)
2009 {
2010  bool variadic = get_fn_expr_variadic(fcinfo->flinfo);
2011  Datum *args_res;
2012  bool *nulls_res;
2013  Oid *types_res;
2014  int nargs,
2015  i;
2016 
2017  *args = NULL;
2018  *types = NULL;
2019  *nulls = NULL;
2020 
2021  if (variadic)
2022  {
2024  Oid element_type;
2025  bool typbyval;
2026  char typalign;
2027  int16 typlen;
2028 
2029  Assert(PG_NARGS() == variadic_start + 1);
2030 
2031  if (PG_ARGISNULL(variadic_start))
2032  return -1;
2033 
2034  array_in = PG_GETARG_ARRAYTYPE_P(variadic_start);
2035  element_type = ARR_ELEMTYPE(array_in);
2036 
2037  get_typlenbyvalalign(element_type,
2038  &typlen, &typbyval, &typalign);
2039  deconstruct_array(array_in, element_type, typlen, typbyval,
2040  typalign, &args_res, &nulls_res,
2041  &nargs);
2042 
2043  /* All the elements of the array have the same type */
2044  types_res = (Oid *) palloc0(nargs * sizeof(Oid));
2045  for (i = 0; i < nargs; i++)
2046  types_res[i] = element_type;
2047  }
2048  else
2049  {
2050  nargs = PG_NARGS() - variadic_start;
2051  Assert(nargs > 0);
2052  nulls_res = (bool *) palloc0(nargs * sizeof(bool));
2053  args_res = (Datum *) palloc0(nargs * sizeof(Datum));
2054  types_res = (Oid *) palloc0(nargs * sizeof(Oid));
2055 
2056  for (i = 0; i < nargs; i++)
2057  {
2058  nulls_res[i] = PG_ARGISNULL(i + variadic_start);
2059  types_res[i] = get_fn_expr_argtype(fcinfo->flinfo,
2060  i + variadic_start);
2061 
2062  /*
2063  * Turn a constant (more or less literal) value that's of unknown
2064  * type into text if required. Unknowns come in as a cstring
2065  * pointer. Note: for functions declared as taking type "any", the
2066  * parser will not do any type conversion on unknown-type literals
2067  * (that is, undecorated strings or NULLs).
2068  */
2069  if (convert_unknown &&
2070  types_res[i] == UNKNOWNOID &&
2071  get_fn_expr_arg_stable(fcinfo->flinfo, i + variadic_start))
2072  {
2073  types_res[i] = TEXTOID;
2074 
2075  if (PG_ARGISNULL(i + variadic_start))
2076  args_res[i] = (Datum) 0;
2077  else
2078  args_res[i] =
2079  CStringGetTextDatum(PG_GETARG_POINTER(i + variadic_start));
2080  }
2081  else
2082  {
2083  /* no conversion needed, just take the datum as given */
2084  args_res[i] = PG_GETARG_DATUM(i + variadic_start);
2085  }
2086 
2087  if (!OidIsValid(types_res[i]) ||
2088  (convert_unknown && types_res[i] == UNKNOWNOID))
2089  ereport(ERROR,
2090  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
2091  errmsg("could not determine data type for argument %d",
2092  i + 1)));
2093  }
2094  }
2095 
2096  /* Fill in results */
2097  *args = args_res;
2098  *nulls = nulls_res;
2099  *types = types_res;
2100 
2101  return nargs;
2102 }
#define ARR_NDIM(a)
Definition: array.h:283
#define PG_GETARG_ARRAYTYPE_P(n)
Definition: array.h:256
#define ARR_DATA_PTR(a)
Definition: array.h:315
#define DatumGetArrayTypeP(X)
Definition: array.h:254
#define ARR_ELEMTYPE(a)
Definition: array.h:285
#define ARR_DIMS(a)
Definition: array.h:287
#define ARR_HASNULL(a)
Definition: array.h:284
void deconstruct_array(ArrayType *array, Oid elmtype, int elmlen, bool elmbyval, char elmalign, Datum **elemsp, bool **nullsp, int *nelemsp)
Definition: arrayfuncs.c:3602
void deconstruct_array_builtin(ArrayType *array, Oid elmtype, Datum **elemsp, bool **nullsp, int *nelemsp)
Definition: arrayfuncs.c:3668
Datum array_in(PG_FUNCTION_ARGS)
Definition: arrayfuncs.c:175
int16 AttrNumber
Definition: attnum.h:21
#define CStringGetTextDatum(s)
Definition: builtins.h:88
#define TextDatumGetCString(d)
Definition: builtins.h:89
#define NameStr(name)
Definition: c.h:682
signed short int16
Definition: c.h:429
signed int int32
Definition: c.h:430
uint32 bits32
Definition: c.h:451
#define OidIsValid(objectId)
Definition: c.h:711
struct typedefs * types
Definition: ecpg.c:29
int errcode(int sqlerrcode)
Definition: elog.c:858
int errmsg(const char *fmt,...)
Definition: elog.c:1069
#define ERROR
Definition: elog.h:39
#define ereport(elevel,...)
Definition: elog.h:149
TupleDesc BlessTupleDesc(TupleDesc tupdesc)
Definition: execTuples.c:2071
void UnregisterExprContextCallback(ExprContext *econtext, ExprContextCallbackFunction function, Datum arg)
Definition: execUtils.c:957
void RegisterExprContextCallback(ExprContext *econtext, ExprContextCallbackFunction function, Datum arg)
Definition: execUtils.c:931
@ SFRM_Materialize_Random
Definition: execnodes.h:310
@ SFRM_Materialize
Definition: execnodes.h:309
bool get_fn_expr_arg_stable(FmgrInfo *flinfo, int argnum)
Definition: fmgr.c:1917
bool get_fn_expr_variadic(FmgrInfo *flinfo)
Definition: fmgr.c:1986
Oid get_call_expr_argtype(Node *expr, int argnum)
Definition: fmgr.c:1871
Oid get_fn_expr_argtype(FmgrInfo *flinfo, int argnum)
Definition: fmgr.c:1852
#define DatumGetHeapTupleHeader(X)
Definition: fmgr.h:295
#define PG_GETARG_POINTER(n)
Definition: fmgr.h:276
#define PG_ARGISNULL(n)
Definition: fmgr.h:209
#define PG_GETARG_DATUM(n)
Definition: fmgr.h:268
#define PG_NARGS()
Definition: fmgr.h:203
#define PG_FUNCTION_ARGS
Definition: fmgr.h:193
char * format_type_be(Oid type_oid)
Definition: format_type.c:339
static TypeFuncClass get_type_func_class(Oid typid, Oid *base_typeid)
Definition: funcapi.c:1320
struct polymorphic_actuals polymorphic_actuals
void InitMaterializedSRF(FunctionCallInfo fcinfo, bits32 flags)
Definition: funcapi.c:76
static TypeFuncClass internal_get_result_type(Oid funcid, Node *call_expr, ReturnSetInfo *rsinfo, Oid *resultTypeId, TupleDesc *resultTupleDesc)
Definition: funcapi.c:423
int get_func_trftypes(HeapTuple procTup, Oid **p_trftypes)
Definition: funcapi.c:1467
TupleDesc build_function_result_tupdesc_t(HeapTuple procTuple)
Definition: funcapi.c:1702
TupleDesc build_function_result_tupdesc_d(char prokind, Datum proallargtypes, Datum proargmodes, Datum proargnames)
Definition: funcapi.c:1752
bool resolve_polymorphic_argtypes(int numargs, Oid *argtypes, char *argmodes, Node *call_expr)
Definition: funcapi.c:1056
char * get_func_result_name(Oid functionId)
Definition: funcapi.c:1599
void end_MultiFuncCall(PG_FUNCTION_ARGS, FuncCallContext *funcctx)
Definition: funcapi.c:220
static void resolve_anyelement_from_others(polymorphic_actuals *actuals)
Definition: funcapi.c:581
int get_func_input_arg_names(Datum proargnames, Datum proargmodes, char ***arg_names)
Definition: funcapi.c:1514
int get_func_arg_info(HeapTuple procTup, Oid **p_argtypes, char ***p_argnames, char **p_argmodes)
Definition: funcapi.c:1371
static bool resolve_polymorphic_tupdesc(TupleDesc tupdesc, oidvector *declared_args, Node *call_expr)
Definition: funcapi.c:736
TupleDesc RelationNameGetTupleDesc(const char *relname)
Definition: funcapi.c:1871
TypeFuncClass get_call_result_type(FunctionCallInfo fcinfo, Oid *resultTypeId, TupleDesc *resultTupleDesc)
Definition: funcapi.c:276
int extract_variadic_args(FunctionCallInfo fcinfo, int variadic_start, bool convert_unknown, Datum **args, Oid **types, bool **nulls)
Definition: funcapi.c:2006
FuncCallContext * init_MultiFuncCall(PG_FUNCTION_ARGS)
Definition: funcapi.c:133
static void resolve_anyrange_from_others(polymorphic_actuals *actuals)
Definition: funcapi.c:673
static void shutdown_MultiFuncCall(Datum arg)
Definition: funcapi.c:238
TupleDesc get_expr_result_tupdesc(Node *expr, bool noError)
Definition: funcapi.c:543
static void resolve_anymultirange_from_others(polymorphic_actuals *actuals)
Definition: funcapi.c:702
TupleDesc TypeGetTupleDesc(Oid typeoid, List *colaliases)
Definition: funcapi.c:1904
TypeFuncClass get_func_result_type(Oid functionId, Oid *resultTypeId, TupleDesc *resultTupleDesc)
Definition: funcapi.c:403
static void resolve_anyarray_from_others(polymorphic_actuals *actuals)
Definition: funcapi.c:647
TypeFuncClass get_expr_result_type(Node *expr, Oid *resultTypeId, TupleDesc *resultTupleDesc)
Definition: funcapi.c:292
FuncCallContext * per_MultiFuncCall(PG_FUNCTION_ARGS)
Definition: funcapi.c:208
TypeFuncClass
Definition: funcapi.h:147
@ TYPEFUNC_SCALAR
Definition: funcapi.h:148
@ TYPEFUNC_COMPOSITE
Definition: funcapi.h:149
@ TYPEFUNC_RECORD
Definition: funcapi.h:151
@ TYPEFUNC_COMPOSITE_DOMAIN
Definition: funcapi.h:150
@ TYPEFUNC_OTHER
Definition: funcapi.h:152
#define MAT_SRF_BLESS
Definition: funcapi.h:296
#define MAT_SRF_USE_EXPECTED_DESC
Definition: funcapi.h:295
int work_mem
Definition: globals.c:125
bool heap_attisnull(HeapTuple tup, int attnum, TupleDesc tupleDesc)
Definition: heaptuple.c:359
#define HeapTupleIsValid(tuple)
Definition: htup.h:78
#define HeapTupleHeaderGetTypMod(tup)
Definition: htup_details.h:462
#define HeapTupleHeaderGetTypeId(tup)
Definition: htup_details.h:452
#define GETSTRUCT(TUP)
Definition: htup_details.h:649
int i
Definition: isn.c:73
if(TABLE==NULL||TABLE_index==NULL)
Definition: isn.c:77
Assert(fmt[strlen(fmt) - 1] !='\n')
#define AccessShareLock
Definition: lockdefs.h:36
Oid get_range_subtype(Oid rangeOid)
Definition: lsyscache.c:3372
Oid get_element_type(Oid typid)
Definition: lsyscache.c:2717
Oid get_multirange_range(Oid multirangeOid)
Definition: lsyscache.c:3448
void get_typlenbyvalalign(Oid typid, int16 *typlen, bool *typbyval, char *typalign)
Definition: lsyscache.c:2229
RegProcedure get_opcode(Oid opno)
Definition: lsyscache.c:1267
Oid get_typcollation(Oid typid)
Definition: lsyscache.c:3014
Oid get_range_multirange(Oid rangeOid)
Definition: lsyscache.c:3423
char get_typtype(Oid typid)
Definition: lsyscache.c:2587
Oid getBaseType(Oid typid)
Definition: lsyscache.c:2479
Oid get_array_type(Oid typid)
Definition: lsyscache.c:2745
void * palloc0(Size size)
Definition: mcxt.c:1230
void * MemoryContextAllocZero(MemoryContext context, Size size)
Definition: mcxt.c:1037
void MemoryContextDelete(MemoryContext context)
Definition: mcxt.c:376
void * palloc(Size size)
Definition: mcxt.c:1199
#define AllocSetContextCreate
Definition: memutils.h:129
#define ALLOCSET_SMALL_SIZES
Definition: memutils.h:163
void namestrcpy(Name name, const char *str)
Definition: name.c:233
RangeVar * makeRangeVarFromNameList(List *names)
Definition: namespace.c:3105
Oid exprType(const Node *expr)
Definition: nodeFuncs.c:43
Oid exprInputCollation(const Node *expr)
Definition: nodeFuncs.c:972
int32 exprTypmod(const Node *expr)
Definition: nodeFuncs.c:266
Oid exprCollation(const Node *expr)
Definition: nodeFuncs.c:764
#define IsA(nodeptr, _type_)
Definition: nodes.h:168
static MemoryContext MemoryContextSwitchTo(MemoryContext context)
Definition: palloc.h:135
NameData attname
Definition: pg_attribute.h:41
FormData_pg_attribute * Form_pg_attribute
Definition: pg_attribute.h:207
void * arg
static char * label
NameData relname
Definition: pg_class.h:38
#define lfirst(lc)
Definition: pg_list.h:170
static int list_length(const List *l)
Definition: pg_list.h:150
#define NIL
Definition: pg_list.h:66
#define forboth(cell1, list1, cell2, list2)
Definition: pg_list.h:465
#define linitial(l)
Definition: pg_list.h:176
static void * list_nth(const List *list, int n)
Definition: pg_list.h:297
FormData_pg_proc * Form_pg_proc
Definition: pg_proc.h:136
char typalign
Definition: pg_type.h:176
static Datum PointerGetDatum(const void *X)
Definition: postgres.h:670
uintptr_t Datum
Definition: postgres.h:412
static Datum ObjectIdGetDatum(Oid X)
Definition: postgres.h:600
static Pointer DatumGetPointer(Datum X)
Definition: postgres.h:660
#define InvalidOid
Definition: postgres_ext.h:36
unsigned int Oid
Definition: postgres_ext.h:31
char * psprintf(const char *fmt,...)
Definition: psprintf.c:46
List * stringToQualifiedNameList(const char *string)
Definition: regproc.c:1877
#define RelationGetDescr(relation)
Definition: rel.h:527
void relation_close(Relation relation, LOCKMODE lockmode)
Definition: relation.c:206
Relation relation_openrv(const RangeVar *relation, LOCKMODE lockmode)
Definition: relation.c:138
MemoryContext ecxt_per_query_memory
Definition: execnodes.h:254
Definition: fmgr.h:57
void * fn_extra
Definition: fmgr.h:64
fmNodePtr fn_expr
Definition: fmgr.h:66
Oid fn_oid
Definition: fmgr.h:59
void * user_fctx
Definition: funcapi.h:82
uint64 max_calls
Definition: funcapi.h:74
uint64 call_cntr
Definition: funcapi.h:65
AttInMetadata * attinmeta
Definition: funcapi.h:91
MemoryContext multi_call_memory_ctx
Definition: funcapi.h:101
TupleDesc tuple_desc
Definition: funcapi.h:112
fmNodePtr resultinfo
Definition: fmgr.h:89
FmgrInfo * flinfo
Definition: fmgr.h:87
Definition: pg_list.h:52
Definition: nodes.h:118
SetFunctionReturnMode returnMode
Definition: execnodes.h:328
ExprContext * econtext
Definition: execnodes.h:324
TupleDesc setDesc
Definition: execnodes.h:332
Tuplestorestate * setResult
Definition: execnodes.h:331
TupleDesc expectedDesc
Definition: execnodes.h:325
int allowedModes
Definition: execnodes.h:326
Oid row_typeid
Definition: primnodes.h:1208
List * args
Definition: primnodes.h:1207
List * colnames
Definition: primnodes.h:1223
int32 tdtypmod
Definition: tupdesc.h:83
Oid tdtypeid
Definition: tupdesc.h:82
Definition: c.h:662
int dim1
Definition: c.h:667
Oid values[FLEXIBLE_ARRAY_MEMBER]
Definition: c.h:669
Oid anymultirange_type
Definition: funcapi.c:40
void ReleaseSysCache(HeapTuple tuple)
Definition: syscache.c:1221
HeapTuple SearchSysCache1(int cacheId, Datum key1)
Definition: syscache.c:1173
Datum SysCacheGetAttr(int cacheId, HeapTuple tup, AttrNumber attributeNumber, bool *isNull)
Definition: syscache.c:1434
@ PROCOID
Definition: syscache.h:79
TupleDesc CreateTemplateTupleDesc(int natts)
Definition: tupdesc.c:45
TupleDesc CreateTupleDescCopy(TupleDesc tupdesc)
Definition: tupdesc.c:111
void TupleDescInitEntryCollation(TupleDesc desc, AttrNumber attributeNumber, Oid collationid)
Definition: tupdesc.c:763
void TupleDescInitEntry(TupleDesc desc, AttrNumber attributeNumber, const char *attributeName, Oid oidtypeid, int32 typmod, int attdim)
Definition: tupdesc.c:583
#define TupleDescAttr(tupdesc, i)
Definition: tupdesc.h:92
Tuplestorestate * tuplestore_begin_heap(bool randomAccess, bool interXact, int maxKBytes)
Definition: tuplestore.c:318
void assign_record_type_typmod(TupleDesc tupDesc)
Definition: typcache.c:1944
TupleDesc lookup_rowtype_tupdesc_copy(Oid type_id, int32 typmod)
Definition: typcache.c:1858
#define strVal(v)
Definition: value.h:82