Merge pull request #85 from cvng/feat/create-proc

Author: psteinroe

Diff for: crates/codegen/src/get_node_properties.rs

@@ -451,7 +451,11 @@ fn custom_handlers(node: &Node) -> TokenStream {
         },
         "CreateFunctionStmt" => quote! {
             tokens.push(TokenProperty::from(Token::Create));
-            tokens.push(TokenProperty::from(Token::Function));
+            if n.is_procedure {
+                tokens.push(TokenProperty::from(Token::Procedure));
+            } else {
+                tokens.push(TokenProperty::from(Token::Function));
+            }
             if n.replace {
                 tokens.push(TokenProperty::from(Token::Or));
                 tokens.push(TokenProperty::from(Token::Replace));

Diff for: crates/parser/src/codegen.rs

@@ -224,4 +224,19 @@ mod tests {
             ],
         )
     }
+
+    #[test]
+    fn test_create_procedure() {
+        test_get_node_properties(
+            "create procedure insert_data(a integer)
+                language sql
+                as $$insert into tbl values (a);$$;",
+            SyntaxKind::CreateFunctionStmt,
+            vec![
+                TokenProperty::from(SyntaxKind::Create),
+                TokenProperty::from(SyntaxKind::Procedure),
+                TokenProperty::from("insert_data".to_string()),
+            ],
+        )
+    }
 }

Diff for: crates/parser/tests/data/statements/valid/0052.sql

@@ -0,0 +1,2 @@
+CREATE PROCEDURE insert_data(a integer, b integer) LANGUAGE SQL AS $$INSERT INTO tbl VALUES (a); INSERT INTO tbl VALUES (b);$$;
+CREATE PROCEDURE insert_data(a integer, b integer) LANGUAGE SQL BEGIN ATOMIC INSERT INTO tbl VALUES (a); INSERT INTO tbl VALUES (b); END;