Add as_arc_str() and as_arc_slice() methods on FieldValue. (#414)

obi1kenobi · Jul 29, 2023 · c38126b · c38126b
1 parent 9cc00b1
commit c38126b
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/trustfall_core/src/ir/value.rs b/trustfall_core/src/ir/value.rs
@@ -176,6 +176,13 @@ impl FieldValue {
         }
     }
 
+    pub fn as_arc_str(&self) -> Option<&Arc<str>> {
+        match self {
+            FieldValue::String(s) => Some(s),
+            _ => None,
+        }
+    }
+
     pub fn as_bool(&self) -> Option<bool> {
         match self {
             FieldValue::Boolean(b) => Some(*b),
@@ -190,6 +197,13 @@ impl FieldValue {
         }
     }
 
+    pub fn as_arc_slice(&self) -> Option<&Arc<[FieldValue]>> {
+        match self {
+            FieldValue::List(l) => Some(l),
+            _ => None,
+        }
+    }
+
     pub fn as_vec_with<'a, T>(
         &'a self,
         inner: impl Fn(&'a FieldValue) -> Option<T>,