#include "postgres.h"
#include <ctype.h>
#include "access/detoast.h"
#include "access/htup_details.h"
#include "access/tupconvert.h"
#include "catalog/pg_proc.h"
#include "catalog/pg_type.h"
#include "executor/execExpr.h"
#include "executor/spi.h"
#include "executor/tstoreReceiver.h"
#include "funcapi.h"
#include "mb/stringinfo_mb.h"
#include "miscadmin.h"
#include "nodes/nodeFuncs.h"
#include "optimizer/optimizer.h"
#include "parser/parse_coerce.h"
#include "parser/parse_type.h"
#include "plpgsql.h"
#include "storage/proc.h"
#include "tcop/cmdtag.h"
#include "tcop/pquery.h"
#include "utils/array.h"
#include "utils/builtins.h"
#include "utils/datum.h"
#include "utils/fmgroids.h"
#include "utils/lsyscache.h"
#include "utils/memutils.h"
#include "utils/rel.h"
#include "utils/snapmgr.h"
#include "utils/syscache.h"
#include "utils/typcache.h"

Include dependency graph for pl_exec.c:

Data Structures
struct	SimpleEcontextStackEntry

struct	plpgsql_CastHashKey

struct	plpgsql_CastExprHashEntry

struct	plpgsql_CastHashEntry

Macros
#define	get_eval_mcontext(estate) ((estate)->eval_econtext->ecxt_per_tuple_memory)

#define	eval_mcontext_alloc(estate, sz) MemoryContextAlloc(get_eval_mcontext(estate), sz)

#define	eval_mcontext_alloc0(estate, sz) MemoryContextAllocZero(get_eval_mcontext(estate), sz)

#define	LOOP_RC_PROCESSING(looplabel, exit_action)

#define	SET_RAISE_OPTION_TEXT(opt, name)

Typedefs
typedef struct SimpleEcontextStackEntry	SimpleEcontextStackEntry

Functions
static void	coerce_function_result_tuple (PLpgSQL_execstate *estate, TupleDesc tupdesc)

static void	plpgsql_exec_error_callback (void *arg)

static void	copy_plpgsql_datums (PLpgSQL_execstate estate, PLpgSQL_function func)

static void	plpgsql_fulfill_promise (PLpgSQL_execstate estate, PLpgSQL_var var)

static MemoryContext	get_stmt_mcontext (PLpgSQL_execstate *estate)

static void	push_stmt_mcontext (PLpgSQL_execstate *estate)

static void	pop_stmt_mcontext (PLpgSQL_execstate *estate)

static int	exec_toplevel_block (PLpgSQL_execstate estate, PLpgSQL_stmt_block block)

static int	exec_stmt_block (PLpgSQL_execstate estate, PLpgSQL_stmt_block block)

static int	exec_stmts (PLpgSQL_execstate estate, List stmts)

static int	exec_stmt_assign (PLpgSQL_execstate estate, PLpgSQL_stmt_assign stmt)

static int	exec_stmt_perform (PLpgSQL_execstate estate, PLpgSQL_stmt_perform stmt)

static int	exec_stmt_call (PLpgSQL_execstate estate, PLpgSQL_stmt_call stmt)

static int	exec_stmt_getdiag (PLpgSQL_execstate estate, PLpgSQL_stmt_getdiag stmt)

static int	exec_stmt_if (PLpgSQL_execstate estate, PLpgSQL_stmt_if stmt)

static int	exec_stmt_case (PLpgSQL_execstate estate, PLpgSQL_stmt_case stmt)

static int	exec_stmt_loop (PLpgSQL_execstate estate, PLpgSQL_stmt_loop stmt)

static int	exec_stmt_while (PLpgSQL_execstate estate, PLpgSQL_stmt_while stmt)

static int	exec_stmt_fori (PLpgSQL_execstate estate, PLpgSQL_stmt_fori stmt)

static int	exec_stmt_fors (PLpgSQL_execstate estate, PLpgSQL_stmt_fors stmt)

static int	exec_stmt_forc (PLpgSQL_execstate estate, PLpgSQL_stmt_forc stmt)

static int	exec_stmt_foreach_a (PLpgSQL_execstate estate, PLpgSQL_stmt_foreach_a stmt)

static int	exec_stmt_open (PLpgSQL_execstate estate, PLpgSQL_stmt_open stmt)

static int	exec_stmt_fetch (PLpgSQL_execstate estate, PLpgSQL_stmt_fetch stmt)

static int	exec_stmt_close (PLpgSQL_execstate estate, PLpgSQL_stmt_close stmt)

static int	exec_stmt_exit (PLpgSQL_execstate estate, PLpgSQL_stmt_exit stmt)

static int	exec_stmt_return (PLpgSQL_execstate estate, PLpgSQL_stmt_return stmt)

static int	exec_stmt_return_next (PLpgSQL_execstate estate, PLpgSQL_stmt_return_next stmt)

static int	exec_stmt_return_query (PLpgSQL_execstate estate, PLpgSQL_stmt_return_query stmt)

static int	exec_stmt_raise (PLpgSQL_execstate estate, PLpgSQL_stmt_raise stmt)

static int	exec_stmt_assert (PLpgSQL_execstate estate, PLpgSQL_stmt_assert stmt)

static int	exec_stmt_execsql (PLpgSQL_execstate estate, PLpgSQL_stmt_execsql stmt)

static int	exec_stmt_dynexecute (PLpgSQL_execstate estate, PLpgSQL_stmt_dynexecute stmt)

static int	exec_stmt_dynfors (PLpgSQL_execstate estate, PLpgSQL_stmt_dynfors stmt)

static int	exec_stmt_commit (PLpgSQL_execstate estate, PLpgSQL_stmt_commit stmt)

static int	exec_stmt_rollback (PLpgSQL_execstate estate, PLpgSQL_stmt_rollback stmt)

static void	plpgsql_estate_setup (PLpgSQL_execstate estate, PLpgSQL_function func, ReturnSetInfo rsi, EState simple_eval_estate, ResourceOwner simple_eval_resowner)

static void	exec_eval_cleanup (PLpgSQL_execstate *estate)

static void	exec_prepare_plan (PLpgSQL_execstate estate, PLpgSQL_expr expr, int cursorOptions)

static void	exec_simple_check_plan (PLpgSQL_execstate estate, PLpgSQL_expr expr)

static bool	exec_is_simple_query (PLpgSQL_expr *expr)

static void	exec_save_simple_expr (PLpgSQL_expr expr, CachedPlan cplan)

static void	exec_check_rw_parameter (PLpgSQL_expr *expr)

static void	exec_check_assignable (PLpgSQL_execstate *estate, int dno)

static bool	exec_eval_simple_expr (PLpgSQL_execstate estate, PLpgSQL_expr expr, Datum result, bool isNull, Oid rettype, int32 rettypmod)

static void	exec_assign_expr (PLpgSQL_execstate estate, PLpgSQL_datum target, PLpgSQL_expr *expr)

static void	exec_assign_c_string (PLpgSQL_execstate estate, PLpgSQL_datum target, const char *str)

static void	exec_assign_value (PLpgSQL_execstate estate, PLpgSQL_datum target, Datum value, bool isNull, Oid valtype, int32 valtypmod)

static void	exec_eval_datum (PLpgSQL_execstate estate, PLpgSQL_datum datum, Oid typeid, int32 typetypmod, Datum value, bool isnull)

static int	exec_eval_integer (PLpgSQL_execstate estate, PLpgSQL_expr expr, bool *isNull)

static bool	exec_eval_boolean (PLpgSQL_execstate estate, PLpgSQL_expr expr, bool *isNull)

static Datum	exec_eval_expr (PLpgSQL_execstate estate, PLpgSQL_expr expr, bool isNull, Oid rettype, int32 *rettypmod)

static int	exec_run_select (PLpgSQL_execstate estate, PLpgSQL_expr expr, long maxtuples, Portal *portalP)

static int	exec_for_query (PLpgSQL_execstate estate, PLpgSQL_stmt_forq stmt, Portal portal, bool prefetch_ok)

static ParamListInfo	setup_param_list (PLpgSQL_execstate estate, PLpgSQL_expr expr)

static ParamExternData *	plpgsql_param_fetch (ParamListInfo params, int paramid, bool speculative, ParamExternData *prm)

static void	plpgsql_param_compile (ParamListInfo params, Param param, ExprState state, Datum resv, bool resnull)

static void	plpgsql_param_eval_var (ExprState state, ExprEvalStep op, ExprContext *econtext)

static void	plpgsql_param_eval_var_ro (ExprState state, ExprEvalStep op, ExprContext *econtext)

static void	plpgsql_param_eval_recfield (ExprState state, ExprEvalStep op, ExprContext *econtext)

static void	plpgsql_param_eval_generic (ExprState state, ExprEvalStep op, ExprContext *econtext)

static void	plpgsql_param_eval_generic_ro (ExprState state, ExprEvalStep op, ExprContext *econtext)

static void	exec_move_row (PLpgSQL_execstate estate, PLpgSQL_variable target, HeapTuple tup, TupleDesc tupdesc)

static void	revalidate_rectypeid (PLpgSQL_rec *rec)

static ExpandedRecordHeader *	make_expanded_record_for_rec (PLpgSQL_execstate estate, PLpgSQL_rec rec, TupleDesc srctupdesc, ExpandedRecordHeader *srcerh)

static void	exec_move_row_from_fields (PLpgSQL_execstate estate, PLpgSQL_variable target, ExpandedRecordHeader newerh, Datum values, bool *nulls, TupleDesc tupdesc)

static bool	compatible_tupdescs (TupleDesc src_tupdesc, TupleDesc dst_tupdesc)

static HeapTuple	make_tuple_from_row (PLpgSQL_execstate estate, PLpgSQL_row row, TupleDesc tupdesc)

static TupleDesc	deconstruct_composite_datum (Datum value, HeapTupleData *tmptup)

static void	exec_move_row_from_datum (PLpgSQL_execstate estate, PLpgSQL_variable target, Datum value)

static void	instantiate_empty_record_variable (PLpgSQL_execstate estate, PLpgSQL_rec rec)

static char *	convert_value_to_string (PLpgSQL_execstate *estate, Datum value, Oid valtype)

static Datum	exec_cast_value (PLpgSQL_execstate estate, Datum value, bool isnull, Oid valtype, int32 valtypmod, Oid reqtype, int32 reqtypmod)

static Datum	do_cast_value (PLpgSQL_execstate estate, Datum value, bool isnull, Oid valtype, int32 valtypmod, Oid reqtype, int32 reqtypmod)

static plpgsql_CastHashEntry *	get_cast_hashentry (PLpgSQL_execstate *estate, Oid srctype, int32 srctypmod, Oid dsttype, int32 dsttypmod)

static void	exec_init_tuple_store (PLpgSQL_execstate *estate)

static void	exec_set_found (PLpgSQL_execstate *estate, bool state)

static void	plpgsql_create_econtext (PLpgSQL_execstate *estate)

static void	plpgsql_destroy_econtext (PLpgSQL_execstate *estate)

static void	assign_simple_var (PLpgSQL_execstate estate, PLpgSQL_var var, Datum newvalue, bool isnull, bool freeable)

static void	assign_text_var (PLpgSQL_execstate estate, PLpgSQL_var var, const char *str)

static void	assign_record_var (PLpgSQL_execstate estate, PLpgSQL_rec rec, ExpandedRecordHeader *erh)

static ParamListInfo	exec_eval_using_params (PLpgSQL_execstate estate, List params)

static Portal	exec_dynquery_with_params (PLpgSQL_execstate estate, PLpgSQL_expr dynquery, List params, const char portalname, int cursorOptions)

static char *	format_expr_params (PLpgSQL_execstate estate, const PLpgSQL_expr expr)

static char *	format_preparedparamsdata (PLpgSQL_execstate *estate, ParamListInfo paramLI)

static PLpgSQL_variable *	make_callstmt_target (PLpgSQL_execstate estate, PLpgSQL_expr expr)

Datum	plpgsql_exec_function (PLpgSQL_function func, FunctionCallInfo fcinfo, EState simple_eval_estate, ResourceOwner simple_eval_resowner, ResourceOwner procedure_resowner, bool atomic)

HeapTuple	plpgsql_exec_trigger (PLpgSQL_function func, TriggerData trigdata)

void	plpgsql_exec_event_trigger (PLpgSQL_function func, EventTriggerData trigdata)

static bool	exception_matches_conditions (ErrorData edata, PLpgSQL_condition cond)

Oid	plpgsql_exec_get_datum_type (PLpgSQL_execstate estate, PLpgSQL_datum datum)

void	plpgsql_exec_get_datum_type_info (PLpgSQL_execstate estate, PLpgSQL_datum datum, Oid typeId, int32 typMod, Oid *collation)

void	plpgsql_xact_cb (XactEvent event, void *arg)

void	plpgsql_subxact_cb (SubXactEvent event, SubTransactionId mySubid, SubTransactionId parentSubid, void *arg)

Variables
static EState *	shared_simple_eval_estate = NULL

static SimpleEcontextStackEntry *	simple_econtext_stack = NULL

static ResourceOwner	shared_simple_eval_resowner = NULL

static HTAB *	cast_expr_hash = NULL

static HTAB *	shared_cast_hash = NULL

Macro Definition Documentation

◆ eval_mcontext_alloc

#define eval_mcontext_alloc	(	estate,
		sz
	)	MemoryContextAlloc(get_eval_mcontext(estate), sz)

Definition at line 128 of file pl_exec.c.

◆ eval_mcontext_alloc0

#define eval_mcontext_alloc0	(	estate,
		sz
	)	MemoryContextAllocZero(get_eval_mcontext(estate), sz)

Definition at line 130 of file pl_exec.c.

◆ get_eval_mcontext

#define get_eval_mcontext ( estate ) ((estate)->eval_econtext->ecxt_per_tuple_memory)

Definition at line 126 of file pl_exec.c.

◆ LOOP_RC_PROCESSING

#define LOOP_RC_PROCESSING	(	looplabel,
		exit_action
	)

Definition at line 203 of file pl_exec.c.

◆ SET_RAISE_OPTION_TEXT

#define SET_RAISE_OPTION_TEXT	(	opt,
		name
	)

Value:

do { \
    if (opt) \
        ereport(ERROR, \
                (errcode(ERRCODE_SYNTAX_ERROR), \
                 errmsg("RAISE option already specified: %s", \
                        name))); \
    opt = MemoryContextStrdup(stmt_mcontext, extval); \
} while (0)

Definition at line 3708 of file pl_exec.c.

Typedef Documentation

◆ SimpleEcontextStackEntry

typedef struct SimpleEcontextStackEntry SimpleEcontextStackEntry

Function Documentation

◆ assign_record_var()

static void assign_record_var	(	PLpgSQL_execstate *	estate,
		PLpgSQL_rec *	rec,
		ExpandedRecordHeader *	erh
	)

static

Definition at line 8627 of file pl_exec.c.

{
    Assert(rec->dtype == PLPGSQL_DTYPE_REC);
 
    /* Transfer new record object into datum_context */
    TransferExpandedRecord(erh, estate->datum_context);
 
    /* Free the old value ... */
    if (rec->erh)
        DeleteExpandedObject(ExpandedRecordGetDatum(rec->erh));
 
    /* ... and install the new */
    rec->erh = erh;
}

References Assert, PLpgSQL_execstate::datum_context, DeleteExpandedObject(), PLpgSQL_rec::dtype, PLpgSQL_rec::erh, ExpandedRecordGetDatum(), PLPGSQL_DTYPE_REC, and TransferExpandedRecord.

Referenced by exec_move_row(), exec_move_row_from_datum(), and exec_move_row_from_fields().

◆ assign_simple_var()

static void assign_simple_var	(	PLpgSQL_execstate *	estate,
		PLpgSQL_var *	var,
		Datum	newvalue,
		bool	isnull,
		bool	freeable
	)

static

Definition at line 8551 of file pl_exec.c.

{
    Assert(var->dtype == PLPGSQL_DTYPE_VAR ||
           var->dtype == PLPGSQL_DTYPE_PROMISE);
 
    /*
     * In non-atomic contexts, we do not want to store TOAST pointers in
     * variables, because such pointers might become stale after a commit.
     * Forcibly detoast in such cases.  We don't want to detoast (flatten)
     * expanded objects, however; those should be OK across a transaction
     * boundary since they're just memory-resident objects.  (Elsewhere in
     * this module, operations on expanded records likewise need to request
     * detoasting of record fields when !estate->atomic.  Expanded arrays are
     * not a problem since all array entries are always detoasted.)
     */
    if (!estate->atomic && !isnull && var->datatype->typlen == -1 &&
        VARATT_IS_EXTERNAL_NON_EXPANDED(DatumGetPointer(newvalue)))
    {
        MemoryContext oldcxt;
        Datum       detoasted;
 
        /*
         * Do the detoasting in the eval_mcontext to avoid long-term leakage
         * of whatever memory toast fetching might leak.  Then we have to copy
         * the detoasted datum to the function's main context, which is a
         * pain, but there's little choice.
         */
        oldcxt = MemoryContextSwitchTo(get_eval_mcontext(estate));
        detoasted = PointerGetDatum(detoast_external_attr((struct varlena *) DatumGetPointer(newvalue)));
        MemoryContextSwitchTo(oldcxt);
        /* Now's a good time to not leak the input value if it's freeable */
        if (freeable)
            pfree(DatumGetPointer(newvalue));
        /* Once we copy the value, it's definitely freeable */
        newvalue = datumCopy(detoasted, false, -1);
        freeable = true;
        /* Can't clean up eval_mcontext here, but it'll happen before long */
    }
 
    /* Free the old value if needed */
    if (var->freeval)
    {
        if (DatumIsReadWriteExpandedObject(var->value,
                                           var->isnull,
                                           var->datatype->typlen))
            DeleteExpandedObject(var->value);
        else
            pfree(DatumGetPointer(var->value));
    }
    /* Assign new value to datum */
    var->value = newvalue;
    var->isnull = isnull;
    var->freeval = freeable;
 
    /*
     * If it's a promise variable, then either we just assigned the promised
     * value, or the user explicitly assigned an overriding value.  Either
     * way, cancel the promise.
     */
    var->promise = PLPGSQL_PROMISE_NONE;
}

References Assert, PLpgSQL_execstate::atomic, PLpgSQL_var::datatype, datumCopy(), DatumGetPointer(), DatumIsReadWriteExpandedObject, DeleteExpandedObject(), detoast_external_attr(), PLpgSQL_var::dtype, PLpgSQL_var::freeval, get_eval_mcontext, PLpgSQL_var::isnull, MemoryContextSwitchTo(), pfree(), PLPGSQL_DTYPE_PROMISE, PLPGSQL_DTYPE_VAR, PLPGSQL_PROMISE_NONE, PointerGetDatum(), PLpgSQL_var::promise, PLpgSQL_type::typlen, PLpgSQL_var::value, and VARATT_IS_EXTERNAL_NON_EXPANDED.

Referenced by assign_text_var(), exec_assign_value(), exec_set_found(), exec_stmt_block(), exec_stmt_case(), exec_stmt_forc(), exec_stmt_fori(), plpgsql_exec_function(), and plpgsql_fulfill_promise().

◆ assign_text_var()

static void assign_text_var	(	PLpgSQL_execstate *	estate,
		PLpgSQL_var *	var,
		const char *	str
	)

static

Definition at line 8618 of file pl_exec.c.

{
    assign_simple_var(estate, var, CStringGetTextDatum(str), false, true);
}

References assign_simple_var(), CStringGetTextDatum, and str.

Referenced by exec_stmt_block(), exec_stmt_forc(), exec_stmt_open(), and plpgsql_fulfill_promise().

◆ coerce_function_result_tuple()

static void coerce_function_result_tuple	(	PLpgSQL_execstate *	estate,
		TupleDesc	tupdesc
	)

static

Definition at line 808 of file pl_exec.c.

{
    HeapTuple   rettup;
    TupleDesc   retdesc;
    TupleConversionMap *tupmap;
 
    /* We assume exec_stmt_return verified that result is composite */
    Assert(type_is_rowtype(estate->rettype));
 
    /* We can special-case expanded records for speed */
    if (VARATT_IS_EXTERNAL_EXPANDED(DatumGetPointer(estate->retval)))
    {
        ExpandedRecordHeader *erh = (ExpandedRecordHeader *) DatumGetEOHP(estate->retval);
 
        Assert(erh->er_magic == ER_MAGIC);
 
        /* Extract record's TupleDesc */
        retdesc = expanded_record_get_tupdesc(erh);
 
        /* check rowtype compatibility */
        tupmap = convert_tuples_by_position(retdesc,
                                            tupdesc,
                                            gettext_noop("returned record type does not match expected record type"));
 
        /* it might need conversion */
        if (tupmap)
        {
            rettup = expanded_record_get_tuple(erh);
            Assert(rettup);
            rettup = execute_attr_map_tuple(rettup, tupmap);
 
            /*
             * Copy tuple to upper executor memory, as a tuple Datum.  Make
             * sure it is labeled with the caller-supplied tuple type.
             */
            estate->retval = PointerGetDatum(SPI_returntuple(rettup, tupdesc));
            /* no need to free map, we're about to return anyway */
        }
        else if (!(tupdesc->tdtypeid == erh->er_decltypeid ||
                   (tupdesc->tdtypeid == RECORDOID &&
                    !ExpandedRecordIsDomain(erh))))
        {
            /*
             * The expanded record has the right physical tupdesc, but the
             * wrong type ID.  (Typically, the expanded record is RECORDOID
             * but the function is declared to return a named composite type.
             * As in exec_move_row_from_datum, we don't allow returning a
             * composite-domain record from a function declared to return
             * RECORD.)  So we must flatten the record to a tuple datum and
             * overwrite its type fields with the right thing.  spi.c doesn't
             * provide any easy way to deal with this case, so we end up
             * duplicating the guts of datumCopy() :-(
             */
            Size        resultsize;
            HeapTupleHeader tuphdr;
 
            resultsize = EOH_get_flat_size(&erh->hdr);
            tuphdr = (HeapTupleHeader) SPI_palloc(resultsize);
            EOH_flatten_into(&erh->hdr, tuphdr, resultsize);
            HeapTupleHeaderSetTypeId(tuphdr, tupdesc->tdtypeid);
            HeapTupleHeaderSetTypMod(tuphdr, tupdesc->tdtypmod);
            estate->retval = PointerGetDatum(tuphdr);
        }
        else
        {
            /*
             * We need only copy result into upper executor memory context.
             * However, if we have a R/W expanded datum, we can just transfer
             * its ownership out to the upper executor context.
             */
            estate->retval = SPI_datumTransfer(estate->retval,
                                               false,
                                               -1);
        }
    }
    else
    {
        /* Convert composite datum to a HeapTuple and TupleDesc */
        HeapTupleData tmptup;
 
        retdesc = deconstruct_composite_datum(estate->retval, &tmptup);
        rettup = &tmptup;
 
        /* check rowtype compatibility */
        tupmap = convert_tuples_by_position(retdesc,
                                            tupdesc,
                                            gettext_noop("returned record type does not match expected record type"));
 
        /* it might need conversion */
        if (tupmap)
            rettup = execute_attr_map_tuple(rettup, tupmap);
 
        /*
         * Copy tuple to upper executor memory, as a tuple Datum.  Make sure
         * it is labeled with the caller-supplied tuple type.
         */
        estate->retval = PointerGetDatum(SPI_returntuple(rettup, tupdesc));
 
        /* no need to free map, we're about to return anyway */
 
        ReleaseTupleDesc(retdesc);
    }
}

