Revision control

Copy as Markdown

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at https://mozilla.org/MPL/2.0/. */
use crate::{enrollment::ExperimentEnrollment, EnrolledFeature, EnrollmentStatus};
use serde_derive::{Deserialize, Serialize};
pub trait MetricsHandler: Send + Sync {
fn record_enrollment_statuses(&self, enrollment_status_extras: Vec<EnrollmentStatusExtraDef>);
#[cfg(feature = "stateful")]
fn record_feature_activation(&self, event: FeatureExposureExtraDef);
#[cfg(feature = "stateful")]
fn record_feature_exposure(&self, event: FeatureExposureExtraDef);
#[cfg(feature = "stateful")]
fn record_malformed_feature_config(&self, event: MalformedFeatureConfigExtraDef);
}
#[derive(Serialize, Deserialize, Clone)]
pub struct EnrollmentStatusExtraDef {
pub branch: Option<String>,
pub conflict_slug: Option<String>,
pub error_string: Option<String>,
pub reason: Option<String>,
pub slug: Option<String>,
pub status: Option<String>,
#[cfg(not(feature = "stateful"))]
pub user_id: Option<String>,
}
#[cfg(test)]
impl EnrollmentStatusExtraDef {
pub fn branch(&self) -> &str {
self.branch.as_ref().unwrap()
}
pub fn conflict_slug(&self) -> &str {
self.conflict_slug.as_ref().unwrap()
}
pub fn error_string(&self) -> &str {
self.error_string.as_ref().unwrap()
}
pub fn reason(&self) -> &str {
self.reason.as_ref().unwrap()
}
pub fn slug(&self) -> &str {
self.slug.as_ref().unwrap()
}
pub fn status(&self) -> &str {
self.status.as_ref().unwrap()
}
#[cfg(not(feature = "stateful"))]
pub fn user_id(&self) -> &str {
self.user_id.as_ref().unwrap()
}
}
impl From<ExperimentEnrollment> for EnrollmentStatusExtraDef {
fn from(enrollment: ExperimentEnrollment) -> Self {
let mut branch_value: Option<String> = None;
let mut reason_value: Option<String> = None;
let mut error_value: Option<String> = None;
match &enrollment.status {
EnrollmentStatus::Enrolled { reason, branch, .. } => {
branch_value = Some(branch.to_owned());
reason_value = Some(reason.to_string());
}
EnrollmentStatus::Disqualified { reason, branch, .. } => {
branch_value = Some(branch.to_owned());
reason_value = Some(reason.to_string());
}
EnrollmentStatus::NotEnrolled { reason } => {
reason_value = Some(reason.to_string());
}
EnrollmentStatus::WasEnrolled { branch, .. } => branch_value = Some(branch.to_owned()),
EnrollmentStatus::Error { reason } => {
error_value = Some(reason.to_owned());
}
}
EnrollmentStatusExtraDef {
branch: branch_value,
conflict_slug: None,
error_string: error_value,
reason: reason_value,
slug: Some(enrollment.slug),
status: Some(enrollment.status.name()),
#[cfg(not(feature = "stateful"))]
user_id: None,
}
}
}
#[derive(Clone)]
pub struct FeatureExposureExtraDef {
pub branch: Option<String>,
pub slug: String,
pub feature_id: String,
}
impl From<EnrolledFeature> for FeatureExposureExtraDef {
fn from(value: EnrolledFeature) -> Self {
Self {
feature_id: value.feature_id,
branch: value.branch,
slug: value.slug,
}
}
}
#[derive(Clone, Debug, Default, PartialEq)]
pub struct MalformedFeatureConfigExtraDef {
pub slug: Option<String>,
pub branch: Option<String>,
pub feature_id: String,
pub part: String,
}
#[cfg(feature = "stateful")]
impl MalformedFeatureConfigExtraDef {
pub(crate) fn from(value: EnrolledFeature, part: String) -> Self {
Self {
slug: Some(value.slug),
branch: value.branch,
feature_id: value.feature_id,
part,
}
}
pub(crate) fn new(feature_id: String, part: String) -> Self {
Self {
feature_id,
part,
..Default::default()
}
}
}