oai-statsig-python-core 0.32.0rc7__tar.gz → 0.33.0__tar.gz

{oai_statsig_python_core-0.32.0rc7 → oai_statsig_python_core-0.33.0}/Cargo.lock

@@ -498,8 +498,8 @@ dependencies = [
  "compression-core",
  "flate2",
  "memchr",
- "zstd 0.14.0",
- "zstd-safe 8.0.0",
+ "zstd",
+ "zstd-safe",
 ]
 
 [[package]]
@@ -1692,9 +1692,9 @@ dependencies = [
 
 [[package]]
 name = "napi"
-version = "3.12.4"
+version = "3.13.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "3d2c1d080c842ec1347ebf3cfb9347c15f8c2262f143cc30eea41ddf1c7da720"
+checksum = "3b12638855cec016784a7e17cc4ecbdee7ca8b0a3456aa50640d0ee5ed1a42f8"
 dependencies = [
  "anyhow",
  "bitflags 2.13.2",
@@ -1712,15 +1712,15 @@ dependencies = [
 
 [[package]]
 name = "napi-build"
-version = "2.4.2"
+version = "2.5.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "860e7c40864f95cfb83cde99f9ebadd88ef3d9bdccd7dd2cee0cc96a2dd4ffa7"
+checksum = "0941983559cd62bfa7f04646a16428967a847365adeef27711e4e3927046bbed"
 
 [[package]]
 name = "napi-derive"
-version = "3.6.5"
+version = "3.6.9"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "350057056a30368aa76c11a0d406b0aa61321710be2c36656ab4f6efe0b785a2"
+checksum = "6b1d232c24dc6754979b27efdf2047a5849efc7c92c1245de87e6bedbb9f5c3c"
 dependencies = [
  "convert_case 0.12.0",
  "ctor",
@@ -1732,9 +1732,9 @@ dependencies = [
 
 [[package]]
 name = "napi-derive-backend"
-version = "6.1.3"
+version = "6.1.4"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "4c1c87a71568f3fe5c736b10878ff55064b250bb4ea4b48c8055713e07b9e463"
+checksum = "fd6856fe1686867da90ed0dc292d7c0c87a695eb7c578b913360f33619ee01e3"
 dependencies = [
  "convert_case 0.12.0",
  "proc-macro2",
@@ -1745,9 +1745,9 @@ dependencies = [
 
 [[package]]
 name = "napi-sys"
-version = "3.3.1"
+version = "3.3.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "4d29a242104e456960e6c6479847570be731c1a92b8f8c05ddfb5439e5c09915"
+checksum = "6801fc65ba4d7035636e93635e281113f00410401a237dcd4f55b1565ead2115"
 dependencies = [
  "libloading",
 ]
@@ -1853,7 +1853,7 @@ dependencies = [
 
 [[package]]
 name = "oai-statsig-grpc"
-version = "0.32.0-rc.7"
+version = "0.33.0"
 dependencies = [
  "async-trait",
  "chrono",
@@ -1869,7 +1869,7 @@ dependencies = [
 
 [[package]]
 name = "oai-statsig-rust"
-version = "0.32.0-rc.7"
+version = "0.33.0"
 dependencies = [
  "ahash",
  "arc-swap",
@@ -1899,7 +1899,7 @@ dependencies = [
  "percent-encoding",
  "pin-project-lite",
  "prost",
- "rand 0.8.8",
+ "rand 0.10.3",
  "reqwest",
  "rkyv",
  "rusty-fork",
@@ -1919,7 +1919,7 @@ dependencies = [
  "uuid",
  "windows-sys 0.61.2",
  "wiremock",
- "zstd 0.13.3",
+ "zstd",
 ]
 
 [[package]]
@@ -2373,7 +2373,7 @@ dependencies = [
  "bytes",
  "getrandom 0.4.3",
  "lru-slab",
- "rand 0.10.2",
+ "rand 0.10.3",
  "rand_pcg",
  "ring",
  "rustc-hash 2.1.3",
@@ -2453,9 +2453,9 @@ dependencies = [
 
 [[package]]
 name = "rand"
-version = "0.10.2"
+version = "0.10.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "c7f5fa3a058cd35567ef9bfa5e75732bee0f9e4c55fa90477bef2dfcdbc4be80"
+checksum = "65c9fb96cbc91e3478eaae79a69fcd3f1ae4ad052e471fe6732fff548984b4af"
 dependencies = [
  "chacha20",
  "getrandom 0.4.3",
@@ -3102,7 +3102,7 @@ checksum = "a2eb9349b6444b326872e140eb1cf5e7c522154d69e7a0ffb0fb81c06b37543f"
 
 [[package]]
 name = "statsig-node"
-version = "0.32.0-rc.7"
+version = "0.33.0"
 dependencies = [
  "async-trait",
  "napi",
@@ -3117,7 +3117,7 @@ dependencies = [
 
 [[package]]
 name = "statsig-pyo3"
-version = "0.32.0-rc.7"
+version = "0.33.0"
 dependencies = [
  "async-trait",
  "lazy_static",
@@ -3132,7 +3132,7 @@ dependencies = [
 
 [[package]]
 name = "statsig_ffi"
-version = "0.32.0-rc.7"
+version = "0.33.0"
 dependencies = [
  "async-trait",
  "cbindgen",
@@ -3793,7 +3793,7 @@ checksum = "2ef6dac1e96601b4fb3acccccff2139741fcb757cb9a36089bf5be91cfb285ce"
 dependencies = [
  "getrandom 0.4.3",
  "js-sys",
- "rand 0.10.2",
+ "rand 0.10.3",
  "wasm-bindgen",
 ]
 
@@ -4347,31 +4347,13 @@ version = "1.0.23"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "29666d0abbfad1e3dc4dcf6144730dd3a3ab225bbbdac83319345b1b44ccfc1b"
 
-[[package]]
-name = "zstd"
-version = "0.13.3"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "e91ee311a569c327171651566e07972200e76fcfe2242a4fa446149a3881c08a"
-dependencies = [
- "zstd-safe 7.3.0",
-]
-
 [[package]]
 name = "zstd"
 version = "0.14.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "bf06bd8162af0734b344780deb55b42a2429ae430870d13fcc12f238e880fe6e"
 dependencies = [
- "zstd-safe 8.0.0",
-]
-
-[[package]]
-name = "zstd-safe"
-version = "7.3.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "64d80649ab6db9d9f6f9c80a40becd948eda4714a0a5ac8c4d157a32231c7882"
-dependencies = [
- "zstd-sys",
+ "zstd-safe",
 ]
 
 [[package]]

{oai_statsig_python_core-0.32.0rc7 → oai_statsig_python_core-0.33.0}/Cargo.toml

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

{oai_statsig_python_core-0.32.0rc7 → oai_statsig_python_core-0.33.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: oai_statsig_python_core
-Version: 0.32.0rc7
+Version: 0.33.0
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

oai_statsig_python_core-0.33.0/statsig-pyo3/requirements-typing.txt

@@ -0,0 +1,3 @@
+mypy==2.3.1
+types-requests==2.33.0.20260906
+types-protobuf==6.32.1.20260221

{oai_statsig_python_core-0.32.0rc7 → oai_statsig_python_core-0.33.0}/statsig-rust/Cargo.toml

@@ -28,7 +28,7 @@ memchr = "2.8.3"
 parking_lot = "0.12.5"
 pin-project-lite = "0.2.17"
 percent-encoding = "2.3.2"
-rand = "0.8.8"
+rand = "0.10.3"
 fancy-regex = "0.17.0"
 reqwest = { version = "0.12.28", features = [
     "json",
@@ -47,14 +47,14 @@ serde_json = { version = "1.0.151", features = [
 ] }
 serde_with = "3.23.0"
 sha2 = "0.11.0"
-oai-statsig-grpc = { path = "../statsig-grpc", version = "0.32.0-rc.7", optional = true }
+oai-statsig-grpc = { path = "../statsig-grpc", version = "0.33.0", optional = true }
 simple_logger = { version = "5.2.0" }
 tempfile = "3.27.0"
 tokio = { version = "1.53.1", features = ["full"] }
 url = "2.5.8"
 uaparser = "0.6.4"
 uuid = { version = "1.26.1", features = ["v4", "fast-rng"] }
-zstd = { version = "0.13.3", features = ["zdict_builder"] }
+zstd = { version = "0.14.0", features = ["zdict_builder"] }
 tonic = "0.14.6"
 brotli = "9.0.0"
 bytes = "1.12.1"
@@ -99,3 +99,8 @@ with_zstd = []
 pyo3_event_zstd = []
 ffi-support = []
 ordered_user_data_maps = ["dep:indexmap"]
+
+[[test]]
+name = "fork_random"
+path = "tests/fork_random.rs"
+harness = false

{oai_statsig_python_core-0.32.0rc7 → oai_statsig_python_core-0.33.0}/statsig-rust/src/evaluation/cmab_evaluator.rs

@@ -8,7 +8,7 @@ use crate::interned_string::InternedString;
 use crate::specs_response::cmab_types::{CMABConfig, CMABGroup, CMABGroupConfig};
 use crate::unwrap_or_return;
 use lazy_static::lazy_static;
-use rand::Rng;
+use rand::RngExt;
 use rand::seq::SliceRandom;
 use serde::{Deserialize, Serialize};
 use serde_json::Value;
@@ -200,7 +200,7 @@ fn get_passes_targeting<'a>(ctx: &mut EvaluatorContext<'a>, cmab: &'a CMABConfig
 
 fn get_shuffled_groups(cmab: &CMABConfig) -> Vec<&CMABGroup> {
     let mut groups = cmab.groups.iter().collect::<Vec<&CMABGroup>>();
-    let mut rng = rand::thread_rng();
+    let mut rng = crate::utils::random::rng();
     groups.shuffle(&mut rng);
     groups
 }
@@ -241,8 +241,8 @@ fn apply_sampling_group<'a>(
     }
 
     let mut sum: f64 = 0.0;
-    let mut rng = rand::thread_rng();
-    let value: f64 = rng.gen::<f64>();
+    let mut rng = crate::utils::random::rng();
+    let value: f64 = rng.random::<f64>();
     for group in &cmab.groups {
         let cur_count = match config.get(group.id.as_str()) {
             Some(config_for_group) => config_for_group.records + 1,
@@ -289,8 +289,8 @@ fn apply_best_group<'a>(
 
     // if we had no score for any group randomly select one
     if !has_score {
-        let mut rng = rand::thread_rng();
-        let random: f64 = rng.gen::<f64>();
+        let mut rng = crate::utils::random::rng();
+        let random: f64 = rng.random::<f64>();
         best_group = &cmab.groups[(random * cmab.groups.len() as f64).floor() as usize];
     }
     ctx.result.bool_value = true;

{oai_statsig_python_core-0.32.0rc7 → oai_statsig_python_core-0.33.0}/statsig-rust/src/event_logging_adapter/statsig_local_file_event_logging_adapter.rs

@@ -13,7 +13,7 @@ use crate::{
 };
 use async_trait::async_trait;
 use file_guard::Lock;
-use rand::Rng;
+use rand::RngExt;
 use serde::{Deserialize, Serialize};
 use serde_json::json;
 
@@ -196,7 +196,7 @@ fn create_merge_key(event: &StatsigEventInternal) -> String {
 // PHP initializes per request, so we get a diagnostics event per request.
 // This samples quite aggressively to compensate for that
 fn should_sample_sdk_diagnostics() -> bool {
-    let random_number = rand::thread_rng().gen_range(0..10000);
+    let random_number = crate::utils::random::rng().random_range(0..10000);
     random_number < 1
 }
 

{oai_statsig_python_core-0.32.0rc7 → oai_statsig_python_core-0.33.0}/statsig-rust/src/id_lists_adapter/id_list.rs

@@ -26,16 +26,20 @@ impl IdList {
     }
 
     pub fn apply_update(&mut self, update: IdListUpdate) {
+        let changeset_data = unwrap_or_noop!(&update.raw_changeset);
         let updated_meta = update.new_metadata;
         let current_meta = &self.metadata;
 
-        if updated_meta.file_id != current_meta.file_id
-            && updated_meta.creation_time >= current_meta.creation_time
+        // The HTTP adapter downloads from byte zero when creationTime advances,
+        // even if the file ID is unchanged. Apply that body as a full replay so
+        // the consumed cursor and timestamp match the next manifest poll.
+        if updated_meta.creation_time > current_meta.creation_time
+            || (updated_meta.file_id != current_meta.file_id
+                && updated_meta.creation_time >= current_meta.creation_time)
         {
             self.update_metadata(updated_meta);
         }
 
-        let changeset_data = unwrap_or_noop!(&update.raw_changeset);
         let ids = Arc::make_mut(&mut self.ids);
 
         for change in changeset_data.lines() {

oai_statsig_python_core-0.33.0/statsig-rust/src/id_lists_adapter/id_list_cursor_tests.rs

@@ -0,0 +1,228 @@
+use super::*;
+use crate::SpecStore;
+use crate::id_lists_adapter::IdList;
+use crate::sdk_event_emitter::SdkEventEmitter;
+use mockito::Server;
+use std::collections::HashSet;
+
+fn metadata() -> IdListMetadata {
+    IdListMetadata {
+        name: "employees".into(),
+        url: "https://example.com/employees".into(),
+        file_id: Some("file-1".into()),
+        size: 0,
+        creation_time: 1,
+    }
+}
+
+#[test]
+fn same_file_and_creation_time_appends_changes() {
+    let mut list = IdList::new(metadata());
+    list.apply_update(IdListUpdate {
+        raw_changeset: Some("+alice\n+bob\n".to_string()),
+        new_metadata: metadata(),
+    });
+    list.apply_update(IdListUpdate {
+        raw_changeset: Some("-alice\n+carol\n".to_string()),
+        new_metadata: metadata(),
+    });
+
+    assert_eq!(
+        *list.ids,
+        HashSet::from(["bob".to_string(), "carol".to_string()])
+    );
+    assert_eq!(
+        list.metadata.size,
+        "+alice\n+bob\n-alice\n+carol\n".len() as u64
+    );
+}
+
+#[test]
+fn new_file_at_same_or_newer_creation_time_replaces_members_and_cursor() {
+    for creation_time in [1, 2] {
+        let mut list = IdList::new(metadata());
+        list.apply_update(IdListUpdate {
+            raw_changeset: Some("+alice\n".to_string()),
+            new_metadata: metadata(),
+        });
+        let mut replacement = metadata();
+        replacement.file_id = Some("file-2".to_string());
+        replacement.creation_time = creation_time;
+        list.apply_update(IdListUpdate {
+            raw_changeset: Some("+bob\n".to_string()),
+            new_metadata: replacement,
+        });
+
+        assert_eq!(*list.ids, HashSet::from(["bob".to_string()]));
+        assert_eq!(list.metadata.size, "+bob\n".len() as u64);
+        assert_eq!(list.metadata.file_id.as_deref(), Some("file-2"));
+        assert_eq!(list.metadata.creation_time, creation_time);
+    }
+}
+
+#[test]
+fn missing_body_preserves_members_cursor_and_generation() {
+    for file_id in ["file-1", "file-2"] {
+        let mut list = IdList::new(metadata());
+        list.apply_update(IdListUpdate {
+            raw_changeset: Some("+alice\n".into()),
+            new_metadata: metadata(),
+        });
+        let previous = list.clone();
+        let mut next = metadata();
+        next.file_id = Some(file_id.into());
+        next.creation_time = 2;
+        list.apply_update(IdListUpdate {
+            raw_changeset: None,
+            new_metadata: next,
+        });
+        assert_eq!(*list.ids, *previous.ids);
+        assert!(Arc::ptr_eq(&list.ids, &previous.ids));
+        assert_eq!(list.metadata.creation_time, previous.metadata.creation_time);
+        assert_eq!(list.metadata.file_id, previous.metadata.file_id);
+        assert_eq!(list.metadata.size, previous.metadata.size);
+    }
+}
+
+#[test]
+fn same_file_full_replay_can_replace_members_with_an_empty_set() {
+    let mut list = IdList::new(metadata());
+    list.apply_update(IdListUpdate {
+        raw_changeset: Some("+alice\n".into()),
+        new_metadata: metadata(),
+    });
+    let previous = list.clone();
+    let mut next = metadata();
+    next.creation_time = 2;
+    list.apply_update(IdListUpdate {
+        raw_changeset: Some(String::new()),
+        new_metadata: next,
+    });
+    assert!(list.ids.is_empty());
+    assert_eq!(list.metadata.creation_time, 2);
+    assert_eq!(list.metadata.size, 0);
+    assert!(previous.ids.contains("alice"));
+}
+
+#[tokio::test]
+async fn test_same_file_full_replay_updates_cursor_and_only_downloads_once() {
+    let mut server = Server::new_async().await;
+    let list_name = "employees";
+    let initial_body = "+alice\n+bob\n";
+    let full_body = "+alice\n+bob\n-alice\n+carol\n";
+    let appended_body = "+dan\n";
+    let mut metadata = IdListMetadata {
+        name: list_name.to_string(),
+        url: format!("{}/employees", server.url()),
+        file_id: Some("file-a".to_string()),
+        size: initial_body.len() as u64,
+        creation_time: 1,
+    };
+    let listener = Arc::new(SpecStore::new(
+        "secret-cursor-test",
+        "cursor-test".into(),
+        StatsigRuntime::get_runtime(),
+        Arc::new(SdkEventEmitter::default()),
+        None,
+    ));
+    // Seed the overcounted cursor left by repeated full replays in an older SDK.
+    listener.did_receive_id_list_updates(HashMap::from([(
+        list_name.to_string(),
+        IdListUpdate {
+            raw_changeset: Some(initial_body.repeat(3)),
+            new_metadata: metadata.clone(),
+        },
+    )]));
+    let original = listener.load_data();
+    assert_eq!(
+        original.id_lists[list_name].metadata.size,
+        3 * initial_body.len() as u64
+    );
+    let adapter = StatsigHttpIdListsAdapter::new(
+        "secret-key",
+        &StatsigOptions {
+            id_lists_url: Some(format!("{}/get_id_lists", server.url())),
+            ..StatsigOptions::default()
+        },
+    );
+    adapter.set_listener(listener.clone());
+
+    metadata.creation_time = 2;
+    metadata.size = full_body.len() as u64;
+    let manifest = server
+        .mock("POST", "/get_id_lists")
+        .with_status(200)
+        .with_body(serde_json::to_string(&HashMap::from([(list_name, &metadata)])).unwrap())
+        .expect(2)
+        .create_async()
+        .await;
+    let full_download = server
+        .mock("GET", "/employees")
+        .match_header("range", "bytes=0-")
+        .with_status(206)
+        .with_body(full_body)
+        .expect(1)
+        .create_async()
+        .await;
+
+    adapter.sync_id_lists().await.unwrap();
+    adapter.sync_id_lists().await.unwrap();
+
+    manifest.assert_async().await;
+    full_download.assert_async().await;
+    let current = listener.get_current_id_list_metadata();
+    assert_eq!(current[list_name].creation_time, 2);
+    assert_eq!(current[list_name].size, full_body.len() as u64);
+    assert!(
+        !listener.load_data().id_lists[list_name]
+            .ids
+            .contains("alice")
+    );
+    assert!(listener.load_data().id_lists[list_name].ids.contains("bob"));
+    assert!(
+        listener.load_data().id_lists[list_name]
+            .ids
+            .contains("carol")
+    );
+
+    assert!(original.id_lists[list_name].ids.contains("alice"));
+    assert!(!original.id_lists[list_name].ids.contains("carol"));
+    let replayed = listener.load_data();
+
+    // With unchanged creationTime, subsequent growth must read just the tail.
+    manifest.remove_async().await;
+    metadata.size += appended_body.len() as u64;
+    let manifest = server
+        .mock("POST", "/get_id_lists")
+        .with_status(200)
+        .with_body(serde_json::to_string(&HashMap::from([(list_name, &metadata)])).unwrap())
+        .expect(1)
+        .create_async()
+        .await;
+    let tail_download = server
+        .mock("GET", "/employees")
+        .match_header("range", format!("bytes={}-", full_body.len()).as_str())
+        .with_status(206)
+        .with_body(appended_body)
+        .expect(1)
+        .create_async()
+        .await;
+
+    adapter.sync_id_lists().await.unwrap();
+
+    manifest.assert_async().await;
+    tail_download.assert_async().await;
+    full_download.assert_async().await;
+    assert_eq!(
+        listener.get_current_id_list_metadata()[list_name].size,
+        metadata.size
+    );
+    assert!(listener.load_data().id_lists[list_name].ids.contains("dan"));
+    assert!(listener.load_data().id_lists[list_name].ids.contains("bob"));
+    assert!(
+        !listener.load_data().id_lists[list_name]
+            .ids
+            .contains("alice")
+    );
+    assert!(!replayed.id_lists[list_name].ids.contains("dan"));
+}

{oai_statsig_python_core-0.32.0rc7 → oai_statsig_python_core-0.33.0}/statsig-rust/src/id_lists_adapter/id_list_propagation_tests.rs

@@ -377,9 +377,10 @@ async fn id_list_propagation_replay_cannot_invent_contiguous_coverage() {
     fixture.set("users", "A", 2, 0, "+a\n", Some(old));
     fixture.adapter.sync_id_lists().await.unwrap();
     assert!(fixture.metrics().is_empty());
-    // Preserve the existing cursor behavior: replay counts bytes again. Telemetry must not.
-    assert_eq!(fixture.store.load_data().id_lists["users"].metadata.size, 6);
-    fixture.set("users", "A", 1, 6, "+p\n", Some(new));
+    // Full replay installs the byte count once. A later response that skips
+    // bytes must still not invent contiguous coverage for telemetry.
+    assert_eq!(fixture.store.load_data().id_lists["users"].metadata.size, 3);
+    fixture.set("users", "A", 2, 6, "+p\n", Some(new));
     fixture.adapter.sync_id_lists().await.unwrap();
     assert!(
         fixture.metrics().is_empty(),
@@ -390,7 +391,7 @@ async fn id_list_propagation_replay_cannot_invent_contiguous_coverage() {
             .ids
             .contains("p")
     );
-    assert_eq!(fixture.ranges(), ["bytes=0-", "bytes=0-", "bytes=6-"]);
+    assert_eq!(fixture.ranges(), ["bytes=0-", "bytes=0-", "bytes=3-"]);
 }
 
 #[tokio::test]

{oai_statsig_python_core-0.32.0rc7 → oai_statsig_python_core-0.33.0}/statsig-rust/src/id_lists_adapter/statsig_http_id_lists_adapter.rs

@@ -1744,3 +1744,7 @@ mod refresh_tests;
 #[cfg(all(test, feature = "custom_network_provider"))]
 #[path = "id_list_propagation_tests.rs"]
 mod propagation_tests;
+
+#[cfg(test)]
+#[path = "id_list_cursor_tests.rs"]
+mod cursor_tests;

{oai_statsig_python_core-0.32.0rc7 → oai_statsig_python_core-0.33.0}/statsig-rust/src/sdk_diagnostics/diagnostics.rs

@@ -9,7 +9,7 @@ use crate::log_w;
 
 use crate::event_logging::event_logger::EventLogger;
 use parking_lot::Mutex;
-use rand::Rng;
+use rand::RngExt;
 use serde::Serialize;
 use std::collections::HashMap;
 use std::fmt;
@@ -148,8 +148,8 @@ impl Diagnostics {
 
     pub fn should_sample(&self, context: &ContextType, key: Option<KeyType>) -> bool {
         fn check_sampling_rate(sampling_rate: Option<&f64>) -> bool {
-            let mut rng = rand::thread_rng();
-            let rand_value = rng.gen::<f64>() * MAX_SAMPLING_RATE;
+            let mut rng = crate::utils::random::rng();
+            let rand_value = rng.random::<f64>() * MAX_SAMPLING_RATE;
 
             match sampling_rate {
                 Some(sampling_rate) => rand_value < *sampling_rate,