tdm_server_rust/telemetry/skywalking/
context.rs

1//! 请求级 SkyWalking TracingContext（tokio task_local）
2
3use ::skywalking::trace::{
4    span::{HandleSpanObject, Span},
5    trace_context::TracingContext,
6    tracer::create_trace_context,
7};
8use std::cell::RefCell;
9use std::future::Future;
10
11tokio::task_local! {
12    static SW_TRACE_CTX: RefCell<Option<TracingContext>>;
13}
14
15/// 读取当前 SW trace_id（32hex 字符串）
16pub fn current_trace_id() -> Option<String> {
17    SW_TRACE_CTX
18        .try_with(|cell: &RefCell<Option<TracingContext>>| {
19            cell.borrow()
20                .as_ref()
21                .map(|ctx: &TracingContext| ctx.trace_id().to_string())
22        })
23        .ok()
24        .flatten()
25}
26
27/// 是否处于 HTTP 请求 SW 上下文内
28pub fn is_in_request_context() -> bool {
29    SW_TRACE_CTX
30        .try_with(|cell: &RefCell<Option<TracingContext>>| cell.borrow().is_some())
31        .unwrap_or(false)
32}
33
34/// 在 SW 上下文作用域内执行 future；结束时 wait 全部 async span 再释放 ctx
35pub async fn with_context<F, R>(ctx: TracingContext, fut: F) -> R
36where
37    F: Future<Output = R>,
38{
39    SW_TRACE_CTX
40        .scope(RefCell::new(Some(ctx)), async move {
41            let result = fut.await;
42            if let Some(c) = SW_TRACE_CTX
43                .try_with(|cell: &RefCell<Option<TracingContext>>| cell.borrow_mut().take())
44                .ok()
45                .flatten()
46            {
47                c.wait();
48            }
49            result
50        })
51        .await
52}
53
54/// 可变访问当前 TracingContext
55pub fn with_current_mut<F, R>(f: F) -> Option<R>
56where
57    F: FnOnce(&mut TracingContext) -> R,
58{
59    SW_TRACE_CTX
60        .try_with(|cell: &RefCell<Option<TracingContext>>| cell.borrow_mut().as_mut().map(f))
61        .ok()
62        .flatten()
63}
64
65/// 只读访问当前 TracingContext
66pub fn with_current<F, R>(f: F) -> Option<R>
67where
68    F: FnOnce(&TracingContext) -> R,
69{
70    SW_TRACE_CTX
71        .try_with(|cell: &RefCell<Option<TracingContext>>| cell.borrow().as_ref().map(f))
72        .ok()
73        .flatten()
74}
75
76/// 创建 LocalSpan（需在 Entry 上下文内；无 Entry 时返回 None 而非 panic）
77pub fn create_local_span(operation: &str) -> Option<Span> {
78    std::panic::catch_unwind(std::panic::AssertUnwindSafe(|| {
79        with_current_mut(|ctx| ctx.create_local_span(operation))
80    }))
81    .ok()
82    .flatten()
83}
84
85/// 后台任务独立 TraceContext（tag layer=bootstrap，不进 HTTP 日志）
86pub async fn with_bootstrap_context<F, R>(operation: &str, tag: &str, fut: F) -> R
87where
88    F: Future<Output = R>,
89{
90    if !super::reporter::is_ready() {
91        return fut.await;
92    }
93    let mut ctx = create_trace_context();
94    let mut entry = ctx.create_entry_span(operation);
95    entry.add_tag("layer", "bootstrap");
96    entry.add_tag("task", tag);
97    with_context(ctx, async move {
98        let _entry = entry;
99        fut.await
100    })
101    .await
102}
tdm_server_rust/telemetry/skywalking/context.rs

tdm_server_rust/telemetry/skywalking/
context.rs
