feat: support DefineStmt (aggregate)

Author: cvng

crates/codegen/src/get_node_properties.rs

@@ -527,6 +527,28 @@ fn custom_handlers(node: &Node) -> TokenStream {
  _ => panic!("Unknown CreateCastStmt {:#?}", n.context())
  }
  },
+ "DefineStmt" => quote! {
+ tokens.push(TokenProperty::from(Token::Create));
+ match n.kind() {
+ protobuf::ObjectType::ObjectAggregate => tokens.push(TokenProperty::from(Token::Aggregate)),
+ _ => todo!("Unknown DefineStmt {:#?}", n.kind())
+ }
+ if let Some(node) = &n.args[0].node {
+ if let NodeEnum::List(n) = node {
+ if n.items.len() == 2 {
+ tokens.push(TokenProperty::from(Token::Order));
+ tokens.push(TokenProperty::from(Token::By));
+ tokens.push(TokenProperty::from(Token::Ident));
+ tokens.push(TokenProperty::from(Token::Ident));
+ }
+ }
+ } else {
+ tokens.push(TokenProperty::from(Token::Ascii42));
+ }
+ if n.definition.len() > 0 {
+ tokens.push(TokenProperty::from(Token::Ident));
+ }
+ },
  _ => quote! {},
  }
 }

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

@@ -0,0 +1,6 @@
+CREATE AGGREGATE aggregate1 (int4) (sfunc = sfunc1, stype = stype1);
+CREATE AGGREGATE aggregate1 (int4, bool) (sfunc = sfunc1, stype = stype1);
+CREATE AGGREGATE aggregate1 (*) (sfunc = sfunc1, stype = stype1);
+CREATE AGGREGATE aggregate1 (int4) (sfunc = sfunc1, stype = stype1, finalfunc_extra, mfinalfuncextra);
+CREATE AGGREGATE aggregate1 (int4) (sfunc = sfunc1, stype = stype1, finalfunc_modify = read_only, parallel = restricted);
+CREATE AGGREGATE percentile_disc (float8 ORDER BY anyelement) (sfunc = ordered_set_transition, stype = internal, finalfunc = percentile_disc_final, finalfunc_extra);