SPI manager.

src/backend/executor/functions.c

@@ -8,7 +8,7 @@
  *
  *
  * IDENTIFICATION
- *    $Header: /cvsroot/pgsql/src/backend/executor/functions.c,v 1.6 1997/08/12 22:52:35 momjian Exp $
+ *    $Header: /cvsroot/pgsql/src/backend/executor/functions.c,v 1.7 1997/08/29 09:02:50 vadim Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -365,9 +365,19 @@ postquel_execute(execution_state  *es,
 Datum
 postquel_function(Func *funcNode, char **args, bool *isNull, bool *isDone)
 {
-    execution_state  *es;
-    Datum            result = 0;
-    FunctionCachePtr fcache = funcNode->func_fcache;
+    execution_state	*es;
+    Datum		result = 0;
+    FunctionCachePtr	fcache = funcNode->func_fcache;
+    CommandId		savedId;
+    
+    /*
+     * Before we start do anything we must save CurrentScanCommandId
+     * to restore it before return to upper Executor. Also, we have to
+     * set CurrentScanCommandId equal to CurrentCommandId.
+     *		- vadim 08/29/97
+     */
+    savedId = GetScanCommandId ();
+    SetScanCommandId (GetCurrentCommandId ());
     
     es = (execution_state *) fcache->func_state;
     if (es == NULL)
@@ -401,22 +411,23 @@ postquel_function(Func *funcNode, char **args, bool *isNull, bool *isDone)
      * If we've gone through every command in this function, we are done.
      */
     if (es == (execution_state *)NULL)
+    {
+	/*
+	 * Reset the execution states to start over again
+	 */
+	es = (execution_state *)fcache->func_state;
+	while (es)
 	{
-	    /*
-	     * Reset the execution states to start over again
-	     */
-	    es = (execution_state *)fcache->func_state;
-	    while (es)
-		{
-		    es->status = F_EXEC_START;
-		    es = es->next;
-		}
-	    /*
-	     * Let caller know we're finished.
-	     */
-	    *isDone = true;
-	    return (fcache->oneResult) ? result : (Datum)NULL;
+	    es->status = F_EXEC_START;
+	    es = es->next;
 	}
+	/*
+	 * Let caller know we're finished.
+	 */
+	*isDone = true;
+    	SetScanCommandId (savedId);
+	return (fcache->oneResult) ? result : (Datum)NULL;
+    }
     /*
      * If we got a result from a command within the function it has
      * to be the final command.  All others shouldn't be returing
@@ -425,5 +436,6 @@ postquel_function(Func *funcNode, char **args, bool *isNull, bool *isDone)
     Assert ( LAST_POSTQUEL_COMMAND(es) );
     *isDone = false;
     
+    SetScanCommandId (savedId);
     return result;
 }

src/include/executor/spi.h

+/*-------------------------------------------------------------------------
+ *
+ * spi.h--
+ *    
+ *
+ *-------------------------------------------------------------------------
+ */
+#ifndef	SPI_H
+#define SPI_H
+
+#include <string.h>
+#include "postgres.h"
+#include "nodes/primnodes.h"
+#include "nodes/relation.h"
+#include "nodes/execnodes.h"
+#include "nodes/plannodes.h"
+#include "catalog/pg_proc.h"
+#include "parser/parse_query.h"
+#include "tcop/pquery.h"
+#include "tcop/tcopprot.h"
+#include "tcop/utility.h"
+#include "tcop/dest.h"
+#include "nodes/params.h"
+#include "utils/fcache.h"
+#include "utils/datum.h"
+#include "utils/elog.h"
+#include "utils/palloc.h"
+#include "utils/syscache.h"
+#include "utils/mcxt.h"
+#include "utils/portal.h"
+#include "catalog/pg_language.h"
+#include "access/heapam.h"
+#include "access/xact.h"
+#include "executor/executor.h"
+#include "executor/execdefs.h"
+
+typedef struct {
+    uint32	alloced;	/* # of alloced vals */
+    uint32	free;		/* # of free vals */
+    TupleDesc	tupdesc;	/* tuple descriptor */
+    HeapTuple	*vals;		/* tuples */
+} SPITupleTable;
+
+#define SPI_ERROR_CONNECT	-1
+#define SPI_ERROR_COPY		-2
+#define SPI_ERROR_OPUNKNOWN	-3
+#define SPI_ERROR_UNCONNECTED	-4
+#define SPI_ERROR_CURSOR	-5
+#define SPI_ERROR_TRANSACTION	-6
+
+#define SPI_OK_CONNECT		0
+#define SPI_OK_FINISH		1
+#define SPI_OK_FETCH		2
+#define SPI_OK_UTILITY		3
+#define SPI_OK_SELECT		4
+#define SPI_OK_SELINTO		5
+#define SPI_OK_INSERT		6
+#define SPI_OK_DELETE		7
+#define SPI_OK_UPDATE		8
+#define SPI_OK_CURSOR		9
+
+extern uint32 SPI_processed;
+extern SPITupleTable *SPI_tuptable;
+
+extern int SPI_connect (void);
+extern int SPI_finish (void);
+extern int SPI_exec (char *src);
+
+#endif /* SPI_H */