mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-12-27 08:39:28 +08:00
Ensure that the contents of a holdable cursor don't depend on out-of-line
toasted values, since those could get dropped once the cursor's transaction is over. Per bug #4553 from Andrew Gierth. Back-patch as far as 8.1. The bug actually exists back to 7.4 when holdable cursors were introduced, but this patch won't work before 8.1 without significant adjustments. Given the lack of field complaints, it doesn't seem worth the work (and risk of introducing new bugs) to try to make a patch for the older branches.
This commit is contained in:
Tom Lane
parent
eea4890c08
commit
ad807ff65d
@ -14,7 +14,7 @@
|
||||
*
|
||||
*
|
||||
* IDENTIFICATION
|
||||
* $PostgreSQL: pgsql/src/backend/commands/portalcmds.c,v 1.57.2.1 2007/02/06 22:49:30 tgl Exp $
|
||||
* $PostgreSQL: pgsql/src/backend/commands/portalcmds.c,v 1.57.2.2 2008/12/01 17:06:35 tgl Exp $
|
||||
*
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
@ -26,6 +26,7 @@
|
||||
#include "access/xact.h"
|
||||
#include "commands/portalcmds.h"
|
||||
#include "executor/executor.h"
|
||||
#include "executor/tstoreReceiver.h"
|
||||
#include "optimizer/planner.h"
|
||||
#include "rewrite/rewriteHandler.h"
|
||||
#include "tcop/pquery.h"
|
||||
@ -375,8 +376,12 @@ PersistHoldablePortal(Portal portal)
|
||||
*/
|
||||
ExecutorRewind(queryDesc);
|
||||
|
||||
/* Change the destination to output to the tuplestore */
|
||||
/*
|
||||
* Change the destination to output to the tuplestore. Note we
|
||||
* tell the tuplestore receiver to detoast all data passed through it.
|
||||
*/
|
||||
queryDesc->dest = CreateDestReceiver(DestTuplestore, portal);
|
||||
SetTuplestoreDestReceiverDeToast(queryDesc->dest, true);
|
||||
|
||||
/* Fetch the result set into the tuplestore */
|
||||
ExecutorRun(queryDesc, ForwardScanDirection, 0L);
|
||||
|
||||
@ -1,46 +1,98 @@
|
||||
/*-------------------------------------------------------------------------
|
||||
*
|
||||
* tstoreReceiver.c
|
||||
* an implementation of DestReceiver that stores the result tuples in
|
||||
* a Tuplestore
|
||||
* An implementation of DestReceiver that stores the result tuples in
|
||||
* a Tuplestore.
|
||||
*
|
||||
* Optionally, we can force detoasting (but not decompression) of out-of-line
|
||||
* toasted values. This is to support cursors WITH HOLD, which must retain
|
||||
* data even if the underlying table is dropped.
|
||||
*
|
||||
*
|
||||
* Portions Copyright (c) 1996-2006, PostgreSQL Global Development Group
|
||||
* Portions Copyright (c) 1996-2008, PostgreSQL Global Development Group
|
||||
* Portions Copyright (c) 1994, Regents of the University of California
|
||||
*
|
||||
* IDENTIFICATION
|
||||
* $PostgreSQL: pgsql/src/backend/executor/tstoreReceiver.c,v 1.17 2006/06/27 02:51:39 tgl Exp $
|
||||
* $PostgreSQL: pgsql/src/backend/executor/tstoreReceiver.c,v 1.17.2.1 2008/12/01 17:06:35 tgl Exp $
|
||||
*
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
|
||||
#include "postgres.h"
|
||||
|
||||
#include "access/heapam.h"
|
||||
#include "access/tuptoaster.h"
|
||||
#include "executor/tstoreReceiver.h"
|
||||
|
||||
|
||||
typedef struct
|
||||
{
|
||||
DestReceiver pub;
|
||||
Tuplestorestate *tstore;
|
||||
MemoryContext cxt;
|
||||
/* parameters: */
|
||||
Tuplestorestate *tstore; /* where to put the data */
|
||||
MemoryContext cxt; /* context containing tstore */
|
||||
bool detoast; /* were we told to detoast? */
|
||||
/* workspace: */
|
||||
Datum *outvalues; /* values array for result tuple */
|
||||
Datum *tofree; /* temp values to be pfree'd */
|
||||
} TStoreState;
|
||||
|
||||
|
||||
static void tstoreReceiveSlot_notoast(TupleTableSlot *slot, DestReceiver *self);
|
||||
static void tstoreReceiveSlot_detoast(TupleTableSlot *slot, DestReceiver *self);
|
||||
|
||||
|
||||
/*
|
||||
* Prepare to receive tuples from executor.
|
||||
*/
|
||||
static void
|
||||
tstoreStartupReceiver(DestReceiver *self, int operation, TupleDesc typeinfo)
|
||||
{
|
||||
/* do nothing */
|
||||
TStoreState *myState = (TStoreState *) self;
|
||||
bool needtoast = false;
|
||||
Form_pg_attribute *attrs = typeinfo->attrs;
|
||||
int natts = typeinfo->natts;
|
||||
int i;
|
||||
|
||||
/* Check if any columns require detoast work */
|
||||
if (myState->detoast)
|
||||
{
|
||||
for (i = 0; i < natts; i++)
|
||||
{
|
||||
if (attrs[i]->attisdropped)
|
||||
continue;
|
||||
if (attrs[i]->attlen == -1)
|
||||
{
|
||||
needtoast = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/* Set up appropriate callback */
|
||||
if (needtoast)
|
||||
{
|
||||
myState->pub.receiveSlot = tstoreReceiveSlot_detoast;
|
||||
/* Create workspace */
|
||||
myState->outvalues = (Datum *)
|
||||
MemoryContextAlloc(myState->cxt, natts * sizeof(Datum));
|
||||
myState->tofree = (Datum *)
|
||||
MemoryContextAlloc(myState->cxt, natts * sizeof(Datum));
|
||||
}
|
||||
else
|
||||
{
|
||||
myState->pub.receiveSlot = tstoreReceiveSlot_notoast;
|
||||
myState->outvalues = NULL;
|
||||
myState->tofree = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
* Receive a tuple from the executor and store it in the tuplestore.
|
||||
* This is for the easy case where we don't have to detoast.
|
||||
*/
|
||||
static void
|
||||
tstoreReceiveSlot(TupleTableSlot *slot, DestReceiver *self)
|
||||
tstoreReceiveSlot_notoast(TupleTableSlot *slot, DestReceiver *self)
|
||||
{
|
||||
TStoreState *myState = (TStoreState *) self;
|
||||
MemoryContext oldcxt = MemoryContextSwitchTo(myState->cxt);
|
||||
@ -50,13 +102,80 @@ tstoreReceiveSlot(TupleTableSlot *slot, DestReceiver *self)
|
||||
MemoryContextSwitchTo(oldcxt);
|
||||
}
|
||||
|
||||
/*
|
||||
* Receive a tuple from the executor and store it in the tuplestore.
|
||||
* This is for the case where we have to detoast any toasted values.
|
||||
*/
|
||||
static void
|
||||
tstoreReceiveSlot_detoast(TupleTableSlot *slot, DestReceiver *self)
|
||||
{
|
||||
TStoreState *myState = (TStoreState *) self;
|
||||
TupleDesc typeinfo = slot->tts_tupleDescriptor;
|
||||
Form_pg_attribute *attrs = typeinfo->attrs;
|
||||
int natts = typeinfo->natts;
|
||||
int nfree;
|
||||
int i;
|
||||
HeapTuple tuple;
|
||||
MemoryContext oldcxt;
|
||||
|
||||
/* Make sure the tuple is fully deconstructed */
|
||||
slot_getallattrs(slot);
|
||||
|
||||
/*
|
||||
* Fetch back any out-of-line datums. We build the new datums array in
|
||||
* myState->outvalues[] (but we can re-use the slot's isnull array).
|
||||
* Also, remember the fetched values to free afterwards.
|
||||
*/
|
||||
nfree = 0;
|
||||
for (i = 0; i < natts; i++)
|
||||
{
|
||||
Datum val = slot->tts_values[i];
|
||||
|
||||
if (!attrs[i]->attisdropped &&
|
||||
attrs[i]->attlen == -1 &&
|
||||
!slot->tts_isnull[i])
|
||||
{
|
||||
if (VARATT_IS_EXTERNAL(DatumGetPointer(val)))
|
||||
{
|
||||
val = PointerGetDatum(heap_tuple_fetch_attr((varattrib *)
|
||||
DatumGetPointer(val)));
|
||||
myState->tofree[nfree++] = val;
|
||||
}
|
||||
}
|
||||
|
||||
myState->outvalues[i] = val;
|
||||
}
|
||||
|
||||
/*
|
||||
* Push the modified tuple into the tuplestore.
|
||||
*/
|
||||
tuple = heap_form_tuple(typeinfo,
|
||||
myState->outvalues, slot->tts_isnull);
|
||||
oldcxt = MemoryContextSwitchTo(myState->cxt);
|
||||
tuplestore_puttuple(myState->tstore, tuple);
|
||||
MemoryContextSwitchTo(oldcxt);
|
||||
heap_freetuple(tuple);
|
||||
|
||||
/* And release any temporary detoasted values */
|
||||
for (i = 0; i < nfree; i++)
|
||||
pfree(DatumGetPointer(myState->tofree[i]));
|
||||
}
|
||||
|
||||
/*
|
||||
* Clean up at end of an executor run
|
||||
*/
|
||||
static void
|
||||
tstoreShutdownReceiver(DestReceiver *self)
|
||||
{
|
||||
/* do nothing */
|
||||
TStoreState *myState = (TStoreState *) self;
|
||||
|
||||
/* Release workspace if any */
|
||||
if (myState->outvalues)
|
||||
pfree(myState->outvalues);
|
||||
myState->outvalues = NULL;
|
||||
if (myState->tofree)
|
||||
pfree(myState->tofree);
|
||||
myState->tofree = NULL;
|
||||
}
|
||||
|
||||
/*
|
||||
@ -75,9 +194,9 @@ DestReceiver *
|
||||
CreateTuplestoreDestReceiver(Tuplestorestate *tStore,
|
||||
MemoryContext tContext)
|
||||
{
|
||||
TStoreState *self = (TStoreState *) palloc(sizeof(TStoreState));
|
||||
TStoreState *self = (TStoreState *) palloc0(sizeof(TStoreState));
|
||||
|
||||
self->pub.receiveSlot = tstoreReceiveSlot;
|
||||
self->pub.receiveSlot = tstoreReceiveSlot_notoast;
|
||||
self->pub.rStartup = tstoreStartupReceiver;
|
||||
self->pub.rShutdown = tstoreShutdownReceiver;
|
||||
self->pub.rDestroy = tstoreDestroyReceiver;
|
||||
@ -85,6 +204,20 @@ CreateTuplestoreDestReceiver(Tuplestorestate *tStore,
|
||||
|
||||
self->tstore = tStore;
|
||||
self->cxt = tContext;
|
||||
self->detoast = false;
|
||||
|
||||
return (DestReceiver *) self;
|
||||
}
|
||||
|
||||
/*
|
||||
* Set parameters for a TuplestoreDestReceiver
|
||||
*/
|
||||
void
|
||||
SetTuplestoreDestReceiverDeToast(DestReceiver *self,
|
||||
bool detoast)
|
||||
{
|
||||
TStoreState *myState = (TStoreState *) self;
|
||||
|
||||
Assert(myState->pub.mydest == DestTuplestore);
|
||||
myState->detoast = detoast;
|
||||
}
|
||||
|
||||
@ -7,7 +7,7 @@
|
||||
* Portions Copyright (c) 1996-2006, PostgreSQL Global Development Group
|
||||
* Portions Copyright (c) 1994, Regents of the University of California
|
||||
*
|
||||
* $PostgreSQL: pgsql/src/include/executor/tstoreReceiver.h,v 1.8 2006/03/05 15:58:56 momjian Exp $
|
||||
* $PostgreSQL: pgsql/src/include/executor/tstoreReceiver.h,v 1.8.2.1 2008/12/01 17:06:35 tgl Exp $
|
||||
*
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
@ -22,4 +22,7 @@
|
||||
extern DestReceiver *CreateTuplestoreDestReceiver(Tuplestorestate *tStore,
|
||||
MemoryContext tContext);
|
||||
|
||||
extern void SetTuplestoreDestReceiverDeToast(DestReceiver *self,
|
||||
bool detoast);
|
||||
|
||||
#endif /* TSTORE_RECEIVER_H */
|
||||
|
||||
Loading…
Reference in New Issue
Block a user