rebase

Author: cvng

crates/codegen/src/get_node_properties.rs

@@ -664,6 +664,19 @@ fn custom_handlers(node: &Node) -> TokenStream {
  _ => panic!("Unknown IndexElem {:#?}", n.nulls_ordering()),
  }
  },
+ "DefineStmt" => quote! {
+ tokens.push(TokenProperty::from(Token::Create));
+ if n.replace {
+ tokens.push(TokenProperty::from(Token::Or));
+ tokens.push(TokenProperty::from(Token::Replace));
+ }
+ match n.kind() {
+ protobuf::ObjectType::ObjectOperator => {
+ tokens.push(TokenProperty::from(Token::Operator));
+ },
+ _ => todo!("Unknown DefineStmt {:#?}", n.kind())
+ }
+ },
  _ => quote! {},
  }
 }

crates/parser/src/codegen.rs

@@ -224,4 +224,21 @@ mod tests {
  ],
  )
  }
+
+ #[test]
+ fn test_create_operator() {
+ test_get_node_properties(
+ "create operator + (
+ procedure = plusfunc,
+ leftarg = int4,
+ rightarg = int4
+ );",
+ SyntaxKind::DefineStmt,
+ vec![
+ TokenProperty::from(SyntaxKind::Create),
+ TokenProperty::from(SyntaxKind::Operator),
+ TokenProperty::from("+".to_string()),
+ ],
+ )
+ }
 }

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

@@ -0,0 +1,3 @@
+CREATE OPERATOR + (procedure = plusfunc);
+CREATE OPERATOR + (procedure = plusfunc, leftarg = int4, rightarg = int4);
+CREATE OPERATOR + (procedure = plusfunc, hashes, merges);