statsig-rust 0.19.1-beta.2604130314

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
108
109
110
111
112
113
114
115
116
117
118
119
120
121

use crate::event_emitter::{SdkEvent, SdkEventEmitter};
use std::sync::{
    atomic::{AtomicUsize, Ordering},
    Arc,
};

fn sub(event_emitter: &mut SdkEventEmitter, event_name: &str) -> (String, Arc<AtomicUsize>) {
    let counter = Arc::new(AtomicUsize::new(0));
    let counter_clone = counter.clone();
    let id: String = event_emitter.subscribe(event_name, move |_| {
        counter_clone.fetch_add(1, Ordering::SeqCst);
    });

    (id, counter)
}

fn emit(event_emitter: &mut SdkEventEmitter, event_name: &str) {
    match event_name {
        SdkEvent::RULESETS_UPDATED => {
            event_emitter.emit(SdkEvent::RulesetsUpdated {
                lcut: 0,
                raw_value: "".to_string(),
            });
        }
        SdkEvent::INIT_SUCCESS => {
            event_emitter.emit(SdkEvent::InitSuccess { duration: 1.0 });
        }
        _ => {
            panic!("Unsupported event: {event_name}");
        }
    }
}

#[test]
fn test_unsub_by_event() {
    let mut event_emitter = SdkEventEmitter::default();

    let (_, first_counter) = sub(&mut event_emitter, SdkEvent::RULESETS_UPDATED);
    let (_, second_counter) = sub(&mut event_emitter, SdkEvent::RULESETS_UPDATED);
    let (_, third_counter) = sub(&mut event_emitter, SdkEvent::INIT_SUCCESS);

    emit(&mut event_emitter, SdkEvent::RULESETS_UPDATED);

    assert_eq!(first_counter.load(Ordering::SeqCst), 1);
    assert_eq!(second_counter.load(Ordering::SeqCst), 1);
    assert_eq!(third_counter.load(Ordering::SeqCst), 0);

    emit(&mut event_emitter, SdkEvent::INIT_SUCCESS);
    assert_eq!(third_counter.load(Ordering::SeqCst), 1);

    event_emitter.unsubscribe(SdkEvent::RULESETS_UPDATED);
    emit(&mut event_emitter, SdkEvent::RULESETS_UPDATED);

    assert_eq!(first_counter.load(Ordering::SeqCst), 1);
    assert_eq!(second_counter.load(Ordering::SeqCst), 1);
    assert_eq!(third_counter.load(Ordering::SeqCst), 1);

    emit(&mut event_emitter, SdkEvent::INIT_SUCCESS);
    assert_eq!(third_counter.load(Ordering::SeqCst), 2);
}

#[test]
fn test_unsub_by_event_and_id() {
    let mut event_emitter = SdkEventEmitter::default();

    let (first_id, first_counter) = sub(&mut event_emitter, SdkEvent::RULESETS_UPDATED);
    let (_, second_counter) = sub(&mut event_emitter, SdkEvent::RULESETS_UPDATED);
    let (_, third_counter) = sub(&mut event_emitter, SdkEvent::INIT_SUCCESS);

    emit(&mut event_emitter, SdkEvent::RULESETS_UPDATED);
    emit(&mut event_emitter, SdkEvent::INIT_SUCCESS);

    assert_eq!(first_counter.load(Ordering::SeqCst), 1);
    assert_eq!(second_counter.load(Ordering::SeqCst), 1);
    assert_eq!(third_counter.load(Ordering::SeqCst), 1);

    event_emitter.unsubscribe_by_id(SdkEvent::RULESETS_UPDATED, &first_id);
    emit(&mut event_emitter, SdkEvent::RULESETS_UPDATED);
    emit(&mut event_emitter, SdkEvent::INIT_SUCCESS);

    assert_eq!(first_counter.load(Ordering::SeqCst), 1);
    assert_eq!(second_counter.load(Ordering::SeqCst), 2);
    assert_eq!(third_counter.load(Ordering::SeqCst), 2);
}

#[test]
fn test_unsub_all() {
    let mut event_emitter = SdkEventEmitter::default();

    let (_, first_counter) = sub(&mut event_emitter, SdkEvent::RULESETS_UPDATED);
    let (_, second_counter) = sub(&mut event_emitter, SdkEvent::RULESETS_UPDATED);
    let (_, third_counter) = sub(&mut event_emitter, SdkEvent::INIT_SUCCESS);

    emit(&mut event_emitter, SdkEvent::RULESETS_UPDATED);
    emit(&mut event_emitter, SdkEvent::INIT_SUCCESS);

    assert_eq!(first_counter.load(Ordering::SeqCst), 1);
    assert_eq!(second_counter.load(Ordering::SeqCst), 1);
    assert_eq!(third_counter.load(Ordering::SeqCst), 1);

    event_emitter.unsubscribe_all();

    emit(&mut event_emitter, SdkEvent::RULESETS_UPDATED);
    emit(&mut event_emitter, SdkEvent::INIT_SUCCESS);

    assert_eq!(first_counter.load(Ordering::SeqCst), 1);
    assert_eq!(second_counter.load(Ordering::SeqCst), 1);
    assert_eq!(third_counter.load(Ordering::SeqCst), 1);
}

#[test]
fn test_sub_all() {
    let mut event_emitter = SdkEventEmitter::default();
    let (_, counter) = sub(&mut event_emitter, SdkEvent::ALL);

    emit(&mut event_emitter, SdkEvent::RULESETS_UPDATED);
    assert_eq!(counter.load(Ordering::SeqCst), 1);

    emit(&mut event_emitter, SdkEvent::INIT_SUCCESS);
    assert_eq!(counter.load(Ordering::SeqCst), 2);
}