feat: support CreateTableSpaceStmt

Author: cvng

crates/codegen/src/get_node_properties.rs

@@ -668,6 +668,17 @@ fn custom_handlers(node: &Node) -> TokenStream {
                 _ => panic!("Unknown IndexElem {:#?}", n.nulls_ordering()),
             }
         },
+        "CreateTableSpaceStmt" => quote! {
+            tokens.push(TokenProperty::from(Token::Create));
+            tokens.push(TokenProperty::from(Token::Tablespace));
+            tokens.push(TokenProperty::from(Token::Location));
+            if n.owner.is_some() {
+                tokens.push(TokenProperty::from(Token::Owner));
+            }
+            if n.options.len() > 0 {
+                tokens.push(TokenProperty::from(Token::With));
+            }
+        },
         _ => quote! {},
     }
 }

crates/parser/src/codegen.rs

@@ -239,4 +239,21 @@ mod tests {
             ],
         )
     }
+
+    #[test]
+    fn test_create_tablespace() {
+        test_get_node_properties(
+            "create tablespace x owner a location 'b' with (seq_page_cost=3);",
+            SyntaxKind::CreateTableSpaceStmt,
+            vec![
+                TokenProperty::from(SyntaxKind::Create),
+                TokenProperty::from(SyntaxKind::Tablespace),
+                TokenProperty::from(SyntaxKind::Location),
+                TokenProperty::from(SyntaxKind::Owner),
+                TokenProperty::from(SyntaxKind::With),
+                TokenProperty::from("x".to_string()),
+                TokenProperty::from("b".to_string()),
+            ],
+        )
+    }
 }

crates/parser/tests/data/statements/valid/0048.sql

@@ -0,0 +1,2 @@
+CREATE TABLESPACE x LOCATION 'a';
+CREATE TABLESPACE x OWNER a LOCATION 'b' WITH (random_page_cost=42, seq_page_cost=3);