Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 15 additions & 0 deletions src/parser/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -675,6 +675,7 @@ impl<'a> Parser<'a> {
self.parse_throw().map(Into::into)
}
Keyword::ROLLBACK => self.parse_rollback(),
Keyword::ABORT => self.parse_abort(),
Keyword::ASSERT => self.parse_assert(),
// `PREPARE`, `EXECUTE` and `DEALLOCATE` are Postgres-specific
// syntaxes. They are used for Postgres prepared statement.
Expand Down Expand Up @@ -19275,6 +19276,20 @@ impl<'a> Parser<'a> {
Ok(Statement::Rollback { chain, savepoint })
}

/// Parse an 'ABORT' statement
///
/// ```sql
/// ABORT [ TRANSACTION | WORK ] [ AND [ NO ] CHAIN ]
/// ```
pub fn parse_abort(&mut self) -> Result<Statement, ParserError> {
let chain = self.parse_commit_rollback_chain()?;

Ok(Statement::Rollback {
chain,
savepoint: None,
})
}

/// Parse an optional `AND [NO] CHAIN` clause for `COMMIT` and `ROLLBACK` statements
pub fn parse_commit_rollback_chain(&mut self) -> Result<bool, ParserError> {
let _ = self.parse_one_of_keywords(&[Keyword::TRANSACTION, Keyword::WORK, Keyword::TRAN]);
Expand Down
16 changes: 16 additions & 0 deletions tests/sqlparser_common.rs
Original file line number Diff line number Diff line change
Expand Up @@ -9475,6 +9475,22 @@ fn parse_rollback() {
);
}

#[test]
fn parse_abort() {
one_statement_parses_to("ABORT", "ROLLBACK");
one_statement_parses_to("ABORT TRANSACTION", "ROLLBACK");
one_statement_parses_to("ABORT WORK", "ROLLBACK");
one_statement_parses_to("ABORT AND CHAIN", "ROLLBACK AND CHAIN");
one_statement_parses_to("ABORT AND NO CHAIN", "ROLLBACK");
one_statement_parses_to("ABORT TRANSACTION AND CHAIN", "ROLLBACK AND CHAIN");
one_statement_parses_to("ABORT WORK AND NO CHAIN", "ROLLBACK");

assert_eq!(
parse_sql_statements("ABORT TO test1").unwrap_err(),
ParserError::ParserError("Expected: end of statement, found: TO".to_string()),
);
}

#[test]
#[should_panic(expected = "Parse results with GenericDialect are different from PostgreSqlDialect")]
fn ensure_multiple_dialects_are_tested() {
Expand Down
Loading