language-servers/rust-analyzer
1
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer.git synced 2025-10-02 06:41:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

better profilig

Browse source
This commit is contained in:
Aleksey Kladov 2019-05-22 00:00:58 +03:00
parent 366ad6f03b
commit 6fed943bd2
3 changed files with 29 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

1
Cargo.lock generated
View file

@ -1211,6 +1211,7 @@ dependencies = [
name = "ra_prof" name = "ra_prof"
version = "0.1.0" version = "0.1.0"
dependencies = [ dependencies = [
"itertools 0.8.0 (registry+https://github.com/rust-lang/crates.io-index)",
"once_cell 0.2.0 (registry+https://github.com/rust-lang/crates.io-index)", "once_cell 0.2.0 (registry+https://github.com/rust-lang/crates.io-index)",
] ]

1
crates/ra_prof/Cargo.toml
View file

@ -7,3 +7,4 @@ publish = false
[dependencies] [dependencies]
once_cell = "0.2.0" once_cell = "0.2.0"
itertools = "0.8.0"

35
crates/ra_prof/src/lib.rs
View file

@ -9,6 +9,7 @@ use std::{
}; };
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use itertools::Itertools;
/// Set profiling filter. It specifies descriptions allowed to profile. /// Set profiling filter. It specifies descriptions allowed to profile.
/// This is helpful when call stack has too many nested profiling scopes. /// This is helpful when call stack has too many nested profiling scopes.
@ -179,8 +180,9 @@ impl Drop for Profiler {
stack.messages.push(Message { level, duration, message }); stack.messages.push(Message { level, duration, message });
if level == 0 { if level == 0 {
let stdout = stderr(); let stdout = stderr();
if duration >= stack.filter_data.longer_than { let longer_than = stack.filter_data.longer_than;
print(0, &stack.messages, &mut stdout.lock()); if duration >= longer_than {
print(0, &stack.messages, &mut stdout.lock(), longer_than);
} }
stack.messages.clear(); stack.messages.clear();
} }
@ -191,20 +193,37 @@ impl Drop for Profiler {
} }
} }
fn print(lvl: usize, msgs: &[Message], out: &mut impl Write) { fn print(lvl: usize, msgs: &[Message], out: &mut impl Write, longer_than: Duration) {
let mut last = 0; let mut last = 0;
let indent = repeat(" ").take(lvl + 1).collect::<String>(); let indent = repeat(" ").take(lvl + 1).collect::<String>();
for (i, &Message { level: l, duration: dur, message: ref msg }) in msgs.iter().enumerate() { // We output hierarchy for long calls, but sum up all short calls
if l != lvl { let mut short = Vec::new();
for (i, &Message { level, duration, message: ref msg }) in msgs.iter().enumerate() {
if level != lvl {
continue; continue;
} }
if duration >= longer_than {
writeln!(out, "{} {:6}ms - {}", indent, duration.as_millis(), msg)
.expect("printing profiling info to stdout");
writeln!(out, "{} {:6}ms - {}", indent, dur.as_millis(), msg) print(lvl + 1, &msgs[last..i], out, longer_than);
.expect("printing profiling info to stdout"); } else {
short.push((msg, duration))
}
print(lvl + 1, &msgs[last..i], out);
last = i; last = i;
} }
short.sort_by_key(|(msg, _time)| *msg);
for (msg, entires) in short.iter().group_by(|(msg, _time)| msg).into_iter() {
let mut count = 0;
let mut total_duration = Duration::default();
entires.for_each(|(_msg, time)| {
count += 1;
total_duration += *time;
});
writeln!(out, "{} {:6}ms - {} ({} calls)", indent, total_duration.as_millis(), msg, count)
.expect("printing profiling info to stdout");
}
} }
#[cfg(test)] #[cfg(test)]