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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
use crate::span_context::HSpanContext;
use crate::span_wrap::SpanWrap;
use rustracing::sampler::*;
use rustracing::span::StartSpanOptions;
use rustracing_jaeger::Span as RjSpan;
use rustracing_jaeger::{span::SpanContextState, Tracer};
use std::borrow::Cow;
lazy_static! {
pub(crate) static ref NOOP_SPAN: HSpan = HSpan::noop();
pub(crate) static ref NULL_TRACER: Tracer = Tracer::new(NullSampler).0;
}
#[derive(Debug, Shrinkwrap)]
#[shrinkwrap(mutable)]
pub struct HSpan(pub RjSpan);
impl From<RjSpan> for HSpan {
fn from(span: RjSpan) -> HSpan {
HSpan(span)
}
}
impl HSpan {
pub fn event<S: Into<Cow<'static, str>>>(&mut self, msg: S) {
self.0.log(|l| {
l.std().event(msg);
})
}
pub fn error<S: Into<Cow<'static, str>>>(&mut self, kind: S, msg: S) {
self.0.log(|l| {
l.error().kind(kind).message(msg);
})
}
pub fn context(&self) -> Option<HSpanContext> {
self.0.context().map(|ctx| HSpanContext(ctx.to_owned()))
}
pub fn child_<N: Into<Cow<'static, str>>, F>(&self, operation_name: N, f: F) -> RjSpan
where
F: FnOnce(StartSpanOptions<AllSampler, SpanContextState>) -> RjSpan,
{
self.0.child(operation_name, f)
}
pub fn follower_<N: Into<Cow<'static, str>>, F>(&self, operation_name: N, f: F) -> RjSpan
where
F: FnOnce(StartSpanOptions<AllSampler, SpanContextState>) -> RjSpan,
{
self.0.follower(operation_name, f)
}
pub fn child<S: Into<Cow<'static, str>>>(&self, operation_name: S) -> Self {
self.0.child(operation_name, |o| o.start()).into()
}
pub fn follower<S: Into<Cow<'static, str>>>(&self, operation_name: S) -> Self {
self.0.follower(operation_name, |o| o.start()).into()
}
pub fn wrap<T>(self, data: T) -> SpanWrap<T> {
SpanWrap::new(data, self.0.context().cloned().map(HSpanContext))
}
pub fn noop() -> Self {
noop("no-op, intentionally disconnected RjSpan")
}
pub fn todo(reason: &'static str) -> Self {
noop(&format!("TODO: {}", reason))
}
pub fn fixme() -> Self {
noop("not yet hooked up")
}
}
pub fn null_tracer() -> Tracer {
NULL_TRACER.clone()
}
pub fn noop(name: &str) -> HSpan {
null_tracer().span(name.to_string()).start().into()
}
pub fn test_span() -> HSpan {
noop("noop test span")
}
#[cfg(test)]
mod tests {
use super::*;
use crate::{tracer_console, Tag};
#[test]
fn trace_test() {
let (tracer, mut reporter) = tracer_console::new_tracer_with_console_reporter();
{
let parent_span: HSpan = tracer.span("parent").start().into();
{
let mut child_span = parent_span.child("child_span");
child_span.set_tag(|| Tag::new("id", "A"));
child_span.event("a log message");
std::thread::sleep(std::time::Duration::from_millis(10));
let mut child_b_span = parent_span.child("child_b_span");
child_b_span.set_tag(|| Tag::new("id", "B"));
let mut _grand_child_span = child_span.child("grand_child_span");
let mut _follower_span = child_span.follower("child_follower_span");
}
let parent_follower_span = parent_span.follower("parent_follower_span");
let _parent_follower_b_span = parent_span.follower("parent_follower_b_span");
let mut _parent_follower_follower_span =
parent_follower_span.follower("parent_follower_follower_span");
}
let count = reporter.drain();
assert_eq!(8, count);
reporter.print(false);
}
}