statsig_rust/event_logging/event_queue/
queued_experiment_expo.rs

1use crate::{
2    evaluation::evaluation_types::ExtraExposureInfo,
3    event_logging::{
4        event_logger::ExposureTrigger,
5        exposure_sampling::{EvtSamplingDecision, ExposureSamplingKey},
6        exposure_utils::{get_metadata_with_details, get_statsig_metadata_with_sampling_decision},
7        statsig_event::StatsigEvent,
8        statsig_event_internal::{StatsigEventInternal, CONFIG_EXPOSURE_EVENT_NAME},
9    },
10    interned_string::InternedString,
11    statsig_types::Experiment,
12    user::{StatsigUserInternal, StatsigUserLoggable},
13    EvaluationDetails, SecondaryExposure,
14};
15
16use super::queued_event::{EnqueueOperation, QueuedEvent, QueuedExposure};
17use crate::event_logging::statsig_event::string_metadata_to_value_metadata;
18
19pub struct EnqueueExperimentExpoOp<'a> {
20    pub user: &'a StatsigUserInternal<'a, 'a>,
21    pub experiment: &'a Experiment,
22    pub trigger: ExposureTrigger,
23    pub exposure_time: u64,
24}
25
26impl EnqueueOperation for EnqueueExperimentExpoOp<'_> {
27    fn as_exposure(&self) -> Option<&impl QueuedExposure<'_>> {
28        Some(self)
29    }
30
31    fn into_queued_event(self, sampling_decision: EvtSamplingDecision) -> QueuedEvent {
32        let evaluation = self.experiment.__evaluation.as_ref();
33        let secondary_exposures = evaluation.map(|eval| &eval.base.secondary_exposures);
34        let exposure_info = evaluation.and_then(|eval| eval.base.exposure_info.as_ref());
35        let (version, override_config_name) = exposure_info
36            .map(|info| (info.version, info.override_config_name.clone()))
37            .unwrap_or_default();
38
39        QueuedEvent::ExperimentExposure(QueuedExperimentExposureEvent {
40            user: self.user.to_loggable(),
41            experiment_name: self.experiment.name.clone(),
42            rule_id: self.experiment.rule_id.clone(),
43            secondary_exposures: secondary_exposures.cloned(),
44            evaluation_details: self.experiment.details.clone(),
45            version,
46            exposure_trigger: self.trigger,
47            sampling_decision,
48            override_config_name,
49            exposure_time: self.exposure_time,
50        })
51    }
52}
53
54impl<'a> QueuedExposure<'a> for EnqueueExperimentExpoOp<'a> {
55    fn create_exposure_sampling_key(&self) -> ExposureSamplingKey {
56        let user_data = &self.user.user_ref.data;
57        let evaluation = self.experiment.__evaluation.as_ref().map(|e| &e.base);
58
59        let in_experiment = self
60            .experiment
61            .__evaluation
62            .as_ref()
63            .is_some_and(|e| e.is_user_in_experiment == Some(true));
64
65        ExposureSamplingKey::new(evaluation, user_data.as_ref(), in_experiment as u64)
66    }
67
68    fn get_rule_id_ref(&self) -> &'a str {
69        &self.experiment.rule_id
70    }
71
72    fn get_extra_exposure_info_ref(&self) -> Option<&'a ExtraExposureInfo> {
73        self.experiment
74            .__evaluation
75            .as_ref()?
76            .base
77            .exposure_info
78            .as_ref()
79    }
80}
81
82pub struct QueuedExperimentExposureEvent {
83    pub user: StatsigUserLoggable,
84    pub experiment_name: String,
85    pub rule_id: String,
86    pub secondary_exposures: Option<Vec<SecondaryExposure>>,
87    pub evaluation_details: EvaluationDetails,
88    pub version: Option<u32>,
89    pub exposure_trigger: ExposureTrigger,
90    pub sampling_decision: EvtSamplingDecision,
91    pub override_config_name: Option<InternedString>,
92    pub exposure_time: u64,
93}
94
95impl QueuedExperimentExposureEvent {
96    pub fn into_statsig_event_internal(self) -> StatsigEventInternal {
97        let mut metadata = get_metadata_with_details(self.evaluation_details);
98        metadata.insert("config".into(), self.experiment_name);
99        metadata.insert("ruleID".into(), self.rule_id);
100
101        if let Some(version) = self.version {
102            metadata.insert("configVersion".into(), version.to_string());
103        }
104
105        if self.exposure_trigger == ExposureTrigger::Manual {
106            metadata.insert("isManualExposure".into(), "true".into());
107        }
108
109        if let Some(override_config_name) = self.override_config_name {
110            metadata.insert(
111                "overrideConfigName".into(),
112                override_config_name.unperformant_to_string(),
113            );
114        }
115
116        let statsig_metadata = get_statsig_metadata_with_sampling_decision(self.sampling_decision);
117
118        let event = StatsigEvent {
119            event_name: CONFIG_EXPOSURE_EVENT_NAME.into(),
120            value: None,
121            metadata: Some(string_metadata_to_value_metadata(metadata)),
122            statsig_metadata: Some(statsig_metadata),
123        };
124
125        StatsigEventInternal::new(
126            self.exposure_time,
127            self.user,
128            event,
129            Some(self.secondary_exposures.unwrap_or_default()),
130        )
131    }
132}
statsig_rust/event_logging/event_queue/queued_experiment_expo.rs

statsig_rust/event_logging/event_queue/
queued_experiment_expo.rs
