BABEL: fix babelfish commit/rollback inside subtransaction post commi…

…t 25d639ee Signed-off-by: Tanzeel Khan <[email protected]> (cherry picked from commit 831461774c33b0845df71e7473e51ca8dfe0019e) (cherry picked from commit f11c2b9)
babelfish-for-postgresql · Dec 11, 2024 · d0c5bcc · d0c5bcc
1 parent e18d34a
commit d0c5bcc
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/backend/executor/spi.c b/src/backend/executor/spi.c
@@ -2442,8 +2442,15 @@ _SPI_execute_plan(SPIPlanPtr plan, const SPIExecuteOptions *options,
 	 * not inside a subtransaction.  The latter two tests match whether
 	 * _SPI_commit() would allow a commit; see there for more commentary.
 	 */
-	allow_nonatomic = options->allow_nonatomic &&
-		!_SPI_current->atomic && !IsSubTransaction();
+	if (check_pltsql_support_tsql_transactions_hook && (*check_pltsql_support_tsql_transactions_hook)())
+	{
+		allow_nonatomic = options->allow_nonatomic && !_SPI_current->atomic;
+	}
+	else
+	{
+		allow_nonatomic = options->allow_nonatomic &&
+			!_SPI_current->atomic && !IsSubTransaction();
+	}
 
 	/*
 	 * Setup error traceback support for ereport()