Fix a lint, add CREATE TABLE ... LIKE ...

Author: hntd187

src/ast/mod.rs

@@ -513,6 +513,7 @@ pub enum Statement {
         location: Option<String>,
         query: Option<Box<Query>>,
         without_rowid: bool,
+        like: Option<ObjectName>
     },
     /// SQLite's `CREATE VIRTUAL TABLE .. USING <module_name> (<module_args>)`
     CreateVirtualTable {
@@ -808,6 +809,7 @@ impl fmt::Display for Statement {
                 location,
                 query,
                 without_rowid,
+                like
             } => {
                 // We want to allow the following options
                 // Empty column list, allowed by PostgreSQL:
@@ -829,7 +831,7 @@ impl fmt::Display for Statement {
                         write!(f, ", ")?;
                     }
                     write!(f, "{})", display_comma_separated(constraints))?;
-                } else if query.is_none() {
+                } else if query.is_none() && like.is_none() {
                     // PostgreSQL allows `CREATE TABLE t ();`, but requires empty parens
                     write!(f, " ()")?;
                 }
@@ -838,9 +840,13 @@ impl fmt::Display for Statement {
                     write!(f, " WITHOUT ROWID")?;
                 }
 
+                // Only for Hive
+                if let Some(l) = like {
+                    write!(f, " LIKE {}", l)?;
+                }
                 match hive_distribution {
                     HiveDistributionStyle::PARTITIONED { columns } => {
-                        write!(f, " PARTITIONED BY ({})", display_comma_separated(&columns))?
+                        write!(f, " PARTITIONED BY ({})", display_comma_separated(&columns))?;
                     }
                     HiveDistributionStyle::CLUSTERED {
                         columns,

src/parser.rs

@@ -1183,6 +1183,7 @@ impl Parser {
             location: Some(location),
             query: None,
             without_rowid: false,
+            like: None
         })
     }
 
@@ -1327,6 +1328,9 @@ impl Parser {
     pub fn parse_create_table(&mut self) -> Result<Statement, ParserError> {
         let if_not_exists = self.parse_keywords(&[Keyword::IF, Keyword::NOT, Keyword::EXISTS]);
         let table_name = self.parse_object_name()?;
+        let like = if self.parse_keyword(Keyword::LIKE) {
+            self.parse_object_name().ok()
+        } else { None };
         // parse optional column list (schema)
         let (columns, constraints) = self.parse_columns()?;
 
@@ -1358,6 +1362,7 @@ impl Parser {
             location: None,
             query,
             without_rowid,
+            like
         })
     }
 

tests/sqlparser_hive.rs

@@ -75,14 +75,14 @@ fn drop_table_purge() {
     hive().verified_stmt(purge);
 }
 
-fn hive() -> TestedDialects {
-    TestedDialects {
-        dialects: vec![Box::new(HiveDialect {})],
-    }
+#[test]
+fn create_table_like() {
+    let like = "CREATE TABLE db.table_name LIKE db.other_table";
+    hive().verified_stmt(like);
 }
 
-fn hive_and_generic() -> TestedDialects {
+fn hive() -> TestedDialects {
     TestedDialects {
-        dialects: vec![Box::new(HiveDialect {}), Box::new(GenericDialect {})],
+        dialects: vec![Box::new(HiveDialect {})],
     }
 }