Merge pull request #19 from MaterializeInc/parse_show_objects_regex

Parse LIKE regex for Statement::ShowObjects

Author: JLDLaughlin

src/ast/mod.rs

@@ -620,7 +620,10 @@ pub enum Statement {
     /// SHOW SOURCES;
     /// SHOW SINKS;
     /// ```
-    ShowObjects { object_type: ObjectType },
+    ShowObjects {
+        object_type: ObjectType,
+        like: Option<String>,
+    },
     /// `SHOW COLUMNS`
     ///
     /// Note: this is a MySQL-specific statement.
@@ -880,7 +883,7 @@ impl fmt::Display for Statement {
                 write!(f, "{} = {}", variable, value)
             }
             Statement::ShowVariable { variable } => write!(f, "SHOW {}", variable),
-            Statement::ShowObjects { object_type } => {
+            Statement::ShowObjects { object_type, like } => {
                 use ObjectType::*;
                 write!(
                     f,
@@ -892,7 +895,11 @@ impl fmt::Display for Statement {
                         Sink => "SINKS",
                         Index => unreachable!(),
                     }
-                )
+                )?;
+                if let Some(like) = like {
+                    write!(f, " LIKE '{}'", value::escape_single_quote_string(like))?;
+                }
+                Ok(())
             }
             Statement::ShowColumns {
                 extended,

src/ast/visit_macro.rs

@@ -505,7 +505,7 @@ macro_rules! make_visitor {
                 visit_show_variable(self, variable)
             }
 
-            fn visit_show_objects(&mut self, _object_type: ObjectType) {}
+            fn visit_show_objects(&mut self, _object_type: ObjectType, _like: &Option<String>) {}
 
             fn visit_show_columns(
                 &mut self,
@@ -654,7 +654,7 @@ macro_rules! make_visitor {
                     value,
                 } => visitor.visit_set_variable(*local, variable, value),
                 Statement::ShowVariable { variable } => visitor.visit_show_variable(variable),
-                Statement::ShowObjects { object_type } => visitor.visit_show_objects(*object_type),
+                Statement::ShowObjects { object_type, like } => visitor.visit_show_objects(*object_type, like),
                 Statement::ShowColumns {
                     extended,
                     full,

src/parser.rs

@@ -1940,6 +1940,7 @@ impl Parser {
                         val
                     ),
                 },
+                like: self.parse_like_filter()?,
             })
         } else if self.parse_keywords(vec!["CREATE", "VIEW"]) {
             Ok(Statement::ShowCreateView {

tests/sqlparser_common.rs

@@ -1912,7 +1912,10 @@ fn parse_show_objects() {
         let sql = format!("SHOW {}", s);
         assert_eq!(
             verified_stmt(&sql),
-            Statement::ShowObjects { object_type: *ot }
+            Statement::ShowObjects {
+                object_type: *ot,
+                like: None
+            }
         )
     }
 }
@@ -2744,6 +2747,18 @@ fn parse_create_sources_with_like_regex() {
     }
 }
 
+#[test]
+fn parse_show_objects_with_like_regex() {
+    let sql = "SHOW TABLES LIKE '%foo%'";
+    match verified_stmt(sql) {
+        Statement::ShowObjects { object_type, like } => {
+            assert_eq!(like.unwrap(), "%foo%");
+            assert_eq!(ObjectType::Table, object_type);
+        }
+        _ => panic!("invalid SHOW OBJECTS statement"),
+    }
+}
+
 #[test]
 fn parse_create_sink() {
     let sql = "CREATE SINK foo FROM bar INTO 'baz' WITH (name = 'val')";