Fix SPI clean up incase of errors for insert into execute with @table…

…varible (#274) Introduce functions to fetch SPI connected value (spi stack depth). Extension PR: babelfish-for-postgresql/babelfish_extensions#2183 Issues Resolved: BABEL-4360 Signed-off-by: Tanzeel Khan <[email protected]>
babelfish-for-postgresql · Jan 9, 2024 · 50fba00 · 50fba00
1 parent e00cbef
commit 50fba00
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/backend/executor/spi.c b/src/backend/executor/spi.c
@@ -3432,3 +3432,9 @@ SPI_setCurrentInternalTxnMode(bool mode)
 {
 	_SPI_current->internal_xact = mode;
 }
+
+int
+SPI_get_depth(void)
+{
+	return _SPI_connected;
+}
diff --git a/src/include/executor/spi.h b/src/include/executor/spi.h
@@ -213,4 +213,5 @@ extern void AtEOSubXact_SPI(bool isCommit, SubTransactionId mySubid);
 extern bool SPI_inside_nonatomic_context(void);
 
 extern void SPI_setCurrentInternalTxnMode(bool mode);
+extern int  SPI_get_depth(void);
 #endif							/* SPI_H */