msgpack-tracing 0.1.2

Compact storage for tracing using msgpack
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
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

use crate::tape::{Instruction, InstructionSet, SpanRecords, TapeMachine};
use std::{collections::HashMap, num::NonZeroU64};

pub struct RestartableMachine<T> {
    forward: T,
    span: HashMap<NonZeroU64, SpanRecords>,
    current_span: Option<(NonZeroU64, SpanRecords)>,
}
impl<T> RestartableMachine<T>
where
    T: TapeMachine<InstructionSet>,
{
    pub fn new(forward: T) -> Self {
        Self {
            forward,
            span: Default::default(),
            current_span: None,
        }
    }
}
impl<T> TapeMachine<InstructionSet> for RestartableMachine<T>
where
    T: TapeMachine<InstructionSet>,
{
    fn needs_restart(&mut self) -> bool {
        self.forward.needs_restart()
    }

    fn handle(&mut self, instruction: Instruction) {
        match instruction {
            Instruction::Restart => {
                self.forward.handle(Instruction::Restart);

                for (span, records) in self.span.iter() {
                    self.forward.handle(Instruction::NewSpan {
                        parent: records.parent,
                        span: *span,
                        name: records.name.as_ref(),
                    });

                    for record in records.records.iter() {
                        self.forward.handle(Instruction::AddValue(record.as_ref()));
                    }

                    self.forward.handle(Instruction::FinishedSpan);
                }
            }
            Instruction::NewSpan { parent, span, name } => {
                assert!(self.current_span.is_none());
                self.current_span = Some((
                    span,
                    SpanRecords {
                        parent,
                        name: name.to_owned(),
                        records: Default::default(),
                    },
                ));

                self.forward
                    .handle(Instruction::NewSpan { parent, span, name });
            }
            Instruction::FinishedSpan => {
                let (k, v) = self.current_span.take().unwrap();
                self.span.insert(k, v);
                self.forward.handle(Instruction::FinishedSpan)
            }
            Instruction::NewRecord(span) => {
                assert!(self.current_span.is_none());
                self.current_span = Some(self.span.remove_entry(&span).unwrap());
                self.forward.handle(Instruction::NewRecord(span));
            }
            Instruction::FinishedRecord => {
                let (k, v) = self.current_span.take().unwrap();
                self.span.insert(k, v);
                self.forward.handle(Instruction::FinishedRecord)
            }
            Instruction::StartEvent {
                time,
                span,
                target,
                priority,
            } => {
                self.forward.handle(Instruction::StartEvent {
                    time,
                    span,
                    target,
                    priority,
                });
            }
            Instruction::FinishedEvent => self.forward.handle(Instruction::FinishedEvent),
            Instruction::AddValue(field_value) => {
                if let Some((_, current_span)) = self.current_span.as_mut() {
                    current_span.records.push(field_value.to_owned());
                }
                self.forward.handle(Instruction::AddValue(field_value));
            }
            Instruction::DeleteSpan(span) => {
                self.span.remove(&span);
                self.forward.handle(Instruction::DeleteSpan(span));
            }
        }
    }
}