optimizely 0.5.0

An unofficial Rust SDK for Optimizely Feature Experimentation
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

// External imports
use serde::{Deserialize, Deserializer};
use std::collections::HashMap;

// Imports from super
use super::{TrafficAllocation, Variation, VariationMap};

#[derive(Deserialize, Debug)]
#[serde(rename_all = "camelCase")]
pub struct Experiment {
    #[serde()]
    id: String,
    #[serde(rename = "layerId")]
    campaign_id: String,
    traffic_allocation: TrafficAllocation,
    audience_ids: Vec<String>, // TODO: use audienceConditions instead of audienceIds
    variations: VariationMap,
}

impl Experiment {
    pub fn id(&self) -> &str {
        &self.id
    }

    pub fn campaign_id(&self) -> &str {
        &self.campaign_id
    }

    pub fn traffic_allocation(&self) -> &TrafficAllocation {
        &self.traffic_allocation
    }

    pub fn variation(&self, variation_id: &str) -> Option<&Variation> {
        self.variations.get(variation_id)
    }

    pub fn audience_ids(&self) -> &[String] {
        &self.audience_ids
    }
}

#[derive(Debug)]
pub struct ExperimentMap(HashMap<String, Experiment>);

impl<'de> Deserialize<'de> for ExperimentMap {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        let mut map = HashMap::new();
        for experiment in Vec::<Experiment>::deserialize(deserializer)? {
            map.insert(experiment.id.clone(), experiment);
        }

        Ok(Self(map))
    }
}

impl ExperimentMap {
    pub fn get(&self, id: &str) -> Option<&Experiment> {
        self.0.get(id)
    }
}