feat(expr): support jsonb - and #- operator (#13118)

Signed-off-by: Runji Wang <[email protected]>
risingwavelabs · Oct 30, 2023 · c0ab3ca · c0ab3ca
1 parent a9d9bc9
commit c0ab3ca
Show file tree

Hide file tree

Showing 12 changed files with 444 additions and 48 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/proto/expr.proto b/proto/expr.proto
@@ -237,6 +237,13 @@ message ExprNode {
     JSONB_EXISTS_ANY = 611;
     // jsonb ?& text[]
     JSONB_EXISTS_ALL = 612;
+    // see SUBTRACT for:
+    // jsonb - text -> jsonb
+    // jsonb - text[] -> jsonb
+    // jsonb - integer -> jsonb
+    //
+    // jsonb #- text[] -> jsonb
+    JSONB_DELETE_PATH = 615;
 
     // Non-pure functions below (> 1000)
     // ------------------------

diff --git a/src/common/src/types/jsonb.rs b/src/common/src/types/jsonb.rs
@@ -293,6 +293,14 @@ impl<'a> JsonbRef<'a> {
         self.0.as_null().is_some()
     }
 
+    /// Returns true if this is a jsonb null, boolean, number or string.
+    pub fn is_scalar(&self) -> bool {
+        matches!(
+            self.0,
+            ValueRef::Null | ValueRef::Bool(_) | ValueRef::Number(_) | ValueRef::String(_)
+        )
+    }
+
     /// Returns true if this is a jsonb array.
     pub fn is_array(&self) -> bool {
         matches!(self.0, ValueRef::Array(_))