References Assert, convert_tuples_by_position(), DatumGetEOHP(), DatumGetPointer(), deconstruct_composite_datum(), EOH_flatten_into(), EOH_get_flat_size(), ExpandedRecordHeader::er_decltypeid, ER_MAGIC, ExpandedRecordHeader::er_magic, execute_attr_map_tuple(), expanded_record_get_tupdesc(), expanded_record_get_tuple(), ExpandedRecordIsDomain, gettext_noop, ExpandedRecordHeader::hdr, HeapTupleHeaderSetTypeId, HeapTupleHeaderSetTypMod, PointerGetDatum(), ReleaseTupleDesc, PLpgSQL_execstate::rettype, PLpgSQL_execstate::retval, SPI_datumTransfer(), SPI_palloc(), SPI_returntuple(), TupleDescData::tdtypeid, TupleDescData::tdtypmod, type_is_rowtype(), and VARATT_IS_EXTERNAL_EXPANDED.

Referenced by plpgsql_exec_function().

◆ compatible_tupdescs()

static bool compatible_tupdescs	(	TupleDesc	src_tupdesc,
		TupleDesc	dst_tupdesc
	)

static

Definition at line 7293 of file pl_exec.c.

{
    int         i;
 
    /* Possibly we could allow src_tupdesc to have extra columns? */
    if (dst_tupdesc->natts != src_tupdesc->natts)
        return false;
 
    for (i = 0; i < dst_tupdesc->natts; i++)
    {
        Form_pg_attribute dattr = TupleDescAttr(dst_tupdesc, i);
        Form_pg_attribute sattr = TupleDescAttr(src_tupdesc, i);
 
        if (dattr->attisdropped != sattr->attisdropped)
            return false;
        if (!dattr->attisdropped)
        {
            /* Normal columns must match by type and typmod */
            if (dattr->atttypid != sattr->atttypid ||
                (dattr->atttypmod >= 0 &&
                 dattr->atttypmod != sattr->atttypmod))
                return false;
        }
        else
        {
            /* Dropped columns are OK as long as length/alignment match */
            if (dattr->attlen != sattr->attlen ||
                dattr->attalign != sattr->attalign)
                return false;
        }
    }
    return true;
}

References i, TupleDescData::natts, and TupleDescAttr().

Referenced by exec_for_query(), and exec_move_row().

◆ convert_value_to_string()

static char * convert_value_to_string	(	PLpgSQL_execstate *	estate,
		Datum	value,
		Oid	valtype
	)

static

Definition at line 7691 of file pl_exec.c.

{
    char       *result;
    MemoryContext oldcontext;
    Oid         typoutput;
    bool        typIsVarlena;
 
    oldcontext = MemoryContextSwitchTo(get_eval_mcontext(estate));
    getTypeOutputInfo(valtype, &typoutput, &typIsVarlena);
    result = OidOutputFunctionCall(typoutput, value);
    MemoryContextSwitchTo(oldcontext);
 
    return result;
}

References get_eval_mcontext, getTypeOutputInfo(), MemoryContextSwitchTo(), OidOutputFunctionCall(), and value.

Referenced by exec_dynquery_with_params(), exec_stmt_assert(), exec_stmt_dynexecute(), exec_stmt_raise(), exec_stmt_return_query(), format_expr_params(), and format_preparedparamsdata().

◆ copy_plpgsql_datums()

static void copy_plpgsql_datums	(	PLpgSQL_execstate *	estate,
		PLpgSQL_function *	func
	)

static

Definition at line 1294 of file pl_exec.c.

{
    int         ndatums = estate->ndatums;
    PLpgSQL_datum **indatums;
    PLpgSQL_datum **outdatums;
    char       *workspace;
    char       *ws_next;
    int         i;
 
    /* Allocate local datum-pointer array */
    estate->datums = (PLpgSQL_datum **)
        palloc(sizeof(PLpgSQL_datum *) * ndatums);
 
    /*
     * To reduce palloc overhead, we make a single palloc request for all the
     * space needed for locally-instantiated datums.
     */
    workspace = palloc(func->copiable_size);
    ws_next = workspace;
 
    /* Fill datum-pointer array, copying datums into workspace as needed */
    indatums = func->datums;
    outdatums = estate->datums;
    for (i = 0; i < ndatums; i++)
    {
        PLpgSQL_datum *indatum = indatums[i];
        PLpgSQL_datum *outdatum;
 
        /* This must agree with plpgsql_finish_datums on what is copiable */
        switch (indatum->dtype)
        {
            case PLPGSQL_DTYPE_VAR:
            case PLPGSQL_DTYPE_PROMISE:
                outdatum = (PLpgSQL_datum *) ws_next;
                memcpy(outdatum, indatum, sizeof(PLpgSQL_var));
                ws_next += MAXALIGN(sizeof(PLpgSQL_var));
                break;
 
            case PLPGSQL_DTYPE_REC:
                outdatum = (PLpgSQL_datum *) ws_next;
                memcpy(outdatum, indatum, sizeof(PLpgSQL_rec));
                ws_next += MAXALIGN(sizeof(PLpgSQL_rec));
                break;
 
            case PLPGSQL_DTYPE_ROW:
            case PLPGSQL_DTYPE_RECFIELD:
 
                /*
                 * These datum records are read-only at runtime, so no need to
                 * copy them (well, RECFIELD contains cached data, but we'd
                 * just as soon centralize the caching anyway).
                 */
                outdatum = indatum;
                break;
 
            default:
                elog(ERROR, "unrecognized dtype: %d", indatum->dtype);
                outdatum = NULL;    /* keep compiler quiet */
                break;
        }
 
        outdatums[i] = outdatum;
    }
 
    Assert(ws_next == workspace + func->copiable_size);
}

References Assert, PLpgSQL_function::copiable_size, PLpgSQL_function::datums, PLpgSQL_execstate::datums, PLpgSQL_datum::dtype, elog, ERROR, i, MAXALIGN, PLpgSQL_execstate::ndatums, palloc(), PLPGSQL_DTYPE_PROMISE, PLPGSQL_DTYPE_REC, PLPGSQL_DTYPE_RECFIELD, PLPGSQL_DTYPE_ROW, and PLPGSQL_DTYPE_VAR.

Referenced by plpgsql_exec_event_trigger(), plpgsql_exec_function(), and plpgsql_exec_trigger().

◆ deconstruct_composite_datum()

static TupleDesc deconstruct_composite_datum	(	Datum	value,
		HeapTupleData *	tmptup
	)

static

Definition at line 7392 of file pl_exec.c.

{
    HeapTupleHeader td;
    Oid         tupType;
    int32       tupTypmod;
 
    /* Get tuple body (note this could involve detoasting) */
    td = DatumGetHeapTupleHeader(value);
 
    /* Build a temporary HeapTuple control structure */
    tmptup->t_len = HeapTupleHeaderGetDatumLength(td);
    ItemPointerSetInvalid(&(tmptup->t_self));
    tmptup->t_tableOid = InvalidOid;
    tmptup->t_data = td;
 
    /* Extract rowtype info and find a tupdesc */
    tupType = HeapTupleHeaderGetTypeId(td);
    tupTypmod = HeapTupleHeaderGetTypMod(td);
    return lookup_rowtype_tupdesc(tupType, tupTypmod);
}

References DatumGetHeapTupleHeader, HeapTupleHeaderGetDatumLength, HeapTupleHeaderGetTypeId, HeapTupleHeaderGetTypMod, InvalidOid, ItemPointerSetInvalid(), lookup_rowtype_tupdesc(), HeapTupleData::t_data, HeapTupleData::t_len, HeapTupleData::t_self, HeapTupleData::t_tableOid, and value.

Referenced by coerce_function_result_tuple(), exec_stmt_return_next(), and plpgsql_exec_trigger().

◆ do_cast_value()

static Datum do_cast_value	(	PLpgSQL_execstate *	estate,
		Datum	value,
		bool *	isnull,
		Oid	valtype,
		int32	valtypmod,
		Oid	reqtype,
		int32	reqtypmod
	)

static

Definition at line 7744 of file pl_exec.c.

{
    plpgsql_CastHashEntry *cast_entry;
 
    cast_entry = get_cast_hashentry(estate,
                                    valtype, valtypmod,
                                    reqtype, reqtypmod);
    if (cast_entry)
    {
        ExprContext *econtext = estate->eval_econtext;
        MemoryContext oldcontext;
 
        oldcontext = MemoryContextSwitchTo(get_eval_mcontext(estate));
 
        econtext->caseValue_datum = value;
        econtext->caseValue_isNull = *isnull;
 
        cast_entry->cast_in_use = true;
 
        value = ExecEvalExpr(cast_entry->cast_exprstate, econtext,
                             isnull);
 
        cast_entry->cast_in_use = false;
 
        MemoryContextSwitchTo(oldcontext);
    }
 
    return value;
}

References ExprContext::caseValue_datum, ExprContext::caseValue_isNull, plpgsql_CastHashEntry::cast_exprstate, plpgsql_CastHashEntry::cast_in_use, PLpgSQL_execstate::eval_econtext, ExecEvalExpr(), get_cast_hashentry(), get_eval_mcontext, MemoryContextSwitchTo(), and value.

Referenced by exec_cast_value().

◆ exception_matches_conditions()

static bool exception_matches_conditions	(	ErrorData *	edata,
		PLpgSQL_condition *	cond
	)

static

Definition at line 1579 of file pl_exec.c.

{
    for (; cond != NULL; cond = cond->next)
    {
        int         sqlerrstate = cond->sqlerrstate;
 
        /*
         * OTHERS matches everything *except* query-canceled and
         * assert-failure.  If you're foolish enough, you can match those
         * explicitly.
         */
        if (sqlerrstate == 0)
        {
            if (edata->sqlerrcode != ERRCODE_QUERY_CANCELED &&
                edata->sqlerrcode != ERRCODE_ASSERT_FAILURE)
                return true;
        }
        /* Exact match? */
        else if (edata->sqlerrcode == sqlerrstate)
            return true;
        /* Category match? */
        else if (ERRCODE_IS_CATEGORY(sqlerrstate) &&
                 ERRCODE_TO_CATEGORY(edata->sqlerrcode) == sqlerrstate)
            return true;
    }
    return false;
}

References ERRCODE_IS_CATEGORY, ERRCODE_TO_CATEGORY, PLpgSQL_condition::next, ErrorData::sqlerrcode, and PLpgSQL_condition::sqlerrstate.

Referenced by exec_stmt_block().

◆ exec_assign_c_string()

static void exec_assign_c_string	(	PLpgSQL_execstate *	estate,
		PLpgSQL_datum *	target,
		const char *	str
	)

static

Definition at line 5051 of file pl_exec.c.

{
    text       *value;
    MemoryContext oldcontext;
 
    /* Use eval_mcontext for short-lived text value */
    oldcontext = MemoryContextSwitchTo(get_eval_mcontext(estate));
    if (str != NULL)
        value = cstring_to_text(str);
    else
        value = cstring_to_text("");
    MemoryContextSwitchTo(oldcontext);
 
    exec_assign_value(estate, target, PointerGetDatum(value), false,
                      TEXTOID, -1);
}

References cstring_to_text(), exec_assign_value(), get_eval_mcontext, MemoryContextSwitchTo(), PointerGetDatum(), str, and value.

Referenced by exec_stmt_getdiag().

◆ exec_assign_expr()

static void exec_assign_expr	(	PLpgSQL_execstate *	estate,
		PLpgSQL_datum *	target,
		PLpgSQL_expr *	expr
	)

static

Definition at line 5007 of file pl_exec.c.

{
    Datum       value;
    bool        isnull;
    Oid         valtype;
    int32       valtypmod;
 
    /*
     * If first time through, create a plan for this expression.
     */
    if (expr->plan == NULL)
    {
        /*
         * Mark the expression as being an assignment source, if target is a
         * simple variable.  (This is a bit messy, but it seems cleaner than
         * modifying the API of exec_prepare_plan for the purpose.  We need to
         * stash the target dno into the expr anyway, so that it will be
         * available if we have to replan.)
         */
        if (target->dtype == PLPGSQL_DTYPE_VAR)
            expr->target_param = target->dno;
        else
            expr->target_param = -1;    /* should be that already */
 
        exec_prepare_plan(estate, expr, 0);
    }
 
    value = exec_eval_expr(estate, expr, &isnull, &valtype, &valtypmod);
    exec_assign_value(estate, target, value, isnull, valtype, valtypmod);
    exec_eval_cleanup(estate);
}

References PLpgSQL_datum::dno, PLpgSQL_datum::dtype, exec_assign_value(), exec_eval_cleanup(), exec_eval_expr(), exec_prepare_plan(), PLpgSQL_expr::plan, PLPGSQL_DTYPE_VAR, PLpgSQL_expr::target_param, and value.

Referenced by exec_stmt_assign(), exec_stmt_block(), and plpgsql_estate_setup().

◆ exec_assign_value()

static void exec_assign_value	(	PLpgSQL_execstate *	estate,
		PLpgSQL_datum *	target,
		Datum	value,
		bool	isNull,
		Oid	valtype,
		int32	valtypmod
	)

static

Definition at line 5079 of file pl_exec.c.

{
    switch (target->dtype)
    {
        case PLPGSQL_DTYPE_VAR:
        case PLPGSQL_DTYPE_PROMISE:
            {
                /*
                 * Target is a variable
                 */
                PLpgSQL_var *var = (PLpgSQL_var *) target;
                Datum       newvalue;
 
                newvalue = exec_cast_value(estate,
                                           value,
                                           &isNull,
                                           valtype,
                                           valtypmod,
                                           var->datatype->typoid,
                                           var->datatype->atttypmod);
 
                if (isNull && var->notnull)
                    ereport(ERROR,
                            (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                             errmsg("null value cannot be assigned to variable \"%s\" declared NOT NULL",
                                    var->refname)));
 
                /*
                 * If type is by-reference, copy the new value (which is
                 * probably in the eval_mcontext) into the procedure's main
                 * memory context.  But if it's a read/write reference to an
                 * expanded object, no physical copy needs to happen; at most
                 * we need to reparent the object's memory context.
                 *
                 * If it's an array, we force the value to be stored in R/W
                 * expanded form.  This wins if the function later does, say,
                 * a lot of array subscripting operations on the variable, and
                 * otherwise might lose.  We might need to use a different
                 * heuristic, but it's too soon to tell.  Also, are there
                 * cases where it'd be useful to force non-array values into
                 * expanded form?
                 */
                if (!var->datatype->typbyval && !isNull)
                {
                    if (var->datatype->typisarray &&
                        !VARATT_IS_EXTERNAL_EXPANDED_RW(DatumGetPointer(newvalue)))
                    {
                        /* array and not already R/W, so apply expand_array */
                        newvalue = expand_array(newvalue,
                                                estate->datum_context,
                                                NULL);
                    }
                    else
                    {
                        /* else transfer value if R/W, else just datumCopy */
                        newvalue = datumTransfer(newvalue,
                                                 false,
                                                 var->datatype->typlen);
                    }
                }
 
                /*
                 * Now free the old value, if any, and assign the new one. But
                 * skip the assignment if old and new values are the same.
                 * Note that for expanded objects, this test is necessary and
                 * cannot reliably be made any earlier; we have to be looking
                 * at the object's standard R/W pointer to be sure pointer
                 * equality is meaningful.
                 *
                 * Also, if it's a promise variable, we should disarm the
                 * promise in any case --- otherwise, assigning null to an
                 * armed promise variable would fail to disarm the promise.
                 */
                if (var->value != newvalue || var->isnull || isNull)
                    assign_simple_var(estate, var, newvalue, isNull,
                                      (!var->datatype->typbyval && !isNull));
                else
                    var->promise = PLPGSQL_PROMISE_NONE;
                break;
            }
 
        case PLPGSQL_DTYPE_ROW:
            {
                /*
                 * Target is a row variable
                 */
                PLpgSQL_row *row = (PLpgSQL_row *) target;
 
                if (isNull)
                {
                    /* If source is null, just assign nulls to the row */
                    exec_move_row(estate, (PLpgSQL_variable *) row,
                                  NULL, NULL);
                }
                else
                {
                    /* Source must be of RECORD or composite type */
                    if (!type_is_rowtype(valtype))
                        ereport(ERROR,
                                (errcode(ERRCODE_DATATYPE_MISMATCH),
                                 errmsg("cannot assign non-composite value to a row variable")));
                    exec_move_row_from_datum(estate, (PLpgSQL_variable *) row,
                                             value);
                }
                break;
            }
 
        case PLPGSQL_DTYPE_REC:
            {
                /*
                 * Target is a record variable
                 */
                PLpgSQL_rec *rec = (PLpgSQL_rec *) target;
 
                if (isNull)
                {
                    if (rec->notnull)
                        ereport(ERROR,
                                (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                                 errmsg("null value cannot be assigned to variable \"%s\" declared NOT NULL",
                                        rec->refname)));
 
                    /* Set variable to a simple NULL */
                    exec_move_row(estate, (PLpgSQL_variable *) rec,
                                  NULL, NULL);
                }
                else
                {
                    /* Source must be of RECORD or composite type */
                    if (!type_is_rowtype(valtype))
                        ereport(ERROR,
                                (errcode(ERRCODE_DATATYPE_MISMATCH),
                                 errmsg("cannot assign non-composite value to a record variable")));
                    exec_move_row_from_datum(estate, (PLpgSQL_variable *) rec,
                                             value);
                }
                break;
            }
 
        case PLPGSQL_DTYPE_RECFIELD:
            {
                /*
                 * Target is a field of a record
                 */
                PLpgSQL_recfield *recfield = (PLpgSQL_recfield *) target;
                PLpgSQL_rec *rec;
                ExpandedRecordHeader *erh;
 
                rec = (PLpgSQL_rec *) (estate->datums[recfield->recparentno]);
                erh = rec->erh;
 
                /*
                 * If record variable is NULL, instantiate it if it has a
                 * named composite type, else complain.  (This won't change
                 * the logical state of the record, but if we successfully
                 * assign below, the unassigned fields will all become NULLs.)
                 */
                if (erh == NULL)
                {
                    instantiate_empty_record_variable(estate, rec);
                    erh = rec->erh;
                }
 
                /*
                 * Look up the field's properties if we have not already, or
                 * if the tuple descriptor ID changed since last time.
                 */
                if (unlikely(recfield->rectupledescid != erh->er_tupdesc_id))
                {
                    if (!expanded_record_lookup_field(erh,
                                                      recfield->fieldname,
                                                      &recfield->finfo))
                        ereport(ERROR,
                                (errcode(ERRCODE_UNDEFINED_COLUMN),
                                 errmsg("record \"%s\" has no field \"%s\"",
                                        rec->refname, recfield->fieldname)));
                    recfield->rectupledescid = erh->er_tupdesc_id;
                }
 
                /* We don't support assignments to system columns. */
                if (recfield->finfo.fnumber <= 0)
                    ereport(ERROR,
                            (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                             errmsg("cannot assign to system column \"%s\"",
                                    recfield->fieldname)));
 
                /* Cast the new value to the right type, if needed. */
                value = exec_cast_value(estate,
                                        value,
                                        &isNull,
                                        valtype,
                                        valtypmod,
                                        recfield->finfo.ftypeid,
                                        recfield->finfo.ftypmod);
 
                /* And assign it. */
                expanded_record_set_field(erh, recfield->finfo.fnumber,
                                          value, isNull, !estate->atomic);
                break;
            }
 
        default:
            elog(ERROR, "unrecognized dtype: %d", target->dtype);
    }
}

References assign_simple_var(), PLpgSQL_execstate::atomic, PLpgSQL_type::atttypmod, PLpgSQL_var::datatype, PLpgSQL_execstate::datum_context, DatumGetPointer(), PLpgSQL_execstate::datums, datumTransfer(), PLpgSQL_datum::dtype, elog, ExpandedRecordHeader::er_tupdesc_id, ereport, PLpgSQL_rec::erh, errcode(), errmsg(), ERROR, exec_cast_value(), exec_move_row(), exec_move_row_from_datum(), expand_array(), expanded_record_lookup_field(), expanded_record_set_field, PLpgSQL_recfield::fieldname, PLpgSQL_recfield::finfo, ExpandedRecordFieldInfo::fnumber, ExpandedRecordFieldInfo::ftypeid, ExpandedRecordFieldInfo::ftypmod, instantiate_empty_record_variable(), PLpgSQL_var::isnull, PLpgSQL_var::notnull, PLpgSQL_rec::notnull, PLPGSQL_DTYPE_PROMISE, PLPGSQL_DTYPE_REC, PLPGSQL_DTYPE_RECFIELD, PLPGSQL_DTYPE_ROW, PLPGSQL_DTYPE_VAR, PLPGSQL_PROMISE_NONE, PLpgSQL_var::promise, PLpgSQL_recfield::recparentno, PLpgSQL_recfield::rectupledescid, PLpgSQL_var::refname, PLpgSQL_rec::refname, PLpgSQL_type::typbyval, type_is_rowtype(), PLpgSQL_type::typisarray, PLpgSQL_type::typlen, PLpgSQL_type::typoid, unlikely, value, PLpgSQL_var::value, and VARATT_IS_EXTERNAL_EXPANDED_RW.

