add support for abort and end

Author: iskakaushik

src/parser.rs

@@ -427,7 +427,11 @@ impl<'a> Parser<'a> {
                 Keyword::BEGIN => Ok(self.parse_begin()?),
                 Keyword::SAVEPOINT => Ok(self.parse_savepoint()?),
                 Keyword::COMMIT => Ok(self.parse_commit()?),
+                Keyword::END if dialect_of!(self is PostgreSqlDialect) => Ok(self.parse_commit()?),
                 Keyword::ROLLBACK => Ok(self.parse_rollback()?),
+                Keyword::ABORT if dialect_of!(self is PostgreSqlDialect) => {
+                    Ok(self.parse_rollback()?)
+                }
                 Keyword::ASSERT => Ok(self.parse_assert()?),
                 // `PREPARE`, `EXECUTE` and `DEALLOCATE` are Postgres-specific
                 // syntaxes. They are used for Postgres prepared statement.

tests/sqlparser_postgres.rs

@@ -2878,3 +2878,25 @@ fn parse_create_mirror() {
         _ => unreachable!(),
     }
 }
+
+#[test]
+fn parse_abort() {
+    match pg().verified_stmt("ROLLBACK") {
+        Statement::Rollback { chain: false } => (),
+        _ => unreachable!(),
+    }
+    pg().one_statement_parses_to("ABORT", "ROLLBACK");
+    pg().one_statement_parses_to("ABORT WORK", "ROLLBACK");
+    pg().one_statement_parses_to("ABORT TRANSACTION", "ROLLBACK");
+}
+
+#[test]
+fn parse_end() {
+    match pg().verified_stmt("COMMIT") {
+        Statement::Commit { chain: false } => (),
+        _ => unreachable!(),
+    }
+    pg().one_statement_parses_to("END", "COMMIT");
+    pg().one_statement_parses_to("END WORK", "COMMIT");
+    pg().one_statement_parses_to("END TRANSACTION", "COMMIT");
+}