Allow the parameters to PL/PgSQL's RAISE statement to be expressions,

instead of just scalar variables. Add regression tests and update the
documentation. Along the way, remove some redundant error checking
code from exec_stmt_perform().

Original patch from Pavel Stehule, reworked by Neil Conway.

src/pl/plpgsql/src/gram.y

@@ -4,7 +4,7 @@
  *						  procedural language
  *
  * IDENTIFICATION
- *	  $PostgreSQL: pgsql/src/pl/plpgsql/src/gram.y,v 1.75 2005/06/10 16:23:11 neilc Exp $
+ *	  $PostgreSQL: pgsql/src/pl/plpgsql/src/gram.y,v 1.76 2005/06/14 06:43:14 neilc Exp $
  *
  *	  This software is copyrighted by Jan Wieck - Hamburg.
  *
@@ -135,8 +135,8 @@ static	void			 plpgsql_sql_error_callback(void *arg);
 %type <exception>	proc_exception
 %type <condition>	proc_conditions
 
-%type <list>	raise_params
-%type <ival>	raise_level raise_param
+
+%type <ival>	raise_level
 %type <str>		raise_msg
 
 %type <list>	getdiag_list
@@ -1157,23 +1157,10 @@ stmt_return_next: K_RETURN_NEXT lno
 					}
 				;
 
-stmt_raise		: K_RAISE lno raise_level raise_msg raise_params ';'
-					{
-						PLpgSQL_stmt_raise		*new;
-
-						new = palloc(sizeof(PLpgSQL_stmt_raise));
-
-						new->cmd_type	= PLPGSQL_STMT_RAISE;
-						new->lineno		= $2;
-						new->elog_level = $3;
-						new->message	= $4;
-						new->params		= $5;
-
-						$$ = (PLpgSQL_stmt *)new;
-					}
-				| K_RAISE lno raise_level raise_msg ';'
+stmt_raise		: K_RAISE lno raise_level raise_msg
 					{
 						PLpgSQL_stmt_raise		*new;
+						int	tok;
 
 						new = palloc(sizeof(PLpgSQL_stmt_raise));
 
@@ -1183,6 +1170,32 @@ stmt_raise		: K_RAISE lno raise_level raise_msg raise_params ';'
 						new->message	= $4;
 						new->params		= NIL;
 
+						tok = yylex();
+
+						/*
+						 * We expect either a semi-colon, which
+						 * indicates no parameters, or a comma that
+						 * begins the list of parameter expressions
+						 */
+						if (tok != ',' && tok != ';')
+							yyerror("syntax error");
+
+						if (tok == ',')
+						{
+							PLpgSQL_expr *expr;
+							int term;
+
+							for (;;)
+							{
+								expr = read_sql_construct(',', ';', ", or ;",
+														  "SELECT ",
+														  true, true, &term);
+								new->params = lappend(new->params, expr);
+								if (term == ';')
+									break;
+							}
+						}
+
 						$$ = (PLpgSQL_stmt *)new;
 					}
 				;
@@ -1219,22 +1232,6 @@ raise_level		: K_EXCEPTION
 					}
 				;
 
-raise_params	: raise_params raise_param
-					{
-						$$ = lappend_int($1, $2);
-					}
-				| raise_param
-					{
-						$$ = list_make1_int($1);
-					}
-				;
-
-raise_param		: ',' T_SCALAR
-					{
-						$$ = yylval.scalar->dno;
-					}
-				;
-
 loop_body		: proc_sect K_END K_LOOP ';'
 					{ $$ = $1; }
 				;
@@ -1658,7 +1655,7 @@ read_sql_stmt(const char *sqlstart)
  * expected:	text to use in complaining that terminator was not found
  * sqlstart:	text to prefix to the accumulated SQL text
  * isexpression: whether to say we're reading an "expression" or a "statement"
