Reset cross-db flag for each statement (babelfish-for-postgresql#3237)

Earlier, while parsing the batch, We were marking the all the subsequent statements of the batch as cross-database statements once we encounter any cross-database statement in batch which was incorrect. This commit fixes above behaviour by resetting 'is_cross_db' maintained at mutator level once we mark some statement as cross_db in the parse tree. Task: BABEL-5447 Signed-off-by: Harsh Lunagariya <[email protected]>
amazon-aurora · Dec 24, 2024 · 316e1c9 · 316e1c9
1 parent 92e34b5
commit 316e1c9
Show file tree

Hide file tree

Showing 3 changed files with 1,621 additions and 0 deletions.
diff --git a/contrib/babelfishpg_tsql/src/tsqlIface.cpp b/contrib/babelfishpg_tsql/src/tsqlIface.cpp
@@ -1857,6 +1857,7 @@ class tsqlBuilder : public tsqlCommonMutator
             graft(makeSQL(ctx), peekContainer());
         }
 
+		is_cross_db = false;
 		// prepare rewriting
 		clear_rewritten_query_fragment();
 		PLtsql_stmt_execsql *stmt = (PLtsql_stmt_execsql *) getPLtsql_fragment(ctx);
@@ -1917,7 +1918,10 @@ class tsqlBuilder : public tsqlCommonMutator
 
 		// record whether stmt is cross-db
 		if (is_cross_db)
+		{
 			stmt->is_cross_db = true;
+			is_cross_db = false;
+		}
 		// record that the stmt is dml
 	 	stmt->is_dml = true;
 		// record if a function call