mirrors/datafusion-sqlparse
1
0
Fork 0
mirror of https://github.com/apache/datafusion-sqlparser-rs.git synced 2025-12-23 11:12:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Add RETURNS TABLE() support for CREATE FUNCTION in Postgresql (#1687)

Browse source 
Co-authored-by: Ifeanyi Ubah <ify1992@yahoo.com>
This commit is contained in:
Rémy SAISSY 2025-02-03 08:27:37 +01:00 • committed by GitHub
parent 906f395341
commit 257da5a82c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 28 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/parser/mod.rs
View file

@ -8867,6 +8867,10 @@ impl<'a> Parser<'a> {
let _ = self.parse_keyword(Keyword::TYPE);
Ok(DataType::AnyType)
}
Keyword::TABLE => {
let columns = self.parse_returns_table_columns()?;
Ok(DataType::Table(columns))
}
_ => {
self.prev_token();
let type_name = self.parse_object_name(false)?;
@ -8894,6 +8898,24 @@ impl<'a> Parser<'a> {
Ok((data, trailing_bracket))
}
fn parse_returns_table_column(&mut self) -> Result<ColumnDef, ParserError> {
let name = self.parse_identifier()?;
let data_type = self.parse_data_type()?;
Ok(ColumnDef {
name,
data_type,
collation: None,
options: Vec::new(), // No constraints expected here
})
}
fn parse_returns_table_columns(&mut self) -> Result<Vec<ColumnDef>, ParserError> {
self.expect_token(&Token::LParen)?;
let columns = self.parse_comma_separated(Parser::parse_returns_table_column)?;
self.expect_token(&Token::RParen)?;
Ok(columns)
}
pub fn parse_string_values(&mut self) -> Result<Vec<String>, ParserError> {
self.expect_token(&Token::LParen)?;
let mut values = Vec::new();