Referenced by exec_assign_c_string(), exec_assign_expr(), exec_move_row_from_fields(), exec_stmt_block(), exec_stmt_case(), exec_stmt_foreach_a(), exec_stmt_getdiag(), and plpgsql_estate_setup().

◆ exec_cast_value()

static Datum exec_cast_value	(	PLpgSQL_execstate *	estate,
		Datum	value,
		bool *	isnull,
		Oid	valtype,
		int32	valtypmod,
		Oid	reqtype,
		int32	reqtypmod
	)

inlinestatic

Definition at line 7720 of file pl_exec.c.

{
    /*
     * If the type of the given value isn't what's requested, convert it.
     */
    if (valtype != reqtype ||
        (valtypmod != reqtypmod && reqtypmod != -1))
    {
        /* We keep the slow path out-of-line. */
        value = do_cast_value(estate, value, isnull, valtype, valtypmod,
                              reqtype, reqtypmod);
    }
 
    return value;
}

References do_cast_value(), and value.

Referenced by exec_assign_value(), exec_eval_boolean(), exec_eval_integer(), exec_move_row_from_fields(), exec_stmt_fori(), exec_stmt_return_next(), plpgsql_estate_setup(), and plpgsql_exec_function().

◆ exec_check_assignable()

static void exec_check_assignable	(	PLpgSQL_execstate *	estate,
		int	dno
	)

static

Definition at line 8339 of file pl_exec.c.

{
    PLpgSQL_datum *datum;
 
    Assert(dno >= 0 && dno < estate->ndatums);
    datum = estate->datums[dno];
    switch (datum->dtype)
    {
        case PLPGSQL_DTYPE_VAR:
        case PLPGSQL_DTYPE_PROMISE:
        case PLPGSQL_DTYPE_REC:
            if (((PLpgSQL_variable *) datum)->isconst)
                ereport(ERROR,
                        (errcode(ERRCODE_ERROR_IN_ASSIGNMENT),
                         errmsg("variable \"%s\" is declared CONSTANT",
                                ((PLpgSQL_variable *) datum)->refname)));
            break;
        case PLPGSQL_DTYPE_ROW:
            /* always assignable; member vars were checked at compile time */
            break;
        case PLPGSQL_DTYPE_RECFIELD:
            /* assignable if parent record is */
            exec_check_assignable(estate,
                                  ((PLpgSQL_recfield *) datum)->recparentno);
            break;
        default:
            elog(ERROR, "unrecognized dtype: %d", datum->dtype);
            break;
    }
}

References Assert, PLpgSQL_execstate::datums, PLpgSQL_datum::dtype, PLpgSQL_variable::dtype, elog, ereport, errcode(), errmsg(), ERROR, exec_check_assignable(), PLPGSQL_DTYPE_PROMISE, PLPGSQL_DTYPE_REC, PLPGSQL_DTYPE_RECFIELD, PLPGSQL_DTYPE_ROW, and PLPGSQL_DTYPE_VAR.

Referenced by exec_check_assignable(), exec_stmt_forc(), exec_stmt_open(), and make_callstmt_target().

◆ exec_check_rw_parameter()

static void exec_check_rw_parameter ( PLpgSQL_expr * expr )

static

Definition at line 8228 of file pl_exec.c.

{
    int         target_dno;
    Oid         funcid;
    List       *fargs;
    ListCell   *lc;
 
    /* Assume unsafe */
    expr->expr_rw_param = NULL;
 
    /* Done if expression isn't an assignment source */
    target_dno = expr->target_param;
    if (target_dno < 0)
        return;
 
    /*
     * If target variable isn't referenced by expression, no need to look
     * further.
     */
    if (!bms_is_member(target_dno, expr->paramnos))
        return;
 
    /* Shouldn't be here for non-simple expression */
    Assert(expr->expr_simple_expr != NULL);
 
    /*
     * Top level of expression must be a simple FuncExpr, OpExpr, or
     * SubscriptingRef, else we can't optimize.
     */
    if (IsA(expr->expr_simple_expr, FuncExpr))
    {
        FuncExpr   *fexpr = (FuncExpr *) expr->expr_simple_expr;
 
        funcid = fexpr->funcid;
        fargs = fexpr->args;
    }
    else if (IsA(expr->expr_simple_expr, OpExpr))
    {
        OpExpr     *opexpr = (OpExpr *) expr->expr_simple_expr;
 
        funcid = opexpr->opfuncid;
        fargs = opexpr->args;
    }
    else if (IsA(expr->expr_simple_expr, SubscriptingRef))
    {
        SubscriptingRef *sbsref = (SubscriptingRef *) expr->expr_simple_expr;
 
        /* We only trust standard varlena arrays to be safe */
        if (get_typsubscript(sbsref->refcontainertype, NULL) !=
            F_ARRAY_SUBSCRIPT_HANDLER)
            return;
 
        /* We can optimize the refexpr if it's the target, otherwise not */
        if (sbsref->refexpr && IsA(sbsref->refexpr, Param))
        {
            Param      *param = (Param *) sbsref->refexpr;
 
            if (param->paramkind == PARAM_EXTERN &&
                param->paramid == target_dno + 1)
            {
                /* Found the Param we want to pass as read/write */
                expr->expr_rw_param = param;
                return;
            }
        }
 
        return;
    }
    else
        return;
 
    /*
     * The top-level function must be one that we trust to be "safe".
     * Currently we hard-wire the list, but it would be very desirable to
     * allow extensions to mark their functions as safe ...
     */
    if (!(funcid == F_ARRAY_APPEND ||
          funcid == F_ARRAY_PREPEND))
        return;
 
    /*
     * The target variable (in the form of a Param) must appear as a direct
     * argument of the top-level function.  References further down in the
     * tree can't be optimized; but on the other hand, they don't invalidate
     * optimizing the top-level call, since that will be executed last.
     */
    foreach(lc, fargs)
    {
        Node       *arg = (Node *) lfirst(lc);
 
        if (arg && IsA(arg, Param))
        {
            Param      *param = (Param *) arg;
 
            if (param->paramkind == PARAM_EXTERN &&
                param->paramid == target_dno + 1)
            {
                /* Found the Param we want to pass as read/write */
                expr->expr_rw_param = param;
                return;
            }
        }
    }
}

References arg, FuncExpr::args, OpExpr::args, Assert, bms_is_member(), PLpgSQL_expr::expr_rw_param, PLpgSQL_expr::expr_simple_expr, FuncExpr::funcid, get_typsubscript(), if(), IsA, lfirst, PARAM_EXTERN, Param::paramid, Param::paramkind, PLpgSQL_expr::paramnos, SubscriptingRef::refexpr, and PLpgSQL_expr::target_param.

Referenced by exec_save_simple_expr().

◆ exec_dynquery_with_params()

static Portal exec_dynquery_with_params	(	PLpgSQL_execstate *	estate,
		PLpgSQL_expr *	dynquery,
		List *	params,
		const char *	portalname,
		int	cursorOptions
	)

static

Definition at line 8732 of file pl_exec.c.

{
    Portal      portal;
    Datum       query;
    bool        isnull;
    Oid         restype;
    int32       restypmod;
    char       *querystr;
    SPIParseOpenOptions options;
    MemoryContext stmt_mcontext = get_stmt_mcontext(estate);
 
    /*
     * Evaluate the string expression after the EXECUTE keyword. Its result is
     * the querystring we have to execute.
     */
    query = exec_eval_expr(estate, dynquery, &isnull, &restype, &restypmod);
    if (isnull)
        ereport(ERROR,
                (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                 errmsg("query string argument of EXECUTE is null")));
 
    /* Get the C-String representation */
    querystr = convert_value_to_string(estate, query, restype);
 
    /* copy it into the stmt_mcontext before we clean up */
    querystr = MemoryContextStrdup(stmt_mcontext, querystr);
 
    exec_eval_cleanup(estate);
 
    /*
     * Open an implicit cursor for the query.  We use SPI_cursor_parse_open
     * even when there are no params, because this avoids making and freeing
     * one copy of the plan.
     */
    memset(&options, 0, sizeof(options));
    options.params = exec_eval_using_params(estate, params);
    options.cursorOptions = cursorOptions;
    options.read_only = estate->readonly_func;
 
    portal = SPI_cursor_parse_open(portalname, querystr, &options);
 
    if (portal == NULL)
        elog(ERROR, "could not open implicit cursor for query \"%s\": %s",
             querystr, SPI_result_code_string(SPI_result));
 
    /* Release transient data */
    MemoryContextReset(stmt_mcontext);
 
    return portal;
}

References convert_value_to_string(), elog, ereport, errcode(), errmsg(), ERROR, exec_eval_cleanup(), exec_eval_expr(), exec_eval_using_params(), get_stmt_mcontext(), MemoryContextReset(), MemoryContextStrdup(), options, PLpgSQL_execstate::readonly_func, SPI_cursor_parse_open(), SPI_result, and SPI_result_code_string().

Referenced by exec_stmt_dynfors(), and exec_stmt_open().

◆ exec_eval_boolean()

static bool exec_eval_boolean	(	PLpgSQL_execstate *	estate,
		PLpgSQL_expr *	expr,
		bool *	isNull
	)

static

Definition at line 5660 of file pl_exec.c.

{
    Datum       exprdatum;
    Oid         exprtypeid;
    int32       exprtypmod;
 
    exprdatum = exec_eval_expr(estate, expr, isNull, &exprtypeid, &exprtypmod);
    exprdatum = exec_cast_value(estate, exprdatum, isNull,
                                exprtypeid, exprtypmod,
                                BOOLOID, -1);
    return DatumGetBool(exprdatum);
}

References DatumGetBool(), exec_cast_value(), and exec_eval_expr().

Referenced by exec_stmt_assert(), exec_stmt_case(), exec_stmt_exit(), exec_stmt_if(), and exec_stmt_while().

◆ exec_eval_cleanup()

static void exec_eval_cleanup ( PLpgSQL_execstate * estate )

static

Definition at line 4134 of file pl_exec.c.

{
    /* Clear result of a full SPI_execute */
    if (estate->eval_tuptable != NULL)
        SPI_freetuptable(estate->eval_tuptable);
    estate->eval_tuptable = NULL;
 
    /*
     * Clear result of exec_eval_simple_expr (but keep the econtext).  This
     * also clears any short-lived allocations done via get_eval_mcontext.
     */
    if (estate->eval_econtext != NULL)
        ResetExprContext(estate->eval_econtext);
}

References PLpgSQL_execstate::eval_econtext, PLpgSQL_execstate::eval_tuptable, ResetExprContext, and SPI_freetuptable().

Referenced by exec_assign_expr(), exec_dynquery_with_params(), exec_eval_using_params(), exec_for_query(), exec_run_select(), exec_stmt_assert(), exec_stmt_block(), exec_stmt_call(), exec_stmt_case(), exec_stmt_dynexecute(), exec_stmt_execsql(), exec_stmt_exit(), exec_stmt_fetch(), exec_stmt_forc(), exec_stmt_foreach_a(), exec_stmt_fori(), exec_stmt_getdiag(), exec_stmt_if(), exec_stmt_open(), exec_stmt_perform(), exec_stmt_raise(), exec_stmt_return_next(), exec_stmt_return_query(), exec_stmt_while(), plpgsql_exec_event_trigger(), plpgsql_exec_function(), and plpgsql_exec_trigger().

◆ exec_eval_datum()

static void exec_eval_datum	(	PLpgSQL_execstate *	estate,
		PLpgSQL_datum *	datum,
		Oid *	typeid,
		int32 *	typetypmod,
		Datum *	value,
		bool *	isnull
	)

static

Definition at line 5306 of file pl_exec.c.

{
    MemoryContext oldcontext;
 
    switch (datum->dtype)
    {
        case PLPGSQL_DTYPE_PROMISE:
            /* fulfill promise if needed, then handle like regular var */
            plpgsql_fulfill_promise(estate, (PLpgSQL_var *) datum);
 
            /* FALL THRU */
 
        case PLPGSQL_DTYPE_VAR:
            {
                PLpgSQL_var *var = (PLpgSQL_var *) datum;
 
                *typeid = var->datatype->typoid;
                *typetypmod = var->datatype->atttypmod;
                *value = var->value;
                *isnull = var->isnull;
                break;
            }
 
        case PLPGSQL_DTYPE_ROW:
            {
                PLpgSQL_row *row = (PLpgSQL_row *) datum;
                HeapTuple   tup;
 
                /* We get here if there are multiple OUT parameters */
                if (!row->rowtupdesc)   /* should not happen */
                    elog(ERROR, "row variable has no tupdesc");
                /* Make sure we have a valid type/typmod setting */
                BlessTupleDesc(row->rowtupdesc);
                oldcontext = MemoryContextSwitchTo(get_eval_mcontext(estate));
                tup = make_tuple_from_row(estate, row, row->rowtupdesc);
                if (tup == NULL)    /* should not happen */
                    elog(ERROR, "row not compatible with its own tupdesc");
                *typeid = row->rowtupdesc->tdtypeid;
                *typetypmod = row->rowtupdesc->tdtypmod;
                *value = HeapTupleGetDatum(tup);
                *isnull = false;
                MemoryContextSwitchTo(oldcontext);
                break;
            }
 
        case PLPGSQL_DTYPE_REC:
            {
                PLpgSQL_rec *rec = (PLpgSQL_rec *) datum;
 
                if (rec->erh == NULL)
                {
                    /* Treat uninstantiated record as a simple NULL */
                    *value = (Datum) 0;
                    *isnull = true;
                    /* Report variable's declared type */
                    *typeid = rec->rectypeid;
                    *typetypmod = -1;
                }
                else
                {
                    if (ExpandedRecordIsEmpty(rec->erh))
                    {
                        /* Empty record is also a NULL */
                        *value = (Datum) 0;
                        *isnull = true;
                    }
                    else
                    {
                        *value = ExpandedRecordGetDatum(rec->erh);
                        *isnull = false;
                    }
                    if (rec->rectypeid != RECORDOID)
                    {
                        /* Report variable's declared type, if not RECORD */
                        *typeid = rec->rectypeid;
                        *typetypmod = -1;
                    }
                    else
                    {
                        /* Report record's actual type if declared RECORD */
                        *typeid = rec->erh->er_typeid;
                        *typetypmod = rec->erh->er_typmod;
                    }
                }
                break;
            }
 
        case PLPGSQL_DTYPE_RECFIELD:
            {
                PLpgSQL_recfield *recfield = (PLpgSQL_recfield *) datum;
                PLpgSQL_rec *rec;
                ExpandedRecordHeader *erh;
 
                rec = (PLpgSQL_rec *) (estate->datums[recfield->recparentno]);
                erh = rec->erh;
 
                /*
                 * If record variable is NULL, instantiate it if it has a
                 * named composite type, else complain.  (This won't change
                 * the logical state of the record: it's still NULL.)
                 */
                if (erh == NULL)
                {
                    instantiate_empty_record_variable(estate, rec);
                    erh = rec->erh;
                }
 
                /*
                 * Look up the field's properties if we have not already, or
                 * if the tuple descriptor ID changed since last time.
                 */
                if (unlikely(recfield->rectupledescid != erh->er_tupdesc_id))
                {
                    if (!expanded_record_lookup_field(erh,
                                                      recfield->fieldname,
                                                      &recfield->finfo))
                        ereport(ERROR,
                                (errcode(ERRCODE_UNDEFINED_COLUMN),
                                 errmsg("record \"%s\" has no field \"%s\"",
                                        rec->refname, recfield->fieldname)));
                    recfield->rectupledescid = erh->er_tupdesc_id;
                }
 
                /* Report type data. */
                *typeid = recfield->finfo.ftypeid;
                *typetypmod = recfield->finfo.ftypmod;
 
                /* And fetch the field value. */
                *value = expanded_record_get_field(erh,
                                                   recfield->finfo.fnumber,
                                                   isnull);
                break;
            }
 
        default:
            elog(ERROR, "unrecognized dtype: %d", datum->dtype);
    }
}

References PLpgSQL_type::atttypmod, BlessTupleDesc(), PLpgSQL_var::datatype, PLpgSQL_execstate::datums, PLpgSQL_datum::dtype, PLpgSQL_var::dtype, elog, ExpandedRecordHeader::er_tupdesc_id, ExpandedRecordHeader::er_typeid, ExpandedRecordHeader::er_typmod, ereport, PLpgSQL_rec::erh, errcode(), errmsg(), ERROR, expanded_record_get_field(), expanded_record_lookup_field(), ExpandedRecordGetDatum(), ExpandedRecordIsEmpty, PLpgSQL_recfield::fieldname, PLpgSQL_recfield::finfo, ExpandedRecordFieldInfo::fnumber, ExpandedRecordFieldInfo::ftypeid, ExpandedRecordFieldInfo::ftypmod, get_eval_mcontext, HeapTupleGetDatum(), instantiate_empty_record_variable(), PLpgSQL_var::isnull, make_tuple_from_row(), MemoryContextSwitchTo(), PLPGSQL_DTYPE_PROMISE, PLPGSQL_DTYPE_REC, PLPGSQL_DTYPE_RECFIELD, PLPGSQL_DTYPE_ROW, PLPGSQL_DTYPE_VAR, plpgsql_fulfill_promise(), PLpgSQL_recfield::recparentno, PLpgSQL_recfield::rectupledescid, PLpgSQL_rec::rectypeid, PLpgSQL_rec::refname, PLpgSQL_row::rowtupdesc, TupleDescData::tdtypeid, TupleDescData::tdtypmod, PLpgSQL_type::typoid, unlikely, value, and PLpgSQL_var::value.

Referenced by exec_stmt_return(), format_expr_params(), make_tuple_from_row(), plpgsql_estate_setup(), plpgsql_param_eval_generic(), plpgsql_param_eval_generic_ro(), and plpgsql_param_fetch().

◆ exec_eval_expr()

static Datum exec_eval_expr	(	PLpgSQL_execstate *	estate,
		PLpgSQL_expr *	expr,
		bool *	isNull,
		Oid *	rettype,
		int32 *	rettypmod
	)

static

Definition at line 5683 of file pl_exec.c.

{
    Datum       result = 0;
    int         rc;
    Form_pg_attribute attr;
 
    /*
     * If first time through, create a plan for this expression.
     */
    if (expr->plan == NULL)
        exec_prepare_plan(estate, expr, CURSOR_OPT_PARALLEL_OK);
 
    /*
     * If this is a simple expression, bypass SPI and use the executor
     * directly
     */
    if (exec_eval_simple_expr(estate, expr,
                              &result, isNull, rettype, rettypmod))
        return result;
 
    /*
     * Else do it the hard way via exec_run_select
     */
    rc = exec_run_select(estate, expr, 2, NULL);
    if (rc != SPI_OK_SELECT)
        ereport(ERROR,
                (errcode(ERRCODE_WRONG_OBJECT_TYPE),
                 errmsg("query did not return data"),
                 errcontext("query: %s", expr->query)));
 
    /*
     * Check that the expression returns exactly one column...
     */
    if (estate->eval_tuptable->tupdesc->natts != 1)
        ereport(ERROR,
                (errcode(ERRCODE_SYNTAX_ERROR),
                 errmsg_plural("query returned %d column",
                               "query returned %d columns",
                               estate->eval_tuptable->tupdesc->natts,
                               estate->eval_tuptable->tupdesc->natts),
                 errcontext("query: %s", expr->query)));
 
    /*
     * ... and get the column's datatype.
     */
    attr = TupleDescAttr(estate->eval_tuptable->tupdesc, 0);
    *rettype = attr->atttypid;
    *rettypmod = attr->atttypmod;
 
    /*
     * If there are no rows selected, the result is a NULL of that type.
     */
    if (estate->eval_processed == 0)
    {
        *isNull = true;
        return (Datum) 0;
    }
 
    /*
     * Check that the expression returned no more than one row.
     */
    if (estate->eval_processed != 1)
        ereport(ERROR,
                (errcode(ERRCODE_CARDINALITY_VIOLATION),
                 errmsg("query returned more than one row"),
                 errcontext("query: %s", expr->query)));
 
    /*
     * Return the single result Datum.
     */
    return SPI_getbinval(estate->eval_tuptable->vals[0],
                         estate->eval_tuptable->tupdesc, 1, isNull);
}

References CURSOR_OPT_PARALLEL_OK, ereport, errcode(), errcontext, errmsg(), errmsg_plural(), ERROR, PLpgSQL_execstate::eval_processed, PLpgSQL_execstate::eval_tuptable, exec_eval_simple_expr(), exec_prepare_plan(), exec_run_select(), TupleDescData::natts, PLpgSQL_expr::plan, PLpgSQL_expr::query, SPI_getbinval(), SPI_OK_SELECT, SPITupleTable::tupdesc, TupleDescAttr(), and SPITupleTable::vals.

Referenced by exec_assign_expr(), exec_dynquery_with_params(), exec_eval_boolean(), exec_eval_integer(), exec_eval_using_params(), exec_stmt_assert(), exec_stmt_case(), exec_stmt_dynexecute(), exec_stmt_foreach_a(), exec_stmt_fori(), exec_stmt_raise(), exec_stmt_return(), exec_stmt_return_next(), and exec_stmt_return_query().

◆ exec_eval_integer()

static int exec_eval_integer	(	PLpgSQL_execstate *	estate,
		PLpgSQL_expr *	expr,
		bool *	isNull
	)

static

Definition at line 5637 of file pl_exec.c.

{
    Datum       exprdatum;
    Oid         exprtypeid;
    int32       exprtypmod;
 
    exprdatum = exec_eval_expr(estate, expr, isNull, &exprtypeid, &exprtypmod);
    exprdatum = exec_cast_value(estate, exprdatum, isNull,
                                exprtypeid, exprtypmod,
                                INT4OID, -1);
    return DatumGetInt32(exprdatum);
}

References DatumGetInt32(), exec_cast_value(), and exec_eval_expr().

Referenced by exec_stmt_fetch().

◆ exec_eval_simple_expr()

static bool exec_eval_simple_expr	(	PLpgSQL_execstate *	estate,
		PLpgSQL_expr *	expr,
		Datum *	result,
		bool *	isNull,
		Oid *	rettype,
		int32 *	rettypmod
	)

static

Definition at line 6033 of file pl_exec.c.

