Merge pull request #23 from wangandi/showindexes

Support the MySQL version of SHOW INDEXES.

Author: wangandi

src/ast/mod.rs

@@ -624,6 +624,13 @@ pub enum Statement {
         object_type: ObjectType,
         filter: Option<ShowStatementFilter>,
     },
+    /// `SHOW INDEX|INDEXES|KEYS`
+    ///
+    /// Note: this is a MySQL-specific statement
+    ShowIndexes {
+        table_name: ObjectName,
+        filter: Option<ShowStatementFilter>,
+    },
     /// `SHOW COLUMNS`
     ///
     /// Note: this is a MySQL-specific statement.
@@ -906,6 +913,13 @@ impl fmt::Display for Statement {
                 }
                 Ok(())
             }
+            Statement::ShowIndexes { table_name, filter } => {
+                write!(f, "SHOW INDEXES FROM {}", table_name)?;
+                if let Some(filter) = filter {
+                    write!(f, " {}", filter)?;
+                }
+                Ok(())
+            }
             Statement::ShowColumns {
                 extended,
                 full,

src/ast/visit_macro.rs

@@ -509,6 +509,10 @@ macro_rules! make_visitor {
                 visit_show_objects(self, object_type, filter)
             }
 
+            fn visit_show_indexes(&mut self, table_name: &'ast $($mut)* ObjectName, filter: Option<&'ast $($mut)* ShowStatementFilter>) {
+                visit_show_indexes(self, table_name, filter)
+            }
+
             fn visit_show_columns(
                 &mut self,
                 extended: bool,
@@ -666,6 +670,9 @@ macro_rules! make_visitor {
                 Statement::ShowObjects { object_type, filter } => {
                     visitor.visit_show_objects(*object_type, filter.as_auto_ref())
                 }
+                Statement::ShowIndexes { table_name, filter } => {
+                    visitor.visit_show_indexes(table_name, filter.as_auto_ref())
+                }
                 Statement::ShowColumns {
                     extended,
                     full,
@@ -1559,6 +1566,17 @@ macro_rules! make_visitor {
             }
         }
 
+        pub fn visit_show_indexes<'ast, V: $name<'ast> + ?Sized>(
+            visitor: &mut V,
+            table_name: &'ast $($mut)* ObjectName,
+            filter: Option<&'ast $($mut)* ShowStatementFilter>
+        ) {
+            visitor.visit_object_name(table_name);
+            if let Some(filter) = filter {
+                visitor.visit_show_statement_filter(filter);
+            }
+        }
+
         pub fn visit_show_columns<'ast, V: $name<'ast> + ?Sized>(
             visitor: &mut V,
             _extended: bool,

src/dialect/keywords.rs

@@ -205,6 +205,7 @@ define_keywords!(
     IMMEDIATE,
     IN,
     INDEX,
+    INDEXES,
     INDICATOR,
     INNER,
     INOUT,
@@ -220,6 +221,7 @@ define_keywords!(
     ISOLATION,
     JOIN,
     KEY,
+    KEYS,
     LAG,
     LANGUAGE,
     LARGE,

src/parser.rs

@@ -2013,6 +2013,22 @@ impl Parser {
                 },
                 filter: self.parse_show_statement_filter()?,
             })
+        } else if self
+            .parse_one_of_keywords(&["INDEX", "INDEXES", "KEYS"])
+            .is_some()
+        {
+            match self.parse_one_of_keywords(&["FROM", "IN"]) {
+                Some(_) => {
+                    let table_name = self.parse_object_name()?;
+                    let filter = if self.parse_keyword("WHERE") {
+                        Some(ShowStatementFilter::Where(self.parse_expr()?))
+                    } else {
+                        None
+                    };
+                    Ok(Statement::ShowIndexes { table_name, filter })
+                }
+                None => self.expected("FROM or IN after SHOW INDEXES", self.peek_token()),
+            }
         } else if self.parse_keywords(vec!["CREATE", "VIEW"]) {
             Ok(Statement::ShowCreateView {
                 view_name: self.parse_object_name()?,

tests/sqlparser_common.rs

@@ -2007,6 +2007,58 @@ fn parse_show_objects_with_like_regex() {
     }
 }
 
+#[test]
+fn parse_show_indexes() {
+    let canonical_sql = "SHOW INDEXES FROM foo";
+    assert_eq!(
+        verified_stmt(&canonical_sql),
+        Statement::ShowIndexes {
+            table_name: ObjectName(vec!["foo".into()]),
+            filter: None,
+        }
+    );
+    one_statement_parses_to("SHOW INDEXES IN foo", &canonical_sql);
+    let index_alias = ["INDEX", "KEYS"];
+    let from_alias = ["FROM", "IN"];
+    for i in &index_alias {
+        for f in &from_alias {
+            let sql = format!("SHOW {} {} foo", i, f);
+            one_statement_parses_to(&sql, &canonical_sql);
+        }
+    }
+}
+
+#[test]
+fn parse_show_indexes_with_where_expr() {
+    let canonical_sql = "SHOW INDEXES FROM foo WHERE index_name = 'bar'";
+    match verified_stmt(canonical_sql) {
+        Statement::ShowIndexes { table_name, filter } => {
+            assert_eq!(
+                filter.unwrap(),
+                ShowStatementFilter::Where(Expr::BinaryOp {
+                    left: Box::new(Expr::Identifier(Ident::new("index_name"))),
+                    op: BinaryOperator::Eq,
+                    right: Box::new(Expr::Value(Value::SingleQuotedString("bar".to_string()))),
+                })
+            );
+            assert_eq!(table_name, ObjectName(vec!["foo".into()]));
+        }
+        _ => panic!("invalid SHOW INDEXES statement"),
+    }
+    one_statement_parses_to(
+        "SHOW INDEXES IN foo WHERE index_name = 'bar'",
+        &canonical_sql,
+    );
+    let index_alias = ["INDEX", "KEYS"];
+    let from_alias = ["FROM", "IN"];
+    for i in &index_alias {
+        for f in &from_alias {
+            let sql = format!("SHOW {} {} foo WHERE index_name = 'bar'", i, f);
+            one_statement_parses_to(&sql, &canonical_sql);
+        }
+    }
+}
+
 #[test]
 fn parse_show_create_view() {
     assert_eq!(