SupportSELECT AS VALUE and SELECT AS STRUCT for BigQuery (#1135)

This commit is contained in:
Lukasz Stefaniak 2024-02-29 13:14:00 +01:00 committed by GitHub
parent 6a9b6f547d
commit 1cf6585649
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 89 additions and 10 deletions

View file

@ -6826,6 +6826,19 @@ impl<'a> Parser<'a> {
/// Parse a restricted `SELECT` statement (no CTEs / `UNION` / `ORDER BY`),
/// assuming the initial `SELECT` was already consumed
pub fn parse_select(&mut self) -> Result<Select, ParserError> {
let value_table_mode =
if dialect_of!(self is BigQueryDialect) && self.parse_keyword(Keyword::AS) {
if self.parse_keyword(Keyword::VALUE) {
Some(ValueTableMode::AsValue)
} else if self.parse_keyword(Keyword::STRUCT) {
Some(ValueTableMode::AsStruct)
} else {
self.expected("VALUE or STRUCT", self.peek_token())?
}
} else {
None
};
let distinct = self.parse_all_or_distinct()?;
let top = if self.parse_keyword(Keyword::TOP) {
@ -6962,6 +6975,7 @@ impl<'a> Parser<'a> {
having,
named_window: named_windows,
qualify,
value_table_mode,
})
}