Fix regression with BEGIN..END and BigQuery

nene · Dec 2, 2024 · d902de2 · d902de2
1 parent b984abe
commit d902de2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/parser.pegjs b/src/parser.pegjs
@@ -64,7 +64,7 @@ program
 
 statement
   = non_transaction_statement
-  / transaction_statement  // math BEGIN transaction after BEGIN..END block
+  / transaction_statement  // match BEGIN transaction after BEGIN..END block
   / &{ return isAcceptUnsupportedGrammar(); } x:unsupported_grammar_stmt { return x; }
 
 // This is referenced by BEGIN..END blocks of FUNCTION definitions,
@@ -198,8 +198,10 @@ inner_program
     });
   }
 
+// BigQuery allows transactions inside BEGIN..END blocks, Postgres doesn't allow it.
 inner_program_statement
-  = non_transaction_statement
+  = &postgres x:non_transaction_statement { return x; }
+  / &bigquery x:statement { return x; }
   / &postgres x:return_stmt { return x; }
 
 /**