mirrors/datafusion-sqlparse
1
0
Fork 0
mirror of https://github.com/apache/datafusion-sqlparser-rs.git synced 2025-10-14 15:59:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Add support for DuckDB functions named arguments with assignment operator (#1195)

Browse source 
Co-authored-by: Andrew Lamb <andrew@nerdnetworks.org>
This commit is contained in:
gstvg 2024-04-06 13:46:36 -03:00 • committed by GitHub
parent e747c9c2af
commit 14b33ac493
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 56 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/parser/mod.rs
View file

@ -3485,7 +3485,7 @@ impl<'a> Parser<'a> {
let name = self.parse_identifier(false)?;
let default_expr =
if self.consume_token(&Token::DuckAssignment) || self.consume_token(&Token::RArrow) {
if self.consume_token(&Token::Assignment) || self.consume_token(&Token::RArrow) {
Some(self.parse_expr()?)
} else {
None
@ -4183,7 +4183,7 @@ impl<'a> Parser<'a> {
self.next_token(); // Skip `DEFAULT`
Some(DeclareAssignment::Default(Box::new(self.parse_expr()?)))
}
Token::DuckAssignment => {
Token::Assignment => {
self.next_token(); // Skip `:=`
Some(DeclareAssignment::DuckAssignment(Box::new(
self.parse_expr()?,
@ -8602,6 +8602,19 @@ impl<'a> Parser<'a> {
arg,
operator: FunctionArgOperator::Equals,
})
} else if dialect_of!(self is DuckDbDialect | GenericDialect)
&& self.peek_nth_token(1) == Token::Assignment
{
let name = self.parse_identifier(false)?;
self.expect_token(&Token::Assignment)?;
let arg = self.parse_expr()?.into();
Ok(FunctionArg::Named {
name,
arg,
operator: FunctionArgOperator::Assignment,
})
} else {
Ok(FunctionArg::Unnamed(self.parse_wildcard_expr()?.into()))
}