diff options
Diffstat (limited to 'contrib/plpgsql/src/pl_handler.c')
| -rw-r--r-- | contrib/plpgsql/src/pl_handler.c | 193 |
1 files changed, 193 insertions, 0 deletions
diff --git a/contrib/plpgsql/src/pl_handler.c b/contrib/plpgsql/src/pl_handler.c new file mode 100644 index 00000000000..fb01e38b697 --- /dev/null +++ b/contrib/plpgsql/src/pl_handler.c @@ -0,0 +1,193 @@ +/********************************************************************** + * pl_handler.c - Handler for the PL/pgSQL + * procedural language + * + * IDENTIFICATION + * $Header: /cvsroot/pgsql/contrib/plpgsql/src/Attic/pl_handler.c,v 1.1 1998/08/22 12:38:32 momjian Exp $ + * + * This software is copyrighted by Jan Wieck - Hamburg. + * + * The author hereby grants permission to use, copy, modify, + * distribute, and license this software and its documentation + * for any purpose, provided that existing copyright notices are + * retained in all copies and that this notice is included + * verbatim in any distributions. No written agreement, license, + * or royalty fee is required for any of the authorized uses. + * Modifications to this software may be copyrighted by their + * author and need not follow the licensing terms described + * here, provided that the new terms are clearly indicated on + * the first page of each file where they apply. + * + * IN NO EVENT SHALL THE AUTHOR OR DISTRIBUTORS BE LIABLE TO ANY + * PARTY FOR DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR + * CONSEQUENTIAL DAMAGES ARISING OUT OF THE USE OF THIS + * SOFTWARE, ITS DOCUMENTATION, OR ANY DERIVATIVES THEREOF, EVEN + * IF THE AUTHOR HAVE BEEN ADVISED OF THE POSSIBILITY OF SUCH + * DAMAGE. + * + * THE AUTHOR AND DISTRIBUTORS SPECIFICALLY DISCLAIM ANY + * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR + * PURPOSE, AND NON-INFRINGEMENT. THIS SOFTWARE IS PROVIDED ON + * AN "AS IS" BASIS, AND THE AUTHOR AND DISTRIBUTORS HAVE NO + * OBLIGATION TO PROVIDE MAINTENANCE, SUPPORT, UPDATES, + * ENHANCEMENTS, OR MODIFICATIONS. + * + **********************************************************************/ + +#include <stdio.h> +#include <stdlib.h> +#include <stdarg.h> +#include <unistd.h> +#include <fcntl.h> +#include <string.h> + +#include "plpgsql.h" +#include "pl.tab.h" + +#include "executor/spi.h" +#include "commands/trigger.h" +#include "utils/elog.h" +#include "utils/builtins.h" +#include "fmgr.h" +#include "access/heapam.h" + +#include "utils/syscache.h" +#include "catalog/pg_proc.h" +#include "catalog/pg_type.h" + + +static PLpgSQL_function *compiled_functions = NULL; + + +Datum plpgsql_call_handler(FmgrInfo *proinfo, + FmgrValues *proargs, bool *isNull); + +static Datum plpgsql_func_handler(FmgrInfo *proinfo, + FmgrValues *proargs, bool *isNull); + +static HeapTuple plpgsql_trigger_handler(FmgrInfo *proinfo); + + +/* ---------- + * plpgsql_call_handler - This is the only visible function + * of the PL interpreter. The PostgreSQL + * function manager and trigger manager + * call this function for execution of + * PL/pgSQL procedures. + * ---------- + */ +Datum +plpgsql_call_handler(FmgrInfo *proinfo, + FmgrValues *proargs, + bool *isNull) +{ + Datum retval; + + /* ---------- + * Connect to SPI manager + * ---------- + */ + if (SPI_connect() != SPI_OK_CONNECT) { + elog(ERROR, "plpgsql: cannot connect to SPI manager"); + } + + /* ---------- + * Determine if called as function or trigger and + * call appropriate subhandler + * ---------- + */ + if (CurrentTriggerData == NULL) { + retval = plpgsql_func_handler(proinfo, proargs, isNull); + } else { + retval = (Datum)plpgsql_trigger_handler(proinfo); + } + + /* ---------- + * Disconnect from SPI manager + * ---------- + */ + if (SPI_finish() != SPI_OK_FINISH) { + elog(ERROR, "plpgsql: SPI_finish() failed"); + } + + return retval; +} + + +/* ---------- + * plpgsql_func_handler() - Handler for regular function calls + * ---------- + */ +static Datum +plpgsql_func_handler(FmgrInfo *proinfo, + FmgrValues *proargs, + bool *isNull) +{ + PLpgSQL_function *func; + + /* ---------- + * Check if we already compiled this function + * ---------- + */ + for (func = compiled_functions; func != NULL; func = func->next) { + if (proinfo->fn_oid == func->fn_oid) + break; + } + + /* ---------- + * If not, do so and add it to the compiled ones + * ---------- + */ + if (func == NULL) { + func = plpgsql_compile(proinfo->fn_oid, T_FUNCTION); + + func->next = compiled_functions; + compiled_functions = func; + } + + return plpgsql_exec_function(func, proargs, isNull); +} + + +/* ---------- + * plpgsql_trigger_handler() - Handler for trigger calls + * ---------- + */ +static HeapTuple +plpgsql_trigger_handler(FmgrInfo *proinfo) +{ + TriggerData *trigdata; + PLpgSQL_function *func; + + /* ---------- + * Save the current trigger data local + * ---------- + */ + trigdata = CurrentTriggerData; + CurrentTriggerData = NULL; + + /* ---------- + * Check if we already compiled this trigger procedure + * ---------- + */ + for (func = compiled_functions; func != NULL; func = func->next) { + if (proinfo->fn_oid == func->fn_oid) + break; + } + + /* ---------- + * If not, do so and add it to the compiled ones + * ---------- + */ + if (func == NULL) { + func = plpgsql_compile(proinfo->fn_oid, T_TRIGGER); + + func->next = compiled_functions; + compiled_functions = func; + } + + return plpgsql_exec_trigger(func, trigdata); +} + + |