Add DROP PARTITIONED INDEX

Author: srh

src/ast/mod.rs

@@ -6170,6 +6170,8 @@ pub enum ObjectType {
     Sequence,
     Stage,
     Type,
+    // CubeStore extension.
+    PartitionedIndex,
 }
 
 impl fmt::Display for ObjectType {
@@ -6184,6 +6186,7 @@ impl fmt::Display for ObjectType {
             ObjectType::Sequence => "SEQUENCE",
             ObjectType::Stage => "STAGE",
             ObjectType::Type => "TYPE",
+            ObjectType::PartitionedIndex => "PARTITIONED INDEX",
         })
     }
 }

src/parser/mod.rs

@@ -5551,6 +5551,8 @@ impl<'a> Parser<'a> {
             ObjectType::Schema
         } else if self.parse_keyword(Keyword::DATABASE) {
             ObjectType::Database
+        } 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) {
@@ -5571,7 +5573,7 @@ impl<'a> Parser<'a> {
             return self.parse_drop_extension();
         } else {
             return self.expected(
-                "DATABASE, EXTENSION, FUNCTION, INDEX, POLICY, PROCEDURE, ROLE, SCHEMA, SECRET, SEQUENCE, STAGE, TABLE, TRIGGER, TYPE, or VIEW after DROP",
+                "DATABASE, EXTENSION, FUNCTION, INDEX, PARTITIONED INDEX, POLICY, PROCEDURE, ROLE, SCHEMA, SECRET, SEQUENCE, STAGE, TABLE, TRIGGER, TYPE, or VIEW after DROP",
                 self.peek_token(),
             );
         };

tests/sqlparser_common.rs

@@ -3764,6 +3764,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";