- * valid_sql:   whether to check the syntax of the expression (plus sqlstart)
+ * valid_sql:   whether to check the syntax of the expr (prefixed with sqlstart)
  * endtoken:	if not NULL, ending token is stored at *endtoken
  *				(this is only interesting if until2 isn't zero)
  */

src/pl/plpgsql/src/pl_exec.c

@@ -3,7 +3,7 @@
  *			  procedural language
  *
  * IDENTIFICATION
- *	  $PostgreSQL: pgsql/src/pl/plpgsql/src/pl_exec.c,v 1.143 2005/06/10 16:23:11 neilc Exp $
+ *	  $PostgreSQL: pgsql/src/pl/plpgsql/src/pl_exec.c,v 1.144 2005/06/14 06:43:14 neilc Exp $
  *
  *	  This software is copyrighted by Jan Wieck - Hamburg.
  *
@@ -594,7 +594,7 @@ plpgsql_exec_trigger(PLpgSQL_function *func,
 	error_context_stack = plerrcontext.previous;
 
 	/*
-	 * Return the triggers result
+	 * Return the trigger's result
 	 */
 	return rettup;
 }
@@ -1095,22 +1095,9 @@ static int
 exec_stmt_perform(PLpgSQL_execstate *estate, PLpgSQL_stmt_perform *stmt)
 {
 	PLpgSQL_expr *expr = stmt->expr;
-	int			rc;
-
-	/*
-	 * If not already done create a plan for this expression
-	 */
-	if (expr->plan == NULL)
-		exec_prepare_plan(estate, expr);
-
-	rc = exec_run_select(estate, expr, 0, NULL);
-	if (rc != SPI_OK_SELECT)
-		ereport(ERROR,
-				(errcode(ERRCODE_WRONG_OBJECT_TYPE),
-			   errmsg("query \"%s\" did not return data", expr->query)));
 
+	(void) exec_run_select(estate, expr, 0, NULL);
 	exec_set_found(estate, (estate->eval_processed != 0));
-
 	exec_eval_cleanup(estate);
 
 	return PLPGSQL_RC_OK;
@@ -1941,15 +1928,18 @@ exec_stmt_raise(PLpgSQL_execstate *estate, PLpgSQL_stmt_raise *stmt)
 						(errcode(ERRCODE_SYNTAX_ERROR),
 						 errmsg("too few parameters specified for RAISE")));
 
-			exec_eval_datum(estate, estate->datums[lfirst_int(current_param)],
-							InvalidOid,
-							&paramtypeid, &paramvalue, &paramisnull);
+			paramvalue = exec_eval_expr(estate,
+										(PLpgSQL_expr *) lfirst(current_param),
+										&paramisnull,
+										&paramtypeid);
+
 			if (paramisnull)
 				extval = "<NULL>";
 			else
 				extval = convert_value_to_string(paramvalue, paramtypeid);
 			plpgsql_dstring_append(&ds, extval);
 			current_param = lnext(current_param);
+			exec_eval_cleanup(estate);
 			continue;
 		}
 

src/pl/plpgsql/src/pl_funcs.c

@@ -3,7 +3,7 @@
  *			  procedural language
  *
  * IDENTIFICATION
- *	  $PostgreSQL: pgsql/src/pl/plpgsql/src/pl_funcs.c,v 1.42 2005/06/14 00:10:02 neilc Exp $
+ *	  $PostgreSQL: pgsql/src/pl/plpgsql/src/pl_funcs.c,v 1.43 2005/06/14 06:43:14 neilc Exp $
  *
  *	  This software is copyrighted by Jan Wieck - Hamburg.
  *
@@ -885,13 +885,20 @@ dump_return_next(PLpgSQL_stmt_return_next *stmt)
 static void
 dump_raise(PLpgSQL_stmt_raise *stmt)
 {
-	ListCell *l;
+	ListCell *lc;
+	int i = 0;
 
 	dump_ind();
-	printf("RAISE '%s'", stmt->message);
-	foreach (l, stmt->params)
-		printf(" %d", lfirst_int(l));
-	printf("\n");
+	printf("RAISE '%s'\n", stmt->message);
+	dump_indent += 2;
+	foreach (lc, stmt->params)
+	{
+		dump_ind();
+		printf("    parameter %d: ", i++);
+		dump_expr((PLpgSQL_expr *) lfirst(lc));
+		printf("\n");
+	}
+	dump_indent -= 2;
 }
 
 static void
@@ -916,7 +923,8 @@ dump_dynexecute(PLpgSQL_stmt_dynexecute *stmt)
 	{
 		dump_ind();
 		printf("    target = %d %s\n", stmt->rec->recno, stmt->rec->refname);
-	} else if (stmt->row != NULL)
+	}
+	else if (stmt->row != NULL)
 	{
 		dump_ind();
 		printf("    target = %d %s\n", stmt->row->rowno, stmt->row->refname);

src/pl/plpgsql/src/plpgsql.h

@@ -3,7 +3,7 @@
  *			  procedural language
  *
  * IDENTIFICATION
- *	  $PostgreSQL: pgsql/src/pl/plpgsql/src/plpgsql.h,v 1.62 2005/06/10 16:23:11 neilc Exp $
+ *	  $PostgreSQL: pgsql/src/pl/plpgsql/src/plpgsql.h,v 1.63 2005/06/14 06:43:14 neilc Exp $
  *
  *	  This software is copyrighted by Jan Wieck - Hamburg.
  *
@@ -515,7 +515,7 @@ typedef struct
 	int			lineno;
 	int			elog_level;
 	char	   *message;
-	List	   *params;
+	List	   *params;			/* list of expressions */
 } PLpgSQL_stmt_raise;