Added: SPI_copytuple() & SPI_modifytuple()

diff --git a/src/backend/executor/spi.c b/src/backend/executor/spi.c
index f3aae45ce2a0dfd53075f837150d9e0be215582d..186c0f0313ef145e066d27d3397e3974f6872301 100644 (file)
--- a/src/backend/executor/spi.c
+++ b/src/backend/executor/spi.c
@@ -48,7 +48,7 @@ static void _SPI_fetch(FetchStmt * stmt);
 #endif
 static int
 _SPI_execute_plan(_SPI_plan * plan,
-                 Datum *Values, char *Nulls, int tcount);
+                 Datum * Values, char *Nulls, int tcount);
 
 #define _SPI_CPLAN_CURCXT  0
 #define _SPI_CPLAN_PROCXT  1
@@ -199,7 +199,7 @@ SPI_exec(char *src, int tcount)
 }
 
 int
-SPI_execp(void *plan, Datum *Values, char *Nulls, int tcount)
+SPI_execp(void *plan, Datum * Values, char *Nulls, int tcount)
 {
    int         res;
 
@@ -278,11 +278,108 @@ SPI_saveplan(void *plan)
 
 }
 
+HeapTuple
+SPI_copytuple(HeapTuple tuple)
+{
+   MemoryContext oldcxt = NULL;
+   HeapTuple   ctuple;
+
+   if (tuple == NULL)
+   {
+       SPI_result = SPI_ERROR_ARGUMENT;
+       return (NULL);
+   }
+
+   if (_SPI_curid + 1 == _SPI_connected)       /* connected */
+   {
+       if (_SPI_current != &(_SPI_stack[_SPI_curid + 1]))
+           elog(FATAL, "SPI: stack corrupted");
+       oldcxt = MemoryContextSwitchTo(_SPI_current->savedcxt);
+   }
+
+   ctuple = heap_copytuple(tuple);
+
+   if (oldcxt)
+       MemoryContextSwitchTo(oldcxt);
+
+   return (ctuple);
+}
+
+HeapTuple
+SPI_modifytuple(Relation rel, HeapTuple tuple, int natts, int *attnum,
+               Datum * Values, char *Nulls)
+{
+   MemoryContext oldcxt = NULL;
+   HeapTuple   mtuple;
+   int         numberOfAttributes;
+   uint8       infomask;
+   Datum      *v;
+   char       *n;
+   bool        isnull;
+   int         i;
+
+   if (rel == NULL || tuple == NULL || natts <= 0 || attnum == NULL || Values == NULL)
+   {
+       SPI_result = SPI_ERROR_ARGUMENT;
+       return (NULL);
+   }
+
+   if (_SPI_curid + 1 == _SPI_connected)       /* connected */
+   {
+       if (_SPI_current != &(_SPI_stack[_SPI_curid + 1]))
+           elog(FATAL, "SPI: stack corrupted");
+       oldcxt = MemoryContextSwitchTo(_SPI_current->savedcxt);
+   }
+   SPI_result = 0;
+   numberOfAttributes = rel->rd_att->natts;
+   v = (Datum *) palloc(numberOfAttributes * sizeof(Datum));
+   n = (char *) palloc(numberOfAttributes * sizeof(char));
+
+   /* fetch old values and nulls */
+   for (i = 0; i < numberOfAttributes; i++)
+   {
+       v[i] = heap_getattr(tuple, InvalidBuffer, i + 1, rel->rd_att, &isnull);
+       n[i] = (isnull) ? 'n' : ' ';
+   }
+
+   /* replace values and nulls */
+   for (i = 0; i < natts; i++)
+   {
+       if (attnum[i] <= 0 || attnum[i] > numberOfAttributes)
+           break;
+       v[attnum[i] - 1] = Values[i];
+       n[attnum[i] - 1] = (Nulls && Nulls[i] == 'n') ? 'n' : ' ';
+   }
+
+   if (i == natts)             /* no errors in attnum[] */
+   {
+       mtuple = heap_formtuple(rel->rd_att, v, n);
+       infomask = mtuple->t_infomask;
+       memmove(&(mtuple->t_ctid), &(tuple->t_ctid),
+               ((char *) &(tuple->t_hoff) - (char *) &(tuple->t_ctid)));
+       mtuple->t_infomask = infomask;
+       mtuple->t_natts = numberOfAttributes;
+   }
+   else
+   {
+       mtuple = NULL;
+       SPI_result = SPI_ERROR_NOATTRIBUTE;
+   }
+
+   pfree(v);
+   pfree(n);
+
+   if (oldcxt)
+       MemoryContextSwitchTo(oldcxt);
+
+   return (mtuple);
+}
+
 int
 SPI_fnumber(TupleDesc tupdesc, char *fname)
 {
    int         res;
-   
+
    for (res = 0; res < tupdesc->natts; res++)
    {
        if (strcasecmp(tupdesc->attrs[res]->attname.data, fname) == 0)
@@ -333,7 +430,7 @@ SPI_getvalue(HeapTuple tuple, TupleDesc tupdesc, int fnumber)
 Datum
 SPI_getbinval(HeapTuple tuple, TupleDesc tupdesc, int fnumber, bool * isnull)
 {
-   Datum   val;
+   Datum       val;
 
    *isnull = true;
    SPI_result = 0;
@@ -539,7 +636,7 @@ _SPI_execute(char *src, int tcount, _SPI_plan * plan)
 }
 
 static int
-_SPI_execute_plan(_SPI_plan * plan, Datum *Values, char *Nulls, int tcount)
+_SPI_execute_plan(_SPI_plan * plan, Datum * Values, char *Nulls, int tcount)
 {
    QueryTreeList *queryTree_list = plan->qtlist;
    List       *planTree_list = plan->ptlist;
@@ -591,7 +688,7 @@ _SPI_execute_plan(_SPI_plan * plan, Datum *Values, char *Nulls, int tcount)
                {
                    paramLI->kind = PARAM_NUM;
                    paramLI->id = k + 1;
-                   paramLI->isnull = (Nulls != NULL && Nulls[k] != 'n');
+                   paramLI->isnull = (Nulls && Nulls[k] == 'n');
                    paramLI->value = Values[k];
                }
                paramLI->kind = PARAM_INVALID;

diff --git a/src/include/executor/spi.h b/src/include/executor/spi.h
index 6eab4b8544fce19073a92d4a217cf1c92d17f997..58346be4921786f171254237fec69e0fbb46b73f 100644 (file)
--- a/src/include/executor/spi.h
+++ b/src/include/executor/spi.h
@@ -73,10 +73,14 @@ extern int  SPI_result;
 extern int SPI_connect(void);
 extern int SPI_finish(void);
 extern int SPI_exec(char *src, int tcount);
-extern int SPI_execp(void *plan, Datum *values, char *Nulls, int tcount);
+extern int SPI_execp(void *plan, Datum * values, char *Nulls, int tcount);
 extern void *SPI_prepare(char *src, int nargs, Oid * argtypes);
 extern void *SPI_saveplan(void *plan);
 
+extern HeapTuple SPI_copytuple(HeapTuple tuple);
+extern HeapTuple
+SPI_modifytuple(Relation rel, HeapTuple tuple, int natts,
+               int *attnum, Datum * Values, char *Nulls);
 extern int SPI_fnumber(TupleDesc tupdesc, char *fname);
 extern char *SPI_fname(TupleDesc tupdesc, int fnumber);
 extern char *SPI_getvalue(HeapTuple tuple, TupleDesc tupdesc, int fnumber);