Syntax for PARTITIONED INDEX (CubeStore extension)

Author: srh

src/ast/mod.rs

@@ -2555,6 +2555,12 @@ pub enum Statement {
     /// `CREATE INDEX`
     /// ```
     CreateIndex(CreateIndex),
+    // Cubestore extension.
+    CreatePartitionedIndex {
+        name: ObjectName,
+        columns: Vec<ColumnDef>,
+        if_not_exists: bool,
+    },
     /// ```sql
     /// CREATE ROLE
     /// ```
@@ -5064,6 +5070,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)
+                )
+            }
             Statement::LISTEN { channel } => {
                 write!(f, "LISTEN {channel}")?;
                 Ok(())

src/ast/spans.rs

@@ -395,6 +395,12 @@ impl Spanned for Statement {
                     .chain(module_args.iter().map(|i| i.span)),
             ),
             Statement::CreateIndex(create_index) => create_index.span(),
+            // Cubestore extension.
+            Statement::CreatePartitionedIndex {
+                name: _,
+                columns: _,
+                if_not_exists: _,
+            } => Span::empty(),  // Not implemented -- maybe if_not_exists being a naked bool is an issue.
             Statement::CreateRole { .. } => Span::empty(),
             Statement::CreateSecret { .. } => Span::empty(),
             Statement::AlterTable {

src/parser/mod.rs

@@ -4209,6 +4209,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) {
@@ -6130,6 +6132,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]);