hal8174/tracing-perfetto
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

something is working

Browse source
This commit is contained in:
hal8174 2025-02-14 23:26:51 +01:00
parent e7f75310ab
commit 45236f8bb8
2 changed files with 172 additions and 133 deletions
Download patch file Download diff file Expand all files Collapse all files

165
src/lib.rs
View file

@ -1,6 +1,83 @@
use std::{
sync::{atomic::AtomicU64, Mutex},
time::Instant,
};
mod perfetto {
include!(concat!(env!("OUT_DIR"), "/perfetto.trace.rs"));
}
use prost::Message;
use tracing::{span, Subscriber};
struct PerfettoSubscriber {}
pub struct PerfettoSubscriber {
id_counter: AtomicU64,
start_time: Instant,
writer: Mutex<Vec<perfetto::TracePacket>>,
}
impl PerfettoSubscriber {
pub fn new() -> Self {
let mut packets = Vec::new();
packets.push(perfetto::TracePacket {
timestamp: None,
data: Some(perfetto::trace_packet::Data::TrackDescriptor(
perfetto::TrackDescriptor {
uuid: Some(894893984),
parent_uuid: None,
static_or_dynamic_name: None,
process: Some(perfetto::ProcessDescriptor {
pid: Some(1234),
process_name: Some(String::from("test")),
}),
thread: None,
},
)),
optional_trusted_packet_sequence_id: Some(
perfetto::trace_packet::OptionalTrustedPacketSequenceId::TrustedPacketSequenceId(
3903809,
),
),
});
packets.push(perfetto::TracePacket {
timestamp: None,
data: Some(perfetto::trace_packet::Data::TrackDescriptor(
perfetto::TrackDescriptor {
uuid: Some(49083589894),
parent_uuid: None,
static_or_dynamic_name: None,
process: None,
thread: Some(perfetto::ThreadDescriptor {
pid: Some(1234),
thread_name: Some(String::from("test_thread")),
tid: Some(10),
}),
},
)),
optional_trusted_packet_sequence_id: Some(
perfetto::trace_packet::OptionalTrustedPacketSequenceId::TrustedPacketSequenceId(
3903809,
),
),
});
Self {
id_counter: AtomicU64::new(1),
writer: Mutex::new(packets),
start_time: Instant::now(),
}
}
fn flush(&self) {
let mut data = self.writer.lock().unwrap();
let trace = perfetto::Trace {
packet: data.to_vec(),
};
let buf = trace.encode_to_vec();
let _ = std::fs::write("out.trace", buf);
}
}
impl Subscriber for PerfettoSubscriber {
fn enabled(&self, metadata: &tracing::Metadata<'_>) -> bool {
@ -8,26 +85,100 @@ impl Subscriber for PerfettoSubscriber {
}
fn new_span(&self, span: &span::Attributes<'_>) -> span::Id {
todo!()
println!("new_span {:?}", span);
let id = self
.id_counter
.fetch_add(1, std::sync::atomic::Ordering::Relaxed);
span::Id::from_u64(id)
}
fn record(&self, span: &span::Id, values: &span::Record<'_>) {
todo!()
println!("record {:?} {:?}", span, values);
}
fn record_follows_from(&self, span: &span::Id, follows: &span::Id) {
todo!()
println!("record_follows_from {:?} {:?}", span, follows);
}
fn event(&self, event: &tracing::Event<'_>) {
todo!()
println!("event {:?}", event);
{
let time = self.start_time.elapsed().as_nanos() as u64;
let mut vec = self.writer.lock().unwrap();
vec.push(perfetto::TracePacket {
timestamp: Some(time),
data: Some(perfetto::trace_packet::Data::TrackEvent(
perfetto::TrackEvent {
track_uuid: Some(49083589894),
categories: vec![],
category_iids: vec![],
name_field: Some(perfetto::track_event::NameField::Name(String::from("Test"))),
r#type: Some(perfetto::track_event::Type::Instant as i32),
},
)),
optional_trusted_packet_sequence_id: Some(
perfetto::trace_packet::OptionalTrustedPacketSequenceId::TrustedPacketSequenceId(
3903809,
),
),
});
}
self.flush();
}
fn enter(&self, span: &span::Id) {
todo!()
println!("enter {:?}", span);
{
let time = self.start_time.elapsed().as_nanos() as u64;
let mut vec = self.writer.lock().unwrap();
vec.push(perfetto::TracePacket {
timestamp: Some(time),
data: Some(perfetto::trace_packet::Data::TrackEvent(
perfetto::TrackEvent {
track_uuid: Some(49083589894),
categories: vec![],
category_iids: vec![],
name_field: Some(perfetto::track_event::NameField::Name(String::from("Test"))),
r#type: Some(perfetto::track_event::Type::SliceBegin as i32),
},
)),
optional_trusted_packet_sequence_id: Some(
perfetto::trace_packet::OptionalTrustedPacketSequenceId::TrustedPacketSequenceId(
3903809,
),
),
});
}
self.flush();
}
fn exit(&self, span: &span::Id) {
todo!()
println!("exit {:?}", span);
{
let time = self.start_time.elapsed().as_nanos() as u64;
let mut vec = self.writer.lock().unwrap();
vec.push(perfetto::TracePacket {
timestamp: Some(time),
data: Some(perfetto::trace_packet::Data::TrackEvent(
perfetto::TrackEvent {
track_uuid: Some(49083589894),
categories: vec![],
category_iids: vec![],
name_field: Some(perfetto::track_event::NameField::Name(String::from("Test"))),
r#type: Some(perfetto::track_event::Type::SliceEnd as i32),
},
)),
optional_trusted_packet_sequence_id: Some(
perfetto::trace_packet::OptionalTrustedPacketSequenceId::TrustedPacketSequenceId(
3903809,
),
),
});
}
self.flush();
}
}