parse with connection clause

Author: kitta65

src/parser.rs

@@ -2051,7 +2051,10 @@ impl Parser {
             default.push_node("next_keyword", collate);
             create.push_node("default_collate", default);
         }
-        // TODO with connection
+        if self.get_token(1)?.is("WITH") && self.get_token(2)?.is("CONNECTION") {
+            self.next_token()?; // -> WITH
+            create.push_node("with_connection", self.parse_with_connection_clause()?);
+        }
         if self.get_token(1)?.is("OPTIONS") {
             self.next_token()?; // OPTIONS
             create.push_node("options", self.parse_keyword_with_grouped_exprs(false)?);

src/parser/tests/tests_ddl.rs

@@ -77,6 +77,7 @@ what:
         Box::new(SuccessTestCase::new(
             "\
 CREATE EXTERNAL SCHEMA dataset_name
+WITH CONNECTION connection_name
 ",
             "\
 self: CREATE (CreateSchemaStatement)
@@ -86,6 +87,12 @@ ident:
   self: dataset_name (Identifier)
 what:
   self: SCHEMA (Keyword)
+with_connection:
+  self: WITH (KeywordSequence)
+  next_keyword:
+    self: CONNECTION (KeywordWithExpr)
+    expr:
+      self: connection_name (Identifier)
 ",
             0,
         )),

src/types.rs

@@ -649,10 +649,12 @@ export type CreateRowAccessPolicyStatement = XXXStatement & {
 export type CreateSchemaStatement = XXXStatement & {
   node_type: "CreateSchemaStatement";
   children: {
+    external?: NodeChild;
     what: NodeChild;
     if_not_exists?: NodeVecChild;
     ident: NodeChild;
     default_collate: NodeChild;
+    with_connection?: NodeChild;
     options?: NodeChild;
   };
 };