Added: SPI_copytuple() & SPI_modifytuple()

src/backend/executor/spi.c

@@ -278,6 +278,103 @@ 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)
 {
@@ -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;

src/include/executor/spi.h

@@ -77,6 +77,10 @@ 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);