Syntax for PARTITIONED INDEX (CubeStore extension)

Author: srh

src/ast/mod.rs

@@ -2169,6 +2169,12 @@ pub enum Statement {
     /// `CREATE INDEX`
     /// ```
     CreateIndex(CreateIndex),
+    // Cubestore extension.
+    CreatePartitionedIndex {
+        name: ObjectName,
+        columns: Vec<ColumnDef>,
+        if_not_exists: bool,
+    },
     /// ```sql
     /// CREATE ROLE
     /// ```
@@ -4529,6 +4535,19 @@ impl fmt::Display for Statement {
                 }
                 Ok(())
             }
+            Statement::CreatePartitionedIndex {
+                name,
+                columns,
+                if_not_exists,
+            } => {
+                write!(
+                    f,
+                    "CREATE PARTITIONED INDEX{} {}({})",
+                    if *if_not_exists { " IF NOT EXISTS" } else { "" },
+                    name,
+                    display_comma_separated(&columns)
+                )
+            }
         }
     }
 }

src/parser/mod.rs

@@ -3554,6 +3554,8 @@ impl<'a> Parser<'a> {
             self.parse_create_virtual_table()
         } else if self.parse_keyword(Keyword::SCHEMA) {
             self.parse_create_schema()
+        } else if self.parse_keywords(&[Keyword::PARTITIONED, Keyword::INDEX]) {
+            self.parse_create_partitioned_index()
         } else if self.parse_keyword(Keyword::DATABASE) {
             self.parse_create_database()
         } else if self.parse_keyword(Keyword::ROLE) {
@@ -5288,6 +5290,19 @@ impl<'a> Parser<'a> {
         Ok(Statement::Discard { object_type })
     }
 
+    pub fn parse_create_partitioned_index(&mut self) -> Result<Statement, ParserError> {
+        let if_not_exists = self.parse_keywords(&[Keyword::IF, Keyword::NOT, Keyword::EXISTS]);
+        let name = self.parse_object_name(false)?;
+        self.expect_token(&Token::LParen)?;
+        let columns = self.parse_comma_separated(Parser::parse_column_def)?;
+        self.expect_token(&Token::RParen)?;
+        Ok(Statement::CreatePartitionedIndex {
+            name,
+            columns,
+            if_not_exists,
+        })
+    }
+
     pub fn parse_create_index(&mut self, unique: bool) -> Result<Statement, ParserError> {
         let concurrently = self.parse_keyword(Keyword::CONCURRENTLY);
         let if_not_exists = self.parse_keywords(&[Keyword::IF, Keyword::NOT, Keyword::EXISTS]);