oai-statsig-rust 0.27.0

Statsig Rust SDK for usage in multi-user server environments.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

use serde::Serialize;

use crate::{
    id_lists_adapter::{IdListMetadata, IdListUpdate},
    unwrap_or_noop,
};
use std::{collections::HashSet, sync::Arc};

#[derive(Clone, Serialize)]
pub struct IdList {
    pub metadata: IdListMetadata,

    #[serde(skip_serializing)]
    pub ids: Arc<HashSet<String>>,
}

impl IdList {
    pub fn new(metadata: IdListMetadata) -> Self {
        let mut local_metadata = metadata;
        local_metadata.size = 0;

        Self {
            metadata: local_metadata,
            ids: Arc::new(HashSet::new()),
        }
    }

    pub fn apply_update(&mut self, update: IdListUpdate) {
        let updated_meta = update.new_metadata;
        let current_meta = &self.metadata;

        if updated_meta.file_id != current_meta.file_id
            && updated_meta.creation_time >= current_meta.creation_time
        {
            self.update_metadata(updated_meta);
        }

        let changeset_data = unwrap_or_noop!(&update.raw_changeset);
        let ids = Arc::make_mut(&mut self.ids);

        for change in changeset_data.lines() {
            let trimmed = change.trim();
            if trimmed.len() <= 1 {
                continue;
            }

            let op = change.chars().next();
            let id = &change[1..];

            match op {
                Some('+') => {
                    ids.insert(id.to_string());
                }
                Some('-') => {
                    ids.remove(id);
                }
                _ => continue,
            }
        }

        self.metadata.size += changeset_data.len() as u64;
    }

    fn update_metadata(&mut self, metadata: IdListMetadata) {
        self.metadata = metadata;
        self.metadata.size = 0;
        self.ids = Arc::new(HashSet::new());
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    fn metadata() -> IdListMetadata {
        IdListMetadata {
            name: "employees".to_string(),
            url: "https://example.com/employees".to_string(),
            file_id: Some("file-1".to_string()),
            size: 0,
            creation_time: 1,
        }
    }

    #[test]
    fn cloned_list_copies_ids_only_when_updated() {
        let mut original = IdList::new(metadata());
        original.apply_update(IdListUpdate {
            raw_changeset: Some("+alice".to_string()),
            new_metadata: metadata(),
        });

        let mut updated = original.clone();
        assert!(Arc::ptr_eq(&original.ids, &updated.ids));

        updated.apply_update(IdListUpdate {
            raw_changeset: Some("+bob".to_string()),
            new_metadata: metadata(),
        });

        assert!(!Arc::ptr_eq(&original.ids, &updated.ids));
        assert!(original.ids.contains("alice"));
        assert!(!original.ids.contains("bob"));
        assert!(updated.ids.contains("alice"));
        assert!(updated.ids.contains("bob"));
    }
}