{
    ExprContext *econtext = estate->eval_econtext;
    LocalTransactionId curlxid = MyProc->vxid.lxid;
    ParamListInfo paramLI;
    void       *save_setup_arg;
    bool        need_snapshot;
    MemoryContext oldcontext;
 
    /*
     * Forget it if expression wasn't simple before.
     */
    if (expr->expr_simple_expr == NULL)
        return false;
 
    /*
     * If expression is in use in current xact, don't touch it.
     */
    if (unlikely(expr->expr_simple_in_use) &&
        expr->expr_simple_lxid == curlxid)
        return false;
 
    /*
     * Ensure that there's a portal-level snapshot, in case this simple
     * expression is the first thing evaluated after a COMMIT or ROLLBACK.
     * We'd have to do this anyway before executing the expression, so we
     * might as well do it now to ensure that any possible replanning doesn't
     * need to take a new snapshot.
     */
    EnsurePortalSnapshotExists();
 
    /*
     * Check to see if the cached plan has been invalidated.  If not, and this
     * is the first use in the current transaction, save a plan refcount in
     * the simple-expression resowner.
     */
    if (likely(CachedPlanIsSimplyValid(expr->expr_simple_plansource,
                                       expr->expr_simple_plan,
                                       (expr->expr_simple_plan_lxid != curlxid ?
                                        estate->simple_eval_resowner : NULL))))
    {
        /*
         * It's still good, so just remember that we have a refcount on the
         * plan in the current transaction.  (If we already had one, this
         * assignment is a no-op.)
         */
        expr->expr_simple_plan_lxid = curlxid;
    }
    else
    {
        /* Need to replan */
        CachedPlan *cplan;
 
        /*
         * If we have a valid refcount on some previous version of the plan,
         * release it, so we don't leak plans intra-transaction.
         */
        if (expr->expr_simple_plan_lxid == curlxid)
            ReleaseCachedPlan(expr->expr_simple_plan,
                              estate->simple_eval_resowner);
 
        /*
         * Reset to "not simple" to leave sane state (with no dangling
         * pointers) in case we fail while replanning.  expr_simple_plansource
         * can be left alone however, as that cannot move.
         */
        expr->expr_simple_expr = NULL;
        expr->expr_rw_param = NULL;
        expr->expr_simple_plan = NULL;
        expr->expr_simple_plan_lxid = InvalidLocalTransactionId;
 
        /* Do the replanning work in the eval_mcontext */
        oldcontext = MemoryContextSwitchTo(get_eval_mcontext(estate));
        cplan = SPI_plan_get_cached_plan(expr->plan);
        MemoryContextSwitchTo(oldcontext);
 
        /*
         * We can't get a failure here, because the number of
         * CachedPlanSources in the SPI plan can't change from what
         * exec_simple_check_plan saw; it's a property of the raw parsetree
         * generated from the query text.
         */
        Assert(cplan != NULL);
 
        /*
         * Recheck exec_is_simple_query, which could now report false in
         * edge-case scenarios such as a non-SRF having been replaced with a
         * SRF.  Also recheck CachedPlanAllowsSimpleValidityCheck, just to be
         * sure.  If either test fails, cope by declaring the plan to be
         * non-simple.  On success, we'll acquire a refcount on the new plan,
         * stored in simple_eval_resowner.
         */
        if (exec_is_simple_query(expr) &&
            CachedPlanAllowsSimpleValidityCheck(expr->expr_simple_plansource,
                                                cplan,
                                                estate->simple_eval_resowner))
        {
            /* Remember that we have the refcount */
            expr->expr_simple_plan = cplan;
            expr->expr_simple_plan_lxid = curlxid;
        }
        else
        {
            /* Release SPI_plan_get_cached_plan's refcount */
            ReleaseCachedPlan(cplan, CurrentResourceOwner);
            return false;
        }
 
        /*
         * SPI_plan_get_cached_plan acquired a plan refcount stored in the
         * active resowner.  We don't need that anymore, so release it.
         */
        ReleaseCachedPlan(cplan, CurrentResourceOwner);
 
        /* Extract desired scalar expression from cached plan */
        exec_save_simple_expr(expr, cplan);
    }
 
    /*
     * Pass back previously-determined result type.
     */
    *rettype = expr->expr_simple_type;
    *rettypmod = expr->expr_simple_typmod;
 
    /*
     * Set up ParamListInfo to pass to executor.  For safety, save and restore
     * estate->paramLI->parserSetupArg around our use of the param list.
     */
    paramLI = estate->paramLI;
    save_setup_arg = paramLI->parserSetupArg;
 
    /*
     * We can skip using setup_param_list() in favor of just doing this
     * unconditionally, because there's no need for the optimization of
     * possibly setting ecxt_param_list_info to NULL; we've already forced use
     * of a generic plan.
     */
    paramLI->parserSetupArg = expr;
    econtext->ecxt_param_list_info = paramLI;
 
    /*
     * Prepare the expression for execution, if it's not been done already in
     * the current transaction.  (This will be forced to happen if we called
     * exec_save_simple_expr above.)
     */
    if (unlikely(expr->expr_simple_lxid != curlxid))
    {
        oldcontext = MemoryContextSwitchTo(estate->simple_eval_estate->es_query_cxt);
        expr->expr_simple_state =
            ExecInitExprWithParams(expr->expr_simple_expr,
                                   econtext->ecxt_param_list_info);
        expr->expr_simple_in_use = false;
        expr->expr_simple_lxid = curlxid;
        MemoryContextSwitchTo(oldcontext);
    }
 
    /*
     * We have to do some of the things SPI_execute_plan would do, in
     * particular push a new snapshot so that stable functions within the
     * expression can see updates made so far by our own function.  However,
     * we can skip doing that (and just invoke the expression with the same
     * snapshot passed to our function) in some cases, which is useful because
     * it's quite expensive relative to the cost of a simple expression.  We
     * can skip it if the expression contains no stable or volatile functions;
     * immutable functions shouldn't need to see our updates.  Also, if this
     * is a read-only function, we haven't made any updates so again it's okay
     * to skip.
     */
    oldcontext = MemoryContextSwitchTo(get_eval_mcontext(estate));
    need_snapshot = (expr->expr_simple_mutable && !estate->readonly_func);
    if (need_snapshot)
    {
        CommandCounterIncrement();
        PushActiveSnapshot(GetTransactionSnapshot());
    }
 
    /*
     * Mark expression as busy for the duration of the ExecEvalExpr call.
     */
    expr->expr_simple_in_use = true;
 
    /*
     * Finally we can call the executor to evaluate the expression
     */
    *result = ExecEvalExpr(expr->expr_simple_state,
                           econtext,
                           isNull);
 
    /* Assorted cleanup */
    expr->expr_simple_in_use = false;
 
    econtext->ecxt_param_list_info = NULL;
 
    paramLI->parserSetupArg = save_setup_arg;
 
    if (need_snapshot)
        PopActiveSnapshot();
 
    MemoryContextSwitchTo(oldcontext);
 
    /*
     * That's it.
     */
    return true;
}

Referenced by exec_eval_expr().

◆ exec_eval_using_params()

static ParamListInfo exec_eval_using_params	(	PLpgSQL_execstate *	estate,
		List *	params
	)

static

Definition at line 8650 of file pl_exec.c.

{
    ParamListInfo paramLI;
    int         nargs;
    MemoryContext stmt_mcontext;
    MemoryContext oldcontext;
    int         i;
    ListCell   *lc;
 
    /* Fast path for no parameters: we can just return NULL */
    if (params == NIL)
        return NULL;
 
    nargs = list_length(params);
    stmt_mcontext = get_stmt_mcontext(estate);
    oldcontext = MemoryContextSwitchTo(stmt_mcontext);
    paramLI = makeParamList(nargs);
    MemoryContextSwitchTo(oldcontext);
 
    i = 0;
    foreach(lc, params)
    {
        PLpgSQL_expr *param = (PLpgSQL_expr *) lfirst(lc);
        ParamExternData *prm = &paramLI->params[i];
        int32       ppdtypmod;
 
        /*
         * Always mark params as const, since we only use the result with
         * one-shot plans.
         */
        prm->pflags = PARAM_FLAG_CONST;
 
        prm->value = exec_eval_expr(estate, param,
                                    &prm->isnull,
                                    &prm->ptype,
                                    &ppdtypmod);
 
        oldcontext = MemoryContextSwitchTo(stmt_mcontext);
 
        if (prm->ptype == UNKNOWNOID)
        {
            /*
             * Treat 'unknown' parameters as text, since that's what most
             * people would expect.  The SPI functions can coerce unknown
             * constants in a more intelligent way, but not unknown Params.
             * This code also takes care of copying into the right context.
             * Note we assume 'unknown' has the representation of C-string.
             */
            prm->ptype = TEXTOID;
            if (!prm->isnull)
                prm->value = CStringGetTextDatum(DatumGetCString(prm->value));
        }
        /* pass-by-ref non null values must be copied into stmt_mcontext */
        else if (!prm->isnull)
        {
            int16       typLen;
            bool        typByVal;
 
            get_typlenbyval(prm->ptype, &typLen, &typByVal);
            if (!typByVal)
                prm->value = datumCopy(prm->value, typByVal, typLen);
        }
 
        MemoryContextSwitchTo(oldcontext);
 
        exec_eval_cleanup(estate);
 
        i++;
    }
 
    return paramLI;
}

References CStringGetTextDatum, datumCopy(), DatumGetCString(), exec_eval_cleanup(), exec_eval_expr(), get_stmt_mcontext(), get_typlenbyval(), i, ParamExternData::isnull, lfirst, list_length(), makeParamList(), MemoryContextSwitchTo(), NIL, PARAM_FLAG_CONST, ParamListInfoData::params, ParamExternData::pflags, ParamExternData::ptype, and ParamExternData::value.

Referenced by exec_dynquery_with_params(), exec_stmt_dynexecute(), and exec_stmt_return_query().

◆ exec_for_query()

static int exec_for_query	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_forq *	stmt,
		Portal	portal,
		bool	prefetch_ok
	)

static

Definition at line 5851 of file pl_exec.c.

{
    PLpgSQL_variable *var;
    SPITupleTable *tuptab;
    bool        found = false;
    int         rc = PLPGSQL_RC_OK;
    uint64      previous_id = INVALID_TUPLEDESC_IDENTIFIER;
    bool        tupdescs_match = true;
    uint64      n;
 
    /* Fetch loop variable's datum entry */
    var = (PLpgSQL_variable *) estate->datums[stmt->var->dno];
 
    /*
     * Make sure the portal doesn't get closed by the user statements we
     * execute.
     */
    PinPortal(portal);
 
    /*
     * In a non-atomic context, we dare not prefetch, even if it would
     * otherwise be safe.  Aside from any semantic hazards that that might
     * create, if we prefetch toasted data and then the user commits the
     * transaction, the toast references could turn into dangling pointers.
     * (Rows we haven't yet fetched from the cursor are safe, because the
     * PersistHoldablePortal mechanism handles this scenario.)
     */
    if (!estate->atomic)
        prefetch_ok = false;
 
    /*
     * Fetch the initial tuple(s).  If prefetching is allowed then we grab a
     * few more rows to avoid multiple trips through executor startup
     * overhead.
     */
    SPI_cursor_fetch(portal, true, prefetch_ok ? 10 : 1);
    tuptab = SPI_tuptable;
    n = SPI_processed;
 
    /*
     * If the query didn't return any rows, set the target to NULL and fall
     * through with found = false.
     */
    if (n == 0)
    {
        exec_move_row(estate, var, NULL, tuptab->tupdesc);
        exec_eval_cleanup(estate);
    }
    else
        found = true;           /* processed at least one tuple */
 
    /*
     * Now do the loop
     */
    while (n > 0)
    {
        uint64      i;
 
        for (i = 0; i < n; i++)
        {
            /*
             * Assign the tuple to the target.  Here, because we know that all
             * loop iterations should be assigning the same tupdesc, we can
             * optimize away repeated creations of expanded records with
             * identical tupdescs.  Testing for changes of er_tupdesc_id is
             * reliable even if the loop body contains assignments that
             * replace the target's value entirely, because it's assigned from
             * a process-global counter.  The case where the tupdescs don't
             * match could possibly be handled more efficiently than this
             * coding does, but it's not clear extra effort is worthwhile.
             */
            if (var->dtype == PLPGSQL_DTYPE_REC)
            {
                PLpgSQL_rec *rec = (PLpgSQL_rec *) var;
 
                if (rec->erh &&
                    rec->erh->er_tupdesc_id == previous_id &&
                    tupdescs_match)
                {
                    /* Only need to assign a new tuple value */
                    expanded_record_set_tuple(rec->erh, tuptab->vals[i],
                                              true, !estate->atomic);
                }
                else
                {
                    /*
                     * First time through, or var's tupdesc changed in loop,
                     * or we have to do it the hard way because type coercion
                     * is needed.
                     */
                    exec_move_row(estate, var,
                                  tuptab->vals[i], tuptab->tupdesc);
 
                    /*
                     * Check to see if physical assignment is OK next time.
                     * Once the tupdesc comparison has failed once, we don't
                     * bother rechecking in subsequent loop iterations.
                     */
                    if (tupdescs_match)
                    {
                        tupdescs_match =
                            (rec->rectypeid == RECORDOID ||
                             rec->rectypeid == tuptab->tupdesc->tdtypeid ||
                             compatible_tupdescs(tuptab->tupdesc,
                                                 expanded_record_get_tupdesc(rec->erh)));
                    }
                    previous_id = rec->erh->er_tupdesc_id;
                }
            }
            else
                exec_move_row(estate, var, tuptab->vals[i], tuptab->tupdesc);
 
            exec_eval_cleanup(estate);
 
            /*
             * Execute the statements
             */
            rc = exec_stmts(estate, stmt->body);
 
            LOOP_RC_PROCESSING(stmt->label, goto loop_exit);
        }
 
        SPI_freetuptable(tuptab);
 
        /*
         * Fetch more tuples.  If prefetching is allowed, grab 50 at a time.
         */
        SPI_cursor_fetch(portal, true, prefetch_ok ? 50 : 1);
        tuptab = SPI_tuptable;
        n = SPI_processed;
    }
 
loop_exit:
 
    /*
     * Release last group of tuples (if any)
     */
    SPI_freetuptable(tuptab);
 
    UnpinPortal(portal);
 
    /*
     * Set the FOUND variable to indicate the result of executing the loop
     * (namely, whether we looped one or more times). This must be set last so
     * that it does not interfere with the value of the FOUND variable inside
     * the loop processing itself.
     */
    exec_set_found(estate, found);
 
    return rc;
}

References PLpgSQL_execstate::atomic, compatible_tupdescs(), PLpgSQL_execstate::datums, PLpgSQL_variable::dtype, ExpandedRecordHeader::er_tupdesc_id, PLpgSQL_rec::erh, exec_eval_cleanup(), exec_move_row(), exec_set_found(), exec_stmts(), expanded_record_get_tupdesc(), expanded_record_set_tuple(), i, INVALID_TUPLEDESC_IDENTIFIER, LOOP_RC_PROCESSING, PinPortal(), PLPGSQL_DTYPE_REC, PLPGSQL_RC_OK, PLpgSQL_rec::rectypeid, SPI_cursor_fetch(), SPI_freetuptable(), SPI_processed, SPI_tuptable, stmt, TupleDescData::tdtypeid, SPITupleTable::tupdesc, UnpinPortal(), and SPITupleTable::vals.

Referenced by exec_stmt_dynfors(), exec_stmt_forc(), and exec_stmt_fors().

◆ exec_init_tuple_store()

static void exec_init_tuple_store ( PLpgSQL_execstate * estate )

static

Definition at line 3667 of file pl_exec.c.

{
    ReturnSetInfo *rsi = estate->rsi;
    MemoryContext oldcxt;
    ResourceOwner oldowner;
 
    /*
     * Check caller can handle a set result in the way we want
     */
    if (!rsi || !IsA(rsi, ReturnSetInfo))
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("set-valued function called in context that cannot accept a set")));
 
    if (!(rsi->allowedModes & SFRM_Materialize) ||
        rsi->expectedDesc == NULL)
        ereport(ERROR,
                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                 errmsg("materialize mode required, but it is not allowed in this context")));
 
    /*
     * Switch to the right memory context and resource owner for storing the
     * tuplestore for return set. If we're within a subtransaction opened for
     * an exception-block, for example, we must still create the tuplestore in
     * the resource owner that was active when this function was entered, and
     * not in the subtransaction resource owner.
     */
    oldcxt = MemoryContextSwitchTo(estate->tuple_store_cxt);
    oldowner = CurrentResourceOwner;
    CurrentResourceOwner = estate->tuple_store_owner;
 
    estate->tuple_store =
        tuplestore_begin_heap(rsi->allowedModes & SFRM_Materialize_Random,
                              false, work_mem);
 
    CurrentResourceOwner = oldowner;
    MemoryContextSwitchTo(oldcxt);
 
    estate->tuple_store_desc = rsi->expectedDesc;
}

References ReturnSetInfo::allowedModes, CurrentResourceOwner, ereport, errcode(), errmsg(), ERROR, ReturnSetInfo::expectedDesc, IsA, MemoryContextSwitchTo(), PLpgSQL_execstate::rsi, SFRM_Materialize, SFRM_Materialize_Random, PLpgSQL_execstate::tuple_store, PLpgSQL_execstate::tuple_store_cxt, PLpgSQL_execstate::tuple_store_desc, PLpgSQL_execstate::tuple_store_owner, tuplestore_begin_heap(), and work_mem.

Referenced by exec_stmt_return_next(), and exec_stmt_return_query().

◆ exec_is_simple_query()

static bool exec_is_simple_query ( PLpgSQL_expr * expr )

static

Definition at line 8050 of file pl_exec.c.

{
    List       *plansources;
    CachedPlanSource *plansource;
    Query      *query;
 
    /*
     * We can only test queries that resulted in exactly one CachedPlanSource.
     */
    plansources = SPI_plan_get_plan_sources(expr->plan);
    if (list_length(plansources) != 1)
        return false;
    plansource = (CachedPlanSource *) linitial(plansources);
 
    /*
     * 1. There must be one single querytree.
     */
    if (list_length(plansource->query_list) != 1)
        return false;
    query = (Query *) linitial(plansource->query_list);
 
    /*
     * 2. It must be a plain SELECT query without any input tables.
     */
    if (!IsA(query, Query))
        return false;
    if (query->commandType != CMD_SELECT)
        return false;
    if (query->rtable != NIL)
        return false;
 
    /*
     * 3. Can't have any subplans, aggregates, qual clauses either.  (These
     * tests should generally match what inline_function() checks before
     * inlining a SQL function; otherwise, inlining could change our
     * conclusion about whether an expression is simple, which we don't want.)
     */
    if (query->hasAggs ||
        query->hasWindowFuncs ||
        query->hasTargetSRFs ||
        query->hasSubLinks ||
        query->cteList ||
        query->jointree->fromlist ||
        query->jointree->quals ||
        query->groupClause ||
        query->groupingSets ||
        query->havingQual ||
        query->windowClause ||
        query->distinctClause ||
        query->sortClause ||
        query->limitOffset ||
        query->limitCount ||
        query->setOperations)
        return false;
 
    /*
     * 4. The query must have a single attribute as result.
     */
    if (list_length(query->targetList) != 1)
        return false;
 
    /*
     * OK, we can treat it as a simple plan.
     */
    return true;
}

References CMD_SELECT, Query::commandType, Query::cteList, Query::distinctClause, FromExpr::fromlist, Query::groupClause, Query::groupingSets, Query::havingQual, IsA, Query::jointree, Query::limitCount, Query::limitOffset, linitial, list_length(), NIL, PLpgSQL_expr::plan, FromExpr::quals, CachedPlanSource::query_list, Query::rtable, Query::setOperations, Query::sortClause, SPI_plan_get_plan_sources(), Query::targetList, and Query::windowClause.

Referenced by exec_eval_simple_expr(), and exec_simple_check_plan().

◆ exec_move_row()

static void exec_move_row	(	PLpgSQL_execstate *	estate,
		PLpgSQL_variable *	target,
		HeapTuple	tup,
		TupleDesc	tupdesc
	)

static

Definition at line 6749 of file pl_exec.c.

{
    ExpandedRecordHeader *newerh = NULL;
 
    /*
     * If target is RECORD, we may be able to avoid field-by-field processing.
     */
    if (target->dtype == PLPGSQL_DTYPE_REC)
    {
        PLpgSQL_rec *rec = (PLpgSQL_rec *) target;
 
        /*
         * If we have no source tupdesc, just set the record variable to NULL.
         * (If we have a source tupdesc but not a tuple, we'll set the
         * variable to a row of nulls, instead.  This is odd perhaps, but
         * backwards compatible.)
         */
        if (tupdesc == NULL)
        {
            if (rec->datatype &&
                rec->datatype->typtype == TYPTYPE_DOMAIN)
            {
                /*
                 * If it's a composite domain, NULL might not be a legal
                 * value, so we instead need to make an empty expanded record
                 * and ensure that domain type checking gets done.  If there
                 * is already an expanded record, piggyback on its lookups.
                 */
                newerh = make_expanded_record_for_rec(estate, rec,
                                                      NULL, rec->erh);
                expanded_record_set_tuple(newerh, NULL, false, false);
                assign_record_var(estate, rec, newerh);
            }
            else
            {
                /* Just clear it to NULL */
                if (rec->erh)
                    DeleteExpandedObject(ExpandedRecordGetDatum(rec->erh));
                rec->erh = NULL;
            }
            return;
        }
 
        /*
         * Build a new expanded record with appropriate tupdesc.
         */
        newerh = make_expanded_record_for_rec(estate, rec, tupdesc, NULL);
 
        /*
         * If the rowtypes match, or if we have no tuple anyway, we can
         * complete the assignment without field-by-field processing.
         *
         * The tests here are ordered more or less in order of cheapness.  We
         * can easily detect it will work if the target is declared RECORD or
         * has the same typeid as the source.  But when assigning from a query
         * result, it's common to have a source tupdesc that's labeled RECORD
         * but is actually physically compatible with a named-composite-type
         * target, so it's worth spending extra cycles to check for that.
         */
        if (rec->rectypeid == RECORDOID ||
            rec->rectypeid == tupdesc->tdtypeid ||
            !HeapTupleIsValid(tup) ||
            compatible_tupdescs(tupdesc, expanded_record_get_tupdesc(newerh)))
        {
            if (!HeapTupleIsValid(tup))
            {
                /* No data, so force the record into all-nulls state */
                deconstruct_expanded_record(newerh);
            }
            else
            {
                /* No coercion is needed, so just assign the row value */
                expanded_record_set_tuple(newerh, tup, true, !estate->atomic);
            }
 
            /* Complete the assignment */
            assign_record_var(estate, rec, newerh);
 
            return;
        }
    }
 
    /*
     * Otherwise, deconstruct the tuple and do field-by-field assignment,
     * using exec_move_row_from_fields.
     */
    if (tupdesc && HeapTupleIsValid(tup))
    {
        int         td_natts = tupdesc->natts;
        Datum      *values;
        bool       *nulls;
        Datum       values_local[64];
        bool        nulls_local[64];
 
        /*
         * Need workspace arrays.  If td_natts is small enough, use local
         * arrays to save doing a palloc.  Even if it's not small, we can
         * allocate both the Datum and isnull arrays in one palloc chunk.
         */
        if (td_natts <= lengthof(values_local))
        {
            values = values_local;
            nulls = nulls_local;
        }
        else
        {
            char       *chunk;
 
            chunk = eval_mcontext_alloc(estate,
                                        td_natts * (sizeof(Datum) + sizeof(bool)));
            values = (Datum *) chunk;
            nulls = (bool *) (chunk + td_natts * sizeof(Datum));
        }
 
        heap_deform_tuple(tup, tupdesc, values, nulls);
 
        exec_move_row_from_fields(estate, target, newerh,
                                  values, nulls, tupdesc);
    }
    else
    {
        /*
         * Assign all-nulls.
         */
        exec_move_row_from_fields(estate, target, newerh,
                                  NULL, NULL, NULL);
    }
}

