mirrors/datafusion-sqlparse
1
0
Fork 0
mirror of https://github.com/apache/datafusion-sqlparser-rs.git synced 2025-09-04 13:10:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

cleanup parse_statement (#1407)

Browse source
This commit is contained in:
Samuel Colvin 2024-09-01 12:23:29 +01:00 • committed by GitHub
parent df86f259ce
commit d64beea4d4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 50 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

100
src/parser/mod.rs
View file

@ -478,88 +478,88 @@ impl<'a> Parser<'a> {
let next_token = self.next_token(); let next_token = self.next_token();
match &next_token.token { match &next_token.token {
Token::Word(w) => match w.keyword { Token::Word(w) => match w.keyword {
Keyword::KILL => Ok(self.parse_kill()?), Keyword::KILL => self.parse_kill(),
Keyword::FLUSH => Ok(self.parse_flush()?), Keyword::FLUSH => self.parse_flush(),
Keyword::DESC => Ok(self.parse_explain(DescribeAlias::Desc)?), Keyword::DESC => self.parse_explain(DescribeAlias::Desc),
Keyword::DESCRIBE => Ok(self.parse_explain(DescribeAlias::Describe)?), Keyword::DESCRIBE => self.parse_explain(DescribeAlias::Describe),
Keyword::EXPLAIN => Ok(self.parse_explain(DescribeAlias::Explain)?), Keyword::EXPLAIN => self.parse_explain(DescribeAlias::Explain),
Keyword::ANALYZE => Ok(self.parse_analyze()?), Keyword::ANALYZE => self.parse_analyze(),
Keyword::SELECT | Keyword::WITH | Keyword::VALUES => { Keyword::SELECT | Keyword::WITH | Keyword::VALUES => {
self.prev_token(); self.prev_token();
Ok(Statement::Query(self.parse_boxed_query()?)) self.parse_boxed_query().map(Statement::Query)
} }
Keyword::TRUNCATE => Ok(self.parse_truncate()?), Keyword::TRUNCATE => self.parse_truncate(),
Keyword::ATTACH => { Keyword::ATTACH => {
if dialect_of!(self is DuckDbDialect) { if dialect_of!(self is DuckDbDialect) {
Ok(self.parse_attach_duckdb_database()?) self.parse_attach_duckdb_database()
} else { } else {
Ok(self.parse_attach_database()?) self.parse_attach_database()
} }
} }
Keyword::DETACH if dialect_of!(self is DuckDbDialect | GenericDialect) => { Keyword::DETACH if dialect_of!(self is DuckDbDialect | GenericDialect) => {
Ok(self.parse_detach_duckdb_database()?) self.parse_detach_duckdb_database()
} }
Keyword::MSCK => Ok(self.parse_msck()?), Keyword::MSCK => self.parse_msck(),
Keyword::CREATE => Ok(self.parse_create()?), Keyword::CREATE => self.parse_create(),
Keyword::CACHE => Ok(self.parse_cache_table()?), Keyword::CACHE => self.parse_cache_table(),
Keyword::DROP => Ok(self.parse_drop()?), Keyword::DROP => self.parse_drop(),
Keyword::DISCARD => Ok(self.parse_discard()?), Keyword::DISCARD => self.parse_discard(),
Keyword::DECLARE => Ok(self.parse_declare()?), Keyword::DECLARE => self.parse_declare(),
Keyword::FETCH => Ok(self.parse_fetch_statement()?), Keyword::FETCH => self.parse_fetch_statement(),
Keyword::DELETE => Ok(self.parse_delete()?), Keyword::DELETE => self.parse_delete(),
Keyword::INSERT => Ok(self.parse_insert()?), Keyword::INSERT => self.parse_insert(),
Keyword::REPLACE => Ok(self.parse_replace()?), Keyword::REPLACE => self.parse_replace(),
Keyword::UNCACHE => Ok(self.parse_uncache_table()?), Keyword::UNCACHE => self.parse_uncache_table(),
Keyword::UPDATE => Ok(self.parse_update()?), Keyword::UPDATE => self.parse_update(),
Keyword::ALTER => Ok(self.parse_alter()?), Keyword::ALTER => self.parse_alter(),
Keyword::CALL => Ok(self.parse_call()?), Keyword::CALL => self.parse_call(),
Keyword::COPY => Ok(self.parse_copy()?), Keyword::COPY => self.parse_copy(),
Keyword::CLOSE => Ok(self.parse_close()?), Keyword::CLOSE => self.parse_close(),
Keyword::SET => Ok(self.parse_set()?), Keyword::SET => self.parse_set(),
Keyword::SHOW => Ok(self.parse_show()?), Keyword::SHOW => self.parse_show(),
Keyword::USE => Ok(self.parse_use()?), Keyword::USE => self.parse_use(),
Keyword::GRANT => Ok(self.parse_grant()?), Keyword::GRANT => self.parse_grant(),
Keyword::REVOKE => Ok(self.parse_revoke()?), Keyword::REVOKE => self.parse_revoke(),
Keyword::START => Ok(self.parse_start_transaction()?), Keyword::START => self.parse_start_transaction(),
// `BEGIN` is a nonstandard but common alias for the // `BEGIN` is a nonstandard but common alias for the
// standard `START TRANSACTION` statement. It is supported // standard `START TRANSACTION` statement. It is supported
// by at least PostgreSQL and MySQL. // by at least PostgreSQL and MySQL.
Keyword::BEGIN => Ok(self.parse_begin()?), Keyword::BEGIN => self.parse_begin(),
// `END` is a nonstandard but common alias for the // `END` is a nonstandard but common alias for the
// standard `COMMIT TRANSACTION` statement. It is supported // standard `COMMIT TRANSACTION` statement. It is supported
// by PostgreSQL. // by PostgreSQL.
Keyword::END => Ok(self.parse_end()?), Keyword::END => self.parse_end(),
Keyword::SAVEPOINT => Ok(self.parse_savepoint()?), Keyword::SAVEPOINT => self.parse_savepoint(),
Keyword::RELEASE => Ok(self.parse_release()?), Keyword::RELEASE => self.parse_release(),
Keyword::COMMIT => Ok(self.parse_commit()?), Keyword::COMMIT => self.parse_commit(),
Keyword::ROLLBACK => Ok(self.parse_rollback()?), Keyword::ROLLBACK => self.parse_rollback(),
Keyword::ASSERT => Ok(self.parse_assert()?), Keyword::ASSERT => self.parse_assert(),
// `PREPARE`, `EXECUTE` and `DEALLOCATE` are Postgres-specific // `PREPARE`, `EXECUTE` and `DEALLOCATE` are Postgres-specific
// syntaxes. They are used for Postgres prepared statement. // syntaxes. They are used for Postgres prepared statement.
Keyword::DEALLOCATE => Ok(self.parse_deallocate()?), Keyword::DEALLOCATE => self.parse_deallocate(),
Keyword::EXECUTE => Ok(self.parse_execute()?), Keyword::EXECUTE => self.parse_execute(),
Keyword::PREPARE => Ok(self.parse_prepare()?), Keyword::PREPARE => self.parse_prepare(),
Keyword::MERGE => Ok(self.parse_merge()?), Keyword::MERGE => self.parse_merge(),
// `PRAGMA` is sqlite specific https://www.sqlite.org/pragma.html // `PRAGMA` is sqlite specific https://www.sqlite.org/pragma.html
Keyword::PRAGMA => Ok(self.parse_pragma()?), Keyword::PRAGMA => self.parse_pragma(),
Keyword::UNLOAD => Ok(self.parse_unload()?), Keyword::UNLOAD => self.parse_unload(),
// `INSTALL` is duckdb specific https://duckdb.org/docs/extensions/overview // `INSTALL` is duckdb specific https://duckdb.org/docs/extensions/overview
Keyword::INSTALL if dialect_of!(self is DuckDbDialect | GenericDialect) => { Keyword::INSTALL if dialect_of!(self is DuckDbDialect | GenericDialect) => {
Ok(self.parse_install()?) self.parse_install()
} }
// `LOAD` is duckdb specific https://duckdb.org/docs/extensions/overview // `LOAD` is duckdb specific https://duckdb.org/docs/extensions/overview
Keyword::LOAD if dialect_of!(self is DuckDbDialect | GenericDialect) => { Keyword::LOAD if dialect_of!(self is DuckDbDialect | GenericDialect) => {
Ok(self.parse_load()?) self.parse_load()
} }
// `OPTIMIZE` is clickhouse specific https://clickhouse.tech/docs/en/sql-reference/statements/optimize/ // `OPTIMIZE` is clickhouse specific https://clickhouse.tech/docs/en/sql-reference/statements/optimize/
Keyword::OPTIMIZE if dialect_of!(self is ClickHouseDialect | GenericDialect) => { Keyword::OPTIMIZE if dialect_of!(self is ClickHouseDialect | GenericDialect) => {
Ok(self.parse_optimize_table()?) self.parse_optimize_table()
} }
_ => self.expected("an SQL statement", next_token), _ => self.expected("an SQL statement", next_token),
}, },
Token::LParen => { Token::LParen => {
self.prev_token(); self.prev_token();
Ok(Statement::Query(self.parse_boxed_query()?)) self.parse_boxed_query().map(Statement::Query)
} }
_ => self.expected("an SQL statement", next_token), _ => self.expected("an SQL statement", next_token),
} }