parse SET ( storage_parameters )

Author: achristmascarl

src/ast/ddl.rs

@@ -351,6 +351,10 @@ pub enum AlterTableOperation {
     ValidateConstraint {
         name: Ident,
     },
+    /// `SET ( storage_parameter [= value] [, ... ] )`
+    SetStorageParameters {
+        storage_parameters: Vec<SqlOption>,
+    },
 }
 
 /// An `ALTER Policy` (`Statement::AlterPolicy`) operation
@@ -791,6 +795,9 @@ impl fmt::Display for AlterTableOperation {
             AlterTableOperation::ValidateConstraint { name } => {
                 write!(f, "VALIDATE CONSTRAINT {name}")
             }
+            AlterTableOperation::SetStorageParameters { storage_parameters } => {
+                write!(f, "SET ({})", display_comma_separated(storage_parameters))
+            }
         }
     }
 }

src/ast/spans.rs

@@ -1201,6 +1201,9 @@ impl Spanned for AlterTableOperation {
             AlterTableOperation::Lock { .. } => Span::empty(),
             AlterTableOperation::ReplicaIdentity { .. } => Span::empty(),
             AlterTableOperation::ValidateConstraint { name } => name.span,
+            AlterTableOperation::SetStorageParameters { storage_parameters } => {
+                union_spans(storage_parameters.iter().map(|i| i.span()))
+            }
         }
     }
 }

src/parser/mod.rs

@@ -8920,17 +8920,23 @@ impl<'a> Parser<'a> {
             let name = self.parse_identifier()?;
             AlterTableOperation::ValidateConstraint { name }
         } else {
-            let options: Vec<SqlOption> =
-                self.parse_options_with_keywords(&[Keyword::SET, Keyword::TBLPROPERTIES])?;
-            if !options.is_empty() {
-                AlterTableOperation::SetTblProperties {
-                    table_properties: options,
+            let maybe_options = self.maybe_parse_options(Keyword::SET)?;
+            if let Some(options) = maybe_options {
+                AlterTableOperation::SetStorageParameters {
+                    storage_parameters: options,
                 }
             } else {
-                return self.expected(
-                    "ADD, RENAME, PARTITION, SWAP, DROP, REPLICA IDENTITY, or SET TBLPROPERTIES after ALTER TABLE",
+                let options: Vec<SqlOption> = self.parse_options(Keyword::TBLPROPERTIES)?;
+                if !options.is_empty() {
+                    AlterTableOperation::SetTblProperties {
+                        table_properties: options,
+                    }
+                } else {
+                    return self.expected(
+                    "ADD, RENAME, PARTITION, SWAP, DROP, REPLICA IDENTITY, SET, or SET TBLPROPERTIES after ALTER TABLE",
                     self.peek_token(),
-                );
+                  );
+                }
             }
         };
         Ok(operation)