References assign_record_var(), PLpgSQL_execstate::atomic, chunk, compatible_tupdescs(), PLpgSQL_rec::datatype, deconstruct_expanded_record(), DeleteExpandedObject(), PLpgSQL_variable::dtype, PLpgSQL_rec::erh, eval_mcontext_alloc, exec_move_row_from_fields(), expanded_record_get_tupdesc(), expanded_record_set_tuple(), ExpandedRecordGetDatum(), heap_deform_tuple(), HeapTupleIsValid, lengthof, make_expanded_record_for_rec(), TupleDescData::natts, PLPGSQL_DTYPE_REC, PLpgSQL_rec::rectypeid, TupleDescData::tdtypeid, PLpgSQL_type::typtype, and values.

Referenced by exec_assign_value(), exec_for_query(), exec_move_row_from_datum(), exec_stmt_block(), exec_stmt_call(), exec_stmt_dynexecute(), exec_stmt_execsql(), exec_stmt_fetch(), and plpgsql_exec_function().

◆ exec_move_row_from_datum()

static void exec_move_row_from_datum	(	PLpgSQL_execstate *	estate,
		PLpgSQL_variable *	target,
		Datum	value
	)

static

Definition at line 7423 of file pl_exec.c.

{
    /* Check to see if source is an expanded record */
    if (VARATT_IS_EXTERNAL_EXPANDED(DatumGetPointer(value)))
    {
        ExpandedRecordHeader *erh = (ExpandedRecordHeader *) DatumGetEOHP(value);
        ExpandedRecordHeader *newerh = NULL;
 
        Assert(erh->er_magic == ER_MAGIC);
 
        /* These cases apply if the target is record not row... */
        if (target->dtype == PLPGSQL_DTYPE_REC)
        {
            PLpgSQL_rec *rec = (PLpgSQL_rec *) target;
 
            /*
             * If it's the same record already stored in the variable, do
             * nothing.  This would happen only in silly cases like "r := r",
             * but we need some check to avoid possibly freeing the variable's
             * live value below.  Note that this applies even if what we have
             * is a R/O pointer.
             */
            if (erh == rec->erh)
                return;
 
            /*
             * Make sure rec->rectypeid is up-to-date before using it.
             */
            revalidate_rectypeid(rec);
 
            /*
             * If we have a R/W pointer, we're allowed to just commandeer
             * ownership of the expanded record.  If it's of the right type to
             * put into the record variable, do that.  (Note we don't accept
             * an expanded record of a composite-domain type as a RECORD
             * value.  We'll treat it as the base composite type instead;
             * compare logic in make_expanded_record_for_rec.)
             */
            if (VARATT_IS_EXTERNAL_EXPANDED_RW(DatumGetPointer(value)) &&
                (rec->rectypeid == erh->er_decltypeid ||
                 (rec->rectypeid == RECORDOID &&
                  !ExpandedRecordIsDomain(erh))))
            {
                assign_record_var(estate, rec, erh);
                return;
            }
 
            /*
             * If we already have an expanded record object in the target
             * variable, and the source record contains a valid tuple
             * representation with the right rowtype, then we can skip making
             * a new expanded record and just assign the tuple with
             * expanded_record_set_tuple.  (We can't do the equivalent if we
             * have to do field-by-field assignment, since that wouldn't be
             * atomic if there's an error.)  We consider that there's a
             * rowtype match only if it's the same named composite type or
             * same registered rowtype; checking for matches of anonymous
             * rowtypes would be more expensive than this is worth.
             */
            if (rec->erh &&
                (erh->flags & ER_FLAG_FVALUE_VALID) &&
                erh->er_typeid == rec->erh->er_typeid &&
                (erh->er_typeid != RECORDOID ||
                 (erh->er_typmod == rec->erh->er_typmod &&
                  erh->er_typmod >= 0)))
            {
                expanded_record_set_tuple(rec->erh, erh->fvalue,
                                          true, !estate->atomic);
                return;
            }
 
            /*
             * Otherwise we're gonna need a new expanded record object.  Make
             * it here in hopes of piggybacking on the source object's
             * previous typcache lookup.
             */
            newerh = make_expanded_record_for_rec(estate, rec, NULL, erh);
 
            /*
             * If the expanded record contains a valid tuple representation,
             * and we don't need rowtype conversion, then just copying the
             * tuple is probably faster than field-by-field processing.  (This
             * isn't duplicative of the previous check, since here we will
             * catch the case where the record variable was previously empty.)
             */
            if ((erh->flags & ER_FLAG_FVALUE_VALID) &&
                (rec->rectypeid == RECORDOID ||
                 rec->rectypeid == erh->er_typeid))
            {
                expanded_record_set_tuple(newerh, erh->fvalue,
                                          true, !estate->atomic);
                assign_record_var(estate, rec, newerh);
                return;
            }
 
            /*
             * Need to special-case empty source record, else code below would
             * leak newerh.
             */
            if (ExpandedRecordIsEmpty(erh))
            {
                /* Set newerh to a row of NULLs */
                deconstruct_expanded_record(newerh);
                assign_record_var(estate, rec, newerh);
                return;
            }
        }                       /* end of record-target-only cases */
 
        /*
         * If the source expanded record is empty, we should treat that like a
         * NULL tuple value.  (We're unlikely to see such a case, but we must
         * check this; deconstruct_expanded_record would cause a change of
         * logical state, which is not OK.)
         */
        if (ExpandedRecordIsEmpty(erh))
        {
            exec_move_row(estate, target, NULL,
                          expanded_record_get_tupdesc(erh));
            return;
        }
 
        /*
         * Otherwise, ensure that the source record is deconstructed, and
         * assign from its field values.
         */
        deconstruct_expanded_record(erh);
        exec_move_row_from_fields(estate, target, newerh,
                                  erh->dvalues, erh->dnulls,
                                  expanded_record_get_tupdesc(erh));
    }
    else
    {
        /*
         * Nope, we've got a plain composite Datum.  Deconstruct it; but we
         * don't use deconstruct_composite_datum(), because we may be able to
         * skip calling lookup_rowtype_tupdesc().
         */
        HeapTupleHeader td;
        HeapTupleData tmptup;
        Oid         tupType;
        int32       tupTypmod;
        TupleDesc   tupdesc;
        MemoryContext oldcontext;
 
        /* Ensure that any detoasted data winds up in the eval_mcontext */
        oldcontext = MemoryContextSwitchTo(get_eval_mcontext(estate));
        /* Get tuple body (note this could involve detoasting) */
        td = DatumGetHeapTupleHeader(value);
        MemoryContextSwitchTo(oldcontext);
 
        /* Build a temporary HeapTuple control structure */
        tmptup.t_len = HeapTupleHeaderGetDatumLength(td);
        ItemPointerSetInvalid(&(tmptup.t_self));
        tmptup.t_tableOid = InvalidOid;
        tmptup.t_data = td;
 
        /* Extract rowtype info */
        tupType = HeapTupleHeaderGetTypeId(td);
        tupTypmod = HeapTupleHeaderGetTypMod(td);
 
        /* Now, if the target is record not row, maybe we can optimize ... */
        if (target->dtype == PLPGSQL_DTYPE_REC)
        {
            PLpgSQL_rec *rec = (PLpgSQL_rec *) target;
 
            /*
             * If we already have an expanded record object in the target
             * variable, and the source datum has a matching rowtype, then we
             * can skip making a new expanded record and just assign the tuple
             * with expanded_record_set_tuple.  We consider that there's a
             * rowtype match only if it's the same named composite type or
             * same registered rowtype.  (Checking to reject an anonymous
             * rowtype here should be redundant, but let's be safe.)
             */
            if (rec->erh &&
                tupType == rec->erh->er_typeid &&
                (tupType != RECORDOID ||
                 (tupTypmod == rec->erh->er_typmod &&
                  tupTypmod >= 0)))
            {
                expanded_record_set_tuple(rec->erh, &tmptup,
                                          true, !estate->atomic);
                return;
            }
 
            /*
             * If the source datum has a rowtype compatible with the target
             * variable, just build a new expanded record and assign the tuple
             * into it.  Using make_expanded_record_from_typeid() here saves
             * one typcache lookup compared to the code below.
             */
            if (rec->rectypeid == RECORDOID || rec->rectypeid == tupType)
            {
                ExpandedRecordHeader *newerh;
                MemoryContext mcontext = get_eval_mcontext(estate);
 
                newerh = make_expanded_record_from_typeid(tupType, tupTypmod,
                                                          mcontext);
                expanded_record_set_tuple(newerh, &tmptup,
                                          true, !estate->atomic);
                assign_record_var(estate, rec, newerh);
                return;
            }
 
            /*
             * Otherwise, we're going to need conversion, so fall through to
             * do it the hard way.
             */
        }
 
        /*
         * ROW target, or unoptimizable RECORD target, so we have to expend a
         * lookup to obtain the source datum's tupdesc.
         */
        tupdesc = lookup_rowtype_tupdesc(tupType, tupTypmod);
 
        /* Do the move */
        exec_move_row(estate, target, &tmptup, tupdesc);
 
        /* Release tupdesc usage count */
        ReleaseTupleDesc(tupdesc);
    }
}

Referenced by exec_assign_value(), and plpgsql_exec_function().

◆ exec_move_row_from_fields()

static void exec_move_row_from_fields	(	PLpgSQL_execstate *	estate,
		PLpgSQL_variable *	target,
		ExpandedRecordHeader *	newerh,
		Datum *	values,
		bool *	nulls,
		TupleDesc	tupdesc
	)

static

Definition at line 7027 of file pl_exec.c.

{
    int         td_natts = tupdesc ? tupdesc->natts : 0;
    int         fnum;
    int         anum;
    int         strict_multiassignment_level = 0;
 
    /*
     * The extra check strict strict_multi_assignment can be active, only when
     * input tupdesc is specified.
     */
    if (tupdesc != NULL)
    {
        if (plpgsql_extra_errors & PLPGSQL_XCHECK_STRICTMULTIASSIGNMENT)
            strict_multiassignment_level = ERROR;
        else if (plpgsql_extra_warnings & PLPGSQL_XCHECK_STRICTMULTIASSIGNMENT)
            strict_multiassignment_level = WARNING;
    }
 
    /* Handle RECORD-target case */
    if (target->dtype == PLPGSQL_DTYPE_REC)
    {
        PLpgSQL_rec *rec = (PLpgSQL_rec *) target;
        TupleDesc   var_tupdesc;
        Datum       newvalues_local[64];
        bool        newnulls_local[64];
 
        Assert(newerh != NULL); /* caller must have built new object */
 
        var_tupdesc = expanded_record_get_tupdesc(newerh);
 
        /*
         * Coerce field values if needed.  This might involve dealing with
         * different sets of dropped columns and/or coercing individual column
         * types.  That's sort of a pain, but historically plpgsql has allowed
         * it, so we preserve the behavior.  However, it's worth a quick check
         * to see if the tupdescs are identical.  (Since expandedrecord.c
         * prefers to use refcounted tupdescs from the typcache, expanded
         * records with the same rowtype will have pointer-equal tupdescs.)
         */
        if (var_tupdesc != tupdesc)
        {
            int         vtd_natts = var_tupdesc->natts;
            Datum      *newvalues;
            bool       *newnulls;
 
            /*
             * Need workspace arrays.  If vtd_natts is small enough, use local
             * arrays to save doing a palloc.  Even if it's not small, we can
             * allocate both the Datum and isnull arrays in one palloc chunk.
             */
            if (vtd_natts <= lengthof(newvalues_local))
            {
                newvalues = newvalues_local;
                newnulls = newnulls_local;
            }
            else
            {
                char       *chunk;
 
                chunk = eval_mcontext_alloc(estate,
                                            vtd_natts * (sizeof(Datum) + sizeof(bool)));
                newvalues = (Datum *) chunk;
                newnulls = (bool *) (chunk + vtd_natts * sizeof(Datum));
            }
 
            /* Walk over destination columns */
            anum = 0;
            for (fnum = 0; fnum < vtd_natts; fnum++)
            {
                Form_pg_attribute attr = TupleDescAttr(var_tupdesc, fnum);
                Datum       value;
                bool        isnull;
                Oid         valtype;
                int32       valtypmod;
 
                if (attr->attisdropped)
                {
                    /* expanded_record_set_fields should ignore this column */
                    continue;   /* skip dropped column in record */
                }
 
                while (anum < td_natts &&
                       TupleDescAttr(tupdesc, anum)->attisdropped)
                    anum++;     /* skip dropped column in tuple */
 
                if (anum < td_natts)
                {
                    value = values[anum];
                    isnull = nulls[anum];
                    valtype = TupleDescAttr(tupdesc, anum)->atttypid;
                    valtypmod = TupleDescAttr(tupdesc, anum)->atttypmod;
                    anum++;
                }
                else
                {
                    /* no source for destination column */
                    value = (Datum) 0;
                    isnull = true;
                    valtype = UNKNOWNOID;
                    valtypmod = -1;
 
                    /* When source value is missing */
                    if (strict_multiassignment_level)
                        ereport(strict_multiassignment_level,
                                (errcode(ERRCODE_DATATYPE_MISMATCH),
                                 errmsg("number of source and target fields in assignment does not match"),
                        /* translator: %s represents a name of an extra check */
                                 errdetail("%s check of %s is active.",
                                           "strict_multi_assignment",
                                           strict_multiassignment_level == ERROR ? "extra_errors" :
                                           "extra_warnings"),
                                 errhint("Make sure the query returns the exact list of columns.")));
                }
 
                /* Cast the new value to the right type, if needed. */
                newvalues[fnum] = exec_cast_value(estate,
                                                  value,
                                                  &isnull,
                                                  valtype,
                                                  valtypmod,
                                                  attr->atttypid,
                                                  attr->atttypmod);
                newnulls[fnum] = isnull;
            }
 
            /*
             * When strict_multiassignment extra check is active, then ensure
             * there are no unassigned source attributes.
             */
            if (strict_multiassignment_level && anum < td_natts)
            {
                /* skip dropped columns in the source descriptor */
                while (anum < td_natts &&
                       TupleDescAttr(tupdesc, anum)->attisdropped)
                    anum++;
 
                if (anum < td_natts)
                    ereport(strict_multiassignment_level,
                            (errcode(ERRCODE_DATATYPE_MISMATCH),
                             errmsg("number of source and target fields in assignment does not match"),
                    /* translator: %s represents a name of an extra check */
                             errdetail("%s check of %s is active.",
                                       "strict_multi_assignment",
                                       strict_multiassignment_level == ERROR ? "extra_errors" :
                                       "extra_warnings"),
                             errhint("Make sure the query returns the exact list of columns.")));
            }
 
            values = newvalues;
            nulls = newnulls;
        }
 
        /* Insert the coerced field values into the new expanded record */
        expanded_record_set_fields(newerh, values, nulls, !estate->atomic);
 
        /* Complete the assignment */
        assign_record_var(estate, rec, newerh);
 
        return;
    }
 
    /* newerh should not have been passed in non-RECORD cases */
    Assert(newerh == NULL);
 
    /*
     * For a row, we assign the individual field values to the variables the
     * row points to.
     *
     * NOTE: both this code and the record code above silently ignore extra
     * columns in the source and assume NULL for missing columns.  This is
     * pretty dubious but it's the historical behavior.
     *
     * If we have no input data at all, we'll assign NULL to all columns of
     * the row variable.
     */
    if (target->dtype == PLPGSQL_DTYPE_ROW)
    {
        PLpgSQL_row *row = (PLpgSQL_row *) target;
 
        anum = 0;
        for (fnum = 0; fnum < row->nfields; fnum++)
        {
            PLpgSQL_var *var;
            Datum       value;
            bool        isnull;
            Oid         valtype;
            int32       valtypmod;
 
            var = (PLpgSQL_var *) (estate->datums[row->varnos[fnum]]);
 
            while (anum < td_natts &&
                   TupleDescAttr(tupdesc, anum)->attisdropped)
                anum++;         /* skip dropped column in tuple */
 
            if (anum < td_natts)
            {
                value = values[anum];
                isnull = nulls[anum];
                valtype = TupleDescAttr(tupdesc, anum)->atttypid;
                valtypmod = TupleDescAttr(tupdesc, anum)->atttypmod;
                anum++;
            }
            else
            {
                /* no source for destination column */
                value = (Datum) 0;
                isnull = true;
                valtype = UNKNOWNOID;
                valtypmod = -1;
 
                if (strict_multiassignment_level)
                    ereport(strict_multiassignment_level,
                            (errcode(ERRCODE_DATATYPE_MISMATCH),
                             errmsg("number of source and target fields in assignment does not match"),
                    /* translator: %s represents a name of an extra check */
                             errdetail("%s check of %s is active.",
                                       "strict_multi_assignment",
                                       strict_multiassignment_level == ERROR ? "extra_errors" :
                                       "extra_warnings"),
                             errhint("Make sure the query returns the exact list of columns.")));
            }
 
            exec_assign_value(estate, (PLpgSQL_datum *) var,
                              value, isnull, valtype, valtypmod);
        }
 
        /*
         * When strict_multiassignment extra check is active, ensure there are
         * no unassigned source attributes.
         */
        if (strict_multiassignment_level && anum < td_natts)
        {
            while (anum < td_natts &&
                   TupleDescAttr(tupdesc, anum)->attisdropped)
                anum++;         /* skip dropped column in tuple */
 
            if (anum < td_natts)
                ereport(strict_multiassignment_level,
                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                         errmsg("number of source and target fields in assignment does not match"),
                /* translator: %s represents a name of an extra check */
                         errdetail("%s check of %s is active.",
                                   "strict_multi_assignment",
                                   strict_multiassignment_level == ERROR ? "extra_errors" :
                                   "extra_warnings"),
                         errhint("Make sure the query returns the exact list of columns.")));
        }
 
        return;
    }
 
    elog(ERROR, "unsupported target type: %d", target->dtype);
}

References Assert, assign_record_var(), PLpgSQL_execstate::atomic, chunk, PLpgSQL_execstate::datums, PLpgSQL_variable::dtype, elog, ereport, errcode(), errdetail(), errhint(), errmsg(), ERROR, eval_mcontext_alloc, exec_assign_value(), exec_cast_value(), expanded_record_get_tupdesc(), expanded_record_set_fields(), lengthof, TupleDescData::natts, PLpgSQL_row::nfields, PLPGSQL_DTYPE_REC, PLPGSQL_DTYPE_ROW, plpgsql_extra_errors, plpgsql_extra_warnings, PLPGSQL_XCHECK_STRICTMULTIASSIGNMENT, TupleDescAttr(), value, values, PLpgSQL_row::varnos, and WARNING.

Referenced by exec_move_row(), and exec_move_row_from_datum().

◆ exec_prepare_plan()

static void exec_prepare_plan	(	PLpgSQL_execstate *	estate,
		PLpgSQL_expr *	expr,
		int	cursorOptions
	)

static

Definition at line 4171 of file pl_exec.c.

{
    SPIPlanPtr  plan;
    SPIPrepareOptions options;
 
    /*
     * The grammar can't conveniently set expr->func while building the parse
     * tree, so make sure it's set before parser hooks need it.
     */
    expr->func = estate->func;
 
    /*
     * Generate and save the plan
     */
    memset(&options, 0, sizeof(options));
    options.parserSetup = (ParserSetupHook) plpgsql_parser_setup;
    options.parserSetupArg = expr;
    options.parseMode = expr->parseMode;
    options.cursorOptions = cursorOptions;
    plan = SPI_prepare_extended(expr->query, &options);
    if (plan == NULL)
        elog(ERROR, "SPI_prepare_extended failed for \"%s\": %s",
             expr->query, SPI_result_code_string(SPI_result));
 
    SPI_keepplan(plan);
    expr->plan = plan;
 
    /* Check to see if it's a simple expression */
    exec_simple_check_plan(estate, expr);
}

References elog, ERROR, exec_simple_check_plan(), PLpgSQL_expr::func, PLpgSQL_execstate::func, options, PLpgSQL_expr::parseMode, PLpgSQL_expr::plan, plan, plpgsql_parser_setup(), PLpgSQL_expr::query, SPI_keepplan(), SPI_prepare_extended(), SPI_result, and SPI_result_code_string().

Referenced by exec_assign_expr(), exec_eval_expr(), exec_run_select(), exec_stmt_call(), exec_stmt_execsql(), exec_stmt_forc(), exec_stmt_open(), and exec_stmt_return_query().

◆ exec_run_select()

static int exec_run_select	(	PLpgSQL_execstate *	estate,
		PLpgSQL_expr *	expr,
		long	maxtuples,
		Portal *	portalP
	)

static

Definition at line 5767 of file pl_exec.c.

