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

…t 25d639ee Signed-off-by: Tanzeel Khan <[email protected]> (cherry picked from commit 00bce91f5e1b2b1909c80a073acebb980172c07e)
babelfish-for-postgresql · Dec 6, 2024 · a35b28b · a35b28b
1 parent f8601f3
commit a35b28b
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/backend/executor/spi.c b/src/backend/executor/spi.c
@@ -2447,8 +2447,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()