Add DROP PARTITIONED INDEX

Author: ilya-biryukov

src/ast/mod.rs

@@ -1464,6 +1464,8 @@ pub enum ObjectType {
     View,
     Index,
     Schema,
+    // CubeStore extension.
+    PartitionedIndex,
 }
 
 impl fmt::Display for ObjectType {
@@ -1473,6 +1475,7 @@ impl fmt::Display for ObjectType {
             ObjectType::View => "VIEW",
             ObjectType::Index => "INDEX",
             ObjectType::Schema => "SCHEMA",
+            ObjectType::PartitionedIndex => "PARTITIONED INDEX",
         })
     }
 }

src/parser.rs

@@ -1413,8 +1413,13 @@ impl<'a> Parser<'a> {
             ObjectType::Index
         } else if self.parse_keyword(Keyword::SCHEMA) {
             ObjectType::Schema
+        } else if self.parse_keywords(&[Keyword::PARTITIONED, Keyword::INDEX]) {
+            ObjectType::PartitionedIndex
         } else {
-            return self.expected("TABLE, VIEW, INDEX or SCHEMA after DROP", self.peek_token());
+            return self.expected(
+                "TABLE, VIEW, INDEX, PARTITIONED INDEX or SCHEMA after DROP",
+                self.peek_token(),
+            );
         };
         // Many dialects support the non standard `IF EXISTS` clause and allow
         // specifying multiple objects to delete in a single statement

tests/sqlparser_common.rs

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