{
    ParamListInfo paramLI;
    int         rc;
 
    /*
     * On the first call for this expression generate the plan.
     *
     * If we don't need to return a portal, then we're just going to execute
     * the query immediately, which means it's OK to use a parallel plan, even
     * if the number of rows being fetched is limited.  If we do need to
     * return a portal (i.e., this is for a FOR loop), the user's code might
     * invoke additional operations inside the FOR loop, making parallel query
     * unsafe.  In any case, we don't expect any cursor operations to be done,
     * so specify NO_SCROLL for efficiency and semantic safety.
     */
    if (expr->plan == NULL)
    {
        int         cursorOptions = CURSOR_OPT_NO_SCROLL;
 
        if (portalP == NULL)
            cursorOptions |= CURSOR_OPT_PARALLEL_OK;
        exec_prepare_plan(estate, expr, cursorOptions);
    }
 
    /*
     * Set up ParamListInfo to pass to executor
     */
    paramLI = setup_param_list(estate, expr);
 
    /*
     * If a portal was requested, put the query and paramlist into the portal
     */
    if (portalP != NULL)
    {
        *portalP = SPI_cursor_open_with_paramlist(NULL, expr->plan,
                                                  paramLI,
                                                  estate->readonly_func);
        if (*portalP == NULL)
            elog(ERROR, "could not open implicit cursor for query \"%s\": %s",
                 expr->query, SPI_result_code_string(SPI_result));
        exec_eval_cleanup(estate);
        return SPI_OK_CURSOR;
    }
 
    /*
     * Execute the query
     */
    rc = SPI_execute_plan_with_paramlist(expr->plan, paramLI,
                                         estate->readonly_func, maxtuples);
    if (rc != SPI_OK_SELECT)
    {
        /*
         * SELECT INTO deserves a special error message, because "query is not
         * a SELECT" is not very helpful in that case.
         */
        if (rc == SPI_OK_SELINTO)
            ereport(ERROR,
                    (errcode(ERRCODE_SYNTAX_ERROR),
                     errmsg("query is SELECT INTO, but it should be plain SELECT"),
                     errcontext("query: %s", expr->query)));
        else
            ereport(ERROR,
                    (errcode(ERRCODE_SYNTAX_ERROR),
                     errmsg("query is not a SELECT"),
                     errcontext("query: %s", expr->query)));
    }
 
    /* Save query results for eventual cleanup */
    Assert(estate->eval_tuptable == NULL);
    estate->eval_tuptable = SPI_tuptable;
    estate->eval_processed = SPI_processed;
 
    return rc;
}

References Assert, CURSOR_OPT_NO_SCROLL, CURSOR_OPT_PARALLEL_OK, elog, ereport, errcode(), errcontext, errmsg(), ERROR, PLpgSQL_execstate::eval_processed, PLpgSQL_execstate::eval_tuptable, exec_eval_cleanup(), exec_prepare_plan(), PLpgSQL_expr::plan, PLpgSQL_expr::query, PLpgSQL_execstate::readonly_func, setup_param_list(), SPI_cursor_open_with_paramlist(), SPI_execute_plan_with_paramlist(), SPI_OK_CURSOR, SPI_OK_SELECT, SPI_OK_SELINTO, SPI_processed, SPI_result, SPI_result_code_string(), and SPI_tuptable.

Referenced by exec_eval_expr(), exec_stmt_fors(), and exec_stmt_perform().

◆ exec_save_simple_expr()

static void exec_save_simple_expr	(	PLpgSQL_expr *	expr,
		CachedPlan *	cplan
	)

static

Definition at line 8121 of file pl_exec.c.

{
    PlannedStmt *stmt;
    Plan       *plan;
    Expr       *tle_expr;
 
    /*
     * Given the checks that exec_simple_check_plan did, none of the Asserts
     * here should ever fail.
     */
 
    /* Extract the single PlannedStmt */
    Assert(list_length(cplan->stmt_list) == 1);
    stmt = linitial_node(PlannedStmt, cplan->stmt_list);
    Assert(stmt->commandType == CMD_SELECT);
 
    /*
     * Ordinarily, the plan node should be a simple Result.  However, if
     * debug_parallel_query is on, the planner might've stuck a Gather node
     * atop that.  The simplest way to deal with this is to look through the
     * Gather node.  The Gather node's tlist would normally contain a Var
     * referencing the child node's output, but it could also be a Param, or
     * it could be a Const that setrefs.c copied as-is.
     */
    plan = stmt->planTree;
    for (;;)
    {
        /* Extract the single tlist expression */
        Assert(list_length(plan->targetlist) == 1);
        tle_expr = linitial_node(TargetEntry, plan->targetlist)->expr;
 
        if (IsA(plan, Result))
        {
            Assert(plan->lefttree == NULL &&
                   plan->righttree == NULL &&
                   plan->initPlan == NULL &&
                   plan->qual == NULL &&
                   ((Result *) plan)->resconstantqual == NULL);
            break;
        }
        else if (IsA(plan, Gather))
        {
            Assert(plan->lefttree != NULL &&
                   plan->righttree == NULL &&
                   plan->initPlan == NULL &&
                   plan->qual == NULL);
            /* If setrefs.c copied up a Const, no need to look further */
            if (IsA(tle_expr, Const))
                break;
            /* Otherwise, it had better be a Param or an outer Var */
            Assert(IsA(tle_expr, Param) || (IsA(tle_expr, Var) &&
                                            ((Var *) tle_expr)->varno == OUTER_VAR));
            /* Descend to the child node */
            plan = plan->lefttree;
        }
        else
            elog(ERROR, "unexpected plan node type: %d",
                 (int) nodeTag(plan));
    }
 
    /*
     * Save the simple expression, and initialize state to "not valid in
     * current transaction".
     */
    expr->expr_simple_expr = tle_expr;
    expr->expr_simple_state = NULL;
    expr->expr_simple_in_use = false;
    expr->expr_simple_lxid = InvalidLocalTransactionId;
    /* Also stash away the expression result type */
    expr->expr_simple_type = exprType((Node *) tle_expr);
    expr->expr_simple_typmod = exprTypmod((Node *) tle_expr);
    /* We also want to remember if it is immutable or not */
    expr->expr_simple_mutable = contain_mutable_functions((Node *) tle_expr);
 
    /*
     * Lastly, check to see if there's a possibility of optimizing a
     * read/write parameter.
     */
    exec_check_rw_parameter(expr);
}

References Assert, CMD_SELECT, contain_mutable_functions(), elog, ERROR, exec_check_rw_parameter(), PLpgSQL_expr::expr_simple_expr, PLpgSQL_expr::expr_simple_in_use, PLpgSQL_expr::expr_simple_lxid, PLpgSQL_expr::expr_simple_mutable, PLpgSQL_expr::expr_simple_state, PLpgSQL_expr::expr_simple_type, PLpgSQL_expr::expr_simple_typmod, exprType(), exprTypmod(), InvalidLocalTransactionId, IsA, linitial_node, list_length(), nodeTag, OUTER_VAR, plan, stmt, and CachedPlan::stmt_list.

Referenced by exec_eval_simple_expr(), and exec_simple_check_plan().

◆ exec_set_found()

static void exec_set_found	(	PLpgSQL_execstate *	estate,
		bool	state
	)

static

Definition at line 8375 of file pl_exec.c.

{
    PLpgSQL_var *var;
 
    var = (PLpgSQL_var *) (estate->datums[estate->found_varno]);
    assign_simple_var(estate, var, BoolGetDatum(state), false, false);
}

References assign_simple_var(), BoolGetDatum(), PLpgSQL_execstate::datums, and PLpgSQL_execstate::found_varno.

Referenced by exec_for_query(), exec_stmt_execsql(), exec_stmt_fetch(), exec_stmt_foreach_a(), exec_stmt_fori(), exec_stmt_perform(), exec_stmt_return_query(), plpgsql_exec_function(), and plpgsql_exec_trigger().

◆ exec_simple_check_plan()

static void exec_simple_check_plan	(	PLpgSQL_execstate *	estate,
		PLpgSQL_expr *	expr
	)

static

Definition at line 7979 of file pl_exec.c.

{
    List       *plansources;
    CachedPlanSource *plansource;
    CachedPlan *cplan;
    MemoryContext oldcontext;
 
    /*
     * Initialize to "not simple".
     */
    expr->expr_simple_expr = NULL;
    expr->expr_rw_param = NULL;
 
    /*
     * Check the analyzed-and-rewritten form of the query to see if we will be
     * able to treat it as a simple expression.  Since this function is only
     * called immediately after creating the CachedPlanSource, we need not
     * worry about the query being stale.
     */
    if (!exec_is_simple_query(expr))
        return;
 
    /* exec_is_simple_query verified that there's just one CachedPlanSource */
    plansources = SPI_plan_get_plan_sources(expr->plan);
    plansource = (CachedPlanSource *) linitial(plansources);
 
    /*
     * Get the generic plan for the query.  If replanning is needed, do that
     * work in the eval_mcontext.  (Note that replanning could throw an error,
     * in which case the expr is left marked "not simple", which is fine.)
     */
    oldcontext = MemoryContextSwitchTo(get_eval_mcontext(estate));
    cplan = SPI_plan_get_cached_plan(expr->plan);
    MemoryContextSwitchTo(oldcontext);
 
    /* Can't fail, because we checked for a single CachedPlanSource above */
    Assert(cplan != NULL);
 
    /*
     * Verify that plancache.c thinks the plan is simple enough to use
     * CachedPlanIsSimplyValid.  Given the restrictions above, it's unlikely
     * that this could fail, but if it does, just treat plan as not simple. On
     * success, save a refcount on the plan in the simple-expression resowner.
     */
    if (CachedPlanAllowsSimpleValidityCheck(plansource, cplan,
                                            estate->simple_eval_resowner))
    {
        /* Remember that we have the refcount */
        expr->expr_simple_plansource = plansource;
        expr->expr_simple_plan = cplan;
        expr->expr_simple_plan_lxid = MyProc->vxid.lxid;
 
        /* Share the remaining work with the replan code path */
        exec_save_simple_expr(expr, cplan);
    }
 
    /*
     * Release the plan refcount obtained by SPI_plan_get_cached_plan.  (This
     * refcount is held by the wrong resowner, so we can't just repurpose it.)
     */
    ReleaseCachedPlan(cplan, CurrentResourceOwner);
}

References Assert, CachedPlanAllowsSimpleValidityCheck(), CurrentResourceOwner, exec_is_simple_query(), exec_save_simple_expr(), PLpgSQL_expr::expr_rw_param, PLpgSQL_expr::expr_simple_expr, PLpgSQL_expr::expr_simple_plan, PLpgSQL_expr::expr_simple_plan_lxid, PLpgSQL_expr::expr_simple_plansource, get_eval_mcontext, linitial, PGPROC::lxid, MemoryContextSwitchTo(), MyProc, PLpgSQL_expr::plan, ReleaseCachedPlan(), PLpgSQL_execstate::simple_eval_resowner, SPI_plan_get_cached_plan(), SPI_plan_get_plan_sources(), and PGPROC::vxid.

Referenced by exec_prepare_plan().

◆ exec_stmt_assert()

static int exec_stmt_assert	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_assert *	stmt
	)

static

Definition at line 3934 of file pl_exec.c.

{
    bool        value;
    bool        isnull;
 
    /* do nothing when asserts are not enabled */
    if (!plpgsql_check_asserts)
        return PLPGSQL_RC_OK;
 
    value = exec_eval_boolean(estate, stmt->cond, &isnull);
    exec_eval_cleanup(estate);
 
    if (isnull || !value)
    {
        char       *message = NULL;
 
        if (stmt->message != NULL)
        {
            Datum       val;
            Oid         typeid;
            int32       typmod;
 
            val = exec_eval_expr(estate, stmt->message,
                                 &isnull, &typeid, &typmod);
            if (!isnull)
                message = convert_value_to_string(estate, val, typeid);
            /* we mustn't do exec_eval_cleanup here */
        }
 
        ereport(ERROR,
                (errcode(ERRCODE_ASSERT_FAILURE),
                 message ? errmsg_internal("%s", message) :
                 errmsg("assertion failed")));
    }
 
    return PLPGSQL_RC_OK;
}

References convert_value_to_string(), ereport, errcode(), errmsg(), errmsg_internal(), ERROR, exec_eval_boolean(), exec_eval_cleanup(), exec_eval_expr(), plpgsql_check_asserts, PLPGSQL_RC_OK, stmt, val, and value.

Referenced by exec_stmts().

◆ exec_stmt_assign()

static int exec_stmt_assign	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_assign *	stmt
	)

static

Definition at line 2148 of file pl_exec.c.

{
    Assert(stmt->varno >= 0);
 
    exec_assign_expr(estate, estate->datums[stmt->varno], stmt->expr);
 
    return PLPGSQL_RC_OK;
}

References Assert, PLpgSQL_execstate::datums, exec_assign_expr(), PLPGSQL_RC_OK, and stmt.

Referenced by exec_stmts().

◆ exec_stmt_block()

static int exec_stmt_block	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_block *	block
	)

static

Definition at line 1647 of file pl_exec.c.

{
    volatile int rc = -1;
    int         i;
 
    /*
     * First initialize all variables declared in this block
     */
    estate->err_text = gettext_noop("during statement block local variable initialization");
 
    for (i = 0; i < block->n_initvars; i++)
    {
        int         n = block->initvarnos[i];
        PLpgSQL_datum *datum = estate->datums[n];
 
        /*
         * The set of dtypes handled here must match plpgsql_add_initdatums().
         *
         * Note that we currently don't support promise datums within blocks,
         * only at a function's outermost scope, so we needn't handle those
         * here.
         *
         * Since RECFIELD isn't a supported case either, it's okay to cast the
         * PLpgSQL_datum to PLpgSQL_variable.
         */
        estate->err_var = (PLpgSQL_variable *) datum;
 
        switch (datum->dtype)
        {
            case PLPGSQL_DTYPE_VAR:
                {
                    PLpgSQL_var *var = (PLpgSQL_var *) datum;
 
                    /*
                     * Free any old value, in case re-entering block, and
                     * initialize to NULL
                     */
                    assign_simple_var(estate, var, (Datum) 0, true, false);
 
                    if (var->default_val == NULL)
                    {
                        /*
                         * If needed, give the datatype a chance to reject
                         * NULLs, by assigning a NULL to the variable.  We
                         * claim the value is of type UNKNOWN, not the var's
                         * datatype, else coercion will be skipped.
                         */
                        if (var->datatype->typtype == TYPTYPE_DOMAIN)
                            exec_assign_value(estate,
                                              (PLpgSQL_datum *) var,
                                              (Datum) 0,
                                              true,
                                              UNKNOWNOID,
                                              -1);
 
                        /* parser should have rejected NOT NULL */
                        Assert(!var->notnull);
                    }
                    else
                    {
                        exec_assign_expr(estate, (PLpgSQL_datum *) var,
                                         var->default_val);
                    }
                }
                break;
 
            case PLPGSQL_DTYPE_REC:
                {
                    PLpgSQL_rec *rec = (PLpgSQL_rec *) datum;
 
                    /*
                     * Deletion of any existing object will be handled during
                     * the assignments below, and in some cases it's more
                     * efficient for us not to get rid of it beforehand.
                     */
                    if (rec->default_val == NULL)
                    {
                        /*
                         * If needed, give the datatype a chance to reject
                         * NULLs, by assigning a NULL to the variable.
                         */
                        exec_move_row(estate, (PLpgSQL_variable *) rec,
                                      NULL, NULL);
 
                        /* parser should have rejected NOT NULL */
                        Assert(!rec->notnull);
                    }
                    else
                    {
                        exec_assign_expr(estate, (PLpgSQL_datum *) rec,
                                         rec->default_val);
                    }
                }
                break;
 
            default:
                elog(ERROR, "unrecognized dtype: %d", datum->dtype);
        }
    }
 
    estate->err_var = NULL;
 
    if (block->exceptions)
    {
        /*
         * Execute the statements in the block's body inside a sub-transaction
         */
        MemoryContext oldcontext = CurrentMemoryContext;
        ResourceOwner oldowner = CurrentResourceOwner;
        ExprContext *old_eval_econtext = estate->eval_econtext;
        ErrorData  *save_cur_error = estate->cur_error;
        MemoryContext stmt_mcontext;
 
        estate->err_text = gettext_noop("during statement block entry");
 
        /*
         * We will need a stmt_mcontext to hold the error data if an error
         * occurs.  It seems best to force it to exist before entering the
         * subtransaction, so that we reduce the risk of out-of-memory during
         * error recovery, and because this greatly simplifies restoring the
         * stmt_mcontext stack to the correct state after an error.  We can
         * ameliorate the cost of this by allowing the called statements to
         * use this mcontext too; so we don't push it down here.
         */
        stmt_mcontext = get_stmt_mcontext(estate);
 
        BeginInternalSubTransaction(NULL);
        /* Want to run statements inside function's memory context */
        MemoryContextSwitchTo(oldcontext);
 
        PG_TRY();
        {
            /*
             * We need to run the block's statements with a new eval_econtext
             * that belongs to the current subtransaction; if we try to use
             * the outer econtext then ExprContext shutdown callbacks will be
             * called at the wrong times.
             */
            plpgsql_create_econtext(estate);
 
            estate->err_text = NULL;
 
            /* Run the block's statements */
            rc = exec_stmts(estate, block->body);
 
            estate->err_text = gettext_noop("during statement block exit");
 
            /*
             * If the block ended with RETURN, we may need to copy the return
             * value out of the subtransaction eval_context.  We can avoid a
             * physical copy if the value happens to be a R/W expanded object.
             */
            if (rc == PLPGSQL_RC_RETURN &&
                !estate->retisset &&
                !estate->retisnull)
            {
                int16       resTypLen;
                bool        resTypByVal;
 
                get_typlenbyval(estate->rettype, &resTypLen, &resTypByVal);
                estate->retval = datumTransfer(estate->retval,
                                               resTypByVal, resTypLen);
            }
 
            /* Commit the inner transaction, return to outer xact context */
            ReleaseCurrentSubTransaction();
            MemoryContextSwitchTo(oldcontext);
            CurrentResourceOwner = oldowner;
 
            /* Assert that the stmt_mcontext stack is unchanged */
            Assert(stmt_mcontext == estate->stmt_mcontext);
 
            /*
             * Revert to outer eval_econtext.  (The inner one was
             * automatically cleaned up during subxact exit.)
             */
            estate->eval_econtext = old_eval_econtext;
        }
        PG_CATCH();
        {
            ErrorData  *edata;
            ListCell   *e;
 
            estate->err_text = gettext_noop("during exception cleanup");
 
            /* Save error info in our stmt_mcontext */
            MemoryContextSwitchTo(stmt_mcontext);
            edata = CopyErrorData();
            FlushErrorState();
 
            /* Abort the inner transaction */
            RollbackAndReleaseCurrentSubTransaction();
            MemoryContextSwitchTo(oldcontext);
            CurrentResourceOwner = oldowner;
 
            /*
             * Set up the stmt_mcontext stack as though we had restored our
             * previous state and then done push_stmt_mcontext().  The push is
             * needed so that statements in the exception handler won't
             * clobber the error data that's in our stmt_mcontext.
             */
            estate->stmt_mcontext_parent = stmt_mcontext;
            estate->stmt_mcontext = NULL;
 
            /*
             * Now we can delete any nested stmt_mcontexts that might have
             * been created as children of ours.  (Note: we do not immediately
             * release any statement-lifespan data that might have been left
             * behind in stmt_mcontext itself.  We could attempt that by doing
             * a MemoryContextReset on it before collecting the error data
             * above, but it seems too risky to do any significant amount of
             * work before collecting the error.)
             */
            MemoryContextDeleteChildren(stmt_mcontext);
 
            /* Revert to outer eval_econtext */
            estate->eval_econtext = old_eval_econtext;
 
            /*
             * Must clean up the econtext too.  However, any tuple table made
             * in the subxact will have been thrown away by SPI during subxact
             * abort, so we don't need to (and mustn't try to) free the
             * eval_tuptable.
             */
            estate->eval_tuptable = NULL;
            exec_eval_cleanup(estate);
 
            /* Look for a matching exception handler */
            foreach(e, block->exceptions->exc_list)
            {
                PLpgSQL_exception *exception = (PLpgSQL_exception *) lfirst(e);
 
                if (exception_matches_conditions(edata, exception->conditions))
                {
                    /*
                     * Initialize the magic SQLSTATE and SQLERRM variables for
                     * the exception block; this also frees values from any
                     * prior use of the same exception. We needn't do this
                     * until we have found a matching exception.
                     */
                    PLpgSQL_var *state_var;
                    PLpgSQL_var *errm_var;
 
                    state_var = (PLpgSQL_var *)
                        estate->datums[block->exceptions->sqlstate_varno];
                    errm_var = (PLpgSQL_var *)
                        estate->datums[block->exceptions->sqlerrm_varno];
 
                    assign_text_var(estate, state_var,
                                    unpack_sql_state(edata->sqlerrcode));
                    assign_text_var(estate, errm_var, edata->message);
 
                    /*
                     * Also set up cur_error so the error data is accessible
                     * inside the handler.
                     */
                    estate->cur_error = edata;
 
                    estate->err_text = NULL;
 
                    rc = exec_stmts(estate, exception->action);
 
                    break;
                }
            }
 
            /*
             * Restore previous state of cur_error, whether or not we executed
             * a handler.  This is needed in case an error got thrown from
             * some inner block's exception handler.
             */
            estate->cur_error = save_cur_error;
 
            /* If no match found, re-throw the error */
            if (e == NULL)
                ReThrowError(edata);
 
            /* Restore stmt_mcontext stack and release the error data */
            pop_stmt_mcontext(estate);
            MemoryContextReset(stmt_mcontext);
        }
        PG_END_TRY();
 
        Assert(save_cur_error == estate->cur_error);
    }
    else
    {
        /*
         * Just execute the statements in the block's body
         */
        estate->err_text = NULL;
 
        rc = exec_stmts(estate, block->body);
    }
 
    estate->err_text = NULL;
 
    /*
     * Handle the return code.  This is intentionally different from
     * LOOP_RC_PROCESSING(): CONTINUE never matches a block, and EXIT matches
     * a block only if there is a label match.
     */
    switch (rc)
    {
        case PLPGSQL_RC_OK:
        case PLPGSQL_RC_RETURN:
        case PLPGSQL_RC_CONTINUE:
            return rc;
 
        case PLPGSQL_RC_EXIT:
            if (estate->exitlabel == NULL)
                return PLPGSQL_RC_EXIT;
            if (block->label == NULL)
                return PLPGSQL_RC_EXIT;
            if (strcmp(block->label, estate->exitlabel) != 0)
                return PLPGSQL_RC_EXIT;
            estate->exitlabel = NULL;
            return PLPGSQL_RC_OK;
 
        default:
            elog(ERROR, "unrecognized rc: %d", rc);
    }
 
    return PLPGSQL_RC_OK;
}

