elfo-core 0.1.13

The core of the system
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use std::{cell::Cell, future::Future, sync::Arc};

use crate::{object::ObjectMeta, trace_id::TraceId};

tokio::task_local! {
    static META: Arc<ObjectMeta>;
    static TRACE_ID: Cell<TraceId>;
}

/// Returns the current trace id.
///
/// # Panics
/// This function will panic if called ouside actors.
pub fn trace_id() -> TraceId {
    TRACE_ID.with(Cell::get)
}

/// Returns the current trace id if inside the actor system.
pub fn try_trace_id() -> Option<TraceId> {
    TRACE_ID.try_with(Cell::get).ok()
}

/// Replaces the current trace id with the provided one.
///
/// # Panics
/// This function will panic if called ouside actors.
pub fn set_trace_id(trace_id: TraceId) {
    TRACE_ID.with(|stored| stored.set(trace_id));
}

/// Returns the current object's meta.
///
/// # Panics
/// This function will panic if called ouside actors.
pub fn meta() -> Arc<ObjectMeta> {
    META.with(Arc::clone)
}

/// Returns the current object's meta if inside the actor system.
pub fn try_meta() -> Option<Arc<ObjectMeta>> {
    META.try_with(Arc::clone).ok()
}

pub async fn scope<F: Future>(meta: Arc<ObjectMeta>, trace_id: TraceId, f: F) -> F::Output {
    META.scope(meta, TRACE_ID.scope(Cell::new(trace_id), f))
        .await
}

pub fn sync_scope<R>(meta: Arc<ObjectMeta>, trace_id: TraceId, f: impl FnOnce() -> R) -> R {
    META.sync_scope(meta, || TRACE_ID.sync_scope(Cell::new(trace_id), f))
}