Support ABORT as alias for ROLLBACK

Fixes #99
nene · Dec 3, 2024 · 3e89684 · 3e89684
1 parent 94b4c59
commit 3e89684
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/src/cst/Transaction.ts b/src/cst/Transaction.ts
@@ -69,7 +69,7 @@ export interface CommitTransactionStmt extends BaseNode {
 
 export interface RollbackTransactionStmt extends BaseNode {
   type: "rollback_transaction_stmt";
-  rollbackKw: Keyword<"ROLLBACK">;
+  rollbackKw: Keyword<"ROLLBACK" | "ABORT">;
   transactionKw?: Keyword<"TRANSACTION" | "WORK">;
   savepoint?: RollbackToSavepoint;
   chain?: TransactionChainClause | TransactionNoChainClause;

diff --git a/src/parser.pegjs b/src/parser.pegjs
@@ -4771,7 +4771,7 @@ commit_kw
   / kw:END (&sqlite / &postgres) { return kw; }
 
 rollback_transaction_stmt
-  = kw:ROLLBACK tKw:(__ transaction_kw)?
+  = kw:rollback_kw tKw:(__ transaction_kw)?
     sp:(__ rollback_to_savepoint)?
     chain:(__ transaction_chain_clause)? {
       return loc({
@@ -4783,6 +4783,10 @@ rollback_transaction_stmt
       });
     }
 
+rollback_kw
+  = ROLLBACK
+  / &postgres x:ABORT { return x; }
+
 rollback_to_savepoint
   = toKw:(TO __) spKw:(SAVEPOINT __)? id:ident {
     return loc({

diff --git a/test/transactions.test.ts b/test/transactions.test.ts
@@ -124,6 +124,18 @@ describe("transactions", () => {
     });
   });
 
+  dialect("postgresql", () => {
+    describe("ABORT as alias for ROLLBACK", () => {
+      it("supports ABORT", () => {
+        testWc("ABORT");
+        testWc("ABORT WORK");
+        testWc("ABORT TRANSACTION");
+        testWc("ABORT AND CHAIN");
+        testWc("ABORT AND NO CHAIN");
+      });
+    });
+  });
+
   dialect(["mysql", "mariadb", "sqlite", "postgresql"], () => {
     describe("creating savepoints", () => {
       it("supports SAVEPOINT", () => {