Referenced by exec_stmts(), and exec_toplevel_block().

◆ exec_stmt_call()

static int exec_stmt_call	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_call *	stmt
	)

static

Definition at line 2181 of file pl_exec.c.

{
    PLpgSQL_expr *expr = stmt->expr;
    LocalTransactionId before_lxid;
    LocalTransactionId after_lxid;
    ParamListInfo paramLI;
    SPIExecuteOptions options;
    int         rc;
 
    /*
     * Make a plan if we don't have one already.
     */
    if (expr->plan == NULL)
        exec_prepare_plan(estate, expr, 0);
 
    /*
     * A CALL or DO can never be a simple expression.
     */
    Assert(!expr->expr_simple_expr);
 
    /*
     * Also construct a DTYPE_ROW datum representing the plpgsql variables
     * associated with the procedure's output arguments.  Then we can use
     * exec_move_row() to do the assignments.
     */
    if (stmt->is_call && stmt->target == NULL)
        stmt->target = make_callstmt_target(estate, expr);
 
    paramLI = setup_param_list(estate, expr);
 
    before_lxid = MyProc->vxid.lxid;
 
    /*
     * If we have a procedure-lifespan resowner, use that to hold the refcount
     * for the plan.  This avoids refcount leakage complaints if the called
     * procedure ends the current transaction.
     *
     * Also, tell SPI to allow non-atomic execution.
     */
    memset(&options, 0, sizeof(options));
    options.params = paramLI;
    options.read_only = estate->readonly_func;
    options.allow_nonatomic = true;
    options.owner = estate->procedure_resowner;
 
    rc = SPI_execute_plan_extended(expr->plan, &options);
 
    if (rc < 0)
        elog(ERROR, "SPI_execute_plan_extended failed executing query \"%s\": %s",
             expr->query, SPI_result_code_string(rc));
 
    after_lxid = MyProc->vxid.lxid;
 
    if (before_lxid != after_lxid)
    {
        /*
         * If we are in a new transaction after the call, we need to build new
         * simple-expression infrastructure.
         */
        estate->simple_eval_estate = NULL;
        estate->simple_eval_resowner = NULL;
        plpgsql_create_econtext(estate);
    }
 
    /*
     * Check result rowcount; if there's one row, assign procedure's output
     * values back to the appropriate variables.
     */
    if (SPI_processed == 1)
    {
        SPITupleTable *tuptab = SPI_tuptable;
 
        if (!stmt->is_call)
            elog(ERROR, "DO statement returned a row");
 
        exec_move_row(estate, stmt->target, tuptab->vals[0], tuptab->tupdesc);
    }
    else if (SPI_processed > 1)
        elog(ERROR, "procedure call returned more than one row");
 
    exec_eval_cleanup(estate);
    SPI_freetuptable(SPI_tuptable);
 
    return PLPGSQL_RC_OK;
}

References Assert, elog, ERROR, exec_eval_cleanup(), exec_move_row(), exec_prepare_plan(), PLpgSQL_expr::expr_simple_expr, PGPROC::lxid, make_callstmt_target(), MyProc, options, PLpgSQL_expr::plan, plpgsql_create_econtext(), PLPGSQL_RC_OK, PLpgSQL_execstate::procedure_resowner, PLpgSQL_expr::query, PLpgSQL_execstate::readonly_func, setup_param_list(), PLpgSQL_execstate::simple_eval_estate, PLpgSQL_execstate::simple_eval_resowner, SPI_execute_plan_extended(), SPI_freetuptable(), SPI_processed, SPI_result_code_string(), SPI_tuptable, stmt, SPITupleTable::tupdesc, SPITupleTable::vals, and PGPROC::vxid.

Referenced by exec_stmts().

◆ exec_stmt_case()

static int exec_stmt_case	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_case *	stmt
	)

static

Definition at line 2540 of file pl_exec.c.

{
    PLpgSQL_var *t_var = NULL;
    bool        isnull;
    ListCell   *l;
 
    if (stmt->t_expr != NULL)
    {
        /* simple case */
        Datum       t_val;
        Oid         t_typoid;
        int32       t_typmod;
 
        t_val = exec_eval_expr(estate, stmt->t_expr,
                               &isnull, &t_typoid, &t_typmod);
 
        t_var = (PLpgSQL_var *) estate->datums[stmt->t_varno];
 
        /*
         * When expected datatype is different from real, change it. Note that
         * what we're modifying here is an execution copy of the datum, so
         * this doesn't affect the originally stored function parse tree. (In
         * theory, if the expression datatype keeps changing during execution,
         * this could cause a function-lifespan memory leak.  Doesn't seem
         * worth worrying about though.)
         */
        if (t_var->datatype->typoid != t_typoid ||
            t_var->datatype->atttypmod != t_typmod)
            t_var->datatype = plpgsql_build_datatype(t_typoid,
                                                     t_typmod,
                                                     estate->func->fn_input_collation,
                                                     NULL);
 
        /* now we can assign to the variable */
        exec_assign_value(estate,
                          (PLpgSQL_datum *) t_var,
                          t_val,
                          isnull,
                          t_typoid,
                          t_typmod);
 
        exec_eval_cleanup(estate);
    }
 
    /* Now search for a successful WHEN clause */
    foreach(l, stmt->case_when_list)
    {
        PLpgSQL_case_when *cwt = (PLpgSQL_case_when *) lfirst(l);
        bool        value;
 
        value = exec_eval_boolean(estate, cwt->expr, &isnull);
        exec_eval_cleanup(estate);
        if (!isnull && value)
        {
            /* Found it */
 
            /* We can now discard any value we had for the temp variable */
            if (t_var != NULL)
                assign_simple_var(estate, t_var, (Datum) 0, true, false);
 
            /* Evaluate the statement(s), and we're done */
            return exec_stmts(estate, cwt->stmts);
        }
    }
 
    /* We can now discard any value we had for the temp variable */
    if (t_var != NULL)
        assign_simple_var(estate, t_var, (Datum) 0, true, false);
 
    /* SQL2003 mandates this error if there was no ELSE clause */
    if (!stmt->have_else)
        ereport(ERROR,
                (errcode(ERRCODE_CASE_NOT_FOUND),
                 errmsg("case not found"),
                 errhint("CASE statement is missing ELSE part.")));
 
    /* Evaluate the ELSE statements, and we're done */
    return exec_stmts(estate, stmt->else_stmts);
}

References assign_simple_var(), PLpgSQL_type::atttypmod, PLpgSQL_var::datatype, PLpgSQL_execstate::datums, ereport, errcode(), errhint(), errmsg(), ERROR, exec_assign_value(), exec_eval_boolean(), exec_eval_cleanup(), exec_eval_expr(), exec_stmts(), PLpgSQL_case_when::expr, PLpgSQL_function::fn_input_collation, PLpgSQL_execstate::func, if(), lfirst, plpgsql_build_datatype(), stmt, PLpgSQL_case_when::stmts, PLpgSQL_type::typoid, and value.

Referenced by exec_stmts().

◆ exec_stmt_close()

static int exec_stmt_close	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_close *	stmt
	)

static

Definition at line 4917 of file pl_exec.c.

{
    PLpgSQL_var *curvar;
    Portal      portal;
    char       *curname;
    MemoryContext oldcontext;
 
    /* ----------
     * Get the portal of the cursor by name
     * ----------
     */
    curvar = (PLpgSQL_var *) (estate->datums[stmt->curvar]);
    if (curvar->isnull)
        ereport(ERROR,
                (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                 errmsg("cursor variable \"%s\" is null", curvar->refname)));
 
    /* Use eval_mcontext for short-lived string */
    oldcontext = MemoryContextSwitchTo(get_eval_mcontext(estate));
    curname = TextDatumGetCString(curvar->value);
    MemoryContextSwitchTo(oldcontext);
 
    portal = SPI_cursor_find(curname);
    if (portal == NULL)
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_CURSOR),
                 errmsg("cursor \"%s\" does not exist", curname)));
 
    /* ----------
     * And close it.
     * ----------
     */
    SPI_cursor_close(portal);
 
    return PLPGSQL_RC_OK;
}

References PLpgSQL_execstate::datums, ereport, errcode(), errmsg(), ERROR, get_eval_mcontext, PLpgSQL_var::isnull, MemoryContextSwitchTo(), PLPGSQL_RC_OK, PLpgSQL_var::refname, SPI_cursor_close(), SPI_cursor_find(), stmt, TextDatumGetCString, and PLpgSQL_var::value.

Referenced by exec_stmts().

◆ exec_stmt_commit()

static int exec_stmt_commit	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_commit *	stmt
	)

static

Definition at line 4960 of file pl_exec.c.

{
    if (stmt->chain)
        SPI_commit_and_chain();
    else
        SPI_commit();
 
    /*
     * We need to build new simple-expression infrastructure, since the old
     * data structures are gone.
     */
    estate->simple_eval_estate = NULL;
    estate->simple_eval_resowner = NULL;
    plpgsql_create_econtext(estate);
 
    return PLPGSQL_RC_OK;
}

References plpgsql_create_econtext(), PLPGSQL_RC_OK, PLpgSQL_execstate::simple_eval_estate, PLpgSQL_execstate::simple_eval_resowner, SPI_commit(), SPI_commit_and_chain(), and stmt.

Referenced by exec_stmts().

◆ exec_stmt_dynexecute()

static int exec_stmt_dynexecute	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_dynexecute *	stmt
	)

static

Definition at line 4444 of file pl_exec.c.

{
    Datum       query;
    bool        isnull;
    Oid         restype;
    int32       restypmod;
    char       *querystr;
    int         exec_res;
    ParamListInfo paramLI;
    SPIExecuteOptions options;
    MemoryContext stmt_mcontext = get_stmt_mcontext(estate);
 
    /*
     * First we evaluate the string expression after the EXECUTE keyword. Its
     * result is the querystring we have to execute.
     */
    query = exec_eval_expr(estate, stmt->query, &isnull, &restype, &restypmod);
    if (isnull)
        ereport(ERROR,
                (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                 errmsg("query string argument of EXECUTE is null")));
 
    /* Get the C-String representation */
    querystr = convert_value_to_string(estate, query, restype);
 
    /* copy it into the stmt_mcontext before we clean up */
    querystr = MemoryContextStrdup(stmt_mcontext, querystr);
 
    exec_eval_cleanup(estate);
 
    /*
     * Execute the query without preparing a saved plan.
     */
    paramLI = exec_eval_using_params(estate, stmt->params);
 
    memset(&options, 0, sizeof(options));
    options.params = paramLI;
    options.read_only = estate->readonly_func;
 
    exec_res = SPI_execute_extended(querystr, &options);
 
    switch (exec_res)
    {
        case SPI_OK_SELECT:
        case SPI_OK_INSERT:
        case SPI_OK_UPDATE:
        case SPI_OK_DELETE:
        case SPI_OK_MERGE:
        case SPI_OK_INSERT_RETURNING:
        case SPI_OK_UPDATE_RETURNING:
        case SPI_OK_DELETE_RETURNING:
        case SPI_OK_MERGE_RETURNING:
        case SPI_OK_UTILITY:
        case SPI_OK_REWRITTEN:
            break;
 
        case 0:
 
            /*
             * Also allow a zero return, which implies the querystring
             * contained no commands.
             */
            break;
 
        case SPI_OK_SELINTO:
 
            /*
             * We want to disallow SELECT INTO for now, because its behavior
             * is not consistent with SELECT INTO in a normal plpgsql context.
             * (We need to reimplement EXECUTE to parse the string as a
             * plpgsql command, not just feed it to SPI_execute.)  This is not
             * a functional limitation because CREATE TABLE AS is allowed.
             */
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("EXECUTE of SELECT ... INTO is not implemented"),
                     errhint("You might want to use EXECUTE ... INTO or EXECUTE CREATE TABLE ... AS instead.")));
            break;
 
            /* Some SPI errors deserve specific error messages */
        case SPI_ERROR_COPY:
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("cannot COPY to/from client in PL/pgSQL")));
            break;
 
        case SPI_ERROR_TRANSACTION:
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("EXECUTE of transaction commands is not implemented")));
            break;
 
        default:
            elog(ERROR, "SPI_execute_extended failed executing query \"%s\": %s",
                 querystr, SPI_result_code_string(exec_res));
            break;
    }
 
    /* Save result info for GET DIAGNOSTICS */
    estate->eval_processed = SPI_processed;
 
    /* Process INTO if present */
    if (stmt->into)
    {
        SPITupleTable *tuptab = SPI_tuptable;
        uint64      n = SPI_processed;
        PLpgSQL_variable *target;
 
        /* If the statement did not return a tuple table, complain */
        if (tuptab == NULL)
            ereport(ERROR,
                    (errcode(ERRCODE_SYNTAX_ERROR),
                     errmsg("INTO used with a command that cannot return data")));
 
        /* Fetch target's datum entry */
        target = (PLpgSQL_variable *) estate->datums[stmt->target->dno];
 
        /*
         * If SELECT ... INTO specified STRICT, and the query didn't find
         * exactly one row, throw an error.  If STRICT was not specified, then
         * allow the query to find any number of rows.
         */
        if (n == 0)
        {
            if (stmt->strict)
            {
                char       *errdetail;
 
                if (estate->func->print_strict_params)
                    errdetail = format_preparedparamsdata(estate, paramLI);
                else
                    errdetail = NULL;
 
                ereport(ERROR,
                        (errcode(ERRCODE_NO_DATA_FOUND),
                         errmsg("query returned no rows"),
                         errdetail ? errdetail_internal("parameters: %s", errdetail) : 0));
            }
            /* set the target to NULL(s) */
            exec_move_row(estate, target, NULL, tuptab->tupdesc);
        }
        else
        {
            if (n > 1 && stmt->strict)
            {
                char       *errdetail;
 
                if (estate->func->print_strict_params)
                    errdetail = format_preparedparamsdata(estate, paramLI);
                else
                    errdetail = NULL;
 
                ereport(ERROR,
                        (errcode(ERRCODE_TOO_MANY_ROWS),
                         errmsg("query returned more than one row"),
                         errdetail ? errdetail_internal("parameters: %s", errdetail) : 0));
            }
 
            /* Put the first result row into the target */
            exec_move_row(estate, target, tuptab->vals[0], tuptab->tupdesc);
        }
        /* clean up after exec_move_row() */
        exec_eval_cleanup(estate);
    }
    else
    {
        /*
         * It might be a good idea to raise an error if the query returned
         * tuples that are being ignored, but historically we have not done
         * that.
         */
    }
 
    /* Release any result from SPI_execute, as well as transient data */
    SPI_freetuptable(SPI_tuptable);
    MemoryContextReset(stmt_mcontext);
 
    return PLPGSQL_RC_OK;
}

References convert_value_to_string(), PLpgSQL_execstate::datums, elog, ereport, errcode(), errdetail(), errdetail_internal(), errhint(), errmsg(), ERROR, PLpgSQL_execstate::eval_processed, exec_eval_cleanup(), exec_eval_expr(), exec_eval_using_params(), exec_move_row(), format_preparedparamsdata(), PLpgSQL_execstate::func, get_stmt_mcontext(), if(), MemoryContextReset(), MemoryContextStrdup(), options, PLPGSQL_RC_OK, PLpgSQL_function::print_strict_params, PLpgSQL_execstate::readonly_func, SPI_ERROR_COPY, SPI_ERROR_TRANSACTION, SPI_execute_extended(), SPI_freetuptable(), SPI_OK_DELETE, SPI_OK_DELETE_RETURNING, SPI_OK_INSERT, SPI_OK_INSERT_RETURNING, SPI_OK_MERGE, SPI_OK_MERGE_RETURNING, SPI_OK_REWRITTEN, SPI_OK_SELECT, SPI_OK_SELINTO, SPI_OK_UPDATE, SPI_OK_UPDATE_RETURNING, SPI_OK_UTILITY, SPI_processed, SPI_result_code_string(), SPI_tuptable, stmt, SPITupleTable::tupdesc, and SPITupleTable::vals.

Referenced by exec_stmts().

◆ exec_stmt_dynfors()

static int exec_stmt_dynfors	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_dynfors *	stmt
	)

static

Definition at line 4634 of file pl_exec.c.

{
    Portal      portal;
    int         rc;
 
    portal = exec_dynquery_with_params(estate, stmt->query, stmt->params,
                                       NULL, CURSOR_OPT_NO_SCROLL);
 
    /*
     * Execute the loop
     */
    rc = exec_for_query(estate, (PLpgSQL_stmt_forq *) stmt, portal, true);
 
    /*
     * Close the implicit cursor
     */
    SPI_cursor_close(portal);
 
    return rc;
}

References CURSOR_OPT_NO_SCROLL, exec_dynquery_with_params(), exec_for_query(), SPI_cursor_close(), and stmt.

Referenced by exec_stmts().

◆ exec_stmt_execsql()

static int exec_stmt_execsql	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_execsql *	stmt
	)

static

Definition at line 4212 of file pl_exec.c.

{
    ParamListInfo paramLI;
    long        tcount;
    int         rc;
    PLpgSQL_expr *expr = stmt->sqlstmt;
    int         too_many_rows_level = 0;
 
    if (plpgsql_extra_errors & PLPGSQL_XCHECK_TOOMANYROWS)
        too_many_rows_level = ERROR;
    else if (plpgsql_extra_warnings & PLPGSQL_XCHECK_TOOMANYROWS)
        too_many_rows_level = WARNING;
 
    /*
     * On the first call for this statement generate the plan, and detect
     * whether the statement is INSERT/UPDATE/DELETE/MERGE
     */
    if (expr->plan == NULL)
        exec_prepare_plan(estate, expr, CURSOR_OPT_PARALLEL_OK);
 
    if (!stmt->mod_stmt_set)
    {
        ListCell   *l;
 
        stmt->mod_stmt = false;
        foreach(l, SPI_plan_get_plan_sources(expr->plan))
        {
            CachedPlanSource *plansource = (CachedPlanSource *) lfirst(l);
 
            /*
             * We could look at the raw_parse_tree, but it seems simpler to
             * check the command tag.  Note we should *not* look at the Query
             * tree(s), since those are the result of rewriting and could be
             * stale, or could have been transmogrified into something else
             * entirely.
             */
            if (plansource->commandTag == CMDTAG_INSERT ||
                plansource->commandTag == CMDTAG_UPDATE ||
                plansource->commandTag == CMDTAG_DELETE ||
                plansource->commandTag == CMDTAG_MERGE)
            {
                stmt->mod_stmt = true;
                break;
            }
        }
        stmt->mod_stmt_set = true;
    }
 
    /*
     * Set up ParamListInfo to pass to executor
     */
    paramLI = setup_param_list(estate, expr);
 
    /*
     * If we have INTO, then we only need one row back ... but if we have INTO
     * STRICT or extra check too_many_rows, ask for two rows, so that we can
     * verify the statement returns only one.  INSERT/UPDATE/DELETE/MERGE are
     * always treated strictly. Without INTO, just run the statement to
     * completion (tcount = 0).
     *
     * We could just ask for two rows always when using INTO, but there are
     * some cases where demanding the extra row costs significant time, eg by
     * forcing completion of a sequential scan.  So don't do it unless we need
     * to enforce strictness.
     */
    if (stmt->into)
    {
        if (stmt->strict || stmt->mod_stmt || too_many_rows_level)
            tcount = 2;
        else
            tcount = 1;
    }
    else
        tcount = 0;
 
    /*
     * Execute the plan
     */
    rc = SPI_execute_plan_with_paramlist(expr->plan, paramLI,
                                         estate->readonly_func, tcount);
 
    /*
     * Check for error, and set FOUND if appropriate (for historical reasons
     * we set FOUND only for certain query types).  Also Assert that we
     * identified the statement type the same as SPI did.
     */
    switch (rc)
    {
        case SPI_OK_SELECT:
            Assert(!stmt->mod_stmt);
            exec_set_found(estate, (SPI_processed != 0));
            break;
 
        case SPI_OK_INSERT:
        case SPI_OK_UPDATE:
        case SPI_OK_DELETE:
        case SPI_OK_MERGE:
        case SPI_OK_INSERT_RETURNING:
        case SPI_OK_UPDATE_RETURNING:
        case SPI_OK_DELETE_RETURNING:
        case SPI_OK_MERGE_RETURNING:
            Assert(stmt->mod_stmt);
            exec_set_found(estate, (SPI_processed != 0));
            break;
 
        case SPI_OK_SELINTO:
        case SPI_OK_UTILITY:
            Assert(!stmt->mod_stmt);
            break;
 
        case SPI_OK_REWRITTEN:
 
            /*
             * The command was rewritten into another kind of command. It's
             * not clear what FOUND would mean in that case (and SPI doesn't
             * return the row count either), so just set it to false.  Note
             * that we can't assert anything about mod_stmt here.
             */
            exec_set_found(estate, false);
            break;
 
            /* Some SPI errors deserve specific error messages */
        case SPI_ERROR_COPY:
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("cannot COPY to/from client in PL/pgSQL")));
            break;
 
        case SPI_ERROR_TRANSACTION:
            ereport(ERROR,
                    (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                     errmsg("unsupported transaction command in PL/pgSQL")));
            break;
 
        default:
            elog(ERROR, "SPI_execute_plan_with_paramlist failed executing query \"%s\": %s",
                 expr->query, SPI_result_code_string(rc));
            break;
    }
 
    /* All variants should save result info for GET DIAGNOSTICS */
    estate->eval_processed = SPI_processed;
 
    /* Process INTO if present */
    if (stmt->into)
    {
        SPITupleTable *tuptab = SPI_tuptable;
        uint64      n = SPI_processed;
        PLpgSQL_variable *target;
 
        /* If the statement did not return a tuple table, complain */
        if (tuptab == NULL)
            ereport(ERROR,
                    (errcode(ERRCODE_SYNTAX_ERROR),
                     errmsg("INTO used with a command that cannot return data")));
 
        /* Fetch target's datum entry */
        target = (PLpgSQL_variable *) estate->datums[stmt->target->dno];
 
        /*
         * If SELECT ... INTO specified STRICT, and the query didn't find
         * exactly one row, throw an error.  If STRICT was not specified, then
         * allow the query to find any number of rows.
         */
        if (n == 0)
        {
            if (stmt->strict)
            {
                char       *errdetail;
 
                if (estate->func->print_strict_params)
                    errdetail = format_expr_params(estate, expr);
                else
                    errdetail = NULL;
 
                ereport(ERROR,
                        (errcode(ERRCODE_NO_DATA_FOUND),
                         errmsg("query returned no rows"),
                         errdetail ? errdetail_internal("parameters: %s", errdetail) : 0));
            }
            /* set the target to NULL(s) */
            exec_move_row(estate, target, NULL, tuptab->tupdesc);
        }
        else
        {
            if (n > 1 && (stmt->strict || stmt->mod_stmt || too_many_rows_level))
            {
                char       *errdetail;
                int         errlevel;
 
                if (estate->func->print_strict_params)
                    errdetail = format_expr_params(estate, expr);
                else
                    errdetail = NULL;
 
                errlevel = (stmt->strict || stmt->mod_stmt) ? ERROR : too_many_rows_level;
 
                ereport(errlevel,
                        (errcode(ERRCODE_TOO_MANY_ROWS),
                         errmsg("query returned more than one row"),
                         errdetail ? errdetail_internal("parameters: %s", errdetail) : 0,
                         errhint("Make sure the query returns a single row, or use LIMIT 1.")));
            }
            /* Put the first result row into the target */
            exec_move_row(estate, target, tuptab->vals[0], tuptab->tupdesc);
        }
 
        /* Clean up */
        exec_eval_cleanup(estate);
        SPI_freetuptable(SPI_tuptable);
    }
    else
    {
        /* If the statement returned a tuple table, complain */
        if (SPI_tuptable != NULL)
            ereport(ERROR,
                    (errcode(ERRCODE_SYNTAX_ERROR),
                     errmsg("query has no destination for result data"),
                     (rc == SPI_OK_SELECT) ? errhint("If you want to discard the results of a SELECT, use PERFORM instead.") : 0));
    }
 
    return PLPGSQL_RC_OK;
}

