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

feat: Support expression in SET statement (#574)

Browse source 
Co-authored-by: Alex Vasilev <vaspiring@gmail.com>
This commit is contained in:
Dmitry Patsura 2022-08-18 20:29:55 +03:00 • committed by GitHub
parent eb7f1b005e
commit 6d8aacd85b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 63 additions and 52 deletions
Download patch file Download diff file Expand all files Collapse all files

34
tests/sqlparser_postgres.rs
View file

@ -18,7 +18,6 @@
mod test_utils;
use test_utils::*;
use sqlparser::ast::Value::Boolean;
use sqlparser::ast::*;
use sqlparser::dialect::{GenericDialect, PostgreSqlDialect};
use sqlparser::parser::ParserError;
@ -782,7 +781,10 @@ fn parse_set() {
local: false,
hivevar: false,
variable: ObjectName(vec![Ident::new("a")]),
value: vec![SetVariableValue::Ident("b".into())],
value: vec![Expr::Identifier(Ident {
value: "b".into(),
quote_style: None
})],
}
);
@ -793,9 +795,7 @@ fn parse_set() {
local: false,
hivevar: false,
variable: ObjectName(vec![Ident::new("a")]),
value: vec![SetVariableValue::Literal(Value::SingleQuotedString(
"b".into()
))],
value: vec![Expr::Value(Value::SingleQuotedString("b".into()))],
}
);
@ -806,7 +806,13 @@ fn parse_set() {
local: false,
hivevar: false,
variable: ObjectName(vec![Ident::new("a")]),
value: vec![SetVariableValue::Literal(number("0"))],
value: vec![Expr::Value(Value::Number(
#[cfg(not(feature = "bigdecimal"))]
"0".to_string(),
#[cfg(feature = "bigdecimal")]
bigdecimal::BigDecimal::from(0),
false,
))],
}
);
@ -817,7 +823,10 @@ fn parse_set() {
local: false,
hivevar: false,
variable: ObjectName(vec![Ident::new("a")]),
value: vec![SetVariableValue::Ident("DEFAULT".into())],
value: vec![Expr::Identifier(Ident {
value: "DEFAULT".into(),
quote_style: None
})],
}
);
@ -828,7 +837,7 @@ fn parse_set() {
local: true,
hivevar: false,
variable: ObjectName(vec![Ident::new("a")]),
value: vec![SetVariableValue::Ident("b".into())],
value: vec![Expr::Identifier("b".into())],
}
);
@ -839,7 +848,10 @@ fn parse_set() {
local: false,
hivevar: false,
variable: ObjectName(vec![Ident::new("a"), Ident::new("b"), Ident::new("c")]),
value: vec![SetVariableValue::Ident("b".into())],
value: vec![Expr::Identifier(Ident {
value: "b".into(),
quote_style: None
})],
}
);
@ -859,7 +871,7 @@ fn parse_set() {
Ident::new("reducer"),
Ident::new("parallelism")
]),
value: vec![SetVariableValue::Literal(Boolean(false))],
value: vec![Expr::Value(Value::Boolean(false))],
}
);
@ -1107,7 +1119,7 @@ fn parse_pg_unary_ops() {
];
for (str_op, op) in pg_unary_ops {
let select = pg().verified_only_select(&format!("SELECT {} a", &str_op));
let select = pg().verified_only_select(&format!("SELECT {}a", &str_op));
assert_eq!(
SelectItem::UnnamedExpr(Expr::UnaryOp {
op: op.clone(),