Add DROP PARTITIONED INDEX

Author: srh

src/ast/mod.rs

@@ -5378,6 +5378,8 @@ pub enum ObjectType {
     Role,
     Sequence,
     Stage,
+    // CubeStore extension.
+    PartitionedIndex,
 }
 
 impl fmt::Display for ObjectType {
@@ -5390,6 +5392,7 @@ impl fmt::Display for ObjectType {
             ObjectType::Role => "ROLE",
             ObjectType::Sequence => "SEQUENCE",
             ObjectType::Stage => "STAGE",
+            ObjectType::PartitionedIndex => "PARTITIONED INDEX",
         })
     }
 }

src/parser/mod.rs

@@ -4739,6 +4739,8 @@ impl<'a> Parser<'a> {
             ObjectType::Role
         } else if self.parse_keyword(Keyword::SCHEMA) {
             ObjectType::Schema
+        } else if self.parse_keywords(&[Keyword::PARTITIONED, Keyword::INDEX]) {
+            ObjectType::PartitionedIndex
         } else if self.parse_keyword(Keyword::SEQUENCE) {
             ObjectType::Sequence
         } else if self.parse_keyword(Keyword::STAGE) {
@@ -4753,7 +4755,7 @@ impl<'a> Parser<'a> {
             return self.parse_drop_trigger();
         } else {
             return self.expected(
-                "TABLE, VIEW, INDEX, ROLE, SCHEMA, FUNCTION, PROCEDURE, STAGE, TRIGGER, SECRET or SEQUENCE after DROP",
+                "TABLE, VIEW, INDEX, PARTITIONED INDEX, ROLE, SCHEMA, FUNCTION, PROCEDURE, STAGE, TRIGGER, SECRET or SEQUENCE after DROP",
                 self.peek_token(),
             );
         };

tests/sqlparser_common.rs

@@ -3469,6 +3469,18 @@ fn parse_drop_schema() {
     }
 }
 
+#[test]
+fn parse_drop_partitioned_index() {
+    let sql = "DROP PARTITIONED INDEX X";
+
+    match verified_stmt(sql) {
+        Statement::Drop { object_type, .. } => {
+            assert_eq!(object_type, ObjectType::PartitionedIndex)
+        }
+        _ => unreachable!(),
+    }
+}
+
 #[test]
 fn parse_create_table_as() {
     let sql = "CREATE TABLE t AS SELECT * FROM a";