nemo-flow 0.2.0

Core Rust SDK for NeMo Flow observability, scope management, and runtime instrumentation.
Documentation 
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

// SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0

//! Process-global access to the shared runtime context state.
//!
//! The public API layer uses this module to resolve the single
//! [`NemoFlowContextState`] instance that owns middleware registrations and
//! runtime extensions for the current process.

use std::sync::{Arc, RwLock};

use crate::api::runtime::state::NemoFlowContextState;

static GLOBAL_CONTEXT: std::sync::OnceLock<Arc<RwLock<NemoFlowContextState>>> =
    std::sync::OnceLock::new();

/// Return the process-global runtime context state handle.
///
/// This lazily initializes the shared [`NemoFlowContextState`] on first use and
/// returns a cloned [`Arc`] handle to the same underlying [`RwLock`] on every
/// subsequent call.
///
/// # Returns
/// An [`Arc`] pointing at the singleton [`RwLock`] that stores the runtime
/// context state for the current process.
///
/// # Notes
/// All callers share the same underlying state. Mutations made through one
/// handle are visible through every other handle returned by this function.
pub fn global_context() -> Arc<RwLock<NemoFlowContextState>> {
    GLOBAL_CONTEXT
        .get_or_init(|| Arc::new(RwLock::new(NemoFlowContextState::new())))
        .clone()
}