statsig-python-core 0.11.0__tar.gz → 0.11.1__tar.gz

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/Cargo.lock

@@ -2374,7 +2374,7 @@ dependencies = [
 
 [[package]]
 name = "sigstat-grpc"
-version = "0.11.0"
+version = "0.11.1"
 dependencies = [
  "async-trait",
  "chrono",
@@ -2438,7 +2438,7 @@ checksum = "a8f112729512f8e442d81f95a8a7ddf2b7c6b8a1a6f509a95864142b30cab2d3"
 
 [[package]]
 name = "statsig-node"
-version = "0.11.0"
+version = "0.11.1"
 dependencies = [
  "async-trait",
  "napi",
@@ -2452,7 +2452,7 @@ dependencies = [
 
 [[package]]
 name = "statsig-pyo3"
-version = "0.11.0"
+version = "0.11.1"
 dependencies = [
  "async-trait",
  "lazy_static",
@@ -2466,7 +2466,7 @@ dependencies = [
 
 [[package]]
 name = "statsig-rust"
-version = "0.11.0"
+version = "0.11.1"
 dependencies = [
  "ahash",
  "arc-swap",
@@ -2507,7 +2507,7 @@ dependencies = [
 
 [[package]]
 name = "statsig_elixir"
-version = "0.11.0"
+version = "0.11.1"
 dependencies = [
  "parking_lot",
  "rustler",
@@ -2519,7 +2519,7 @@ dependencies = [
 
 [[package]]
 name = "statsig_ffi"
-version = "0.11.0"
+version = "0.11.1"
 dependencies = [
  "async-trait",
  "cbindgen",

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/Cargo.toml

@@ -18,7 +18,7 @@ exclude = [
 [workspace.package]
 edition = "2021"
 license = "ISC"
-version = "0.11.0"
+version = "0.11.1"
 homepage = "https://statsig.com/"
 authors = ["Statsig", "Daniel Loomb <daniel@statsig.com>"]
 readme = "README.md"

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: statsig_python_core
-Version: 0.11.0
+Version: 0.11.1
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/statsig-rust/Cargo.toml

@@ -38,7 +38,7 @@ serde_json = { version = "1.0.143", features = [
 ] }
 serde_with = "3.4.0"
 sha2 = "0.10.8"
-sigstat-grpc = { path = "../statsig-grpc", version = "0.11.0", optional = true }
+sigstat-grpc = { path = "../statsig-grpc", version = "0.11.1", optional = true }
 simple_logger = { version = "5.0.0" }
 tempfile = "3.8.1"
 tokio = { version = "1.39.1", features = ["full"] }

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/statsig-rust/src/__tests__/spec_store_tests.rs

@@ -6,6 +6,7 @@ use parking_lot::Mutex;
 use crate::{
     data_store_interface::{DataStoreResponse, DataStoreTrait, RequestPath},
     networking::ResponseData,
+    sdk_event_emitter::SdkEventEmitter,
     SpecStore, SpecsSource, SpecsUpdate, StatsigErr, StatsigRuntime,
 };
 
@@ -65,6 +66,7 @@ async fn test_spec_store_data_store_updates_forwarded_to_data_store() {
         "test",
         "test".to_string(),
         StatsigRuntime::get_runtime(),
+        Arc::new(SdkEventEmitter::default()),
         Some(data_store.clone()),
     );
 
@@ -97,6 +99,7 @@ fn test_failure_to_update() {
         "test",
         "test".to_string(),
         StatsigRuntime::get_runtime(),
+        Arc::new(SdkEventEmitter::default()),
         None,
     );
 
@@ -116,6 +119,7 @@ fn test_no_updates() {
         "test",
         "test".to_string(),
         StatsigRuntime::get_runtime(),
+        Arc::new(SdkEventEmitter::default()),
         None,
     );
 

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/statsig-rust/src/sdk_event_emitter/event_types.rs

@@ -1,4 +1,8 @@
-use crate::statsig_types::{DynamicConfig, Experiment, Layer};
+use crate::{
+    specs_response::spec_types::SpecsResponseFull,
+    statsig_types::{DynamicConfig, Experiment, Layer},
+    SpecsSource,
+};
 use serde::Serialize;
 use serde_json::json;
 use std::borrow::Cow;
@@ -12,6 +16,7 @@ pub enum SdkEventCode {
     DynamicConfigEvaluated = 3,
     ExperimentEvaluated = 4,
     LayerEvaluated = 5,
+    SpecsUpdated = 6,
 }
 
 impl SdkEventCode {
@@ -22,6 +27,7 @@ impl SdkEventCode {
             SdkEvent::DYNAMIC_CONFIG_EVALUATED => SdkEventCode::DynamicConfigEvaluated,
             SdkEvent::EXPERIMENT_EVALUATED => SdkEventCode::ExperimentEvaluated,
             SdkEvent::LAYER_EVALUATED => SdkEventCode::LayerEvaluated,
+            SdkEvent::SPECS_UPDATED => SdkEventCode::SpecsUpdated,
             _ => SdkEventCode::Unknown,
         }
     }
@@ -33,6 +39,11 @@ impl SdkEventCode {
 
 #[derive(Serialize, Clone)]
 pub enum SdkEvent<'a> {
+    SpecsUpdated {
+        source: &'a SpecsSource,
+        source_api: &'a Option<String>,
+        values: &'a SpecsResponseFull,
+    },
     GateEvaluated {
         gate_name: Cow<'a, str>,
         rule_id: Cow<'a, str>,
@@ -56,6 +67,7 @@ impl SdkEvent<'_> {
     pub const DYNAMIC_CONFIG_EVALUATED: &'static str = "dynamic_config_evaluated";
     pub const EXPERIMENT_EVALUATED: &'static str = "experiment_evaluated";
     pub const LAYER_EVALUATED: &'static str = "layer_evaluated";
+    pub const SPECS_UPDATED: &'static str = "specs_updated";
 
     pub fn get_code(&self) -> SdkEventCode {
         match self {
@@ -63,6 +75,7 @@ impl SdkEvent<'_> {
             SdkEvent::DynamicConfigEvaluated { .. } => SdkEventCode::DynamicConfigEvaluated,
             SdkEvent::ExperimentEvaluated { .. } => SdkEventCode::ExperimentEvaluated,
             SdkEvent::LayerEvaluated { .. } => SdkEventCode::LayerEvaluated,
+            SdkEvent::SpecsUpdated { .. } => SdkEventCode::SpecsUpdated,
         }
     }
 
@@ -73,6 +86,7 @@ impl SdkEvent<'_> {
             SdkEventCode::DynamicConfigEvaluated => SdkEvent::DYNAMIC_CONFIG_EVALUATED,
             SdkEventCode::ExperimentEvaluated => SdkEvent::EXPERIMENT_EVALUATED,
             SdkEventCode::LayerEvaluated => SdkEvent::LAYER_EVALUATED,
+            SdkEventCode::SpecsUpdated => SdkEvent::SPECS_UPDATED,
             SdkEventCode::Unknown => "unknown",
         }
     }
@@ -103,6 +117,15 @@ impl SdkEvent<'_> {
             SdkEvent::LayerEvaluated { layer } => {
                 map.insert("layer".to_string(), json!(layer));
             }
+            SdkEvent::SpecsUpdated {
+                source,
+                source_api,
+                values,
+            } => {
+                map.insert("source".to_string(), json!(source));
+                map.insert("source_api".to_string(), json!(source_api));
+                map.insert("values".to_string(), json!(values));
+            }
         }
 
         map

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/statsig-rust/src/spec_store.rs

@@ -5,6 +5,7 @@ use crate::networking::ResponseData;
 use crate::observability::observability_client_adapter::{MetricType, ObservabilityEvent};
 use crate::observability::ops_stats::{OpsStatsForInstance, OPS_STATS};
 use crate::observability::sdk_errors_observer::ErrorBoundaryEvent;
+use crate::sdk_event_emitter::{SdkEvent, SdkEventEmitter};
 use crate::specs_response::spec_types::{SpecsResponseFull, SpecsResponseNoUpdates};
 use crate::utils::maybe_trim_malloc;
 use crate::{
@@ -36,6 +37,7 @@ pub struct SpecStore {
     statsig_runtime: Arc<StatsigRuntime>,
     ops_stats: Arc<OpsStatsForInstance>,
     global_configs: Arc<GlobalConfigs>,
+    event_emitter: Arc<SdkEventEmitter>,
 }
 
 impl SpecStore {
@@ -44,6 +46,7 @@ impl SpecStore {
         sdk_key: &str,
         hashed_sdk_key: String,
         statsig_runtime: Arc<StatsigRuntime>,
+        event_emitter: Arc<SdkEventEmitter>,
         data_store: Option<Arc<dyn DataStoreTrait>>,
     ) -> SpecStore {
         SpecStore {
@@ -56,6 +59,7 @@ impl SpecStore {
                 source_api: None,
                 id_lists: HashMap::new(),
             })),
+            event_emitter,
             data_store,
             statsig_runtime,
             ops_stats: OPS_STATS.get_for_instance(sdk_key),
@@ -189,6 +193,9 @@ impl SpecStore {
 
                 data.time_received_at = Some(now);
                 data.source_api = source_api;
+
+                self.emit_specs_updated_sdk_event(&data.source, &data.source_api, &data.values);
+
                 Ok((prev_source, prev_lcut, data.values.time))
             }
             None => {
@@ -200,6 +207,19 @@ impl SpecStore {
         }
     }
 
+    fn emit_specs_updated_sdk_event(
+        &self,
+        source: &SpecsSource,
+        source_api: &Option<String>,
+        values: &SpecsResponseFull,
+    ) {
+        self.event_emitter.emit(SdkEvent::SpecsUpdated {
+            source,
+            source_api,
+            values,
+        });
+    }
+
     fn ops_stats_log_no_update(&self, source: SpecsSource, source_api: Option<String>) {
         log_d!(TAG, "No Updates");
         self.ops_stats.log(ObservabilityEvent::new_event(

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/statsig-rust/src/statsig.rs

@@ -83,7 +83,7 @@ lazy_static::lazy_static! {
 pub struct Statsig {
     pub statsig_runtime: Arc<StatsigRuntime>,
     pub options: Arc<StatsigOptions>,
-    pub event_emitter: SdkEventEmitter,
+    pub event_emitter: Arc<SdkEventEmitter>,
 
     sdk_key: String,
     event_logger: Arc<EventLogger>,
@@ -167,10 +167,13 @@ impl Statsig {
             &options.observability_client,
         );
 
+        let event_emitter = Arc::new(SdkEventEmitter::default());
+
         let spec_store = Arc::new(SpecStore::new(
             sdk_key,
             hashing.sha256(sdk_key),
             statsig_runtime.clone(),
+            event_emitter.clone(),
             options.data_store.clone(),
         ));
 
@@ -206,7 +209,7 @@ impl Statsig {
             background_tasks_started: Arc::new(AtomicBool::new(false)),
             persistent_values_manager,
             initialize_details: Mutex::new(InitializeDetails::default()),
-            event_emitter: SdkEventEmitter::default(),
+            event_emitter,
         }
     }
 

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/statsig-rust/src/statsig_metadata.rs

@@ -49,7 +49,7 @@ pub struct StatsigMetadataWithLogEventExtras {
 impl StatsigMetadata {
     fn new() -> Self {
         Self {
-            sdk_version: "0.11.0".to_string(),
+            sdk_version: "0.11.1".to_string(),
             sdk_type: "statsig-server-core".to_string(),
             session_id: Uuid::new_v4().to_string(),
             os: None,

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/statsig-rust/src/user/statsig_user_builder.rs

@@ -18,6 +18,8 @@ pub struct StatsigUserBuilder {
 
     pub custom: Option<HashMap<String, DynamicValue>>,
     pub private_attributes: Option<HashMap<String, DynamicValue>>,
+
+    pub statsig_environment: Option<HashMap<String, DynamicValue>>,
 }
 
 impl StatsigUserBuilder {
@@ -50,6 +52,7 @@ impl StatsigUserBuilder {
             custom: None,
             private_attributes: None,
             custom_ids: None,
+            statsig_environment: None,
         }
     }
 
@@ -117,6 +120,16 @@ impl StatsigUserBuilder {
         self
     }
 
+    pub fn statsig_environment(
+        mut self,
+        statsig_environment: Option<HashMap<String, String>>,
+    ) -> Self {
+        if let Some(statsig_environment) = statsig_environment {
+            self.statsig_environment = Some(convert_str_map_to_dyn_values(statsig_environment));
+        }
+        self
+    }
+
     // todo: support HashMap<String, String | Number | Boolean | Array<String>>
     pub fn custom_from_str_map(mut self, custom: Option<HashMap<String, String>>) -> Self {
         if let Some(custom) = custom {
@@ -164,6 +177,7 @@ impl StatsigUserBuilder {
             app_version: self.app_version,
             custom: self.custom,
             private_attributes: self.private_attributes,
+            statsig_environment: self.statsig_environment,
             custom_ids: self
                 .custom_ids
                 .map(|m| m.into_iter().map(|(k, v)| (k, v.into())).collect()),

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/statsig-rust/src/user/statsig_user_internal.rs

@@ -1,3 +1,5 @@
+use std::collections::HashMap;
+
 use super::StatsigUserLoggable;
 use crate::evaluation::dynamic_value::DynamicValue;
 use crate::hashing::djb2_number;
@@ -109,6 +111,12 @@ impl<'statsig, 'user> StatsigUserInternal<'statsig, 'user> {
     ) -> Option<DynamicValue> {
         let field = field.as_ref()?;
 
+        if let Some(statsig_environment) = &self.user_ref.data.statsig_environment {
+            if let Some(result) = statsig_environment.get(field.value.as_str()) {
+                return Some(result.clone());
+            }
+        }
+
         if let Some(result) = self.statsig_instance?.get_from_statsig_env(&field.value) {
             return Some(result);
         }
@@ -118,13 +126,15 @@ impl<'statsig, 'user> StatsigUserInternal<'statsig, 'user> {
     }
 
     pub fn to_loggable(&self) -> StatsigUserLoggable {
-        let (environment, global_custom) = match self.statsig_instance {
-            Some(statsig) => (
-                statsig.use_statsig_env(|e| e.cloned()),
-                statsig.use_global_custom_fields(|gc| gc.cloned()),
-            ),
-            None => (None, None),
-        };
+        let mut environment = self.user_ref.data.statsig_environment.clone();
+        let mut global_custom: Option<HashMap<String, DynamicValue>> = None;
+
+        if let Some(statsig_instance) = &self.statsig_instance {
+            if environment.is_none() {
+                environment = statsig_instance.use_statsig_env(|e| e.cloned());
+            }
+            global_custom = statsig_instance.use_global_custom_fields(|gc| gc.cloned());
+        }
 
         StatsigUserLoggable::new(&self.user_ref.data, environment, global_custom)
     }

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/statsig-rust/src/user/user_data.rs

@@ -20,6 +20,7 @@ pub struct UserData {
     pub country: Option<DynamicValue>,
     pub locale: Option<DynamicValue>,
     pub app_version: Option<DynamicValue>,
+    pub statsig_environment: Option<HashMap<String, DynamicValue>>,
 
     #[serde(skip_serializing)]
     pub private_attributes: Option<HashMap<String, DynamicValue>>,

{statsig_python_core-0.11.0 → statsig_python_core-0.11.1}/statsig-rust/tests/global_configs_test.rs

@@ -1,6 +1,6 @@
 use statsig_rust::{
-    global_configs::GlobalConfigs, networking::ResponseData, SpecStore, SpecsSource, SpecsUpdate,
-    StatsigRuntime,
+    global_configs::GlobalConfigs, networking::ResponseData, sdk_event_emitter::SdkEventEmitter,
+    SpecStore, SpecsSource, SpecsUpdate, StatsigRuntime,
 };
 use std::fs;
 
@@ -9,6 +9,7 @@ fn create_test_spec_store(sdk_key: &str) -> SpecStore {
         sdk_key,
         sdk_key.to_string(),
         StatsigRuntime::get_runtime(),
+        std::sync::Arc::new(SdkEventEmitter::default()),
         None,
     )
 }

statsig_python_core-0.11.1/statsig-rust/tests/statsig_environment_tests.rs

@@ -0,0 +1,203 @@
+mod utils;
+
+use crate::utils::mock_specs_adapter::MockSpecsAdapter;
+use serde_json::json;
+use statsig_rust::{Statsig, StatsigOptions, StatsigUser, StatsigUserBuilder};
+use std::collections::HashMap;
+use std::sync::atomic::Ordering;
+use std::sync::Arc;
+use utils::mock_event_logging_adapter::MockEventLoggingAdapter;
+
+const DCS_EVAL_PROJ: &str = "eval_proj_dcs";
+
+async fn setup_with_environment(
+    environment: Option<String>,
+) -> (Statsig, Arc<MockEventLoggingAdapter>) {
+    let logging_adapter = Arc::new(MockEventLoggingAdapter::new());
+    let dcs_path = format!("tests/data/{DCS_EVAL_PROJ}.json");
+    let specs_adapter = Arc::new(MockSpecsAdapter::with_data(&dcs_path));
+
+    let mut options = StatsigOptions::new();
+    options.specs_adapter = Some(specs_adapter);
+    options.event_logging_adapter = Some(logging_adapter.clone());
+    options.environment = environment;
+
+    let uuid = uuid::Uuid::new_v4();
+    let statsig = Statsig::new(&format!("secret-{uuid}"), Some(Arc::new(options)));
+    statsig.initialize().await.unwrap();
+
+    (statsig, logging_adapter)
+}
+
+#[tokio::test]
+async fn test_environment_from_statsig_options() {
+    let (statsig, logging_adapter) = setup_with_environment(Some("production".to_string())).await;
+
+    let user = StatsigUser::with_user_id("test_user".to_string());
+    let _ = statsig.check_gate(&user, "test_public");
+    statsig.shutdown().await.unwrap();
+
+    assert_eq!(
+        logging_adapter
+            .no_diagnostics_logged_event_count
+            .load(Ordering::SeqCst),
+        1
+    );
+
+    let event = logging_adapter.force_get_first_event();
+    assert_eq!(event["eventName"], "statsig::gate_exposure");
+
+    let user_data = event["user"].as_object().expect("user should be an object");
+    let statsig_environment = user_data
+        .get("statsigEnvironment")
+        .expect("statsigEnvironment should exist");
+
+    let env_obj = statsig_environment
+        .as_object()
+        .expect("statsigEnvironment should be an object");
+    assert_eq!(env_obj.get("tier"), Some(&json!("production")));
+}
+
+#[tokio::test]
+async fn test_environment_from_user_overrides_options() {
+    // Test Case 2: Set environment on both StatsigOptions and StatsigUser
+    let (statsig, logging_adapter) = setup_with_environment(Some("production".to_string())).await;
+
+    let mut user_env = HashMap::new();
+    user_env.insert("tier".to_string(), "development".to_string());
+
+    let user = StatsigUserBuilder::new_with_user_id("test_user".to_string())
+        .statsig_environment(Some(user_env))
+        .build();
+
+    let _ = statsig.check_gate(&user, "test_public");
+    statsig.shutdown().await.unwrap();
+
+    assert_eq!(
+        logging_adapter
+            .no_diagnostics_logged_event_count
+            .load(Ordering::SeqCst),
+        1
+    );
+
+    let event = logging_adapter.force_get_first_event();
+    assert_eq!(event["eventName"], "statsig::gate_exposure");
+
+    let user_data = event["user"].as_object().expect("user should be an object");
+    let statsig_environment = user_data
+        .get("statsigEnvironment")
+        .expect("statsigEnvironment should exist");
+
+    let env_obj = statsig_environment
+        .as_object()
+        .expect("statsigEnvironment should be an object");
+
+    assert_eq!(
+        env_obj.get("tier"),
+        Some(&json!("development")),
+        "User's tier value should override options tier value"
+    );
+}
+
+#[tokio::test]
+async fn test_environment_only_from_user() {
+    // Test Case 3: Set environment just on StatsigUser
+    let (statsig, logging_adapter) = setup_with_environment(None).await;
+
+    let mut user_env = HashMap::new();
+    user_env.insert("tier".to_string(), "staging".to_string());
+
+    let user = StatsigUserBuilder::new_with_user_id("test_user".to_string())
+        .statsig_environment(Some(user_env))
+        .build();
+
+    let _ = statsig.check_gate(&user, "test_public");
+    statsig.shutdown().await.unwrap();
+
+    assert_eq!(
+        logging_adapter
+            .no_diagnostics_logged_event_count
+            .load(Ordering::SeqCst),
+        1
+    );
+
+    let event = logging_adapter.force_get_first_event();
+    assert_eq!(event["eventName"], "statsig::gate_exposure");
+
+    // Verify user's environment values are used
+    let user_data = event["user"].as_object().expect("user should be an object");
+    let statsig_environment = user_data
+        .get("statsigEnvironment")
+        .expect("statsigEnvironment should exist when set on user");
+
+    let env_obj = statsig_environment
+        .as_object()
+        .expect("statsigEnvironment should be an object");
+
+    assert_eq!(
+        env_obj.get("tier"),
+        Some(&json!("staging")),
+        "User's tier value should be present"
+    );
+}
+
+#[tokio::test]
+async fn test_no_environment_set() {
+    // Test Case 4: Neither StatsigOptions nor StatsigUser has environment
+    let (statsig, logging_adapter) = setup_with_environment(None).await;
+
+    let user = StatsigUser::with_user_id("test_user".to_string());
+    let _ = statsig.check_gate(&user, "test_public");
+    statsig.shutdown().await.unwrap();
+
+    assert_eq!(
+        logging_adapter
+            .no_diagnostics_logged_event_count
+            .load(Ordering::SeqCst),
+        1
+    );
+
+    let event = logging_adapter.force_get_first_event();
+    assert_eq!(event["eventName"], "statsig::gate_exposure");
+
+    // Verify statsigEnvironment is not present when no environment is set
+    let user_data = event["user"].as_object().expect("user should be an object");
+    assert!(
+        user_data.get("statsigEnvironment").is_none(),
+        "statsigEnvironment should not exist when no environment is set"
+    );
+}
+
+#[tokio::test]
+async fn test_environment_override_behavior() {
+    // Test Case 5: StatsigOptions has tier="production", user sets tier="testing"
+    let (statsig, logging_adapter) = setup_with_environment(Some("production".to_string())).await;
+
+    let mut user_env = HashMap::new();
+    user_env.insert("tier".to_string(), "testing".to_string());
+
+    let user = StatsigUserBuilder::new_with_user_id("test_user".to_string())
+        .statsig_environment(Some(user_env))
+        .build();
+
+    let _ = statsig.check_gate(&user, "test_public");
+    statsig.shutdown().await.unwrap();
+
+    let event = logging_adapter.force_get_first_event();
+
+    let user_data = event["user"].as_object().expect("user should be an object");
+    let statsig_environment = user_data
+        .get("statsigEnvironment")
+        .expect("statsigEnvironment should exist");
+
+    let env_obj = statsig_environment
+        .as_object()
+        .expect("statsigEnvironment should be an object");
+
+    // User's tier value should override options tier value
+    assert_eq!(
+        env_obj.get("tier"),
+        Some(&json!("testing")),
+        "User's tier value should override options tier value"
+    );
+}