statsig_rust/event_logging_adapter/
event_logging_adapter_trait.rs

1use std::sync::Arc;
2
3use crate::log_event_payload::LogEventRequest;
4use crate::{StatsigErr, StatsigRuntime};
5use async_trait::async_trait;
6
7#[async_trait]
8pub trait EventLoggingAdapter: Send + Sync {
9    async fn start(&self, statsig_runtime: &Arc<StatsigRuntime>) -> Result<(), StatsigErr>;
10    async fn log_events(&self, request: LogEventRequest) -> Result<bool, StatsigErr>;
11    async fn shutdown(&self) -> Result<(), StatsigErr>;
12    fn should_schedule_background_flush(&self) -> bool;
13}
14
15#[cfg(test)]
16mod tests {
17    use serde_json::{json, Value};
18
19    use crate::event_logging::statsig_event::StatsigEvent;
20    use crate::event_logging::statsig_event_internal::make_custom_event;
21    use crate::log_event_payload::{LogEventPayload, LogEventRequest};
22    use crate::statsig_metadata::StatsigMetadata;
23    use crate::statsig_user_internal::StatsigUserInternal;
24    use crate::StatsigUser;
25
26    #[test]
27    fn test_request_jsonify() {
28        let user = StatsigUser::with_user_id("a-user".to_string());
29        let user_internal = StatsigUserInternal::new(&user, None, None);
30        let event = StatsigEvent {
31            event_name: "my_custom_event".to_string(),
32            value: None,
33            metadata: None,
34            statsig_metadata: None,
35        };
36
37        let event = make_custom_event(user_internal, event);
38
39        let payload = LogEventPayload {
40            events: json!([event]),
41            statsig_metadata: StatsigMetadata::get_as_json(),
42        };
43
44        let request = LogEventRequest {
45            payload,
46            event_count: 1,
47        };
48
49        let serialized = serde_json::to_string(&request).unwrap();
50        println!("{serialized}");
51
52        let deserialized: Value = serde_json::from_str(&serialized).unwrap();
53
54        let event_count = deserialized.get("eventCount").unwrap().as_u64().unwrap();
55        assert_eq!(event_count, 1);
56
57        let payload_json = deserialized.get("payload").unwrap();
58
59        let statsig_metadata_json = payload_json
60            .get("statsigMetadata")
61            .unwrap()
62            .as_object()
63            .unwrap();
64
65        assert!(statsig_metadata_json.contains_key("sdkType"));
66        assert!(statsig_metadata_json.contains_key("sdkVersion"));
67        assert!(statsig_metadata_json.contains_key("sessionID"));
68
69        let events_json = payload_json.get("events").unwrap();
70        assert_eq!(events_json.as_array().unwrap().len(), 1);
71    }
72
73    #[test]
74    fn test_json_deserialization() {
75        let request_json = r#"{"payload":{"events":[{"eventName":"my_custom_event","metadata":null,"secondaryExposures":null,"time":1730831508904,"user":{"statsigEnvironment":null,"userID":"a-user"},"value":null}],"statsigMetadata":{"sdkType":"statsig-server-core","sdkVersion":"0.0.1","sessionId":"1ff863ed-a9ab-4785-bb0e-1a7b0140c040"}},"eventCount":1}"#;
76        let request: LogEventRequest = serde_json::from_str(request_json).unwrap();
77        assert_eq!(request.event_count, 1);
78    }
79}
statsig_rust/event_logging_adapter/event_logging_adapter_trait.rs

statsig_rust/event_logging_adapter/
event_logging_adapter_trait.rs
