openai_api_dispatch/
utils.rs

1#[cfg(all(target_has_atomic = "32", not(feature = "std")))]
2/// Generates a process-local counter ID using 32-bit atomics.
3///
4/// The counters reset on restart and their values are added, not concatenated;
5/// IDs can repeat after wraparound. Supply your own IDs when uniqueness matters.
6pub fn id() -> u128 {
7    use core::sync::atomic::{AtomicU32, Ordering};
8
9    static ID_A: AtomicU32 = AtomicU32::new(0);
10    static ID_B: AtomicU32 = AtomicU32::new(0);
11
12    // affordable risk of acq u32::max, serve 0 to another consumer so it has duplicated (0, 0),
13    // then increment b
14    let a = ID_A.fetch_add(1, Ordering::SeqCst);
15    let b = if a == u32::MAX {
16        ID_B.fetch_add(1, Ordering::SeqCst)
17    } else {
18        ID_B.load(Ordering::SeqCst)
19    };
20
21    a as u128 + b as u128
22}
23
24#[cfg(feature = "std")]
25/// Generates a time-based UUID v7 represented as a `u128`.
26pub fn id() -> u128 {
27    uuid::Uuid::now_v7().as_u128()
28}
29
30#[cfg(all(not(target_has_atomic = "32"), not(feature = "std")))]
31compile_error!(
32    "atomic sync for u32 not supported; std not enabled; no fallback implementation for `id`"
33);
34
35#[cfg(feature = "std")]
36/// Reads `OPENAI_API_DEFAULT_MODEL` into a shared optional value.
37///
38/// Unset or non-Unicode values become `None`. Empty strings are accepted, and
39/// the returned value does not track subsequent environment changes.
40pub fn get_default_model() -> std::sync::Arc<Option<String>> {
41    let default_model = std::env::var("OPENAI_API_DEFAULT_MODEL").ok();
42    std::sync::Arc::new(default_model)
43}
44
45#[test]
46fn id_increments() {
47    let a = (0..1_000).map(|_| id());
48    let b = (0..1_000).map(|_| id());
49
50    a.zip(b).for_each(|(a, b)| assert!(a < b));
51}
openai_api_dispatch/utils.rs

openai_api_dispatch/
utils.rs
