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

Support multiple SET variables (#1252)

Browse source
This commit is contained in:
Ifeanyi Ubah 2024-05-07 18:51:39 +02:00 • committed by GitHub
parent c4f3ef9600
commit eb36bd7138
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 145 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/ast/mod.rs
View file

@ -2275,7 +2275,8 @@ pub enum Statement {
role_name: Option<Ident>,
},
/// ```sql
/// SET <variable>
/// SET <variable> = expression;
/// SET (variable[, ...]) = (expression[, ...]);
/// ```
///
/// Note: this is not a standard SQL statement, but it is supported by at
@ -2284,7 +2285,7 @@ pub enum Statement {
SetVariable {
local: bool,
hivevar: bool,
variable: ObjectName,
variables: OneOrManyWithParens<ObjectName>,
value: Vec<Expr>,
},
/// ```sql
@ -3823,7 +3824,7 @@ impl fmt::Display for Statement {
}
Statement::SetVariable {
local,
variable,
variables,
hivevar,
value,
} => {
@ -3831,12 +3832,15 @@ impl fmt::Display for Statement {
if *local {
f.write_str("LOCAL ")?;
}
let parenthesized = matches!(variables, OneOrManyWithParens::Many(_));
write!(
f,
"{hivevar}{name} = {value}",
"{hivevar}{name} = {l_paren}{value}{r_paren}",
hivevar = if *hivevar { "HIVEVAR:" } else { "" },
name = variable,
value = display_comma_separated(value)
name = variables,
l_paren = parenthesized.then_some("(").unwrap_or_default(),
value = display_comma_separated(value),
r_paren = parenthesized.then_some(")").unwrap_or_default(),
)
}
Statement::SetTimeZone { local, value } => {