fix(parse): create partition of table schema (#146)

related #145

Author: chdsbd

parser/src/ast.rs

@@ -729,7 +729,7 @@ pub struct CreateStmt {
     /// relation to create
     pub relation: RelationKind,
     /// column definitions (list of ColumnDef)
-    #[serde(rename = "tableElts")]
+    #[serde(rename = "tableElts", default)]
     pub table_elts: Vec<TableElt>,
     /// relations to inherit from (list of inhRelation)
     #[serde(rename = "inhRelations")]

parser/src/parse.rs

@@ -1224,6 +1224,16 @@ CREATE TABLE example (
     c integer,
     PRIMARY KEY (a, c)
 );
+"#;
+        let res = parse_sql_query(sql);
+        assert_debug_snapshot!(res);
+    }
+
+    #[test]
+    fn test_parse_create_table_partition() {
+        let sql = r#"
+CREATE TABLE measurement_y2006m02 PARTITION OF measurement
+    FOR VALUES FROM ('2006-02-01') TO ('2006-03-01');
 "#;
         let res = parse_sql_query(sql);
         assert_debug_snapshot!(res);

parser/src/snapshots/squawk_parser__parse__tests__parse_create_table_partition.snap

@@ -0,0 +1,121 @@
+---
+source: parser/src/parse.rs
+expression: res
+---
+Ok(
+    [
+        RawStmt(
+            RawStmt {
+                stmt: CreateStmt(
+                    CreateStmt {
+                        relation: RangeVar(
+                            RangeVar {
+                                catalogname: None,
+                                schemaname: None,
+                                relname: "measurement_y2006m02",
+                                inh: true,
+                                relpersistence: "p",
+                                alias: None,
+                                location: 14,
+                            },
+                        ),
+                        table_elts: [],
+                        inh_relations: [
+                            Object({
+                                "RangeVar": Object({
+                                    "inh": Bool(
+                                        true,
+                                    ),
+                                    "location": Number(
+                                        48,
+                                    ),
+                                    "relname": String(
+                                        "measurement",
+                                    ),
+                                    "relpersistence": String(
+                                        "p",
+                                    ),
+                                }),
+                            }),
+                        ],
+                        partbound: Some(
+                            Object({
+                                "PartitionBoundSpec": Object({
+                                    "location": Number(
+                                        75,
+                                    ),
+                                    "lowerdatums": Array([
+                                        Object({
+                                            "PartitionRangeDatum": Object({
+                                                "kind": Number(
+                                                    0,
+                                                ),
+                                                "location": Number(
+                                                    81,
+                                                ),
+                                                "value": Object({
+                                                    "A_Const": Object({
+                                                        "location": Number(
+                                                            81,
+                                                        ),
+                                                        "val": Object({
+                                                            "String": Object({
+                                                                "str": String(
+                                                                    "2006-02-01",
+                                                                ),
+                                                            }),
+                                                        }),
+                                                    }),
+                                                }),
+                                            }),
+                                        }),
+                                    ]),
+                                    "strategy": String(
+                                        "r",
+                                    ),
+                                    "upperdatums": Array([
+                                        Object({
+                                            "PartitionRangeDatum": Object({
+                                                "kind": Number(
+                                                    0,
+                                                ),
+                                                "location": Number(
+                                                    99,
+                                                ),
+                                                "value": Object({
+                                                    "A_Const": Object({
+                                                        "location": Number(
+                                                            99,
+                                                        ),
+                                                        "val": Object({
+                                                            "String": Object({
+                                                                "str": String(
+                                                                    "2006-03-01",
+                                                                ),
+                                                            }),
+                                                        }),
+                                                    }),
+                                                }),
+                                            }),
+                                        }),
+                                    ]),
+                                }),
+                            }),
+                        ),
+                        partspec: None,
+                        of_typename: None,
+                        constraints: [],
+                        options: [],
+                        oncommit: Noop,
+                        tablespacename: None,
+                        if_not_exists: false,
+                    },
+                ),
+                stmt_location: 0,
+                stmt_len: Some(
+                    112,
+                ),
+            },
+        ),
+    ],
+)