Fix partition clauses with values

Author: hntd187

src/ast/mod.rs

@@ -460,7 +460,7 @@ pub enum Statement {
         /// A SQL query that specifies what to insert
         source: Box<Query>,
         /// partitioned insert (Hive)
-        partitioned: Option<Vec<Ident>>
+        partitioned: Option<Vec<Expr>>
     },
     Copy {
         /// TABLE

src/parser.rs

@@ -2358,7 +2358,7 @@ impl Parser {
 
         let partitioned = if self.parse_keyword(Keyword::PARTITION) {
             self.expect_token(&Token::LParen)?;
-            let r = Some(self.parse_comma_separated(Parser::parse_identifier)?);
+            let r = Some(self.parse_comma_separated(Parser::parse_expr)?);
             self.expect_token(&Token::RParen)?;
             r
         } else {

tests/sqlparser_hive.rs

@@ -31,7 +31,7 @@ fn parse_table_create() {
 
 #[test]
 fn parse_insert_overwrite() {
-    let insert_partitions = r#"INSERT OVERWRITE TABLE db.new_table PARTITION (a, b) SELECT a, b, c FROM db.table"#;
+    let insert_partitions = r#"INSERT OVERWRITE TABLE db.new_table PARTITION (a = '1', b) SELECT a, b, c FROM db.table"#;
     hive().verified_stmt(insert_partitions);
 }