fix: parse json path with double quote

Author: hantmac

src/common/jsonb/src/functions.rs

@@ -31,7 +31,7 @@ use super::value::Value;
 /// Build `JSONB` array from items.
 /// Assuming that the input values is valid JSONB data.
 pub fn build_array<'a>(
-    items: impl IntoIterator<Item = &'a [u8]>,
+    items: impl IntoIterator<Item=&'a [u8]>,
     buf: &mut Vec<u8>,
 ) -> Result<(), Error> {
     let start = buf.len();
@@ -69,7 +69,7 @@ pub fn build_array<'a>(
 /// Build `JSONB` object from items.
 /// Assuming that the input values is valid JSONB data.
 pub fn build_object<'a, K: AsRef<str>>(
-    items: impl IntoIterator<Item = (K, &'a [u8])>,
+    items: impl IntoIterator<Item=(K, &'a [u8])>,
     buf: &mut Vec<u8>,
 ) -> Result<(), Error> {
     let start = buf.len();
@@ -836,6 +836,25 @@ pub fn parse_json_path(path: &[u8]) -> Result<Vec<JsonPath>, Error> {
                     return Err(Error::InvalidToken);
                 }
             }
+        } else if c == b'"' {
+            prev_idx = idx;
+            loop {
+                let c = read_char(path, &mut idx)?;
+                if c == b'\\' {
+                    idx += 1;
+                    let c = read_char(path, &mut idx)?;
+                    if c == b'"' {
+                        idx += 1;
+                    }
+                } else if c != b'"' {
+                    idx += 1;
+                } else {
+                    break;
+                }
+            }
+            let s = std::str::from_utf8(&path[prev_idx..idx-1])?;
+            let json_path = JsonPath::String(Cow::Borrowed(s));
+            json_paths.push(json_path);
         } else {
             if c == b':' || c == b'.' {
                 if idx == 1 {

src/common/jsonb/tests/it/functions.rs

@@ -361,6 +361,10 @@ fn test_parse_json_path() {
             JsonPath::String(Cow::from("k2")),
             JsonPath::String(Cow::from("k3")),
         ]),
+        ("\"k1\"", vec![JsonPath::String(Cow::from("k1")),
+        ]),
+        ("\"k1k2\"", vec![JsonPath::String(Cow::from("k1k2")),
+        ]),
         (r#"k1["k2"][1]"#, vec![
             JsonPath::String(Cow::from("k1")),
             JsonPath::String(Cow::from("k2")),
@@ -413,7 +417,7 @@ fn test_as_type() {
 
     let mut buf: Vec<u8> = Vec::new();
     for (s, expect_null, expect_bool, expect_number, expect_str, expect_array, expect_object) in
-        sources
+    sources
     {
         let res = as_null(s.as_bytes());
         match expect_null {