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

support "set time zone to 'some-timezone'" (#617)

Browse source 
* support "set time zone"

* fix clippy

* fix test cases
This commit is contained in:
Wei-Ting Kuo 2022-09-20 21:44:51 +08:00 • committed by GitHub
parent 48fa79d744
commit fccae77c5e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 52 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/parser.rs
View file

@ -3774,7 +3774,12 @@ impl<'a> Parser<'a> {
});
}
let variable = self.parse_object_name()?;
let variable = if self.parse_keywords(&[Keyword::TIME, Keyword::ZONE]) {
ObjectName(vec!["TIMEZONE".into()])
} else {
self.parse_object_name()?
};
if variable.to_string().eq_ignore_ascii_case("NAMES")
&& dialect_of!(self is MySqlDialect | GenericDialect)
{

46
tests/sqlparser_common.rs
View file

@ -4663,6 +4663,52 @@ fn parse_set_transaction() {
}
}
#[test]
fn parse_set_variable() {
match verified_stmt("SET SOMETHING = '1'") {
Statement::SetVariable {
local,
hivevar,
variable,
value,
} => {
assert!(!local);
assert!(!hivevar);
assert_eq!(variable, ObjectName(vec!["SOMETHING".into()]));
assert_eq!(
value,
vec![Expr::Value(Value::SingleQuotedString("1".into()))]
);
}
_ => unreachable!(),
}
one_statement_parses_to("SET SOMETHING TO '1'", "SET SOMETHING = '1'");
}
#[test]
fn parse_set_time_zone() {
match verified_stmt("SET TIMEZONE = 'UTC'") {
Statement::SetVariable {
local,
hivevar,
variable,
value,
} => {
assert!(!local);
assert!(!hivevar);
assert_eq!(variable, ObjectName(vec!["TIMEZONE".into()]));
assert_eq!(
value,
vec![Expr::Value(Value::SingleQuotedString("UTC".into()))]
);
}
_ => unreachable!(),
}
one_statement_parses_to("SET TIME ZONE TO 'UTC'", "SET TIMEZONE = 'UTC'");
}
#[test]
fn parse_commit() {
match verified_stmt("COMMIT") {