them to define operators or use them in functional indexes.
-
Advantages of Using <span class="marked">PL/pgSQL</span>
+
Advantages of Using <span class="marked"><application>PL/pgSQL</application></span>
- There is a special variable named FOUND of type
- boolean that can be used immediately after a SELECT
- INTO to check if an assignment had success (that is, at least one
- row was returned by the SELECT). For example,
+ There is a special variable named FOUND of
+ type boolean. The initial value of
+ FOUND is false; it is set to true when one of
+ the following events occurs:
+
+
+ A SELECT INTO statement is executed, and it returns one or
+ more rows.
+
+
+
+ A UPDATE, INSERT, or DELETE statement is executed, and it
+ affects one or more rows.
+
+
+
+ A PERFORM statement is executed, and it discards one or more
+ rows.
+
+
+
+ A FETCH statement is executed, and it returns an additional
+ row.
+
+
+
+ A FOR statement is executed, and it iterates one or more
+ times. This applies to all three variants of the FOR statement
+ (integer FOR loops, record-set FOR loops, and dynamic
+ record-set FOR loops). FOUND is only set
+ when the FOR loop exits: inside the execution of the loop,
+ FOUND is not modified, although it may be
+ set by the execution of other statements.
+
+
+
+ If none of these events occur, FOUND is set to
+ false. FOUND is a local variable; any changes
+ to it effect only the current
PL/pgSQL+ function.
+
+
+ You can use FOUND immediately after a SELECT
+ INTO statement to determine whether the assignment was successful
+ (that is, at least one row was was returned by the SELECT
+ statement). For example:
SELECT INTO myrec * FROM EMP WHERE empname = myname;
This executes a SELECT
query and discards the
- result.
PL/pgSQL variables are
substituted- in the query as usual. Also, the special variable FOUND is set to
- true if the query produced at least one row, or false if it produced
- no rows.
+ result.
PL/pgSQL variables are
+ substituted in the query as usual. Also, the special variable
+ FOUND is set to true if the query produced at
+ least one row, or false if it produced no rows.
FETCH retrieves the next row from the cursor into a target,
which may be a row variable, a record variable, or a comma-separated
list of simple variables, just like SELECT INTO. As with
- SELECT INTO, the special variable FOUND may be checked to see
- whether a row was obtained or not.
+ SELECT INTO, the special variable FOUND may be
+ checked to see whether a row was obtained or not.
FETCH curs1 INTO rowvar;
* procedural language
*
* IDENTIFICATION
- * $Header: /cvsroot/pgsql/src/pl/plpgsql/src/pl_exec.c,v 1.56 2002/06/24 23:12:06 tgl Exp $
+ * $Header: /cvsroot/pgsql/src/pl/plpgsql/src/pl_exec.c,v 1.57 2002/08/20 05:28:23 momjian Exp $
*
* This software is copyrighted by Jan Wieck - Hamburg.
*
Datum value;
Oid valtype;
bool isnull = false;
- int rc;
+ bool found = false;
+ int rc = PLPGSQL_RC_OK;
var = (PLpgSQL_var *) (estate->datums[stmt->var->varno]);
/*
* Now do the loop
*/
- exec_set_found(estate, false);
for (;;)
{
/*
if ((int4) (var->value) > (int4) value)
break;
}
- exec_set_found(estate, true);
+
+ found = true; /* looped at least once */
/*
* Execute the statements
*/
rc = exec_stmts(estate, stmt->body);
- /*
- * Check returncode
- */
- switch (rc)
+ if (rc == PLPGSQL_RC_RETURN)
+ break; /* return from function */
+ else if (rc == PLPGSQL_RC_EXIT)
{
- case PLPGSQL_RC_OK:
- break;
-
- case PLPGSQL_RC_EXIT:
- if (estate->exitlabel == NULL)
- return PLPGSQL_RC_OK;
- if (stmt->label == NULL)
- return PLPGSQL_RC_EXIT;
- if (strcmp(stmt->label, estate->exitlabel))
- return PLPGSQL_RC_EXIT;
+ if (estate->exitlabel == NULL)
+ /* unlabelled exit, finish the current loop */
+ rc = PLPGSQL_RC_OK;
+ else if (stmt->label != NULL &&
+ strcmp(stmt->label, estate->exitlabel) == 0)
+ {
+ /* labelled exit, matches the current stmt's label */
estate->exitlabel = NULL;
- return PLPGSQL_RC_OK;
+ rc = PLPGSQL_RC_OK;
+ }
- case PLPGSQL_RC_RETURN:
- return PLPGSQL_RC_RETURN;
+ /*
+ * otherwise, we processed a labelled exit that does not
+ * match the current statement's label, if any: return
+ * RC_EXIT so that the EXIT continues to recurse upward.
+ */
- default:
- elog(ERROR, "unknown rc %d from exec_stmts()", rc);
+ break;
}
/*
var->value++;
}
- return PLPGSQL_RC_OK;
+ /*
+ * 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;
}
PLpgSQL_row *row = NULL;
SPITupleTable *tuptab;
Portal portal;
- int rc;
+ bool found = false;
+ int rc = PLPGSQL_RC_OK;
int i;
int n;
- /*
- * Initialize the global found variable to false
- */
- exec_set_found(estate, false);
-
/*
* Determine if we assign to a record or a row
*/
tuptab = SPI_tuptable;
/*
- * If the query didn't return any row, set the target to NULL and
- * return.
+ * If the query didn't return any rows, set the target to NULL and
+ * return with FOUND = false.
*/
if (n == 0)
- {
exec_move_row(estate, rec, row, NULL, NULL);
- SPI_cursor_close(portal);
- return PLPGSQL_RC_OK;
- }
-
- /*
- * There are tuples, so set found to true
- */
- exec_set_found(estate, true);
+ else
+ found = true; /* processed at least one tuple */
/*
* Now do the loop
*/
- for (;;)
+ while (n > 0)
{
for (i = 0; i < n; i++)
{
*/
rc = exec_stmts(estate, stmt->body);
- /*
- * Check returncode
- */
- switch (rc)
+ if (rc != PLPGSQL_RC_OK)
{
- case PLPGSQL_RC_OK:
- break;
-
- case PLPGSQL_RC_EXIT:
- SPI_freetuptable(tuptab);
- SPI_cursor_close(portal);
-
+ /*
+ * We're aborting the loop, so cleanup and set FOUND
+ */
+ exec_set_found(estate, found);
+ SPI_freetuptable(tuptab);
+ SPI_cursor_close(portal);
+
+ if (rc == PLPGSQL_RC_EXIT)
+ {
if (estate->exitlabel == NULL)
- return PLPGSQL_RC_OK;
- if (stmt->label == NULL)
- return PLPGSQL_RC_EXIT;
- if (strcmp(stmt->label, estate->exitlabel))
- return PLPGSQL_RC_EXIT;
- estate->exitlabel = NULL;
- return PLPGSQL_RC_OK;
-
- case PLPGSQL_RC_RETURN:
- SPI_freetuptable(tuptab);
- SPI_cursor_close(portal);
+ /* unlabelled exit, finish the current loop */
+ rc = PLPGSQL_RC_OK;
+ else if (stmt->label != NULL &&
+ strcmp(stmt->label, estate->exitlabel) == 0)
+ {
+ /* labelled exit, matches the current stmt's label */
+ estate->exitlabel = NULL;
+ rc = PLPGSQL_RC_OK;
+ }
- return PLPGSQL_RC_RETURN;
+ /*
+ * otherwise, we processed a labelled exit that does not
+ * match the current statement's label, if any: return
+ * RC_EXIT so that the EXIT continues to recurse upward.
+ */
+ }
- default:
- elog(ERROR, "unknown rc %d from exec_stmts()", rc);
+ return rc;
}
}
SPI_cursor_fetch(portal, true, 50);
n = SPI_processed;
tuptab = SPI_tuptable;
-
- if (n == 0)
- break;
}
/*
*/
SPI_cursor_close(portal);
- return PLPGSQL_RC_OK;
+ /*
+ * 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;
}
/* ----------
* exec_stmt_select Run a query and assign the first
* row to a record or rowtype.
- * ----------
+ * ----------
*/
static int
exec_stmt_select(PLpgSQL_execstate * estate, PLpgSQL_stmt_select * stmt)
PLpgSQL_expr *expr = stmt->sqlstmt;
bool isnull;
+ /*
+ * Set magic FOUND variable to false
+ */
+ exec_set_found(estate, false);
+
/*
* On the first call for this expression generate the plan
*/
{
case SPI_OK_UTILITY:
case SPI_OK_SELINTO:
+ break;
+
+ /*
+ * If the INSERT, DELETE, or UPDATE query affected at least
+ * one tuple, set the magic 'FOUND' variable to true. This
+ * conforms with the behavior of PL/SQL.
+ */
case SPI_OK_INSERT:
case SPI_OK_DELETE:
case SPI_OK_UPDATE:
+ if (SPI_processed > 0)
+ exec_set_found(estate, true);
break;
case SPI_OK_SELECT:
"\n\tIf you want to discard the results, use PERFORM instead.");
default:
- elog(ERROR, "error executing query \"%s\"",
- expr->query);
+ elog(ERROR, "error executing query \"%s\"", expr->query);
}
/*
PLpgSQL_rec *rec = NULL;
PLpgSQL_row *row = NULL;
SPITupleTable *tuptab;
- int rc;
+ int rc = PLPGSQL_RC_OK;
int i;
int n;
HeapTuple typetup;
FmgrInfo finfo_output;
void *plan;
Portal portal;
-
- /*
- * Initialize the global found variable to false
- */
- exec_set_found(estate, false);
+ bool found = false;
/*
* Determine if we assign to a record or a row
tuptab = SPI_tuptable;
/*
- * If the query didn't return any row, set the target to NULL and
- * return.
+ * If the query didn't return any rows, set the target to NULL and
+ * return with FOUND = false.
*/
if (n == 0)
- {
exec_move_row(estate, rec, row, NULL, NULL);
- SPI_cursor_close(portal);
- return PLPGSQL_RC_OK;
- }
-
- /*
- * There are tuples, so set found to true
- */
- exec_set_found(estate, true);
+ else
+ found = true;
/*
* Now do the loop
*/
- for (;;)
+ while (n > 0)
{
for (i = 0; i < n; i++)
{
rc = exec_stmts(estate, stmt->body);
/*
- * Check returncode
+ * We're aborting the loop, so cleanup and set FOUND
*/
- switch (rc)
+ if (rc != PLPGSQL_RC_OK)
{
- case PLPGSQL_RC_OK:
- break;
-
- case PLPGSQL_RC_EXIT:
- SPI_freetuptable(tuptab);
- SPI_cursor_close(portal);
+ exec_set_found(estate, found);
+ SPI_freetuptable(tuptab);
+ SPI_cursor_close(portal);
+ if (rc == PLPGSQL_RC_EXIT)
+ {
if (estate->exitlabel == NULL)
- return PLPGSQL_RC_OK;
- if (stmt->label == NULL)
- return PLPGSQL_RC_EXIT;
- if (strcmp(stmt->label, estate->exitlabel))
- return PLPGSQL_RC_EXIT;
- estate->exitlabel = NULL;
- return PLPGSQL_RC_OK;
-
- case PLPGSQL_RC_RETURN:
- SPI_freetuptable(tuptab);
- SPI_cursor_close(portal);
+ /* unlabelled exit, finish the current loop */
+ rc = PLPGSQL_RC_OK;
+ else if (stmt->label != NULL &&
+ strcmp(stmt->label, estate->exitlabel) == 0)
+ {
+ /* labelled exit, matches the current stmt's label */
+ estate->exitlabel = NULL;
+ rc = PLPGSQL_RC_OK;
+ }
- return PLPGSQL_RC_RETURN;
+ /*
+ * otherwise, we processed a labelled exit that does not
+ * match the current statement's label, if any: return
+ * RC_EXIT so that the EXIT continues to recurse upward.
+ */
+ }
- default:
- elog(ERROR, "unknown rc %d from exec_stmts()", rc);
+ return rc;
}
}
SPI_cursor_fetch(portal, true, 50);
n = SPI_processed;
tuptab = SPI_tuptable;
-
- if (n == 0)
- break;
}
/*
*/
SPI_cursor_close(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 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 PLPGSQL_RC_OK;
}
/* ----------
- * exec_assign_expr Put an expressions result into
+ * exec_assign_expr Put an expression's result into
* a variable.
* ----------
*/
projects / postgresql.git / commitdiff