rustmeter-beacon-target 0.2.0

Target side library for rustmeter-beacon
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

//! Embassy tracing integration for Rustmeter Beacon
//! Endpoints for Embassy's tracing hooks to send events to Rustmeter Beacon

use rustmeter_beacon_core::{
    compressed_task_id,
    protocol::{
        EventPayload, TypeDefinitionPayload,
        raw_writers::{
            write_embassy_executor_idle, write_embassy_executor_poll_start,
            write_embassy_task_exec_begin, write_embassy_task_exec_end, write_embassy_task_ready,
        },
    },
    tracing::write_tracing_event,
};

use crate::executor_registry::ExecutorRegistry;

// Registry to map long executor IDs to short IDs
static EXECUTOR_REGISTRY: ExecutorRegistry = ExecutorRegistry::new();

#[unsafe(no_mangle)]
fn _embassy_trace_poll_start(executor_id: u32) {
    let executor_id = EXECUTOR_REGISTRY.lookup_or_register(executor_id).unwrap();

    write_embassy_executor_poll_start(executor_id);
}

#[unsafe(no_mangle)]
fn _embassy_trace_executor_idle(executor_id: u32) {
    let executor_id = EXECUTOR_REGISTRY.lookup_or_register(executor_id).unwrap();

    write_embassy_executor_idle(executor_id);
}

#[unsafe(no_mangle)]
fn _embassy_trace_task_new(executor_id: u32, task_id: u32) {
    critical_section::with(|_| {
        let payload = EventPayload::TypeDefinition(TypeDefinitionPayload::EmbassyTaskCreated {
            task_id: task_id,
            executor_id_long: executor_id,
            executor_id_short: EXECUTOR_REGISTRY.lookup_or_register(executor_id).unwrap(),
        });

        write_tracing_event(payload);
    });
}

#[unsafe(no_mangle)]
fn _embassy_trace_task_end(executor_id: u32, task_id: u32) {
    critical_section::with(|_| {
        let payload = EventPayload::TypeDefinition(TypeDefinitionPayload::EmbassyTaskEnded {
            task_id: task_id,
            executor_id_long: executor_id,
            executor_id_short: EXECUTOR_REGISTRY.lookup_or_register(executor_id).unwrap(),
        });

        write_tracing_event(payload);
    });
}

#[unsafe(no_mangle)]
fn _embassy_trace_task_exec_begin(executor_id: u32, task_id: u32) {
    let executor_id = EXECUTOR_REGISTRY.lookup_or_register(executor_id).unwrap();

    write_embassy_task_exec_begin(compressed_task_id(task_id), executor_id);
}

#[unsafe(no_mangle)]
fn _embassy_trace_task_exec_end(executor_id: u32, _task_id: u32) {
    let executor_id = EXECUTOR_REGISTRY.lookup_or_register(executor_id).unwrap();

    write_embassy_task_exec_end(executor_id);
}

#[unsafe(no_mangle)]
#[cfg_attr(
    any(
        feature = "esp32",
        feature = "esp32c2",
        feature = "esp32c3",
        feature = "esp32c6",
        feature = "esp32h2",
        feature = "esp32s2",
        feature = "esp32s3"
    ),
    esp_hal::ram
)]
fn _embassy_trace_task_ready_begin(executor_id: u32, task_id: u32) {
    let executor_id = EXECUTOR_REGISTRY.lookup_or_register(executor_id).unwrap();
    // let start_time = crate::timing::get_tracing_raw_ticks();

    write_embassy_task_ready(compressed_task_id(task_id), executor_id);

    // let end_time = crate::timing::get_tracing_raw_ticks();
    // defmt::info!(
    //     "Task ready begin: task_id={}, executor_id={}, time_taken={}us",
    //     task_id,
    //     executor_id.as_u8(),
    //     (end_time - start_time)
    // );
}