diff options
Diffstat (limited to 'src/backend/executor/spi.c')
| -rw-r--r-- | src/backend/executor/spi.c | 27 |
1 files changed, 26 insertions, 1 deletions
diff --git a/src/backend/executor/spi.c b/src/backend/executor/spi.c index 7a99a32b9a4..d20e06a0778 100644 --- a/src/backend/executor/spi.c +++ b/src/backend/executor/spi.c @@ -8,7 +8,7 @@ * * * IDENTIFICATION - * $PostgreSQL: pgsql/src/backend/executor/spi.c,v 1.188.2.3 2008/10/16 13:23:28 tgl Exp $ + * $PostgreSQL: pgsql/src/backend/executor/spi.c,v 1.188.2.4 2009/01/07 20:39:05 tgl Exp $ * *------------------------------------------------------------------------- */ @@ -286,6 +286,31 @@ SPI_pop(void) _SPI_curid--; } +/* Conditional push: push only if we're inside a SPI procedure */ +bool +SPI_push_conditional(void) +{ + bool pushed = (_SPI_curid != _SPI_connected); + + if (pushed) + { + _SPI_curid++; + /* We should now be in a state where SPI_connect would succeed */ + Assert(_SPI_curid == _SPI_connected); + } + return pushed; +} + +/* Conditional pop: pop only if SPI_push_conditional pushed */ +void +SPI_pop_conditional(bool pushed) +{ + /* We should be in a state where SPI_connect would succeed */ + Assert(_SPI_curid == _SPI_connected); + if (pushed) + _SPI_curid--; +} + /* Restore state of SPI stack after aborting a subtransaction */ void SPI_restore_connection(void) |