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

Add support for BigQuery ANY TYPE data type (#1602)

Browse source 
Co-authored-by: Martin Abelson Sahlen <sahlen@Martins-MacBook-Air.local>
Co-authored-by: Martin Abelson Sahlen <sahlen@Mac.lan>
This commit is contained in:
Martin Abelson Sahlen 2024-12-15 11:39:42 +02:00 • committed by GitHub
parent 885aa93465
commit 7bc6ddb8fb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 25 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/ast/data_type.rs
View file

@ -373,6 +373,10 @@ pub enum DataType {
/// ///
/// [postgresql]: https://www.postgresql.org/docs/current/plpgsql-trigger.html /// [postgresql]: https://www.postgresql.org/docs/current/plpgsql-trigger.html
Trigger, Trigger,
/// Any data type, used in BigQuery UDF definitions for templated parameters
///
/// [bigquery]: https://cloud.google.com/bigquery/docs/user-defined-functions#templated-sql-udf-parameters
AnyType,
} }
impl fmt::Display for DataType { impl fmt::Display for DataType {
@ -383,7 +387,6 @@ impl fmt::Display for DataType {
DataType::CharacterVarying(size) => { DataType::CharacterVarying(size) => {
format_character_string_type(f, "CHARACTER VARYING", size) format_character_string_type(f, "CHARACTER VARYING", size)
} }
DataType::CharVarying(size) => format_character_string_type(f, "CHAR VARYING", size), DataType::CharVarying(size) => format_character_string_type(f, "CHAR VARYING", size),
DataType::Varchar(size) => format_character_string_type(f, "VARCHAR", size), DataType::Varchar(size) => format_character_string_type(f, "VARCHAR", size),
DataType::Nvarchar(size) => format_character_string_type(f, "NVARCHAR", size), DataType::Nvarchar(size) => format_character_string_type(f, "NVARCHAR", size),
@ -626,6 +629,7 @@ impl fmt::Display for DataType {
} }
DataType::Unspecified => Ok(()), DataType::Unspecified => Ok(()),
DataType::Trigger => write!(f, "TRIGGER"), DataType::Trigger => write!(f, "TRIGGER"),
DataType::AnyType => write!(f, "ANY TYPE"),
} }
} }
} }

4
src/parser/mod.rs
View file

@ -8382,6 +8382,10 @@ impl<'a> Parser<'a> {
Ok(DataType::Tuple(field_defs)) Ok(DataType::Tuple(field_defs))
} }
Keyword::TRIGGER => Ok(DataType::Trigger), Keyword::TRIGGER => Ok(DataType::Trigger),
Keyword::ANY if self.peek_keyword(Keyword::TYPE) => {
let _ = self.parse_keyword(Keyword::TYPE);
Ok(DataType::AnyType)
}
_ => { _ => {
self.prev_token(); self.prev_token();
let type_name = self.parse_object_name(false)?; let type_name = self.parse_object_name(false)?;

16
tests/sqlparser_bigquery.rs
View file

@ -2212,3 +2212,19 @@ fn test_any_value() {
bigquery_and_generic().verified_expr("ANY_VALUE(fruit HAVING MAX sold)"); bigquery_and_generic().verified_expr("ANY_VALUE(fruit HAVING MAX sold)");
bigquery_and_generic().verified_expr("ANY_VALUE(fruit HAVING MIN sold)"); bigquery_and_generic().verified_expr("ANY_VALUE(fruit HAVING MIN sold)");
} }
#[test]
fn test_any_type() {
bigquery().verified_stmt(concat!(
"CREATE OR REPLACE TEMPORARY FUNCTION ",
"my_function(param1 ANY TYPE) ",
"AS (",
"(SELECT 1)",
")",
));
}
#[test]
fn test_any_type_dont_break_custom_type() {
bigquery_and_generic().verified_stmt("CREATE TABLE foo (x ANY)");
}