Fix timeout issue

Author: monk0707

contrib/babelfishpg_tsql/src/iterative_exec.c

@@ -1,4 +1,5 @@
 #include "access/xact.h"
+#include "access/parallel.h"
 #include "commands/explain.h"
 #include "pltsql.h"
 #include "pltsql-2.h"
@@ -1329,8 +1330,14 @@ dispatch_stmt_handle_error(PLtsql_execstate *estate,
 		 *    when the error is caught by TRY-CATCH
 		 * 3. This causes "cannot DROP ... because it is being used" errors
 		 *    when the procedure later tries to DROP the temp table
+		 *
+		 * Note: exec_state_call_stack can be NULL in parallel workers, so we
+		 * must check before calling is_part_of_pltsql_trycatch_block.
+		 *
+		 * Also skip internal savepoints in parallel workers because
+		 * BeginInternalSubTransaction is not allowed in parallel mode.
 		 */
-		in_trycatch = is_part_of_pltsql_trycatch_block(estate);
+		in_trycatch = (exec_state_call_stack != NULL) ? is_part_of_pltsql_trycatch_block(estate) : false;
 		insert_exec_active = pltsql_insert_exec_rewrite_active();
 		/*
 		 * We need internal savepoints in two cases:
@@ -1346,7 +1353,7 @@ dispatch_stmt_handle_error(PLtsql_execstate *estate,
 		txn_active_or_insert_exec_trycatch = IsTransactionBlockActive() || need_savepoint_for_insert_exec_trycatch;
 
 		if (!ro_func && !pltsql_disable_internal_savepoint && !is_batch_command(stmt) && txn_active_or_insert_exec_trycatch && !is_set_tran_isolation(stmt) && 
-			(!insert_exec_active || in_trycatch))
+			(!insert_exec_active || in_trycatch) && !IsParallelWorker())
 		{
 			elog(DEBUG5, "TSQL TXN Start internal savepoint");
 			BeginInternalSubTransaction(NULL);
@@ -1430,7 +1437,7 @@ dispatch_stmt_handle_error(PLtsql_execstate *estate,
 			}
 			else if (!IsTransactionBlockActive())
 			{
-				if (is_part_of_pltsql_trycatch_block(estate) && !pltsql_insert_exec_rewrite_active())
+				if (exec_state_call_stack != NULL && is_part_of_pltsql_trycatch_block(estate) && !pltsql_insert_exec_rewrite_active())
 				{
 					HOLD_INTERRUPTS();
 					elog(DEBUG1, "TSQL TXN PG semantics : Rollback current transaction");