References Assert, CachedPlanSource::commandTag, CURSOR_OPT_PARALLEL_OK, PLpgSQL_execstate::datums, elog, ereport, errcode(), errdetail(), errdetail_internal(), errhint(), errmsg(), ERROR, PLpgSQL_execstate::eval_processed, exec_eval_cleanup(), exec_move_row(), exec_prepare_plan(), exec_set_found(), format_expr_params(), PLpgSQL_execstate::func, if(), lfirst, PLpgSQL_expr::plan, plpgsql_extra_errors, plpgsql_extra_warnings, PLPGSQL_RC_OK, PLPGSQL_XCHECK_TOOMANYROWS, PLpgSQL_function::print_strict_params, PLpgSQL_expr::query, PLpgSQL_execstate::readonly_func, setup_param_list(), SPI_ERROR_COPY, SPI_ERROR_TRANSACTION, SPI_execute_plan_with_paramlist(), SPI_freetuptable(), SPI_OK_DELETE, SPI_OK_DELETE_RETURNING, SPI_OK_INSERT, SPI_OK_INSERT_RETURNING, SPI_OK_MERGE, SPI_OK_MERGE_RETURNING, SPI_OK_REWRITTEN, SPI_OK_SELECT, SPI_OK_SELINTO, SPI_OK_UPDATE, SPI_OK_UPDATE_RETURNING, SPI_OK_UTILITY, SPI_plan_get_plan_sources(), SPI_processed, SPI_result_code_string(), SPI_tuptable, stmt, SPITupleTable::tupdesc, SPITupleTable::vals, and WARNING.

Referenced by exec_stmt_forc(), exec_stmt_open(), and exec_stmts().

◆ exec_stmt_exit()

static int exec_stmt_exit	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_exit *	stmt
	)

static

Definition at line 3148 of file pl_exec.c.

{
    /*
     * If the exit / continue has a condition, evaluate it
     */
    if (stmt->cond != NULL)
    {
        bool        value;
        bool        isnull;
 
        value = exec_eval_boolean(estate, stmt->cond, &isnull);
        exec_eval_cleanup(estate);
        if (isnull || value == false)
            return PLPGSQL_RC_OK;
    }
 
    estate->exitlabel = stmt->label;
    if (stmt->is_exit)
        return PLPGSQL_RC_EXIT;
    else
        return PLPGSQL_RC_CONTINUE;
}

References exec_eval_boolean(), exec_eval_cleanup(), PLpgSQL_execstate::exitlabel, PLPGSQL_RC_CONTINUE, PLPGSQL_RC_EXIT, PLPGSQL_RC_OK, stmt, and value.

Referenced by exec_stmts().

◆ exec_stmt_fetch()

static int exec_stmt_fetch	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_fetch *	stmt
	)

static

Definition at line 4826 of file pl_exec.c.

{
    PLpgSQL_var *curvar;
    long        how_many = stmt->how_many;
    SPITupleTable *tuptab;
    Portal      portal;
    char       *curname;
    uint64      n;
    MemoryContext oldcontext;
 
    /* ----------
     * Get the portal of the cursor by name
     * ----------
     */
    curvar = (PLpgSQL_var *) (estate->datums[stmt->curvar]);
    if (curvar->isnull)
        ereport(ERROR,
                (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                 errmsg("cursor variable \"%s\" is null", curvar->refname)));
 
    /* Use eval_mcontext for short-lived string */
    oldcontext = MemoryContextSwitchTo(get_eval_mcontext(estate));
    curname = TextDatumGetCString(curvar->value);
    MemoryContextSwitchTo(oldcontext);
 
    portal = SPI_cursor_find(curname);
    if (portal == NULL)
        ereport(ERROR,
                (errcode(ERRCODE_UNDEFINED_CURSOR),
                 errmsg("cursor \"%s\" does not exist", curname)));
 
    /* Calculate position for FETCH_RELATIVE or FETCH_ABSOLUTE */
    if (stmt->expr)
    {
        bool        isnull;
 
        /* XXX should be doing this in LONG not INT width */
        how_many = exec_eval_integer(estate, stmt->expr, &isnull);
 
        if (isnull)
            ereport(ERROR,
                    (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                     errmsg("relative or absolute cursor position is null")));
 
        exec_eval_cleanup(estate);
    }
 
    if (!stmt->is_move)
    {
        PLpgSQL_variable *target;
 
        /* ----------
         * Fetch 1 tuple from the cursor
         * ----------
         */
        SPI_scroll_cursor_fetch(portal, stmt->direction, how_many);
        tuptab = SPI_tuptable;
        n = SPI_processed;
 
        /* ----------
         * Set the target appropriately.
         * ----------
         */
        target = (PLpgSQL_variable *) estate->datums[stmt->target->dno];
        if (n == 0)
            exec_move_row(estate, target, NULL, tuptab->tupdesc);
        else
            exec_move_row(estate, target, tuptab->vals[0], tuptab->tupdesc);
 
        exec_eval_cleanup(estate);
        SPI_freetuptable(tuptab);
    }
    else
    {
        /* Move the cursor */
        SPI_scroll_cursor_move(portal, stmt->direction, how_many);
        n = SPI_processed;
    }
 
    /* Set the ROW_COUNT and the global FOUND variable appropriately. */
    estate->eval_processed = n;
    exec_set_found(estate, n != 0);
 
    return PLPGSQL_RC_OK;
}

References PLpgSQL_execstate::datums, ereport, errcode(), errmsg(), ERROR, PLpgSQL_execstate::eval_processed, exec_eval_cleanup(), exec_eval_integer(), exec_move_row(), exec_set_found(), get_eval_mcontext, if(), PLpgSQL_var::isnull, MemoryContextSwitchTo(), PLPGSQL_RC_OK, PLpgSQL_var::refname, SPI_cursor_find(), SPI_freetuptable(), SPI_processed, SPI_scroll_cursor_fetch(), SPI_scroll_cursor_move(), SPI_tuptable, stmt, TextDatumGetCString, SPITupleTable::tupdesc, SPITupleTable::vals, and PLpgSQL_var::value.

Referenced by exec_stmts().

◆ exec_stmt_forc()

static int exec_stmt_forc	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_forc *	stmt
	)

static

Definition at line 2852 of file pl_exec.c.

{
    PLpgSQL_var *curvar;
    MemoryContext stmt_mcontext = NULL;
    char       *curname = NULL;
    PLpgSQL_expr *query;
    ParamListInfo paramLI;
    Portal      portal;
    int         rc;
 
    /* ----------
     * Get the cursor variable and if it has an assigned name, check
     * that it's not in use currently.
     * ----------
     */
    curvar = (PLpgSQL_var *) (estate->datums[stmt->curvar]);
    if (!curvar->isnull)
    {
        MemoryContext oldcontext;
 
        /* We only need stmt_mcontext to hold the cursor name string */
        stmt_mcontext = get_stmt_mcontext(estate);
        oldcontext = MemoryContextSwitchTo(stmt_mcontext);
        curname = TextDatumGetCString(curvar->value);
        MemoryContextSwitchTo(oldcontext);
 
        if (SPI_cursor_find(curname) != NULL)
            ereport(ERROR,
                    (errcode(ERRCODE_DUPLICATE_CURSOR),
                     errmsg("cursor \"%s\" already in use", curname)));
    }
 
    /* ----------
     * Open the cursor just like an OPEN command
     *
     * Note: parser should already have checked that statement supplies
     * args iff cursor needs them, but we check again to be safe.
     * ----------
     */
    if (stmt->argquery != NULL)
    {
        /* ----------
         * OPEN CURSOR with args.  We fake a SELECT ... INTO ...
         * statement to evaluate the args and put 'em into the
         * internal row.
         * ----------
         */
        PLpgSQL_stmt_execsql set_args;
 
        if (curvar->cursor_explicit_argrow < 0)
            ereport(ERROR,
                    (errcode(ERRCODE_SYNTAX_ERROR),
                     errmsg("arguments given for cursor without arguments")));
 
        memset(&set_args, 0, sizeof(set_args));
        set_args.cmd_type = PLPGSQL_STMT_EXECSQL;
        set_args.lineno = stmt->lineno;
        set_args.sqlstmt = stmt->argquery;
        set_args.into = true;
        /* XXX historically this has not been STRICT */
        set_args.target = (PLpgSQL_variable *)
            (estate->datums[curvar->cursor_explicit_argrow]);
 
        if (exec_stmt_execsql(estate, &set_args) != PLPGSQL_RC_OK)
            elog(ERROR, "open cursor failed during argument processing");
    }
    else
    {
        if (curvar->cursor_explicit_argrow >= 0)
            ereport(ERROR,
                    (errcode(ERRCODE_SYNTAX_ERROR),
                     errmsg("arguments required for cursor")));
    }
 
    query = curvar->cursor_explicit_expr;
    Assert(query);
 
    if (query->plan == NULL)
        exec_prepare_plan(estate, query, curvar->cursor_options);
 
    /*
     * Set up ParamListInfo for this query
     */
    paramLI = setup_param_list(estate, query);
 
    /*
     * Open the cursor (the paramlist will get copied into the portal)
     */
    portal = SPI_cursor_open_with_paramlist(curname, query->plan,
                                            paramLI,
                                            estate->readonly_func);
    if (portal == NULL)
        elog(ERROR, "could not open cursor: %s",
             SPI_result_code_string(SPI_result));
 
    /*
     * If cursor variable was NULL, store the generated portal name in it,
     * after verifying it's okay to assign to.
     */
    if (curname == NULL)
    {
        exec_check_assignable(estate, stmt->curvar);
        assign_text_var(estate, curvar, portal->name);
    }
 
    /*
     * Clean up before entering exec_for_query
     */
    exec_eval_cleanup(estate);
    if (stmt_mcontext)
        MemoryContextReset(stmt_mcontext);
 
    /*
     * Execute the loop.  We can't prefetch because the cursor is accessible
     * to the user, for instance via UPDATE WHERE CURRENT OF within the loop.
     */
    rc = exec_for_query(estate, (PLpgSQL_stmt_forq *) stmt, portal, false);
 
    /* ----------
     * Close portal, and restore cursor variable if it was initially NULL.
     * ----------
     */
    SPI_cursor_close(portal);
 
    if (curname == NULL)
        assign_simple_var(estate, curvar, (Datum) 0, true, false);
 
    return rc;
}

References Assert, assign_simple_var(), assign_text_var(), PLpgSQL_stmt_execsql::cmd_type, PLpgSQL_var::cursor_explicit_argrow, PLpgSQL_var::cursor_explicit_expr, PLpgSQL_var::cursor_options, PLpgSQL_execstate::datums, elog, ereport, errcode(), errmsg(), ERROR, exec_check_assignable(), exec_eval_cleanup(), exec_for_query(), exec_prepare_plan(), exec_stmt_execsql(), get_stmt_mcontext(), PLpgSQL_stmt_execsql::into, PLpgSQL_var::isnull, PLpgSQL_stmt_execsql::lineno, MemoryContextReset(), MemoryContextSwitchTo(), PortalData::name, PLpgSQL_expr::plan, PLPGSQL_RC_OK, PLPGSQL_STMT_EXECSQL, PLpgSQL_execstate::readonly_func, setup_param_list(), SPI_cursor_close(), SPI_cursor_find(), SPI_cursor_open_with_paramlist(), SPI_result, SPI_result_code_string(), PLpgSQL_stmt_execsql::sqlstmt, stmt, PLpgSQL_stmt_execsql::target, TextDatumGetCString, and PLpgSQL_var::value.

Referenced by exec_stmts().

◆ exec_stmt_foreach_a()

static int exec_stmt_foreach_a	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_foreach_a *	stmt
	)

static

Definition at line 2992 of file pl_exec.c.

{
    ArrayType  *arr;
    Oid         arrtype;
    int32       arrtypmod;
    PLpgSQL_datum *loop_var;
    Oid         loop_var_elem_type;
    bool        found = false;
    int         rc = PLPGSQL_RC_OK;
    MemoryContext stmt_mcontext;
    MemoryContext oldcontext;
    ArrayIterator array_iterator;
    Oid         iterator_result_type;
    int32       iterator_result_typmod;
    Datum       value;
    bool        isnull;
 
    /* get the value of the array expression */
    value = exec_eval_expr(estate, stmt->expr, &isnull, &arrtype, &arrtypmod);
    if (isnull)
        ereport(ERROR,
                (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                 errmsg("FOREACH expression must not be null")));
 
    /*
     * Do as much as possible of the code below in stmt_mcontext, to avoid any
     * leaks from called subroutines.  We need a private stmt_mcontext since
     * we'll be calling arbitrary statement code.
     */
    stmt_mcontext = get_stmt_mcontext(estate);
    push_stmt_mcontext(estate);
    oldcontext = MemoryContextSwitchTo(stmt_mcontext);
 
    /* check the type of the expression - must be an array */
    if (!OidIsValid(get_element_type(arrtype)))
        ereport(ERROR,
                (errcode(ERRCODE_DATATYPE_MISMATCH),
                 errmsg("FOREACH expression must yield an array, not type %s",
                        format_type_be(arrtype))));
 
    /*
     * We must copy the array into stmt_mcontext, else it will disappear in
     * exec_eval_cleanup.  This is annoying, but cleanup will certainly happen
     * while running the loop body, so we have little choice.
     */
    arr = DatumGetArrayTypePCopy(value);
 
    /* Clean up any leftover temporary memory */
    exec_eval_cleanup(estate);
 
    /* Slice dimension must be less than or equal to array dimension */
    if (stmt->slice < 0 || stmt->slice > ARR_NDIM(arr))
        ereport(ERROR,
                (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
                 errmsg("slice dimension (%d) is out of the valid range 0..%d",
                        stmt->slice, ARR_NDIM(arr))));
 
    /* Set up the loop variable and see if it is of an array type */
    loop_var = estate->datums[stmt->varno];
    if (loop_var->dtype == PLPGSQL_DTYPE_REC ||
        loop_var->dtype == PLPGSQL_DTYPE_ROW)
    {
        /*
         * Record/row variable is certainly not of array type, and might not
         * be initialized at all yet, so don't try to get its type
         */
        loop_var_elem_type = InvalidOid;
    }
    else
        loop_var_elem_type = get_element_type(plpgsql_exec_get_datum_type(estate,
                                                                          loop_var));
 
    /*
     * Sanity-check the loop variable type.  We don't try very hard here, and
     * should not be too picky since it's possible that exec_assign_value can
     * coerce values of different types.  But it seems worthwhile to complain
     * if the array-ness of the loop variable is not right.
     */
    if (stmt->slice > 0 && loop_var_elem_type == InvalidOid)
        ereport(ERROR,
                (errcode(ERRCODE_DATATYPE_MISMATCH),
                 errmsg("FOREACH ... SLICE loop variable must be of an array type")));
    if (stmt->slice == 0 && loop_var_elem_type != InvalidOid)
        ereport(ERROR,
                (errcode(ERRCODE_DATATYPE_MISMATCH),
                 errmsg("FOREACH loop variable must not be of an array type")));
 
    /* Create an iterator to step through the array */
    array_iterator = array_create_iterator(arr, stmt->slice, NULL);
 
    /* Identify iterator result type */
    if (stmt->slice > 0)
    {
        /* When slicing, nominal type of result is same as array type */
        iterator_result_type = arrtype;
        iterator_result_typmod = arrtypmod;
    }
    else
    {
        /* Without slicing, results are individual array elements */
        iterator_result_type = ARR_ELEMTYPE(arr);
        iterator_result_typmod = arrtypmod;
    }
 
    /* Iterate over the array elements or slices */
    while (array_iterate(array_iterator, &value, &isnull))
    {
        found = true;           /* looped at least once */
 
        /* exec_assign_value and exec_stmts must run in the main context */
        MemoryContextSwitchTo(oldcontext);
 
        /* Assign current element/slice to the loop variable */
        exec_assign_value(estate, loop_var, value, isnull,
                          iterator_result_type, iterator_result_typmod);
 
        /* In slice case, value is temporary; must free it to avoid leakage */
        if (stmt->slice > 0)
            pfree(DatumGetPointer(value));
 
        /*
         * Execute the statements
         */
        rc = exec_stmts(estate, stmt->body);
 
        LOOP_RC_PROCESSING(stmt->label, break);
 
        MemoryContextSwitchTo(stmt_mcontext);
    }
 
    /* Restore memory context state */
    MemoryContextSwitchTo(oldcontext);
    pop_stmt_mcontext(estate);
 
    /* Release temporary memory, including the array value */
    MemoryContextReset(stmt_mcontext);
 
    /*
     * Set the FOUND variable to indicate the result of executing the loop
     * (namely, whether we looped one or more times). This must be set here so
     * that it does not interfere with the value of the FOUND variable inside
     * the loop processing itself.
     */
    exec_set_found(estate, found);
 
    return rc;
}

References ARR_ELEMTYPE, ARR_NDIM, array_create_iterator(), array_iterate(), array_iterator(), DatumGetArrayTypePCopy, DatumGetPointer(), PLpgSQL_execstate::datums, PLpgSQL_datum::dtype, ereport, errcode(), errmsg(), ERROR, exec_assign_value(), exec_eval_cleanup(), exec_eval_expr(), exec_set_found(), exec_stmts(), format_type_be(), get_element_type(), get_stmt_mcontext(), InvalidOid, LOOP_RC_PROCESSING, MemoryContextReset(), MemoryContextSwitchTo(), OidIsValid, pfree(), PLPGSQL_DTYPE_REC, PLPGSQL_DTYPE_ROW, plpgsql_exec_get_datum_type(), PLPGSQL_RC_OK, pop_stmt_mcontext(), push_stmt_mcontext(), stmt, and value.

Referenced by exec_stmts().

◆ exec_stmt_fori()

static int exec_stmt_fori	(	PLpgSQL_execstate *	estate,
		PLpgSQL_stmt_fori *	stmt
	)

static

Definition at line 2680 of file pl_exec.c.

{
    PLpgSQL_var *var;
    Datum       value;
    bool        isnull;
    Oid         valtype;
    int32       valtypmod;
    int32       loop_value;
    int32       end_value;
    int32       step_value;
    bool        found = false;
    int         rc = PLPGSQL_RC_OK;
 
    var = (PLpgSQL_var *) (estate->datums[stmt->var->dno]);
 
    /*
     * Get the value of the lower bound
     */
    value = exec_eval_expr(estate, stmt->lower,
                           &isnull, &valtype, &valtypmod);

Data Structures

Macros

Typedefs

Functions

Variables

Macro Definition Documentation

◆ eval_mcontext_alloc

◆ eval_mcontext_alloc0

◆ get_eval_mcontext

◆ LOOP_RC_PROCESSING

◆ SET_RAISE_OPTION_TEXT

Typedef Documentation

◆ SimpleEcontextStackEntry

Function Documentation

◆ assign_record_var()

◆ assign_simple_var()

◆ assign_text_var()

◆ coerce_function_result_tuple()

◆ compatible_tupdescs()

◆ convert_value_to_string()

◆ copy_plpgsql_datums()

◆ deconstruct_composite_datum()

◆ do_cast_value()

◆ exception_matches_conditions()

◆ exec_assign_c_string()

◆ exec_assign_expr()

◆ exec_assign_value()

◆ exec_cast_value()

◆ exec_check_assignable()

◆ exec_check_rw_parameter()

◆ exec_dynquery_with_params()

◆ exec_eval_boolean()

◆ exec_eval_cleanup()

◆ exec_eval_datum()

◆ exec_eval_expr()

◆ exec_eval_integer()

◆ exec_eval_simple_expr()

◆ exec_eval_using_params()

◆ exec_for_query()

◆ exec_init_tuple_store()

◆ exec_is_simple_query()

◆ exec_move_row()

◆ exec_move_row_from_datum()

◆ exec_move_row_from_fields()

◆ exec_prepare_plan()

◆ exec_run_select()

◆ exec_save_simple_expr()

◆ exec_set_found()

◆ exec_simple_check_plan()

◆ exec_stmt_assert()

◆ exec_stmt_assign()

◆ exec_stmt_block()

◆ exec_stmt_call()

◆ exec_stmt_case()

◆ exec_stmt_close()

◆ exec_stmt_commit()

◆ exec_stmt_dynexecute()

◆ exec_stmt_dynfors()

◆ exec_stmt_execsql()

◆ exec_stmt_exit()

◆ exec_stmt_fetch()

◆ exec_stmt_forc()

◆ exec_stmt_foreach_a()

◆ exec_stmt_fori()