Mirrors/rust-analyzer
2
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer synced 2024-12-26 13:03:31 +00:00
Code Issues Projects Releases Packages Wiki Activity

internal: cleanup hprof

Browse source
This commit is contained in:
Aleksey Kladov 2021-03-29 20:54:15 +03:00
parent 5dd6b93138
commit 2381a54c2f
1 changed files with 13 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
crates/profile/src/hprof.rs
View file

@ -1,5 +1,4 @@
//! Simple hierarchical profiler
use once_cell::sync::Lazy;
use std::{
cell::RefCell,
collections::{BTreeMap, HashSet},
@ -12,6 +11,8 @@ use std::{
time::{Duration, Instant},
};
use once_cell::sync::Lazy;
use crate::tree::{Idx, Tree};
/// Filtering syntax
@ -56,12 +57,12 @@ type Label = &'static str;
/// 0ms - profile
/// 0ms - profile2
/// ```
#[inline]
pub fn span(label: Label) -> ProfileSpan {
assert!(!label.is_empty());
debug_assert!(!label.is_empty());
if PROFILING_ENABLED.load(Ordering::Relaxed)
&& PROFILE_STACK.with(|stack| stack.borrow_mut().push(label))
{
let enabled = PROFILING_ENABLED.load(Ordering::Relaxed);
if enabled && with_profile_stack(|stack| stack.push(label)) {
ProfileSpan(Some(ProfilerImpl { label, detail: None }))
} else {
ProfileSpan(None)
@ -85,14 +86,19 @@ impl ProfileSpan {
}
impl Drop for ProfilerImpl {
#[inline]
fn drop(&mut self) {
PROFILE_STACK.with(|it| it.borrow_mut().pop(self.label, self.detail.take()));
with_profile_stack(|it| it.pop(self.label, self.detail.take()));
}
}
static PROFILING_ENABLED: AtomicBool = AtomicBool::new(false);
static FILTER: Lazy<RwLock<Filter>> = Lazy::new(Default::default);
thread_local!(static PROFILE_STACK: RefCell<ProfileStack> = RefCell::new(ProfileStack::new()));
fn with_profile_stack<T>(f: impl FnOnce(&mut ProfileStack) -> T) -> T {
thread_local!(static STACK: RefCell<ProfileStack> = RefCell::new(ProfileStack::new()));
STACK.with(|it| f(&mut *it.borrow_mut()))
}
#[derive(Default, Clone, Debug)]
struct Filter {