samply 0.13.1

A command line profiler for macOS and Linux.
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

use std::fmt::Debug;

use fxprof_processed_profile::{Frame, FrameInfo, Profile, StringHandle, ThreadHandle, Timestamp};

use crate::shared::context_switch::ThreadContextSwitchData;
use crate::shared::unresolved_samples::UnresolvedStackHandle;

#[derive(Debug)]
pub struct Thread {
    pub profile_thread: ThreadHandle,
    pub context_switch_data: ThreadContextSwitchData,
    pub last_sample_timestamp: Option<u64>,

    /// Some() between sched_switch and the next context switch IN
    ///
    /// Refers to a stack in the containing Process's UnresolvedSamples stack table.
    pub off_cpu_stack: Option<UnresolvedStackHandle>,
    pub name: Option<String>,
    pub thread_label_frame: FrameInfo,
}

impl Thread {
    pub fn new(
        thread_handle: ThreadHandle,
        thread_label_frame: FrameInfo,
        name: Option<String>,
    ) -> Self {
        Self {
            profile_thread: thread_handle,
            context_switch_data: Default::default(),
            last_sample_timestamp: None,
            off_cpu_stack: None,
            name,
            thread_label_frame,
        }
    }

    pub fn thread_label(&self) -> StringHandle {
        match self.thread_label_frame.frame {
            Frame::Label(s) => s,
            _ => panic!(),
        }
    }

    pub fn rename_with_recycling(
        &mut self,
        name: String,
        (thread_handle, thread_label_frame): (ThreadHandle, FrameInfo),
    ) -> (Option<String>, (ThreadHandle, FrameInfo)) {
        let old_thread_handle = std::mem::replace(&mut self.profile_thread, thread_handle);
        let old_thread_label_frame =
            std::mem::replace(&mut self.thread_label_frame, thread_label_frame);
        let old_name = std::mem::replace(&mut self.name, Some(name));
        (old_name, (old_thread_handle, old_thread_label_frame))
    }

    pub fn rename_without_recycling(
        &mut self,
        name: String,
        thread_label_frame: FrameInfo,
        profile: &mut Profile,
    ) {
        profile.set_thread_name(self.profile_thread, &name);
        self.thread_label_frame = thread_label_frame;
        self.name = Some(name);
    }

    pub fn notify_dead(&mut self, end_time: Timestamp, profile: &mut Profile) {
        profile.set_thread_end_time(self.profile_thread, end_time);
    }

    pub fn finish(self) -> (Option<String>, (ThreadHandle, FrameInfo)) {
        (self.name, (self.profile_thread, self.thread_label_frame))
    }
}