try to fix lexical precedence

datafusion-contrib · Jun 28, 2024 · f573a39 · f573a39
1 parent b364da4
commit f573a39
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/rewrite.rs b/src/rewrite.rs
@@ -141,6 +141,10 @@ impl CustomOperator for JsonOperator {
             JsonOperator::LongArrow => "LongArrow",
         }
     }
+
+    fn precedence(&self) -> u8 {
+        255
+    }
 }
 
 impl TryFrom<&str> for JsonOperator {

diff --git a/tests/main.rs b/tests/main.rs
@@ -856,3 +856,10 @@ async fn test_arrow_nested_double_columns() {
     let batches = run_query(sql).await.unwrap();
     assert_batches_eq!(expected, &batches);
 }
+
+#[tokio::test]
+async fn test_lexical_precedence_wrong() {
+    let sql = r#"select '{"a": "b"}'->>'a'='b' as v"#;
+    let err = run_query(sql).await.unwrap_err();
+    assert_eq!(err.to_string(), "Error during planning: Unexpected argument type to 'LongArrow' at position 2, expected string or int, got Boolean.")
+}