Support SHOW COLLATION

sqlparser-rs · Aug 9, 2022 · cc29d03 · cc29d03
1 parent 231370a
commit cc29d03
Show file tree

Hide file tree

Showing 4 changed files with 41 additions and 0 deletions.
diff --git a/src/ast/mod.rs b/src/ast/mod.rs
@@ -1012,6 +1012,10 @@ pub enum Statement {
         table_name: ObjectName,
         filter: Option<ShowStatementFilter>,
     },
+    /// SHOW COLLATION
+    ///
+    /// Note: this is a MySQL-specific statement.
+    ShowCollation { filter: Option<ShowStatementFilter> },
     /// `{ BEGIN [ TRANSACTION | WORK ] | START TRANSACTION } ...`
     StartTransaction { modes: Vec<TransactionMode> },
     /// `SET TRANSACTION ...`
@@ -1814,6 +1818,13 @@ impl fmt::Display for Statement {
                 }
                 Ok(())
             }
+            Statement::ShowCollation { filter } => {
+                write!(f, "SHOW COLLATION")?;
+                if let Some(filter) = filter {
+                    write!(f, " {}", filter)?;
+                }
+                Ok(())
+            }
             Statement::StartTransaction { modes } => {
                 write!(f, "START TRANSACTION")?;
                 if !modes.is_empty() {

diff --git a/src/keywords.rs b/src/keywords.rs
@@ -131,6 +131,7 @@ define_keywords!(
     CLUSTER,
     COALESCE,
     COLLATE,
+    COLLATION,
     COLLECT,
     COLUMN,
     COLUMNS,

diff --git a/src/parser.rs b/src/parser.rs
@@ -3694,6 +3694,8 @@ impl<'a> Parser<'a> {
             Ok(self.parse_show_columns()?)
         } else if self.parse_one_of_keywords(&[Keyword::CREATE]).is_some() {
             Ok(self.parse_show_create()?)
+        } else if self.parse_keyword(Keyword::COLLATION) {
+            Ok(self.parse_show_collation()?)
         } else {
             Ok(Statement::ShowVariable {
                 variable: self.parse_identifiers()?,
@@ -3745,6 +3747,11 @@ impl<'a> Parser<'a> {
         })
     }
 
+    pub fn parse_show_collation(&mut self) -> Result<Statement, ParserError> {
+        let filter = self.parse_show_statement_filter()?;
+        Ok(Statement::ShowCollation { filter })
+    }
+
     pub fn parse_show_statement_filter(
         &mut self,
     ) -> Result<Option<ShowStatementFilter>, ParserError> {

diff --git a/tests/sqlparser_mysql.rs b/tests/sqlparser_mysql.rs
@@ -140,6 +140,28 @@ fn parse_show_create() {
     }
 }
 
+#[test]
+fn parse_show_collation() {
+    assert_eq!(
+        mysql_and_generic().verified_stmt("SHOW COLLATION"),
+        Statement::ShowCollation { filter: None }
+    );
+    assert_eq!(
+        mysql_and_generic().verified_stmt("SHOW COLLATION LIKE 'pattern'"),
+        Statement::ShowCollation {
+            filter: Some(ShowStatementFilter::Like("pattern".into())),
+        }
+    );
+    assert_eq!(
+        mysql_and_generic().verified_stmt("SHOW COLLATION WHERE 1 = 2"),
+        Statement::ShowCollation {
+            filter: Some(ShowStatementFilter::Where(
+                mysql_and_generic().verified_expr("1 = 2")
+            )),
+        }
+    );
+}
+
 #[test]
 fn parse_create_table_auto_increment() {
     let sql = "CREATE TABLE foo (bar INT PRIMARY KEY AUTO_INCREMENT)";