statsig_rust/evaluation/
dynamic_returnable.rs

1use std::{borrow::Cow, collections::HashMap, sync::Arc};
2
3use serde::{Deserialize, Deserializer, Serialize, Serializer};
4use serde_json::{
5    value::{to_raw_value, RawValue},
6    Value as JsonValue,
7};
8
9use crate::{impl_interned_value, interned_value_store::FromRawValue, log_e};
10
11const TAG: &str = "DynamicReturnable";
12
13lazy_static::lazy_static! {
14    static ref EMPTY_DYNAMIC_RETURNABLE: DynamicReturnable = DynamicReturnable {
15        hash: 0,
16        value: Arc::new(MemoizedValue {
17            raw_value: RawValue::NULL.to_owned(),
18        }),
19    };
20
21}
22
23#[derive(Clone, PartialEq, Debug)]
24pub struct DynamicReturnable {
25    hash: u64,
26    value: Arc<MemoizedValue>,
27}
28
29impl_interned_value!(DynamicReturnable, MemoizedValue);
30
31impl DynamicReturnable {
32    pub fn empty() -> Self {
33        EMPTY_DYNAMIC_RETURNABLE.clone()
34    }
35
36    pub fn from_map(value: HashMap<String, JsonValue>) -> Self {
37        let raw_value = match to_raw_value(&value) {
38            Ok(raw_value) => raw_value,
39            Err(e) => {
40                log_e!(TAG, "Failed to convert map to raw value: {}", e);
41                return Self::empty();
42            }
43        };
44
45        let (hash, value) = DynamicReturnable::get_or_create_memoized(Cow::Owned(raw_value));
46        Self { hash, value }
47    }
48
49    pub fn get_bool(&self) -> Option<bool> {
50        match self.value.raw_value.get() {
51            "true" => Some(true),
52            "false" => Some(false),
53            _ => None,
54        }
55    }
56
57    pub fn get_json(&self) -> Option<HashMap<String, JsonValue>> {
58        let raw_json = self.value.raw_value.get();
59
60        match raw_json {
61            "null" | "true" | "false" => None,
62            _ => match serde_json::from_str(raw_json) {
63                Ok(json) => Some(json),
64                Err(e) => {
65                    log_e!(TAG, "Failed to parse json: {}. Error: {}", raw_json, e);
66                    None
67                }
68            },
69        }
70    }
71}
72
73impl<'de> Deserialize<'de> for DynamicReturnable {
74    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
75    where
76        D: Deserializer<'de>,
77    {
78        let raw_value_ref: Box<RawValue> = Deserialize::deserialize(deserializer)?;
79        let (hash, value) = DynamicReturnable::get_or_create_memoized(Cow::Owned(raw_value_ref));
80        Ok(DynamicReturnable { hash, value })
81    }
82}
83
84impl Serialize for DynamicReturnable {
85    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
86    where
87        S: Serializer,
88    {
89        self.value.raw_value.serialize(serializer)
90    }
91}
92
93#[derive(Debug, Clone)]
94pub(crate) struct MemoizedValue {
95    pub(crate) raw_value: Box<RawValue>,
96}
97
98impl FromRawValue for MemoizedValue {
99    fn from_raw_value(raw_value: Cow<'_, RawValue>) -> Self {
100        Self {
101            raw_value: raw_value.into_owned(),
102        }
103    }
104}
105
106impl PartialEq for MemoizedValue {
107    fn eq(&self, other: &Self) -> bool {
108        self.raw_value.get() == other.raw_value.get()
109    }
110}
statsig_rust/evaluation/dynamic_returnable.rs

statsig_rust/evaluation/
dynamic_returnable.rs
