Add support for PartiQL queries in Redshift

Author: yoavcloud

src/parser/mod.rs

@@ -3087,7 +3087,7 @@ impl<'a> Parser<'a> {
         }
 
         debug_assert!(!path.is_empty());
-        Ok(JsonPath{ path })
+        Ok(JsonPath { path })
     }
 
     pub fn parse_map_access(&mut self, expr: Expr) -> Result<Expr, ParserError> {
@@ -10311,12 +10311,12 @@ impl<'a> Parser<'a> {
             self.parse_open_json_table_factor()
         } else {
             let name = self.parse_object_name(true)?;
-            
+
             let partiql = match self.peek_token().token {
                 Token::LBracket if self.dialect.supports_partiql() => Some(self.parse_json_path()?),
-                _ => None
+                _ => None,
             };
-        
+
             let partitions: Vec<Ident> = if dialect_of!(self is MySqlDialect | GenericDialect)
                 && self.parse_keyword(Keyword::PARTITION)
             {

tests/sqlparser_redshift.rs

@@ -206,15 +206,20 @@ fn test_redshift_json_path() {
     let select = redshift().verified_only_select(sql);
 
     assert_eq!(
-        &Expr::JsonAccess{
+        &Expr::JsonAccess {
             value: Box::new(Expr::CompoundIdentifier(vec![
                 Ident::new("cust"),
                 Ident::new("c_orders")
             ])),
-            path: JsonPath{
+            path: JsonPath {
                 path: vec![
-                    JsonPathElem::Bracket{key: Expr::Value(Value::Number("0".to_string(), false))},
-                    JsonPathElem::Dot{key: "o_orderkey".to_string(), quoted: false}
+                    JsonPathElem::Bracket {
+                        key: Expr::Value(Value::Number("0".parse().unwrap(), false))
+                    },
+                    JsonPathElem::Dot {
+                        key: "o_orderkey".to_string(),
+                        quoted: false
+                    }
                 ]
             }
         },
@@ -224,15 +229,19 @@ fn test_redshift_json_path() {
     let sql = "SELECT cust.c_orders[0]['id'] FROM customer_orders_lineitem";
     let select = redshift().verified_only_select(sql);
     assert_eq!(
-        &Expr::JsonAccess{
+        &Expr::JsonAccess {
             value: Box::new(Expr::CompoundIdentifier(vec![
                 Ident::new("cust"),
                 Ident::new("c_orders")
             ])),
-            path: JsonPath{
+            path: JsonPath {
                 path: vec![
-                    JsonPathElem::Bracket{key: Expr::Value(Value::Number("0".to_string(), false))},
-                    JsonPathElem::Bracket{key: Expr::Value(Value::SingleQuotedString("id".to_owned()))}
+                    JsonPathElem::Bracket {
+                        key: Expr::Value(Value::Number("0".parse().unwrap(), false))
+                    },
+                    JsonPathElem::Bracket {
+                        key: Expr::Value(Value::SingleQuotedString("id".to_owned()))
+                    }
                 ]
             }
         },
@@ -248,10 +257,15 @@ fn test_parse_json_path_from() {
             assert_eq!(name, &ObjectName(vec![Ident::new("src")]));
             assert_eq!(
                 partiql,
-                &Some(JsonPath{
+                &Some(JsonPath {
                     path: vec![
-                        JsonPathElem::Bracket{key: Expr::Value(Value::Number("0".to_string(), false))},
-                        JsonPathElem::Dot{key: "a".to_string(), quoted: false}
+                        JsonPathElem::Bracket {
+                            key: Expr::Value(Value::Number("0".parse().unwrap(), false))
+                        },
+                        JsonPathElem::Dot {
+                            key: "a".to_string(),
+                            quoted: false
+                        }
                     ]
                 })
             );
@@ -265,12 +279,22 @@ fn test_parse_json_path_from() {
             assert_eq!(name, &ObjectName(vec![Ident::new("src")]));
             assert_eq!(
                 partiql,
-                &Some(JsonPath{
+                &Some(JsonPath {
                     path: vec![
-                        JsonPathElem::Bracket{key: Expr::Value(Value::Number("0".to_string(), false))},
-                        JsonPathElem::Dot{key: "a".to_string(), quoted: false},
-                        JsonPathElem::Bracket{key: Expr::Value(Value::Number("1".to_string(), false))},
-                        JsonPathElem::Dot{key: "b".to_string(), quoted: false},
+                        JsonPathElem::Bracket {
+                            key: Expr::Value(Value::Number("0".parse().unwrap(), false))
+                        },
+                        JsonPathElem::Dot {
+                            key: "a".to_string(),
+                            quoted: false
+                        },
+                        JsonPathElem::Bracket {
+                            key: Expr::Value(Value::Number("1".parse().unwrap(), false))
+                        },
+                        JsonPathElem::Dot {
+                            key: "b".to_string(),
+                            quoted: false
+                        },
                     ]
                 })
             );
@@ -281,10 +305,12 @@ fn test_parse_json_path_from() {
     let select = redshift().verified_only_select("SELECT * FROM src.a.b");
     match &select.from[0].relation {
         TableFactor::Table { name, partiql, .. } => {
-            assert_eq!(name, &ObjectName(vec![Ident::new("src"), Ident::new("a"), Ident::new("b")]));
+            assert_eq!(
+                name,
+                &ObjectName(vec![Ident::new("src"), Ident::new("a"), Ident::new("b")])
+            );
             assert_eq!(partiql, &None);
         }
         _ => panic!(),
     }
-
-}
+}