feat: Support USE db

Author: MazterQyou

src/ast/mod.rs

@@ -894,6 +894,10 @@ pub enum Statement {
     ///
     /// Note: this is a MySQL-specific statement.
     ShowCollation { filter: Option<ShowStatementFilter> },
+    /// USE
+    ///
+    /// Note: this is a MySQL-specific statement.
+    Use { db_name: Ident },
     /// `{ BEGIN [ TRANSACTION | WORK ] | START TRANSACTION } ...`
     StartTransaction { modes: Vec<TransactionMode> },
     /// `SET TRANSACTION ...`
@@ -1613,6 +1617,10 @@ impl fmt::Display for Statement {
                 }
                 Ok(())
             }
+            Statement::Use { db_name } => {
+                write!(f, "USE {}", db_name)?;
+                Ok(())
+            }
             Statement::StartTransaction { modes } => {
                 write!(f, "START TRANSACTION")?;
                 if !modes.is_empty() {

src/keywords.rs

@@ -520,6 +520,7 @@ define_keywords!(
     UPDATE,
     UPPER,
     USAGE,
+    USE,
     USER,
     USING,
     UUID,

src/parser.rs

@@ -175,6 +175,7 @@ impl<'a> Parser<'a> {
                 Keyword::SHOW => Ok(self.parse_show()?),
                 Keyword::GRANT => Ok(self.parse_grant()?),
                 Keyword::REVOKE => Ok(self.parse_revoke()?),
+                Keyword::USE => Ok(self.parse_use()?),
                 Keyword::START => Ok(self.parse_start_transaction()?),
                 // `BEGIN` is a nonstandard but common alias for the
                 // standard `START TRANSACTION` statement. It is supported
@@ -3385,6 +3386,11 @@ impl<'a> Parser<'a> {
         }
     }
 
+    pub fn parse_use(&mut self) -> Result<Statement, ParserError> {
+        let db_name = self.parse_identifier()?;
+        Ok(Statement::Use { db_name })
+    }
+
     pub fn parse_table_and_joins(&mut self) -> Result<TableWithJoins, ParserError> {
         let relation = self.parse_table_factor()?;
 

tests/sqlparser_mysql.rs

@@ -184,6 +184,16 @@ fn parse_show_collation() {
     );
 }
 
+#[test]
+fn parse_use() {
+    assert_eq!(
+        mysql_and_generic().verified_stmt("USE database_name"),
+        Statement::Use {
+            db_name: Ident::new("database_name")
+        }
+    );
+}
+
 #[test]
 fn parse_show_create() {
     let obj_name = ObjectName(vec![Ident::new("myident")]);