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

Handle optional datatypes properly in CREATE FUNCTION statements (#1826)

Browse source 
Co-authored-by: Ifeanyi Ubah <ify1992@yahoo.com>
This commit is contained in:
Luca Cappelletti 2025-05-21 05:49:28 +02:00 • committed by GitHub
parent 3f4d5f96ee
commit 05d7ffb1d5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 225 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

19
src/parser/mod.rs
View file

@ -5273,12 +5273,21 @@ impl<'a> Parser<'a> {
// parse: [ argname ] argtype
let mut name = None;
let mut data_type = self.parse_data_type()?;
if let DataType::Custom(n, _) = &data_type {
// the first token is actually a name
match n.0[0].clone() {
ObjectNamePart::Identifier(ident) => name = Some(ident),
// To check whether the first token is a name or a type, we need to
// peek the next token, which if it is another type keyword, then the
// first token is a name and not a type in itself.
let data_type_idx = self.get_current_index();
if let Some(next_data_type) = self.maybe_parse(|parser| parser.parse_data_type())? {
let token = self.token_at(data_type_idx);
// We ensure that the token is a `Word` token, and not other special tokens.
if !matches!(token.token, Token::Word(_)) {
return self.expected("a name or type", token.clone());
}
data_type = self.parse_data_type()?;
name = Some(Ident::new(token.to_string()));
data_type = next_data_type;
}
let default_expr = if self.parse_keyword(Keyword::DEFAULT) || self.consume_token(&Token::Eq)